diff --git a/README.md b/README.md index d855edd..76bf6fa 100644 --- a/README.md +++ b/README.md @@ -30,7 +30,7 @@ USAGE: fragilebyte [OPTIONS] OPTIONS: - -b, --buffersize Buffersize used in the file transfer (bytes) [default: 8192] + -b, --chunksize Chunksize used in the file transfer (bytes) [default: 8192] -f, --fileroot Path to the folder where the files are outputted as a client or served from as a server [default: './output' / './data'] -h, --help Print help information diff --git a/src/client.rs b/src/client.rs index 6f78fb5..ca37dcf 100644 --- a/src/client.rs +++ b/src/client.rs @@ -45,13 +45,13 @@ pub async fn connect( } } - // Receive buffersize - let buffersize = recv_msg_string(&mut reader, &mut buf) + // Receive chunksize + let chunksize = recv_msg_string(&mut reader, &mut buf) .await? .parse::()?; - println!("[+] Selected buffersize: {}", buffersize); + println!("[+] Selected chunksize: {}", chunksize); - // ACK buffersize + // ACK chunksize send_msg(&mut writer, "ACK\n").await?; // Receive metadata @@ -69,7 +69,7 @@ pub async fn connect( &mut reader, &mut writer, rx, - &buffersize, + &chunksize, &metadata, &fileroot, &download_all, @@ -224,7 +224,7 @@ async fn handle_file_reqs( reader: &mut BufReader>, writer: &mut BufWriter>, rx: Receiver, - buffersize: &usize, + chunksize: &usize, metadata: &HashMap, fileroot: &PathBuf, download_all: &bool, @@ -268,7 +268,7 @@ async fn handle_file_reqs( // Receive the file itself let filesize = metadata.get(input_string.as_str()).unwrap().clone(); - receive_file(reader, &mut file_buf, &filesize, buffersize).await?; + receive_file(reader, &mut file_buf, &filesize, chunksize).await?; // ACK file send_msg(writer, "ACK\n").await?; @@ -285,13 +285,13 @@ async fn receive_file( reader: &mut BufReader>, file_buf: &mut BufWriter, filesize: &u64, - buffersize: &usize, + chunksize: &usize, ) -> Result<(), Box> { let mut remaining_data = *filesize; - let mut buf = vec![0u8; *buffersize]; + let mut buf = vec![0u8; *chunksize]; while remaining_data != 0 { - if remaining_data >= *buffersize as u64 { + if remaining_data >= *chunksize as u64 { let read_result = reader.read(&mut buf); match read_result.await { diff --git a/src/main.rs b/src/main.rs index 6ff33fd..7f1098f 100644 --- a/src/main.rs +++ b/src/main.rs @@ -16,8 +16,8 @@ struct Args { /// Port where the service is hosted port: u16, #[clap(default_value_t = 8192usize, short = 'b', long, value_parser = validate_arg::)] - /// Buffersize used in the file transfer (bytes) - buffersize: usize, + /// Chunksize used in the file transfer (bytes) + chunksize: usize, #[clap(default_value_t = false, long, action)] /// Run only in the local network localhost: bool, @@ -66,7 +66,7 @@ async fn main() -> Result<(), Box> { server::listen( args.port, fileroot, - args.buffersize, + args.chunksize, args.localhost, args.timeout, false, diff --git a/src/server.rs b/src/server.rs index 01f0bd2..57b861f 100644 --- a/src/server.rs +++ b/src/server.rs @@ -23,7 +23,7 @@ use tokio::{ pub async fn listen( port: u16, fileroot: PathBuf, - buffersize: usize, + chunksize: usize, localhost: bool, timeout_duration: u64, use_testing_key: bool, @@ -79,10 +79,10 @@ pub async fn listen( return Ok::<(), Box>(()); } - // Send buffersize - send_msg(&mut writer, (buffersize.to_string() + "\n").as_str()).await?; + // Send chunksize + send_msg(&mut writer, (chunksize.to_string() + "\n").as_str()).await?; - // ACK buffersize + // ACK chunksize if recv_msg_string(&mut reader, &mut vec_buf).await? != "ACK" { return Ok::<(), Box>(()); } @@ -104,7 +104,7 @@ pub async fn listen( &mut writer, &mut vec_buf, &alt_fileroot, - &buffersize, + &chunksize, &addr, ) .await? @@ -239,7 +239,7 @@ async fn handle_file_reqs( writer: &mut BufWriter>, buf: &mut Vec, fileroot: &PathBuf, - buffersize: &usize, + chunksize: &usize, addr: &SocketAddr, ) -> Result, Box> { loop { @@ -256,7 +256,7 @@ async fn handle_file_reqs( println!("\n[REQ] {}: {:#?}", addr, input_path); let mut file = File::open(input_path.clone()).await?; let mut remaining_data = file.metadata().await?.len(); - let mut filebuf = vec![0u8; *buffersize]; + let mut filebuf = vec![0u8; *chunksize]; // Serve the file itself while remaining_data != 0 {