1 files changed, 69 insertions, 0 deletions
diff --git a/tests/lshr.c b/tests/lshr.c
new file mode 100644
index 0000000..89b9084
--- /dev/null
+++ b/tests/lshr.c
@@ -0,0 +1,69 @@
+#include "test.h"
+
+static void
+run_test(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);
+	jit_load_args_2(j, jit_operand_gpr (JIT_OPERAND_ABI_WORD, JIT_R0),
+	                jit_operand_gpr (JIT_OPERAND_ABI_WORD, JIT_R1));
+
+	jit_lshr(j, JIT_R0, JIT_R0, JIT_R1);
+	jit_leave_jit_abi(j, 0, 0, align);
+	jit_retr(j, JIT_R0);
+
+	size_t size = 0;
+	void* ret = jit_end(j, &size);
+
+	jit_word_t (*f)(jit_word_t, jit_word_t) = ret;
+
+	ASSERT(f(0x7f, 1) == 0xfe);
+	ASSERT(f(0x7fff, 2) == 0x1fffc);
+	ASSERT(f(0x81, 16) == 0x810000);
+	ASSERT(f(0xff, 15) == 0x7f8000);
+	ASSERT(f(0x7fffffff, 0) == 0x7fffffff);
+#if EJIT_WORDSIZE == 32
+	ASSERT(f(0xffffffff, 8) == 0xffffff00);
+	ASSERT(f(0x7fffffff, 3) == 0xfffffff8);
+	ASSERT(f(-0x7f, 31) == 0x80000000);
+	ASSERT(f(-0x7fff, 30) == 0x40000000);
+	ASSERT(f(-0x7fffffff, 29) == 0x20000000);
+	ASSERT(f(0x80000001, 28) == 0x10000000);
+	ASSERT(f(0x8001, 17) == 0x20000);
+	ASSERT(f(0x80000001, 18) == 0x40000);
+	ASSERT(f(-0xffff, 24) == 0x1000000);
+#else
+	ASSERT(f(0xffffffff, 8) == 0xffffffff00);
+	ASSERT(f(0x7fffffff, 3) == 0x3fffffff8);
+	ASSERT(f(-0x7f, 31) == 0xffffffc080000000);
+	ASSERT(f(-0x7fff, 30) == 0xffffe00040000000);
+	ASSERT(f(-0x7fffffff, 29) == 0xf000000020000000);
+	ASSERT(f(0x80000001, 28) == 0x800000010000000);
+	ASSERT(f(0x8001, 17) == 0x100020000);
+	ASSERT(f(0x80000001, 18) == 0x2000000040000);
+	ASSERT(f(-0xffff, 24) == 0xffffff0001000000);
+	ASSERT(f(0x7f, 33) == 0xfe00000000);
+	ASSERT(f(0x7ffff, 34) == 0x1ffffc00000000);
+	ASSERT(f(0x7fffffff, 35) == 0xfffffff800000000);
+	ASSERT(f(-0x7f, 63) == 0x8000000000000000);
+	ASSERT(f(-0x7fff, 62) == 0x4000000000000000);
+	ASSERT(f(-0x7fffffff, 61) == 0x2000000000000000);
+	ASSERT(f(0x80000001, 60) == 0x1000000000000000);
+	ASSERT(f(0x81, 48) == 0x81000000000000);
+	ASSERT(f(0x8001, 49) == 0x2000000000000);
+	ASSERT(f(0x80000001, 40) == 0x10000000000);
+	ASSERT(f(0xff, 47) == 0x7f800000000000);
+	ASSERT(f(0xffff0001, 56) == 0x100000000000000);
+	ASSERT(f(0xffffffff, 40) == 0xffffff0000000000);
+	ASSERT(f(0x7fffffffff, 33) == 0xfffffffe00000000);
+	ASSERT(f(-0x7fffffffff, 63) == 0x8000000000000000);
+	ASSERT(f(0x8000000001, 48) == 0x1000000000000);
+	ASSERT(f(0xffffffffff, 47) == 0xffff800000000000);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+	return main_helper(argc, argv, run_test);
+}