From 87c50384f9e6cd66b3aa68b57326cbdcbc755b06 Mon Sep 17 00:00:00 2001
From: Jianfeng Jiang <jiangjianfeng.jjf@antgroup.com>
Date: Mon, 31 Jul 2023 19:24:00 +0800
Subject: [PATCH] Add syscall socketpair

---
 .../libs/jinux-std/src/syscall/socketpair.rs  | 45 +++++++++++++++++++
 1 file changed, 45 insertions(+)
 create mode 100644 services/libs/jinux-std/src/syscall/socketpair.rs

diff --git a/services/libs/jinux-std/src/syscall/socketpair.rs b/services/libs/jinux-std/src/syscall/socketpair.rs
new file mode 100644
index 000000000..2a483db09
--- /dev/null
+++ b/services/libs/jinux-std/src/syscall/socketpair.rs
@@ -0,0 +1,45 @@
+use crate::fs::file_table::FileDescripter;
+use crate::net::socket::unix::UnixStreamSocket;
+use crate::util::net::{Protocol, SaFamily, SockFlags, SockType, SOCK_TYPE_MASK};
+use crate::util::write_val_to_user;
+use crate::{log_syscall_entry, prelude::*};
+
+use super::SyscallReturn;
+use super::SYS_SOCKETPAIR;
+
+pub fn sys_socketpair(domain: i32, type_: i32, protocol: i32, sv: Vaddr) -> Result<SyscallReturn> {
+    log_syscall_entry!(SYS_SOCKETPAIR);
+    let domain = SaFamily::try_from(domain)?;
+    let sock_type = SockType::try_from(type_ & SOCK_TYPE_MASK)?;
+    let sock_flags = SockFlags::from_bits_truncate(type_ & !SOCK_TYPE_MASK);
+    let protocol = Protocol::try_from(protocol)?;
+
+    debug!(
+        "domain = {:?}, sock_type = {:?}, sock_flags = {:?}, protocol = {:?}",
+        domain, sock_type, sock_flags, protocol
+    );
+    // TODO: deal with all sock_flags and protocol
+    let nonblocking = sock_flags.contains(SockFlags::SOCK_NONBLOCK);
+    let (socket_a, socket_b) = match (domain, sock_type) {
+        (SaFamily::AF_UNIX, SockType::SOCK_STREAM) => UnixStreamSocket::new_pair(nonblocking)?,
+        _ => return_errno_with_message!(
+            Errno::EAFNOSUPPORT,
+            "cannot create socket pair for this family"
+        ),
+    };
+
+    let socket_fds = {
+        let current = current!();
+        let mut filetable = current.file_table().lock();
+        let fd_a = filetable.insert(socket_a);
+        let fd_b = filetable.insert(socket_b);
+        SocketFds(fd_a, fd_b)
+    };
+
+    write_val_to_user(sv, &socket_fds)?;
+    Ok(SyscallReturn::Return(0))
+}
+
+#[repr(C)]
+#[derive(Debug, Clone, Copy, Pod)]
+struct SocketFds(FileDescripter, FileDescripter);