Add sourceline output and logging to coverage example (#2753)

Author: ranweiler

src/agent/Cargo.lock

@@ -24,4 +24,5 @@ procfs = { version = "0.12", default-features = false, features=["flate2"] }
 
 [dev-dependencies]
 clap = { version = "4.0", features = ["derive"] }
+env_logger = "0.10.0"
 pretty_assertions = "1.3.0"

src/agent/coverage/Cargo.toml

@@ -19,6 +19,9 @@ struct Args {
     #[arg(short, long)]
     timeout: Option<u64>,
 
+    #[arg(short, long)]
+    source: bool,
+
     command: Vec<String>,
 }
 
@@ -27,6 +30,8 @@ const DEFAULT_TIMEOUT: Duration = Duration::from_secs(5);
 fn main() -> Result<()> {
     let args = Args::parse();
 
+    env_logger::init();
+
     let timeout = args
         .timeout
         .map(Duration::from_millis)
@@ -54,7 +59,11 @@ fn main() -> Result<()> {
         .timeout(timeout)
         .record()?;
 
-    dump_modoff(&recorded.coverage)?;
+    if args.source {
+        dump_source_line(&recorded.coverage)?;
+    } else {
+        dump_modoff(&recorded.coverage)?;
+    }
 
     Ok(())
 }
@@ -70,3 +79,15 @@ fn dump_modoff(coverage: &BinaryCoverage) -> Result<()> {
 
     Ok(())
 }
+
+fn dump_source_line(binary: &BinaryCoverage) -> Result<()> {
+    let source = coverage::source::binary_to_source_coverage(binary)?;
+
+    for (path, file) in &source.files {
+        for (line, count) in &file.lines {
+            println!("{}:{} {}", path, line.number(), count.0);
+        }
+    }
+
+    Ok(())
+}