Namularbre/UdpSocketServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Global refactor of the code

Browse Source
This commit is contained in:
namu
2024-11-14 17:38:07 +01:00
parent 2ae74269ea
commit a7197fc0b4
3 changed files with 105 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/main.rs
View File

@@ -5,6 +5,7 @@ use colored::Colorize;
use crate::config::Config;
mod config;
mod vec_to_string;
fn main() {
let config = Config::from_json("config.json".to_string());
@@ -13,20 +14,34 @@ fn main() {
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.");
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; 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_owned().to_string()) {
message.push_str(std::str::from_utf8(&buf[..num_bytes]).expect("Invalid UTF-8 data"));
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.");
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(), message);
}