[0.2.x] +getCurrentProcessExeName

git-svn-id: https://pykd.svn.codeplex.com/svn@83522 9b283d60-5439-405e-af05-b73fd8c4d996

pykd/dbgengine.h

@@ -172,6 +172,7 @@ ULONG getCurrentThreadId();
 void setCurrentProcess( ULONG64 processAddr );
 void setImplicitThread( ULONG64 threadAddr );
 void getAllProcessThreads( std::vector<ULONG64> &threadsArray );
+std::string getCurrentProcessExecutableName();
 
 // Symbol path
 std::string getSymbolPath();

pykd/pykd_2008.vcproj

@@ -593,6 +593,10 @@
 				RelativePath=".\win\dbgmod.cpp"
 				>
 			</File>
+			<File
+				RelativePath=".\win\dbgps.cpp"
+				>
+			</File>
 			<File
 				RelativePath=".\win\lastevent.cpp"
 				>

pykd/python/pymod.cpp

@@ -304,8 +304,8 @@ BOOST_PYTHON_MODULE( pykd )
         "Return PID of the current process ( user mode only )" );
     python::def( "getCurrentThreadId", &getCurrentThreadId,
         "Return TID of the current thread ( user mode only )" );
-
+    python::def( "getCurrentProcessExeName", &getCurrentProcessExecutableName,
-
+        "Return name of executable file loaded in the current process");
 
     // symbol path
     python::def( "getSymbolPath", &getSymbolPath, "Returns current symbol path");

pykd/win/dbgeng.cpp

@@ -17,118 +17,6 @@ DebugEngine     g_dbgEng;
 
 ///////////////////////////////////////////////////////////////////////////////////
 
-ULONG startProcess( const std::wstring  &processName )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres;
-
-    ULONG       opt;
-    hres = g_dbgEng->control->GetEngineOptions( &opt );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::GetEngineOptions failed" );
-
-    opt |= DEBUG_ENGOPT_INITIAL_BREAK;
-    hres = g_dbgEng->control->SetEngineOptions( opt );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::SetEngineOptions failed" );
-
-    std::vector< std::wstring::value_type >      cmdLine( processName.size() + 1 );
-    wcscpy_s( &cmdLine[0], cmdLine.size(), processName.c_str() );
-
-    hres = g_dbgEng->client->CreateProcessWide( 0, &cmdLine[0], DEBUG_PROCESS | DETACHED_PROCESS );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugClient4::CreateProcessWide failed" );
-
-    hres = g_dbgEng->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::WaitForEvent failed" );
-
-    ULONG processId = -1;
-    hres = g_dbgEng->system->GetCurrentProcessId( &processId );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects::GetCurrentProcessId failed" );
-
-    return processId;
-}
-
-///////////////////////////////////////////////////////////////////////////////////
-
-ULONG attachProcess( ULONG pid )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres;
-
-    ULONG       opt;
-    hres = g_dbgEng->control->GetEngineOptions( &opt );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::GetEngineOptions failed" );
-
-    opt |= DEBUG_ENGOPT_INITIAL_BREAK;
-    hres = g_dbgEng->control->SetEngineOptions( opt );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::SetEngineOptions failed" );
-    
-    hres = g_dbgEng->client->AttachProcess( 0, pid, 0 );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugClient::AttachProcess failed" );
-
-    hres = g_dbgEng->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::WaitForEvent failed" );
-
-    ULONG processId = -1;
-    hres = g_dbgEng->system->GetCurrentProcessId( &processId );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects::GetCurrentProcessId failed" );
-
-    return processId;
-}
-
-///////////////////////////////////////////////////////////////////////////////////
-
-void detachProcess( ULONG processId )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT  hres;
-
-    if ( processId != -1 )
-    {
-        hres = g_dbgEng->system->SetCurrentProcessId(processId);
-        if ( FAILED(hres) )
-            throw DbgException( "IDebugSystemObjects::SetCurrentProcessId failed" );
-    }
-
-    hres = g_dbgEng->client->DetachCurrentProcess();
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugClient::DetachCurrentProcess failed" );
-}
-
-///////////////////////////////////////////////////////////////////////////////////
-
-void terminateProcess( ULONG processId )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres;
-
-    if ( processId != -1 )
-    {
-        hres = g_dbgEng->system->SetCurrentProcessId(processId);
-        if ( FAILED(hres) )
-            throw DbgException( "IDebugSystemObjects::SetCurrentProcessId failed" );
-    }
-
-    hres = g_dbgEng->client->TerminateCurrentProcess();
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugClient::TerminateCurrentProcess", hres );
-
-}
-
-///////////////////////////////////////////////////////////////////////////////////
-
 void loadDump( const std::wstring &fileName )
 {
     PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
@@ -1331,157 +1219,6 @@ DebugEngine::DbgEngBind* DebugEngine::operator->()
 
 ///////////////////////////////////////////////////////////////////////////////
 
-ULONG64
-getCurrentProcess()
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT         hres;  
-    ULONG64         processAddr = 0;
-    
-    hres = g_dbgEng->system->GetImplicitProcessDataOffset( &processAddr );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::GetImplicitProcessDataOffset  failed" ); 
-        
-     return processAddr; 
-}
-
-
-///////////////////////////////////////////////////////////////////////////////
-
-ULONG getCurrentProcessId()
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT      hres;
-    ULONG        pid;
-
-    hres = g_dbgEng->system->GetCurrentProcessSystemId( &pid );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::GetCurrentProcessSystemId  failed" ); 
-        
-     return pid; 
-}
-
-
-///////////////////////////////////////////////////////////////////////////////
-
-ULONG64 
-getImplicitThread()
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres; 
-    ULONG64     threadOffset = -1;
-
-    hres = g_dbgEng->system->GetImplicitThreadDataOffset( &threadOffset );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::GetImplicitThreadDataOffset  failed" ); 
-        
-    return threadOffset;
-}
-
-///////////////////////////////////////////////////////////////////////////////
-
-ULONG getCurrentThreadId()
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT      hres;
-    ULONG        tid;
-
-    hres = g_dbgEng->system->GetCurrentThreadSystemId( &tid );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::GetCurrentThreadSystemId  failed" ); 
-        
-     return tid; 
-}
-
-///////////////////////////////////////////////////////////////////////////////
-
-void setCurrentProcess( ULONG64 processAddr )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres;
-
-    processAddr = addr64NoSafe(processAddr);
-    hres = g_dbgEng->system->SetImplicitProcessDataOffset( processAddr );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::SetImplicitProcessDataOffset  failed" );
-}
-
-///////////////////////////////////////////////////////////////////////////////
-
-void setImplicitThread( ULONG64 threadAddr )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT  hres;
-
-    threadAddr = addr64NoSafe(threadAddr);
-    hres = g_dbgEng->system->SetImplicitThreadDataOffset( threadAddr );
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugSystemObjects2::SetImplicitThreadDataOffset  failed" );
-}
-
-///////////////////////////////////////////////////////////////////////////////
-
-void getAllProcessThreads( std::vector<ULONG64> &threadsArray )
-{
-    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
-
-    HRESULT     hres;
-    ULONG       debugClass, debugQualifier;
-    
-    hres = g_dbgEng->control->GetDebuggeeType( &debugClass, &debugQualifier );
-    
-    if ( FAILED( hres ) )
-        throw DbgException( "IDebugControl::GetDebuggeeType  failed" );   
-         
-    if  ( debugClass != DEBUG_CLASS_USER_WINDOWS )
-        throw DbgException( "getAllProcessThreads routine only for usermode" );
-
-    ULONG  threadsNumber = 0;
-
-    hres = g_dbgEng->system->GetNumberThreads( &threadsNumber );
-    if ( FAILED(hres) )
-        throw DbgException( "IDebugSystemObjects::GetNumberThreads failed" );
-
-    std::vector<ULONG>  ids(threadsNumber);
-
-    hres = g_dbgEng->system->GetThreadIdsByIndex( 0, threadsNumber, &ids[0], NULL );
-    if ( FAILED(hres) )
-        throw DbgException( "IDebugSystemObjects::GetThreadIdsByIndex failed" );
-
-    ULONG  currentThreadId;
-    hres = g_dbgEng->system->GetCurrentThreadId( &currentThreadId );
-    if ( FAILED(hres) )
-        throw DbgException( "IDebugSystemObjects::GetCurrentThreadId failed" );
-
-    threadsArray.resize( threadsNumber );
-    for ( size_t i = 0; i < threadsNumber; ++i )
-    {
-        hres = g_dbgEng->system->SetCurrentThreadId( ids[i] );
-        if ( FAILED(hres) )
-        {   
-            g_dbgEng->system->SetCurrentThreadId( currentThreadId );
-            throw DbgException( "IDebugSystemObjects::SetCurrentThreadId failed" );
-        }
-
-        hres = g_dbgEng->system->GetCurrentThreadTeb( &threadsArray[i] );
-        if ( FAILED(hres) )
-        {   
-            g_dbgEng->system->SetCurrentThreadId( currentThreadId );
-            throw DbgException( "IDebugSystemObjects::GetCurrentThreadTeb failed" );
-        }
-    }
-
-     g_dbgEng->system->SetCurrentThreadId( currentThreadId );
-}
-
-///////////////////////////////////////////////////////////////////////////////
-
 ULONG64 loadExtension(const std::wstring &extPath )
 {
     PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );

pykd/win/dbgps.cpp

@@ -0,0 +1,298 @@
+// 
+// Debug engine: processes and treads
+// 
+
+///////////////////////////////////////////////////////////////////////////////
+
+#include "stdafx.h"
+#include "dbgeng.h"
+
+///////////////////////////////////////////////////////////////////////////////
+
+namespace pykd {
+
+///////////////////////////////////////////////////////////////////////////////////
+
+ULONG startProcess( const std::wstring  &processName )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres;
+
+    ULONG       opt;
+    hres = g_dbgEng->control->GetEngineOptions( &opt );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::GetEngineOptions failed" );
+
+    opt |= DEBUG_ENGOPT_INITIAL_BREAK;
+    hres = g_dbgEng->control->SetEngineOptions( opt );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::SetEngineOptions failed" );
+
+    std::vector< std::wstring::value_type >      cmdLine( processName.size() + 1 );
+    wcscpy_s( &cmdLine[0], cmdLine.size(), processName.c_str() );
+
+    hres = g_dbgEng->client->CreateProcessWide( 0, &cmdLine[0], DEBUG_PROCESS | DETACHED_PROCESS );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugClient4::CreateProcessWide failed" );
+
+    hres = g_dbgEng->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::WaitForEvent failed" );
+
+    ULONG processId = -1;
+    hres = g_dbgEng->system->GetCurrentProcessId( &processId );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects::GetCurrentProcessId failed" );
+
+    return processId;
+}
+
+///////////////////////////////////////////////////////////////////////////////////
+
+ULONG attachProcess( ULONG pid )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres;
+
+    ULONG       opt;
+    hres = g_dbgEng->control->GetEngineOptions( &opt );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::GetEngineOptions failed" );
+
+    opt |= DEBUG_ENGOPT_INITIAL_BREAK;
+    hres = g_dbgEng->control->SetEngineOptions( opt );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::SetEngineOptions failed" );
+    
+    hres = g_dbgEng->client->AttachProcess( 0, pid, 0 );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugClient::AttachProcess failed" );
+
+    hres = g_dbgEng->control->WaitForEvent(DEBUG_WAIT_DEFAULT, INFINITE);
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::WaitForEvent failed" );
+
+    ULONG processId = -1;
+    hres = g_dbgEng->system->GetCurrentProcessId( &processId );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects::GetCurrentProcessId failed" );
+
+    return processId;
+}
+
+///////////////////////////////////////////////////////////////////////////////////
+
+void detachProcess( ULONG processId )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT  hres;
+
+    if ( processId != -1 )
+    {
+        hres = g_dbgEng->system->SetCurrentProcessId(processId);
+        if ( FAILED(hres) )
+            throw DbgException( "IDebugSystemObjects::SetCurrentProcessId failed" );
+    }
+
+    hres = g_dbgEng->client->DetachCurrentProcess();
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugClient::DetachCurrentProcess failed" );
+}
+
+///////////////////////////////////////////////////////////////////////////////////
+
+void terminateProcess( ULONG processId )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres;
+
+    if ( processId != -1 )
+    {
+        hres = g_dbgEng->system->SetCurrentProcessId(processId);
+        if ( FAILED(hres) )
+            throw DbgException( "IDebugSystemObjects::SetCurrentProcessId failed" );
+    }
+
+    hres = g_dbgEng->client->TerminateCurrentProcess();
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugClient::TerminateCurrentProcess", hres );
+
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+ULONG64
+getCurrentProcess()
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT         hres;  
+    ULONG64         processAddr = 0;
+
+    hres = g_dbgEng->system->GetImplicitProcessDataOffset( &processAddr );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::GetImplicitProcessDataOffset  failed" ); 
+
+     return processAddr; 
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+ULONG getCurrentProcessId()
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT      hres;
+    ULONG        pid;
+
+    hres = g_dbgEng->system->GetCurrentProcessSystemId( &pid );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::GetCurrentProcessSystemId  failed" ); 
+        
+     return pid; 
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+ULONG64 
+getImplicitThread()
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres; 
+    ULONG64     threadOffset = -1;
+
+    hres = g_dbgEng->system->GetImplicitThreadDataOffset( &threadOffset );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::GetImplicitThreadDataOffset  failed" ); 
+        
+    return threadOffset;
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+ULONG getCurrentThreadId()
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT      hres;
+    ULONG        tid;
+
+    hres = g_dbgEng->system->GetCurrentThreadSystemId( &tid );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::GetCurrentThreadSystemId  failed" ); 
+        
+     return tid; 
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+void setCurrentProcess( ULONG64 processAddr )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres;
+
+    processAddr = addr64NoSafe(processAddr);
+    hres = g_dbgEng->system->SetImplicitProcessDataOffset( processAddr );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::SetImplicitProcessDataOffset  failed" );
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+void setImplicitThread( ULONG64 threadAddr )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT  hres;
+
+    threadAddr = addr64NoSafe(threadAddr);
+    hres = g_dbgEng->system->SetImplicitThreadDataOffset( threadAddr );
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugSystemObjects2::SetImplicitThreadDataOffset  failed" );
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+void getAllProcessThreads( std::vector<ULONG64> &threadsArray )
+{
+    PyThread_StateRestore pyThreadRestore( g_dbgEng->pystate );
+
+    HRESULT     hres;
+    ULONG       debugClass, debugQualifier;
+
+    hres = g_dbgEng->control->GetDebuggeeType( &debugClass, &debugQualifier );
+
+    if ( FAILED( hres ) )
+        throw DbgException( "IDebugControl::GetDebuggeeType  failed" );   
+         
+    if  ( debugClass != DEBUG_CLASS_USER_WINDOWS )
+        throw DbgException( "getAllProcessThreads routine only for usermode" );
+
+    ULONG  threadsNumber = 0;
+
+    hres = g_dbgEng->system->GetNumberThreads( &threadsNumber );
+    if ( FAILED(hres) )
+        throw DbgException( "IDebugSystemObjects::GetNumberThreads failed" );
+
+    std::vector<ULONG>  ids(threadsNumber);
+
+    hres = g_dbgEng->system->GetThreadIdsByIndex( 0, threadsNumber, &ids[0], NULL );
+    if ( FAILED(hres) )
+        throw DbgException( "IDebugSystemObjects::GetThreadIdsByIndex failed" );
+
+    ULONG  currentThreadId;
+    hres = g_dbgEng->system->GetCurrentThreadId( &currentThreadId );
+    if ( FAILED(hres) )
+        throw DbgException( "IDebugSystemObjects::GetCurrentThreadId failed" );
+
+    threadsArray.resize( threadsNumber );
+    for ( size_t i = 0; i < threadsNumber; ++i )
+    {
+        hres = g_dbgEng->system->SetCurrentThreadId( ids[i] );
+        if ( FAILED(hres) )
+        {   
+            g_dbgEng->system->SetCurrentThreadId( currentThreadId );
+            throw DbgException( "IDebugSystemObjects::SetCurrentThreadId failed" );
+        }
+
+        hres = g_dbgEng->system->GetCurrentThreadTeb( &threadsArray[i] );
+        if ( FAILED(hres) )
+        {   
+            g_dbgEng->system->SetCurrentThreadId( currentThreadId );
+            throw DbgException( "IDebugSystemObjects::GetCurrentThreadTeb failed" );
+        }
+    }
+
+     g_dbgEng->system->SetCurrentThreadId( currentThreadId );
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+std::string getCurrentProcessExecutableName()
+{
+    boost::scoped_array< CHAR > exeName( new CHAR[MAXWORD+1] );
+    memset(&exeName[0], 0, MAXWORD+1);
+    ULONG tmp;
+    HRESULT hres = 
+        g_dbgEng->system->GetCurrentProcessExecutableName(
+            &exeName[0],
+            MAXWORD,
+            &tmp);
+    if (S_OK != hres)
+        throw DbgException("IDebugSystemObjects::GetCurrentProcessExecutableName", hres);
+
+    return std::string(&exeName[0]);
+}
+
+///////////////////////////////////////////////////////////////////////////////
+
+}   // namespace pykd
+
+///////////////////////////////////////////////////////////////////////////////
+