JitCore_aarch64.c File Reference

#include <Python.h>
#include "../JitCore.h"
#include "structmember.h"
#include <stdint.h>
#include <inttypes.h>
#include "../queue.h"
#include "../vm_mngr.h"
#include "../vm_mngr_py.h"
#include "JitCore_aarch64.h"

Include dependency graph for JitCore_aarch64.c:

Go to the source code of this file.

Functions
PyObject *	cpu_get_gpreg (JitCpu *self)
PyObject *	cpu_set_gpreg (JitCpu *self, PyObject *args)
PyObject *	cpu_init_regs (JitCpu *self)
void	dump_gpregs (vm_cpu_t *vmcpu)
PyObject *	cpu_dump_gpregs (JitCpu *self, PyObject *args)
PyObject *	cpu_set_exception (JitCpu *self, PyObject *args)
PyObject *	cpu_get_exception (JitCpu *self, PyObject *args)
void	check_automod (JitCpu *jitcpu, uint64_t addr, uint64_t size)
void	MEM_WRITE_08 (JitCpu *jitcpu, uint64_t addr, uint8_t src)
void	MEM_WRITE_16 (JitCpu *jitcpu, uint64_t addr, uint16_t src)
void	MEM_WRITE_32 (JitCpu *jitcpu, uint64_t addr, uint32_t src)
void	MEM_WRITE_64 (JitCpu *jitcpu, uint64_t addr, uint64_t src)
PyObject *	vm_set_mem (JitCpu *self, PyObject *args)
static int	JitCpu_init (JitCpu *self, PyObject *args, PyObject *kwds)
	getset_reg_u64 (X0)
	getset_reg_u64 (X1)
	getset_reg_u64 (X2)
	getset_reg_u64 (X3)
	getset_reg_u64 (X4)
	getset_reg_u64 (X5)
	getset_reg_u64 (X6)
	getset_reg_u64 (X7)
	getset_reg_u64 (X8)
	getset_reg_u64 (X9)
	getset_reg_u64 (X10)
	getset_reg_u64 (X11)
	getset_reg_u64 (X12)
	getset_reg_u64 (X13)
	getset_reg_u64 (X14)
	getset_reg_u64 (X15)
	getset_reg_u64 (X16)
	getset_reg_u64 (X17)
	getset_reg_u64 (X18)
	getset_reg_u64 (X19)
	getset_reg_u64 (X20)
	getset_reg_u64 (X21)
	getset_reg_u64 (X22)
	getset_reg_u64 (X23)
	getset_reg_u64 (X24)
	getset_reg_u64 (X25)
	getset_reg_u64 (X26)
	getset_reg_u64 (X27)
	getset_reg_u64 (X28)
	getset_reg_u64 (X29)
	getset_reg_u64 (LR)
	getset_reg_u64 (SP)
	getset_reg_u64 (PC)
	getset_reg_u32 (zf)
	getset_reg_u32 (nf)
	getset_reg_u32 (of)
	getset_reg_u32 (cf)
PyObject *	get_gpreg_offset_all (void)
PyMODINIT_FUNC	initJitCore_aarch64 (void)

Variables
reg_dict	gpreg_dict []
static PyMemberDef	JitCpu_members []
static PyMethodDef	JitCpu_methods []
static PyGetSetDef	JitCpu_getseters []
static PyTypeObject	JitCpuType
static PyMethodDef	JitCore_aarch64_Methods []
static PyObject *	JitCore_aarch64_Error

Function Documentation

void check_automod	(	JitCpu *	jitcpu,
		uint64_t	addr,
		uint64_t	size
	)

Definition at line 215 of file JitCore_aarch64.c.

{
    PyObject *result;

    if (!(((VmMngr*)jitcpu->pyvm)->vm_mngr.exception_flags & EXCEPT_CODE_AUTOMOD))
        return;
    result = PyObject_CallMethod(jitcpu->jitter, "automod_cb", "LL", addr, size);
    Py_DECREF(result);

}

Here is the caller graph for this function:

PyObject* cpu_dump_gpregs	(	JitCpu *	self,
		PyObject *	args
	)

Definition at line 180 of file JitCore_aarch64.c.

{
    vm_cpu_t* vmcpu;

    vmcpu = self->cpu;
    dump_gpregs(vmcpu);
    Py_INCREF(Py_None);
    return Py_None;
}

Here is the call graph for this function:

PyObject* cpu_get_exception	(	JitCpu *	self,
		PyObject *	args
	)

Definition at line 206 of file JitCore_aarch64.c.

{
    return PyLong_FromUnsignedLongLong((uint64_t)(((vm_cpu_t*)self->cpu)->exception_flags));
}

PyObject* cpu_get_gpreg

(

JitCpu *

self

)

Definition at line 60 of file JitCore_aarch64.c.

{
    PyObject *dict = PyDict_New();
    PyObject *o;

    get_reg(X0);
    get_reg(X1);
    get_reg(X2);
    get_reg(X3);
    get_reg(X4);
    get_reg(X5);
    get_reg(X6);
    get_reg(X7);
    get_reg(X8);
    get_reg(X9);
    get_reg(X10);
    get_reg(X11);
    get_reg(X12);
    get_reg(X13);
    get_reg(X14);
    get_reg(X15);
    get_reg(X16);
    get_reg(X17);
    get_reg(X18);
    get_reg(X19);
    get_reg(X20);
    get_reg(X21);
    get_reg(X22);
    get_reg(X23);
    get_reg(X24);
    get_reg(X25);
    get_reg(X26);
    get_reg(X27);
    get_reg(X28);
    get_reg(X29);
    get_reg(LR);
    get_reg(SP);
    get_reg(PC);

    get_reg(zf);
    get_reg(nf);
    get_reg(of);
    get_reg(cf);

    return dict;
}

PyObject* cpu_init_regs

(

JitCpu *

self

)

Definition at line 146 of file JitCore_aarch64.c.

{
    memset(self->cpu, 0, sizeof(vm_cpu_t));

    Py_INCREF(Py_None);
    return Py_None;
}

PyObject* cpu_set_exception	(	JitCpu *	self,
		PyObject *	args
	)

Definition at line 191 of file JitCore_aarch64.c.

{
    PyObject *item1;
    uint64_t i;

    if (!PyArg_ParseTuple(args, "O", &item1))
        return NULL;

    PyGetInt(item1, i);

    ((vm_cpu_t*)self->cpu)->exception_flags = i;
    Py_INCREF(Py_None);
    return Py_None;
}

PyObject* cpu_set_gpreg	(	JitCpu *	self,
		PyObject *	args
	)

Definition at line 109 of file JitCore_aarch64.c.

{
    PyObject* dict;
    PyObject *d_key, *d_value = NULL;
    Py_ssize_t pos = 0;
    uint64_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)){
        if(!PyString_Check(d_key))
            RAISE(PyExc_TypeError, "key must be str");

        PyGetInt(d_value, val);

        found = 0;
        for (i=0; i < sizeof(gpreg_dict)/sizeof(reg_dict); i++){
            if (strcmp(PyString_AsString(d_key), gpreg_dict[i].name))
                continue;
            *((uint32_t*)(((char*)(self->cpu)) + gpreg_dict[i].offset)) = val;
            found = 1;
            break;
        }

        if (found)
            continue;
        fprintf(stderr, "unkown key: %s\n", PyString_AsString(d_key));
        RAISE(PyExc_ValueError, "unkown reg");
    }
    Py_INCREF(Py_None);
    return Py_None;
}

void dump_gpregs

(

vm_cpu_t *

vmcpu

)

Definition at line 154 of file JitCore_aarch64.c.

{
    printf("X0  %.16"PRIX64" X1  %.16"PRIX64" X2  %.16"PRIX64" X3  %.16"PRIX64" "\
           "X4  %.16"PRIX64" X5  %.16"PRIX64" X6  %.16"PRIX64" X7  %.16"PRIX64"\n",
           vmcpu->X0, vmcpu->X1, vmcpu->X2, vmcpu->X3, vmcpu->X4, vmcpu->X5, vmcpu->X6, vmcpu->X7);
    printf("X8  %.16"PRIX64" X9  %.16"PRIX64" X10 %.16"PRIX64" X11 %.16"PRIX64" "\
           "X12 %.16"PRIX64" X13 %.16"PRIX64" X14 %.16"PRIX64" X15 %.16"PRIX64"\n",
           vmcpu->X8, vmcpu->X9, vmcpu->X10, vmcpu->X11,
           vmcpu->X12, vmcpu->X13, vmcpu->X14, vmcpu->X15);
    printf("X16 %.16"PRIX64" X17 %.16"PRIX64" X18 %.16"PRIX64" X19 %.16"PRIX64" "\
           "X20 %.16"PRIX64" X21 %.16"PRIX64" X22 %.16"PRIX64" X23 %.16"PRIX64"\n",
           vmcpu->X16, vmcpu->X17, vmcpu->X18, vmcpu->X19,
           vmcpu->X20, vmcpu->X21, vmcpu->X22, vmcpu->X23);
    printf("X24 %.16"PRIX64" X25 %.16"PRIX64" X26 %.16"PRIX64" X27 %.16"PRIX64" "\
           "X28 %.16"PRIX64" X29 %.16"PRIX64" LR  %.16"PRIX64"\n",
           vmcpu->X24, vmcpu->X25, vmcpu->X26, vmcpu->X27,
           vmcpu->X28, vmcpu->X29, vmcpu->LR);


    printf("SP  %.16"PRIX64" PC  %.16"PRIX64" "\
           "zf  %.16"PRIX32" nf  %.16"PRIX32" of  %.16"PRIX32" cf  %.16"PRIX32"\n",
           vmcpu->SP, vmcpu->PC,
           vmcpu->zf, vmcpu->nf, vmcpu->of, vmcpu->cf);
}

Here is the caller graph for this function:

PyObject* get_gpreg_offset_all

(

void

)

Definition at line 383 of file JitCore_aarch64.c.

{
    PyObject *dict = PyDict_New();
    PyObject *o;

    get_reg_off(exception_flags);
    get_reg_off(exception_flags_new);

    get_reg_off(X0);
    get_reg_off(X1);
    get_reg_off(X2);
    get_reg_off(X3);
    get_reg_off(X4);
    get_reg_off(X5);
    get_reg_off(X6);
    get_reg_off(X7);
    get_reg_off(X8);
    get_reg_off(X9);
    get_reg_off(X10);
    get_reg_off(X11);
    get_reg_off(X12);
    get_reg_off(X13);
    get_reg_off(X14);
    get_reg_off(X15);
    get_reg_off(X16);
    get_reg_off(X17);
    get_reg_off(X18);
    get_reg_off(X19);
    get_reg_off(X20);
    get_reg_off(X21);
    get_reg_off(X22);
    get_reg_off(X23);
    get_reg_off(X24);
    get_reg_off(X25);
    get_reg_off(X26);
    get_reg_off(X27);
    get_reg_off(X28);
    get_reg_off(X29);
    get_reg_off(LR);
    get_reg_off(SP);
    get_reg_off(PC);


    get_reg_off(X0_new);
    get_reg_off(X1_new);
    get_reg_off(X2_new);
    get_reg_off(X3_new);
    get_reg_off(X4_new);
    get_reg_off(X5_new);
    get_reg_off(X6_new);
    get_reg_off(X7_new);
    get_reg_off(X8_new);
    get_reg_off(X9_new);
    get_reg_off(X10_new);
    get_reg_off(X11_new);
    get_reg_off(X12_new);
    get_reg_off(X13_new);
    get_reg_off(X14_new);
    get_reg_off(X15_new);
    get_reg_off(X16_new);
    get_reg_off(X17_new);
    get_reg_off(X18_new);
    get_reg_off(X19_new);
    get_reg_off(X20_new);
    get_reg_off(X21_new);
    get_reg_off(X22_new);
    get_reg_off(X23_new);
    get_reg_off(X24_new);
    get_reg_off(X25_new);
    get_reg_off(X26_new);
    get_reg_off(X27_new);
    get_reg_off(X28_new);
    get_reg_off(X29_new);
    get_reg_off(LR_new);
    get_reg_off(SP_new);
    get_reg_off(PC_new);



    /* eflag */
    get_reg_off(zf);
    get_reg_off(nf);
    get_reg_off(of);
    get_reg_off(cf);

    get_reg_off(zf_new);
    get_reg_off(nf_new);
    get_reg_off(of_new);
    get_reg_off(cf_new);


    get_reg_off(pfmem08_0);
    get_reg_off(pfmem08_1);
    get_reg_off(pfmem08_2);
    get_reg_off(pfmem08_3);
    get_reg_off(pfmem08_4);
    get_reg_off(pfmem08_5);
    get_reg_off(pfmem08_6);
    get_reg_off(pfmem08_7);
    get_reg_off(pfmem08_8);
    get_reg_off(pfmem08_9);
    get_reg_off(pfmem08_10);
    get_reg_off(pfmem08_11);
    get_reg_off(pfmem08_12);
    get_reg_off(pfmem08_13);
    get_reg_off(pfmem08_14);
    get_reg_off(pfmem08_15);
    get_reg_off(pfmem08_16);
    get_reg_off(pfmem08_17);
    get_reg_off(pfmem08_18);
    get_reg_off(pfmem08_19);


    get_reg_off(pfmem16_0);
    get_reg_off(pfmem16_1);
    get_reg_off(pfmem16_2);
    get_reg_off(pfmem16_3);
    get_reg_off(pfmem16_4);
    get_reg_off(pfmem16_5);
    get_reg_off(pfmem16_6);
    get_reg_off(pfmem16_7);
    get_reg_off(pfmem16_8);
    get_reg_off(pfmem16_9);
    get_reg_off(pfmem16_10);
    get_reg_off(pfmem16_11);
    get_reg_off(pfmem16_12);
    get_reg_off(pfmem16_13);
    get_reg_off(pfmem16_14);
    get_reg_off(pfmem16_15);
    get_reg_off(pfmem16_16);
    get_reg_off(pfmem16_17);
    get_reg_off(pfmem16_18);
    get_reg_off(pfmem16_19);


    get_reg_off(pfmem32_0);
    get_reg_off(pfmem32_1);
    get_reg_off(pfmem32_2);
    get_reg_off(pfmem32_3);
    get_reg_off(pfmem32_4);
    get_reg_off(pfmem32_5);
    get_reg_off(pfmem32_6);
    get_reg_off(pfmem32_7);
    get_reg_off(pfmem32_8);
    get_reg_off(pfmem32_9);
    get_reg_off(pfmem32_10);
    get_reg_off(pfmem32_11);
    get_reg_off(pfmem32_12);
    get_reg_off(pfmem32_13);
    get_reg_off(pfmem32_14);
    get_reg_off(pfmem32_15);
    get_reg_off(pfmem32_16);
    get_reg_off(pfmem32_17);
    get_reg_off(pfmem32_18);
    get_reg_off(pfmem32_19);


    get_reg_off(pfmem64_0);
    get_reg_off(pfmem64_1);
    get_reg_off(pfmem64_2);
    get_reg_off(pfmem64_3);
    get_reg_off(pfmem64_4);
    get_reg_off(pfmem64_5);
    get_reg_off(pfmem64_6);
    get_reg_off(pfmem64_7);
    get_reg_off(pfmem64_8);
    get_reg_off(pfmem64_9);
    get_reg_off(pfmem64_10);
    get_reg_off(pfmem64_11);
    get_reg_off(pfmem64_12);
    get_reg_off(pfmem64_13);
    get_reg_off(pfmem64_14);
    get_reg_off(pfmem64_15);
    get_reg_off(pfmem64_16);
    get_reg_off(pfmem64_17);
    get_reg_off(pfmem64_18);
    get_reg_off(pfmem64_19);

    return dict;
}

getset_reg_u32

(

zf

)

getset_reg_u32

(

nf

)

getset_reg_u32

(

of

)

getset_reg_u32

(

cf

)

getset_reg_u64

(

X0

)

getset_reg_u64

(

X1

)

getset_reg_u64

(

X2

)

getset_reg_u64

(

X3

)

getset_reg_u64

(

X4

)

getset_reg_u64

(

X5

)

getset_reg_u64

(

X6

)

getset_reg_u64

(

X7

)

getset_reg_u64

(

X8

)

getset_reg_u64

(

X9

)

getset_reg_u64

(

X10

)

getset_reg_u64

(

X11

)

getset_reg_u64

(

X12

)

getset_reg_u64

(

X13

)

getset_reg_u64

(

X14

)

getset_reg_u64

(

X15

)

getset_reg_u64

(

X16

)

getset_reg_u64

(

X17

)

getset_reg_u64

(

X18

)

getset_reg_u64

(

X19

)

getset_reg_u64

(

X20

)

getset_reg_u64

(

X21

)

getset_reg_u64

(

X22

)

getset_reg_u64

(

X23

)

getset_reg_u64

(

X24

)

getset_reg_u64

(

X25

)

getset_reg_u64

(

X26

)

getset_reg_u64

(

X27

)

getset_reg_u64

(

X28

)

getset_reg_u64

(

X29

)

getset_reg_u64

(

LR

)

getset_reg_u64

(

SP

)

getset_reg_u64

(

PC

)

PyMODINIT_FUNC initJitCore_aarch64

(

void

)

Definition at line 680 of file JitCore_aarch64.c.

{
    PyObject *m;

    if (PyType_Ready(&JitCpuType) < 0)
    return;

    m = Py_InitModule("JitCore_aarch64", JitCore_aarch64_Methods);
    if (m == NULL)
        return;

    JitCore_aarch64_Error = PyErr_NewException("JitCore_aarch64.error", NULL, NULL);
    Py_INCREF(JitCore_aarch64_Error);
    PyModule_AddObject(m, "error", JitCore_aarch64_Error);

    Py_INCREF(&JitCpuType);
    PyModule_AddObject(m, "JitCpu", (PyObject *)&JitCpuType);

}

static int JitCpu_init ( JitCpu *  self, PyObject *  args, PyObject *  kwds  )

static

Definition at line 330 of file JitCore_aarch64.c.

{
    self->cpu = malloc(sizeof(vm_cpu_t));
    if (self->cpu == NULL) {
        fprintf(stderr, "cannot alloc vm_cpu_t\n");
        exit(0);
    }
    return 0;
}

void MEM_WRITE_08	(	JitCpu *	jitcpu,
		uint64_t	addr,
		uint8_t	src
	)

Definition at line 226 of file JitCore_aarch64.c.

{
    vm_MEM_WRITE_08(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
    check_automod(jitcpu, addr, 8);
}

void MEM_WRITE_16	(	JitCpu *	jitcpu,
		uint64_t	addr,
		uint16_t	src
	)

Definition at line 232 of file JitCore_aarch64.c.

{
    vm_MEM_WRITE_16(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
    check_automod(jitcpu, addr, 16);
}

void MEM_WRITE_32	(	JitCpu *	jitcpu,
		uint64_t	addr,
		uint32_t	src
	)

Definition at line 238 of file JitCore_aarch64.c.

{
    vm_MEM_WRITE_32(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
    check_automod(jitcpu, addr, 32);
}

void MEM_WRITE_64	(	JitCpu *	jitcpu,
		uint64_t	addr,
		uint64_t	src
	)

Definition at line 244 of file JitCore_aarch64.c.

{
    vm_MEM_WRITE_64(&((VmMngr*)jitcpu->pyvm)->vm_mngr, addr, src);
    check_automod(jitcpu, addr, 64);
}

PyObject* vm_set_mem	(	JitCpu *	self,
		PyObject *	args
	)

Definition at line 255 of file JitCore_aarch64.c.

{
       PyObject *py_addr;
       PyObject *py_buffer;
       Py_ssize_t py_length;

       char * buffer;
       uint64_t size;
       uint64_t addr;
       int ret = 0x1337;

       if (!PyArg_ParseTuple(args, "OO", &py_addr, &py_buffer))
           return NULL;

       PyGetInt(py_addr, addr);

       if(!PyString_Check(py_buffer))
           RAISE(PyExc_TypeError,"arg must be str");

       size = PyString_Size(py_buffer);
       PyString_AsStringAndSize(py_buffer, &buffer, &py_length);

       ret = vm_write_mem(&(((VmMngr*)self->pyvm)->vm_mngr), addr, buffer, size);
       if (ret < 0)
           RAISE(PyExc_TypeError,"arg must be str");
       check_automod(self, addr, size*8);

       Py_INCREF(Py_None);
       return Py_None;
}

Here is the call graph for this function:

Variable Documentation

reg_dict gpreg_dict[]

Definition at line 13 of file JitCore_aarch64.c.

PyObject* JitCore_aarch64_Error

static

Definition at line 677 of file JitCore_aarch64.c.

PyMethodDef JitCore_aarch64_Methods[]

static

Initial value:

= {
    {"get_gpreg_offset_all", (PyCFunction)get_gpreg_offset_all, METH_NOARGS},
    {NULL, NULL, 0, NULL}        

}

Definition at line 671 of file JitCore_aarch64.c.

PyGetSetDef JitCpu_getseters[]

static

Definition at line 565 of file JitCore_aarch64.c.

PyMemberDef JitCpu_members[]

static

Initial value:

= {
    {NULL}  
}

Definition at line 305 of file JitCore_aarch64.c.

PyMethodDef JitCpu_methods[]

static

Initial value:

= {
    {"init_regs", (PyCFunction)cpu_init_regs, METH_NOARGS,
     "X"},
    {"dump_gpregs", (PyCFunction)cpu_dump_gpregs, METH_NOARGS,
     "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"},
    {"set_mem", (PyCFunction)vm_set_mem, METH_VARARGS,
     "X"},
    {"get_mem", (PyCFunction)vm_get_mem, METH_VARARGS,
     "X"},
    {NULL}  
}

Definition at line 309 of file JitCore_aarch64.c.

PyTypeObject JitCpuType

static

Definition at line 627 of file JitCore_aarch64.c.