wip5

Author: hvitved

rust/ql/lib/codeql/rust/internal/TypeInference.qll

@@ -1161,18 +1161,32 @@ private module MethodCallResolution {
     )
   }
 
-  pragma[nomagic]
-  private predicate traitMethodInfo(string name, int arity, Trait trait) {
-    exists(ImplItemNode i |
-      methodInfo(_, name, arity, i, _, _, _) and
-      trait = i.resolveTraitTy()
-    )
+  /**
+   * todo
+   *
+   * Holds if method `f` with the name `name` and the arity `arity` exists in
+   * `i`, and the type of the `self` parameter is `selfType`.
+   *
+   * `rootTypePath` points to the type `rootType` inside `selfType`, which is
+   * the (possibly `&`-stripped) root type of `selfType`.
+   */
+  pragma[inline]
+  predicate methodInfoMatch(
+    Function f, string name, int arity, ImplOrTraitItemNode i, FunctionType selfType,
+    TypePath rootTypePath, Type rootType
+  ) {
+    methodInfo(f, name, arity, i, selfType, rootTypePath, rootType)
+    or
+    methodInfo(f, name, arity, i, selfType, rootTypePath, TTypeParamTypeParameter(_))
   }
 
   pragma[nomagic]
   private predicate methodCallTraitCandidate(Element mc, Trait trait) {
     exists(string name, int arity | mc.(MethodCall).isMethodCall(name, arity) |
-      traitMethodInfo(name, arity, trait)
+      exists(ImplItemNode i |
+        methodInfo(_, name, arity, i, _, _, _) and
+        trait = i.resolveTraitTy()
+      )
       or
       methodInfo(_, name, arity, trait, _, _, _)
     )
@@ -1209,7 +1223,7 @@ private module MethodCallResolution {
   ) {
     exists(string name, int arity |
       mc.isMethodCall(name, arity) and
-      methodInfo(_, name, arity, i, self, rootTypePath, rootType)
+      methodInfoMatch(_, name, arity, i, self, rootTypePath, rootType)
     |
       i =
         any(Impl impl |
@@ -1472,7 +1486,7 @@ private module MethodCallResolution {
         this.isMethodCall(_, rootTypePath, rootType, name, arity) and
         isRefStrippedRoot(rootTypePath, rootType) and
         forall(Impl i |
-          methodInfo(_, name, arity, i, _, rootTypePath, rootType) and
+          methodInfoMatch(_, name, arity, i, _, rootTypePath, rootType) and
           not i.hasTrait()
         |
           this.isNotInherentCandidate(i)
@@ -1883,7 +1897,7 @@ private module FunctionCallResolution {
       exists(TypePath rootTypePath, Type rootType |
         f = call.getPathResolutionResolvedFunctionOrImplementation(resolved) and
         trait = call.(Call).getTrait() and
-        MethodCallResolution::methodInfo(f, _, _, i, self, rootTypePath, rootType) and
+        MethodCallResolution::methodInfoMatch(f, _, _, i, self, rootTypePath, rootType) and
         call.getTypeAt(rootTypePath) = rootType
       )
     }
@@ -2178,12 +2192,12 @@ private module OperationResolution {
   private module OperationIsInstantiationOfInput implements
     IsInstantiationOfInputSig<Op, FunctionType>
   {
-    pragma[nomagic]
-    private predicate methodInfo(
+    pragma[inline]
+    private predicate methodInfoMatch(
       TypeAbstraction abs, FunctionType constraint, Trait trait, string name, int arity,
       TypePath rootTypePath, Type rootType
     ) {
-      MethodCallResolution::methodInfo(_, name, arity, abs, constraint, rootTypePath, rootType) and
+      MethodCallResolution::methodInfoMatch(_, name, arity, abs, constraint, rootTypePath, rootType) and
       (
         trait = abs.(ImplItemNode).resolveTraitTy()
         or
@@ -2195,7 +2209,7 @@ private module OperationResolution {
     predicate potentialInstantiationOf(Op op, TypeAbstraction abs, FunctionType constraint) {
       exists(Trait trait, string name, int arity, TypePath rootTypePath, Type rootType |
         op.isOperation(rootTypePath, rootType, trait, name, arity) and
-        methodInfo(abs, constraint, trait, name, arity, rootTypePath, rootType)
+        methodInfoMatch(abs, constraint, trait, name, arity, rootTypePath, rootType)
       )
     }