most is implemented
This commit is contained in:
parent
95a65f981c
commit
6017a6ca47
1 changed files with 47 additions and 12 deletions
57
src/main.rs
57
src/main.rs
|
@ -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
|
||||
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);
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue