Learned to handled labeled statements

Previously we treated labels as placeholders. But they actually hold the next statement inside them!

src/control-flow/cfg.ts

@@ -267,9 +267,13 @@ export class CFGBuilder {
     return { entry: gotoNode, exit: null, gotos: [{ node: gotoNode, label: name }] }
   }
   private processLabeledStatement(labelSyntax: Parser.SyntaxNode): BasicBlock {
+    let blockHandler = new BlockHandler();
     let name = this.getChildFieldText(labelSyntax, "label");
     let labelNode = this.addNode("LABEL", name);
-    return { entry: labelNode, exit: labelNode, labels: new Map([[name, labelNode]]) }
+    console.log("label", labelSyntax.namedChildCount);
+    const { entry: labeledEntry, exit: labeledExit } = blockHandler.update(this.processBlock(labelSyntax.namedChildren[1]))
+    if (labeledEntry) this.addEdge(labelNode, labeledEntry)
+    return blockHandler.update({ entry: labelNode, exit: labeledExit, labels: new Map([[name, labelNode]]) });
   }
   private processContinueStatement(continueSyntax: Parser.SyntaxNode): BasicBlock {
     const continueNode = this.addNode("CONTINUE", "CONTINUE");

src/frontend/src/assets/samples/anotherLabelTestCase.go

@@ -0,0 +1,17 @@
+func anotherLabelTest() {
+	switch x {
+	case 1:
+	case 2:
+		goto label
+	case 3:
+	case 4:
+	}
+	return
+
+label:
+	if x {
+		f()
+	} else {
+		g()
+	}
+}

src/frontend/src/assets/samples/labeledStatement.go

@@ -0,0 +1,5 @@
+func labeledStatement() {
+label:
+	if x {
+	}
+}