Add coverage file path configuration and rename filename to path in DrCovModule of QEMU (#3525)

* feat(qemu): in drcov module - rename filename to path, add methods

* feat(qemu): use generic arg's type for path method of DrCovModule

Author: saruman9

crates/libafl_qemu/src/modules/drcov.rs

@@ -3,7 +3,10 @@ use std::{
     cmp::{max, min},
     ops::Range,
 };
-use std::{path::PathBuf, sync::Mutex};
+use std::{
+    path::{Path, PathBuf},
+    sync::Mutex,
+};
 
 use hashbrown::{HashMap, hash_map::Entry};
 use libafl::{HasMetadata, executors::ExitKind, observers::ObserversTuple};
@@ -55,7 +58,7 @@ libafl_bolts::impl_serdeany!(DrCovMetadata);
 pub struct DrCovModuleBuilder<F> {
     filter: Option<F>,
     module_mapping: Option<RangeMap<u64, (u16, String)>>,
-    filename: Option<PathBuf>,
+    path: Option<PathBuf>,
     full_trace: bool,
 }
 
@@ -66,7 +69,7 @@ where
     pub fn build(self) -> DrCovModule<F> {
         DrCovModule::new(
             self.filter.unwrap(),
-            self.filename.unwrap(),
+            self.path.unwrap(),
             self.module_mapping,
             self.full_trace,
         )
@@ -76,7 +79,7 @@ where
         DrCovModuleBuilder {
             filter: Some(filter),
             module_mapping: self.module_mapping,
-            filename: self.filename,
+            path: self.path,
             full_trace: self.full_trace,
         }
     }
@@ -86,17 +89,17 @@ where
         Self {
             filter: self.filter,
             module_mapping: Some(module_mapping),
-            filename: self.filename,
+            path: self.path,
             full_trace: self.full_trace,
         }
     }
 
     #[must_use]
-    pub fn filename(self, filename: PathBuf) -> Self {
+    pub fn path<P: Into<PathBuf>>(self, path: P) -> Self {
         Self {
             filter: self.filter,
             module_mapping: self.module_mapping,
-            filename: Some(filename),
+            path: Some(path.into()),
             full_trace: self.full_trace,
         }
     }
@@ -106,7 +109,7 @@ where
         Self {
             filter: self.filter,
             module_mapping: self.module_mapping,
-            filename: self.filename,
+            path: self.path,
             full_trace,
         }
     }
@@ -116,7 +119,7 @@ where
 pub struct DrCovModule<F> {
     filter: F,
     module_mapping: Option<RangeMap<u64, (u16, String)>>,
-    filename: PathBuf,
+    path: PathBuf,
     full_trace: bool,
     drcov_len: usize,
 }
@@ -376,16 +379,16 @@ impl DrCovModule<NopAddressFilter> {
             filter: Some(NopAddressFilter),
             module_mapping: None,
             full_trace: false,
-            filename: None,
+            path: None,
         }
     }
 }
 
 impl<F> DrCovModule<F> {
     #[must_use]
-    pub fn new(
+    pub fn new<P: Into<PathBuf>>(
         filter: F,
-        filename: PathBuf,
+        path: P,
         module_mapping: Option<RangeMap<u64, (u16, String)>>,
         full_trace: bool,
     ) -> Self {
@@ -399,12 +402,20 @@ impl<F> DrCovModule<F> {
         Self {
             filter,
             module_mapping,
-            filename,
+            path: path.into(),
             full_trace,
             drcov_len: 0,
         }
     }
 
+    pub fn set_path<P: Into<PathBuf>>(&mut self, path: P) {
+        self.path = path.into();
+    }
+
+    pub fn path(&self) -> &Path {
+        self.path.as_path()
+    }
+
     pub fn flush(&mut self) {
         let lengths_opt = DRCOV_LENGTHS.lock().unwrap();
         let lengths = lengths_opt.as_ref().unwrap();
@@ -451,7 +462,7 @@ impl<F> DrCovModule<F> {
                 // Module mapping is already set. It's checked or filled when the module is first run.
                 unsafe {
                     DrCovWriter::new(self.module_mapping.as_ref().unwrap_unchecked())
-                        .write(&self.filename, &drcov_vec)
+                        .write(&self.path, &drcov_vec)
                         .expect("Failed to write coverage file");
                 }
             }
@@ -499,7 +510,7 @@ impl<F> DrCovModule<F> {
                 // Module mapping is already set. It's checked or filled when the module is first run.
                 unsafe {
                     DrCovWriter::new(self.module_mapping.as_ref().unwrap_unchecked())
-                        .write(&self.filename, &drcov_vec)
+                        .write(&self.path, &drcov_vec)
                         .expect("Failed to write coverage file");
                 }
             }