most is implemented

This commit is contained in:
Martin Brodbeck 2023-10-03 19:53:32 +02:00
parent 95a65f981c
commit 6017a6ca47

View file

@ -1,4 +1,4 @@
use chrono::{DateTime, Local};
use chrono::{DateTime, Local, Duration};
use std::{
collections::{HashMap, HashSet},
net::UdpSocket,
@ -25,7 +25,7 @@ fn strip_header(msg: &[u8]) -> Vec<u8> {
let mut stripped = Vec::new();
stripped.push(msg[1] & 3);
stripped.append(msg[2..].to_vec().as_mut());
print_msg(&stripped);
//print_msg(&stripped);
stripped.to_owned()
}
@ -100,16 +100,41 @@ fn mopp(speed: u8, data: &[u8]) -> Vec<u8> {
m.push_str("00"); // EOC
}
m.push_str("11"); // EOW
m.replace_range(m.len() - 2.., "11"); // EOW
println!("MOPP: {1:0$}", (8 as f32 * (m.len() as f32 / 8 as f32).ceil()) as usize, m);
println!(
"MOPP: {1:0$}",
(8 as f32 * (m.len() as f32 / 8 as f32).ceil()) as usize,
m
);
m.as_bytes().to_owned()
m = format!(
"{1:0<0$}",
(8 as f32 * (m.len() as f32 / 8 as f32).ceil()) as usize,
m
);
let mut res = Vec::<u8>::new();
for i in (0..m.len()).step_by(8) {
let value = u8::from_str_radix(&m[i..i + 8], 2).unwrap();
res.push(value);
}
res.to_owned()
}
fn broadcast(socket: &UdpSocket, client: &Client, data: &[u8]) {}
fn broadcast(socket: &UdpSocket, receivers: &HashSet<Client>, client: &Client, data: &[u8]) {
for rec in receivers {
// Do not broadcast to origin
if rec.name == client.name {
continue;
}
socket.send_to(&data, &rec.name).unwrap();
}
}
fn main() -> std::io::Result<()> {
let socket = UdpSocket::bind("0.0.0.0:7373")?;
socket
.set_read_timeout(None)
@ -130,23 +155,33 @@ fn main() -> std::io::Result<()> {
let data = &buf[0..number_of_bytes];
println!("Data: {:?}", strip_header(data));
println!("Repl: {:?}", strip_header(mopp(speed, b"hi").as_slice()));
if receivers.iter().any(|x| x.name == client_addr) {
println!("Client already known.");
broadcast(&socket, &receivers, &client, data);
} else if strip_header(data) == strip_header(mopp(speed, b"hi").as_slice()) {
println!("Welcome!");
if receivers.len() < MAX_CLIENTS {
receivers.insert(client.clone());
// TODO: welcome
socket
.send_to(mopp(speed, b"hi").as_slice(), &client.name)
.unwrap();
} else {
// TODO: reject
println!("Unknown client - Ignoring");
socket
.send_to(mopp(speed, b":qrl").as_slice(), &client.name)
.unwrap();
}
} else {
println!("Unknown client - Ignoring");
}
let timestamp = Local::now();
for cl in &receivers {
if cl.time + Duration::seconds(CLIENT_TIMEOUT as i64) < timestamp {
socket
.send_to(mopp(speed, b":bye").as_slice(), &client.name)
.unwrap();
}
}
receivers.retain(|cl| cl.time + Duration::seconds(CLIENT_TIMEOUT as i64) >= timestamp);
}
}