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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

// Copyright (C) 2020 - Will Glozer. All rights reserved.

pub use crate::socket::RawSocket;

pub use socket2::Domain;
pub use socket2::Type;
pub use socket2::Protocol;

pub mod control;
pub mod ffi;
pub mod option;
pub mod prelude;

mod socket;

#[cfg(feature = "async-tokio")]
pub mod tokio;

#[cfg(test)]
mod test {
    use std::io::{Error, ErrorKind, IoSlice, IoSliceMut, Result};
    use std::net::{IpAddr, SocketAddr};
    use std::thread::sleep;
    use std::time::Duration;
    use libc::{c_int, SOCK_DGRAM, SOCK_STREAM};
    use crate::{RawSocket, Domain, Type};
    use crate::option::{Level, Name};

    #[test]
    fn get_sockopt() -> Result<()> {
        let ipv4  = Domain::ipv4();

        let sock0 = RawSocket::new(ipv4, Type::dgram(),  None)?;
        let sock1 = RawSocket::new(ipv4, Type::stream(), None)?;

        assert_eq!(SOCK_DGRAM,  sock0.get_sockopt(Level::SOCKET, Name::SO_TYPE)?);
        assert_eq!(SOCK_STREAM, sock1.get_sockopt(Level::SOCKET, Name::SO_TYPE)?);

        Ok(())
    }

    #[test]
    fn set_sockopt() -> Result<()> {
        let level = Level::SOCKET;
        let name  = Name::SO_KEEPALIVE;

        let sock = RawSocket::new(Domain::ipv4(), Type::stream(), None)?;

        let mut get: c_int;
        get = sock.get_sockopt(level, name)?;
        assert!(get == 0);

        let set: c_int = 1;
        sock.set_sockopt(level, name, &set)?;

        get = sock.get_sockopt(level, name)?;
        assert!(get != 0);

        Ok(())
    }

    #[test]
    fn send_recv_msg() -> Result<()> {
        let addr = SocketAddr::new(IpAddr::from([127, 0, 0, 1]), 0);

        let send = RawSocket::new(Domain::ipv4(), Type::dgram(), None)?;
        let recv = RawSocket::new(Domain::ipv4(), Type::dgram(), None)?;

        send.bind(&addr)?;
        recv.bind(&addr)?;

        let sent = [0u8; 64];
        let addr = recv.local_addr()?;
        let n = send.send_msg(&addr, &[IoSlice::new(&sent)], &[])?;

        assert_eq!(n, sent.len());

        let mut data = [0u8; 64];

        let (n, from) = loop {
            let iovec = &[IoSliceMut::new(&mut data)];
            let delay = Duration::from_secs(1);

            let is_wb = |e: &Error| e.kind() == ErrorKind::WouldBlock;

            match recv.recv_msg(iovec, &mut []) {
                Ok((n, from))          => break (n, from),
                Err(ref e) if is_wb(e) => sleep(delay),
                Err(e)                 => return Err(e),
            }
        };

        assert_eq!(n, data.len());
        assert_eq!(&sent[..], &data[..]);
        assert_eq!(from, send.local_addr()?);

        Ok(())
    }
}