[feat] adds time in udp message logs

This commit is contained in:
namu
2025-03-09 00:55:32 +01:00
parent a7197fc0b4
commit f9e0090a52
3 changed files with 300 additions and 23 deletions

View File

@@ -1,6 +1,6 @@
use std::net::UdpSocket;
use chrono;
use colored::Colorize;
use std::net::UdpSocket;
use crate::config::Config;
@@ -16,34 +16,38 @@ fn listen(config: &Config) {
let serv_config = config;
let addr = config.get_server_addr();
println!("{}: {}", "listening on".green(), serv_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.");
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)
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"));
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();
let forwarding_addr = forwarding_addr_opt.unwrap();
socket.send_to(message.as_bytes(), forwarding_addr)
socket
.send_to(message.as_bytes(), forwarding_addr)
.expect("Error forwarding message to another ip.");
}
println!("{}: {}", src_addr.to_string().green(), message);
println!("{}-{}: {}", src_addr.to_string().green(), chrono::Local::now(), message);
}
}
}