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

yippie?

Browse source
This commit is contained in:
janis 2025-11-03 15:16:18 +01:00
parent 8508018b0a
commit 52cfc7dc54
Signed by: janis
SSH key fingerprint: SHA256:bB1qbbqmDXZNT0KKD5c2Dfjg53JGhj7B3CFcLIzSqq8
2 changed files with 176 additions and 22 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

196
lang/src/codegen.asm
View file

@ -254,8 +254,8 @@ stackvar_cmp:
push rbp
mov rbp, rsp
mov rax, [rdi + 0] ; a.index
mov rcx, [rsi + 0] ; b.index
mov rax, [rsi + 0] ; a.index
mov rcx, [rdx + 0] ; b.index
cmp rax, 0
jl .less
jg .greater
@ -624,8 +624,8 @@ codegen_function:
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 104] ; src
lea rdx, [rsp + 80] ; dst
lea rsi, [rsp + 80] ; dst
lea rdx, [rsp + 104] ; src
call codegen_move_dst_src
inc r14
@ -1368,11 +1368,178 @@ codegen_expr:
jmp .done
.place_to_value:
.assignment:
; codegen inner expr
mov rdi, [rsp] ; ctx
mov rsi, [rsp + 8] ; &function_ctx
mov rdx, [rax + 8] ; AstNode.data
call codegen_expr
mov [rsp + 16], rax
mov [rsp + 24], rdx
cmp al, OPERAND_RBP_OFFSET
je .ptv_rbp_offset
cmp al, OPERAND_ADDRESS_PLACE
je .ptv_address_place
cmp al, OPERAND_REGISTER_PLACE
je .ptv_register_place
cmp al, OPERAND_RBP_PLACE
je .ptv_rbp_place
.ptv_rbp_offset:
mov byte [rsp + 16], OPERAND_RBP_PVALUE
jmp .ptv_done
.ptv_address_place:
mov byte [rsp + 16], OPERAND_ADDRESS_VALUE
jmp .ptv_done
.ptv_register_place:
mov rax, [rsp + 16]
mov rdx, [rsp + 24]
mov [rsp + 32], rax
mov [rsp + 40], rdx
; mov rdx, [rdx]
mov byte [rsp + 16], OPERAND_REGISTER
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 16] ; dst
lea rdx, [rsp + 32] ; src
mov rcx, 'mov '
call codegen_binary_op_unchecked
jmp .ptv_done
.ptv_rbp_place:
; mov rax, [src]
; mov rax, [rax]
; mov [src], rax
mov rdi, [rsp]
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel OPERAND_RAX] ; dst
lea rdx, [rsp + 16] ; src
call codegen_move_dst_src
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rel OPERAND_RAX] ; rax
lea rdx, [rel OPERAND_RAX_P] ; [rax]
mov byte [rsp + 16], OPERAND_RBP_VALUE
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 16] ; dst
lea rdx, [rel OPERAND_RAX_P] ; rax
jmp .ptv_done
.ptv_done:
mov rax, [rsp + 16]
mov rdx, [rsp + 24]
jmp .done
.value_to_place:
; codegen inner expr
mov rdi, [rsp] ; ctx
mov rsi, [rsp + 8] ; &function_ctx
mov rdx, [rax + 8] ; AstNode.data
call codegen_expr
mov [rsp + 16], rax
mov [rsp + 24], rdx
cmp al, OPERAND_RBP_PVALUE
je .vtp_rbp_pvalue
cmp al, OPERAND_ADDRESS_VALUE
je .vtp_address_value
cmp al, OPERAND_LAST_VALUE
jg .panic
; dst = allocate_place
mov rdi, [rsp + 8] ; &function_ctx
movzx rsi, word [rsp + 16 + 2] ; width
call codegen_allocate_place
mov [rsp + 32], rax
mov [rsp + 40], rdx
; mov dst, src
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 32] ; dst
lea rdx, [rsp + 16] ; src
call codegen_move_dst_src
; free src
mov rdi, [rsp + 8] ; &function_ctx
lea rsi, [rsp + 16] ; src
call codegen_free_operand
; return dst
mov rax, [rsp + 32]
mov rdx, [rsp + 40]
jmp .done
.vtp_rbp_pvalue:
mov byte [rsp + 16], OPERAND_RBP_OFFSET
jmp .vtp_done
.vtp_address_value:
mov byte [rsp + 16], OPERAND_ADDRESS_PLACE
jmp .vtp_done
.vtp_done:
mov rax, [rsp + 16]
mov rdx, [rsp + 24]
jmp .done
.assignment:
mov rcx, [rax + 8] ; AstNode.data
mov rdx, [rax + 16] ; AstNode.extra
mov [rsp + 16], rcx ; dst
mov [rsp + 24], rdx ; src
; codegen src
mov rdi, [rsp] ; ctx
mov rsi, [rsp + 8] ; &function_ctx
call codegen_expr
mov [rsp + 32], rax ; src
mov [rsp + 40], rdx
mov rdi, [rsp] ; ctx
mov rsi, [rsp + 8] ; &function_ctx
mov rdx, [rsp + 16] ; dst
call codegen_expr
mov [rsp + 48], rax ; dst
mov [rsp + 56], rdx
; mov dst, src
mov rdi, [rsp] ; ctx
lea rdi, [rdi + 8] ; &ctx.text
lea rsi, [rsp + 48] ; dst
lea rdx, [rsp + 32] ; src
call codegen_move_dst_src
; free [dst, src]
mov rdi, [rsp + 8] ; &function_ctx
lea rsi, [rsp + 48] ; dst
call codegen_free_operand
mov rdi, [rsp + 8] ; &function_ctx
lea rsi, [rsp + 48] ; src
call codegen_free_operand
xor rax, rax
xor rdx, rdx
jmp .done
.deref:
.address_of:
; TODO
; codegen inner expr
mov rdi, [rsp] ; ctx
mov rsi, [rsp + 8] ; &function_ctx
mov rdx, [rax + 8] ; AstNode.data
call codegen_expr
jmp .done
.done:
add rsp, 80
@ -1410,9 +1577,9 @@ section .rdata
OPERAND_RBP_OFFSET equ 6 ; e.g. mov [rbp - 8], rax or lea rax, [rbp - 8]
; Address of a static-slot containing a Value
OPERAND_ADDRESS_PLACE equ 7 ; e.g. lea [rel OPERAND_ADDRESS]
; Register containing an address (Place)
; Register containing an address pointing at a Value (Place)
OPERAND_REGISTER_PLACE equ 8 ; e.g. [rax]
; RBP-relative stack-slot containing an address (Place)
; RBP-relative stack-slot containing an address pointing at a Value (Place)
OPERAND_RBP_PLACE equ 9 ; e.g. mov rax, [rbp - 8]; mov [rax], src
OPERAND_LAST_PLACE equ 10 ; operand kinds > this are not memory operands
; Immediate constant Value
@ -1537,25 +1704,12 @@ codegen_write_operand:
call vec_push
; if op.kind == OPERAND_RBP_OFFSET
mov rax, [rsp + 8] ; op
mov al, byte [rax + 0] ; Operand.kind
cmp al, OPERAND_RBP_OFFSET
jne .rsp_offset_write
mov qword [rsp + 16], 'rbp '
mov rdi, [rsp] ; *text
lea rsi, [rsp + 16]
mov rdx, 4
call vec_extend
jmp .check_sign
.rsp_offset_write:
mov qword [rsp + 16], 'rsp '
mov rdi, [rsp] ; *text
lea rsi, [rsp + 16]
mov rdx, 4
call vec_extend
.check_sign:
; if value >= 0
mov rax, [rsp + 8] ; op
mov rdi, [rax + 8] ; op.value

2
lang/tests/codegen.rs
View file

@ -93,7 +93,7 @@ fn main() {
print_ast(
b"fn main(a: u32) -> void {
return 2 * 3 + 4 * 5;
return 2 * 3 + 4 * 5 + a;
}",
|ast| unsafe { parse_func(ast) },
);