janis/from-scratch
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

codegen working? pogU?

Browse source
This commit is contained in:
janis 2025-11-01 03:01:48 +01:00
parent 105c052e95
commit 2ca60ace59
Signed by: janis
SSH key fingerprint: SHA256:bB1qbbqmDXZNT0KKD5c2Dfjg53JGhj7B3CFcLIzSqq8
1 changed files with 31 additions and 23 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

54
lang/src/codegen.asm
View file

@ -21,11 +21,9 @@ section .rdata
GLOBAL_LEN equ $ - GLOBAL_
COLON_NL db ":", 10
COLON_NL_LEN equ $ - COLON_NL
RET_NL db "ret", 10
RET_NL_LEN equ $ - RET_NL
PROLOGUE db "push rbp", 10, "mov rbp, rsp", 10
PROLOGUE_LEN equ $ - PROLOGUE
EPILOGUE db "mov rsp, rbp", 10, "pop rbp", 10, "ret", 10
EPILOGUE db "pop rbp", 10, "ret", 10
EPILOGUE_LEN equ $ - EPILOGUE
MOV_RAX db "mov rax, "
MOV_RAX_LEN equ $ - MOV_RAX
@ -49,6 +47,8 @@ section .rdata
SUB_RSP_LEN equ $ - SUB_RSP
ADD_RSP db "add rsp, "
ADD_RSP_LEN equ $ - ADD_RSP
BODY db ".body:", 10
BODY_LEN equ $ - BODY
REGISTER_NAMES db "abcdsidibpspr8r9r10r11r12r13r14r15"
@ -359,6 +359,7 @@ codegen_function:
bts word [rsp + 72], 7 ; mark rsp as used
bts word [rsp + 72], 6 ; mark rbp as used
bts word [rsp + 72], 0 ; mark rax as used
mov word [rsp + 76], 0 ; dirtied_register_bitset = 0
; push "section .text\n"
mov rdi, [rsp] ; ctx
@ -442,9 +443,9 @@ codegen_function:
mov [rsp + 104], rsi ; scratch
mov rax, [rsp + 24] ; current_stack_size
mov [rsp + 112], rax
add rax, 8 ; size_of::<u64>
mov [rsp + 24], rax ; current_stack_size += size_of::<u64>
mov [rsp + 112], rax
lea rdi, [rsp + 32] ; stack-vars
lea rsi, [rsp + 104] ; &scratch: &(index, offset)
mov rdx, stackvar_cmp
@ -472,6 +473,12 @@ codegen_function:
; "jmp .body\n"
; ".body:\n"
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel BODY]
mov rdx, BODY_LEN
call vec_extend
mov rdi, [rsp] ; ctx
lea rsi, [rsp + 24] ; &function_ctx
@ -491,7 +498,7 @@ codegen_function:
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 24] ; &function_ctx
mov rdx, 0 ; push = false
mov rdx, 1 ; pop = true
call codegen_push_pop_dirtied_registers
; "add rsp, {current_stack_size}\n"
@ -508,13 +515,6 @@ codegen_function:
mov rcx, 10 ; radix
call int_to_str2
; ; push "\n"
; mov rdi, [rsp] ; ctx
; lea rdi, [rdi + 8] ; &ctx.text
; lea rsi, [rel COLON_NL]
; inc rsi
; call vec_push
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rax + rdx]
@ -523,11 +523,19 @@ codegen_function:
mov rdx, rdx ; length from int_to_str2
call vec_extend
; push "\n"
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel COLON_NL]
inc rsi
call vec_push
; "pop rbp\n"
; "ret\n"
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel RET_NL]
mov rdx, RET_NL_LEN
lea rsi, [rel EPILOGUE]
mov rdx, EPILOGUE_LEN
call vec_extend
; ".prologue:\n"
@ -551,13 +559,6 @@ codegen_function:
mov rcx, 10 ; radix
call int_to_str2
; ; push "\n"
; mov rdi, [rsp] ; ctx
; lea rdi, [rdi + 8] ; &ctx.text
; lea rsi, [rel COLON_NL]
; inc rsi
; call vec_push
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rax + rdx]
@ -566,11 +567,18 @@ codegen_function:
mov rdx, rdx ; length from int_to_str2
call vec_extend
; push "\n"
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel COLON_NL]
inc rsi
call vec_push
; "push{dirtied registers}\n"
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 24] ; &function_ctx
mov rdx, 1 ; push = false
mov rdx, 0 ; pop = false
call codegen_push_pop_dirtied_registers
; "jmp .args\n"
@ -594,7 +602,7 @@ codegen_function:
;; rdi: *text
;; rsi: &function_ctx
;; rdx: push: bool
;; rdx: pop: bool
codegen_push_pop_dirtied_registers:
push rbp
mov rbp, rsp