1 files changed, 41 insertions, 0 deletions
diff --git a/tests/jmpi.c b/tests/jmpi.c
new file mode 100644
index 0000000..cb12789
--- /dev/null
+++ b/tests/jmpi.c
@@ -0,0 +1,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);
+}