123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310 |
- #include <inttypes.h>
- char* regnames[] = {
- "%rax",
- "%rdi",
- "%rsi",
- "%rdx",
- "%r12",
- "%r13",
- "%r14",
- "%r15",
- "%r8",
- "%r9",
- "%r10",
- "%r11",
- "%rsp"
- };
- enum jit_reg {
- R0 = 0,
- R1,
- R2,
- R3,
- R4,
- R5,
- R6,
- R7,
- R8,
- R9,
- R10,
- R11,
- R12,
- R13,
- R14,
- R15
- };
- enum arg_reg {
- ARGR0 = 1,
- ARGR1 = 2,
- ARGR2 = 3
- };
- #define RSP R12
- void jit_init() {
- }
- void jit_movi(int reg, uint64_t imm) {
- fprintf(jit_out, "movq $%" PRIu64 ", %s\n", imm, regnames[reg]);
- }
- void jit_movr(int dreg, int sreg) {
- if (dreg == sreg) return;
- fprintf(jit_out, "movq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_movneg(int dreg, int sreg) {
- if (dreg == sreg) return;
- fprintf(jit_out, "cmovs %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_movne(int dreg, int sreg) {
- if (dreg == sreg) return;
- fprintf(jit_out, "cmovne %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_moveq(int dreg, int sreg) {
- if (dreg == sreg) return;
- fprintf(jit_out, "cmoveq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_lea(int reg, void* addr) {
- fprintf(jit_out, "movq $%p, %s\n", addr, regnames[reg]);
- }
- void jit_ldr(int reg) {
- fprintf(jit_out, "movq (%s), %s\n", regnames[reg], regnames[reg]);
- }
- void jit_ldr_stack(int dreg, int offset) {
- fprintf(jit_out, "movq %d(%%rsp), %s\n", offset, regnames[dreg]);
- }
- void jit_str_stack(int sreg, int offset) {
- fprintf(jit_out, "movq %s, %d(%%rsp)\n", regnames[sreg], offset);
- }
- void jit_inc_stack(int offset) {
- if (offset == 0) return;
- fprintf(jit_out, "addq $%d, %%rsp\n", offset);
- }
- void jit_dec_stack(int offset) {
- if (offset == 0) return;
- fprintf(jit_out, "subq $%d, %%rsp\n", offset);
- }
- // clobbers rdx!
- void jit_ldrb(int reg) {
- fprintf(jit_out, "movb (%s), %%dl\n", regnames[reg]);
- fprintf(jit_out, "andq $0xff, %%rdx\n");
- if (reg!=3) {
- fprintf(jit_out, "movq %%rdx, %s\n", regnames[reg]);
- }
- }
- // clobbers rdx!
- void jit_ldrs(int reg) {
- fprintf(jit_out, "movw (%s), %%dx\n", regnames[reg]);
- fprintf(jit_out, "andq $0xffff, %%rdx\n");
- if (reg!=3) {
- fprintf(jit_out, "movq %%rdx, %s\n", regnames[reg]);
- }
- }
- // clobbers rdx!
- void jit_ldrw(int reg) {
- fprintf(jit_out, "movl (%s), %%edx\n", regnames[reg]);
- //fprintf(jit_out, "andq $0xffffffff, %rdx\n", regnames[reg]);
- if (reg!=3) {
- fprintf(jit_out, "movq %%rdx, %s\n", regnames[reg]);
- }
- }
- // 8 bit only from rdx!
- void jit_strb(int reg) {
- fprintf(jit_out, "movb %%dl, (%s)\n", regnames[reg]);
- }
- // 16 bit only from rdx!
- void jit_strs(int reg) {
- fprintf(jit_out, "movw %%dx, (%s)\n", regnames[reg]);
- }
- // 32 bit only from rdx!
- void jit_strw(int reg) {
- fprintf(jit_out, "movl %%edx, (%s)\n", regnames[reg]);
- }
- void jit_stra(int reg) {
- fprintf(jit_out, "movq %%rdx, (%s)\n", regnames[reg]);
- }
- void jit_addr(int dreg, int sreg) {
- fprintf(jit_out, "addq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_addi(int dreg, int imm) {
- fprintf(jit_out, "addq $%d, %s\n", imm, regnames[dreg]);
- }
- void jit_andr(int dreg, int sreg) {
- fprintf(jit_out, "andq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_notr(int dreg) {
- fprintf(jit_out, "notq %s\n", regnames[dreg]);
- }
- void jit_orr(int dreg, int sreg) {
- fprintf(jit_out, "orq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_xorr(int dreg, int sreg) {
- fprintf(jit_out, "xorq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_shrr(int dreg, int sreg) {
- fprintf(jit_out, "movq %s, %%rcx\n", regnames[sreg]);
- fprintf(jit_out, "shr %%cl, %s\n", regnames[dreg]);
- }
- void jit_shlr(int dreg, int sreg) {
- fprintf(jit_out, "movq %s, %%rcx\n", regnames[sreg]);
- fprintf(jit_out, "shl %%cl, %s\n", regnames[dreg]);
- }
- void jit_subr(int dreg, int sreg) {
- fprintf(jit_out, "subq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_mulr(int dreg, int sreg) {
- fprintf(jit_out, "imulq %s, %s\n", regnames[sreg], regnames[dreg]);
- }
- void jit_divr(int dreg, int sreg) {
- fprintf(jit_out, "movq %s, %%rax\n", regnames[dreg]);
- fprintf(jit_out, "cqto\n");
- fprintf(jit_out, "idivq %s\n", regnames[sreg]);
- fprintf(jit_out, "movq %%rax, %s\n", regnames[dreg]);
- }
- void jit_host_call_enter() {
- fprintf(jit_out, "movq %%rsp, %%rax\n");
- fprintf(jit_out, "push %%rax\n");
- fprintf(jit_out, "andq $0xfffffffffffffff0, %%rsp\n");
- fprintf(jit_out, "mov %%rax,(%%rsp)\n");
- }
- void jit_host_call_exit() {
- fprintf(jit_out, "pop %%rsp\n");
- }
- void jit_call(void* func, char* note) {
- fprintf(jit_out, "mov $%p, %%rax\n", func);
- fprintf(jit_out, "callq *%%rax # %s\n", note);
- }
- #define jit_call2 jit_call
- #define jit_call3 jit_call
- void jit_callr(int reg) {
- fprintf(jit_out, "callq *%s\n", regnames[reg]);
- }
- int32_t inline_mod(int64_t a, int64_t b) {
- return a%b;
- }
- void jit_modr(int dreg, int sreg) {
- jit_movr(ARGR0,dreg);
- jit_movr(ARGR1,sreg);
- jit_call(inline_mod,"mod");
- if (dreg!=0) jit_movr(dreg,0);
- }
- void jit_cmpi(int sreg, int imm) {
- fprintf(jit_out, "cmp $%d, %s\n", imm, regnames[sreg]);
- }
- void jit_cmpr(int sreg, int dreg) {
- fprintf(jit_out, "cmp %s, %s\n", regnames[dreg], regnames[sreg]);
- }
- void jit_je(char* label) {
- fprintf(jit_out, "je %s\n", label);
- }
- void jit_jne(char* label) {
- fprintf(jit_out, "jne %s\n", label);
- }
- void jit_jge(char* label) {
- fprintf(jit_out, "jge %s\n", label);
- }
- void jit_jneg(char* label) {
- fprintf(jit_out, "js %s\n", label);
- }
- void jit_jmp(char* label) {
- fprintf(jit_out, "jmp %s\n", label);
- }
- void jit_label(char* label) {
- fprintf(jit_out, "%s:\n", label);
- }
- void jit_ret() {
- fprintf(jit_out, "ret\n");
- }
- void jit_push(int r1, int r2) {
- for (int i=r1; i<=r2; i++) {
- fprintf(jit_out, "push %s\n",regnames[i]);
- }
- }
- void jit_pop(int r1, int r2) {
- for (int i=r2; i>=r1; i--) {
- fprintf(jit_out, "pop %s\n",regnames[i]);
- }
- }
- void jit_comment(char* comment) {
- fprintf(jit_out, "# %s\n",comment);
- }
- void debug_handler(char* line, Frame* frame) {
- printf("@ %s\r\n",line);
-
- if (debug_mode==2 && frame) {
- if (frame->f) {
- for (int i=0; i<MAXFRAME; i++) {
- char* typestr = "UNKNOWN";
- Arg a = frame->f[i];
-
- if (a.type) {
- switch (a.type) {
- case ARGT_CONST: typestr = "CONST"; break;
- case ARGT_ENV: typestr = "ENV"; break;
- case ARGT_LAMBDA: typestr = "LAMBDA"; break;
- case ARGT_REG: {
- char buf[10];
- sprintf(buf,"R%d(%s)",a.slot+LBDREG,regnames[a.slot+LBDREG]);
- typestr = buf;
- break;
- }
- case ARGT_REG_INT: typestr = "INT"; break;
- case ARGT_STACK: typestr = "STACK"; break;
- case ARGT_STACK_INT: typestr = "STACK_INT"; break;
- }
-
- printf(" %2d\t%s\t%s\t%d\r\n",i,a.name,typestr,a.slot);
- }
- }
- } else {
- //printf(" empty frame\r\n");
- }
- }
- }
|