summaryrefslogtreecommitdiff
path: root/yjit_utils.c
blob: cbc50a493254145b055979f594b7ad5778718b80 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// This file is a fragment of the yjit.o compilation unit. See yjit.c.

// Save caller-save registers on the stack before a C call
static 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
static 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);
}

RBIMPL_ATTR_MAYBE_UNUSED()
static 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);
}

RBIMPL_ATTR_MAYBE_UNUSED()
static 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
static 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);
}