socket/
socket.rs

1// Copyright (c) 2020 Xu Shaohua <shaohua@biofan.org>. All rights reserved.
2// Use of this source is governed by Apache-2.0 License that can be found
3// in the LICENSE file.
4
5use nc::Errno;
6use std::mem::{size_of, transmute};
7use std::thread;
8
9const SERVER_PORT: u16 = 18080;
10
11#[must_use]
12#[inline]
13const fn htons(host: u16) -> u16 {
14    host.to_be()
15}
16
17fn main() -> Result<(), Errno> {
18    let listen_fd = unsafe { nc::socket(nc::AF_INET, nc::SOCK_STREAM, 0)? };
19    println!("listen fd: {listen_fd}");
20
21    let addr = nc::sockaddr_in_t {
22        sin_family: nc::AF_INET as nc::sa_family_t,
23        sin_port: htons(SERVER_PORT),
24        sin_addr: nc::in_addr_t {
25            s_addr: nc::INADDR_ANY as u32,
26        },
27        ..Default::default()
28    };
29    println!("addr: {addr:?}");
30
31    let ret = unsafe {
32        let addr_alias = transmute::<&nc::sockaddr_in_t, &nc::sockaddr_t>(&addr);
33        nc::bind(listen_fd, addr_alias, size_of::<nc::sockaddr_in_t>() as u32)
34    };
35    assert!(ret.is_ok());
36
37    // Start worker thread
38    thread::spawn(|| {
39        println!("worker thread started");
40        let socket_fd = unsafe { nc::socket(nc::AF_INET, nc::SOCK_STREAM, 0) };
41        assert!(socket_fd.is_ok());
42        if let Ok(socket_fd) = socket_fd {
43            let addr = nc::sockaddr_in_t {
44                sin_family: nc::AF_INET as nc::sa_family_t,
45                sin_port: htons(SERVER_PORT),
46                sin_addr: nc::in_addr_t {
47                    s_addr: nc::INADDR_ANY as u32,
48                },
49                ..Default::default()
50            };
51            unsafe {
52                let addr_alias = (&addr as *const nc::sockaddr_in_t).cast::<nc::sockaddr_t>();
53                let ret = nc::connect(socket_fd, addr_alias, size_of::<nc::sockaddr_in_t>() as u32);
54                assert_eq!(ret, Ok(()));
55            }
56        } else {
57            eprintln!("Failed to create socket");
58        }
59    });
60
61    unsafe {
62        nc::listen(listen_fd, nc::SOCK_STREAM)?;
63    }
64
65    let conn_fd = unsafe { nc::accept4(listen_fd, None, None, nc::SOCK_CLOEXEC)? };
66    println!("conn_fd: {conn_fd}");
67
68    unsafe {
69        nc::close(listen_fd)?;
70    }
71
72    Ok(())
73}
socket/socket.rs

socket/
socket.rs
