1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
#include <Python.h>
#include <inttypes.h>
#include <stdint.h>
#include "queue.h"
#include "vm_mngr.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_bloc(PyObject* self, PyObject* args)
{
uint64_t (*func)(void*, void*, void*, uint8_t*);
uint64_t vm;
uint64_t ret;
JitCpu* jitcpu;
uint8_t status;
PyObject* func_py;
PyObject* lbl2ptr;
PyObject* breakpoints;
PyObject* retaddr = NULL;
if (!PyArg_ParseTuple(args, "OOKOO", &retaddr, &jitcpu, &vm, &lbl2ptr, &breakpoints))
return NULL;
vm_cpu_t* cpu = jitcpu->cpu;
/* The loop will decref retaddr always once */
Py_INCREF(retaddr);
for (;;) {
// 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 breakpoint
if (PyDict_Contains(breakpoints, retaddr))
return retaddr;
}
}
static PyMethodDef LLVMMethods[] = {
{"llvm_exec_bloc", llvm_exec_bloc, METH_VARARGS,
"llvm exec bloc"},
{NULL, NULL, 0, NULL} /* Sentinel */
};
PyMODINIT_FUNC
initJitllvm(void)
{
PyObject *m;
m = Py_InitModule("Jitllvm", LLVMMethods);
if (m == NULL)
return;
}
|