built-in Go

cpg-language-go/src/main/golang/frontend/handler.go

@@ -104,6 +104,11 @@ func (this *GoLanguageFrontend) HandleFile(fset *token.FileSet, file *ast.File,
 		}
 	}
 
+	// create a built-in "go" function
+	goFunc := this.NewFunctionDeclaration(fset, nil, "go", "", true)
+	goFunc.AddParameter(this.NewParamVariableDeclaration(fset, nil, "call"))
+	scope.AddDeclaration((*cpg.Declaration)(goFunc))
+
 	// create a new namespace declaration, representing the package
 	p := this.NewNamespaceDeclaration(fset, nil, file.Name.Name)
 
@@ -658,6 +663,8 @@ func (this *GoLanguageFrontend) handleStmt(fset *token.FileSet, stmt ast.Stmt, p
 		s = (*cpg.Statement)(this.handleAssignStmt(fset, v))
 	case *ast.DeclStmt:
 		s = (*cpg.Statement)(this.handleDeclStmt(fset, v))
+	case *ast.GoStmt:
+		s = (*cpg.Statement)(this.handleGoStmt(fset, v))
 	case *ast.IfStmt:
 		s = (*cpg.Statement)(this.handleIfStmt(fset, v))
 	case *ast.SwitchStmt:
@@ -789,6 +796,20 @@ func (this *GoLanguageFrontend) handleDeclStmt(fset *token.FileSet, declStmt *as
 	return (*cpg.Expression)(stmt)
 }
 
+// handleGoStmt handles the `go` statement, which is a special keyword in go
+// that starts the supplied call expression in a separate Go routine. We cannot
+// model this 1:1, so we basically we create a call expression to a built-in call.
+func (this *GoLanguageFrontend) handleGoStmt(fset *token.FileSet, goStmt *ast.GoStmt) (expr *cpg.Expression) {
+	this.LogTrace("Handling go statement: %+v", *goStmt)
+
+	ref := (*cpg.Expression)(this.NewDeclaredReferenceExpression(fset, nil, "go"))
+
+	call := this.NewCallExpression(fset, goStmt, ref, "go")
+	call.AddArgument(this.handleCallExpr(fset, goStmt.Call))
+
+	return (*cpg.Expression)(call)
+}
+
 func (this *GoLanguageFrontend) handleIfStmt(fset *token.FileSet, ifStmt *ast.IfStmt) (expr *cpg.Expression) {
 	this.LogTrace("Handling if statement: %+v", *ifStmt)
 

cpg-language-go/src/test/kotlin/de/fraunhofer/aisec/cpg/frontends/golang/GoLanguageFrontendTest.kt

@@ -619,6 +619,9 @@ class GoLanguageFrontendTest : BaseTest() {
         val base = call.base as? DeclaredReferenceExpression
         assertNotNull(base)
         assertEquals(c, base.refersTo)
+
+        val go = main.calls["go"]
+        assertNotNull(go)
     }
 
     @Test
@@ -650,6 +653,7 @@ class GoLanguageFrontendTest : BaseTest() {
 
         val bytes = assertIs<DeclaredReferenceExpression>(each.iterable)
         assertLocalName("bytes", bytes)
+        assertNotNull(bytes.refersTo)
 
         val idx = assertIs<DeclarationStatement>(each.variable).variables["idx"]
         assertNotNull(idx)

cpg-language-go/src/test/resources/golang/call.go

@@ -5,4 +5,7 @@ import ("http")
 func main() {
     c := NewMyStruct()
 	c.myOtherFunc()
+
+	go c.MyFunc()
+	go c.MyFunc()
 }