feat(cfg): Update CFG

Author: GeekMasher

ql/lib/codeql/bicep/controlflow/internal/ControlFlowGraphImpl.qll

@@ -63,6 +63,10 @@ private module CfgImpl = Make<Location, Implementation>;
 
 import CfgImpl
 
+class InfrastructureScopeTree extends StandardTree, PreOrderTree, PostOrderTree, Scope::InfrastructureScope {
+  override AstNode getChildNode(int i) { result = super.getStatement(i) }
+}
+
 /**
  * A literal value in a Bicep program.
  */
@@ -87,3 +91,38 @@ class StringLiteralTree extends LeafTree instanceof StringLiteral { }
  * A StringContent literal value in a Bicep program.
  */
 class StringContentLiteralTree extends LeafTree instanceof StringContentLiteral { }
+
+/**
+ *  ParameterDeclarationTree represents a parameter declaration in a Bicep program.
+ */
+class ParameterDeclarationTree extends StandardPostOrderTree instanceof ParameterDeclaration {
+  override AstNode getChildNode(int i) {
+    i = 0 and result = super.getIdentifier()
+    or
+    i = 1 and result = super.getType()
+    or
+    i = 2 and result = super.getDefaultValue()
+  }
+}
+
+class UserDefinedFunctionTree extends StandardPostOrderTree instanceof UserDefinedFunction {
+  override AstNode getChildNode(int i) {
+    i = 0 and result = super.getIdentifier()
+    or
+    i = 1 and result = super.getParameters()
+    or
+    i = 2 and result = super.getReturnType()
+    or
+    i = 3 and result = super.getBody()
+  }
+}
+
+class OutputDeclarationTree extends StandardPostOrderTree instanceof OutputDeclaration {
+  override AstNode getChildNode(int i) {
+    i = 0 and result = super.getIdentifier()
+    or
+    i = 1 and result = super.getType()
+    or
+    i = 2 and result = super.getValue()
+  }
+}

ql/lib/codeql/bicep/controlflow/internal/Scope.qll

@@ -3,10 +3,21 @@ private import codeql.bicep.AST
 private import Completion
 private import ControlFlowGraphImpl
 
-abstract class CfgScope extends AstNode {
+abstract private class CfgScopeImpl extends AstNode {
   /** Holds if `first` is executed first when entering scope. */
   abstract predicate scopeFirst(AstNode first);
 
   /** Holds if scope is exited when `last` finishes with completion `c`. */
   abstract predicate scopeLast(AstNode last, Completion c);
 }
+
+final class CfgScope = CfgScopeImpl;
+
+/**
+ * A Infrastructure is a Sequence of statements.
+ */
+final class InfrastructureScope extends CfgScopeImpl, Infrastructure {
+  override predicate scopeFirst(AstNode first) { first(this.getStatement(0), first) }
+
+  override predicate scopeLast(AstNode last, Completion c) { last(this, last, c) }
+}