Swift: extract DynamicMemberRefExpr

Author: redsun82

swift/extractor/translators/ExprTranslator.cpp

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

swift/extractor/translators/ExprTranslator.h

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

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr.expected

@@ -0,0 +1 @@
+| dynamic_member_ref.swift:12:1:12:3 | .foo(_:) | getBase: | dynamic_member_ref.swift:12:1:12:1 | OpaqueValueExpr |

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr.ql

@@ -0,0 +1,10 @@
+// generated by codegen/codegen.py
+import codeql.swift.elements
+import TestUtils
+
+from DynamicMemberRefExpr x, Expr getBase
+where
+  toBeTested(x) and
+  not x.isUnknown() and
+  getBase = x.getBase()
+select x, "getBase:", getBase

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr_getMember.expected

@@ -0,0 +1 @@
+| dynamic_member_ref.swift:12:1:12:3 | .foo(_:) | dynamic_member_ref.swift:6:9:6:28 | foo(_:) |

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr_getMember.ql

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

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr_getType.expected

@@ -0,0 +1 @@
+| dynamic_member_ref.swift:12:1:12:3 | .foo(_:) | ((Int) -> ())? |

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/DynamicMemberRefExpr_getType.ql

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

swift/ql/test/extractor-tests/generated/expr/DynamicMemberRefExpr/MISSING_SOURCE.txt

@@ -0,0 +1,12 @@
+//codeql-extractor-options: -enable-objc-interop
+//codeql-extractor-expected-status: 1
+
+// Successful compilation would require importing Foundation for `@objc`
+class A {
+  @objc func foo(_ : Int) {}
+};
+
+class B {}
+
+var x : AnyObject = B()
+x.foo!(17)