2010-06-28 14:36:40 +02:00
|
|
|
// Ignore the following symbols
|
|
|
|
%ignore WorkReg;
|
|
|
|
%ignore AbstractRegister;
|
|
|
|
%ignore rginfo;
|
|
|
|
%ignore insn_t::get_canon_mnem;
|
|
|
|
%ignore insn_t::get_canon_feature;
|
|
|
|
%ignore insn_t::is_canon_insn;
|
|
|
|
%ignore bytes_t;
|
|
|
|
%ignore IDPOPT_STR;
|
|
|
|
%ignore IDPOPT_NUM;
|
|
|
|
%ignore IDPOPT_BIT;
|
|
|
|
%ignore IDPOPT_FLT;
|
|
|
|
%ignore IDPOPT_I64;
|
|
|
|
%ignore IDPOPT_OK;
|
|
|
|
%ignore IDPOPT_BADKEY;
|
|
|
|
%ignore IDPOPT_BADTYPE;
|
|
|
|
%ignore IDPOPT_BADVALUE;
|
|
|
|
%ignore set_options_t;
|
|
|
|
%ignore read_user_config_file;
|
|
|
|
|
|
|
|
%ignore s_preline;
|
|
|
|
%ignore ca_operation_t;
|
|
|
|
%ignore _chkarg_cmd;
|
|
|
|
%ignore ENUM_SIZE;
|
|
|
|
|
|
|
|
%ignore asm_t::checkarg_dispatch;
|
|
|
|
%ignore asm_t::func_header;
|
|
|
|
%ignore asm_t::func_footer;
|
|
|
|
%ignore asm_t::get_type_name;
|
|
|
|
%ignore instruc_t;
|
|
|
|
%ignore processor_t;
|
|
|
|
%ignore ph;
|
|
|
|
%ignore IDB_Callback;
|
|
|
|
%ignore IDP_Callback;
|
2011-05-23 17:54:27 +02:00
|
|
|
%ignore _py_getreg;
|
2010-06-28 14:36:40 +02:00
|
|
|
%ignore free_processor_module;
|
|
|
|
%ignore read_config_file;
|
|
|
|
|
|
|
|
%ignore gen_idb_event;
|
|
|
|
|
|
|
|
%include "idp.hpp"
|
|
|
|
%feature("director") IDB_Hooks;
|
|
|
|
%feature("director") IDP_Hooks;
|
|
|
|
%inline %{
|
|
|
|
|
|
|
|
//<inline(py_idp)>
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def AssembleLine(ea, cs, ip, use32, line):
|
|
|
|
"""
|
2011-04-18 18:07:00 +02:00
|
|
|
Assemble an instruction to a string (display a warning if an error is found)
|
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
@param ea: linear address of instruction
|
|
|
|
@param cs: cs of instruction
|
|
|
|
@param ip: ip of instruction
|
|
|
|
@param use32: is 32bit segment
|
|
|
|
@param line: line to assemble
|
|
|
|
@return:
|
|
|
|
- None on failure
|
|
|
|
- or a string containing the assembled instruction
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
2011-04-21 16:23:46 +02:00
|
|
|
static PyObject *AssembleLine(
|
2011-05-23 17:54:27 +02:00
|
|
|
ea_t ea,
|
|
|
|
ea_t cs,
|
|
|
|
ea_t ip,
|
|
|
|
bool use32,
|
2011-04-21 16:23:46 +02:00
|
|
|
const char *line)
|
2010-06-28 14:36:40 +02:00
|
|
|
{
|
|
|
|
int inslen;
|
|
|
|
char buf[MAXSTR];
|
|
|
|
if (ph.notify != NULL &&
|
|
|
|
(inslen = ph.notify(ph.assemble, ea, cs, ip, use32, line, buf)) > 0)
|
|
|
|
{
|
|
|
|
return PyString_FromStringAndSize(buf, inslen);
|
|
|
|
}
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def assemble(ea, cs, ip, use32, line):
|
|
|
|
"""
|
|
|
|
Assemble an instruction into the database (display a warning if an error is found)
|
|
|
|
@param ea: linear address of instruction
|
|
|
|
@param cs: cs of instruction
|
|
|
|
@param ip: ip of instruction
|
|
|
|
@param use32: is 32bit segment?
|
|
|
|
@param line: line to assemble
|
|
|
|
|
|
|
|
@return: Boolean. True on success.
|
|
|
|
"""
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
2011-04-21 16:23:46 +02:00
|
|
|
bool assemble(
|
2011-05-23 17:54:27 +02:00
|
|
|
ea_t ea,
|
|
|
|
ea_t cs,
|
|
|
|
ea_t ip,
|
|
|
|
bool use32,
|
2011-04-21 16:23:46 +02:00
|
|
|
const char *line)
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
|
|
|
int inslen;
|
|
|
|
char buf[MAXSTR];
|
|
|
|
|
|
|
|
if (ph.notify != NULL)
|
|
|
|
{
|
|
|
|
inslen = ph.notify(ph.assemble, ea, cs, ip, use32, line, buf);
|
|
|
|
if (inslen > 0)
|
|
|
|
{
|
|
|
|
patch_many_bytes(ea, buf, inslen);
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
2010-07-16 14:07:49 +02:00
|
|
|
def ph_get_id():
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
2010-07-16 14:07:49 +02:00
|
|
|
Returns the 'ph.id' field
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
2010-07-16 14:07:49 +02:00
|
|
|
static size_t ph_get_id()
|
2010-06-28 14:36:40 +02:00
|
|
|
{
|
2010-07-16 14:07:49 +02:00
|
|
|
return ph.id;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
2010-07-16 14:07:49 +02:00
|
|
|
def ph_get_version():
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
2010-07-16 14:07:49 +02:00
|
|
|
Returns the 'ph.version'
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
2010-07-16 14:07:49 +02:00
|
|
|
static size_t ph_get_version()
|
2010-06-28 14:36:40 +02:00
|
|
|
{
|
2010-07-16 14:07:49 +02:00
|
|
|
return ph.version;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_flag():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.flag'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_flag()
|
|
|
|
{
|
|
|
|
return ph.flag;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_cnbits():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.cnbits'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_cnbits()
|
|
|
|
{
|
|
|
|
return ph.cnbits;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_dnbits():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.dnbits'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_dnbits()
|
|
|
|
{
|
|
|
|
return ph.dnbits;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_regFirstSreg():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.regFirstSreg'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_regFirstSreg()
|
|
|
|
{
|
|
|
|
return ph.regFirstSreg;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_regLastSreg():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.regLastSreg'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_regLastSreg()
|
|
|
|
{
|
|
|
|
return ph.regLastSreg;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_segreg_size():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.segreg_size'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_segreg_size()
|
|
|
|
{
|
|
|
|
return ph.segreg_size;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_regCodeSreg():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.regCodeSreg'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_regCodeSreg()
|
|
|
|
{
|
|
|
|
return ph.regCodeSreg;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_regDataSreg():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.regDataSreg'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_regDataSreg()
|
|
|
|
{
|
|
|
|
return ph.regDataSreg;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_high_fixup_bits():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.high_fixup_bits'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_high_fixup_bits()
|
|
|
|
{
|
|
|
|
return ph.high_fixup_bits;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_icode_return():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.icode_return'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_icode_return()
|
|
|
|
{
|
|
|
|
return ph.icode_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_instruc_start():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.instruc_start'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_instruc_start()
|
|
|
|
{
|
|
|
|
return ph.instruc_start;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_instruc_end():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.instruc_end'
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_instruc_end()
|
|
|
|
{
|
|
|
|
return ph.instruc_end;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_tbyte_size():
|
|
|
|
"""
|
|
|
|
Returns the 'ph.tbyte_size' field as defined in he processor module
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static size_t ph_get_tbyte_size()
|
|
|
|
{
|
|
|
|
return ph.tbyte_size;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_instruc():
|
|
|
|
"""
|
|
|
|
Returns a list of tuples (instruction_name, instruction_feature) containing the
|
|
|
|
instructions list as defined in he processor module
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static PyObject *ph_get_instruc()
|
|
|
|
{
|
|
|
|
Py_ssize_t i = 0;
|
|
|
|
PyObject *py_result = PyTuple_New(ph.instruc_end - ph.instruc_start);
|
|
|
|
for ( instruc_t *p = ph.instruc + ph.instruc_start, *end = ph.instruc + ph.instruc_end;
|
|
|
|
p != end;
|
|
|
|
++p )
|
|
|
|
{
|
|
|
|
PyTuple_SetItem(py_result, i++, Py_BuildValue("(sI)", p->name, p->feature));
|
|
|
|
}
|
|
|
|
return py_result;
|
|
|
|
}
|
|
|
|
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_regnames():
|
|
|
|
"""
|
|
|
|
Returns the list of register names as defined in the processor module
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static PyObject *ph_get_regnames()
|
|
|
|
{
|
|
|
|
Py_ssize_t i = 0;
|
|
|
|
PyObject *py_result = PyList_New(ph.regsNum);
|
|
|
|
for ( Py_ssize_t i=0; i<ph.regsNum; i++ )
|
|
|
|
PyList_SetItem(py_result, i, PyString_FromString(ph.regNames[i]));
|
2011-04-18 18:07:00 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
return py_result;
|
|
|
|
}
|
|
|
|
|
2011-05-23 17:54:27 +02:00
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
static const regval_t *idaapi _py_getreg(
|
|
|
|
const char *name,
|
|
|
|
const regval_t *regvals);
|
|
|
|
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
def ph_get_operand_info():
|
|
|
|
"""
|
|
|
|
Returns the operand information given an ea and operand number.
|
|
|
|
|
|
|
|
@param ea: address
|
|
|
|
@param n: operand number
|
|
|
|
|
|
|
|
@return: Returns an idd_opinfo_t as a tuple: (modified, ea, reg_ival, regidx, value_size).
|
|
|
|
Please refer to idd_opinfo_t structure in the SDK.
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
#</pydoc>
|
|
|
|
*/
|
|
|
|
static PyObject *ph_get_operand_info(
|
|
|
|
ea_t ea,
|
|
|
|
int n)
|
|
|
|
{
|
|
|
|
do
|
|
|
|
{
|
|
|
|
if ( dbg == NULL || n == - 1 )
|
|
|
|
break;
|
|
|
|
|
|
|
|
// Allocate register space
|
|
|
|
thid_t tid = get_current_thread();
|
|
|
|
regvals_t regvalues;
|
2011-06-10 16:54:03 +02:00
|
|
|
regvalues.resize(dbg->registers_size);
|
2011-05-23 17:54:27 +02:00
|
|
|
// Read registers
|
|
|
|
if ( dbg->read_registers(tid, -1, regvalues.begin()) != 1 )
|
|
|
|
break;
|
|
|
|
|
|
|
|
// Call the processor module
|
|
|
|
idd_opinfo_t opinf;
|
|
|
|
if ( ph.notify(ph.get_operand_info,
|
|
|
|
ea,
|
|
|
|
n,
|
|
|
|
tid,
|
|
|
|
_py_getreg,
|
|
|
|
regvalues.begin(),
|
|
|
|
&opinf) != 0 )
|
|
|
|
{
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
return Py_BuildValue("(i" PY_FMT64 "Kii)",
|
|
|
|
opinf.modified,
|
|
|
|
opinf.ea,
|
|
|
|
opinf.value.ival,
|
|
|
|
opinf.debregidx,
|
|
|
|
opinf.value_size);
|
|
|
|
} while (false);
|
|
|
|
Py_RETURN_NONE;
|
|
|
|
}
|
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
/*
|
|
|
|
#<pydoc>
|
|
|
|
class IDP_Hooks(object):
|
2011-04-18 18:07:00 +02:00
|
|
|
def hook(self):
|
|
|
|
"""
|
|
|
|
Creates an IDP hook
|
|
|
|
|
|
|
|
@return: Boolean true on success
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def unhook(self):
|
|
|
|
"""
|
|
|
|
Removes the IDP hook
|
|
|
|
@return: Boolean true on success
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
def custom_ana(self):
|
|
|
|
"""
|
|
|
|
Analyzes and decodes an instruction at idaapi.cmd.ea
|
|
|
|
- cmd.itype must be set >= idaapi.CUSTOM_CMD_ITYPE
|
|
|
|
- cmd.size must be set to the instruction length
|
|
|
|
|
|
|
|
@return: Boolean
|
|
|
|
- False if the instruction is not recognized
|
|
|
|
- True if the instruction was decoded. idaapi.cmd should be filled in that case.
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
def custom_out(self):
|
|
|
|
"""
|
|
|
|
Outputs the instruction defined in idaapi.cmd
|
|
|
|
|
|
|
|
@return: Boolean (whether this instruction can be outputted or not)
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
def custom_emu(self):
|
|
|
|
"""
|
|
|
|
Emulate instruction, create cross-references, plan to analyze
|
|
|
|
subsequent instructions, modify flags etc. Upon entrance to this function
|
|
|
|
all information about the instruction is in 'cmd' structure.
|
|
|
|
|
|
|
|
@return: Boolean (whether this instruction has been emulated or not)
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
def custom_outop(self, op):
|
|
|
|
"""
|
|
|
|
Notification to generate operand text.
|
|
|
|
If False was returned, then the standard operand output function will be called.
|
|
|
|
|
|
|
|
The output buffer is inited with init_output_buffer()
|
|
|
|
and this notification may use out_...() functions to form the operand text
|
|
|
|
|
|
|
|
@return: Boolean (whether the operand has been outputted or not)
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
2010-06-28 14:36:40 +02:00
|
|
|
|
|
|
|
def custom_mnem(self):
|
|
|
|
"""
|
|
|
|
Prints the mnemonic of the instruction defined in idaapi.cmd
|
|
|
|
|
|
|
|
@return:
|
|
|
|
- None: No mnemonic. IDA will use the default mnemonic value if present
|
|
|
|
- String: The desired mnemonic string
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
|
|
|
|
def is_sane_insn(self, no_crefs):
|
|
|
|
"""
|
|
|
|
is the instruction sane for the current file type?
|
|
|
|
@param no_crefs:
|
|
|
|
- 1: the instruction has no code refs to it.
|
|
|
|
ida just tries to convert unexplored bytes
|
|
|
|
to an instruction (but there is no other
|
|
|
|
reason to convert them into an instruction)
|
|
|
|
- 0: the instruction is created because
|
|
|
|
of some coderef, user request or another
|
|
|
|
weighty reason.
|
2011-04-18 18:07:00 +02:00
|
|
|
@return: 1-ok, <=0-no, the instruction isn't likely to appear in the program
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def may_be_func(self, no_crefs):
|
2010-06-28 14:36:40 +02:00
|
|
|
"""
|
2011-04-18 18:07:00 +02:00
|
|
|
Can a function start here?
|
2010-06-28 14:36:40 +02:00
|
|
|
@param state: autoanalysis phase
|
|
|
|
0: creating functions
|
|
|
|
1: creating chunks
|
2011-04-18 18:07:00 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
@return: integer (probability 0..100)
|
|
|
|
"""
|
|
|
|
pass
|
2011-04-18 18:07:00 +02:00
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def closebase(self):
|
|
|
|
"""
|
|
|
|
The database will be closed now
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def savebase(self):
|
|
|
|
"""
|
|
|
|
The database is being saved. Processor module should
|
|
|
|
"""
|
|
|
|
pass
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def rename(self, ea, new_name):
|
|
|
|
"""
|
|
|
|
The kernel is going to rename a byte.
|
|
|
|
|
|
|
|
@param ea: Address
|
|
|
|
@param new_name: The new name
|
|
|
|
|
|
|
|
@return:
|
|
|
|
- If returns value <=0, then the kernel should
|
|
|
|
not rename it. See also the 'renamed' event
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def renamed(self, ea, new_name, local_name):
|
|
|
|
"""
|
|
|
|
The kernel has renamed a byte
|
|
|
|
|
|
|
|
@param ea: Address
|
|
|
|
@param new_name: The new name
|
|
|
|
@param local_name: Is local name
|
|
|
|
|
|
|
|
@return: Ignored
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def undefine(self, ea):
|
|
|
|
"""
|
|
|
|
An item in the database (insn or data) is being deleted
|
|
|
|
@param ea: Address
|
|
|
|
@return:
|
|
|
|
- returns: >0-ok, <=0-the kernel should stop
|
|
|
|
- if the return value is positive:
|
|
|
|
bit0 - ignored
|
|
|
|
bit1 - do not delete srareas at the item end
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def make_code(self, ea, size):
|
|
|
|
"""
|
|
|
|
An instruction is being created
|
|
|
|
@param ea: Address
|
|
|
|
@param size: Instruction size
|
|
|
|
@return: 1-ok, <=0-the kernel should stop
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def make_code(self, ea, size):
|
|
|
|
"""
|
|
|
|
An instruction is being created
|
|
|
|
@param ea: Address
|
|
|
|
@param size: Instruction size
|
|
|
|
@return: 1-ok, <=0-the kernel should stop
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
def make_data(self, ea, flags, tid, len):
|
|
|
|
"""
|
|
|
|
A data item is being created
|
|
|
|
@param ea: Address
|
|
|
|
@param tid: type id
|
|
|
|
@param flags: item flags
|
|
|
|
@param len: data item size
|
|
|
|
@return: 1-ok, <=0-the kernel should stop
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def load_idasgn(self, short_sig_name):
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
FLIRT signature have been loaded for normal processing
|
2011-04-18 18:07:00 +02:00
|
|
|
(not for recognition of startup sequences)
|
|
|
|
@param short_sig_name: signature name
|
|
|
|
@return: Ignored
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def add_func(self, func):
|
|
|
|
"""
|
|
|
|
The kernel has added a function
|
|
|
|
@param func: the func_t instance
|
|
|
|
@return: Ignored
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def del_func(self, func):
|
|
|
|
"""
|
|
|
|
The kernel is about to delete a function
|
|
|
|
@param func: the func_t instance
|
|
|
|
@return: 1-ok,<=0-do not delete
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def is_call_insn(self, ea, func_name):
|
|
|
|
"""
|
|
|
|
Is the instruction a "call"?
|
|
|
|
|
|
|
|
@param ea: instruction address
|
|
|
|
@return: 1-unknown, 0-no, 2-yes
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
def is_ret_insn(self, ea, func_name):
|
|
|
|
"""
|
|
|
|
Is the instruction a "return"?
|
|
|
|
|
|
|
|
@param ea: instruction address
|
|
|
|
@param strict: - True: report only ret instructions
|
|
|
|
False: include instructions like "leave" which begins the function epilog
|
|
|
|
@return: 1-unknown, 0-no, 2-yes
|
|
|
|
"""
|
2011-04-21 16:23:46 +02:00
|
|
|
pass
|
|
|
|
|
|
|
|
|
|
|
|
def assemble(self, ea, cs, ip, use32, line):
|
|
|
|
"""
|
|
|
|
Assembles an instruction
|
|
|
|
|
|
|
|
@param ea: linear address of instruction
|
|
|
|
@param cs: cs of instruction
|
|
|
|
@param ip: ip of instruction
|
|
|
|
@param use32: is 32bit segment?
|
|
|
|
@param line: line to assemble
|
|
|
|
|
|
|
|
@return: - None to let the underlying processor module assemble the line
|
|
|
|
- or a string containing the assembled buffer
|
|
|
|
"""
|
|
|
|
pass
|
2011-04-18 18:07:00 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
#</pydoc>
|
|
|
|
*/
|
2011-04-18 18:07:00 +02:00
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
// IDP hooks
|
|
|
|
//---------------------------------------------------------------------------
|
2010-06-28 14:36:40 +02:00
|
|
|
int idaapi IDP_Callback(void *ud, int notification_code, va_list va);
|
2011-04-18 18:07:00 +02:00
|
|
|
class IDP_Hooks
|
2010-06-28 14:36:40 +02:00
|
|
|
{
|
|
|
|
public:
|
2011-04-18 18:07:00 +02:00
|
|
|
virtual ~IDP_Hooks()
|
2010-06-28 14:36:40 +02:00
|
|
|
{
|
|
|
|
}
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
bool hook()
|
|
|
|
{
|
|
|
|
return hook_to_notification_point(HT_IDP, IDP_Callback, this);
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
bool unhook()
|
|
|
|
{
|
|
|
|
return unhook_from_notification_point(HT_IDP, IDP_Callback, this);
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
virtual bool custom_ana()
|
|
|
|
{
|
|
|
|
return false;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
virtual bool custom_out()
|
|
|
|
{
|
|
|
|
return false;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
virtual bool custom_emu()
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
|
|
|
return false;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
virtual bool custom_outop(PyObject *py_op)
|
|
|
|
{
|
|
|
|
return false;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
virtual PyObject *custom_mnem()
|
|
|
|
{
|
|
|
|
return NULL;
|
2010-06-28 14:36:40 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
virtual int is_sane_insn(int no_crefs)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int may_be_func(int state)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
|
|
|
|
virtual int closebase()
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void savebase()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int rename(ea_t ea, const char *new_name)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void renamed(ea_t ea, const char *new_name, bool local_name)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int undefine(ea_t ea)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int make_code(ea_t ea, asize_t size)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int make_data(ea_t ea, flags_t flags, tid_t tid, asize_t len)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void load_idasgn(const char *short_sig_name)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual void add_func(func_t *func)
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
virtual int del_func(func_t *func)
|
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-06-10 16:54:03 +02:00
|
|
|
virtual int is_call_insn(ea_t /*ea*/)
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-06-10 16:54:03 +02:00
|
|
|
virtual int is_ret_insn(ea_t /*ea*/, bool /*strict*/)
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-04-21 16:23:46 +02:00
|
|
|
virtual PyObject *assemble(
|
2011-06-10 16:54:03 +02:00
|
|
|
ea_t /*ea*/,
|
|
|
|
ea_t /*cs*/,
|
|
|
|
ea_t /*ip*/,
|
|
|
|
bool /*use32*/,
|
|
|
|
const char * /*line*/)
|
2011-04-21 16:23:46 +02:00
|
|
|
{
|
|
|
|
return NULL;
|
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
// IDB hooks
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
int idaapi IDB_Callback(void *ud, int notification_code, va_list va);
|
|
|
|
class IDB_Hooks
|
|
|
|
{
|
|
|
|
public:
|
|
|
|
virtual ~IDB_Hooks() {};
|
|
|
|
|
|
|
|
bool hook()
|
|
|
|
{
|
|
|
|
return hook_to_notification_point(HT_IDB, IDB_Callback, this);
|
|
|
|
}
|
|
|
|
bool unhook()
|
|
|
|
{
|
|
|
|
return unhook_from_notification_point(HT_IDB, IDB_Callback, this);
|
|
|
|
}
|
|
|
|
// Hook functions to override in Python
|
|
|
|
virtual int byte_patched(ea_t ea) { return 0; };
|
|
|
|
virtual int cmt_changed(ea_t, bool repeatable_cmt) { return 0; };
|
|
|
|
virtual int ti_changed(ea_t ea, const type_t *type, const p_list *fnames) { msg("ti_changed hook not supported yet\n"); return 0; };
|
|
|
|
virtual int op_ti_changed(ea_t ea, int n, const type_t *type, const p_list *fnames) { msg("op_ti_changed hook not supported yet\n"); return 0; };
|
|
|
|
virtual int op_type_changed(ea_t ea, int n) { return 0; };
|
|
|
|
virtual int enum_created(enum_t id) { return 0; };
|
|
|
|
virtual int enum_deleted(enum_t id) { return 0; };
|
|
|
|
virtual int enum_bf_changed(enum_t id) { return 0; };
|
|
|
|
virtual int enum_renamed(enum_t id) { return 0; };
|
|
|
|
virtual int enum_cmt_changed(enum_t id) { return 0; };
|
|
|
|
virtual int enum_member_created(enum_t id, const_t cid) { return 0; };
|
|
|
|
virtual int enum_member_deleted(enum_t id, const_t cid) { return 0; };
|
|
|
|
virtual int struc_created(tid_t struc_id) { return 0; };
|
|
|
|
virtual int struc_deleted(tid_t struc_id) { return 0; };
|
|
|
|
virtual int struc_renamed(struc_t *sptr) { return 0; };
|
|
|
|
virtual int struc_expanded(struc_t *sptr) { return 0; };
|
|
|
|
virtual int struc_cmt_changed(tid_t struc_id) { return 0; };
|
|
|
|
virtual int struc_member_created(struc_t *sptr, member_t *mptr) { return 0; };
|
|
|
|
virtual int struc_member_deleted(struc_t *sptr, tid_t member_id) { return 0; };
|
|
|
|
virtual int struc_member_renamed(struc_t *sptr, member_t *mptr) { return 0; };
|
|
|
|
virtual int struc_member_changed(struc_t *sptr, member_t *mptr) { return 0; };
|
|
|
|
virtual int thunk_func_created(func_t *pfn) { return 0; };
|
|
|
|
virtual int func_tail_appended(func_t *pfn, func_t *tail) { return 0; };
|
|
|
|
virtual int func_tail_removed(func_t *pfn, ea_t tail_ea) { return 0; };
|
|
|
|
virtual int tail_owner_changed(func_t *tail, ea_t owner_func) { return 0; };
|
|
|
|
virtual int func_noret_changed(func_t *pfn) { return 0; };
|
|
|
|
virtual int segm_added(segment_t *s) { return 0; };
|
|
|
|
virtual int segm_deleted(ea_t startEA) { return 0; };
|
|
|
|
virtual int segm_start_changed(segment_t *s) { return 0; };
|
|
|
|
virtual int segm_end_changed(segment_t *s) { return 0; };
|
|
|
|
virtual int segm_moved(ea_t from, ea_t to, asize_t size) { return 0; };
|
2010-06-28 14:36:40 +02:00
|
|
|
};
|
|
|
|
|
|
|
|
//</inline(py_idp)>
|
|
|
|
%}
|
|
|
|
|
|
|
|
%{
|
2011-04-18 18:07:00 +02:00
|
|
|
//<code(py_idp)>
|
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
int idaapi IDP_Callback(void *ud, int notification_code, va_list va)
|
|
|
|
{
|
|
|
|
IDP_Hooks *proxy = (IDP_Hooks *)ud;
|
|
|
|
int ret = 0;
|
|
|
|
try
|
|
|
|
{
|
|
|
|
switch ( notification_code )
|
|
|
|
{
|
|
|
|
case processor_t::custom_ana:
|
|
|
|
ret = proxy->custom_ana() ? 1 + cmd.size : 0;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case processor_t::custom_out:
|
|
|
|
ret = proxy->custom_out() ? 2 : 0;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case processor_t::custom_emu:
|
|
|
|
ret = proxy->custom_emu() ? 2 : 0;
|
|
|
|
break;
|
|
|
|
|
|
|
|
case processor_t::custom_outop:
|
|
|
|
{
|
|
|
|
op_t *op = va_arg(va, op_t *);
|
|
|
|
PyObject *py_obj = create_idaapi_linked_class_instance(S_PY_OP_T_CLSNAME, op);
|
|
|
|
if ( py_obj == NULL )
|
|
|
|
break;
|
|
|
|
ret = proxy->custom_outop(py_obj) ? 2 : 0;
|
|
|
|
Py_XDECREF(py_obj);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::custom_mnem:
|
|
|
|
{
|
|
|
|
PyObject *py_ret = proxy->custom_mnem();
|
|
|
|
if ( py_ret != NULL && PyString_Check(py_ret) )
|
|
|
|
{
|
|
|
|
char *outbuffer = va_arg(va, char *);
|
|
|
|
size_t bufsize = va_arg(va, size_t);
|
|
|
|
|
|
|
|
qstrncpy(outbuffer, PyString_AS_STRING(py_ret), bufsize);
|
|
|
|
ret = 2;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
ret = 0;
|
|
|
|
}
|
|
|
|
Py_XDECREF(py_ret);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::is_sane_insn:
|
|
|
|
{
|
|
|
|
int no_crefs = va_arg(va, int);
|
|
|
|
ret = proxy->is_sane_insn(no_crefs);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::may_be_func:
|
|
|
|
{
|
|
|
|
int state = va_arg(va, int);
|
|
|
|
ret = proxy->may_be_func(state);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::closebase:
|
|
|
|
{
|
|
|
|
proxy->closebase();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::savebase:
|
|
|
|
{
|
|
|
|
proxy->savebase();
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::rename:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
const char *new_name = va_arg(va, const char *);
|
|
|
|
ret = proxy->rename(ea, new_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::renamed:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
const char *new_name = va_arg(va, const char *);
|
|
|
|
bool local_name = va_argi(va, bool);
|
|
|
|
proxy->renamed(ea, new_name, local_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::undefine:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
ret = proxy->undefine(ea);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::make_code:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
asize_t size = va_arg(va, asize_t);
|
|
|
|
ret = proxy->make_code(ea, size);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::make_data:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
flags_t flags = va_arg(va, flags_t);
|
|
|
|
tid_t tid = va_arg(va, tid_t);
|
|
|
|
asize_t len = va_arg(va, asize_t);
|
|
|
|
ret = proxy->make_data(ea, flags, tid, len);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::load_idasgn:
|
|
|
|
{
|
|
|
|
const char *short_sig_name = va_arg(va, const char *);
|
|
|
|
proxy->load_idasgn(short_sig_name);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::add_func:
|
|
|
|
{
|
|
|
|
func_t *func = va_arg(va, func_t *);
|
|
|
|
proxy->add_func(func);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::del_func:
|
|
|
|
{
|
|
|
|
func_t *func = va_arg(va, func_t *);
|
|
|
|
ret = proxy->del_func(func);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::is_call_insn:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
ret = proxy->is_call_insn(ea);
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
case processor_t::is_ret_insn:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
bool strict = va_argi(va, bool);
|
|
|
|
ret = proxy->is_ret_insn(ea, strict);
|
|
|
|
break;
|
|
|
|
}
|
2011-04-21 16:23:46 +02:00
|
|
|
|
|
|
|
case processor_t::assemble:
|
|
|
|
{
|
|
|
|
ea_t ea = va_arg(va, ea_t);
|
|
|
|
ea_t cs = va_arg(va, ea_t);
|
|
|
|
ea_t ip = va_arg(va, ea_t);
|
|
|
|
bool use32 = va_argi(va, bool);
|
|
|
|
const char *line = va_arg(va, const char *);
|
|
|
|
// Extract user buffer (we hardcode the MAXSTR size limit)
|
|
|
|
uchar *bin = va_arg(va, uchar *);
|
|
|
|
// Call python
|
|
|
|
PyObject *py_buffer = proxy->assemble(ea, cs, ip, use32, line);
|
|
|
|
if ( py_buffer != NULL && PyString_Check(py_buffer) )
|
|
|
|
{
|
|
|
|
char *s;
|
|
|
|
Py_ssize_t len;
|
|
|
|
if ( PyString_AsStringAndSize(py_buffer, &s, &len) != -1 )
|
|
|
|
{
|
|
|
|
if ( len > MAXSTR )
|
|
|
|
len = MAXSTR;
|
|
|
|
memcpy(bin, s, len);
|
|
|
|
ret = len;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
// ret = 0 otherwise
|
|
|
|
Py_XDECREF(py_buffer);
|
|
|
|
break;
|
|
|
|
}
|
2011-04-18 18:07:00 +02:00
|
|
|
}
|
2011-04-21 16:23:46 +02:00
|
|
|
}
|
2011-06-10 16:54:03 +02:00
|
|
|
catch (Swig::DirectorException &e)
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
2011-06-10 16:54:03 +02:00
|
|
|
msg("Exception in IDP Hook function: %s\n", e.getMessage());
|
2011-04-18 18:07:00 +02:00
|
|
|
if ( PyErr_Occurred() )
|
|
|
|
PyErr_Print();
|
|
|
|
}
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
2010-06-28 14:36:40 +02:00
|
|
|
int idaapi IDB_Callback(void *ud, int notification_code, va_list va)
|
|
|
|
{
|
|
|
|
class IDB_Hooks *proxy = (class IDB_Hooks *)ud;
|
|
|
|
ea_t ea, ea2;
|
|
|
|
bool repeatable_cmt;
|
|
|
|
/*type_t *type;*/
|
|
|
|
/* p_list *fnames; */
|
|
|
|
int n;
|
|
|
|
enum_t id;
|
|
|
|
const_t cid;
|
|
|
|
tid_t struc_id;
|
|
|
|
struc_t *sptr;
|
|
|
|
member_t *mptr;
|
|
|
|
tid_t member_id;
|
|
|
|
func_t *pfn;
|
|
|
|
func_t *tail;
|
|
|
|
segment_t *seg;
|
|
|
|
asize_t size;
|
|
|
|
|
|
|
|
try {
|
|
|
|
switch (notification_code)
|
|
|
|
{
|
|
|
|
case idb_event::byte_patched:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
return proxy->byte_patched(ea);
|
|
|
|
|
|
|
|
case idb_event::cmt_changed:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
repeatable_cmt = va_arg(va, int);
|
|
|
|
return proxy->cmt_changed(ea, repeatable_cmt);
|
|
|
|
#if 0
|
|
|
|
case idb_event::ti_changed:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
type = va_arg(va, type_t *);
|
|
|
|
fnames = va_arg(va, fnames);
|
|
|
|
return proxy->ti_changed(ea, type, fnames);
|
|
|
|
|
|
|
|
case idb_event::op_ti_changed:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
n = va_arg(va, int);
|
|
|
|
type = va_arg(va, type_t *);
|
|
|
|
fnames = va_arg(va, fnames);
|
|
|
|
return proxy->op_ti_changed(ea, n, type, fnames);
|
|
|
|
#endif
|
|
|
|
case idb_event::op_type_changed:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
n = va_arg(va, int);
|
|
|
|
return proxy->op_type_changed(ea, n);
|
|
|
|
|
|
|
|
case idb_event::enum_created:
|
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
return proxy->enum_created(id);
|
|
|
|
|
|
|
|
case idb_event::enum_deleted:
|
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
return proxy->enum_deleted(id);
|
|
|
|
|
|
|
|
case idb_event::enum_bf_changed:
|
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
return proxy->enum_bf_changed(id);
|
|
|
|
|
|
|
|
case idb_event::enum_cmt_changed:
|
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
return proxy->enum_cmt_changed(id);
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
#ifdef NO_OBSOLETE_FUNCS
|
2010-06-28 14:36:40 +02:00
|
|
|
case idb_event::enum_member_created:
|
2011-04-18 18:07:00 +02:00
|
|
|
#else
|
|
|
|
case idb_event::enum_const_created:
|
|
|
|
#endif
|
2010-06-28 14:36:40 +02:00
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
cid = va_arg(va, const_t);
|
|
|
|
return proxy->enum_member_created(id, cid);
|
|
|
|
|
2011-04-18 18:07:00 +02:00
|
|
|
#ifdef NO_OBSOLETE_FUNCS
|
2010-06-28 14:36:40 +02:00
|
|
|
case idb_event::enum_member_deleted:
|
2011-04-18 18:07:00 +02:00
|
|
|
#else
|
|
|
|
case idb_event::enum_const_deleted:
|
|
|
|
#endif
|
2010-06-28 14:36:40 +02:00
|
|
|
id = va_arg(va, enum_t);
|
|
|
|
cid = va_arg(va, const_t);
|
|
|
|
return proxy->enum_member_deleted(id, cid);
|
|
|
|
|
|
|
|
case idb_event::struc_created:
|
|
|
|
struc_id = va_arg(va, tid_t);
|
|
|
|
return proxy->struc_created(struc_id);
|
|
|
|
|
|
|
|
case idb_event::struc_deleted:
|
|
|
|
struc_id = va_arg(va, tid_t);
|
|
|
|
return proxy->struc_deleted(struc_id);
|
|
|
|
|
|
|
|
case idb_event::struc_renamed:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
return proxy->struc_renamed(sptr);
|
|
|
|
|
|
|
|
case idb_event::struc_expanded:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
return proxy->struc_expanded(sptr);
|
|
|
|
|
|
|
|
case idb_event::struc_cmt_changed:
|
|
|
|
struc_id = va_arg(va, tid_t);
|
|
|
|
return proxy->struc_cmt_changed(struc_id);
|
|
|
|
|
|
|
|
case idb_event::struc_member_created:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
mptr = va_arg(va, member_t *);
|
|
|
|
return proxy->struc_member_created(sptr, mptr);
|
|
|
|
|
|
|
|
case idb_event::struc_member_deleted:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
member_id = va_arg(va, tid_t);
|
|
|
|
return proxy->struc_member_deleted(sptr, member_id);
|
|
|
|
|
|
|
|
case idb_event::struc_member_renamed:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
mptr = va_arg(va, member_t *);
|
|
|
|
return proxy->struc_member_renamed(sptr, mptr);
|
|
|
|
|
|
|
|
case idb_event::struc_member_changed:
|
|
|
|
sptr = va_arg(va, struc_t *);
|
|
|
|
mptr = va_arg(va, member_t *);
|
|
|
|
return proxy->struc_member_changed(sptr, mptr);
|
|
|
|
|
|
|
|
case idb_event::thunk_func_created:
|
|
|
|
pfn = va_arg(va, func_t *);
|
|
|
|
return proxy->thunk_func_created(pfn);
|
|
|
|
|
|
|
|
case idb_event::func_tail_appended:
|
|
|
|
pfn = va_arg(va, func_t *);
|
|
|
|
tail = va_arg(va, func_t *);
|
|
|
|
return proxy->func_tail_appended(pfn, tail);
|
|
|
|
|
|
|
|
case idb_event::func_tail_removed:
|
|
|
|
pfn = va_arg(va, func_t *);
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
return proxy->func_tail_removed(pfn, ea);
|
|
|
|
|
|
|
|
case idb_event::tail_owner_changed:
|
|
|
|
tail = va_arg(va, func_t *);
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
return proxy->tail_owner_changed(tail, ea);
|
|
|
|
|
|
|
|
case idb_event::func_noret_changed:
|
|
|
|
pfn = va_arg(va, func_t *);
|
|
|
|
return proxy->func_noret_changed(pfn);
|
|
|
|
|
|
|
|
case idb_event::segm_added:
|
|
|
|
seg = va_arg(va, segment_t *);
|
|
|
|
return proxy->segm_added(seg);
|
|
|
|
|
|
|
|
case idb_event::segm_deleted:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
return proxy->segm_deleted(ea);
|
|
|
|
|
|
|
|
case idb_event::segm_start_changed:
|
|
|
|
seg = va_arg(va, segment_t *);
|
|
|
|
return proxy->segm_start_changed(seg);
|
|
|
|
|
|
|
|
case idb_event::segm_end_changed:
|
|
|
|
seg = va_arg(va, segment_t *);
|
|
|
|
return proxy->segm_end_changed(seg);
|
|
|
|
|
|
|
|
case idb_event::segm_moved:
|
|
|
|
ea = va_arg(va, ea_t);
|
|
|
|
ea2 = va_arg(va, ea_t);
|
|
|
|
size = va_arg(va, asize_t);
|
|
|
|
return proxy->segm_moved(ea, ea2, size);
|
|
|
|
}
|
|
|
|
}
|
2011-06-10 16:54:03 +02:00
|
|
|
catch (Swig::DirectorException &e)
|
2011-04-18 18:07:00 +02:00
|
|
|
{
|
2011-06-10 16:54:03 +02:00
|
|
|
msg("Exception in IDB Hook function: %s\n", e.getMessage());
|
2010-06-28 14:36:40 +02:00
|
|
|
if (PyErr_Occurred())
|
|
|
|
{
|
|
|
|
PyErr_Print();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2011-05-23 17:54:27 +02:00
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
static const regval_t *idaapi _py_getreg(
|
|
|
|
const char *name,
|
|
|
|
const regval_t *regvals)
|
|
|
|
{
|
|
|
|
for ( int i=dbg->registers_size - 1; i >= 0; i-- )
|
|
|
|
{
|
|
|
|
if ( stricmp(name, dbg->registers[i].name) == 0 )
|
|
|
|
return ®vals[i];
|
|
|
|
}
|
|
|
|
static regval_t rv;
|
|
|
|
return &rv;
|
|
|
|
}
|
|
|
|
|
2010-06-28 14:36:40 +02:00
|
|
|
//-------------------------------------------------------------------------
|
|
|
|
//</code(py_idp)>
|
|
|
|
%}
|