The program get the server address in config file
This commit is contained in:
25
src/main.rs
25
src/main.rs
@@ -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());
|
||||
|
||||
Reference in New Issue
Block a user