Enable syscalls and do not preload buffer functions.

Author: AsynchronousAI

.gitignore

@@ -2,10 +2,9 @@
 *.elf
 reasm
 .DS_Store
-main.luau
 bench
 sourcemap.json
 *.c
-main
+main.*
 platform.info
 profile.*

compiler/abstractions.go

@@ -69,16 +69,26 @@ func move(w *OutputWriter, command AssemblyCommand) {
 	src := irArgExpr(w, command.Arguments[1])
 	Emit(w, IRStmtAssign(dst, src))
 }
+func ecall(w *OutputWriter, command AssemblyCommand) {
+	a7 := regVarName(baseRegs["x17"])
+	WriteIndentedString(w, "local syscall_fn = syscalls[%s]\n", a7)
+	WriteIndentedString(w, "if syscall_fn then\n")
+	w.Depth++
+	WriteIndentedString(w, "syscall_fn()\n")
+	w.Depth--
+	WriteIndentedString(w, "else\n")
+	w.Depth++
+	WriteIndentedString(w, "error(\"Undefined syscall: \" .. tostring(%s))\n", a7)
+	w.Depth--
+	WriteIndentedString(w, "end\n")
+}
 
 /* unimplemented */
 func ebreak(w *OutputWriter, command AssemblyCommand) {
-	log.Warn("EBREAK cannot be used (yet).")
-}
-func ecall(w *OutputWriter, command AssemblyCommand) {
-	log.Warn("ECALL cannot be used (yet).")
+	log.Warn("Breakpoints are unsupported")
 }
 func fence(w *OutputWriter, command AssemblyCommand) {
-	log.Warn("FENCE cannot be used.")
+	log.Warn("Fences are unsupported")
 }
 func nop(w *OutputWriter, command AssemblyCommand) {
 	if w.Options.Comments {

compiler/boilerplate.luau

@@ -260,16 +260,12 @@ local functions = {
     end,
 }
 
+local syscalls: {[number]: () -> ()} = {}
+
 -- Extensions
 --{extentions here}
 
 -- Localized Functions
---- buffer
-local writei8, writei16, writei32 = buffer.writei8, buffer.writei16, buffer.writei32
-local readi8, readi16, readi32 = buffer.readi8, buffer.readi16, buffer.readi32
-local writeu8, writeu16, writeu32, writestring, fill = buffer.writeu8, buffer.writeu16, buffer.writeu32, buffer.writestring, buffer.fill
-local readu8, readu16, readu32, readf32, readf64, writef32, writef64 = buffer.readu8, buffer.readu16, buffer.readu32, buffer.readf32, buffer.readf64, buffer.writef32, buffer.writef64
-
 local FUNCS: {[number]: () -> boolean} = {}
 ---- Auto generated code starts here
 --{code here}

compiler/elf.go

@@ -47,25 +47,36 @@ func ParseFromElf(f *elf.File) []AssemblyCommand {
 		return symtab[i].Value < symtab[j].Value
 	})
 
-	symStack := make([]elf.Symbol, len(symtab))
-	copy(symStack, symtab)
-
-	/* TODO: parse the actual data of the file  */
 	instructions := make([]AssemblyCommand, 0)
+
+	symIndex := 0
 	for i := 0; i < len(code); i += 4 {
 		inst, err := riscv64asm.Decode(code[i : i+4])
 		if err != nil {
 			fmt.Println("Decode error:", err)
 			continue
 		}
 
+		addr := text.Addr + uint64(i)
+
 		/* any symbols to be added before this */
-		for _, sym := range symStack {
-			symStack = symStack[:len(symStack)-1] // pop
+		for symIndex < len(symtab) && symtab[symIndex].Value < addr {
+			symIndex++
+		}
+		for symIndex < len(symtab) && symtab[symIndex].Value == addr {
+			instructions = append(instructions, AssemblyCommand{
+				Type:      Label,
+				Name:      symtab[symIndex].Name,
+				Arguments: nil,
+			})
+			symIndex++
+		}
 
+		addrLabel := fmt.Sprintf("%d", addr)
+		if len(instructions) == 0 || instructions[len(instructions)-1].Name != addrLabel {
 			instructions = append(instructions, AssemblyCommand{
 				Type:      Label,
-				Name:      sym.Name,
+				Name:      addrLabel,
 				Arguments: nil,
 			})
 		}

compiler/ir.go

@@ -61,22 +61,22 @@ const (
 	MATH_FLOOR = "math.floor"
 	MATH_CEIL  = "math.ceil"
 
-	BUFFER_READI8   = "readi8"
-	BUFFER_READI16  = "readi16"
-	BUFFER_READI32  = "readi32"
-	BUFFER_READU8   = "readu8"
-	BUFFER_READU16  = "readu16"
-	BUFFER_READU32  = "readu32"
-	BUFFER_READF32  = "readf32"
-	BUFFER_READF64  = "readf64"
-	BUFFER_WRITEI8  = "writei8"
-	BUFFER_WRITEI16 = "writei16"
-	BUFFER_WRITEI32 = "writei32"
-	BUFFER_WRITEU8  = "writeu8"
-	BUFFER_WRITEF32 = "writef32"
-	BUFFER_WRITEF64 = "writef64"
-	BUFFER_WRITESTR = "writestring"
-	BUFFER_FILL     = "fill"
+	BUFFER_READI8   = "buffer.readi8"
+	BUFFER_READI16  = "buffer.readi16"
+	BUFFER_READI32  = "buffer.readi32"
+	BUFFER_READU8   = "buffer.readu8"
+	BUFFER_READU16  = "buffer.readu16"
+	BUFFER_READU32  = "buffer.readu32"
+	BUFFER_READF32  = "buffer.readf32"
+	BUFFER_READF64  = "buffer.readf64"
+	BUFFER_WRITEI8  = "buffer.writei8"
+	BUFFER_WRITEI16 = "buffer.writei16"
+	BUFFER_WRITEI32 = "buffer.writei32"
+	BUFFER_WRITEU8  = "buffer.writeu8"
+	BUFFER_WRITEF32 = "buffer.writef32"
+	BUFFER_WRITEF64 = "buffer.writef64"
+	BUFFER_WRITESTR = "buffer.writestring"
+	BUFFER_FILL     = "buffer.fill"
 	BUFFER_LEN      = "buffer.len"
 
 	RT_I32          = "i32"