YGGverse/titanit
1
0
Fork 0
mirror of https://github.com/YGGverse/titanit.git synced 2026-03-31 17:15:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

handle errors

Browse source
This commit is contained in:
yggverse 2025-02-23 06:29:57 +02:00
parent eff0b836f8
commit f1f419481f
1 changed files with 18 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

21
src/main.rs
View file

@ -17,7 +17,6 @@ use std::{
fn main() -> Result<()> { fn main() -> Result<()> {
use clap::Parser; use clap::Parser;
let argument = Arc::new(Argument::parse()); let argument = Arc::new(Argument::parse());
// https://geminiprotocol.net/docs/protocol-specification.gmi#the-use-of-tls // https://geminiprotocol.net/docs/protocol-specification.gmi#the-use-of-tls
@ -135,21 +134,27 @@ fn gemini(
) { ) {
Ok(()) => loop { Ok(()) => loop {
let mut data = vec![0; argument.chunk]; let mut data = vec![0; argument.chunk];
let l = item.file.read_at(&mut data, read as u64).unwrap(); match item.file.read_at(&mut data, read as u64) {
stream.write_all(&data[..l]).unwrap(); Ok(l) => match stream.write_all(&data[..l]) {
read += l; Ok(()) => {
// println!("[{}] [info] [{peer}] Chunk sent: {l} ({read} total)", now());
// EOF // EOF
if l == 0 { if l == 0 {
println!("[{}] [info] [{peer}] Response: {read} bytes", now()); println!("[{}] [info] [{peer}] Response: {read} bytes", now());
match close(stream) { match close(stream) {
Ok(()) => { Ok(()) => println!(
println!("[{}] [info] [{peer}] Connection closed by server.", now()) "[{}] [info] [{peer}] Connection closed by server.",
} now()
),
Err(e) => println!("[{}] [warning] [{peer}] {e}", now()), Err(e) => println!("[{}] [warning] [{peer}] {e}", now()),
} }
break; break;
} }
read += l;
}
Err(e) => println!("[{}] [error] [{peer}] {e}", now()),
},
Err(e) => println!("[{}] [error] [{peer}] {e}", now()),
}
}, },
Err(e) => println!("[{}] [error] [{peer}] {e}", now()), Err(e) => println!("[{}] [error] [{peer}] {e}", now()),
} }