zjgcjy/pykd
1
0
Fork 0
mirror of https://github.com/ivellioscolin/pykd.git synced 2025-04-21 12:53:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[+] added : getProcessorMode, setProcessorMode routines;

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@58076 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\kernelnet_cp 2010-11-26 13:55:09 +00:00
parent 5cff86598c
commit 40960bd9b7
6 changed files with 221 additions and 160 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pykd/dbgext.cpp
View File

@ -25,7 +25,6 @@
#include "dbgexcept.h" #include "dbgexcept.h"
#include "dbgsession.h" #include "dbgsession.h"
#include "dbgcallback.h" #include "dbgcallback.h"
#include "dbgstack.h"
#include "dbgpath.h" #include "dbgpath.h"
#include "dbginput.h" #include "dbginput.h"
#include "dbgprocess.h" #include "dbgprocess.h"
@ -136,6 +135,8 @@ BOOST_PYTHON_MODULE( pykd )
boost::python::def( "getImplicitThread", &getImplicitThread ); boost::python::def( "getImplicitThread", &getImplicitThread );
boost::python::def( "setImplicitThread", &setImplicitThread ); boost::python::def( "setImplicitThread", &setImplicitThread );
boost::python::def( "getThreadList", &getThreadList ); boost::python::def( "getThreadList", &getThreadList );
boost::python::def( "getProcessorMode", &getProcessorMode );
boost::python::def( "setProcessorMode", &setProcessorMode );
boost::python::class_<typedVarClass>( "typedVarClass" ) boost::python::class_<typedVarClass>( "typedVarClass" )
.def("getAddress", &typedVarClass::getAddress ) .def("getAddress", &typedVarClass::getAddress )
.def("sizeof", &typedVarClass::size ); .def("sizeof", &typedVarClass::size );

185
pykd/dbgprocess.cpp
View File

@ -67,3 +67,188 @@ getThreadList()
} }
///////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////
bool
setImplicitThread(
ULONG64 newThreadAddr )
{
HRESULT hres;
try {
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;
PDEBUG_STACK_FRAME frames = NULL;
try {
frames = new DEBUG_STACK_FRAME [ 1000 ];
ULONG filledFrames;
hres = dbgExt->control->GetStackTrace( 0, 0, 0, frames, 1000, &filledFrames );
if ( FAILED( hres ) )
throw DbgException( "IDebugControl::GetStackTrace failed" );
boost::python::list frameList;
for ( ULONG i = 0; i < filledFrames; ++i )
{
frameList.append( boost::python::object( dbgStackFrameClass( frames[i] ) ) );
}
if ( frames )
delete[] frames;
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" );
}
if ( frames )
delete[] frames;
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" );
}
}
/////////////////////////////////////////////////////////////////////////////////

34
pykd/dbgprocess.h
View File

@ -2,10 +2,44 @@
#include <boost/python.hpp> #include <boost/python.hpp>
#include <boost/python/object.hpp> #include <boost/python/object.hpp>
#include <dbgeng.h>
///////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////
boost::python::object boost::python::object
getThreadList(); getThreadList();
bool
setImplicitThread(
ULONG64 newThreadAddr );
ULONG64
getImplicitThread();
boost::python::object
getCurrentStack();
class dbgStackFrameClass : public DEBUG_STACK_FRAME {
public:
dbgStackFrameClass()
{
memset( static_cast<DEBUG_STACK_FRAME*>( this ), 0, sizeof(DEBUG_STACK_FRAME) );
}
dbgStackFrameClass( const DEBUG_STACK_FRAME &stackFrame )
{
memcpy( static_cast<DEBUG_STACK_FRAME*>( this ), &stackFrame, sizeof(DEBUG_STACK_FRAME) );
}
};
boost::python::object
getProcessorMode();
void
setProcessorMode(
const std::string &mode );
///////////////////////////////////////////////////////////////////////////////// /////////////////////////////////////////////////////////////////////////////////

116
pykd/dbgstack.cpp
View File

@ -1,116 +0,0 @@
#include "stdafx.h"
#include "dbgext.h"
#include "dbgstack.h"
#include "dbgexcept.h"
/////////////////////////////////////////////////////////////////////////////////
bool
setImplicitThread(
ULONG64 newThreadAddr,
PULONG64 oldThreadAddr )
{
HRESULT hres;
try {
*oldThreadAddr = getImplicitThread();
if ( *oldThreadAddr == -1 )
return false;
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;
PDEBUG_STACK_FRAME frames = NULL;
try {
frames = new DEBUG_STACK_FRAME [ 1000 ];
ULONG filledFrames;
hres = dbgExt->control->GetStackTrace( 0, 0, 0, frames, 1000, &filledFrames );
boost::python::list frameList;
for ( ULONG i = 0; i < filledFrames; ++i )
{
frameList.append( boost::python::object( dbgStackFrameClass( frames[i] ) ) );
}
if ( frames )
delete[] frames;
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" );
}
if ( frames )
delete[] frames;
return boost::python::object();
}
/////////////////////////////////////////////////////////////////////////////////

35
pykd/dbgstack.h
View File

@ -1,35 +0,0 @@
#pragma once
#include <boost/python.hpp>
#include <boost/python/object.hpp>
/////////////////////////////////////////////////////////////////////////////////
bool
setImplicitThread(
ULONG64 newThreadAddr,
PULONG64 oldThreadAddr );
ULONG64
getImplicitThread();
boost::python::object
getCurrentStack();
class dbgStackFrameClass : public DEBUG_STACK_FRAME {
public:
dbgStackFrameClass()
{
memset( static_cast<DEBUG_STACK_FRAME*>( this ), 0, sizeof(DEBUG_STACK_FRAME) );
}
dbgStackFrameClass( const DEBUG_STACK_FRAME &stackFrame )
{
memcpy( static_cast<DEBUG_STACK_FRAME*>( this ), &stackFrame, sizeof(DEBUG_STACK_FRAME) );
}
};
/////////////////////////////////////////////////////////////////////////////////

8
pykd/pykd.vcproj
View File

@ -395,10 +395,6 @@
RelativePath=".\dbgsession.cpp" RelativePath=".\dbgsession.cpp"
> >
</File> </File>
<File
RelativePath=".\dbgstack.cpp"
>
</File>
<File <File
RelativePath=".\dbgsym.cpp" RelativePath=".\dbgsym.cpp"
> >
@ -509,10 +505,6 @@
RelativePath=".\dbgsession.h" RelativePath=".\dbgsession.h"
> >
</File> </File>
<File
RelativePath=".\dbgstack.h"
>
</File>
<File <File
RelativePath=".\dbgsym.h" RelativePath=".\dbgsym.h"
> >