mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-19 19:13:22 +08:00
[0.2.x] +getCurrentProcessExeName
git-svn-id: https://pykd.svn.codeplex.com/svn@83522 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\EreTIk_cp
Mikhail I. Izmestev
parent
7a6e212dd4
commit
77bd01b21f
@ -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();
|
||||
|
||||
@ -593,6 +593,10 @@
|
||||
RelativePath=".\win\dbgmod.cpp"
|
||||
>
|
||||
</File>
|
||||
<File
|
||||
RelativePath=".\win\dbgps.cpp"
|
||||
>
|
||||
</File>
|
||||
<File
|
||||
RelativePath=".\win\lastevent.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");
|
||||
|
||||
@ -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( ¤tThreadId );
|
||||
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 );
|
||||
|
||||
298
pykd/win/dbgps.cpp
Normal file
298
pykd/win/dbgps.cpp
Normal file
@ -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( ¤tThreadId );
|
||||
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
|
||||
|
||||
///////////////////////////////////////////////////////////////////////////////
|
||||
|
||||
Loading…
Reference in New Issue
Block a user