cea-sec/miasm

#include <Python.h>

#include <inttypes.h>

#include <stdint.h>
#include "compat_py23.h"
#include "queue.h"
#include "vm_mngr.h"
#include "bn.h"
#include "vm_mngr_py.h"
#include "JitCore.h"
// Needed to get the JitCpu.cpu offset, arch independent
#include "arch/JitCore_x86.h"

PyObject* llvm_exec_block(PyObject* self, PyObject* args)
{
    uint64_t (*func)(void*, void*, void*, uint8_t*);
    struct vm_cpu* cpu;
    vm_mngr_t* vm;
    uint64_t ret;
    JitCpu* jitcpu;
    uint8_t status;
    PyObject* func_py;
    PyObject* lbl2ptr;
    PyObject* stop_offsets;
    PyObject* retaddr = NULL;
    uint64_t max_exec_per_call = 0;
    uint64_t cpt;
    int do_cpt;

    if (!PyArg_ParseTuple(args, "OOOO|K",
                  &retaddr, &jitcpu, &lbl2ptr, &stop_offsets,
                  &max_exec_per_call))
        return NULL;

    cpu = jitcpu->cpu;
    vm = &(jitcpu->pyvm->vm_mngr);
    /* The loop will decref retaddr always once */
    Py_INCREF(retaddr);

    if (max_exec_per_call == 0) {
        do_cpt = 0;
        cpt = 1;
    } else {
        do_cpt = 1;
        cpt = max_exec_per_call;
    }

    for (;;) {
        // Handle cpt
        if (cpt == 0)
            return retaddr;
        if (do_cpt)
            cpt --;

        // Get the expected jitted function address
        func_py = PyDict_GetItem(lbl2ptr, retaddr);
        if (func_py)
            func = PyLong_AsVoidPtr((PyObject*) func_py);
        else
            // retaddr is not jitted yet
            return retaddr;

        // Execute it
        ret = func((void*) jitcpu, (void*)(intptr_t) cpu, (void*)(intptr_t) vm, &status);
        Py_DECREF(retaddr);
        retaddr = PyLong_FromUnsignedLongLong(ret);

        // Check exception
        if (status)
            return retaddr;

        // Check stop offsets
        if (PySet_Contains(stop_offsets, retaddr))
            return retaddr;
    }
}


static PyMethodDef LLVMMethods[] = {
    {"llvm_exec_block",  llvm_exec_block, METH_VARARGS,
     "llvm exec block"},
    {NULL, NULL, 0, NULL}        /* Sentinel */
};




MOD_INIT(Jitllvm)
{
    PyObject *module = NULL;

    MOD_DEF(module, "Jitllvm", "llvm module", LLVMMethods);

    RET_MODULE;
}