Swift: extract OverloadedDeclRefExpr

Author: redsun82

swift/extractor/infra/SwiftTagTraits.h

@@ -16,6 +16,9 @@ using SILBoxTypeTag = SilBoxTypeTag;
 using SILFunctionTypeTag = SilFunctionTypeTag;
 using SILTokenTypeTag = SilTokenTypeTag;
 
+// OverloadSetRefExpr is collapsed with its only derived class OverloadedDeclRefExpr
+using OverloadSetRefExprTag = OverloadedDeclRefExprTag;
+
 #define MAP_TYPE_TO_TAG(TYPE, TAG)    \
   template <>                         \
   struct detail::ToTagFunctor<TYPE> { \

swift/extractor/translators/ExprTranslator.cpp

@@ -537,5 +537,11 @@ codeql::ObjectLiteralExpr ExprTranslator::translateObjectLiteralExpr(
   }
   return entry;
 }
+codeql::OverloadedDeclRefExpr ExprTranslator::translateOverloadedDeclRefExpr(
+    const swift::OverloadedDeclRefExpr& expr) {
+  auto entry = createExprEntry(expr);
+  entry.possible_declarations = dispatcher.fetchRepeatedLabels(expr.getDecls());
+  return entry;
+}
 
 }  // namespace codeql

swift/extractor/translators/ExprTranslator.h

@@ -103,6 +103,8 @@ class ExprTranslator : public AstTranslatorBase<ExprTranslator> {
   codeql::UnresolvedPatternExpr translateUnresolvedPatternExpr(
       const swift::UnresolvedPatternExpr& expr);
   codeql::ObjectLiteralExpr translateObjectLiteralExpr(const swift::ObjectLiteralExpr& expr);
+  codeql::OverloadedDeclRefExpr translateOverloadedDeclRefExpr(
+      const swift::OverloadedDeclRefExpr& expr);
 
  private:
   void fillAbstractClosureExpr(const swift::AbstractClosureExpr& expr,

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

@@ -150,7 +150,6 @@ import codeql.swift.elements.expr.OpenExistentialExpr
 import codeql.swift.elements.expr.OptionalEvaluationExpr
 import codeql.swift.elements.expr.OptionalTryExpr
 import codeql.swift.elements.expr.OtherConstructorDeclRefExpr
-import codeql.swift.elements.expr.OverloadSetRefExpr
 import codeql.swift.elements.expr.OverloadedDeclRefExpr
 import codeql.swift.elements.expr.PackExpr
 import codeql.swift.elements.expr.ParenExpr

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

@@ -1592,8 +1592,8 @@ private module Impl {
     )
   }
 
-  private Element getImmediateChildOfOverloadSetRefExpr(
-    OverloadSetRefExpr e, int index, string partialPredicateCall
+  private Element getImmediateChildOfOverloadedDeclRefExpr(
+    OverloadedDeclRefExpr e, int index, string partialPredicateCall
   ) {
     exists(int b, int bExpr, int n |
       b = 0 and
@@ -2621,22 +2621,6 @@ private module Impl {
     )
   }
 
-  private Element getImmediateChildOfOverloadedDeclRefExpr(
-    OverloadedDeclRefExpr e, int index, string partialPredicateCall
-  ) {
-    exists(int b, int bOverloadSetRefExpr, int n |
-      b = 0 and
-      bOverloadSetRefExpr =
-        b + 1 + max(int i | i = -1 or exists(getImmediateChildOfOverloadSetRefExpr(e, i, _)) | i) and
-      n = bOverloadSetRefExpr and
-      (
-        none()
-        or
-        result = getImmediateChildOfOverloadSetRefExpr(e, index - b, partialPredicateCall)
-      )
-    )
-  }
-
   private Element getImmediateChildOfParenExpr(ParenExpr e, int index, string partialPredicateCall) {
     exists(int b, int bIdentityExpr, int n |
       b = 0 and
@@ -5028,6 +5012,8 @@ private module Impl {
     or
     result = getImmediateChildOfOtherConstructorDeclRefExpr(e, index, partialAccessor)
     or
+    result = getImmediateChildOfOverloadedDeclRefExpr(e, index, partialAccessor)
+    or
     result = getImmediateChildOfPackExpr(e, index, partialAccessor)
     or
     result = getImmediateChildOfPropertyWrapperValuePlaceholderExpr(e, index, partialAccessor)
@@ -5140,8 +5126,6 @@ private module Impl {
     or
     result = getImmediateChildOfOptionalTryExpr(e, index, partialAccessor)
     or
-    result = getImmediateChildOfOverloadedDeclRefExpr(e, index, partialAccessor)
-    or
     result = getImmediateChildOfParenExpr(e, index, partialAccessor)
     or
     result = getImmediateChildOfPointerToPointerExpr(e, index, partialAccessor)

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

@@ -547,7 +547,13 @@ module Raw {
     ConstructorDecl getConstructorDecl() { other_constructor_decl_ref_exprs(this, result) }
   }
 
-  class OverloadSetRefExpr extends @overload_set_ref_expr, Expr { }
+  class OverloadedDeclRefExpr extends @overloaded_decl_ref_expr, Expr {
+    override string toString() { result = "OverloadedDeclRefExpr" }
+
+    ValueDecl getPossibleDeclaration(int index) {
+      overloaded_decl_ref_expr_possible_declarations(this, index, result)
+    }
+  }
 
   class PackExpr extends @pack_expr, Expr {
     override string toString() { result = "PackExpr" }
@@ -849,10 +855,6 @@ module Raw {
     override string toString() { result = "OptionalTryExpr" }
   }
 
-  class OverloadedDeclRefExpr extends @overloaded_decl_ref_expr, OverloadSetRefExpr {
-    override string toString() { result = "OverloadedDeclRefExpr" }
-  }
-
   class ParenExpr extends @paren_expr, IdentityExpr {
     override string toString() { result = "ParenExpr" }
   }

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

@@ -390,7 +390,7 @@ module Synth {
         TKeyPathDotExpr or TKeyPathExpr or TLazyInitializerExpr or TLiteralExpr or TLookupExpr or
         TMakeTemporarilyEscapableExpr or TObjCSelectorExpr or TOneWayExpr or TOpaqueValueExpr or
         TOpenExistentialExpr or TOptionalEvaluationExpr or TOtherConstructorDeclRefExpr or
-        TOverloadSetRefExpr or TPackExpr or TPropertyWrapperValuePlaceholderExpr or
+        TOverloadedDeclRefExpr or TPackExpr or TPropertyWrapperValuePlaceholderExpr or
         TRebindSelfInConstructorExpr or TSequenceExpr or TSuperRefExpr or TTapExpr or
         TTupleElementExpr or TTupleExpr or TTypeExpr or TUnresolvedDeclRefExpr or
         TUnresolvedDotExpr or TUnresolvedMemberExpr or TUnresolvedPatternExpr or
@@ -422,8 +422,6 @@ module Synth {
 
   class TNumberLiteralExpr = TFloatLiteralExpr or TIntegerLiteralExpr;
 
-  class TOverloadSetRefExpr = TOverloadedDeclRefExpr;
-
   class TSelfApplyExpr = TConstructorRefCallExpr or TDotSyntaxCallExpr;
 
   class TPattern =
@@ -1781,7 +1779,7 @@ module Synth {
     or
     result = convertOtherConstructorDeclRefExprFromRaw(e)
     or
-    result = convertOverloadSetRefExprFromRaw(e)
+    result = convertOverloadedDeclRefExprFromRaw(e)
     or
     result = convertPackExprFromRaw(e)
     or
@@ -1923,11 +1921,6 @@ module Synth {
     result = convertIntegerLiteralExprFromRaw(e)
   }
 
-  cached
-  TOverloadSetRefExpr convertOverloadSetRefExprFromRaw(Raw::Element e) {
-    result = convertOverloadedDeclRefExprFromRaw(e)
-  }
-
   cached
   TSelfApplyExpr convertSelfApplyExprFromRaw(Raw::Element e) {
     result = convertConstructorRefCallExprFromRaw(e)
@@ -3491,7 +3484,7 @@ module Synth {
     or
     result = convertOtherConstructorDeclRefExprToRaw(e)
     or
-    result = convertOverloadSetRefExprToRaw(e)
+    result = convertOverloadedDeclRefExprToRaw(e)
     or
     result = convertPackExprToRaw(e)
     or
@@ -3633,11 +3626,6 @@ module Synth {
     result = convertIntegerLiteralExprToRaw(e)
   }
 
-  cached
-  Raw::Element convertOverloadSetRefExprToRaw(TOverloadSetRefExpr e) {
-    result = convertOverloadedDeclRefExprToRaw(e)
-  }
-
   cached
   Raw::Element convertSelfApplyExprToRaw(TSelfApplyExpr e) {
     result = convertConstructorRefCallExprToRaw(e)

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

@@ -1,10 +1,45 @@
 // generated by codegen/codegen.py
 private import codeql.swift.generated.Synth
 private import codeql.swift.generated.Raw
-import codeql.swift.elements.expr.OverloadSetRefExpr
+import codeql.swift.elements.expr.Expr
+import codeql.swift.elements.decl.ValueDecl
 
 module Generated {
-  class OverloadedDeclRefExpr extends Synth::TOverloadedDeclRefExpr, OverloadSetRefExpr {
+  /**
+   * An ambiguous expression that might refer to multiple declarations. This will be present only
+   * for failing compilations.
+   */
+  class OverloadedDeclRefExpr extends Synth::TOverloadedDeclRefExpr, Expr {
     override string getAPrimaryQlClass() { result = "OverloadedDeclRefExpr" }
+
+    /**
+     * Gets the `index`th possible declaration of this overloaded declaration ref expression (0-based).
+     *
+     * 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.
+     */
+    ValueDecl getImmediatePossibleDeclaration(int index) {
+      result =
+        Synth::convertValueDeclFromRaw(Synth::convertOverloadedDeclRefExprToRaw(this)
+              .(Raw::OverloadedDeclRefExpr)
+              .getPossibleDeclaration(index))
+    }
+
+    /**
+     * Gets the `index`th possible declaration of this overloaded declaration ref expression (0-based).
+     */
+    final ValueDecl getPossibleDeclaration(int index) {
+      result = getImmediatePossibleDeclaration(index).resolve()
+    }
+
+    /**
+     * Gets any of the possible declarations of this overloaded declaration ref expression.
+     */
+    final ValueDecl getAPossibleDeclaration() { result = getPossibleDeclaration(_) }
+
+    /**
+     * Gets the number of possible declarations of this overloaded declaration ref expression.
+     */
+    final int getNumberOfPossibleDeclarations() { result = count(getAPossibleDeclaration()) }
   }
 }