Rust: Debug preds

Author: paldepind

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

@@ -215,6 +215,13 @@ private module Input2 implements InputSig2 {
       constraint = object.getTrait()
     )
   }
+
+  private predicate testConditionSatisfiesConstraint(
+    TypeAbstraction abs, TypeMention condition, TypeMention constraint
+  ) {
+    conditionSatisfiesConstraint(abs, condition, constraint) and
+    constraint.resolveType().(TraitType).getTrait().getName().getText() = "TryFuture"
+  }
 }
 
 private module M2 = Make2<Input2>;
@@ -2638,8 +2645,9 @@ private module Debug {
   private Locatable getRelevantLocatable() {
     exists(string filepath, int startline, int startcolumn, int endline, int endcolumn |
       result.getLocation().hasLocationInfo(filepath, startline, startcolumn, endline, endcolumn) and
-      filepath.matches("%/sqlx.rs") and
-      startline = [56 .. 60]
+      filepath.matches("%/path_utils.rs") and
+      startline = [72 .. 75]
+      // startline = [91 .. 97]
     )
   }
 
@@ -2670,6 +2678,14 @@ private module Debug {
     t = inferCallExprBaseType(n, path)
   }
 
+  predicate debugInferCallExprBaseMatchingType(AstNode n, TypePath path, Type t) {
+    n = getRelevantLocatable() and
+    exists(CallExprBaseMatchingInput::Access a, CallExprBaseMatchingInput::AccessPosition apos |
+      n = a.getNodeAt(apos) and
+      t = CallExprBaseMatching::inferAccessType(a, apos, path)
+    )
+  }
+
   predicate debugTypeMention(TypeMention tm, TypePath path, Type type) {
     tm = getRelevantLocatable() and
     tm.resolveTypeAt(path) = type

shared/typeinference/codeql/typeinference/internal/TypeInference.qll

@@ -784,6 +784,13 @@ module Make1<LocationSig Location, InputSig1<Location> Input1> {
         )
       }
 
+      predicate debugConditionSatisfiesConstraintTypeAt(
+        TypeAbstraction abs, TypeMention condition, TypeMention constraint, TypePath path, Type t
+      ) {
+        conditionSatisfiesConstraintTypeAt(abs, condition, constraint, path, t) and
+        constraint.toString().matches("%AsRef%")
+      }
+
       /**
        * Holds if its possible for a type with `conditionRoot` at the root to
        * satisfy a constraint with `constraintRoot` at the root through `abs`,