pykd/pykd/pyeventhandler.h
2017-11-03 15:24:08 +04:00

123 lines
3.4 KiB
C++

#pragma once
#include <map>
#include <boost/thread/recursive_mutex.hpp>
#include "kdlib/dbgengine.h"
#include "kdlib/eventhandler.h"
#include "kdlib/breakpoint.h"
#include "boost/shared_ptr.hpp"
#include "boost/noncopyable.hpp"
#include "boost/python/object.hpp"
#include "boost/python/wrapper.hpp"
#include "pythreadstate.h"
namespace python = boost::python;
namespace pykd {
///////////////////////////////////////////////////////////////////////////////
class EventHandler : public python::wrapper<kdlib::EventHandler>, public kdlib::EventHandler
{
public:
EventHandler();
virtual kdlib::DebugCallbackResult onBreakpoint( kdlib::BREAKPOINT_ID bpId );
virtual kdlib::DebugCallbackResult onException( const kdlib::ExceptionInfo &exceptionInfo );
virtual kdlib::DebugCallbackResult onModuleLoad( kdlib::MEMOFFSET_64 offset, const std::wstring &name );
virtual kdlib::DebugCallbackResult onModuleUnload( kdlib::MEMOFFSET_64 offset, const std::wstring &name );
virtual kdlib::DebugCallbackResult onThreadStart();
virtual kdlib::DebugCallbackResult onThreadStop();
virtual void onExecutionStatusChange(kdlib::ExecutionStatus executionStatus);
virtual void onCurrentThreadChange(kdlib::THREAD_DEBUG_ID threadid);
virtual void onChangeLocalScope();
virtual void onChangeSymbolPaths();
virtual void onChangeBreakpoints();
virtual void onDebugOutput(const std::wstring& text);
virtual void onStartInput();
virtual void onStopInput();
private:
PyThreadState* m_pystate;
};
/////////////////////////////////////////////////////////////////////////////////
class Breakpoint : public python::wrapper<kdlib::BreakpointCallback>, public kdlib::BreakpointCallback
{
public:
static Breakpoint* setSoftwareBreakpoint( kdlib::MEMOFFSET_64 offset, python::object &callback = python::object() );
static Breakpoint* setHardwareBreakpoint( kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType, python::object &callback= python::object() );
static unsigned long getNumberBreakpoints();
static Breakpoint* getBreakpointByIndex(unsigned long index);
static void removeBreakpointByIndex(unsigned long index);
static void removeAllBreakpoints();
public:
explicit Breakpoint(kdlib::BreakpointPtr bp, bool weakbp = true);
explicit Breakpoint(kdlib::MEMOFFSET_64 offset);
Breakpoint(kdlib::MEMOFFSET_64 offset, python::object &callback);
Breakpoint(kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType);
Breakpoint(kdlib::MEMOFFSET_64 offset, size_t size, kdlib::ACCESS_TYPE accessType, python::object &callback);
~Breakpoint();
virtual kdlib::DebugCallbackResult onHit();
virtual void onRemove()
{
m_breakpoint = 0;
}
kdlib::BREAKPOINT_ID getId() const
{
AutoRestorePyState pystate;
return m_breakpoint->getId();
}
kdlib::MEMOFFSET_64 getOffset() const
{
AutoRestorePyState pystate;
return m_breakpoint->getOffset();
}
void remove();
private:
PyThreadState* m_pystate;
kdlib::BreakpointPtr m_breakpoint;
python::object m_callback;
bool m_weakBp;
};
///////////////////////////////////////////////////////////////////////////////
} // end namespace pykd