Swift: add PrintAst

Author: redsun82

config/identical-files.json

@@ -584,19 +584,19 @@
   ],
   "Swift declarations test file": [
     "swift/ql/test/extractor-tests/declarations/declarations.swift",
-    "swift/ql/test/library-tests/parent/declarations.swift"
+    "swift/ql/test/library-tests/ast/declarations.swift"
   ],
   "Swift statements test file": [
     "swift/ql/test/extractor-tests/statements/statements.swift",
-    "swift/ql/test/library-tests/parent/statements.swift"
+    "swift/ql/test/library-tests/ast/statements.swift"
   ],
   "Swift expressions test file": [
     "swift/ql/test/extractor-tests/expressions/expressions.swift",
-    "swift/ql/test/library-tests/parent/expressions.swift"
+    "swift/ql/test/library-tests/ast/expressions.swift"
   ],
   "Swift patterns test file": [
     "swift/ql/test/extractor-tests/patterns/patterns.swift",
-    "swift/ql/test/library-tests/parent/patterns.swift"
+    "swift/ql/test/library-tests/ast/patterns.swift"
   ],
   "IncompleteMultiCharacterSanitization JS/Ruby": [
     "javascript/ql/lib/semmle/javascript/security/IncompleteMultiCharacterSanitizationQuery.qll",

swift/ql/lib/codeql/swift/elements/expr/ArgumentConstructor.qll

@@ -1,4 +1,9 @@
-// generated by codegen/codegen.py, remove this comment if you wish to edit this file
 private import codeql.swift.generated.Raw
 
-predicate constructArgument(Raw::Argument id) { any() }
+predicate constructArgument(Raw::Argument id) {
+  // exclude an argument that will be part of a DotSyntaxCallExpr
+  // that will be transformed into a MethodRefCallExpr
+  not exists(Raw::DotSyntaxCallExpr c |
+    c.getFunction() instanceof Raw::DeclRefExpr and id = c.getArgument(0)
+  )
+}

swift/ql/lib/codeql/swift/printast/PrintAst.qll

@@ -0,0 +1,131 @@
+/**
+ * Provides queries to pretty-print a Go AST as a graph.
+ */
+
+import swift
+import codeql.swift.generated.ParentChild
+
+/**
+ * Hook to customize the files and functions printed by this module.
+ *
+ * For an AstNode to be printed, it always requires `shouldPrintFile(f)` to hold
+ * for its containing file `f`, and additionally requires `shouldPrintFunction(fun)`
+ * to hold if it is, or is a child of, function `fun`.
+ */
+class PrintAstConfiguration extends string {
+  /**
+   * Restrict to a single string, making this a singleton type.
+   */
+  PrintAstConfiguration() { this = "PrintAstConfiguration" }
+
+  /**
+   * Holds if the AST for `e` should be printed. By default, holds for all.
+   */
+  predicate shouldPrint(Locatable e) { any() }
+}
+
+private predicate shouldPrint(Locatable e) { any(PrintAstConfiguration config).shouldPrint(e) }
+
+/**
+ * An AST node that should be printed.
+ */
+private newtype TPrintAstNode =
+  TLocatable(Locatable ast) {
+    // Only consider resolved nodes (that is not within the hidden conversion AST)
+    ast = ast.resolve()
+  }
+
+/**
+ * A node in the output tree.
+ */
+class PrintAstNode extends TPrintAstNode {
+  /**
+   * Gets a textual representation of this node.
+   */
+  abstract string toString();
+
+  /**
+   * Gets the child node at index `index`. Child indices must be unique,
+   * but need not be contiguous.
+   */
+  abstract predicate hasChild(PrintAstNode child, int index, string accessor);
+
+  /**
+   * Holds if this node should be printed in the output.
+   */
+  abstract predicate shouldBePrinted();
+
+  /**
+   * Gets the location of this node in the source code.
+   */
+  abstract Location getLocation();
+
+  /**
+   * Gets the value of an additional property of this node, where the name of
+   * the property is `key`.
+   */
+  string getProperty(string key) { none() }
+}
+
+/**
+ * A graph node representing a real Locatable node.
+ */
+class PrintLocatable extends PrintAstNode, TLocatable {
+  Locatable ast;
+
+  PrintLocatable() { this = TLocatable(ast) }
+
+  final override predicate shouldBePrinted() { shouldPrint(ast) }
+
+  override predicate hasChild(PrintAstNode child, int index, string accessor) {
+    child = TLocatable(getChildAndAccessor(ast, index, accessor))
+  }
+
+  override string toString() { result = "[" + concat(ast.getPrimaryQlClasses(), ", ") + "] " + ast }
+
+  final override Location getLocation() { result = ast.getLocation() }
+}
+
+cached
+private int getOrder(PrintAstNode node) {
+  node =
+    rank[result](PrintAstNode n, Location loc |
+      loc = n.getLocation()
+    |
+      n
+      order by
+        loc.getFile().getName(), loc.getStartLine(), loc.getStartColumn(), loc.getEndLine(),
+        loc.getEndColumn()
+    )
+}
+
+/** Holds if `node` belongs to the output tree, and its property `key` has the given `value`. */
+query predicate nodes(PrintAstNode node, string key, string value) {
+  node.shouldBePrinted() and
+  (
+    key = "semmle.label" and value = node.toString()
+    or
+    key = "semmle.order" and value = getOrder(node).toString()
+    or
+    value = node.getProperty(key)
+  )
+}
+
+/**
+ * Holds if `target` is a child of `source` in the AST, and property `key` of the edge has the
+ * given `value`.
+ */
+query predicate edges(PrintAstNode source, PrintAstNode target, string key, string value) {
+  source.shouldBePrinted() and
+  target.shouldBePrinted() and
+  exists(int index, string accessor | source.hasChild(target, index, accessor) |
+    key = "semmle.label" and value = accessor
+    or
+    key = "semmle.order" and value = index.toString()
+  )
+}
+
+/** Holds if property `key` of the graph has the given `value`. */
+query predicate graphProperties(string key, string value) {
+  key = "semmle.graphKind" and value = "tree"
+}

swift/ql/src/queries/ide-contextual-queries/IDEContextual.qll

@@ -0,0 +1,22 @@
+/**
+ * Provides shared predicates related to contextual queries in the code viewer.
+ */
+
+import swift
+
+/**
+ * Returns the `File` matching the given source file name as encoded by the VS
+ * Code extension.
+ */
+cached
+File getFileBySourceArchiveName(string name) {
+  // The name provided for a file in the source archive by the VS Code extension
+  // has some differences from the absolute path in the database:
+  // 1. colons are replaced by underscores
+  // 2. there's a leading slash, even for Windows paths: "C:/foo/bar" ->
+  //    "/C_/foo/bar"
+  // 3. double slashes in UNC prefixes are replaced with a single slash
+  // We can handle 2 and 3 together by unconditionally adding a leading slash
+  // before replacing double slashes.
+  name = ("/" + result.getAbsolutePath().replaceAll(":", "_")).replaceAll("//", "/")
+}

swift/ql/src/queries/ide-contextual-queries/printast.ql

@@ -0,0 +1,28 @@
+/**
+ * @name Print AST
+ * @description Outputs a representation of a file's Abstract Syntax Tree. This
+ *              query is used by the VS Code extension.
+ * @id swift/print-ast
+ * @kind graph
+ * @tags ide-contextual-queries/print-ast
+ */
+
+import swift
+import codeql.swift.printast.PrintAst
+import IDEContextual
+
+/**
+ * The source file to generate an AST from.
+ */
+external string selectedSourceFile();
+
+class PrintAstConfigurationOverride extends PrintAstConfiguration {
+  /**
+   * Holds if the location matches the selected file in the VS Code extension and
+   * the element is `fromSource`.
+   */
+  override predicate shouldPrint(Locatable e) {
+    super.shouldPrint(e) and
+    e.getFile() = getFileBySourceArchiveName(selectedSourceFile())
+  }
+}