Merge pull request #83 from microsoft/powershell-more-ast-classes

PS: Copy existing AST classes from internal repo

Author: MathiasVP

powershell/ql/lib/powershell.qll

@@ -0,0 +1,70 @@
+import semmle.code.powershell.File
+import semmle.code.powershell.Location
+import semmle.code.powershell.SourceLocation
+import semmle.code.powershell.Ast
+import semmle.code.powershell.Statement
+import semmle.code.powershell.Expression
+import semmle.code.powershell.CommandBase
+import semmle.code.powershell.AttributeBase
+import semmle.code.powershell.PipelineBase
+import semmle.code.powershell.BaseConstantExpression
+import semmle.code.powershell.ConstantExpression
+import semmle.code.powershell.MemberExpressionBase
+import semmle.code.powershell.Attribute
+import semmle.code.powershell.NamedAttributeArgument
+import semmle.code.powershell.TypeConstraint
+import semmle.code.powershell.VariableExpression
+import semmle.code.powershell.Parameter
+import semmle.code.powershell.ModuleSpecification
+import semmle.code.powershell.ParamBlock
+import semmle.code.powershell.NamedBlock
+import semmle.code.powershell.ScriptBlock
+import semmle.code.powershell.StringLiteral
+import semmle.code.powershell.AssignmentStatement
+import semmle.code.powershell.BinaryExpression
+import semmle.code.powershell.ScriptBlockExpr
+import semmle.code.powershell.TernaryExpression
+import semmle.code.powershell.UsingExpression
+import semmle.code.powershell.TrapStatement
+import semmle.code.powershell.StatementBlock
+import semmle.code.powershell.ArrayExpression
+import semmle.code.powershell.ArrayLiteral
+import semmle.code.powershell.CommandElement
+import semmle.code.powershell.Redirection
+import semmle.code.powershell.FileRedirection
+import semmle.code.powershell.MergingRedirection
+import semmle.code.powershell.LoopStmt
+import semmle.code.powershell.DoWhileStmt
+import semmle.code.powershell.DoUntilStmt
+import semmle.code.powershell.WhileStmt
+import semmle.code.powershell.ForStmt
+import semmle.code.powershell.ForEachStmt
+import semmle.code.powershell.GotoStmt
+import semmle.code.powershell.ContinueStmt
+import semmle.code.powershell.BreakStmt
+import semmle.code.powershell.ReturnStmt
+import semmle.code.powershell.UsingStmt
+import semmle.code.powershell.Type
+import semmle.code.powershell.Member
+import semmle.code.powershell.PropertyMember
+import semmle.code.powershell.FunctionMember
+import semmle.code.powershell.TryStmt
+import semmle.code.powershell.IfStmt
+import semmle.code.powershell.ExitStmt
+import semmle.code.powershell.LabeledStmt
+import semmle.code.powershell.DynamicStmt
+import semmle.code.powershell.DataStmt
+import semmle.code.powershell.Configuration
+import semmle.code.powershell.CatchClause
+import semmle.code.powershell.Command
+import semmle.code.powershell.CommandExpression
+import semmle.code.powershell.CommandParameter
+import semmle.code.powershell.ExpandableStringExpression
+import semmle.code.powershell.TypeExpression
+import semmle.code.powershell.ParenExpression
+import semmle.code.powershell.Chainable
+import semmle.code.powershell.Pipeline
+import semmle.code.powershell.StringConstantExpression
+import semmle.code.powershell.FunctionDefinition
+import semmle.code.powershell.InvokeMemberExpression
+import semmle.code.powershell.CommentEntity

powershell/ql/lib/qlpack.yml

@@ -0,0 +1,9 @@
+name: microsoft-sdl/powershell-all
+version: 0.0.1
+groups: 
+  - powershell
+  - microsoft-all
+dbscheme: semmlecode.powershell.dbscheme
+extractor: powershell
+library: true
+warnOnImplicitThis: true

powershell/ql/lib/semmle/code/powershell/ArrayExpression.qll

@@ -0,0 +1,9 @@
+import powershell
+
+class ArrayExpr extends @array_expression, Expr {
+  override SourceLocation getLocation() { array_expression_location(this, result) }
+
+  StmtBlock getStatementBlock() { array_expression(this, result) }
+
+  override string toString() { result = "ArrayExpression at: " + this.getLocation().toString() }
+}

powershell/ql/lib/semmle/code/powershell/ArrayLiteral.qll

@@ -0,0 +1,11 @@
+import powershell
+
+class ArrayLiteral extends @array_literal, Expr {
+  override SourceLocation getLocation() { array_literal_location(this, result) }
+
+  Expr getElement(int index) { array_literal_element(this, index, result) }
+
+  Expr getAnElement() { array_literal_element(this, _, result) }
+
+  override string toString() { result = "ArrayLiteral at: " + this.getLocation().toString() }
+}

powershell/ql/lib/semmle/code/powershell/AssignmentStatement.qll

@@ -0,0 +1,13 @@
+import powershell
+
+class AssignStmt extends @assignment_statement, Stmt {
+  override SourceLocation getLocation() { assignment_statement_location(this, result) }
+
+  int getKind() { assignment_statement(this, result, _, _) }
+
+  Expr getLeftHandSide() { assignment_statement(this, _, result, _) }
+
+  Stmt getRightHandSide() { assignment_statement(this, _, _, result) }
+
+  override string toString() { result = "AssignmentStatement at: " + this.getLocation().toString() }
+}

powershell/ql/lib/semmle/code/powershell/Ast.qll

@@ -0,0 +1,9 @@
+import powershell
+
+class Ast extends @ast {
+  string toString() { none() }
+
+  Ast getParent() { parent(result, this) }
+
+  Location getLocation() { none() }
+}

powershell/ql/lib/semmle/code/powershell/Attribute.qll

@@ -0,0 +1,21 @@
+import powershell
+
+class Attribute extends @attribute, AttributeBase {
+  override string toString() { result = this.getName() }
+
+  override SourceLocation getLocation() { attribute_location(this, result) }
+
+  string getName() { attribute(this, result, _, _) }
+
+  int getNumNamedArguments() { attribute(this, _, result, _) }
+
+  int getNumPositionalArguments() { attribute(this, _, _, result) }
+
+  NamedAttributeArgument getNamedArgument(int i) { attribute_named_argument(this, i, result) }
+
+  NamedAttributeArgument getANamedArgument() { result = this.getNamedArgument(_) }
+
+  Expr getPositionalArgument(int i) { attribute_positional_argument(this, i, result) }
+
+  Expr getAPositionalArgument() { result = this.getPositionalArgument(_) }
+}

powershell/ql/lib/semmle/code/powershell/AttributeBase.qll

@@ -0,0 +1,3 @@
+import powershell
+
+class AttributeBase extends @attribute_base, Ast { }

powershell/ql/lib/semmle/code/powershell/BaseConstantExpression.qll

@@ -0,0 +1,3 @@
+import powershell
+
+class BaseConstExpr extends @base_constant_expression, Expr { }

powershell/ql/lib/semmle/code/powershell/BinaryExpression.qll

@@ -0,0 +1,15 @@
+import powershell
+
+class BinaryExpr extends @binary_expression, Expr {
+  override string toString() {
+    result = "...+..." // TODO
+  }
+
+  override SourceLocation getLocation() { binary_expression_location(this, result) }
+
+  private int getKind() { binary_expression(this, result, _, _) }
+
+  Expr getLeft() { binary_expression(this, _, result, _) }
+
+  Expr getRight() { binary_expression(this, _, _, result) }
+}