add exit function

Author: douyixuan

compiler/compiler/compiler.go

@@ -15,6 +15,8 @@ import (
 	"errors"
 
 	"github.com/llir/llvm/ir"
+	"github.com/llir/llvm/ir/constant"
+	llvmTypes "github.com/llir/llvm/ir/types"
 	llvmValue "github.com/llir/llvm/ir/value"
 )
 
@@ -201,7 +203,6 @@ func (c *Compiler) addGlobal() {
 	global.DefinePkgType("uint8", types.U8)
 	global.DefinePkgType("int16", types.I16)
 	global.DefinePkgType("uint16", types.U16)
-	global.DefinePkgType("exit", types.I64)
 	global.DefinePkgType("int32", types.I32)
 	global.DefinePkgType("uint32", types.U32)
 	global.DefinePkgType("int64", types.I64)
@@ -374,7 +375,7 @@ func (c *Compiler) panic(block *ir.Block, message string) {
 	globMsg := c.module.NewGlobalDef(strings.NextStringName(), strings.Constant("runtime panic: "+message+"\n"))
 	globMsg.Immutable = true
 	block.NewCall(c.osFuncs.Printf.Value.(llvmValue.Named), strings.Toi8Ptr(block, globMsg))
-	// block.NewCall(c.osFuncs.Exit.Value.(llvmValue.Named), constant.NewInt(llvmTypes.I32, 1))
+	block.NewCall(c.osFuncs.Exit.Value.(llvmValue.Named), constant.NewInt(llvmTypes.I8, 1))
 }
 
 type Panic string

compiler/compiler/external_funcs.go

@@ -87,10 +87,10 @@ func (c *Compiler) createExternalPackage() {
 		ir.NewParam("", i64.LLVM()),
 	), false)
 
-	// c.osFuncs.Exit = setExternal("exit", c.module.NewFunc("exit",
-	// 	llvmTypes.Void,
-	// 	ir.NewParam("", i32.LLVM()),
-	// ), false)
+	c.osFuncs.Exit = setExternal("exit", c.module.NewFunc("syscall_exit",
+		llvmTypes.Void,
+		ir.NewParam("", i8.LLVM()),
+	), false)
 
 	c.packages["debug"] = external
 }

compiler/parser/parser.go

@@ -44,7 +44,6 @@ func Parse(input []lexer.Item, debug bool) *FileNode {
 			"uint256": {},
 			"uintptr": {},
 			"string":  {},
-			"exit":    {},
 		},
 	}
 

tests/examples/exit.cell

@@ -0,0 +1,11 @@
+import "debug"
+
+function foo() int64 {
+    debug.Printf("exit from foo")
+    debug.exit(123)
+    return 0
+}
+
+function main() {
+    return foo()
+}

third-party/wrapper.c

@@ -214,3 +214,8 @@ cell_data_t get_utxo_outputs() {
   }
   return outputs;
 }
+
+void syscall_exit(int8_t code) {
+  ckb_exit(code);
+  return;
+}