Merge pull request #11177 from github/redsun82/swift-decls

Swift: extract `PoundDiagnosticDecl` and `MissingMemberDecl`

Author: redsun82

swift/extractor/infra/SwiftDiagnosticKind.h

@@ -0,0 +1,23 @@
+#pragma once
+
+#include <swift/AST/DiagnosticConsumer.h>
+
+namespace codeql {
+
+inline int translateDiagnosticsKind(swift::DiagnosticKind kind) {
+  using Kind = swift::DiagnosticKind;
+  switch (kind) {
+    case Kind::Error:
+      return 1;
+    case Kind::Warning:
+      return 2;
+    case Kind::Note:
+      return 3;
+    case Kind::Remark:
+      return 4;
+    default:
+      return 0;
+  }
+}
+
+}  // namespace codeql

swift/extractor/invocation/SwiftDiagnosticsConsumer.cpp

@@ -1,6 +1,7 @@
 #include "swift/extractor/invocation/SwiftDiagnosticsConsumer.h"
 #include "swift/extractor/trap/generated/TrapEntries.h"
 #include "swift/extractor/trap/TrapDomain.h"
+#include "swift/extractor/infra/SwiftDiagnosticKind.h"
 
 #include <swift/AST/DiagnosticEngine.h>
 #include <swift/Basic/SourceManager.h>
@@ -10,26 +11,12 @@
 
 using namespace codeql;
 
-static int diagnosticsKind(const swift::DiagnosticInfo& diagInfo) {
-  switch (diagInfo.Kind) {
-    case swift::DiagnosticKind::Error:
-      return 1;
-    case swift::DiagnosticKind::Warning:
-      return 2;
-    case swift::DiagnosticKind::Note:
-      return 3;
-    case swift::DiagnosticKind::Remark:
-      return 4;
-  }
-  return 0;
-}
-
 void SwiftDiagnosticsConsumer::handleDiagnostic(swift::SourceManager& sourceManager,
                                                 const swift::DiagnosticInfo& diagInfo) {
   auto message = getDiagMessage(sourceManager, diagInfo);
   DiagnosticsTrap diag{};
   diag.id = trap.createLabel<DiagnosticsTag>();
-  diag.kind = diagnosticsKind(diagInfo);
+  diag.kind = translateDiagnosticsKind(diagInfo.Kind);
   diag.text = message;
   trap.emit(diag);
   locationExtractor.attachLocation(sourceManager, diagInfo.Loc, diagInfo.Loc, diag.id);

swift/extractor/translators/DeclTranslator.cpp

@@ -2,6 +2,7 @@
 
 #include <swift/AST/GenericParamList.h>
 #include <swift/AST/ParameterList.h>
+#include "swift/extractor/infra/SwiftDiagnosticKind.h"
 
 namespace codeql {
 namespace {
@@ -401,4 +402,18 @@ std::optional<codeql::OpaqueTypeDecl> DeclTranslator::translateOpaqueTypeDecl(
   return std::nullopt;
 }
 
+codeql::PoundDiagnosticDecl DeclTranslator::translatePoundDiagnosticDecl(
+    const swift::PoundDiagnosticDecl& decl) {
+  auto entry = createEntry(decl);
+  entry.kind = translateDiagnosticsKind(decl.getKind());
+  entry.message = dispatcher.fetchLabel(decl.getMessage());
+  return entry;
+}
+
+codeql::MissingMemberDecl DeclTranslator::translateMissingMemberDecl(
+    const swift::MissingMemberDecl& decl) {
+  auto entry = createEntry(decl);
+  entry.name = decl.getName().getBaseName().userFacingName().str();
+  return entry;
+}
 }  // namespace codeql

swift/extractor/translators/DeclTranslator.h

@@ -46,6 +46,8 @@ class DeclTranslator : public AstTranslatorBase<DeclTranslator> {
   std::optional<codeql::ModuleDecl> translateModuleDecl(const swift::ModuleDecl& decl);
   codeql::IfConfigDecl translateIfConfigDecl(const swift::IfConfigDecl& decl);
   std::optional<codeql::OpaqueTypeDecl> translateOpaqueTypeDecl(const swift::OpaqueTypeDecl& decl);
+  codeql::PoundDiagnosticDecl translatePoundDiagnosticDecl(const swift::PoundDiagnosticDecl& decl);
+  codeql::MissingMemberDecl translateMissingMemberDecl(const swift::MissingMemberDecl& decl);
 
  private:
   std::string mangledName(const swift::ValueDecl& decl);

swift/ql/lib/codeql/swift/elements/decl/MissingMemberDecl.qll

@@ -1,4 +1,5 @@
-// generated by codegen/codegen.py, remove this comment if you wish to edit this file
 private import codeql.swift.generated.decl.MissingMemberDecl
 
-class MissingMemberDecl extends Generated::MissingMemberDecl { }
+class MissingMemberDecl extends Generated::MissingMemberDecl {
+  override string toString() { result = this.getName() + " (missing)" }
+}

swift/ql/lib/codeql/swift/elements/decl/PoundDiagnosticDecl.qll

@@ -2,6 +2,12 @@ private import codeql.swift.generated.decl.PoundDiagnosticDecl
 
 class PoundDiagnosticDecl extends Generated::PoundDiagnosticDecl {
   override string toString() {
-    result = "#..." // TODO: Once we extract whether this is an error or a warning we can improve this.
+    this.isError() and result = "#error(...)"
+    or
+    this.isWarning() and result = "#warning(...)"
   }
+
+  predicate isError() { this.getKind() = 1 }
+
+  predicate isWarning() { this.getKind() = 2 }
 }

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

@@ -373,14 +373,17 @@ private module Impl {
   private Element getImmediateChildOfPoundDiagnosticDecl(
     PoundDiagnosticDecl e, int index, string partialPredicateCall
   ) {
-    exists(int b, int bDecl, int n |
+    exists(int b, int bDecl, int n, int nMessage |
       b = 0 and
       bDecl = b + 1 + max(int i | i = -1 or exists(getImmediateChildOfDecl(e, i, _)) | i) and
       n = bDecl and
+      nMessage = n + 1 and
       (
         none()
         or
         result = getImmediateChildOfDecl(e, index - b, partialPredicateCall)
+        or
+        index = n and result = e.getImmediateMessage() and partialPredicateCall = "Message()"
       )
     )
   }

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

@@ -115,6 +115,8 @@ module Raw {
 
   class MissingMemberDecl extends @missing_member_decl, Decl {
     override string toString() { result = "MissingMemberDecl" }
+
+    string getName() { missing_member_decls(this, result) }
   }
 
   class OperatorDecl extends @operator_decl, Decl {
@@ -131,6 +133,10 @@ module Raw {
 
   class PoundDiagnosticDecl extends @pound_diagnostic_decl, Decl {
     override string toString() { result = "PoundDiagnosticDecl" }
+
+    int getKind() { pound_diagnostic_decls(this, result, _) }
+
+    StringLiteralExpr getMessage() { pound_diagnostic_decls(this, _, result) }
   }
 
   class PrecedenceGroupDecl extends @precedence_group_decl, Decl {

swift/ql/lib/codeql/swift/generated/decl/MissingMemberDecl.qll

@@ -4,7 +4,17 @@ private import codeql.swift.generated.Raw
 import codeql.swift.elements.decl.Decl
 
 module Generated {
+  /**
+   * A placeholder for missing declarations that can arise on object deserialization.
+   */
   class MissingMemberDecl extends Synth::TMissingMemberDecl, Decl {
     override string getAPrimaryQlClass() { result = "MissingMemberDecl" }
+
+    /**
+     * Gets the name of this missing member declaration.
+     */
+    string getName() {
+      result = Synth::convertMissingMemberDeclToRaw(this).(Raw::MissingMemberDecl).getName()
+    }
   }
 }

swift/ql/lib/codeql/swift/generated/decl/PoundDiagnosticDecl.qll

@@ -2,9 +2,38 @@
 private import codeql.swift.generated.Synth
 private import codeql.swift.generated.Raw
 import codeql.swift.elements.decl.Decl
+import codeql.swift.elements.expr.StringLiteralExpr
 
 module Generated {
+  /**
+   * A diagnostic directive, which is either `#error` or `#warning`.
+   */
   class PoundDiagnosticDecl extends Synth::TPoundDiagnosticDecl, Decl {
     override string getAPrimaryQlClass() { result = "PoundDiagnosticDecl" }
+
+    /**
+     * Gets the This is 1 for `#error` and 2 for `#warning`.
+     */
+    int getKind() {
+      result = Synth::convertPoundDiagnosticDeclToRaw(this).(Raw::PoundDiagnosticDecl).getKind()
+    }
+
+    /**
+     * Gets the message of this pound diagnostic declaration.
+     *
+     * 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.
+     */
+    StringLiteralExpr getImmediateMessage() {
+      result =
+        Synth::convertStringLiteralExprFromRaw(Synth::convertPoundDiagnosticDeclToRaw(this)
+              .(Raw::PoundDiagnosticDecl)
+              .getMessage())
+    }
+
+    /**
+     * Gets the message of this pound diagnostic declaration.
+     */
+    final StringLiteralExpr getMessage() { result = getImmediateMessage().resolve() }
   }
 }