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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#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;
}
|