Tahinli/rust-tcp-file-transfer
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

need_parser_or_smthng_to_send_file_size_correctly

Browse source
This commit is contained in:
Ahmet Kaan GÜMÜŞ 2023-10-09 01:51:15 +03:00
parent 1e2e499352
commit 4a6f1bec74
1 changed files with 85 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

80
src/main.rs
View file

@ -5,7 +5,7 @@ use std::io::{Read, Write, self};
use std::env::{self}; use std::env::{self};
const BUFFER_SIZE:usize = 1024; const BUFFER_SIZE:usize = 100000;
struct FileInfo struct FileInfo
{ {
@ -72,9 +72,32 @@ impl FileInfo
} }
fn file_to_byte(&mut self, stream:&mut TcpStream) fn file_to_byte(&mut self, stream:&mut TcpStream)
{ {
//replace with readbuffer ? //replace with readbuffer ?
let mut iteration = (self.metadata.as_ref().unwrap().len()/BUFFER_SIZE as u64)+1; let mut iteration = (self.metadata.as_ref().unwrap().len()/BUFFER_SIZE as u64)+1;
let mut handshake = [0u8;BUFFER_SIZE];
handshake[..self.metadata.as_ref().unwrap().len().to_string().as_bytes().len()].copy_from_slice(self.metadata.as_ref().unwrap().len().to_string().as_bytes());
match stream.write_all(&mut handshake)
{
Ok(_) =>
{
let mut handshake_callback = [0u8;BUFFER_SIZE];
match stream.read_exact(&mut handshake_callback)
{
Ok(_) =>
{
println!("Done: Handshake -> {}", self.location);
}
Err(err_val) =>
{
println!("Error: Handshake Recv -> {} | Error: {}", self.location, err_val);
}
}
}
Err(err_val) =>
{
println!("Error: Handshake Send -> {} | Error: {}", self.location, err_val);
}
}
while iteration != 0 while iteration != 0
{ {
iteration -= 1; iteration -= 1;
@ -90,19 +113,29 @@ impl FileInfo
{ {
Ok(_) => Ok(_) =>
{ {
println!("Done: Send Bytes -> {}", self.location); println!("Done: Send Bytes -> {} | Iteration = {}", self.location, iteration);
} }
Err(err_val) => Err(err_val) =>
{ {
println!("Error: Send Bytes -> {} | Error: {}", self.location, err_val); println!("Error: Send Bytes -> {} | Error: {}", self.location, err_val);
return;
}
}
match stream.flush()
{
Ok(_) =>
{
println!("Done: Flush -> {}", self.location);
}
Err(err_val) =>
{
println!("Error: Flush -> {} | Error: {}", self.location, err_val);
} }
} }
} }
else else
{ {
println!("Son");
let mut last_buffer:Vec<u8> = (&buffer[..(self.metadata.as_ref().unwrap().len()%BUFFER_SIZE as u64)as usize]).to_vec(); let mut last_buffer:Vec<u8> = (&buffer[..(self.metadata.as_ref().unwrap().len()%BUFFER_SIZE as u64)as usize]).to_vec();
println!("VEC = {:#?}", last_buffer);
match stream.write_all(&mut last_buffer) match stream.write_all(&mut last_buffer)
{ {
Ok(_) => Ok(_) =>
@ -128,15 +161,45 @@ impl FileInfo
} }
fn write_file(&mut self, stream:&mut TcpStream) fn write_file(&mut self, stream:&mut TcpStream)
{ {
//use match, there is a chance to fail creation. don't pass with just some //use match, there is a chance to fail creation. don't pass with just some
self.file = Some(File::create(&self.location).expect("Error: Create File")); self.file = Some(File::create(&self.location).expect("Error: Create File"));
match self.file match self.file
{ {
Some(ref mut file_descriptor) => Some(ref mut file_descriptor) =>
{ {
let mut buffer: Vec<u8> = Vec::new(); let mut handshake = [0u8;BUFFER_SIZE];
match stream.read_to_end(&mut buffer) let mut size:u64 = 0;
match stream.read_exact(&mut handshake)
{
Ok(_) =>
{
todo!();
size = String::from_utf8(handshake.to_vec()).unwrap().parse().unwrap();
println!("Done: Handshake Recv -> {}", self.location);
match stream.write_all(&mut handshake)
{
Ok(_) =>
{
println!("Done: Handshake Send -> {}", self.location);
}
Err(err_val) =>
{
println!("Error: Handshake Send -> {} | Error: {}", self.location, err_val);
}
}
}
Err(err_val) =>
{
println!("Error: Handshake Recv -> {} | Error: {}", self.location, err_val);
}
}
let mut iteration:u64 = (size%BUFFER_SIZE as u64)+1;
while iteration != 0
{
iteration -= 1;
//let mut buffer: Vec<u8> = Vec::new();
let mut buffer = [0u8;BUFFER_SIZE];
match stream.read(&mut buffer)
{ {
Ok(_) => Ok(_) =>
{ {
@ -156,6 +219,7 @@ impl FileInfo
} }
} }
} }
}
None => None =>
{ {
println!("Error: File None"); println!("Error: File None");