X-Git-Url: http://git.euphorik.ch/?p=crypto_lab1.git;a=blobdiff_plain;f=src%2Fmain.rs;h=1fae4c1838621f1c2456490b87d6455f25733a74;hp=5f2fdf93df305e1e64b90692ee699c8ce30cf831;hb=911f8410c05d3d16922587a59266096f3c6899e1;hpb=8226b0eced022af21512d87902bf6e252afe1a67 diff --git a/src/main.rs b/src/main.rs index 5f2fdf9..1fae4c1 100644 --- a/src/main.rs +++ b/src/main.rs @@ -12,21 +12,15 @@ mod crypto; mod packet; mod end_point; -/* -TODO - * Comment stocker les clefs? à quels critères doivent elle répondre? - * -*/ - const PORT: u16 = 4221; fn print_usage() { println!("{} | | ...", os::args()[0]); } -fn main() { +fn main() { let args = os::args(); - + if args.iter().any(|a| a.as_slice() == "--help" || a.as_slice() == "-h") { print_usage(); } else if args.len() > 1 && args[1].as_slice() == "genkey" { @@ -35,13 +29,17 @@ fn main() { Err(e) => println!("Unable to generate a key. Error: {}", e) } } else { - println!("Starting server..., Press any key to quit"); - + println!("Starting server..."); + match Server::new("::1", PORT) { Ok(mut server) => { println!("Server started"); - Client::start_tests("::1", PORT); - io::stdin().read_line().ok().expect("Failed to read line"); + + Client::start_tests("::1", PORT); + + println!("Press any key to quit"); + io::stdin().read_line().ok().expect("Failed to read line"); + server.close().ok().expect("Failed to close the server"); }, Err(e) =>