tcpbin 0.1.2

Simple library/CLI providing TCP echo server
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66

use std::{net, thread, time};
use std::io::{self, Write};

use tcpbin::{DEFAULT_LOCAL_IP, IpEcho, Echo};

#[test]
fn should_detect_ip() {
    const PORT: u16 = 45001;
    const ADDR: net::SocketAddr = net::SocketAddr::new(net::IpAddr::V4(net::Ipv4Addr::LOCALHOST), PORT);

    let _ = thread::spawn(|| {
        let _ = IpEcho::new().with_port(PORT).run_blocking();
    });

    std::thread::sleep(time::Duration::from_secs(1));
    let mut socket = net::TcpStream::connect(ADDR).expect("to connect");
    std::thread::yield_now();

    let mut out = String::new();
    io::Read::read_to_string(&mut socket, &mut out).expect("to read ip");
    assert_eq!(out, "127.0.0.1");
}

#[test]
fn should_detect_ip_via_proxy_v1() {
    const PORT: u16 = 45002;
    const ADDR: net::SocketAddr = net::SocketAddr::new(net::IpAddr::V4(net::Ipv4Addr::LOCALHOST), PORT);

    let proxy = ha_proxy_protocol::v1::Proxy {
        src: net::SocketAddr::new(net::Ipv4Addr::new(255, 1, 2, 3).into(), 80),
        dst: net::SocketAddr::new(DEFAULT_LOCAL_IP, PORT),
    };
    let _ = thread::spawn(|| {
        let _ = IpEcho::new().with_port(PORT).run_blocking();
    });

    std::thread::sleep(time::Duration::from_secs(1));
    let mut socket = net::TcpStream::connect(ADDR).expect("to connect");
    socket.write_fmt(format_args!("{proxy}\r\n")).expect("write proxy");
    socket.flush().expect("flush proxy");
    std::thread::yield_now();

    let mut out = String::new();
    io::Read::read_to_string(&mut socket, &mut out).expect("to read ip");
    assert_eq!(out, "255.1.2.3");
}

#[test]
fn should_echo_data() {
    const PORT: u16 = 45003;
    const ADDR: net::SocketAddr = net::SocketAddr::new(net::IpAddr::V4(net::Ipv4Addr::LOCALHOST), PORT);

    let _ = thread::spawn(|| {
        let _ = Echo::new().with_port(PORT).run_blocking();
    });

    std::thread::sleep(time::Duration::from_secs(1));
    let mut socket = net::TcpStream::connect(ADDR).expect("to connect");
    std::thread::yield_now();

    socket.write_all(b"echo-test").expect("write all");

    let mut out = String::new();
    io::Read::read_to_string(&mut socket, &mut out).expect("to read ip");
    assert_eq!(out, "echo-test");
}