aboutsummaryrefslogtreecommitdiff
path: root/tests/jmpi.c
blob: cb12789fbca146dfb86262a8698262eb5210c4c9 (plain) (blame)
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
#include "test.h"

void *tail;

static void *target;

static void
run_test(jit_state_t *j, uint8_t *arena_base, size_t arena_size)
{
	jit_begin(j, arena_base, arena_size);
	jit_enter_jit_abi(j, 0, 0, 0);
	jit_movi(j, JIT_R0, 42);
	jit_jmpi(j, target);
	// Unreachable.
	jit_breakpoint(j);
	int (*f)(void) = jit_end(j, NULL);
	ASSERT(f() == 42);
}

// Make the tail-call target via a separate main_helper because probably the new
// arena will be allocated farther away, forcing nonlocal jumps.
static void
make_target(jit_state_t *j, uint8_t *arena_base, size_t arena_size)
{
	jit_begin(j, arena_base, arena_size);
	size_t align = jit_enter_jit_abi(j, 0, 0, 0);
	// Tail call target assumes tail caller called enter_jit_abi with compatible
	// parameters.
	target = jit_address(j);
	jit_leave_jit_abi(j, 0, 0, align);
	jit_retr(j, JIT_R0);
	jit_end(j, NULL);

	main_helper(0, NULL, run_test);
}

int
main (int argc, char *argv[])
{
	return main_helper(argc, argv, make_target);
}