mirror of
https://github.com/idapython/src
synced 2026-06-08 14:47:00 +00:00
Arnaud Diederen
98 lines
2.7 KiB
C++
98 lines
2.7 KiB
C++
#ifndef __PY_IDA_NALT__
|
|
#define __PY_IDA_NALT__
|
|
|
|
//<code(py_nalt)>
|
|
|
|
//-------------------------------------------------------------------------
|
|
// callback for enumerating imports
|
|
// ea: import address
|
|
// name: import name (NULL if imported by ordinal)
|
|
// ord: import ordinal (0 for imports by name)
|
|
// param: user parameter passed to enum_import_names()
|
|
// return: 1-ok, 0-stop enumeration
|
|
static int idaapi py_import_enum_cb(
|
|
ea_t ea,
|
|
const char *name,
|
|
uval_t ord,
|
|
void *param)
|
|
{
|
|
// If no name, try to get the name associated with the 'ea'. It may be coming from IDS
|
|
qstring name_buf;
|
|
if ( name == NULL && get_name(&name_buf, ea) > 0 )
|
|
name = name_buf.begin();
|
|
|
|
PYW_GIL_CHECK_LOCKED_SCOPE();
|
|
ref_t py_name;
|
|
if ( name == NULL )
|
|
py_name = borref_t(Py_None);
|
|
else
|
|
py_name = newref_t(IDAPyStr_FromUTF8(name));
|
|
|
|
newref_t py_ord(Py_BuildValue(PY_BV_UVAL, bvuval_t(ord)));
|
|
newref_t py_ea(Py_BuildValue(PY_BV_EA, bvea_t(ea)));
|
|
newref_t py_result(
|
|
PyObject_CallFunctionObjArgs(
|
|
(PyObject *)param,
|
|
py_ea.o,
|
|
py_name.o,
|
|
py_ord.o,
|
|
NULL));
|
|
return py_result != NULL && PyObject_IsTrue(py_result.o) ? 1 : 0;
|
|
}
|
|
//</code(py_nalt)>
|
|
|
|
//<inline(py_nalt)>
|
|
|
|
//-------------------------------------------------------------------------
|
|
/*
|
|
#<pydoc>
|
|
def get_import_module_name(path, fname, callback):
|
|
"""
|
|
Returns the name of an imported module given its index
|
|
@return: None or the module name
|
|
"""
|
|
pass
|
|
#</pydoc>
|
|
*/
|
|
static PyObject *py_get_import_module_name(int mod_index)
|
|
{
|
|
PYW_GIL_CHECK_LOCKED_SCOPE();
|
|
qstring qbuf;
|
|
if ( !get_import_module_name(&qbuf, mod_index) )
|
|
Py_RETURN_NONE;
|
|
|
|
return IDAPyStr_FromUTF8AndSize(qbuf.begin(), qbuf.length());
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
/*
|
|
#<pydoc>
|
|
def enum_import_names(mod_index, callback):
|
|
"""
|
|
Enumerate imports from a specific module.
|
|
Please refer to ex_imports.py example.
|
|
|
|
@param mod_index: The module index
|
|
@param callback: A callable object that will be invoked with an ea, name (could be None) and ordinal.
|
|
@return: 1-finished ok, -1 on error, otherwise callback return value (<=0)
|
|
"""
|
|
pass
|
|
#</pydoc>
|
|
*/
|
|
static int py_enum_import_names(int mod_index, PyObject *py_cb)
|
|
{
|
|
PYW_GIL_CHECK_LOCKED_SCOPE();
|
|
if ( !PyCallable_Check(py_cb) )
|
|
return -1;
|
|
return enum_import_names(mod_index, py_import_enum_cb, py_cb);
|
|
}
|
|
|
|
//-------------------------------------------------------------------------
|
|
static switch_info_t *switch_info_t__from_ptrval__(size_t ptrval)
|
|
{
|
|
return (switch_info_t *) ptrval;
|
|
}
|
|
//</inline(py_nalt)>
|
|
|
|
#endif
|