mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
136 lines
3.7 KiB
C++
136 lines
3.7 KiB
C++
|
|
//<code(py_hexrays_hooks)>
|
|
//---------------------------------------------------------------------------
|
|
ssize_t idaapi Hexrays_Callback(void *ud, hexrays_event_t code, va_list va)
|
|
{
|
|
// hookgenHEXRAYS:safecall=Hexrays_Hooks
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
static qvector<Hexrays_Hooks*> hexrays_hooks_instances;
|
|
|
|
//-------------------------------------------------------------------------
|
|
static void hexrays_unloading__unhook_hooks(void)
|
|
{
|
|
for ( size_t i = 0, n = hexrays_hooks_instances.size(); i < n; ++i )
|
|
hexrays_hooks_instances[i]->unhook();
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
Hexrays_Hooks::Hexrays_Hooks(uint32 _flags, uint32 _hkcb_flags)
|
|
: hooks_base_t("ida_hexrays.Hexrays_Hooks", nullptr, hook_type_t(-1), _flags, _hkcb_flags),
|
|
hooked(false)
|
|
{
|
|
hexrays_hooks_instances.push_back(this);
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
Hexrays_Hooks::~Hexrays_Hooks()
|
|
{
|
|
hexrays_hooks_instances.del(this);
|
|
unhook();
|
|
}
|
|
|
|
// hookgenHEXRAYS:methodsinfo_def
|
|
|
|
//</code(py_hexrays_hooks)>
|
|
|
|
//<inline(py_hexrays_hooks)>
|
|
//-------------------------------------------------------------------------
|
|
// Hexrays hooks
|
|
//---------------------------------------------------------------------------
|
|
ssize_t idaapi Hexrays_Callback(void *ud, hexrays_event_t event, va_list va);
|
|
class control_graph_t;
|
|
|
|
// We'll inherit from hooks_base_t to benefit from some of its
|
|
// goodies, but the [un]hooking mechanism itself will be different.
|
|
struct Hexrays_Hooks : public hooks_base_t
|
|
{
|
|
// hookgenHEXRAYS:methodsinfo_decl
|
|
|
|
bool hooked;
|
|
|
|
Hexrays_Hooks(uint32 _flags=0, uint32 _hkcb_flags=HKCB_GLOBAL);
|
|
virtual ~Hexrays_Hooks();
|
|
|
|
bool hook()
|
|
{
|
|
if ( !hooked )
|
|
hooked = install_hexrays_callback(Hexrays_Callback, this);
|
|
return hooked;
|
|
}
|
|
bool unhook()
|
|
{
|
|
if ( hooked )
|
|
hooked = !remove_hexrays_callback(Hexrays_Callback, this);
|
|
return !hooked;
|
|
}
|
|
#ifdef TESTABLE_BUILD
|
|
PyObject *dump_state(bool assert_all_reimplemented=false) { return hooks_base_t::dump_state(mappings, mappings_size, assert_all_reimplemented); }
|
|
#endif
|
|
|
|
// hookgenHEXRAYS:methods
|
|
|
|
ssize_t dispatch(hexrays_event_t code, va_list va)
|
|
{
|
|
ssize_t ret = 0;
|
|
switch ( code )
|
|
{
|
|
// hookgenHEXRAYS:notifications
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
private:
|
|
static ssize_t handle_create_hint_output(
|
|
PyObject *o,
|
|
vdui_t *,
|
|
qstring *out_hint,
|
|
int *out_implines)
|
|
{
|
|
ssize_t rc = 0;
|
|
if ( o != nullptr && PySequence_Check(o) && PySequence_Size(o) == 3 )
|
|
{
|
|
newref_t py_rc(PySequence_GetItem(o, 0));
|
|
newref_t py_hint(PySequence_GetItem(o, 1));
|
|
newref_t py_implines(PySequence_GetItem(o, 2));
|
|
qstring plugin_hint;
|
|
if ( PyLong_Check(py_rc.o)
|
|
&& PyUnicode_Check(py_hint.o)
|
|
&& PyLong_Check(py_implines.o)
|
|
&& PyUnicode_as_qstring(&plugin_hint, py_hint.o) )
|
|
{
|
|
if ( !out_hint->empty()
|
|
&& out_hint->last() != '\n'
|
|
&& !plugin_hint.empty() )
|
|
{
|
|
out_hint->append('\n');
|
|
}
|
|
out_hint->append(plugin_hint);
|
|
rc = PyLong_AsLong(py_rc.o);
|
|
if ( rc == 2 )
|
|
rc = 0;
|
|
*out_implines += PyLong_AsLong(py_implines.o);
|
|
}
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
static ssize_t handle_build_callinfo_output(
|
|
PyObject *o,
|
|
mblock_t *,
|
|
tinfo_t *,
|
|
mcallinfo_t **out_callinfo)
|
|
{
|
|
if ( o == nullptr || o == Py_None )
|
|
return 0;
|
|
mcallinfo_t *mi = nullptr;
|
|
const int cvt = SWIG_ConvertPtr(o, (void **) &mi, SWIGTYPE_p_mcallinfo_t, 0);
|
|
if ( !SWIG_IsOK(cvt) || mi == nullptr )
|
|
return 0;
|
|
*out_callinfo = new mcallinfo_t(*mi);
|
|
return 0;
|
|
}
|
|
};
|
|
//</inline(py_hexrays_hooks)>
|