Merge pull request github#11169 from github/redsun82/swift-extract-last-stmts

Swift: extract remaining `Stmt`s

Author: redsun82

swift/extractor/translators/StmtTranslator.cpp

@@ -174,4 +174,16 @@ void StmtTranslator::fillLabeledConditionalStmt(const swift::LabeledConditionalS
   fillLabeledStmt(stmt, entry);
 }
 
+codeql::FailStmt StmtTranslator::translateFailStmt(const swift::FailStmt& stmt) {
+  return dispatcher.createEntry(stmt);
+}
+
+codeql::PoundAssertStmt StmtTranslator::translatePoundAssertStmt(
+    const swift::PoundAssertStmt& stmt) {
+  auto entry = dispatcher.createEntry(stmt);
+  entry.condition = dispatcher.fetchLabel(stmt.getCondition());
+  entry.message = stmt.getMessage();
+  return entry;
+}
+
 }  // namespace codeql

swift/extractor/translators/StmtTranslator.h

@@ -31,6 +31,8 @@ class StmtTranslator : public AstTranslatorBase<StmtTranslator> {
   codeql::SwitchStmt translateSwitchStmt(const swift::SwitchStmt& stmt);
   codeql::FallthroughStmt translateFallthroughStmt(const swift::FallthroughStmt& stmt);
   codeql::YieldStmt translateYieldStmt(const swift::YieldStmt& stmt);
+  codeql::FailStmt translateFailStmt(const swift::FailStmt& stmt);
+  codeql::PoundAssertStmt translatePoundAssertStmt(const swift::PoundAssertStmt& stmt);
 
  private:
   void fillLabeledStmt(const swift::LabeledStmt& stmt, codeql::LabeledStmt& entry);

swift/ql/lib/codeql/swift/generated/Raw.qll

@@ -1116,6 +1116,10 @@ module Raw {
 
   class PoundAssertStmt extends @pound_assert_stmt, Stmt {
     override string toString() { result = "PoundAssertStmt" }
+
+    Expr getCondition() { pound_assert_stmts(this, result, _) }
+
+    string getMessage() { pound_assert_stmts(this, _, result) }
   }
 
   class ReturnStmt extends @return_stmt, Stmt {

swift/ql/lib/codeql/swift/generated/stmt/PoundAssertStmt.qll

@@ -1,10 +1,36 @@
 // generated by codegen/codegen.py
 private import codeql.swift.generated.Synth
 private import codeql.swift.generated.Raw
+import codeql.swift.elements.expr.Expr
 import codeql.swift.elements.stmt.Stmt
 
 module Generated {
   class PoundAssertStmt extends Synth::TPoundAssertStmt, Stmt {
     override string getAPrimaryQlClass() { result = "PoundAssertStmt" }
+
+    /**
+     * Gets the condition of this pound assert statement.
+     *
+     * This includes nodes from the "hidden" AST. It can be overridden in subclasses to change the
+     * behavior of both the `Immediate` and non-`Immediate` versions.
+     */
+    Expr getImmediateCondition() {
+      result =
+        Synth::convertExprFromRaw(Synth::convertPoundAssertStmtToRaw(this)
+              .(Raw::PoundAssertStmt)
+              .getCondition())
+    }
+
+    /**
+     * Gets the condition of this pound assert statement.
+     */
+    final Expr getCondition() { result = getImmediateCondition().resolve() }
+
+    /**
+     * Gets the message of this pound assert statement.
+     */
+    string getMessage() {
+      result = Synth::convertPoundAssertStmtToRaw(this).(Raw::PoundAssertStmt).getMessage()
+    }
   }
 }

swift/ql/lib/swift.dbscheme

@@ -1696,7 +1696,9 @@ labeled_stmt_labels(  //dir=stmt
 );
 
 pound_assert_stmts(  //dir=stmt
-  unique int id: @pound_assert_stmt
+  unique int id: @pound_assert_stmt,
+  int condition: @expr_or_none ref,
+  string message: string ref
 );
 
 return_stmts(  //dir=stmt

swift/ql/test/extractor-tests/generated/stmt/FailStmt/FailStmt.expected

@@ -0,0 +1 @@
+| fail.swift:2:15:2:22 | fail |

swift/ql/test/extractor-tests/generated/stmt/FailStmt/FailStmt.ql

@@ -0,0 +1,7 @@
+// generated by codegen/codegen.py
+import codeql.swift.elements
+import TestUtils
+
+from FailStmt x
+where toBeTested(x) and not x.isUnknown()
+select x

swift/ql/test/extractor-tests/generated/stmt/FailStmt/MISSING_SOURCE.txt

@@ -0,0 +1,3 @@
+struct S {
+    init?() { return nil }
+}