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
|
#include <stdio.h>
#include <stdlib.h>
#include <sys/mman.h>
#include <lightening.h>
#define ASSERT(x) \
do { \
if (!(x)) { \
fprintf(stderr, "%s:%d: assertion failed: " #x "\n", \
__FILE__, __LINE__); \
abort(); \
} \
} while (0)
static inline int
main_helper (int argc, char *argv[],
void (*run_test)(jit_state_t*, uint8_t*, size_t))
{
ASSERT(init_jit());
jit_state_t *j = jit_new_state (NULL, NULL);
ASSERT(j);
const size_t arena_size = 4096;
char *arena_base = mmap (NULL, arena_size,
PROT_EXEC | PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (arena_base == MAP_FAILED)
{
perror ("allocating JIT code buffer failed");
return 1;
}
run_test(j, (uint8_t*)arena_base, arena_size);
jit_destroy_state(j);
munmap(arena_base, arena_size);
return 0;
}
static inline int
main_compiler (int argc, char *argv[],
size_t (*run_test)(jit_state_t*, uint8_t*, size_t))
{
ASSERT(init_jit());
jit_state_t *j = jit_new_state (NULL, NULL);
ASSERT(j);
size_t arena_size = 4096, prev_arena_size = arena_size;
uint8_t *arena_base = NULL;
do {
if (arena_base) {
if (munmap(arena_base, prev_arena_size) == -1) {
perror("unmapping arena failed");
return 1;
}
}
prev_arena_size = arena_size;
arena_base = mmap (NULL, arena_size,
PROT_EXEC | PROT_READ | PROT_WRITE,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (arena_base == MAP_FAILED) {
perror ("allocating JIT code buffer failed");
return 1;
}
} while ((arena_size = run_test(j, arena_base, arena_size)) != 0);
jit_destroy_state(j);
munmap(arena_base, arena_size);
return 0;
}
|
