diff --git a/Lucien/Rust/src/client.rs b/Lucien/Rust/src/client.rs index 6da8106..694ba30 100644 --- a/Lucien/Rust/src/client.rs +++ b/Lucien/Rust/src/client.rs @@ -19,14 +19,12 @@ fn write_to_server(stream: TcpStream) { match &*get_entry() { "/quit" => { println!("Disconnecting..."); - // send!("BYE"); writeln!(writer, "BYE").unwrap(); writer.flush().unwrap(); println!("Disconnected!"); return (); } line => { - // send!(line); writeln!(writer, "{}", line).unwrap(); writer.flush().unwrap(); } diff --git a/Lucien/Rust/src/main.rs b/Lucien/Rust/src/main.rs index 0b90c76..df85bcb 100644 --- a/Lucien/Rust/src/main.rs +++ b/Lucien/Rust/src/main.rs @@ -21,7 +21,11 @@ fn main() { // Client opened // /////////////////////////////////////////////////////////////////////// println!("Client connecting on server {}:{}", args[1], args[2]); - let mut serv = args[1].clone(); + let mut serv: String = if args[1] == "localhost" { + String::from("127.0.0.1") + } else { + args[1].clone() + }; serv.push(':'); serv.push_str(&args[2]); client::client(serv);