Skip entire function if entry offset excluded (#3172)

ranweiler · Porges · web-flow · commit a040aa21ed22 · 2023-06-09T19:18:29.000Z
Co-authored-by: George Pollard &lt;gpollard@microsoft.com&gt;
diff --git a/src/agent/coverage/src/binary.rs b/src/agent/coverage/src/binary.rs
@@ -120,6 +120,20 @@ pub fn find_coverage_sites(
     let mut offsets = BTreeSet::new();
 
     for function in debuginfo.functions() {
+        if let Some(location) = symcache.lookup(function.offset.0).next() {
+            if let Some(file) = location.file() {
+                if !allowlist.source_files.is_allowed(file.full_path()) {
+                    debug!(
+                        "skipping sweep of `{}:{}` due to excluded source path `{}`",
+                        module.executable_path(),
+                        function.name,
+                        file.full_path(),
+                    );
+                    continue;
+                }
+            }
+        }
+
         let blocks = block::sweep_region(module, &debuginfo, function.offset, function.size)?;
 
         for block in &blocks {
