42 lines
1.4 KiB
Rust
42 lines
1.4 KiB
Rust
use std::net::SocketAddrV4;
|
|
|
|
use anyhow::Context;
|
|
use chrono::Timelike;
|
|
use roowho2_lib::proto::{Whod, WhodStatusUpdate};
|
|
|
|
const RWHOD_BROADCAST_PORT: u16 = 513;
|
|
|
|
#[tokio::main]
|
|
async fn main() -> anyhow::Result<()> {
|
|
let addr = SocketAddrV4::new(std::net::Ipv4Addr::UNSPECIFIED, RWHOD_BROADCAST_PORT);
|
|
let socket = tokio::net::UdpSocket::bind(addr).await?;
|
|
socket.set_broadcast(true)?;
|
|
|
|
let mut buf = [0u8; Whod::MAX_SIZE];
|
|
loop {
|
|
let (len, src) = socket.recv_from(&mut buf).await?;
|
|
if len < Whod::HEADER_SIZE {
|
|
eprintln!(
|
|
"Received too short packet from {src}: {len} bytes (needs to be at least {} bytes)",
|
|
Whod::HEADER_SIZE
|
|
);
|
|
continue;
|
|
}
|
|
let result: WhodStatusUpdate = Whod::from_bytes(&buf[..len])
|
|
.context("Failed to parse whod packet")?
|
|
.try_into()
|
|
.map(|mut status_update: WhodStatusUpdate| {
|
|
let timestamp = chrono::Utc::now()
|
|
.with_nanosecond(0)
|
|
.unwrap_or(chrono::Utc::now());
|
|
status_update.recvtime = Some(timestamp);
|
|
status_update
|
|
})
|
|
.map_err(|e| anyhow::anyhow!("Invalid whod packet: {}", e))?;
|
|
|
|
println!("Received whod packet from {src}:\n{result:#?}");
|
|
|
|
buf = [0u8; Whod::MAX_SIZE];
|
|
}
|
|
}
|