ruleguard/quasigo: emit implicit return stmt (#308)

quasilyte · Nov 18, 2021 · 05e5d0c · 05e5d0c
1 parent 2d7358b
commit 05e5d0c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/ruleguard/quasigo/compile.go b/ruleguard/quasigo/compile.go
@@ -106,6 +106,10 @@ func (cl *compiler) compileFunc(fn *ast.FuncDecl) *Func {
 	}
 
 	cl.compileStmt(fn.Body)
+	if cl.retType == voidType {
+		cl.emit(opReturn)
+	}
+
 	compiled := &Func{
 		code:         cl.code,
 		constants:    cl.constants,

diff --git a/ruleguard/quasigo/testdata/voidfunc/main.go b/ruleguard/quasigo/testdata/voidfunc/main.go
@@ -5,5 +5,4 @@ func main() {
 	println(1)
 	x := "var"
 	println(x)
-	return
 }