Rust: exclude all function, const, and static bodies for dependencies

Author: aibaars

rust/extractor/src/translate/base.rs

@@ -16,7 +16,7 @@ use ra_ap_ide_db::RootDatabase;
 use ra_ap_ide_db::line_index::{LineCol, LineIndex};
 use ra_ap_parser::SyntaxKind;
 use ra_ap_span::TextSize;
-use ra_ap_syntax::ast::HasName;
+use ra_ap_syntax::ast::{Const, Fn, HasName, Static};
 use ra_ap_syntax::{
     AstNode, NodeOrToken, SyntaxElementChildren, SyntaxError, SyntaxNode, SyntaxToken, TextRange,
     ast,
@@ -598,7 +598,34 @@ impl<'a> Translator<'a> {
         })();
     }
 
-    pub(crate) fn should_be_excluded(&self, item: &impl ast::HasAttrs) -> bool {
+    pub(crate) fn should_be_excluded(&self, item: &(impl ast::HasAttrs + ast::AstNode)) -> bool {
+        if "true"
+            == std::env::var("CODEQL_EXTRACTOR_RUST_OPTION_EXCLUDE_BODIES").unwrap_or_default()
+        {
+            let syntax = item.syntax();
+            if let Some(body) = syntax.parent().and_then(Fn::cast).and_then(|x| x.body()) {
+                if body.syntax() == syntax {
+                    tracing::debug!("Skipping Fn body");
+                    return true;
+                }
+            }
+            if let Some(body) = syntax.parent().and_then(Const::cast).and_then(|x| x.body()) {
+                if body.syntax() == syntax {
+                    tracing::debug!("Skipping Const body");
+                    return true;
+                }
+            }
+            if let Some(body) = syntax
+                .parent()
+                .and_then(Static::cast)
+                .and_then(|x| x.body())
+            {
+                if body.syntax() == syntax {
+                    tracing::debug!("Skipping Static body");
+                    return true;
+                }
+            }
+        }
         self.semantics.is_some_and(|sema| {
             item.attrs().any(|attr| {
                 attr.as_simple_call().is_some_and(|(name, tokens)| {

rust/tools/autobuild.py

@@ -20,17 +20,27 @@ def get_cargo_metadata():
 scratch_dir = os.environ.get("CODEQL_EXTRACTOR_RUST_SCRATCH_DIR")
 metadata = get_cargo_metadata()
 metadata_file = os.path.join(scratch_dir, "metadata_file.yaml")
+os.makedirs(scratch_dir, exist_ok=True)
 with open(metadata_file, "w") as f:
     f.write("---\n")
     f.write(json.dumps(metadata, indent=4))
 
 subprocess.run(["codeql", "database", "index-files", database,
                "-lyaml", "--working-dir", scratch_dir, "--include", "metadata_file.yaml"])
+paths = set()
 for package in metadata['packages']:
     for target in package['targets']:
-        if 'lib' in target['kind']:
+        if 'lib' in target['kind'] or 'bin' in target['kind']:
             src_path = target['src_path']
-            dir = os.path.dirname(src_path)
-            autobuild = "{root}/tools/{platform}/autobuild".format(
-                root=CODEQL_EXTRACTOR_RUST_ROOT, platform=CODEQL_PLATFORM)
-            subprocess.run([autobuild], cwd=dir)
+            paths.add(os.path.dirname(src_path))
+
+autobuild = "{root}/tools/{platform}/autobuild".format(
+    root=CODEQL_EXTRACTOR_RUST_ROOT, platform=CODEQL_PLATFORM)
+
+for path in paths:
+    if not path.startswith(os.getcwd()):
+        env = os.environ.copy()
+        env['CODEQL_EXTRACTOR_RUST_OPTION_EXCLUDE_BODIES'] = "true"
+    else:
+        env = None
+    subprocess.run([autobuild], cwd=path, env=env)