add lightening jit

1 files changed, 79 insertions, 0 deletions

diff --git a/deps/lightening/tests/test.h b/deps/lightening/tests/test.h
new file mode 100644
index 0000000..c4eff2b
--- /dev/null
+++ b/deps/lightening/tests/test.h
@@ -0,0 +1,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;
+}