Merge pull request #43 from genonullfree/41-bug/truncate

Truncate/extend file to correct size on overwrite Closes #41
genonullfree · Oct 29, 2021 · 553b235 · 553b235
2 parents 2acb2ca + 32d919f
commit 553b235
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "teleporter"
-version = "0.5.0"
+version = "0.5.1"
 authors = ["geno nullfree <[email protected]>"]
 license = "BSD-3-Clause"
 description = "A small utility to send files quickly from point A to point B"

diff --git a/src/server.rs b/src/server.rs
@@ -147,7 +147,9 @@ fn recv(mut stream: TcpStream, recv_list: Arc<Mutex<Vec<String>>>) -> Result<(),
 
     // If overwrite and file exists, build TeleportDelta
     let mut chunk_size = 5120;
+    file.set_len(header.filesize)?;
     if meta.len() > 0 {
+        resp = TeleportInitAck::new(TeleportInitStatus::Overwrite);
         resp.delta = match utils::calc_delta_hash(&file) {
             Ok(d) => {
                 chunk_size = d.delta_size + 1024;