Fix testcase set_filename

libafl/src/corpus/ondisk.rs

@@ -229,7 +229,7 @@ where
             };
 
             let filename_str = filename.to_str().expect("Invalid Path");
-            testcase.set_filename(filename_str.into());
+            testcase.set_filename(filename_str.into())?;
         };
         if self.meta_format.is_some() {
             let mut filename = PathBuf::from(testcase.filename().as_ref().unwrap());

libafl/src/corpus/testcase.rs

@@ -4,6 +4,9 @@
 use alloc::string::String;
 use core::{default::Default, option::Option, time::Duration};
 
+#[cfg(feature = "std")]
+use std::{fs, path::Path};
+
 use serde::{Deserialize, Serialize};
 
 use crate::{
@@ -123,8 +126,35 @@ where
 
     /// Set the filename
     #[inline]
-    pub fn set_filename(&mut self, filename: String) {
+    #[cfg(feature = "std")]
+    pub fn set_filename(&mut self, filename: String) -> Result<(), Error> {
+        if self.filename.is_some() {
+            let f = self.filename.clone().unwrap();
+            let old_filename = f.as_str();
+
+            let new_filename = filename.as_str();
+
+            if old_filename.eq(&filename) {
+                return Ok(());
+            }
+
+            let old_lock_filename = format!(".{old_filename}.lafl_lock");
+            if Path::new(&old_lock_filename).exists() {
+                return Err(Error::illegal_state("old lockfile exist"));
+            }
+            let new_lock_filename = format!(".{new_filename}.lafl_lock");
+            if Path::new(&new_lock_filename).exists() {
+                return Err(Error::illegal_state("new lockfile exist"));
+            }
+            fs::rename(old_filename, new_filename)?;
+
+            let old_metadata_filename = format!(".{old_filename}.metadata");
+            let new_metadata_filename = format!(".{new_filename}.metadata");
+            fs::rename(old_metadata_filename, new_metadata_filename)?;
+        }
+
         self.filename = Some(filename);
+        Ok(())
     }
 
     /// Get the execution time of the testcase