1 files changed, 103 insertions, 0 deletions
diff --git a/yjit_utils.c b/yjit_utils.c
new file mode 100644
index 0000000000..02e5dbb3b0
--- /dev/null
+++ b/yjit_utils.c
@@ -0,0 +1,103 @@
+#include <stdio.h>
+#include <string.h>
+#include <assert.h>
+#include "yjit_utils.h"
+#include "yjit_asm.h"
+
+// Save caller-save registers on the stack before a C call
+void push_regs(codeblock_t* cb)
+{
+    push(cb, RAX);
+    push(cb, RCX);
+    push(cb, RDX);
+    push(cb, RSI);
+    push(cb, RDI);
+    push(cb, R8);
+    push(cb, R9);
+    push(cb, R10);
+    push(cb, R11);
+    pushfq(cb);
+}
+
+// Restore caller-save registers from the after a C call
+void pop_regs(codeblock_t* cb)
+{
+    popfq(cb);
+    pop(cb, R11);
+    pop(cb, R10);
+    pop(cb, R9);
+    pop(cb, R8);
+    pop(cb, RDI);
+    pop(cb, RSI);
+    pop(cb, RDX);
+    pop(cb, RCX);
+    pop(cb, RAX);
+}
+
+static void print_int_cfun(int64_t val)
+{
+    fprintf(stderr, "%lld\n", (long long int)val);
+}
+
+void print_int(codeblock_t* cb, x86opnd_t opnd)
+{
+    push_regs(cb);
+
+    if (opnd.num_bits < 64 && opnd.type != OPND_IMM)
+        movsx(cb, RDI, opnd);
+    else
+        mov(cb, RDI, opnd);
+
+    // Call the print function
+    mov(cb, RAX, const_ptr_opnd((void*)&print_int_cfun));
+    call(cb, RAX);
+
+    pop_regs(cb);
+}
+
+static void print_ptr_cfun(void* val)
+{
+    fprintf(stderr, "%p\n", val);
+}
+
+void print_ptr(codeblock_t* cb, x86opnd_t opnd)
+{
+    assert (opnd.num_bits == 64);
+
+    push_regs(cb);
+
+    mov(cb, RDI, opnd);
+    mov(cb, RAX, const_ptr_opnd((void*)&print_ptr_cfun));
+    call(cb, RAX);
+
+    pop_regs(cb);
+}
+
+static void print_str_cfun(const char* str)
+{
+    fprintf(stderr, "%s\n", str);
+}
+
+// Print a constant string to stdout
+void print_str(codeblock_t* cb, const char* str)
+{
+    //as.comment("printStr(\"" ~ str ~ "\")");
+    size_t len = strlen(str);
+
+    push_regs(cb);
+
+    // Load the string address and jump over the string data
+    lea(cb, RDI, mem_opnd(8, RIP, 5));
+    jmp32(cb, (int32_t)len + 1);
+
+    // Write the string chars and a null terminator
+    for (size_t i = 0; i < len; ++i)
+        cb_write_byte(cb, (uint8_t)str[i]);
+    cb_write_byte(cb, 0);
+
+    // Call the print function
+    mov(cb, RAX, const_ptr_opnd((void*)&print_str_cfun));
+    call(cb, RAX);
+
+    pop_regs(cb);
+}