rust udp client

main.rs

use std::{
    io::{stdin, BufRead},
    net::UdpSocket,
    str::from_utf8,
};

fn main() -> std::io::Result<()> {
    let args: Vec<String> = std::env::args().collect();

    if args.len() < 2 {
        println!("Usage: {} hostname", args[0]);
        std::process::exit(1);
    }

    let hostname = &args[1];

    let socket = UdpSocket::bind("0.0.0.0:0")?;

    let peer = format!("{}:{}", hostname, 2000);
    println!("host: {}", peer);

    let stdin = stdin();

    for line in stdin.lock().lines() {
        let line = line.unwrap();

        println!("Line read from stdin {}", line);

        if &line == "BYE" {
            break;
        }

        socket
            .send_to(line.as_bytes(), peer.as_str())
            .expect("Error on send");

        let mut buf = [0; 65535];
        let (read, _) = socket.recv_from(&mut buf)?;

        let echo = from_utf8(&buf[..read]).unwrap();
        println!("Echoed: {}", echo);
    }

    Ok(())
}