static PyObject *Triton_getMemValue(PyObject *self, PyObject *args)
{
PyObject *addr;
PyObject *readSize;
uint64 ad;
uint64 rs;
/* Extract arguments */
PyArg_ParseTuple(args, "O|O", &addr, &readSize);
if (!ap.getCurrentCtxH())
return PyErr_Format(PyExc_TypeError, "getMemValue(): Can't call getMemValue() right now. You must run the program before.");
if (!PyLong_Check(addr) && !PyInt_Check(addr))
return PyErr_Format(PyExc_TypeError, "getMemValue(): expected an address (integer) as argument");
ad = PyLong_AsLong(addr);
rs = PyLong_AsLong(readSize);
if (rs != DQWORD_SIZE && rs != QWORD_SIZE && rs != DWORD_SIZE && rs != WORD_SIZE && rs != BYTE_SIZE)
return PyErr_Format(PyExc_TypeError, "getMemValue(): The readSize argument must be: DQWORD, QWORD, DWORD, WORD or BYTE");
if (PIN_CheckReadAccess(reinterpret_cast<void*>(ad)) == false)
return PyErr_Format(PyExc_TypeError, "getMemValue(): The targeted address memory can not be read");
/* If this is a 128-bits read size, we must use uint128ToPyLongObject() */
if (rs == DQWORD_SIZE){
uint128 value = ap.getMemValue(ad, rs);
return uint128ToPyLongObject(value);
}
return Py_BuildValue("k", ap.getMemValue(ad, rs));
}
static PyObject *Triton_getSyscallReturn(PyObject *self, PyObject *std) {
uint64 ret;
if (!PyLong_Check(std) && !PyInt_Check(std))
return PyErr_Format(PyExc_TypeError, "getSyscallReturn(): expected an id (integer) as argument");
LEVEL_CORE::SYSCALL_STANDARD standard = static_cast<LEVEL_CORE::SYSCALL_STANDARD>(PyLong_AsLong(std));;
CONTEXT *ctx = static_cast<CONTEXT*>(ap.getCurrentCtxH()->getCtx());
ret = PIN_GetSyscallReturn(ctx, standard);
return PyLong_FromLong(ret);
}
static PyObject *Triton_getFlagValue(PyObject *self, PyObject *flagId) {
RegisterOperand flag;
if (!PyLong_Check(flagId) && !PyInt_Check(flagId))
return PyErr_Format(PyExc_TypeError, "getFlagValue(): expected a flag id (IDREF.FLAG) as argument");
if (!ap.getCurrentCtxH())
return PyErr_Format(PyExc_TypeError, "getFlagValue(): Can't call getFlagValue() right now. You must run the program before.");
flag = createTmpFlag(PyLong_AsLong(flagId));
return Py_BuildValue("k", ap.getFlagValue(flag));
}
static PyObject *Triton_getRegValue(PyObject *self, PyObject *reg)
{
uint64 tritonReg;
if (!PyLong_Check(reg) && !PyInt_Check(reg))
return PyErr_Format(PyExc_TypeError, "getRegValue(): expected a register id (IDREF.REG) as argument");
if (!ap.getCurrentCtxH())
return PyErr_Format(PyExc_TypeError, "getRegValue(): Can't call getRegValue() right now. You must run the program before.");
tritonReg = PyLong_AsLong(reg);
if (tritonReg >= ID_XMM0 && tritonReg <= ID_XMM15){
uint128 value = ap.getSSERegisterValue(tritonReg);
return uint128ToPyLongObject(value);
}
return Py_BuildValue("k", ap.getRegisterValue(tritonReg));
}
static PyObject *Triton_getSyscallArgument(PyObject *self, PyObject *args) {
PyObject *num;
PyObject *std;
uint64 ret;
/* Extract arguments */
PyArg_ParseTuple(args, "O|O", &std, &num);
if (!PyLong_Check(std) && !PyInt_Check(std))
return PyErr_Format(PyExc_TypeError, "getSyscallArgument(): expected an id (integer) as first argument");
if (!PyLong_Check(num) && !PyInt_Check(num))
return PyErr_Format(PyExc_TypeError, "getSyscallArgument(): expected an id (integer) as second argument");
LEVEL_CORE::SYSCALL_STANDARD standard = static_cast<LEVEL_CORE::SYSCALL_STANDARD>(PyLong_AsLong(std));;
CONTEXT *ctx = static_cast<CONTEXT*>(ap.getCurrentCtxH()->getCtx());
ret = PIN_GetSyscallArgument(ctx, standard, PyLong_AsLong(num));
return PyLong_FromLong(ret);
}
