1 files changed, 147 insertions, 0 deletions
diff --git a/vendor/rustix/tests/net/unix.rs b/vendor/rustix/tests/net/unix.rs
new file mode 100644
index 000000000..21a6542f8
--- /dev/null
+++ b/vendor/rustix/tests/net/unix.rs
@@ -0,0 +1,147 @@
+//! Test a simple Unix-domain socket server and client.
+//!
+//! The client sends lists of integers and the server sends back sums.
+
+// This test uses `AF_UNIX` with `SOCK_SEQPACKET` which is unsupported on macOS.
+#![cfg(not(any(
+    target_os = "ios",
+    target_os = "macos",
+    target_os = "redox",
+    target_os = "wasi",
+)))]
+// This test uses `DecInt`.
+#![cfg(feature = "itoa")]
+#![cfg(feature = "fs")]
+
+use rustix::fs::{cwd, unlinkat, AtFlags};
+use rustix::io::{read, write};
+use rustix::net::{
+    accept, bind_unix, connect_unix, listen, socket, AddressFamily, Protocol, SocketAddrUnix,
+    SocketType,
+};
+use rustix::path::DecInt;
+use std::path::Path;
+use std::str::FromStr;
+use std::sync::{Arc, Condvar, Mutex};
+use std::thread;
+
+const BUFFER_SIZE: usize = 20;
+
+fn server(ready: Arc<(Mutex<bool>, Condvar)>, path: &Path) {
+    let connection_socket = socket(
+        AddressFamily::UNIX,
+        SocketType::SEQPACKET,
+        Protocol::default(),
+    )
+    .unwrap();
+
+    let name = SocketAddrUnix::new(path).unwrap();
+    bind_unix(&connection_socket, &name).unwrap();
+    listen(&connection_socket, 1).unwrap();
+
+    {
+        let (lock, cvar) = &*ready;
+        let mut started = lock.lock().unwrap();
+        *started = true;
+        cvar.notify_all();
+    }
+
+    let mut buffer = vec![0; BUFFER_SIZE];
+    'exit: loop {
+        let data_socket = accept(&connection_socket).unwrap();
+        let mut sum = 0;
+        loop {
+            let nread = read(&data_socket, &mut buffer).unwrap();
+
+            if &buffer[..nread] == b"exit" {
+                break 'exit;
+            }
+            if &buffer[..nread] == b"sum" {
+                break;
+            }
+
+            sum += i32::from_str(&String::from_utf8_lossy(&buffer[..nread])).unwrap();
+        }
+
+        write(&data_socket, DecInt::new(sum).as_bytes()).unwrap();
+    }
+
+    unlinkat(cwd(), path, AtFlags::empty()).unwrap();
+}
+
+fn client(ready: Arc<(Mutex<bool>, Condvar)>, path: &Path, runs: &[(&[&str], i32)]) {
+    {
+        let (lock, cvar) = &*ready;
+        let mut started = lock.lock().unwrap();
+        while !*started {
+            started = cvar.wait(started).unwrap();
+        }
+    }
+
+    let addr = SocketAddrUnix::new(path).unwrap();
+    let mut buffer = vec![0; BUFFER_SIZE];
+
+    for (args, sum) in runs {
+        let data_socket = socket(
+            AddressFamily::UNIX,
+            SocketType::SEQPACKET,
+            Protocol::default(),
+        )
+        .unwrap();
+        connect_unix(&data_socket, &addr).unwrap();
+
+        for arg in *args {
+            write(&data_socket, arg.as_bytes()).unwrap();
+        }
+        write(&data_socket, b"sum").unwrap();
+
+        let nread = read(&data_socket, &mut buffer).unwrap();
+        assert_eq!(
+            i32::from_str(&String::from_utf8_lossy(&buffer[..nread])).unwrap(),
+            *sum
+        );
+    }
+
+    let data_socket = socket(
+        AddressFamily::UNIX,
+        SocketType::SEQPACKET,
+        Protocol::default(),
+    )
+    .unwrap();
+    connect_unix(&data_socket, &addr).unwrap();
+    write(&data_socket, b"exit").unwrap();
+}
+
+#[test]
+fn test_unix() {
+    let ready = Arc::new((Mutex::new(false), Condvar::new()));
+    let ready_clone = Arc::clone(&ready);
+
+    let tmp = tempfile::tempdir().unwrap();
+    let path = tmp.path().join("soccer");
+    let send_path = path.to_owned();
+    let server = thread::Builder::new()
+        .name("server".to_string())
+        .spawn(move || {
+            server(ready, &send_path);
+        })
+        .unwrap();
+    let send_path = path.to_owned();
+    let client = thread::Builder::new()
+        .name("client".to_string())
+        .spawn(move || {
+            client(
+                ready_clone,
+                &send_path,
+                &[
+                    (&["1", "2"], 3),
+                    (&["4", "77", "103"], 184),
+                    (&["5", "78", "104"], 187),
+                    (&[], 0),
+                ],
+            );
+        })
+        .unwrap();
+    client.join().unwrap();
+    server.join().unwrap();
+}