diff --git a/vlib/v/gen/x64/elf.v b/vlib/v/gen/x64/elf.v index 7bbf1e2d51..2853206eb4 100644 --- a/vlib/v/gen/x64/elf.v +++ b/vlib/v/gen/x64/elf.v @@ -72,7 +72,9 @@ pub fn (mut g Gen) generate_elf_header() { // address: 00070 and a half println('code_start_pos = $g.buf.len.hex()') g.code_start_pos = g.buf.len + g.debug_pos = g.buf.len g.call(PLACEHOLDER) // call main function, it's not guaranteed to be the first, we don't know its address yet + g.println('call fn main') } pub fn (mut g Gen) generate_elf_footer() { @@ -106,5 +108,5 @@ pub fn (mut g Gen) generate_elf_footer() { os.chmod(g.out_name, 0o775) // make it an executable f.write_bytes(g.buf.data, g.buf.len) f.close() - println('x64 elf binary has been successfully generated') + println('\nx64 elf binary has been successfully generated') } diff --git a/vlib/v/gen/x64/gen.v b/vlib/v/gen/x64/gen.v index dd428188bb..fd3bf3008e 100644 --- a/vlib/v/gen/x64/gen.v +++ b/vlib/v/gen/x64/gen.v @@ -5,6 +5,8 @@ module x64 import v.ast import v.util +import term +import strings pub struct Gen { out_name string @@ -20,6 +22,7 @@ mut: fn_addr map[string]i64 var_offset map[string]int // local var stack offset stack_var_pos int + debug_pos int } // string_addr map[string]i64 @@ -197,6 +200,7 @@ fn (mut g Gen) cmp_var(var_name string, val int) { offset := g.get_var_offset(var_name) g.write8(0xff - offset + 1) g.write32(val) + g.println('cmp var `$var_name` $val') } // `add DWORD [rbp-0x4], 1` @@ -205,6 +209,7 @@ fn (mut g Gen) inc_var(var_name string) { offset := g.get_var_offset(var_name) g.write8(0xff - offset + 1) g.write32(1) + g.println('inc_var `$var_name`') } // Returns the position of the address to jump to (set later). @@ -212,6 +217,7 @@ fn (mut g Gen) jne() int { g.write16(0x850f) pos := g.pos() g.write32(PLACEHOLDER) + g.println('jne') return pos } @@ -219,12 +225,14 @@ fn (mut g Gen) jge() int { g.write16(0x8d0f) pos := g.pos() g.write32(PLACEHOLDER) + g.println('jne') return pos } fn (mut g Gen) jmp(addr int) { g.write8(0xe9) g.write32(addr) // 0xffffff + g.println('jmp') } fn abs(a i64) i64 { @@ -241,12 +249,30 @@ fn (mut g Gen) jle(addr i64) { offset := 0xff - int(abs(addr - g.buf.len)) - 1 g.write8(0x7e) g.write8(offset) + g.println('jle') +} + +fn (mut g Gen) println(comment string) { + addr := g.debug_pos.hex() + // println('$g.debug_pos "$addr"') + print(term.red(strings.repeat(`0`, 6 - addr.len) + addr + ' ')) + for i := g.debug_pos; i < g.buf.len; i++ { + s := g.buf[i].hex() + if s.len == 1 { + print(term.blue('0')) + } + print(term.blue(g.buf[i].hex()) + ' ') + } + g.debug_pos = g.buf.len + print(' ' + comment) + println('') } fn (mut g Gen) jl(addr i64) { offset := 0xff - int(abs(addr - g.buf.len)) - 1 g.write8(0x7c) g.write8(offset) + g.println('jl') } fn (g &Gen) abs_to_rel_addr(addr i64) int { @@ -271,6 +297,7 @@ fn (mut g Gen) mov64(reg Register, val i64) { } } g.write64(val) + g.println('mov64 $reg, $val') } fn (mut g Gen) mov_from_reg(var_offset int, reg Register) { @@ -282,6 +309,7 @@ fn (mut g Gen) mov_from_reg(var_offset int, reg Register) { else { verror('mov_from_reg $reg') } } g.write8(0xff - var_offset + 1) + g.println('mov from reg') } fn (mut g Gen) call(addr int) { @@ -290,19 +318,22 @@ fn (mut g Gen) call(addr int) { // +5 is to get the posistion "e8 xx xx xx xx" // Not sure about the -1. rel := 0xffffffff - (g.buf.len + 5 - addr - 1) - println('call addr=$addr.hex() rel_addr=$rel.hex() pos=$g.buf.len') + // println('call addr=$addr.hex() rel_addr=$rel.hex() pos=$g.buf.len') g.write8(0xe8) g.write32(rel) + // g.println('fn call') } fn (mut g Gen) syscall() { // g.write(0x050f) g.write8(0x0f) g.write8(0x05) + g.println('syscall') } pub fn (mut g Gen) ret() { g.write8(0xc3) + g.println('ret') } pub fn (mut g Gen) push(reg Register) { @@ -318,11 +349,13 @@ pub fn (mut g Gen) push(reg Register) { else {} } */ + g.println('push $reg') } pub fn (mut g Gen) pop(reg Register) { g.write8(0x58 + reg) // TODO r8... + g.println('pop $reg') } pub fn (mut g Gen) sub32(reg Register, val int) { @@ -330,6 +363,7 @@ pub fn (mut g Gen) sub32(reg Register, val int) { g.write8(0x81) g.write8(0xe8 + reg) // TODO rax is different? g.write32(val) + g.println('sub $reg,0x$val.hex()') } pub fn (mut g Gen) add(reg Register, val int) { @@ -337,10 +371,12 @@ pub fn (mut g Gen) add(reg Register, val int) { g.write8(0x81) g.write8(0xe8 + reg) // TODO rax is different? g.write32(val) + g.println('add $reg,0x$val.hex()') } fn (mut g Gen) leave() { g.write8(0xc9) + g.println('leave') } // returns label's relative address @@ -419,6 +455,7 @@ fn (mut g Gen) mov(reg Register, val int) { } } g.write32(val) + g.println('mov $reg, $val') } fn (mut g Gen) mov_reg(a, b Register) { @@ -436,6 +473,7 @@ fn (mut g Gen) mov_rbp_rsp() { g.write8(0x48) g.write8(0x89) g.write8(0xe5) + g.println('mov rbp,rsp') } pub fn (mut g Gen) register_function_address(name string) { @@ -446,13 +484,13 @@ pub fn (mut g Gen) register_function_address(name string) { pub fn (mut g Gen) call_fn(node ast.CallExpr) { name := node.name - println('call fn $name') + // println('call fn $name') addr := g.fn_addr[name] if addr == 0 { verror('fn addr of `$name` = 0') } // Copy values to registers (calling convention) - g.mov(.eax, 0) + // g.mov(.eax, 0) for i in 0 .. node.args.len { expr := node.args[i].expr int_lit := expr as ast.IntegerLiteral @@ -462,7 +500,8 @@ pub fn (mut g Gen) call_fn(node ast.CallExpr) { verror('more than 6 args not allowed for now') } g.call(int(addr)) - println('call $name $addr') + g.println('fn call `${name}()`') + // println('call $name $addr') } fn (mut g Gen) stmt(node ast.Stmt) { @@ -480,6 +519,7 @@ fn (mut g Gen) stmt(node ast.Stmt) { ast.ForStmt { g.for_stmt(it) } + ast.Module {} ast.Return { g.gen_exit() g.ret() @@ -548,22 +588,21 @@ fn (mut g Gen) allocate_var(name string, size, initial_val int) { } } // Generate N in `[rbp-N]` - g.write8(0xff - (g.stack_var_pos + size) + 1) + n := g.stack_var_pos + size + g.write8(0xff - n + 1) g.stack_var_pos += size g.var_offset[name] = g.stack_var_pos // Generate the value assigned to the variable g.write32(initial_val) - println('allocate_var(size=$size, initial_val=$initial_val)') + // println('allocate_var(size=$size, initial_val=$initial_val)') + g.println('mov DWORD [rbp-0x$n],$initial_val (Aallocate var `$name`)') } fn (mut g Gen) assign_stmt(node ast.AssignStmt) { // `a := 1` | `a,b := 1,2` for i, ident in node.left { match node.right[0] { - ast.IntegerLiteral { - print(it.val) - g.allocate_var(ident.name, 4, it.val.int()) - } + ast.IntegerLiteral { g.allocate_var(ident.name, 4, it.val.int()) } else {} } } @@ -587,7 +626,7 @@ fn (mut g Gen) if_expr(node ast.IfExpr) { // Now that we know where we need to jump if the condition is false, update the `jne` call. // The value is the relative address, difference between current position and the location // after `jne 00 00 00 00` - println('after if g.pos=$g.pos() jneaddr=$jne_addr') + // println('after if g.pos=$g.pos() jneaddr=$jne_addr') g.write32_at(jne_addr, g.pos() - jne_addr - 4) // 4 is for "00 00 00 00" } @@ -612,18 +651,22 @@ fn (mut g Gen) for_stmt(node ast.ForStmt) { g.jmp(0xffffffff - (g.pos() + 5 - start) + 1) // Update the jump addr to current pos g.write32_at(jump_addr, g.pos() - jump_addr - 4) // 4 is for "00 00 00 00" + g.println('jpm after for') } fn (mut g Gen) fn_decl(node ast.FnDecl) { + println(term.green('\n$node.name:')) + g.stack_var_pos = 0 is_main := node.name == 'main' - println('saving addr $node.name $g.buf.len.hex()') + // println('saving addr $node.name $g.buf.len.hex()') if is_main { g.save_main_fn_addr() } else { g.register_function_address(node.name) - // g.write32(SEVENS) - g.push(.rbp) - g.mov_rbp_rsp() + } + g.push(.rbp) + g.mov_rbp_rsp() + if !is_main { g.sub32(.rsp, 0x20) } if node.args.len > 0 { @@ -642,7 +685,7 @@ fn (mut g Gen) fn_decl(node ast.FnDecl) { // g.stmts(node.stmts) if is_main { - println('end of main: gen exit') + // println('end of main: gen exit') g.gen_exit() // return } diff --git a/vlib/v/gen/x64/tests/if.vv b/vlib/v/gen/x64/tests/if.vv index e404da0875..8a4a1e00a1 100644 --- a/vlib/v/gen/x64/tests/if.vv +++ b/vlib/v/gen/x64/tests/if.vv @@ -28,24 +28,27 @@ fn loop() { } fn foo(a int) { - println('foo') + println('foo:') if a == 1 { - println('foo(1)') + println('a == 1') } a++ - if a == 1 { - println('foo(2)') + if a == 2 { + println('a == 2') + } + if a == 3 { + println('a == 3') } } fn args() { println('===args===') - args(1) - args(2) + foo(1) + foo(2) } fn main() { test() loop() - //args() + args() } diff --git a/vlib/v/gen/x64/tests/if.vv.out b/vlib/v/gen/x64/tests/if.vv.out index 20a3ba5049..4fac90b910 100644 --- a/vlib/v/gen/x64/tests/if.vv.out +++ b/vlib/v/gen/x64/tests/if.vv.out @@ -6,3 +6,10 @@ hello hello hello hello +===args=== +foo: +a == 1 +a == 2 +foo: +a == 3 +