Rust: Repair after Unimplemented.getLocation was removed.

github · Sep 16, 2024 · fb6fbf6 · fb6fbf6
1 parent ab4788a
commit fb6fbf6
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 1,707 deletions.
diff --git a/rust/ql/src/queries/diagnostics/MissingElements.ql b/rust/ql/src/queries/diagnostics/MissingElements.ql
@@ -6,6 +6,16 @@
 
 import rust
 
+/**
+ * Gets a location for an `Unimplemented` node.
+ */
+Location getUnimplementedLocation(Unimplemented node) {
+  result = node.(Locatable).getLocation()
+  or
+  not node instanceof Locatable and
+  result instanceof EmptyLocation
+}
+
 /**
  * Gets `l.toString()`, but with any locations outside of the source location prefix cleaned up.
  */
@@ -29,7 +39,7 @@ string multipleString(int i) {
 query predicate listUnimplemented(string location, string msg) {
   // something that is not extracted yet
   exists(int c |
-    c = strictcount(Unimplemented n | cleanLocationString(n.getLocation()) = location) and
+    c = strictcount(Unimplemented n | cleanLocationString(getUnimplementedLocation(n)) = location) and
     msg = "Not yet implemented" + multipleString(c) + "."
   )
 }