miasm/jitter/arch/JitCore_mep.c
// Inspired from JitCore_mep.c
#include <Python.h>
#include "structmember.h"
#include <stdint.h>
#include <inttypes.h>
#include "../compat_py23.h"
#include "../queue.h"
#include "../vm_mngr.h"
#include "../bn.h"
#include "../vm_mngr_py.h"
#include "../JitCore.h"
#include "JitCore_mep.h"
reg_dict gpreg_dict[] = {
{.name = "R0", .offset = offsetof(struct vm_cpu, R0), .size = 32},
{.name = "R1", .offset = offsetof(struct vm_cpu, R1), .size = 32},
{.name = "R2", .offset = offsetof(struct vm_cpu, R2), .size = 32},
{.name = "R3", .offset = offsetof(struct vm_cpu, R3), .size = 32},
{.name = "R4", .offset = offsetof(struct vm_cpu, R4), .size = 32},
{.name = "R5", .offset = offsetof(struct vm_cpu, R5), .size = 32},
{.name = "R6", .offset = offsetof(struct vm_cpu, R6), .size = 32},
{.name = "R7", .offset = offsetof(struct vm_cpu, R7), .size = 32},
{.name = "R8", .offset = offsetof(struct vm_cpu, R8), .size = 32},
{.name = "R9", .offset = offsetof(struct vm_cpu, R9), .size = 32},
{.name = "R10", .offset = offsetof(struct vm_cpu, R10), .size = 32},
{.name = "R11", .offset = offsetof(struct vm_cpu, R11), .size = 32},
{.name = "R12", .offset = offsetof(struct vm_cpu, R12), .size = 32},
{.name = "TP", .offset = offsetof(struct vm_cpu, TP), .size = 32},
{.name = "GP", .offset = offsetof(struct vm_cpu, GP), .size = 32},
{.name = "SP", .offset = offsetof(struct vm_cpu, SP), .size = 32},
{.name = "PC", .offset = offsetof(struct vm_cpu, PC), .size = 32},
{.name = "LP", .offset = offsetof(struct vm_cpu, LP), .size = 32},
{.name = "SAR", .offset = offsetof(struct vm_cpu, SAR), .size = 32},
{.name = "S3", .offset = offsetof(struct vm_cpu, S3), .size = 32},
{.name = "RPB", .offset = offsetof(struct vm_cpu, RPB), .size = 32},
{.name = "RPE", .offset = offsetof(struct vm_cpu, RPE), .size = 32},
{.name = "RPC", .offset = offsetof(struct vm_cpu, RPC), .size = 32},
{.name = "HI", .offset = offsetof(struct vm_cpu, HI), .size = 32},
{.name = "LO", .offset = offsetof(struct vm_cpu, LO), .size = 32},
{.name = "S9", .offset = offsetof(struct vm_cpu, S9), .size = 32},
{.name = "S10", .offset = offsetof(struct vm_cpu, S10), .size = 32},
{.name = "S11", .offset = offsetof(struct vm_cpu, S11), .size = 32},
{.name = "MB0", .offset = offsetof(struct vm_cpu, MB0), .size = 32},
{.name = "ME0", .offset = offsetof(struct vm_cpu, ME0), .size = 32},
{.name = "MB1", .offset = offsetof(struct vm_cpu, MB1), .size = 32},
{.name = "ME1", .offset = offsetof(struct vm_cpu, ME1), .size = 32},
{.name = "PSW", .offset = offsetof(struct vm_cpu, PSW), .size = 32},
{.name = "ID", .offset = offsetof(struct vm_cpu, ID), .size = 32},
{.name = "TMP", .offset = offsetof(struct vm_cpu, TMP), .size = 32},
{.name = "EPC", .offset = offsetof(struct vm_cpu, EPC), .size = 32},
{.name = "EXC", .offset = offsetof(struct vm_cpu, EXC), .size = 32},
{.name = "CFG", .offset = offsetof(struct vm_cpu, CFG), .size = 32},
{.name = "S22", .offset = offsetof(struct vm_cpu, S22), .size = 32},
{.name = "NPC", .offset = offsetof(struct vm_cpu, NPC), .size = 32},
{.name = "DBG", .offset = offsetof(struct vm_cpu, DBG), .size = 32},
{.name = "DEPC", .offset = offsetof(struct vm_cpu, DEPC), .size = 32},
{.name = "OPT", .offset = offsetof(struct vm_cpu, OPT), .size = 32},
{.name = "RCFG", .offset = offsetof(struct vm_cpu, RCFG), .size = 32},
{.name = "CCFG", .offset = offsetof(struct vm_cpu, CCFG), .size = 32},
{.name = "S29", .offset = offsetof(struct vm_cpu, S29), .size = 32},
{.name = "S30", .offset = offsetof(struct vm_cpu, S30), .size = 32},
{.name = "S31", .offset = offsetof(struct vm_cpu, S31), .size = 32},
{.name = "S32", .offset = offsetof(struct vm_cpu, S32), .size = 32},
{.name = "take_jmp", .offset = offsetof(struct vm_cpu, take_jmp), .size = 32},
{.name = "last_addr", .offset = offsetof(struct vm_cpu, last_addr), .size = 32},
{.name = "is_repeat_end", .offset = offsetof(struct vm_cpu, is_repeat_end), .size = 32},
{.name = "PC_end", .offset = offsetof(struct vm_cpu, PC_end), .size = 32},
{.name = "RPE_instr_count", .offset = offsetof(struct vm_cpu, RPE_instr_count), .size = 32},
{.name = "RPC_current", .offset = offsetof(struct vm_cpu, RPC_current), .size = 32},
};
/************************** JitCpu object **************************/
PyObject* cpu_get_gpreg(JitCpu* self)
{
PyObject *dict = PyDict_New();
PyObject *o;
get_reg(R0);
get_reg(R1);
get_reg(R2);
get_reg(R3);
get_reg(R4);
get_reg(R5);
get_reg(R6);
get_reg(R7);
get_reg(R8);
get_reg(R9);
get_reg(R10);
get_reg(R11);
get_reg(R12);
get_reg(TP);
get_reg(GP);
get_reg(SP);
get_reg(PC);
get_reg(LP);
get_reg(SAR);
get_reg(S3);
get_reg(RPB);
get_reg(RPE);
get_reg(RPC);
get_reg(HI);
get_reg(LO);
get_reg(S9);
get_reg(S10);
get_reg(S11);
get_reg(MB0);
get_reg(ME0);
get_reg(MB1);
get_reg(ME1);
get_reg(PSW);
get_reg(ID);
get_reg(TMP);
get_reg(EPC);
get_reg(EXC);
get_reg(CFG);
get_reg(S22);
get_reg(NPC);
get_reg(DBG);
get_reg(DEPC);
get_reg(OPT);
get_reg(RCFG);
get_reg(CCFG);
get_reg(S29);
get_reg(S30);
get_reg(S31);
get_reg(S32);
get_reg(PC_end);
get_reg(RPE_instr_count);
get_reg(RPC_current);
return dict;
}
PyObject* cpu_set_gpreg(JitCpu* self, PyObject *args)
{
PyObject* dict;
PyObject *d_key, *d_value = NULL;
Py_ssize_t pos = 0;
const char *d_key_name;
uint32_t val;
unsigned int i, found;
if (!PyArg_ParseTuple(args, "O", &dict))
return NULL;
if(!PyDict_Check(dict))
RAISE(PyExc_TypeError, "arg must be dict");
while(PyDict_Next(dict, &pos, &d_key, &d_value)){
PyGetStr(d_key_name, d_key);
PyGetInt_uint32_t(d_value, val);
found = 0;
for (i=0; i < sizeof(gpreg_dict)/sizeof(reg_dict); i++){
if (strcmp(d_key_name, gpreg_dict[i].name))
continue;
*((uint32_t*)(((char*)(self->cpu)) + gpreg_dict[i].offset)) = val;
found = 1;
break;
}
if (found)
continue;
fprintf(stderr, "unknown key: %s\n", d_key_name);
RAISE(PyExc_ValueError, "unknown reg");
}
Py_INCREF(Py_None);
return Py_None;
}
PyObject * cpu_init_regs(JitCpu* self)
{
memset(self->cpu, 0, sizeof(struct vm_cpu));
Py_INCREF(Py_None);
return Py_None;
}
void dump_gpregs(struct vm_cpu* vmcpu)
{
printf("R0 %.4"PRIX32" ", vmcpu->R0);
printf("R1 %.4"PRIX32" ", vmcpu->R1);
printf("R2 %.4"PRIX32" ", vmcpu->R2);
printf("R3 %.4"PRIX32" ", vmcpu->R3);
printf("R4 %.4"PRIX32" ", vmcpu->R4);
printf("R5 %.4"PRIX32" ", vmcpu->R5);
printf("R6 %.4"PRIX32" ", vmcpu->R6);
printf("R7 %.4"PRIX32" ", vmcpu->R7);
printf("R8 %.4"PRIX32" ", vmcpu->R8);
printf("R9 %.4"PRIX32" ", vmcpu->R9);
printf("R10 %.4"PRIX32" ", vmcpu->R10);
printf("R11 %.4"PRIX32" ", vmcpu->R11);
printf("R12 %.4"PRIX32" ", vmcpu->R12);
printf("TP %.4"PRIX32" ", vmcpu->TP);
printf("GP %.4"PRIX32" ", vmcpu->GP);
printf("SP %.4"PRIX32" ", vmcpu->SP);
printf("\n");
}
PyObject * cpu_dump_gpregs(JitCpu* self, PyObject* args)
{
struct vm_cpu* vmcpu;
vmcpu = self->cpu;
dump_gpregs(vmcpu);
Py_INCREF(Py_None);
return Py_None;
}
PyObject * cpu_dump_gpregs_with_attrib(JitCpu* self, PyObject* args)
{
return cpu_dump_gpregs(self, args);
}
PyObject* cpu_set_exception(JitCpu* self, PyObject* args)
{
PyObject *item1;
uint32_t exception_flags;
if (!PyArg_ParseTuple(args, "O", &item1))
return NULL;
PyGetInt_uint32_t(item1, exception_flags);
((struct vm_cpu*)self->cpu)->exception_flags = exception_flags;
Py_INCREF(Py_None);
return Py_None;
}
PyObject* cpu_get_exception(JitCpu* self, PyObject* args)
{
return PyLong_FromUnsignedLongLong((uint64_t)(((struct vm_cpu*)self->cpu)->exception_flags));
}
void MEM_WRITE_08(JitCpu* jitcpu, uint64_t addr, uint8_t src)
{
vm_MEM_WRITE_08(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
}
void MEM_WRITE_16(JitCpu* jitcpu, uint64_t addr, uint16_t src)
{
vm_MEM_WRITE_16(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
}
void MEM_WRITE_32(JitCpu* jitcpu, uint64_t addr, uint32_t src)
{
vm_MEM_WRITE_32(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
}
void MEM_WRITE_64(JitCpu* jitcpu, uint64_t addr, uint64_t src)
{
vm_MEM_WRITE_64(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
}
static PyMemberDef JitCpu_members[] = {
{NULL} /* Sentinel */
};
static PyMethodDef JitCpu_methods[] = {
{"init_regs", (PyCFunction)cpu_init_regs, METH_NOARGS, "X"},
{"dump_gpregs", (PyCFunction)cpu_dump_gpregs, METH_NOARGS, "X"},
{"dump_gpregs_with_attrib", (PyCFunction)cpu_dump_gpregs_with_attrib, METH_VARARGS, "X"},
{"get_gpreg", (PyCFunction)cpu_get_gpreg, METH_NOARGS, "X"},
{"set_gpreg", (PyCFunction)cpu_set_gpreg, METH_VARARGS, "X"},
{"get_exception", (PyCFunction)cpu_get_exception, METH_VARARGS, "X"},
{"set_exception", (PyCFunction)cpu_set_exception, METH_VARARGS, "X"},
{NULL} /* Sentinel */
};
static int
JitCpu_init(JitCpu *self, PyObject *args, PyObject *kwds)
{
self->cpu = malloc(sizeof(struct vm_cpu));
if (self->cpu == NULL) {
fprintf(stderr, "cannot alloc struct vm_cpu\n");
exit(0);
}
return 0;
}
getset_reg_u32(R0);
getset_reg_u32(R1);
getset_reg_u32(R2);
getset_reg_u32(R3);
getset_reg_u32(R4);
getset_reg_u32(R5);
getset_reg_u32(R6);
getset_reg_u32(R7);
getset_reg_u32(R8);
getset_reg_u32(R9);
getset_reg_u32(R10);
getset_reg_u32(R11);
getset_reg_u32(R12);
getset_reg_u32(TP);
getset_reg_u32(GP);
getset_reg_u32(SP);
getset_reg_u32(PC);
getset_reg_u32(LP);
getset_reg_u32(SAR);
getset_reg_u32(S3);
getset_reg_u32(RPB);
getset_reg_u32(RPE);
getset_reg_u32(RPC);
getset_reg_u32(HI);
getset_reg_u32(LO);
getset_reg_u32(S9);
getset_reg_u32(S10);
getset_reg_u32(S11);
getset_reg_u32(MB0);
getset_reg_u32(ME0);
getset_reg_u32(MB1);
getset_reg_u32(ME1);
getset_reg_u32(PSW);
getset_reg_u32(ID);
getset_reg_u32(TMP);
getset_reg_u32(EPC);
getset_reg_u32(EXC);
getset_reg_u32(CFG);
getset_reg_u32(S22);
getset_reg_u32(NPC);
getset_reg_u32(DBG);
getset_reg_u32(DEPC);
getset_reg_u32(OPT);
getset_reg_u32(RCFG);
getset_reg_u32(CCFG);
getset_reg_u32(S29);
getset_reg_u32(S30);
getset_reg_u32(S31);
getset_reg_u32(S32);
getset_reg_u32(PC_end);
getset_reg_u32(RPE_instr_count);
getset_reg_u32(RPC_current);
PyObject* get_gpreg_offset_all(void)
{
PyObject *dict = PyDict_New();
PyObject *o;
get_reg_off(exception_flags);
get_reg_off(R0);
get_reg_off(R1);
get_reg_off(R2);
get_reg_off(R3);
get_reg_off(R4);
get_reg_off(R5);
get_reg_off(R6);
get_reg_off(R7);
get_reg_off(R8);
get_reg_off(R9);
get_reg_off(R10);
get_reg_off(R11);
get_reg_off(R12);
get_reg_off(TP);
get_reg_off(GP);
get_reg_off(SP);
get_reg_off(PC);
get_reg_off(LP);
get_reg_off(SAR);
get_reg_off(S3);
get_reg_off(RPB);
get_reg_off(RPE);
get_reg_off(RPC);
get_reg_off(HI);
get_reg_off(LO);
get_reg_off(S9);
get_reg_off(S10);
get_reg_off(S11);
get_reg_off(MB0);
get_reg_off(ME0);
get_reg_off(MB1);
get_reg_off(ME1);
get_reg_off(PSW);
get_reg_off(ID);
get_reg_off(TMP);
get_reg_off(EPC);
get_reg_off(EXC);
get_reg_off(CFG);
get_reg_off(S22);
get_reg_off(NPC);
get_reg_off(DBG);
get_reg_off(DEPC);
get_reg_off(OPT);
get_reg_off(RCFG);
get_reg_off(CCFG);
get_reg_off(S29);
get_reg_off(S30);
get_reg_off(S31);
get_reg_off(S32);
get_reg_off(PC_end);
get_reg_off(RPE_instr_count);
get_reg_off(RPC_current);
return dict;
}
static PyGetSetDef JitCpu_getseters[] = {
{"vmmngr",
(getter)JitCpu_get_vmmngr, (setter)JitCpu_set_vmmngr,
"vmmngr",
NULL},
{"jitter",
(getter)JitCpu_get_jitter, (setter)JitCpu_set_jitter,
"jitter",
NULL},
{"R0" , (getter)JitCpu_get_R0 , (setter)JitCpu_set_R0 , "R0" , NULL},
{"R1" , (getter)JitCpu_get_R1 , (setter)JitCpu_set_R1 , "R1" , NULL},
{"R2" , (getter)JitCpu_get_R2 , (setter)JitCpu_set_R2 , "R2" , NULL},
{"R3" , (getter)JitCpu_get_R3 , (setter)JitCpu_set_R3 , "R3" , NULL},
{"R4" , (getter)JitCpu_get_R4 , (setter)JitCpu_set_R4 , "R4" , NULL},
{"R5" , (getter)JitCpu_get_R5 , (setter)JitCpu_set_R5 , "R5" , NULL},
{"R6" , (getter)JitCpu_get_R6 , (setter)JitCpu_set_R6 , "R6" , NULL},
{"R7" , (getter)JitCpu_get_R7 , (setter)JitCpu_set_R7 , "R7" , NULL},
{"R8" , (getter)JitCpu_get_R8 , (setter)JitCpu_set_R8 , "R8" , NULL},
{"R9" , (getter)JitCpu_get_R9 , (setter)JitCpu_set_R9 , "R9" , NULL},
{"R10" , (getter)JitCpu_get_R10 , (setter)JitCpu_set_R10 , "R10" , NULL},
{"R11" , (getter)JitCpu_get_R11 , (setter)JitCpu_set_R11 , "R11" , NULL},
{"R12" , (getter)JitCpu_get_R12 , (setter)JitCpu_set_R12 , "R12" , NULL},
{"TP" , (getter)JitCpu_get_TP , (setter)JitCpu_set_TP , "TP" , NULL},
{"GP" , (getter)JitCpu_get_GP , (setter)JitCpu_set_GP , "GP" , NULL},
{"SP" , (getter)JitCpu_get_SP , (setter)JitCpu_set_SP , "SP" , NULL},
{"PC" , (getter)JitCpu_get_PC , (setter)JitCpu_set_PC , "PC" , NULL},
{"LP" , (getter)JitCpu_get_LP , (setter)JitCpu_set_LP , "LP" , NULL},
{"SAR" , (getter)JitCpu_get_SAR , (setter)JitCpu_set_SAR , "SAR" , NULL},
{"S3" , (getter)JitCpu_get_S3 , (setter)JitCpu_set_S3 , "S3" , NULL},
{"RPB" , (getter)JitCpu_get_RPB , (setter)JitCpu_set_RPB , "RPB" , NULL},
{"RPE" , (getter)JitCpu_get_RPE , (setter)JitCpu_set_RPE , "RPE" , NULL},
{"RPC" , (getter)JitCpu_get_RPC , (setter)JitCpu_set_RPC , "RPC" , NULL},
{"HI" , (getter)JitCpu_get_HI , (setter)JitCpu_set_HI , "HI" , NULL},
{"LO" , (getter)JitCpu_get_LO , (setter)JitCpu_set_LO , "LO" , NULL},
{"S9" , (getter)JitCpu_get_S9 , (setter)JitCpu_set_S9 , "S9" , NULL},
{"S10" , (getter)JitCpu_get_S10 , (setter)JitCpu_set_S10 , "S10" , NULL},
{"S11" , (getter)JitCpu_get_S11 , (setter)JitCpu_set_S11 , "S11" , NULL},
{"MB0" , (getter)JitCpu_get_MB0 , (setter)JitCpu_set_MB0 , "MB0" , NULL},
{"ME0" , (getter)JitCpu_get_ME0 , (setter)JitCpu_set_ME0 , "ME0" , NULL},
{"MB1" , (getter)JitCpu_get_MB1 , (setter)JitCpu_set_MB1 , "MB1" , NULL},
{"ME1" , (getter)JitCpu_get_ME1 , (setter)JitCpu_set_ME1 , "ME1" , NULL},
{"PSW" , (getter)JitCpu_get_PSW , (setter)JitCpu_set_PSW , "PSW" , NULL},
{"ID" , (getter)JitCpu_get_ID , (setter)JitCpu_set_ID , "ID" , NULL},
{"TMP" , (getter)JitCpu_get_TMP , (setter)JitCpu_set_TMP , "TMP" , NULL},
{"EPC" , (getter)JitCpu_get_EPC , (setter)JitCpu_set_EPC , "EPC" , NULL},
{"EXC" , (getter)JitCpu_get_EXC , (setter)JitCpu_set_EXC , "EXC" , NULL},
{"CFG" , (getter)JitCpu_get_CFG , (setter)JitCpu_set_CFG , "CFG" , NULL},
{"S22" , (getter)JitCpu_get_S22 , (setter)JitCpu_set_S22 , "S22" , NULL},
{"NPC" , (getter)JitCpu_get_NPC , (setter)JitCpu_set_NPC , "NPC" , NULL},
{"DBG" , (getter)JitCpu_get_DBG , (setter)JitCpu_set_DBG , "DBG" , NULL},
{"DEPC" , (getter)JitCpu_get_DEPC , (setter)JitCpu_set_DEPC , "DEPC" , NULL},
{"OPT" , (getter)JitCpu_get_OPT , (setter)JitCpu_set_OPT , "OPT" , NULL},
{"RCFG" , (getter)JitCpu_get_RCFG , (setter)JitCpu_set_RCFG , "RCFG" , NULL},
{"CCFG" , (getter)JitCpu_get_CCFG , (setter)JitCpu_set_CCFG , "CCFG" , NULL},
{"S29" , (getter)JitCpu_get_S29 , (setter)JitCpu_set_S29 , "S29" , NULL},
{"S30" , (getter)JitCpu_get_S30 , (setter)JitCpu_set_S30 , "S30" , NULL},
{"S31" , (getter)JitCpu_get_S31 , (setter)JitCpu_set_S31 , "S31" , NULL},
{"S32" , (getter)JitCpu_get_S32 , (setter)JitCpu_set_S32 , "S32" , NULL},
{"PC_end" , (getter)JitCpu_get_PC_end , (setter)JitCpu_set_PC_end , "PC_end" , NULL},
{"RPE_instr_count" , (getter)JitCpu_get_RPE_instr_count , (setter)JitCpu_set_RPE_instr_count , "RPE_instr_count" , NULL},
{"RPC_current" , (getter)JitCpu_get_RPC_current , (setter)JitCpu_set_RPC_current , "RPC_current" , NULL},
{NULL} /* Sentinel */
};
static PyTypeObject JitCpuType = {
PyVarObject_HEAD_INIT(NULL, 0)
"JitCore_mep.JitCpu", /*tp_name*/
sizeof(JitCpu), /*tp_basicsize*/
0, /*tp_itemsize*/
(destructor)JitCpu_dealloc,/*tp_dealloc*/
0, /*tp_print*/
0, /*tp_getattr*/
0, /*tp_setattr*/
0, /*tp_compare*/
0, /*tp_repr*/
0, /*tp_as_number*/
0, /*tp_as_sequence*/
0, /*tp_as_mapping*/
0, /*tp_hash */
0, /*tp_call*/
0, /*tp_str*/
0, /*tp_getattro*/
0, /*tp_setattro*/
0, /*tp_as_buffer*/
Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
"JitCpu objects", /* tp_doc */
0, /* tp_traverse */
0, /* tp_clear */
0, /* tp_richcompare */
0, /* tp_weaklistoffset */
0, /* tp_iter */
0, /* tp_iternext */
JitCpu_methods, /* tp_methods */
JitCpu_members, /* tp_members */
JitCpu_getseters, /* tp_getset */
0, /* tp_base */
0, /* tp_dict */
0, /* tp_descr_get */
0, /* tp_descr_set */
0, /* tp_dictoffset */
(initproc)JitCpu_init, /* tp_init */
0, /* tp_alloc */
JitCpu_new, /* tp_new */
};
static PyMethodDef JitCore_mep_Methods[] = {
/*
*/
{"get_gpreg_offset_all", (PyCFunction)get_gpreg_offset_all, METH_NOARGS},
{NULL, NULL, 0, NULL} /* Sentinel */
};
MOD_INIT(JitCore_mep)
{
PyObject *module = NULL;
MOD_DEF(module, "JitCore_mep", "JitCore_mep module", JitCore_mep_Methods);
if (module == NULL)
RET_MODULE;
if (PyType_Ready(&JitCpuType) < 0)
RET_MODULE;
Py_INCREF(&JitCpuType);
if (PyModule_AddObject(module, "JitCpu", (PyObject *)&JitCpuType) < 0)
RET_MODULE;
RET_MODULE;
}