mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-20 19:53:22 +08:00
473 lines
12 KiB
C++
473 lines
12 KiB
C++
#include "stdafx.h"
|
|
|
|
#include "kdlib\eventhandler.h"
|
|
|
|
#include "pyeventhandler.h"
|
|
#include "dbgexcept.h"
|
|
|
|
namespace pykd {
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
EventHandler::EventHandler()
|
|
{
|
|
m_pystate = PyThreadState_Get();
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
kdlib::DebugCallbackResult EventHandler::onBreakpoint( kdlib::BREAKPOINT_ID bpId )
|
|
{
|
|
kdlib::DebugCallbackResult result = kdlib::DebugCallbackNoChange;
|
|
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
do {
|
|
|
|
python::override pythonHandler = get_override( "onBreakpoint" );
|
|
if ( !pythonHandler )
|
|
{
|
|
result = kdlib::EventHandler::onBreakpoint( bpId );
|
|
break;
|
|
}
|
|
|
|
python::object resObj = pythonHandler( bpId );
|
|
|
|
if ( resObj.is_none() )
|
|
{
|
|
result = kdlib::DebugCallbackNoChange;
|
|
break;
|
|
}
|
|
|
|
int retVal = python::extract<int>( resObj );
|
|
|
|
if ( retVal >= kdlib::DebugCallbackMax )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
result = kdlib::DebugCallbackResult(retVal);
|
|
|
|
} while( FALSE );
|
|
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
result = kdlib::DebugCallbackBreak;
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
|
|
return result;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
void EventHandler::onExecutionStatusChange( kdlib::ExecutionStatus executionStatus )
|
|
{
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
python::override pythonHandler = get_override( "onExecutionStatusChange" );
|
|
if ( pythonHandler )
|
|
{
|
|
pythonHandler( executionStatus );
|
|
}
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
kdlib::DebugCallbackResult EventHandler::onException( const kdlib::ExceptionInfo &exceptionInfo )
|
|
{
|
|
kdlib::DebugCallbackResult result = kdlib::DebugCallbackNoChange;
|
|
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
do {
|
|
|
|
python::override pythonHandler = get_override( "onException" );
|
|
if ( !pythonHandler )
|
|
{
|
|
result = kdlib::EventHandler::onException( exceptionInfo );
|
|
break;
|
|
}
|
|
|
|
python::object resObj = pythonHandler( exceptionInfo );
|
|
|
|
if ( resObj.is_none() )
|
|
{
|
|
result = kdlib::DebugCallbackNoChange;
|
|
break;
|
|
}
|
|
|
|
int retVal = python::extract<int>( resObj );
|
|
|
|
if ( retVal >= kdlib::DebugCallbackMax )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
result = kdlib::DebugCallbackResult(retVal);
|
|
|
|
} while( FALSE );
|
|
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
result = kdlib::DebugCallbackBreak;
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
|
|
return result;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
kdlib::DebugCallbackResult EventHandler::onModuleLoad( kdlib::MEMOFFSET_64 offset, const std::wstring &name )
|
|
{
|
|
kdlib::DebugCallbackResult result = kdlib::DebugCallbackNoChange;
|
|
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
do {
|
|
|
|
python::override pythonHandler = get_override( "onModuleLoad" );
|
|
if ( !pythonHandler )
|
|
{
|
|
result = kdlib::EventHandler::onModuleLoad( offset, name );
|
|
break;
|
|
}
|
|
|
|
python::object resObj = pythonHandler( offset, name );
|
|
|
|
if ( resObj.is_none() )
|
|
{
|
|
result = kdlib::DebugCallbackNoChange;
|
|
break;
|
|
}
|
|
|
|
int retVal = python::extract<int>( resObj );
|
|
|
|
if ( retVal >= kdlib::DebugCallbackMax )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
result = kdlib::DebugCallbackResult(retVal);
|
|
|
|
} while( FALSE );
|
|
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
result = kdlib::DebugCallbackBreak;
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
|
|
return result;
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
kdlib::DebugCallbackResult EventHandler::onModuleUnload( kdlib::MEMOFFSET_64 offset, const std::wstring &name )
|
|
{
|
|
kdlib::DebugCallbackResult result = kdlib::DebugCallbackNoChange;
|
|
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
do {
|
|
|
|
python::override pythonHandler = get_override( "onModuleUnload" );
|
|
if ( !pythonHandler )
|
|
{
|
|
result = kdlib::EventHandler::onModuleUnload( offset, name );
|
|
break;
|
|
}
|
|
|
|
python::object resObj = pythonHandler( offset, name );
|
|
|
|
if ( resObj.is_none() )
|
|
{
|
|
result = kdlib::DebugCallbackNoChange;
|
|
break;
|
|
}
|
|
|
|
int retVal = python::extract<int>( resObj );
|
|
|
|
if ( retVal >= kdlib::DebugCallbackMax )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
result = kdlib::DebugCallbackResult(retVal);
|
|
|
|
} while( FALSE );
|
|
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
result = kdlib::DebugCallbackBreak;
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
|
|
return result;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void EventHandler::onCurrentThreadChange(kdlib::THREAD_DEBUG_ID threadid)
|
|
{
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
python::override pythonHandler = get_override("onCurrentThreadChange");
|
|
if ( pythonHandler )
|
|
{
|
|
pythonHandler(threadid);
|
|
}
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void EventHandler::onChangeLocalScope()
|
|
{
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
python::override pythonHandler = get_override("onChangeLocalScope");
|
|
if ( pythonHandler )
|
|
{
|
|
pythonHandler();
|
|
}
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void EventHandler::onChangeBreakpoints()
|
|
{
|
|
PyEval_RestoreThread(m_pystate);
|
|
|
|
try {
|
|
|
|
python::override pythonHandler = get_override("onChangeBreakpoints");
|
|
if (pythonHandler)
|
|
{
|
|
pythonHandler();
|
|
}
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void EventHandler::onDebugOutput(const std::wstring& text)
|
|
{
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
python::override pythonHandler = get_override("onDebugOutput");
|
|
if ( pythonHandler )
|
|
{
|
|
pythonHandler(text);
|
|
}
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Breakpoint::Breakpoint(kdlib::BreakpointPtr bp)
|
|
{
|
|
m_pystate = PyThreadState_Get();
|
|
m_breakpoint = bp;
|
|
}
|
|
|
|
Breakpoint::Breakpoint(kdlib::MEMOFFSET_64 offset)
|
|
{
|
|
AutoRestorePyState pystate(&m_pystate);
|
|
|
|
m_breakpoint = kdlib::softwareBreakPointSet(offset, this);
|
|
}
|
|
|
|
Breakpoint::Breakpoint(kdlib::MEMOFFSET_64 offset, python::object &callback)
|
|
{
|
|
m_callback = callback;
|
|
|
|
AutoRestorePyState pystate(&m_pystate);
|
|
|
|
m_breakpoint = kdlib::softwareBreakPointSet(offset, this);
|
|
}
|
|
|
|
Breakpoint::Breakpoint(kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType)
|
|
{
|
|
AutoRestorePyState pystate(&m_pystate);
|
|
|
|
m_breakpoint = kdlib::hardwareBreakPointSet(offset, size, accessType, this);
|
|
}
|
|
|
|
Breakpoint::Breakpoint(kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType, python::object &callback)
|
|
{
|
|
m_callback = callback;
|
|
|
|
AutoRestorePyState pystate(&m_pystate);
|
|
|
|
m_breakpoint = kdlib::hardwareBreakPointSet(offset, size, accessType, this);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
kdlib::DebugCallbackResult Breakpoint::onHit()
|
|
{
|
|
kdlib::DebugCallbackResult result = kdlib::DebugCallbackNoChange;
|
|
|
|
PyEval_RestoreThread( m_pystate );
|
|
|
|
try {
|
|
|
|
do {
|
|
|
|
python::object resObj;
|
|
|
|
if ( m_callback )
|
|
{
|
|
resObj = m_callback();
|
|
}
|
|
else
|
|
{
|
|
python::override pythonHandler = get_override( "onHit" );
|
|
if ( !pythonHandler )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
resObj = pythonHandler();
|
|
}
|
|
|
|
if ( resObj.is_none() )
|
|
{
|
|
result = kdlib::DebugCallbackNoChange;
|
|
break;
|
|
}
|
|
|
|
int retVal = python::extract<int>( resObj );
|
|
|
|
if ( retVal >= kdlib::DebugCallbackMax )
|
|
{
|
|
result = kdlib::DebugCallbackBreak;
|
|
break;
|
|
}
|
|
|
|
result = kdlib::DebugCallbackResult(retVal);
|
|
|
|
} while( FALSE );
|
|
|
|
}
|
|
catch (const python::error_already_set &)
|
|
{
|
|
printException();
|
|
result = kdlib::DebugCallbackBreak;
|
|
}
|
|
|
|
m_pystate = PyEval_SaveThread();
|
|
|
|
return result;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Breakpoint* Breakpoint::setSoftwareBreakpoint( kdlib::MEMOFFSET_64 offset, python::object &callback )
|
|
{
|
|
Breakpoint *internalBp = new Breakpoint(offset, callback);
|
|
return new Breakpoint(internalBp->m_breakpoint);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Breakpoint* Breakpoint::setHardwareBreakpoint( kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType, python::object &callback )
|
|
{
|
|
Breakpoint *internalBp = new Breakpoint(offset, size, accessType, callback);
|
|
return new Breakpoint(internalBp->m_breakpoint);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
Breakpoint* Breakpoint::getBreakpointByIndex(unsigned long index)
|
|
{
|
|
kdlib::BreakpointPtr bp;
|
|
|
|
{
|
|
AutoRestorePyState pystate;
|
|
bp = kdlib::getBreakpointByIndex(index);
|
|
}
|
|
|
|
return new Breakpoint(bp);
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
void Breakpoint::remove()
|
|
{
|
|
AutoRestorePyState pystate;
|
|
m_breakpoint->remove();
|
|
m_breakpoint = 0;
|
|
}
|
|
|
|
/////////////////////////////////////////////////////////////////////////////////
|
|
|
|
} // end namespace pykd
|