Compare commits
No commits in common. "e79af0192535d7f2a9513bd2f700c873f6e57f6f" and "56354237c63066ef40ecfc5606e0ac5266e55eeb" have entirely different histories.
e79af01925
...
56354237c6
509
lang/src/ast.asm
509
lang/src/ast.asm
|
|
@ -68,7 +68,6 @@ global parse_statement
|
|||
global parse_block
|
||||
global ast_build_symtable
|
||||
global ast_walk_for_each
|
||||
global ast_resolve_var_refs
|
||||
|
||||
;; start very simple, with only functions and addition
|
||||
;; ```rust
|
||||
|
|
@ -83,7 +82,6 @@ global ast_resolve_var_refs
|
|||
;; kind: u8,
|
||||
;; data: *const (),
|
||||
;; extra: usize,
|
||||
;; span: u64,
|
||||
;; }
|
||||
;;
|
||||
;; struct AstArgument {
|
||||
|
|
@ -103,7 +101,6 @@ global ast_resolve_var_refs
|
|||
parse_func:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
sub rsp, 8 ; span
|
||||
push rdi
|
||||
|
||||
; start-structs
|
||||
|
|
@ -117,21 +114,17 @@ parse_func:
|
|||
; }
|
||||
; end-structs
|
||||
|
||||
; span: u64 [64..72]
|
||||
; ast: *mut Ast [56..64]
|
||||
; body: u64 [48..56]
|
||||
; return_type: Type [32..48]
|
||||
; args_len: usize [24..32]
|
||||
; args_ptr: *const u64 [16..24]
|
||||
; name_len: usize [8..16]
|
||||
; name: *const u8 [0..8]
|
||||
sub rsp, 56
|
||||
; name: *const u8 [0..8]
|
||||
; name_len: usize [8..16]
|
||||
; args_ptr: *const u64 [16..24]
|
||||
; args_len: usize [24..32]
|
||||
; return_type: Type [32..48]
|
||||
; body: u64 [48..56]
|
||||
; ast: *mut Ast [56..64]
|
||||
mov qword [rsp + 16], 8 ; <*u64>::dangling()
|
||||
mov qword [rsp + 24], 0 ; args_len
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 64], rax ; span
|
||||
|
||||
mov dil, TOKEN_FN
|
||||
call unwrap_token
|
||||
mov dil, TOKEN_IDENT
|
||||
|
|
@ -168,11 +161,9 @@ parse_func:
|
|||
mov rdx, 56 ; size_of::<AstFunction>
|
||||
call memcpy
|
||||
|
||||
mov byte [rsp], AST_FUNCTION ; AstNode.kind
|
||||
mov [rsp + 8], rdi ; AstNode.data
|
||||
mov qword [rsp + 16], 0 ; AstNode.extra
|
||||
mov rdi, [rsp + 64] ; span
|
||||
mov [rsp + 24], rdi ; AstNode.span
|
||||
mov byte [rsp], AST_FUNCTION ; kind
|
||||
mov [rsp + 8], rdi ; data
|
||||
mov qword [rsp + 16], 0 ; extra
|
||||
mov rdi, [rsp + 56] ; Ast
|
||||
lea rsi, [rsp] ; &AstNode
|
||||
call vec_push
|
||||
|
|
@ -181,7 +172,6 @@ parse_func:
|
|||
dec rax
|
||||
add rsp, 56
|
||||
pop rdi
|
||||
add rsp, 8
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -199,7 +189,6 @@ parse_func:
|
|||
parse_args:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
; span: u64 [80..88]
|
||||
; vec: [40..80]
|
||||
; argument: AstArgument { [8..40]
|
||||
; name: *const u8 [8..16]
|
||||
|
|
@ -207,7 +196,7 @@ parse_args:
|
|||
; arg_type: Type [24..40]
|
||||
; }
|
||||
; ast [0..8]
|
||||
sub rsp, 88
|
||||
sub rsp, 80
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
lea rdi, [rsp + 40] ; vec
|
||||
|
|
@ -221,10 +210,6 @@ parse_args:
|
|||
call expect_token
|
||||
test rax, rax
|
||||
jnz .done_args
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 80], rax ; span
|
||||
|
||||
mov dil, TOKEN_IDENT
|
||||
call unwrap_token
|
||||
mov [rsp + 8], rax ; AstArgument.name
|
||||
|
|
@ -247,8 +232,6 @@ parse_args:
|
|||
mov qword [rsp + 8], AST_ARG ; AstNode.kind
|
||||
mov [rsp + 16], rdi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
mov rdi, [rsp + 80] ; span
|
||||
mov [rsp + 32], rdi ; AstNode.span
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
|
|
@ -273,7 +256,7 @@ parse_args:
|
|||
.done_args:
|
||||
mov rax, [rsp + 40] ; args_ptr
|
||||
mov rdx, [rsp + 48] ; args_len
|
||||
add rsp, 88
|
||||
add rsp, 80
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -334,7 +317,7 @@ parse_number:
|
|||
parse_primary_expr:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
sub rsp, 40
|
||||
sub rsp, 32
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
; start-structs
|
||||
|
|
@ -345,9 +328,6 @@ parse_primary_expr:
|
|||
; }
|
||||
; end-structs
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 32], rax ; span
|
||||
|
||||
mov dil, TOKEN_NUMBER
|
||||
call expect_token
|
||||
test rax, rax
|
||||
|
|
@ -362,7 +342,7 @@ parse_primary_expr:
|
|||
jnz .var_ref
|
||||
jmp .panic
|
||||
.var_ref:
|
||||
mov qword [rsp + 8], -1 ; AstVarRef.resolved
|
||||
mov qword [rsp + 8], 0 ; AstVarRef.resolved
|
||||
mov [rsp + 16], rax ; AstVarRef.name
|
||||
mov [rsp + 24], rdx ; AstVarRef.name_len
|
||||
|
||||
|
|
@ -377,7 +357,6 @@ parse_primary_expr:
|
|||
mov qword [rsp + 8], AST_VAR_REF ; AstNode.kind
|
||||
mov [rsp + 16], rdi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
; mov [rsp + 32], span ; AstNode.span
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
call vec_push
|
||||
|
|
@ -391,10 +370,9 @@ parse_primary_expr:
|
|||
mov rsi, rdx ; lexeme len
|
||||
call parse_number
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov byte [rsp + 8], AST_NUMBER ; AstNode.kind
|
||||
mov [rsp + 16], rax ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
; mov [rsp + 32], [rsp + 32] ; AstNode.span
|
||||
mov byte [rsp + 8], AST_NUMBER ; kind
|
||||
mov [rsp + 16], rax ; data
|
||||
mov qword [rsp + 24], 0 ; extra
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
call vec_push
|
||||
mov rdi, [rsp] ; Ast
|
||||
|
|
@ -410,7 +388,7 @@ parse_primary_expr:
|
|||
call unwrap_token
|
||||
mov rax, [rsp + 8] ; expr
|
||||
.epilogue:
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
pop rbp
|
||||
ret
|
||||
.panic:
|
||||
|
|
@ -434,8 +412,6 @@ parse_binary_expr:
|
|||
; end-structs
|
||||
|
||||
sub rsp, 64
|
||||
; span: u64 [56..64]
|
||||
; AstNode [32..64]
|
||||
; lexeme: Lexeme [32..56]
|
||||
; right: u64 [24..32]
|
||||
; right_placeness: u8 [20..21]
|
||||
|
|
@ -450,9 +426,6 @@ parse_binary_expr:
|
|||
mov byte [rsp + 17], sil ; upper_precedence
|
||||
mov byte [rsp + 16], 0
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 56], rax ; span
|
||||
|
||||
call parse_prefix_expr
|
||||
mov [rsp + 8], rax ; left
|
||||
mov [rsp + 19], dl ; left_placeness
|
||||
|
|
@ -537,7 +510,6 @@ parse_binary_expr:
|
|||
mov byte [rsp + 32], AST_BINARY_OP ; AstNode.kind
|
||||
mov [rsp + 40], rax ; AstNode.data
|
||||
mov qword [rsp + 48], 0 ; AstNode.extra
|
||||
; mov [rsp + 56], [rsp + 56] ; AstNode.span
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 32] ; &AstNode
|
||||
call vec_push
|
||||
|
|
@ -575,13 +547,10 @@ parse_statement:
|
|||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
; AstNode [8..40]
|
||||
; Ast [0..8]
|
||||
sub rsp, 40
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 32], rax ; AstNode.span
|
||||
; Ast [24..32]
|
||||
; AstNode [0..24]
|
||||
sub rsp, 32
|
||||
mov [rsp + 24], rdi ; Ast
|
||||
|
||||
mov dil, TOKEN_RETURN
|
||||
call expect_token
|
||||
|
|
@ -594,21 +563,21 @@ parse_statement:
|
|||
jmp .panic
|
||||
|
||||
.let:
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rdi, [rsp + 24] ; Ast
|
||||
call ast_parse_let
|
||||
mov [rsp], rax ; statement
|
||||
jmp .semi
|
||||
|
||||
.return:
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rdi, [rsp + 24] ; Ast
|
||||
call parse_expr
|
||||
mov byte [rsp + 8], AST_RETURN_STATEMENT ; AstNode.kind
|
||||
mov [rsp + 16], rax ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
mov byte [rsp], AST_RETURN_STATEMENT ; kind
|
||||
mov [rsp + 8], rax ; data
|
||||
mov qword [rsp + 16], 0 ; extra
|
||||
mov rdi, [rsp + 24] ; Ast
|
||||
lea rsi, [rsp] ; &AstNode
|
||||
call vec_push
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rdi, [rsp + 24] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
mov [rsp], rax
|
||||
|
|
@ -617,7 +586,7 @@ parse_statement:
|
|||
mov dil, TOKEN_SEMI
|
||||
call unwrap_token
|
||||
mov rax, [rsp] ; expression
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
pop rbp
|
||||
ret
|
||||
.panic:
|
||||
|
|
@ -629,16 +598,12 @@ parse_block:
|
|||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
; span: u64 [64..72]
|
||||
; Ast: *mut Ast [56..64]
|
||||
; statements: Vec<Statement> [8..56]
|
||||
; statement: u64 [0..8]
|
||||
sub rsp, 72
|
||||
sub rsp, 64
|
||||
mov [rsp + 56], rdi ; Ast
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 64], rax ; span
|
||||
|
||||
mov dil, TOKEN_LBRACE
|
||||
call unwrap_token
|
||||
|
||||
|
|
@ -667,15 +632,13 @@ parse_block:
|
|||
jmp .loop
|
||||
.done:
|
||||
mov rdi, [rsp + 56] ; Ast
|
||||
mov qword [rsp], AST_BLOCK ; AstNode.kind
|
||||
mov rsi, [rsp + 64] ; span
|
||||
mov [rsp + 24], rsi ; AstNode.span
|
||||
mov qword [rsp], AST_BLOCK ; kind
|
||||
lea rsi, [rsp] ; &AstNode
|
||||
call vec_push
|
||||
mov rdi, [rsp + 56] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
add rsp, 72
|
||||
add rsp, 64
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -744,13 +707,10 @@ parse_prefix_expr:
|
|||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
; AstNode [8..40]
|
||||
; ast [0..8]
|
||||
sub rsp, 40
|
||||
mov [rsp], rdi ; Ast
|
||||
sub rsp, 32
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 32], rax ; AstNode.span
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
mov dil, TOKEN_STAR
|
||||
call expect_token
|
||||
|
|
@ -803,7 +763,7 @@ parse_prefix_expr:
|
|||
jmp .done
|
||||
|
||||
.done:
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -813,17 +773,13 @@ parse_assignment_expr:
|
|||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
; span: u64 [32..40]
|
||||
; source [24..32]
|
||||
; dest [16..24]
|
||||
; dest_placeness [8..9]
|
||||
; ast [0..8]
|
||||
sub rsp, 40
|
||||
sub rsp, 32
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 32], rax ; span
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
call parse_binary_expr
|
||||
mov [rsp + 16], rax ; dest
|
||||
|
|
@ -840,7 +796,7 @@ parse_assignment_expr:
|
|||
mov rsi, [rsp + 16] ; dest
|
||||
movzx rdx, byte [rsp + 8] ; placeness
|
||||
call ast_value_to_place
|
||||
mov [rsp + 16], rax ; dest
|
||||
mov [rsp + 16], rax ; source
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
call parse_expr
|
||||
|
|
@ -864,7 +820,7 @@ parse_assignment_expr:
|
|||
.done:
|
||||
mov rax, [rsp + 16] ; dest
|
||||
movzx rdx, byte [rsp + 8] ; placeness
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -887,17 +843,12 @@ ast_parse_let:
|
|||
; }
|
||||
; end-structs
|
||||
|
||||
; expr: u64 [48..56]
|
||||
; *AstVarDecl [40..48]
|
||||
; AstNode [8..40]
|
||||
; AstNode [8..32]
|
||||
; AstVarDecl [8..40]
|
||||
; Ast [0..8]
|
||||
sub rsp, 56
|
||||
sub rsp, 48
|
||||
mov [rsp], rdi ; Ast
|
||||
|
||||
call tokeniser_get_cursor
|
||||
mov [rsp + 40], rax ; span
|
||||
|
||||
; skipped in parse_statement
|
||||
; mov dil, TOKEN_LET
|
||||
; call unwrap_token
|
||||
|
|
@ -923,9 +874,21 @@ ast_parse_let:
|
|||
mov rdx, 32 ; size_of::<AstVarDecl>
|
||||
call memcpy
|
||||
|
||||
mov [rsp + 40], rdi ; AstVarDecl ptr
|
||||
mov qword [rsp + 8], AST_VAR_DECL ; AstNode.kind
|
||||
mov [rsp + 16], rdi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
call vec_push
|
||||
; variable is already a place
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
mov qword [rsp + 8], AST_ASSIGNMENT ; AstNode.kind
|
||||
mov [rsp + 16], rax ; AstNode.data (variable index)
|
||||
|
||||
; parse the expression
|
||||
mov dil, TOKEN_EQUALS
|
||||
call unwrap_token
|
||||
mov rdi, [rsp] ; Ast
|
||||
|
|
@ -933,33 +896,10 @@ ast_parse_let:
|
|||
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rsi, rax ; expr
|
||||
; mov rdx, rdx ; placeness
|
||||
; mov rdx, rdx ; is_placeness
|
||||
call ast_place_to_value
|
||||
mov [rsp + 48], rax ; expr index
|
||||
|
||||
; variable is defined at this point so that the expression cannot reference it
|
||||
call tokeniser_get_cursor
|
||||
|
||||
mov rdi, [rsp + 40] ; AstVarDecl ptr
|
||||
mov qword [rsp + 8], AST_VAR_DECL ; AstNode.kind
|
||||
mov [rsp + 16], rdi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
mov [rsp + 32], rax ; AstNode.span
|
||||
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
call vec_push
|
||||
; variables are places
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
|
||||
mov qword [rsp + 8], AST_ASSIGNMENT ; AstNode.kind
|
||||
mov [rsp + 16], rax ; AstNode.data (variable index)
|
||||
|
||||
mov rax, [rsp + 48] ; expr index
|
||||
mov [rsp + 24], rax ; AstNode.extra (expr index)
|
||||
; reuse span from variable declaration
|
||||
mov rdi, [rsp] ; Ast
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
call vec_push
|
||||
|
|
@ -967,7 +907,7 @@ ast_parse_let:
|
|||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
|
||||
add rsp, 56
|
||||
add rsp, 48
|
||||
xor rdx, rdx ; placeness = false
|
||||
pop rbp
|
||||
ret
|
||||
|
|
@ -984,26 +924,17 @@ ast_value_to_place:
|
|||
mov rax, rsi
|
||||
je .done
|
||||
; create new AST node
|
||||
sub rsp, 40
|
||||
mov [rsp], rdi ; Ast
|
||||
mov [rsp + 8], rsi ; node_index
|
||||
|
||||
; load original node's span
|
||||
call vec_get
|
||||
mov rdi, [rax + 24] ; AstNode.span
|
||||
mov [rsp + 32], rdi ; AstNode.span
|
||||
|
||||
mov rsi, [rsp + 8] ; node_index
|
||||
mov byte [rsp + 8], AST_VALUE_TO_PLACE ; AstNode.kind
|
||||
mov [rsp + 16], rsi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
sub rsp, 32
|
||||
mov [rsp], rdi
|
||||
mov byte [rsp + 8], AST_VALUE_TO_PLACE ; kind
|
||||
mov [rsp + 16], rsi ; data
|
||||
mov qword [rsp + 24], 0 ; extra
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
mov rdi, [rsp] ; Ast
|
||||
call vec_push
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
.done:
|
||||
pop rbp
|
||||
ret
|
||||
|
|
@ -1020,26 +951,17 @@ ast_place_to_value:
|
|||
mov rax, rsi
|
||||
je .done
|
||||
; create new AST node
|
||||
sub rsp, 40
|
||||
mov [rsp], rdi ; Ast
|
||||
mov [rsp + 8], rsi ; node_index
|
||||
|
||||
; load original node's span
|
||||
call vec_get
|
||||
mov rdi, [rax + 24] ; AstNode.span
|
||||
mov [rsp + 32], rdi ; AstNode.span
|
||||
|
||||
mov rsi, [rsp + 8] ; node_index
|
||||
mov byte [rsp + 8], AST_PLACE_TO_VALUE ; AstNode.kind
|
||||
mov [rsp + 16], rsi ; AstNode.data
|
||||
mov qword [rsp + 24], 0 ; AstNode.extra
|
||||
sub rsp, 32
|
||||
mov [rsp], rdi
|
||||
mov byte [rsp + 8], AST_PLACE_TO_VALUE ; kind
|
||||
mov [rsp + 16], rsi ; data
|
||||
mov qword [rsp + 24], 0 ; extra
|
||||
lea rsi, [rsp + 8] ; &AstNode
|
||||
mov rdi, [rsp] ; Ast
|
||||
call vec_push
|
||||
mov rdi, [rsp] ; Ast
|
||||
mov rax, [rdi + 8] ; Ast.nodes.len()
|
||||
dec rax
|
||||
add rsp, 40
|
||||
add rsp, 32
|
||||
.done:
|
||||
pop rbp
|
||||
ret
|
||||
|
|
@ -1054,21 +976,21 @@ symkey_cmp:
|
|||
mov rbp, rsp
|
||||
push rbx
|
||||
|
||||
sub rsp, 16
|
||||
mov [rsp], rsi
|
||||
mov [rsp + 8], rdx
|
||||
|
||||
mov al, byte [rsi] ; a.kind
|
||||
mov bl, byte [rdx] ; b.kind
|
||||
cmp al, bl
|
||||
jl .a_less
|
||||
jg .a_greater
|
||||
|
||||
mov rax, [rsi + 8] ; a.scope_index
|
||||
mov rbx, [rdx + 8] ; b.scope_index
|
||||
cmp rax, rbx
|
||||
jl .a_less
|
||||
jg .a_greater
|
||||
mov rax, [rsi + 16] ; a.span
|
||||
mov rbx, [rdx + 16] ; b.span
|
||||
cmp rax, rbx
|
||||
jl .a_less
|
||||
jg .a_greater
|
||||
|
||||
mov rdi, [rsi + 24] ; a.ident
|
||||
mov rsi, [rsi + 32] ; a.ident_len
|
||||
|
|
@ -1078,15 +1000,6 @@ symkey_cmp:
|
|||
cmp rax, 0
|
||||
jl .a_less
|
||||
jg .a_greater
|
||||
|
||||
mov rsi, [rsp]
|
||||
mov rdx, [rsp + 8]
|
||||
mov rax, [rsi + 16] ; a.span
|
||||
mov rbx, [rdx + 16] ; b.span
|
||||
cmp rax, rbx
|
||||
jl .a_less
|
||||
jg .a_greater
|
||||
|
||||
xor rax, rax
|
||||
jmp .epilogue
|
||||
.a_less:
|
||||
|
|
@ -1095,7 +1008,6 @@ symkey_cmp:
|
|||
.a_greater:
|
||||
mov rax, 1
|
||||
.epilogue:
|
||||
add rsp, 16
|
||||
pop rbx
|
||||
pop rbp
|
||||
ret
|
||||
|
|
@ -1121,17 +1033,23 @@ ast_build_symtable:
|
|||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
; BuildSymtableCtx [24..64]
|
||||
; BuildSymtableCtx [24..104]
|
||||
; *SymbolTable [16..24]
|
||||
; root_index [8..16]
|
||||
; Ast [0..8]
|
||||
sub rsp, 64
|
||||
sub rsp, 104
|
||||
mov [rsp], rdi ; Ast
|
||||
mov [rsp + 8], rsi ; root_index
|
||||
mov [rsp + 16], rdx ; *SymbolTable
|
||||
|
||||
; initialise scope_stack and symtable vecs
|
||||
lea rdi, [rsp + 24] ; &BuildSymtableCtx.symtable
|
||||
lea rdi, [rsp + 24] ; &BuildSymtableCtx.scope_stack
|
||||
mov rsi, 8 ; size of u64
|
||||
mov rdx, 0 ; drop = None
|
||||
mov rcx, 128 ; capacity
|
||||
call vec_init_with
|
||||
|
||||
lea rdi, [rsp + 24 + 40] ; &BuildSymtableCtx.symtable
|
||||
mov rsi, 56 ; size_of::<SymEntry>
|
||||
mov rdx, 0 ; drop = None
|
||||
mov rcx, 128 ; capacity
|
||||
|
|
@ -1146,11 +1064,11 @@ ast_build_symtable:
|
|||
; memcpy symtable out
|
||||
|
||||
mov rdi, [rsp + 16] ; *SymbolTable
|
||||
lea rsi, [rsp + 24] ; &BuildSymtableCtx.symtable
|
||||
lea rsi, [rsp + 24 + 40] ; &BuildSymtableCtx.symtable
|
||||
mov rdx, 40 ; size_of::<Vec<SymEntry>>
|
||||
call memcpy
|
||||
|
||||
add rsp, 64
|
||||
add rsp, 104
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
|
|
@ -1200,6 +1118,7 @@ ast_build_symtable:
|
|||
;;
|
||||
;; #start-structs
|
||||
;; struct BuildSymtableCtx {
|
||||
;; scope_stack: Vec<u64>,
|
||||
;; symtable: Vec<SymEntry>,
|
||||
;; }
|
||||
;; #end-structs
|
||||
|
|
@ -1210,24 +1129,21 @@ ast_build_symtable:
|
|||
;; rdi: Ctx
|
||||
;; rsi: Ast
|
||||
;; rdx: index
|
||||
;; rcx: scope
|
||||
ast_build_symtable_for_each:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
push rbx
|
||||
|
||||
; scope: u64 [88..96]
|
||||
; SymEntry [32..88]
|
||||
; SymKey [32..72]
|
||||
; *AstNode [24..32]
|
||||
; index [16..24]
|
||||
; ctx [8..16]
|
||||
; ast [0..8]
|
||||
sub rsp, 96
|
||||
sub rsp, 88
|
||||
mov [rsp], rsi ; Ast
|
||||
mov [rsp + 8], rdi ; Ctx
|
||||
mov [rsp + 16], rdx ; index
|
||||
mov [rsp + 88], rcx ; scope
|
||||
|
||||
mov rdi, rsi ; Ast
|
||||
mov rsi, rdx ; index
|
||||
|
|
@ -1248,49 +1164,40 @@ ast_build_symtable_for_each:
|
|||
mov byte [rsp + 32], SYM_KEY_SCOPE ; SymKey.kind
|
||||
mov rdx, [rsp + 16] ; index
|
||||
mov qword [rsp + 40], rdx ; SymKey.scope_index
|
||||
mov rdx, [rax + 24] ; AstNode.span
|
||||
mov qword [rsp + 48], rdx ; SymKey.span
|
||||
mov qword [rsp + 56], 1 ; SymKey.ident
|
||||
mov qword [rsp + 64], 0 ; SymKey.ident_len
|
||||
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
mov rdx, [rbx + 0] ; Func.name
|
||||
mov rcx, [rbx + 8] ; Func.name_len
|
||||
|
||||
mov [rsp + 72], rdx ; SymEntry.index
|
||||
mov [rsp + 80], rcx ; SymEntry.extra
|
||||
|
||||
mov rdi, [rsp + 8] ; Ctx.symtable
|
||||
lea rsi, [rsp + 32] ; &SymEntry
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
call vec_insert_sorted
|
||||
|
||||
; add parent scope
|
||||
mov byte [rsp + 32], SYM_KEY_PARENT_SCOPE ; SymKey.kind
|
||||
mov rdx, [rsp + 16] ; index
|
||||
mov qword [rsp + 40], rdx ; SymKey.scope_index
|
||||
mov qword [rsp + 48], 0 ; SymKey.span
|
||||
mov qword [rsp + 56], 1 ; SymKey.ident
|
||||
mov qword [rsp + 64], 0 ; SymKey.ident_len
|
||||
|
||||
mov rdx, [rsp + 88] ; parent scope
|
||||
mov [rsp + 72], rdx ; SymEntry.index
|
||||
mov qword [rsp + 80], 0 ; SymEntry.extra
|
||||
; mov rbx, [rax + 16] ; AstNode.data
|
||||
; mov rdx, [rbx + 8] ; Func.name
|
||||
; mov rcx, [rbx + 16] ; Func.name_len
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
mov rdx, [rbx + 0] ; Func.name
|
||||
mov rcx, [rbx + 8] ; Func.name_len
|
||||
|
||||
mov qword [rsp + 72], rdx ; SymEntry.index
|
||||
mov qword [rsp + 80], rcx ; SymEntry.extra
|
||||
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rdi, [rdi + 40] ; Ctx.symtable
|
||||
lea rsi, [rsp + 32] ; &SymEntry
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
call vec_insert_sorted
|
||||
|
||||
; push scope index onto scope_stack
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rdi, [rdi + 0] ; Ctx.scope_stack
|
||||
lea rsi, [rsp + 16] ; &index
|
||||
call vec_push
|
||||
|
||||
jmp .done
|
||||
.var_decl:
|
||||
; insert variable entry
|
||||
mov byte [rsp + 32], SYM_KEY_VAR ; SymKey.kind
|
||||
|
||||
mov rdx, [rax + 24] ; AstNode.span
|
||||
mov qword [rsp + 48], rdx ; SymKey.span
|
||||
; TODO: set span correctly
|
||||
mov qword [rsp + 48], 0 ; SymKey.span
|
||||
|
||||
mov rbx, [rsp + 24] ; AstNode.data
|
||||
mov rbx, [rbx + 8] ; AstNode.data
|
||||
|
|
@ -1304,10 +1211,17 @@ ast_build_symtable_for_each:
|
|||
mov [rsp + 72], rdx ; SymEntry.index
|
||||
mov qword [rsp + 80], 0 ; SymEntry.extra
|
||||
|
||||
mov rdx, [rsp + 88] ; current scope
|
||||
mov [rsp + 40], rdx ; SymKey.scope_index = default
|
||||
mov qword [rsp + 40], 0 ; SymKey.scope_index = default
|
||||
lea rdx, [rsp + 40]
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
mov rsi, [rdi + 8] ; Ctx.scope_stack.len()
|
||||
dec rsi
|
||||
call vec_get_or
|
||||
mov rax, [rax] ; current scope index
|
||||
mov [rsp + 40], rax ; SymKey.scope_index = scope_stack.last_or(0)
|
||||
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rdi, [rdi + 40] ; Ctx.symtable
|
||||
lea rsi, [rsp + 32] ; &SymEntry
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
|
|
@ -1318,8 +1232,8 @@ ast_build_symtable_for_each:
|
|||
; insert variable entry
|
||||
mov byte [rsp + 32], SYM_KEY_ARG ; SymKey.kind
|
||||
|
||||
mov rdx, [rax + 24] ; AstNode.span
|
||||
mov qword [rsp + 48], rdx ; SymKey.span
|
||||
; TODO: set span correctly
|
||||
mov qword [rsp + 48], 0 ; SymKey.span
|
||||
|
||||
mov rbx, [rsp + 24] ; *AstNode
|
||||
mov rbx, [rbx + 8] ; AstNode.data
|
||||
|
|
@ -1333,18 +1247,24 @@ ast_build_symtable_for_each:
|
|||
mov [rsp + 72], rdx ; SymEntry.index
|
||||
mov qword [rsp + 80], 0 ; SymEntry.extra
|
||||
|
||||
mov rdx, [rsp + 88] ; current scope
|
||||
mov [rsp + 40], rdx ; SymKey.scope_index = default
|
||||
mov qword [rsp + 40], 0 ; SymKey.scope_index = default
|
||||
lea rdx, [rsp + 40]
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
mov rsi, [rdi + 8] ; Ctx.scope_stack.len()
|
||||
dec rsi
|
||||
call vec_get_or
|
||||
mov rax, [rax] ; current scope index
|
||||
mov [rsp + 40], rax ; SymKey.scope_index = scope_stack.last_or(0)
|
||||
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rdi, [rdi + 40] ; Ctx.symtable
|
||||
lea rsi, [rsp + 32] ; &SymEntry
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
call vec_insert_sorted
|
||||
jmp .done
|
||||
|
||||
.done:
|
||||
add rsp, 96
|
||||
add rsp, 88
|
||||
pop rbx
|
||||
pop rbp
|
||||
ret
|
||||
|
|
@ -1353,39 +1273,26 @@ ast_build_symtable_for_each:
|
|||
;; rsi: start_index
|
||||
;; rdx: ctx
|
||||
;; rcx: for_each
|
||||
;; define-fn: fn ast_walk_for_each(ast: *mut Ast, start_index: u64, ctx: *mut (), for_each: unsafe extern "C" fn(ctx: *mut (), *mut Ast, node_index: u64, scope: u64))
|
||||
;; define-fn: fn ast_walk_for_each(ast: *mut Ast, start_index: u64, ctx: *mut (), for_each: unsafe extern "C" fn(ctx: *mut (), *mut Ast, node_index: u64))
|
||||
ast_walk_for_each:
|
||||
push rbp
|
||||
push r15
|
||||
push r14
|
||||
push rbx
|
||||
|
||||
; INVALID_SCOPE [48..56]
|
||||
; current_index [40..48]
|
||||
; *current_scope [32..40]
|
||||
; current_node_ptr [24..32]
|
||||
; current_index [24..32]
|
||||
; for_each [16..24]
|
||||
; ctx [8..16]
|
||||
; ast [0..8]
|
||||
sub rsp, 56
|
||||
sub rsp, 32
|
||||
mov [rsp], rdi ; Ast
|
||||
mov [rsp + 8], rdx ; ctx
|
||||
mov [rsp + 16], rcx ; for_each
|
||||
mov qword [rsp + 24], 0 ; current_node_ptr
|
||||
lea rdi, [rsp + 48] ; &INVALID_SCOPE
|
||||
mov [rsp + 32], rdi ; current_scope
|
||||
mov [rsp + 40], rsi ; current_index
|
||||
mov qword [rsp + 48], -1 ; INVALID_SCOPE
|
||||
mov [rsp + 24], rsi ; current_node_ptr
|
||||
|
||||
mov rbp, rsp
|
||||
push rsi
|
||||
|
||||
; `current_scope` points to the index of the current scope on the stack.
|
||||
; When we enter a new scope, we push `current_scope` onto the stack, then
|
||||
; update it to point to the new scope index.
|
||||
; When `rsp` is equal to `current_scope`, we need to additionally pop into
|
||||
; `current_scope` after popping the current index.
|
||||
|
||||
.loop:
|
||||
cmp rsp, rbp
|
||||
jge .done
|
||||
|
|
@ -1393,8 +1300,6 @@ ast_walk_for_each:
|
|||
mov rdi, [rbp + 8] ; ctx
|
||||
mov rsi, [rbp] ; Ast
|
||||
mov rdx, [rsp] ; current_index
|
||||
mov rcx, [rbp + 32] ; current_scope
|
||||
mov rcx, [rcx] ; current_scope value
|
||||
mov rax, [rbp + 16] ; for_each
|
||||
|
||||
; align stack to 16 bytes before call
|
||||
|
|
@ -1408,8 +1313,6 @@ ast_walk_for_each:
|
|||
; get current_node_ptr
|
||||
mov rdi, [rbp] ; Ast
|
||||
pop rsi ; current_index
|
||||
mov [rbp + 40], rsi ; update current_index
|
||||
|
||||
call vec_get
|
||||
mov [rbp + 24], rax ; current_node_ptr
|
||||
mov bl, byte [rax] ; AstNode.kind
|
||||
|
|
@ -1431,14 +1334,9 @@ ast_walk_for_each:
|
|||
je .address_of
|
||||
cmp bl, AST_RETURN_STATEMENT
|
||||
je .return_statement
|
||||
jmp .check_scope
|
||||
jmp .loop
|
||||
|
||||
.func:
|
||||
; push scope
|
||||
push qword [rbp + 32] ; scope-ptr
|
||||
push qword [rbp + 40] ; current_index
|
||||
mov [rbp + 32], rsp ; update current_scope
|
||||
|
||||
; push child indices to stack
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
|
||||
|
|
@ -1456,7 +1354,7 @@ ast_walk_for_each:
|
|||
inc r14
|
||||
jmp .arg_loop
|
||||
.arg_loop_done:
|
||||
jmp .check_scope
|
||||
jmp .loop
|
||||
|
||||
.block:
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
|
|
@ -1470,7 +1368,7 @@ ast_walk_for_each:
|
|||
push rdx ; push statement index
|
||||
jmp .stmt_loop
|
||||
.stmt_loop_done:
|
||||
jmp .check_scope
|
||||
jmp .loop
|
||||
|
||||
.binary_op:
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
|
|
@ -1478,14 +1376,14 @@ ast_walk_for_each:
|
|||
push rdx ; push right index
|
||||
mov rdx, [rbx + 0] ; left index
|
||||
push rdx ; push left index
|
||||
jmp .check_scope
|
||||
jmp .loop
|
||||
|
||||
.assignment:
|
||||
mov rbx, [rax + 8] ; AstNode.data = dest
|
||||
mov rdx, [rax + 16] ; AstNode.extra = source
|
||||
push rdx ; push source index
|
||||
push rbx ; push dest index
|
||||
jmp .check_scope
|
||||
jmp .loop
|
||||
|
||||
.value_to_place:
|
||||
.place_to_value:
|
||||
|
|
@ -1494,145 +1392,30 @@ ast_walk_for_each:
|
|||
.return_statement:
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
push rbx ; push inner expr index
|
||||
jmp .check_scope
|
||||
|
||||
.check_scope:
|
||||
cmp rsp, [rbp + 32] ; current_scope
|
||||
je .pop_scope
|
||||
jmp .loop
|
||||
|
||||
.pop_scope:
|
||||
; pop current_scope
|
||||
; the stack may look something like this:
|
||||
; current_scope---+
|
||||
; == stack == ^ | points here
|
||||
; scope-ptr[0]-+ |
|
||||
; scope: func0 <-----+
|
||||
; stmt0 | |
|
||||
; stmt1 | |
|
||||
; scope-ptr[1] ------+
|
||||
; scope: block0 <-+
|
||||
; stmt2
|
||||
; stmt3
|
||||
; ...
|
||||
|
||||
pop rax ; scope
|
||||
pop rax ; scope_ptr
|
||||
mov [rbp + 32], rax ; update current_scope pointer
|
||||
jmp .check_scope
|
||||
|
||||
; weird alloca thing
|
||||
.done:
|
||||
add rsp, 56
|
||||
add rsp, 32
|
||||
pop rbx
|
||||
pop r14
|
||||
pop r15
|
||||
pop rbp
|
||||
ret
|
||||
|
||||
;; rdi: *mut SymbolTable
|
||||
;; rsi: *mut Ast
|
||||
;; rdx: node_index
|
||||
;; rcx: scope
|
||||
ast_resolve_var_refs_for_each:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
push rbx
|
||||
|
||||
; lower_bound [88..96]
|
||||
; scope: u64 [80..88]
|
||||
; SymEntry [24..80]
|
||||
; *AstNode [16..24]
|
||||
; *BuildSymtableCtx [8..16]
|
||||
; *Ast [0..8]
|
||||
sub rsp, 96
|
||||
mov [rsp], rsi ; Ast
|
||||
mov [rsp + 8], rdi ; Ctx
|
||||
|
||||
mov [rsp + 80], rcx ; SymKey.scope_index
|
||||
|
||||
mov rdi, rsi ; Ast
|
||||
mov rsi, rdx ; node_index
|
||||
call vec_get
|
||||
mov [rsp + 16], rax ; *AstNode
|
||||
|
||||
mov bl, byte [rax] ; AstNode.kind
|
||||
|
||||
cmp bl, AST_VAR_REF
|
||||
jne .epilogue
|
||||
|
||||
; lookup variable in symbol table
|
||||
|
||||
; binary search lower bound
|
||||
mov byte [rsp + 24 + 0], SYM_KEY_START_LOCALS ; SymKey.kind
|
||||
mov qword [rsp + 24 + 8], 0 ; SymKey.scope_index
|
||||
mov qword [rsp + 24 + 16], 0 ; SymKey.span
|
||||
mov qword [rsp + 24 + 24], 1 ; SymKey.name
|
||||
mov qword [rsp + 24 + 32], 0 ; SymKey.name_len
|
||||
|
||||
; binary search in symbol table
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rsi, [rsp + 24] ; &SymKey
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
call vec_binary_search_by
|
||||
mov [rsp + 88], rax ; lower_bound
|
||||
|
||||
; construct key
|
||||
mov byte [rsp + 24 + 0], SYM_KEY_VAR ; SymKey.kind
|
||||
mov rax, [rsp + 80] ; scope
|
||||
mov [rsp + 24 + 8], rax ; SymKey.scope_index
|
||||
mov rax, [rsp + 16] ; *AstNode
|
||||
mov rbx, [rax + 24] ; AstNode.span
|
||||
mov [rsp + 24 + 16], rbx ; SymKey.span
|
||||
mov rbx, [rax + 8] ; AstNode.data
|
||||
mov rax, [rbx + 8] ; AstVarRef.name
|
||||
mov rbx, [rbx + 16] ; AstVarRef.name_len
|
||||
mov [rsp + 24 + 24], rax ; SymKey.ident
|
||||
mov [rsp + 24 + 32], rbx ; SymKey.ident_len
|
||||
|
||||
; binary search in symbol table
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
lea rsi, [rsp + 24] ; &SymKey
|
||||
mov rdx, symkey_cmp ; cmp
|
||||
mov rcx, 0 ; cmp_ctx
|
||||
call vec_binary_search_by
|
||||
test rdx, rdx
|
||||
jz .fixup
|
||||
dec rax
|
||||
|
||||
.fixup:
|
||||
cmp rax, [rsp + 88] ; lower_bound
|
||||
jl .panic
|
||||
|
||||
mov rdi, [rsp + 8] ; *Ctx
|
||||
mov rsi, rax ; index
|
||||
call vec_get
|
||||
mov rax, [rax + 40] ; SymEntry.index
|
||||
|
||||
mov rdx, [rsp + 16] ; *AstNode
|
||||
mov rdx, [rdx + 8] ; AstNode.data
|
||||
mov [rdx + 0], rax ; AstVarRef.resolved_index
|
||||
|
||||
.epilogue:
|
||||
add rsp, 96
|
||||
pop rbx
|
||||
pop rbp
|
||||
ret
|
||||
.panic:
|
||||
call panic
|
||||
|
||||
;; rdi: Ast
|
||||
;; rsi: *mut SymbolTable
|
||||
;; rdx: root_index
|
||||
;; define-fn: fn ast_resolve_var_refs(ast: *mut Ast, ctx: *mut SymbolTable, root_index: u64)
|
||||
;; define-fn: fn ast_resolve_var_refs(ast: *mut Ast)
|
||||
ast_resolve_var_refs:
|
||||
push rbp
|
||||
mov rbp, rsp
|
||||
|
||||
xchg rsi, rdx
|
||||
mov rcx, ast_resolve_var_refs_for_each
|
||||
call ast_walk_for_each
|
||||
push r15
|
||||
push r14
|
||||
push rbx
|
||||
|
||||
.epilogue:
|
||||
add rsp, 8
|
||||
pop rbx
|
||||
pop r14
|
||||
pop r15
|
||||
pop rbp
|
||||
ret
|
||||
|
|
|
|||
|
|
@ -29,31 +29,22 @@ fn main() {
|
|||
eprintln!("Parsed expression ID: {}", expr_id);
|
||||
println!("{:#}", &ast);
|
||||
|
||||
// unsafe extern "C" fn visit_node(
|
||||
// _this: *mut (),
|
||||
// ast: *mut Ast,
|
||||
// node_id: u64,
|
||||
// scope: u64,
|
||||
// ) {
|
||||
// unsafe extern "C" fn visit_node(_this: *mut (), ast: *mut Ast, node_id: u64) {
|
||||
// let ast = unsafe { &*ast };
|
||||
// let node = ast.nodes.get(node_id as usize).unwrap();
|
||||
// eprintln!("{scope}: Visiting node {node_id}: {node}");
|
||||
// eprintln!("Visiting node {node_id}: {node}");
|
||||
// }
|
||||
|
||||
// util::defs::ast_walk_for_each(&mut ast, expr_id, core::ptr::null_mut(), visit_node);
|
||||
|
||||
let mut symtable = core::mem::MaybeUninit::<util::defs::SymbolTable>::uninit();
|
||||
util::defs::ast_build_symtable(&mut ast, expr_id, &mut symtable);
|
||||
let mut symtable = symtable.assume_init();
|
||||
let symtable = symtable.assume_init();
|
||||
use util::DisplayedSliceExt;
|
||||
println!(
|
||||
"Symbol Table: {:#?}",
|
||||
symtable.symtable.as_slice().displayed()
|
||||
);
|
||||
|
||||
util::defs::ast_resolve_var_refs(&mut ast, &mut symtable, expr_id);
|
||||
|
||||
println!("{:#}", &ast);
|
||||
};
|
||||
}
|
||||
|
||||
|
|
@ -75,14 +66,14 @@ fn main() {
|
|||
// print_ast(b"fn main() -> void { let x: u32 = 4; }", |ast| unsafe {
|
||||
// parse_func(ast)
|
||||
// });
|
||||
// print_ast(
|
||||
// b"fn main(a: u32) -> void { let x: u32 = a + 4; }",
|
||||
// |ast| unsafe { parse_func(ast) },
|
||||
// );
|
||||
print_ast(
|
||||
b"fn main(a: u32) -> void { let x: u32 = a + 4; }",
|
||||
|ast| unsafe { parse_func(ast) },
|
||||
);
|
||||
print_ast(
|
||||
b"fn main(a: u32) -> void {
|
||||
let y: u32 = a + 4;
|
||||
let y: *u32 = &y;
|
||||
let x: u32 = a + 4;
|
||||
let y: *u32 = &x;
|
||||
return *y;
|
||||
}",
|
||||
|ast| unsafe { parse_func(ast) },
|
||||
|
|
@ -119,7 +110,7 @@ impl std::fmt::Display for AstNode {
|
|||
}
|
||||
AST_VAR_REF => {
|
||||
let var_ref = unsafe { self.data.cast::<util::defs::AstVarRef>().read() };
|
||||
if var_ref.resolved != u64::MAX {
|
||||
if var_ref.resolved != 0 {
|
||||
write!(f, "VarRef({})", var_ref.resolved)
|
||||
} else {
|
||||
write!(f, "VarRef(name: {:?})", unsafe {
|
||||
|
|
@ -193,7 +184,7 @@ impl std::fmt::Display for AstNode {
|
|||
AST_VALUE_TO_PLACE => {
|
||||
write!(f, "ValueToPlace(value: {})", self.data as usize)
|
||||
}
|
||||
kind => write!(f, "UnknownNode(kind: {kind})"),
|
||||
_ => write!(f, "UnknownNode"),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
|
|
@ -15,8 +15,8 @@ unsafe extern "C" {
|
|||
pub unsafe fn ast_parse_let(ast: *mut Ast) -> (u64, bool);
|
||||
pub unsafe fn symkey_cmp(a: *const SymKey, b: *const SymKey) -> i32;
|
||||
pub unsafe fn ast_build_symtable(ast: *mut Ast, root_index: u64, symtable: *mut core::mem::MaybeUninit<SymbolTable>);
|
||||
pub unsafe fn ast_walk_for_each(ast: *mut Ast, start_index: u64, ctx: *mut (), for_each: unsafe extern "C" fn(ctx: *mut (), *mut Ast, node_index: u64, scope: u64));
|
||||
pub unsafe fn ast_resolve_var_refs(ast: *mut Ast, ctx: *mut SymbolTable, root_index: u64);
|
||||
pub unsafe fn ast_walk_for_each(ast: *mut Ast, start_index: u64, ctx: *mut (), for_each: unsafe extern "C" fn(ctx: *mut (), *mut Ast, node_index: u64));
|
||||
pub unsafe fn ast_resolve_var_refs(ast: *mut Ast);
|
||||
}
|
||||
|
||||
pub const AST_FUNCTION: u8 = 1;
|
||||
|
|
@ -106,7 +106,6 @@ pub struct AstNode {
|
|||
pub kind: u8,
|
||||
pub data: *const (),
|
||||
pub extra: usize,
|
||||
pub span: u64,
|
||||
}
|
||||
|
||||
#[repr(C)]
|
||||
|
|
|
|||
Loading…
Reference in a new issue