Add output file lock

Cargo.toml

@@ -17,6 +17,7 @@ color-backtrace = "0.6.1"
 criterion = "0.5.1"
 crossterm = "0.27.0"
 dialoguer = "0.11.0"
+fs2 = "0.4.3"
 glob = "0.3.1"
 hex = "0.4.3"
 humantime = "2.1.0"

src/main.rs

@@ -41,6 +41,7 @@ use utils::{
         version::Version,
     },
     color_hash::color_hash,
+    file_lock::FileLock,
     outputs::{
         fingerprint::Fingerprint,
         output::{OpenOutputOptions, Output, OutputOptions},
@@ -668,8 +669,18 @@ fn derive(derive_options: DeriveOptions) {
     let mut output_key = derive_options.output_key;
     let mut checkpoint: Option<Checkpoint> = None;
 
+    let mut _output_lock: Option<FileLock> = None;
     let mut out: Option<Output> = None;
     if let Some(path) = derive_options.output {
+        if path.exists() {
+            panic!("Output file \"{}\" already exists", path.to_string_lossy());
+        }
+
+        _output_lock = match FileLock::try_lock(&path) {
+            Ok(lock) => Some(lock),
+            Err(_) => panic!("Unable to lock {}", path.to_string_lossy()),
+        };
+
         if output_key.is_none() {
             output_key = Some(get_output_key());
         }

src/utils/file_lock.rs

@@ -0,0 +1,43 @@
+use fs2::FileExt;
+use std::fs::File;
+use std::io::{self};
+use std::path::Path;
+
+pub struct FileLock {
+    file: File,
+}
+
+impl FileLock {
+    #[allow(dead_code)]
+    pub fn lock<P: AsRef<Path>>(path: P) -> io::Result<Self> {
+        let file = File::options()
+            .read(true)
+            .write(true)
+            .create(true)
+            .truncate(true)
+            .open(&path)?;
+
+        file.lock_exclusive()?;
+
+        Ok(FileLock { file })
+    }
+
+    pub fn try_lock<P: AsRef<Path>>(path: P) -> io::Result<Self> {
+        let file = File::options()
+            .read(true)
+            .write(true)
+            .create(true)
+            .truncate(true)
+            .open(&path)?;
+
+        file.try_lock_exclusive()?;
+
+        Ok(FileLock { file })
+    }
+}
+
+impl Drop for FileLock {
+    fn drop(&mut self) {
+        let _ = self.file.unlock();
+    }
+}

src/utils/mod.rs

@@ -2,5 +2,6 @@ pub mod algorithms;
 pub mod chacha20poly1305;
 pub mod checkpoints;
 pub mod color_hash;
+pub mod file_lock;
 pub mod outputs;
 pub mod sodium_init;

src/utils/outputs/output.rs

@@ -121,10 +121,6 @@ pub struct Output {
 
 impl Output {
     pub fn new(opts: &OutputOptions) -> Self {
-        if opts.path.exists() {
-            panic!("Output file \"{}\" already exists", opts.path.to_str().unwrap());
-        }
-
         Self {
             path: opts.path.clone(),
             cipher: ChaCha20Poly1305::new(&opts.key),