The program get the server address in config file

This commit is contained in:
Thomas
2024-03-20 14:52:51 +01:00
parent 46e7269a23
commit 8ed155adaa
2 changed files with 19 additions and 18 deletions

View File

@@ -1,33 +1,26 @@
mod config;
use std::io::stdin;
use std::net::UdpSocket;
use colored::Colorize;
use crate::config::Config;
mod config;
fn main() {
let config = Config::from_json("config.json".to_string());
let server_address: String = get_server_address();
listen(&server_address, &config);
listen(&config);
}
fn get_server_address() -> String {
println!("Enter server address (format xxx.xxx.xxx.xxx:port)");
let mut server_address = String::new();
stdin().read_line(&mut server_address).expect("Reading the server address from input");
return server_address.trim().to_string();
}
fn listen(server_address: &str, config: &Config) {
fn listen(config: &Config) {
let serv_config = config;
println!("{}: {}", "listening on".green(), server_address);
let socket: UdpSocket = UdpSocket::bind(server_address).expect("Error while binding the socket.");
println!("{}: {}", "listening on".green(), serv_config.get_server_addr());
let socket: UdpSocket = UdpSocket::bind(serv_config.get_server_addr()).expect("Error while binding the socket.");
loop {
let mut message: String = String::new();
let mut buf: [u8; 1024] = [0; 1024];
let (num_bytes, src_addr) = socket.recv_from(&mut buf).expect("Error while receiving message");
if !serv_config.addr_is_blacklisted(src_addr.to_string()) {
if !serv_config.addr_is_blacklisted(src_addr.to_owned().to_string()) {
message.push_str(std::str::from_utf8(&buf[..num_bytes]).expect("Invalid UTF-8 data"));
let forwarding_addr_opt = serv_config.is_forwarded(src_addr.to_string());