Rust: add some more diagnostics queries

Author: aibaars

rust/ql/src/queries/diagnostics/ExtractionErrors.ql

@@ -0,0 +1,18 @@
+/**
+ * @name Extraction errors
+ * @description List all extraction errors for files in the source code directory.
+ * @kind diagnostic
+ * @id rust/diagnostics/extraction-errors
+ */
+
+import codeql.rust.Diagnostics
+import codeql.files.FileSystem
+
+/** Gets the SARIF severity to associate an error. */
+int getSeverity() { result = 2 }
+
+from ExtractionError error, File f
+where
+  f = error.getLocation().getFile() and
+  exists(f.getRelativePath())
+select error, "Extraction failed in " + f + " with error " + error.getMessage(), getSeverity()

rust/ql/src/queries/summary/NumberOfFilesExtractedWithErrors.ql

@@ -0,0 +1,15 @@
+/**
+ * @id rust/summary/number-of-files-extracted-with-errors
+ * @name Total number of Rust files that were extracted with errors
+ * @description The total number of Rust code files that we extracted, but where
+ *  at least one extraction error occurred in the process.
+ * @kind metric
+ * @tags summary
+ */
+
+import codeql.files.FileSystem
+import codeql.rust.Diagnostics
+
+select count(File f |
+    exists(ExtractionError e | e.getLocation().getFile() = f) and exists(f.getRelativePath())
+  )

rust/ql/src/queries/summary/NumberOfSuccessfullyExtractedFiles.ql

@@ -0,0 +1,15 @@
+/**
+ * @id rust/summary/number-of-successfully-extracted-files
+ * @name Total number of Rust files that were extracted without error
+ * @description The total number of Rust code files that we extracted without
+ *   encountering any extraction errors
+ * @kind metric
+ * @tags summary
+ */
+
+import codeql.rust.Diagnostics
+import codeql.files.FileSystem
+
+select count(File f |
+    not exists(ExtractionError e | e.getLocation().getFile() = f) and exists(f.getRelativePath())
+  )

rust/ql/test/query-tests/diagnostics/ExtractionErrors.expected

@@ -0,0 +1,6 @@
+| does_not_compile.rs:2:6:2:5 | expected SEMICOLON | Extraction failed in does_not_compile.rs with error expected SEMICOLON | 2 |
+| does_not_compile.rs:2:9:2:8 | expected SEMICOLON | Extraction failed in does_not_compile.rs with error expected SEMICOLON | 2 |
+| does_not_compile.rs:2:13:2:12 | expected SEMICOLON | Extraction failed in does_not_compile.rs with error expected SEMICOLON | 2 |
+| does_not_compile.rs:2:21:2:20 | expected SEMICOLON | Extraction failed in does_not_compile.rs with error expected SEMICOLON | 2 |
+| does_not_compile.rs:2:26:2:25 | expected SEMICOLON | Extraction failed in does_not_compile.rs with error expected SEMICOLON | 2 |
+| does_not_compile.rs:2:32:2:31 | expected field name or number | Extraction failed in does_not_compile.rs with error expected field name or number | 2 |

rust/ql/test/query-tests/diagnostics/ExtractionErrors.qlref

@@ -0,0 +1 @@
+queries/diagnostics/ExtractionErrors.ql