Rough out sending a basic UDP notification.

This commit is contained in:
mattbk
2025-09-28 21:20:46 -05:00
parent 6da68c7fe7
commit 3aa395dc4e

View File

@@ -8,6 +8,7 @@ use std::io::{Read, Write};
use std::net::TcpStream; use std::net::TcpStream;
use std::thread::sleep; use std::thread::sleep;
use std::time::Duration; use std::time::Duration;
use std::net::UdpSocket;
/// Validate that the provided port string can be parsed into a u16 and is nonzero. /// Validate that the provided port string can be parsed into a u16 and is nonzero.
fn validate_port(port: &str) -> Result<u16, String> { fn validate_port(port: &str) -> Result<u16, String> {
@@ -47,6 +48,18 @@ struct Cli {
/// Only monitor UI frames /// Only monitor UI frames
#[arg(short = 'u', long, default_value_t = false)] #[arg(short = 'u', long, default_value_t = false)]
ui_only: bool, ui_only: bool,
// Enable UDP
#[arg(short = 'm', long, default_value_t = false)]
udp: bool,
// Send UDP to what IP address (e.g. 127.0.0.1)
#[arg(short = 'b', long)]
uip: std::net::IpAddr,
// Send UDP to what port (e.g. 8000)
#[arg(short = 'k', long, value_parser = validate_port)]
uport: u16,
} }
/// Convert a byte slice into a hex-dump string for debugging purposes. /// Convert a byte slice into a hex-dump string for debugging purposes.
@@ -384,6 +397,12 @@ fn handle_frame(frame: &AgwFrame, cli: &Cli, buffers: &mut BufferManager) {
return; return;
} }
// Send UDP
let uaddr = format!("{}:{}", cli.uip, cli.uport);
let socket = UdpSocket::bind("0.0.0.0:0");
let message = "Packet received";
socket.expect("REASON").send_to(message.as_bytes(), uaddr);
// In non-debug mode, print the session line and any additional lines. // In non-debug mode, print the session line and any additional lines.
if !cli.debug { if !cli.debug {
print_session_line(&timestamp, &source, &final_destination, &summary); print_session_line(&timestamp, &source, &final_destination, &summary);
@@ -403,7 +422,19 @@ fn handle_frame(frame: &AgwFrame, cli: &Cli, buffers: &mut BufferManager) {
fn main() -> Result<()> { fn main() -> Result<()> {
let cli = Cli::parse(); let cli = Cli::parse();
let addr = format!("{}:{}", cli.ip, cli.port); let addr = format!("{}:{}", cli.ip, cli.port);
//let uaddr = format!("{}:{}", cli.uip, cli.uport);
let reconnect_delay_ms = 5000; let reconnect_delay_ms = 5000;
/*
if cli.udp {
// Bind the client socket to any available address and port
let socket = UdpSocket::bind("0.0.0.0:0")?;
println!("UDP client started");
// Send a message to the server
let message = "UDP client connected";
socket.send_to(message.as_bytes(), uaddr)?;
//println!("Sent message to {}: {}", uaddr, message);
}
*/
loop { loop {
println!("Connecting to AGWPE server at {addr}"); println!("Connecting to AGWPE server at {addr}");