From 9926ca5de732db0f146a48631a4677b3cfbe83be Mon Sep 17 00:00:00 2001 From: Mystikfluu Date: Tue, 23 May 2023 19:49:29 +0200 Subject: [PATCH] ask for server ip --- src/main.rs | 19 ++++++++++++++++++- 1 file changed, 18 insertions(+), 1 deletion(-) diff --git a/src/main.rs b/src/main.rs index e843951..acd06f0 100644 --- a/src/main.rs +++ b/src/main.rs @@ -3,6 +3,7 @@ use std::{net::{UdpSocket, SocketAddr}, io::Write}; use serde::{Serialize, Deserialize}; use sha3::{Digest, Sha3_512}; use sha2::Sha512; +use std::io::{self, BufRead}; mod big_array; use big_array::BigArray; @@ -28,10 +29,26 @@ struct Packet { payload: [u8; MAX_PAYLOAD_U], //488 bytes } //512 bytes +fn read_stdin(message: String) -> String { + print!("{}", message); + io::stdout().flush().unwrap(); + + let mut buffer = String::new(); + let stdin = io::stdin(); + let mut handle = stdin.lock(); + + handle.read_line(&mut buffer).unwrap(); + buffer.trim().to_string() +} + fn main() { let timeout = 1; - let server_ip = "213.47.107.152:1337"; + let mut server_ip = read_stdin("Enter server ip: ".to_string());//e.g. 213.47.107.152:1337 + if server_ip.find(":").is_none() { + server_ip.push_str(":1337"); //default port + } + let filename = "data.bin";