mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-20 19:53:22 +08:00
60599dcfb6
[pykd] changed: refactored callbacks engine git-svn-id: https://pykd.svn.codeplex.com/svn@63638 9b283d60-5439-405e-af05-b73fd8c4d996
418 lines
12 KiB
C++
418 lines
12 KiB
C++
#include "stdafx.h"
|
|
|
|
#include <boost/format.hpp>
|
|
#include <boost/scoped_array.hpp>
|
|
|
|
#include "dbgprocess.h"
|
|
#include "dbgext.h"
|
|
#include "dbgexcept.h"
|
|
#include "dbgtype.h"
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
boost::python::object
|
|
getThreadList()
|
|
{
|
|
HRESULT hres;
|
|
ULONG i;
|
|
ULONG oldThreadId = 0;
|
|
|
|
try {
|
|
|
|
ULONG threadCount;
|
|
hres = dbgExt->system->GetNumberThreads( &threadCount );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects::GetNumberThreads failed" );
|
|
|
|
boost::scoped_array<ULONG> ids(new ULONG[threadCount]);
|
|
hres = dbgExt->system->GetThreadIdsByIndex( 0, threadCount, ids.get(), NULL );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects::GetThreadIdsByIndex failed" );
|
|
|
|
hres = dbgExt->system->GetCurrentThreadId( &oldThreadId );
|
|
|
|
boost::python::list threadList;
|
|
|
|
for ( i = 0; i < threadCount; ++i )
|
|
{
|
|
dbgExt->system->SetCurrentThreadId( ids[i] );
|
|
|
|
ULONG64 threadOffset;
|
|
hres = dbgExt->system->GetCurrentThreadDataOffset( &threadOffset );
|
|
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects::GetCurrentThreadDataOffset failed" );
|
|
|
|
threadList.append( threadOffset );
|
|
}
|
|
|
|
return threadList;
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
if ( oldThreadId )
|
|
dbgExt->system->SetCurrentThreadId( oldThreadId );
|
|
|
|
return boost::python::list();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
bool
|
|
setImplicitThread(
|
|
ULONG64 newThreadAddr )
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
newThreadAddr = addr64(newThreadAddr);
|
|
hres = dbgExt->system2->SetImplicitThreadDataOffset( newThreadAddr );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects2::SetImplicitThreadDataOffset failed" );
|
|
|
|
return true;
|
|
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
ULONG64
|
|
getImplicitThread()
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
ULONG64 threadOffset = -1;
|
|
|
|
hres = dbgExt->system2->GetImplicitThreadDataOffset( &threadOffset );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects2::GetImplicitThreadDataOffset failed" );
|
|
|
|
return threadOffset;
|
|
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
return -1;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
boost::python::object
|
|
getCurrentStack()
|
|
{
|
|
HRESULT hres;
|
|
ULONG currentScope = 0;
|
|
|
|
try {
|
|
|
|
hres = dbgExt->symbols3->GetCurrentScopeFrameIndex( ¤tScope );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbol3::GetCurrentScopeFrameIndex failed" );
|
|
|
|
boost::scoped_array<DEBUG_STACK_FRAME> frames(new DEBUG_STACK_FRAME [ 1000 ]);
|
|
|
|
ULONG filledFrames;
|
|
hres = dbgExt->control->GetStackTrace( 0, 0, 0, frames.get(), 1000, &filledFrames );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::GetStackTrace failed" );
|
|
|
|
boost::python::list frameList;
|
|
|
|
for ( ULONG i = 0; i < filledFrames; ++i )
|
|
{
|
|
dbgStackFrameClass frame( frames[i] );
|
|
|
|
boost::python::object frameObj( frame );
|
|
|
|
hres = dbgExt->symbols->SetScope( NULL, &frames[i], NULL, sizeof(DEBUG_STACK_FRAME) );
|
|
if ( SUCCEEDED( hres ) )
|
|
frameObj.attr( "locals" ) = getLocals();
|
|
|
|
frameList.append( frameObj );
|
|
}
|
|
|
|
dbgExt->symbols3->SetScopeFrameByIndex( currentScope );
|
|
|
|
return frameList;
|
|
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
dbgExt->symbols3->SetScopeFrameByIndex( currentScope );
|
|
|
|
return boost::python::object();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
boost::python::object
|
|
getProcessorMode()
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
ULONG processorMode;
|
|
hres = dbgExt->control->GetEffectiveProcessorType( &processorMode );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::GetEffectiveProcessorType failed" );
|
|
|
|
switch( processorMode )
|
|
{
|
|
case IMAGE_FILE_MACHINE_I386:
|
|
return boost::python::str("X86");
|
|
|
|
case IMAGE_FILE_MACHINE_ARM:
|
|
return boost::python::str("ARM");
|
|
|
|
case IMAGE_FILE_MACHINE_IA64:
|
|
return boost::python::str("IA64");
|
|
|
|
case IMAGE_FILE_MACHINE_AMD64:
|
|
return boost::python::str("X64");
|
|
|
|
}
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
return boost::python::object();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void
|
|
setProcessorMode(
|
|
const std::string &mode )
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
ULONG processorMode = ~0;
|
|
|
|
if ( mode == "X86" )
|
|
processorMode = IMAGE_FILE_MACHINE_I386;
|
|
else if ( mode == "ARM" )
|
|
processorMode = IMAGE_FILE_MACHINE_ARM;
|
|
else if ( mode == "IA64" )
|
|
processorMode = IMAGE_FILE_MACHINE_IA64;
|
|
else if ( mode == "X64" )
|
|
processorMode = IMAGE_FILE_MACHINE_AMD64;
|
|
else
|
|
throw DbgException( "Unknown processor type" );
|
|
|
|
hres = dbgExt->control->SetEffectiveProcessorType( processorMode );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugControl::SetEffectiveProcessorType failed" );
|
|
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
ULONG64
|
|
getCurrentProcess()
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
ULONG64 processAddr = 0;
|
|
hres = dbgExt->system2->GetImplicitProcessDataOffset( &processAddr );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects2::GetImplicitProcessDataOffset failed" );
|
|
|
|
return processAddr;
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
VOID
|
|
setCurrentProcess(
|
|
ULONG64 processAddr )
|
|
{
|
|
HRESULT hres;
|
|
|
|
try {
|
|
|
|
processAddr = addr64(processAddr);
|
|
hres = dbgExt->system2->SetImplicitProcessDataOffset( processAddr );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSystemObjects2::SetImplicitProcessDataOffset failed" );
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
dbgStackFrameClass::dbgStackFrameClass( const DEBUG_STACK_FRAME &stackFrame )
|
|
{
|
|
memcpy( static_cast<DEBUG_STACK_FRAME*>( this ), &stackFrame, sizeof(DEBUG_STACK_FRAME) );
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
std::string
|
|
dbgStackFrameClass::print() const
|
|
{
|
|
boost::format fmt(dbgExt->control->IsPointer64Bit() == S_OK ? "%1$4d %2$16x" : "%1$4d %2$08x");
|
|
fmt % FrameNumber % ReturnOffset;
|
|
return fmt.str();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
boost::python::object
|
|
getLocals()
|
|
{
|
|
HRESULT hres;
|
|
IDebugSymbolGroup *localSymbols = NULL;
|
|
IDebugSymbolGroup2 *localSymbols2 = NULL;
|
|
|
|
try {
|
|
|
|
hres = dbgExt->symbols->GetScopeSymbolGroup( DEBUG_SCOPE_GROUP_ARGUMENTS|DEBUG_SCOPE_GROUP_LOCALS, NULL, &localSymbols );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbols::GetScopeSymbolGroup failed" );
|
|
|
|
hres = localSymbols->QueryInterface( __uuidof(IDebugSymbolGroup2), (void**) &localSymbols2 );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbols::QueryInterface failed to get IDebugSymbolGroup2" );
|
|
|
|
ULONG localNumber;
|
|
hres = localSymbols->GetNumberSymbols( &localNumber );
|
|
|
|
boost::python::dict arr;
|
|
|
|
for ( ULONG i = 0; i < localNumber; ++i )
|
|
{
|
|
char varName[0x100];
|
|
|
|
hres = localSymbols->GetSymbolName( i, varName, sizeof(varName), NULL );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbolGroup::GetSymbolName failed" );
|
|
|
|
DEBUG_SYMBOL_PARAMETERS symbolParam = {};
|
|
hres = localSymbols->GetSymbolParameters( i, 1, &symbolParam );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbolGroup::GetSymbolParameters failed" );
|
|
|
|
char typeName[0x100];
|
|
hres = dbgExt->symbols->GetTypeName( symbolParam.Module, symbolParam.TypeId, typeName, sizeof(typeName), NULL );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbols::GetTypeName failed" );
|
|
|
|
char moduleName[0x100];
|
|
hres = dbgExt->symbols2->GetModuleNameString(
|
|
DEBUG_MODNAME_MODULE,
|
|
DEBUG_ANY_ID,
|
|
symbolParam.Module,
|
|
moduleName,
|
|
sizeof(moduleName),
|
|
NULL );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbols2::GetModuleNameString failed" );
|
|
|
|
|
|
ULONG64 varOffset;
|
|
hres = localSymbols2->GetSymbolOffset( i, &varOffset );
|
|
if ( FAILED( hres ) )
|
|
throw DbgException( "IDebugSymbolGroup2::GetSymbolOffset failed" );
|
|
|
|
arr[ varName ] = loadTypedVar( moduleName, typeName, varOffset );
|
|
}
|
|
|
|
return arr;
|
|
}
|
|
catch( std::exception &e )
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd error: %s\n", e.what() );
|
|
}
|
|
catch(...)
|
|
{
|
|
dbgExt->control->Output( DEBUG_OUTPUT_ERROR, "pykd unexpected error\n" );
|
|
}
|
|
|
|
if ( localSymbols )
|
|
localSymbols->Release();
|
|
|
|
if ( localSymbols2 )
|
|
localSymbols2->Release();
|
|
|
|
return boost::python::object();
|
|
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
|