Merge pull request ruby#133 from Shopify/runtime-tests

maximecb · web-flow · commit 010c89aae7e1 · 2021-07-19T20:13:40.000-04:00
Add some YJIT runtime tests
diff --git a/yjit_asm_tests.c b/yjit_asm_tests.c
@@ -58,7 +58,7 @@ void check_bytes(codeblock_t* cb, const char* bytes)
     }
 }
 
-void run_tests()
+void run_assembler_tests()
 {
     printf("Running assembler tests\n");
 
@@ -353,9 +353,64 @@ void run_tests()
     printf("Assembler tests done\n");
 }
 
+void assert_equal(expected, actual)
+{
+    if (expected != actual) {
+        fprintf(stderr, "expected %d, got %d\n", expected, actual);
+        exit(-1);
+    }
+}
+
+void run_runtime_tests()
+{
+    printf("Running runtime tests\n");
+
+    codeblock_t codeblock;
+    codeblock_t* cb = &codeblock;
+
+    uint8_t* mem_block = alloc_exec_mem(4096);
+    cb_init(cb, mem_block, 4096);
+
+    int (*function)(void);
+    function = (int (*)(void))mem_block;
+
+    #define TEST(BODY) cb_set_pos(cb, 0); BODY ret(cb); assert_equal(7, function());
+
+    // add
+    TEST({ mov(cb, RAX, imm_opnd(0)); add(cb, RAX, imm_opnd(7)); })
+    TEST({ mov(cb, RAX, imm_opnd(0)); mov(cb, RCX, imm_opnd(7)); add(cb, RAX, RCX); })
+
+    // and
+    TEST({ mov(cb, RAX, imm_opnd(31)); and(cb, RAX, imm_opnd(7)); })
+    TEST({ mov(cb, RAX, imm_opnd(31)); mov(cb, RCX, imm_opnd(7)); and(cb, RAX, RCX); })
+
+    // or
+    TEST({ mov(cb, RAX, imm_opnd(3)); or(cb, RAX, imm_opnd(4)); })
+    TEST({ mov(cb, RAX, imm_opnd(3)); mov(cb, RCX, imm_opnd(4)); or(cb, RAX, RCX); })
+
+    // push/pop
+    TEST({ mov(cb, RCX, imm_opnd(7)); push(cb, RCX); pop(cb, RAX); })
+
+    // shr
+    TEST({ mov(cb, RAX, imm_opnd(31)); shr(cb, RAX, imm_opnd(2)); })
+
+    // sub
+    TEST({ mov(cb, RAX, imm_opnd(12)); sub(cb, RAX, imm_opnd(5)); })
+    TEST({ mov(cb, RAX, imm_opnd(12)); mov(cb, RCX, imm_opnd(5)); sub(cb, RAX, RCX); })
+
+    // xor
+    TEST({ mov(cb, RAX, imm_opnd(13)); xor(cb, RAX, imm_opnd(10)); })
+    TEST({ mov(cb, RAX, imm_opnd(13)); mov(cb, RCX, imm_opnd(10)); xor(cb, RAX, RCX); })
+
+    #undef TEST
+
+    printf("Runtime tests done\n");
+}
+
 int main(int argc, char** argv)
 {
-    run_tests();
+    run_assembler_tests();
+    run_runtime_tests();
 
     return 0;
 }
