Add lsh/rsh tests

.gitignore

@@ -29,3 +29,9 @@
 /tests/test-orr
 /tests/test-xori
 /tests/test-xorr
+/tests/test-lshi
+/tests/test-lshr
+/tests/test-rshi
+/tests/test-rshr
+/tests/test-rshr_u
+/tests/test-rshi_u

tests/test-lshi.c

@@ -0,0 +1,31 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[1];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 } };
+
+  jit_receive(j, 1, abi, args);
+  jit_load_args(j, 1, abi, args, regs);
+
+  jit_lshi(j, JIT_R0, JIT_R0, 31);
+  jit_retr(j, JIT_R0);
+
+  intmax_t (*f)(intmax_t) = jit_end(j, NULL);
+
+#if __WORDSIZE == 32
+  ASSERT(f(-0x7f) == 0x80000000);
+#else
+  ASSERT(f(-0x7f) == 0xffffffc080000000);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+  return main_helper(argc, argv, run_test);
+}

tests/test-lshr.c

@@ -0,0 +1,72 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX, JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[2];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 }, { .gpr=JIT_R1 }};
+
+  jit_receive(j, 2, abi, args);
+  jit_load_args(j, 2, abi, args, regs);
+
+  jit_lshr(j, JIT_R0, JIT_R0, JIT_R1);
+  jit_retr(j, JIT_R0);
+
+  size_t size = 0;
+  void* ret = jit_end(j, &size);
+
+  intmax_t (*f)(intmax_t, intmax_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 __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);
+}

tests/test-rshi.c

@@ -0,0 +1,32 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[1];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 } };
+
+  jit_receive(j, 1, abi, args);
+  jit_load_args(j, 1, abi, args, regs);
+
+  jit_rshi(j, JIT_R0, JIT_R0, 31);
+  jit_retr(j, JIT_R0);
+
+  intmax_t (*f)(intmax_t) = jit_end(j, NULL);
+
+#if __WORDSIZE == 32
+  ASSERT(f(0x80000000) == -1);
+#else
+  ASSERT(f(0x80000000) == 1);
+  ASSERT(f(0x8000000000000000) == 0xffffffff00000000);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+  return main_helper(argc, argv, run_test);
+}

tests/test-rshi_u.c

@@ -0,0 +1,32 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[1];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 } };
+
+  jit_receive(j, 1, abi, args);
+  jit_load_args(j, 1, abi, args, regs);
+
+  jit_rshi_u(j, JIT_R0, JIT_R0, 31);
+  jit_retr(j, JIT_R0);
+
+  intmax_t (*f)(intmax_t) = jit_end(j, NULL);
+
+#if __WORDSIZE == 32
+  ASSERT(f(0x80000000) == 1);
+#else
+  ASSERT(f(0x80000000) == 1);
+  ASSERT(f(0x8000000000000000) == 0x100000000);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+  return main_helper(argc, argv, run_test);
+}

tests/test-rshr.c

@@ -0,0 +1,66 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX, JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[2];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 }, { .gpr=JIT_R1 }};
+
+  jit_receive(j, 2, abi, args);
+  jit_load_args(j, 2, abi, args, regs);
+
+  jit_rshr(j, JIT_R0, JIT_R0, JIT_R1);
+  jit_retr(j, JIT_R0);
+
+  size_t size = 0;
+  void* ret = jit_end(j, &size);
+
+  intmax_t (*f)(intmax_t, intmax_t) = ret;
+
+  ASSERT(f(0xfe, 1) == 0x7f);
+  ASSERT(f(0x1fffc, 2) == 0x7fff);
+  ASSERT(f(0x40000000, 30) == 1);
+  ASSERT(f(0x20000000, 29) == 1);
+  ASSERT(f(0x10000000, 28) == 1);
+  ASSERT(f(0x810000, 16) == 0x81);
+  ASSERT(f(0x20000, 17) == 1);
+  ASSERT(f(0x40000, 18) == 1);
+  ASSERT(f(0x7f8000, 15) == 0xff);
+  ASSERT(f(0x1000000, 24) == 1);
+  ASSERT(f(0x7fffffff, 0) == 0x7fffffff);
+#if __WORDSIZE == 32
+  ASSERT(f(0xfffffff8, 3) == 0xffffffff);
+  ASSERT(f(0x80000000, 31) == 0xffffffff);
+  ASSERT(f(0xffffff00, 8) == 0xffffffff);
+#else
+  ASSERT(f(0x3fffffff8, 3) == 0x7fffffff);
+  ASSERT(f(0xffffffc080000000, 31) == 0xffffffffffffff81);
+  ASSERT(f(0xffffff00, 8) == 0xffffff);
+  ASSERT(f(0xfe00000000, 33) == 0x7f);
+  ASSERT(f(0x1ffffc00000000, 34) == 0x7ffff);
+  ASSERT(f(0xfffffff800000000, 29) == 0xffffffffffffffc0);
+  ASSERT(f(0x8000000000000000, 63) == 0xffffffffffffffff);
+  ASSERT(f(0x4000000000000000, 62) == 1);
+  ASSERT(f(0x2000000000000000, 61) == 1);
+  ASSERT(f(0x1000000000000000, 60) == 1);
+  ASSERT(f(0x81000000000000, 48) == 0x81);
+  ASSERT(f(0x2000000000000, 49) == 1);
+  ASSERT(f(0x10000000000, 40) == 1);
+  ASSERT(f(0x7f800000000000, 47) == 0xff);
+  ASSERT(f(0x100000000000000, 56) == 1);
+  ASSERT(f(0xffffff0000000000, 40) == 0xffffffffffffffff);
+  ASSERT(f(0xfffffffe00000000, 33) == 0xffffffffffffffff);
+  ASSERT(f(0x8000000000000001, 63) == 0xffffffffffffffff);
+  ASSERT(f(0x1000000000000, 48) == 1);
+  ASSERT(f(0xffff800000000000, 47) == 0xffffffffffffffff);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+  return main_helper(argc, argv, run_test);
+}

tests/test-rshr_u.c

@@ -0,0 +1,65 @@
+#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);
+
+  const jit_arg_abi_t abi[] = { JIT_ARG_ABI_INTMAX, JIT_ARG_ABI_INTMAX };
+  jit_arg_t args[2];
+  const jit_anyreg_t regs[] = { { .gpr=JIT_R0 }, { .gpr=JIT_R1 }};
+
+  jit_receive(j, 2, abi, args);
+  jit_load_args(j, 2, abi, args, regs);
+
+  jit_rshr_u(j, JIT_R0, JIT_R0, JIT_R1);
+  jit_retr(j, JIT_R0);
+
+  size_t size = 0;
+  void* ret = jit_end(j, &size);
+
+  intmax_t (*f)(intmax_t, intmax_t) = ret;
+
+  ASSERT(f(0xfe, 1) == 0x7f);
+  ASSERT(f(0x1fffc, 2) == 0x7fff);
+  ASSERT(f(0x80000000, 31) == 1);
+  ASSERT(f(0x40000000, 30) == 1);
+  ASSERT(f(0x20000000, 29) == 1);
+  ASSERT(f(0x10000000, 28) == 1);
+  ASSERT(f(0x810000, 16) == 0x81);
+  ASSERT(f(0x20000, 17) == 1);
+  ASSERT(f(0x40000, 18) == 1);
+  ASSERT(f(0x7f8000, 15) == 0xff);
+  ASSERT(f(0x1000000, 24) == 1);
+  ASSERT(f(0xffffff00, 8) == 0xffffff);
+  ASSERT(f(0x7fffffff, 0) == 0x7fffffff);
+#if __WORDSIZE == 32
+  ASSERT(f(0xfffffff8, 3) == 0x1fffffff);
+#else
+  ASSERT(f(0x3fffffff8, 3) == 0x7fffffff);
+  ASSERT(f(0xffffffc080000000, 31) == 0x1ffffff81);
+  ASSERT(f(0xfe00000000, 33) == 0x7f);
+  ASSERT(f(0x1ffffc00000000, 34) == 0x7ffff);
+  ASSERT(f(0xfffffff800000000, 29) == 0x7ffffffc0);
+  ASSERT(f(0x8000000000000000, 63) == 1);
+  ASSERT(f(0x4000000000000000, 62) == 1);
+  ASSERT(f(0x2000000000000000, 61) == 1);
+  ASSERT(f(0x1000000000000000, 60) == 1);
+  ASSERT(f(0x81000000000000, 48) == 0x81);
+  ASSERT(f(0x2000000000000, 49) == 1);
+  ASSERT(f(0x10000000000, 40) == 1);
+  ASSERT(f(0x7f800000000000, 47) == 0xff);
+  ASSERT(f(0x100000000000000, 56) == 1);
+  ASSERT(f(0xffffff0000000000, 40) == 0xffffff);
+  ASSERT(f(0xfffffffe00000000, 33) == 0x7fffffff);
+  ASSERT(f(0x8000000000000001, 63) == 1);
+  ASSERT(f(0x1000000000000, 48) == 1);
+  ASSERT(f(0xffff800000000000, 47) == 0x1ffff);
+#endif
+}
+
+int
+main (int argc, char *argv[])
+{
+  return main_helper(argc, argv, run_test);
+}