[refactor] move listen function in server module
This commit is contained in:
47
src/main.rs
47
src/main.rs
@@ -1,53 +1,10 @@
|
||||
use chrono;
|
||||
use colored::Colorize;
|
||||
use std::net::UdpSocket;
|
||||
|
||||
use crate::config::Config;
|
||||
|
||||
mod config;
|
||||
mod vec_to_string;
|
||||
mod server;
|
||||
|
||||
fn main() {
|
||||
let config = Config::from_json("config.json".to_string());
|
||||
listen(&config);
|
||||
}
|
||||
|
||||
fn listen(config: &Config) {
|
||||
let serv_config = config;
|
||||
let addr = config.get_server_addr();
|
||||
|
||||
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.");
|
||||
|
||||
const BUF_SIZE: usize = 1024;
|
||||
|
||||
loop {
|
||||
let mut message: String = String::new();
|
||||
let mut buf: [u8; BUF_SIZE] = [0; BUF_SIZE];
|
||||
let (num_bytes, src_addr) = socket
|
||||
.recv_from(&mut buf)
|
||||
.expect("Error while receiving message");
|
||||
|
||||
if !serv_config.addr_is_blacklisted(&addr) {
|
||||
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());
|
||||
|
||||
if forwarding_addr_opt.is_some() {
|
||||
let forwarding_addr = forwarding_addr_opt.unwrap();
|
||||
|
||||
socket
|
||||
.send_to(message.as_bytes(), forwarding_addr)
|
||||
.expect("Error forwarding message to another ip.");
|
||||
}
|
||||
|
||||
println!("{}-{}: {}", src_addr.to_string().green(), chrono::Local::now(), message);
|
||||
}
|
||||
}
|
||||
server::listen(&config);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user