wip

Author: hvitved

rust/ql/lib/codeql/rust/elements/internal/InvocationExprImpl.qll

@@ -6,7 +6,7 @@ module Impl {
 
   private newtype TArgumentPosition =
     TPositionalArgumentPosition(int i) {
-      i in [0 .. max([any(ParamList l).getNumberOfParams(), any(ArgList l).getNumberOfArgs()]) - 1]
+      i in [0 .. max([any(ParamList l).getNumberOfParams(), any(ArgList l).getNumberOfArgs()])]
     } or
     TSelfArgumentPosition() or
     TTypeQualifierArgumentPosition()

rust/ql/lib/codeql/rust/internal/typeinference/FunctionType.qll

@@ -315,17 +315,17 @@ signature module ArgsAreInstantiationsOfInputSig {
    * If `i` is an inherent implementation, `tp` is a type parameter of the type being
    * implemented, otherwise `tp` is a type parameter of the trait (being implemented).
    *
-   * `pos` is one of the positions in `f` in which the relevant type occours.
+   * `posAdj` is one of the positions in `f` in which the relevant type occurs.
    */
-  predicate toCheck(ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition pos);
+  predicate toCheck(ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition posAdj);
 
   /** A call whose argument types are to be checked. */
   class Call {
     string toString();
 
     Location getLocation();
 
-    Type getArgType(FunctionPosition pos, TypePath path);
+    Type getArgType(FunctionPosition posAdj, TypePath path);
 
     predicate hasTargetCand(ImplOrTraitItemNode i, Function f);
   }
@@ -339,9 +339,9 @@ signature module ArgsAreInstantiationsOfInputSig {
 module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
   pragma[nomagic]
   private predicate toCheckRanked(
-    ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition pos, int rnk
+    ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition posAdj, int rnk
   ) {
-    Input::toCheck(i, f, tp, pos) and
+    Input::toCheck(i, f, tp, posAdj) and
     tp =
       rank[rnk + 1](TypeParameter tp0, int j |
         Input::toCheck(i, f, tp0, _) and
@@ -353,41 +353,45 @@ module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
 
   pragma[nomagic]
   private predicate toCheck(
-    ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition pos, AssocFunctionType t
+    ImplOrTraitItemNode i, Function f, TypeParameter tp, FunctionPosition posAdj,
+    AssocFunctionType t
   ) {
-    Input::toCheck(i, f, tp, pos) and
-    t.appliesTo(f, i, pos)
+    exists(FunctionPosition pos |
+      Input::toCheck(i, f, tp, posAdj) and
+      t.appliesTo(f, i, pos) and
+      posAdj = pos.getFunctionCallAdjusted(f)
+    )
   }
 
   private newtype TCallAndPos =
-    MkCallAndPos(Input::Call call, FunctionPosition pos) { exists(call.getArgType(pos, _)) }
+    MkCallAndPos(Input::Call call, FunctionPosition posAdj) { exists(call.getArgType(posAdj, _)) }
 
   /** A call tagged with a position. */
   private class CallAndPos extends MkCallAndPos {
     Input::Call call;
-    FunctionPosition pos;
+    FunctionPosition posAdj;
 
-    CallAndPos() { this = MkCallAndPos(call, pos) }
+    CallAndPos() { this = MkCallAndPos(call, posAdj) }
 
     Input::Call getCall() { result = call }
 
-    FunctionPosition getPos() { result = pos }
+    FunctionPosition getPosAdj() { result = posAdj }
 
     Location getLocation() { result = call.getLocation() }
 
-    Type getTypeAt(TypePath path) { result = call.getArgType(pos, path) }
+    Type getTypeAt(TypePath path) { result = call.getArgType(posAdj, path) }
 
-    string toString() { result = call.toString() + " [arg " + pos + "]" }
+    string toString() { result = call.toString() + " [arg " + posAdj + "]" }
   }
 
   pragma[nomagic]
   private predicate potentialInstantiationOf0(
-    CallAndPos cp, Input::Call call, TypeParameter tp, FunctionPosition pos, Function f,
+    CallAndPos cp, Input::Call call, TypeParameter tp, FunctionPosition posAdj, Function f,
     TypeAbstraction abs, AssocFunctionType constraint
   ) {
-    cp = MkCallAndPos(call, pragma[only_bind_into](pos)) and
+    cp = MkCallAndPos(call, pragma[only_bind_into](posAdj)) and
     call.hasTargetCand(abs, f) and
-    toCheck(abs, f, tp, pragma[only_bind_into](pos), constraint)
+    toCheck(abs, f, tp, pragma[only_bind_into](posAdj), constraint)
   }
 
   private module ArgIsInstantiationOfToIndexInput implements
@@ -397,9 +401,9 @@ module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
     predicate potentialInstantiationOf(
       CallAndPos cp, TypeAbstraction abs, AssocFunctionType constraint
     ) {
-      exists(Input::Call call, TypeParameter tp, FunctionPosition pos, int rnk, Function f |
-        potentialInstantiationOf0(cp, call, tp, pos, f, abs, constraint) and
-        toCheckRanked(abs, f, tp, pos, rnk)
+      exists(Input::Call call, TypeParameter tp, FunctionPosition posAdj, int rnk, Function f |
+        potentialInstantiationOf0(cp, call, tp, posAdj, f, abs, constraint) and
+        toCheckRanked(abs, f, tp, posAdj, rnk)
       |
         rnk = 0
         or
@@ -415,20 +419,21 @@ module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
 
   pragma[nomagic]
   private predicate argIsInstantiationOf(
-    Input::Call call, FunctionPosition pos, ImplOrTraitItemNode i, Function f, int rnk
+    Input::Call call, ImplOrTraitItemNode i, Function f, int rnk
   ) {
-    ArgIsInstantiationOfToIndex::argIsInstantiationOf(MkCallAndPos(call, pos), i, _) and
-    toCheckRanked(i, f, _, pos, rnk)
+    exists(FunctionPosition posAdj |
+      ArgIsInstantiationOfToIndex::argIsInstantiationOf(MkCallAndPos(call, posAdj), i, _) and
+      toCheckRanked(i, f, _, posAdj, rnk)
+    )
   }
 
   pragma[nomagic]
   private predicate argsAreInstantiationsOfToIndex(
     Input::Call call, ImplOrTraitItemNode i, Function f, int rnk
   ) {
-    exists(FunctionPosition pos |
-      argIsInstantiationOf(call, pos, i, f, rnk) and
-      call.hasTargetCand(i, f)
-    |
+    argIsInstantiationOf(call, i, f, rnk) and
+    call.hasTargetCand(i, f) and
+    (
       rnk = 0
       or
       argsAreInstantiationsOfToIndex(call, i, f, rnk - 1)
@@ -467,9 +472,9 @@ module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
 
   pragma[nomagic]
   private predicate argsAreNotInstantiationsOf0(
-    Input::Call call, FunctionPosition pos, ImplOrTraitItemNode i
+    Input::Call call, FunctionPosition posAdj, ImplOrTraitItemNode i
   ) {
-    ArgsAreNotInstantiationOf::argIsNotInstantiationOf(MkCallAndPos(call, pos), i, _, _)
+    ArgsAreNotInstantiationOf::argIsNotInstantiationOf(MkCallAndPos(call, posAdj), i, _, _)
   }
 
   /**
@@ -480,10 +485,10 @@ module ArgsAreInstantiationsOf<ArgsAreInstantiationsOfInputSig Input> {
    */
   pragma[nomagic]
   predicate argsAreNotInstantiationsOf(Input::Call call, ImplOrTraitItemNode i, Function f) {
-    exists(FunctionPosition pos |
-      argsAreNotInstantiationsOf0(call, pos, i) and
+    exists(FunctionPosition posAdj |
+      argsAreNotInstantiationsOf0(call, posAdj, i) and
       call.hasTargetCand(i, f) and
-      Input::toCheck(i, f, _, pos)
+      Input::toCheck(i, f, _, posAdj)
     )
   }
 }