16046f775S裕依 use core::{cmp::min, ffi::CStr}; 2cde5492fSlogin 3cde5492fSlogin use alloc::{boxed::Box, sync::Arc}; 4cde5492fSlogin use num_traits::{FromPrimitive, ToPrimitive}; 5cde5492fSlogin use smoltcp::wire; 691e9d4abSLoGin use system_error::SystemError; 7cde5492fSlogin 8cde5492fSlogin use crate::{ 9cde5492fSlogin filesystem::vfs::{ 10cde5492fSlogin file::{File, FileMode}, 11cde5492fSlogin syscall::{IoVec, IoVecs}, 126046f775S裕依 FileType, 13cde5492fSlogin }, 14ab5c8ca4Slogin libs::spinlock::SpinLockGuard, 154fda81ceSLoGin mm::{verify_area, VirtAddr}, 16cde5492fSlogin net::socket::{AddressFamily, SOL_SOCKET}, 171496ba7bSLoGin process::ProcessManager, 1891e9d4abSLoGin syscall::Syscall, 19cde5492fSlogin }; 20cde5492fSlogin 21cde5492fSlogin use super::{ 224ad52e57S裕依2439 socket::{new_socket, PosixSocketType, Socket, SocketHandleItem, SocketInode, HANDLE_MAP}, 234ad52e57S裕依2439 Endpoint, Protocol, ShutdownType, 24cde5492fSlogin }; 25cde5492fSlogin 26c47fe904SLoGin /// Flags for socket, socketpair, accept4 27c47fe904SLoGin const SOCK_CLOEXEC: FileMode = FileMode::O_CLOEXEC; 28c47fe904SLoGin const SOCK_NONBLOCK: FileMode = FileMode::O_NONBLOCK; 29c47fe904SLoGin 30ab5c8ca4Slogin impl Syscall { 31cde5492fSlogin /// @brief sys_socket系统调用的实际执行函数 32cde5492fSlogin /// 33cde5492fSlogin /// @param address_family 地址族 34cde5492fSlogin /// @param socket_type socket类型 35cde5492fSlogin /// @param protocol 传输协议 36ab5c8ca4Slogin pub fn socket( 37cde5492fSlogin address_family: usize, 38cde5492fSlogin socket_type: usize, 39cde5492fSlogin protocol: usize, 40ab5c8ca4Slogin ) -> Result<usize, SystemError> { 41cde5492fSlogin let address_family = AddressFamily::try_from(address_family as u16)?; 42cde5492fSlogin let socket_type = PosixSocketType::try_from((socket_type & 0xf) as u8)?; 434ad52e57S裕依2439 let protocol = Protocol::from(protocol as u8); 444ad52e57S裕依2439 454ad52e57S裕依2439 let socket = new_socket(address_family, socket_type, protocol)?; 464ad52e57S裕依2439 476046f775S裕依 if address_family != AddressFamily::Unix { 486046f775S裕依 let handle_item = SocketHandleItem::new(); 4940609970SGnoCiYeH HANDLE_MAP 5040609970SGnoCiYeH .write_irqsave() 5140609970SGnoCiYeH .insert(socket.socket_handle(), handle_item); 526046f775S裕依 } 534ad52e57S裕依2439 54cde5492fSlogin let socketinode: Arc<SocketInode> = SocketInode::new(socket); 55cde5492fSlogin let f = File::new(socketinode, FileMode::O_RDWR)?; 56cde5492fSlogin // 把socket添加到当前进程的文件描述符表中 571496ba7bSLoGin let binding = ProcessManager::current_pcb().fd_table(); 581496ba7bSLoGin let mut fd_table_guard = binding.write(); 591496ba7bSLoGin let fd = fd_table_guard.alloc_fd(f, None).map(|x| x as usize); 601496ba7bSLoGin drop(fd_table_guard); 61cde5492fSlogin return fd; 62cde5492fSlogin } 63cde5492fSlogin 644ad52e57S裕依2439 /// # sys_socketpair系统调用的实际执行函数 654ad52e57S裕依2439 /// 664ad52e57S裕依2439 /// ## 参数 674ad52e57S裕依2439 /// - `address_family`: 地址族 684ad52e57S裕依2439 /// - `socket_type`: socket类型 694ad52e57S裕依2439 /// - `protocol`: 传输协议 704ad52e57S裕依2439 /// - `fds`: 用于返回文件描述符的数组 714ad52e57S裕依2439 pub fn socketpair( 724ad52e57S裕依2439 address_family: usize, 734ad52e57S裕依2439 socket_type: usize, 744ad52e57S裕依2439 protocol: usize, 754ad52e57S裕依2439 fds: &mut [i32], 764ad52e57S裕依2439 ) -> Result<usize, SystemError> { 774ad52e57S裕依2439 let address_family = AddressFamily::try_from(address_family as u16)?; 784ad52e57S裕依2439 let socket_type = PosixSocketType::try_from((socket_type & 0xf) as u8)?; 794ad52e57S裕依2439 let protocol = Protocol::from(protocol as u8); 804ad52e57S裕依2439 814ad52e57S裕依2439 let binding = ProcessManager::current_pcb().fd_table(); 824ad52e57S裕依2439 let mut fd_table_guard = binding.write(); 834ad52e57S裕依2439 846046f775S裕依 // 创建一对socket 856046f775S裕依 let inode0 = SocketInode::new(new_socket(address_family, socket_type, protocol)?); 866046f775S裕依 let inode1 = SocketInode::new(new_socket(address_family, socket_type, protocol)?); 874ad52e57S裕依2439 886046f775S裕依 // 进行pair 896046f775S裕依 unsafe { 906046f775S裕依 inode0 916046f775S裕依 .inner_no_preempt() 926046f775S裕依 .connect(Endpoint::Inode(Some(inode1.clone())))?; 936046f775S裕依 inode1 946046f775S裕依 .inner_no_preempt() 956046f775S裕依 .connect(Endpoint::Inode(Some(inode0.clone())))?; 966046f775S裕依 } 976046f775S裕依 986046f775S裕依 fds[0] = fd_table_guard.alloc_fd(File::new(inode0, FileMode::O_RDWR)?, None)?; 996046f775S裕依 fds[1] = fd_table_guard.alloc_fd(File::new(inode1, FileMode::O_RDWR)?, None)?; 1004ad52e57S裕依2439 1014ad52e57S裕依2439 drop(fd_table_guard); 1024ad52e57S裕依2439 Ok(0) 1034ad52e57S裕依2439 } 1044ad52e57S裕依2439 105cde5492fSlogin /// @brief sys_setsockopt系统调用的实际执行函数 106cde5492fSlogin /// 107cde5492fSlogin /// @param fd 文件描述符 108cde5492fSlogin /// @param level 选项级别 109cde5492fSlogin /// @param optname 选项名称 110cde5492fSlogin /// @param optval 选项值 111cde5492fSlogin /// @param optlen optval缓冲区长度 112ab5c8ca4Slogin pub fn setsockopt( 113cde5492fSlogin fd: usize, 114cde5492fSlogin level: usize, 115cde5492fSlogin optname: usize, 116ab5c8ca4Slogin optval: &[u8], 117ab5c8ca4Slogin ) -> Result<usize, SystemError> { 1181496ba7bSLoGin let socket_inode: Arc<SocketInode> = ProcessManager::current_pcb() 119cde5492fSlogin .get_socket(fd as i32) 120cde5492fSlogin .ok_or(SystemError::EBADF)?; 121cde5492fSlogin // 获取内层的socket(真正的数据) 122ab5c8ca4Slogin let socket: SpinLockGuard<Box<dyn Socket>> = socket_inode.inner(); 123ab5c8ca4Slogin return socket.setsockopt(level, optname, optval).map(|_| 0); 124cde5492fSlogin } 125cde5492fSlogin 126cde5492fSlogin /// @brief sys_getsockopt系统调用的实际执行函数 127cde5492fSlogin /// 128cde5492fSlogin /// 参考:https://man7.org/linux/man-pages/man2/setsockopt.2.html 129cde5492fSlogin /// 130cde5492fSlogin /// @param fd 文件描述符 131cde5492fSlogin /// @param level 选项级别 132cde5492fSlogin /// @param optname 选项名称 133cde5492fSlogin /// @param optval 返回的选项值 134cde5492fSlogin /// @param optlen 返回的optval缓冲区长度 135ab5c8ca4Slogin pub fn getsockopt( 136cde5492fSlogin fd: usize, 137cde5492fSlogin level: usize, 138cde5492fSlogin optname: usize, 139cde5492fSlogin optval: *mut u8, 140cde5492fSlogin optlen: *mut u32, 141ab5c8ca4Slogin ) -> Result<usize, SystemError> { 142cde5492fSlogin // 获取socket 143cde5492fSlogin let optval = optval as *mut u32; 1441496ba7bSLoGin let binding: Arc<SocketInode> = ProcessManager::current_pcb() 145cde5492fSlogin .get_socket(fd as i32) 146cde5492fSlogin .ok_or(SystemError::EBADF)?; 147cde5492fSlogin let socket = binding.inner(); 148cde5492fSlogin 149cde5492fSlogin if level as u8 == SOL_SOCKET { 150ab5c8ca4Slogin let optname = PosixSocketOption::try_from(optname as i32) 151ab5c8ca4Slogin .map_err(|_| SystemError::ENOPROTOOPT)?; 152cde5492fSlogin match optname { 153cde5492fSlogin PosixSocketOption::SO_SNDBUF => { 154cde5492fSlogin // 返回发送缓冲区大小 155cde5492fSlogin unsafe { 1566046f775S裕依 *optval = socket.metadata().tx_buf_size as u32; 157cde5492fSlogin *optlen = core::mem::size_of::<u32>() as u32; 158cde5492fSlogin } 159cde5492fSlogin return Ok(0); 160cde5492fSlogin } 161cde5492fSlogin PosixSocketOption::SO_RCVBUF => { 162cde5492fSlogin // 返回默认的接收缓冲区大小 163cde5492fSlogin unsafe { 1646046f775S裕依 *optval = socket.metadata().rx_buf_size as u32; 165cde5492fSlogin *optlen = core::mem::size_of::<u32>() as u32; 166cde5492fSlogin } 167cde5492fSlogin return Ok(0); 168cde5492fSlogin } 169cde5492fSlogin _ => { 170cde5492fSlogin return Err(SystemError::ENOPROTOOPT); 171cde5492fSlogin } 172cde5492fSlogin } 173cde5492fSlogin } 174cde5492fSlogin drop(socket); 175cde5492fSlogin 176cde5492fSlogin // To manipulate options at any other level the 177cde5492fSlogin // protocol number of the appropriate protocol controlling the 178cde5492fSlogin // option is supplied. For example, to indicate that an option is 179cde5492fSlogin // to be interpreted by the TCP protocol, level should be set to the 180cde5492fSlogin // protocol number of TCP. 181cde5492fSlogin 182cde5492fSlogin let posix_protocol = 183cde5492fSlogin PosixIpProtocol::try_from(level as u16).map_err(|_| SystemError::ENOPROTOOPT)?; 184cde5492fSlogin if posix_protocol == PosixIpProtocol::TCP { 185cde5492fSlogin let optname = PosixTcpSocketOptions::try_from(optname as i32) 186cde5492fSlogin .map_err(|_| SystemError::ENOPROTOOPT)?; 187cde5492fSlogin match optname { 188cde5492fSlogin PosixTcpSocketOptions::Congestion => return Ok(0), 189cde5492fSlogin _ => { 190cde5492fSlogin return Err(SystemError::ENOPROTOOPT); 191cde5492fSlogin } 192cde5492fSlogin } 193cde5492fSlogin } 194cde5492fSlogin return Err(SystemError::ENOPROTOOPT); 195cde5492fSlogin } 196cde5492fSlogin 197cde5492fSlogin /// @brief sys_connect系统调用的实际执行函数 198cde5492fSlogin /// 199cde5492fSlogin /// @param fd 文件描述符 200cde5492fSlogin /// @param addr SockAddr 201cde5492fSlogin /// @param addrlen 地址长度 202cde5492fSlogin /// 203cde5492fSlogin /// @return 成功返回0,失败返回错误码 204ab5c8ca4Slogin pub fn connect(fd: usize, addr: *const SockAddr, addrlen: usize) -> Result<usize, SystemError> { 205cde5492fSlogin let endpoint: Endpoint = SockAddr::to_endpoint(addr, addrlen)?; 2061496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 207cde5492fSlogin .get_socket(fd as i32) 208cde5492fSlogin .ok_or(SystemError::EBADF)?; 2092f6f547aSGnoCiYeH let mut socket = unsafe { socket.inner_no_preempt() }; 210cde5492fSlogin socket.connect(endpoint)?; 2116046f775S裕依 Ok(0) 212cde5492fSlogin } 213cde5492fSlogin 214cde5492fSlogin /// @brief sys_bind系统调用的实际执行函数 215cde5492fSlogin /// 216cde5492fSlogin /// @param fd 文件描述符 217cde5492fSlogin /// @param addr SockAddr 218cde5492fSlogin /// @param addrlen 地址长度 219cde5492fSlogin /// 220cde5492fSlogin /// @return 成功返回0,失败返回错误码 221ab5c8ca4Slogin pub fn bind(fd: usize, addr: *const SockAddr, addrlen: usize) -> Result<usize, SystemError> { 222cde5492fSlogin let endpoint: Endpoint = SockAddr::to_endpoint(addr, addrlen)?; 2231496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 224cde5492fSlogin .get_socket(fd as i32) 225cde5492fSlogin .ok_or(SystemError::EBADF)?; 2262f6f547aSGnoCiYeH let mut socket = unsafe { socket.inner_no_preempt() }; 227cde5492fSlogin socket.bind(endpoint)?; 2286046f775S裕依 Ok(0) 229cde5492fSlogin } 230cde5492fSlogin 231cde5492fSlogin /// @brief sys_sendto系统调用的实际执行函数 232cde5492fSlogin /// 233cde5492fSlogin /// @param fd 文件描述符 234cde5492fSlogin /// @param buf 发送缓冲区 235cde5492fSlogin /// @param flags 标志 236cde5492fSlogin /// @param addr SockAddr 237cde5492fSlogin /// @param addrlen 地址长度 238cde5492fSlogin /// 239cde5492fSlogin /// @return 成功返回发送的字节数,失败返回错误码 240ab5c8ca4Slogin pub fn sendto( 241cde5492fSlogin fd: usize, 242ab5c8ca4Slogin buf: &[u8], 243ab5c8ca4Slogin _flags: u32, 244cde5492fSlogin addr: *const SockAddr, 245cde5492fSlogin addrlen: usize, 246ab5c8ca4Slogin ) -> Result<usize, SystemError> { 247cde5492fSlogin let endpoint = if addr.is_null() { 248cde5492fSlogin None 249cde5492fSlogin } else { 250cde5492fSlogin Some(SockAddr::to_endpoint(addr, addrlen)?) 251cde5492fSlogin }; 252cde5492fSlogin 2531496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 254cde5492fSlogin .get_socket(fd as i32) 255cde5492fSlogin .ok_or(SystemError::EBADF)?; 2562f6f547aSGnoCiYeH let socket = unsafe { socket.inner_no_preempt() }; 257ab5c8ca4Slogin return socket.write(buf, endpoint); 258cde5492fSlogin } 259cde5492fSlogin 260cde5492fSlogin /// @brief sys_recvfrom系统调用的实际执行函数 261cde5492fSlogin /// 262cde5492fSlogin /// @param fd 文件描述符 263cde5492fSlogin /// @param buf 接收缓冲区 264cde5492fSlogin /// @param flags 标志 265cde5492fSlogin /// @param addr SockAddr 266cde5492fSlogin /// @param addrlen 地址长度 267cde5492fSlogin /// 268cde5492fSlogin /// @return 成功返回接收的字节数,失败返回错误码 269ab5c8ca4Slogin pub fn recvfrom( 270cde5492fSlogin fd: usize, 271ab5c8ca4Slogin buf: &mut [u8], 272ab5c8ca4Slogin _flags: u32, 273cde5492fSlogin addr: *mut SockAddr, 274cde5492fSlogin addrlen: *mut u32, 275ab5c8ca4Slogin ) -> Result<usize, SystemError> { 2761496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 277cde5492fSlogin .get_socket(fd as i32) 278cde5492fSlogin .ok_or(SystemError::EBADF)?; 2796046f775S裕依 let socket = unsafe { socket.inner_no_preempt() }; 280cde5492fSlogin 281cde5492fSlogin let (n, endpoint) = socket.read(buf); 282cde5492fSlogin drop(socket); 283cde5492fSlogin 284cde5492fSlogin let n: usize = n?; 285cde5492fSlogin 286cde5492fSlogin // 如果有地址信息,将地址信息写入用户空间 287cde5492fSlogin if !addr.is_null() { 288cde5492fSlogin let sockaddr_in = SockAddr::from(endpoint); 289cde5492fSlogin unsafe { 290cde5492fSlogin sockaddr_in.write_to_user(addr, addrlen)?; 291cde5492fSlogin } 292cde5492fSlogin } 293ab5c8ca4Slogin return Ok(n); 294cde5492fSlogin } 295cde5492fSlogin 296cde5492fSlogin /// @brief sys_recvmsg系统调用的实际执行函数 297cde5492fSlogin /// 298cde5492fSlogin /// @param fd 文件描述符 299cde5492fSlogin /// @param msg MsgHdr 300ab5c8ca4Slogin /// @param flags 标志,暂时未使用 301cde5492fSlogin /// 302cde5492fSlogin /// @return 成功返回接收的字节数,失败返回错误码 303ab5c8ca4Slogin pub fn recvmsg(fd: usize, msg: &mut MsgHdr, _flags: u32) -> Result<usize, SystemError> { 304cde5492fSlogin // 检查每个缓冲区地址是否合法,生成iovecs 305cde5492fSlogin let mut iovs = unsafe { IoVecs::from_user(msg.msg_iov, msg.msg_iovlen, true)? }; 306cde5492fSlogin 3071496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 308cde5492fSlogin .get_socket(fd as i32) 309cde5492fSlogin .ok_or(SystemError::EBADF)?; 3106046f775S裕依 let socket = unsafe { socket.inner_no_preempt() }; 311cde5492fSlogin 312cde5492fSlogin let mut buf = iovs.new_buf(true); 313cde5492fSlogin // 从socket中读取数据 314cde5492fSlogin let (n, endpoint) = socket.read(&mut buf); 315cde5492fSlogin drop(socket); 316cde5492fSlogin 317cde5492fSlogin let n: usize = n?; 318cde5492fSlogin 319cde5492fSlogin // 将数据写入用户空间的iovecs 320cde5492fSlogin iovs.scatter(&buf[..n]); 321cde5492fSlogin 322cde5492fSlogin let sockaddr_in = SockAddr::from(endpoint); 323cde5492fSlogin unsafe { 324cde5492fSlogin sockaddr_in.write_to_user(msg.msg_name, &mut msg.msg_namelen)?; 325cde5492fSlogin } 326ab5c8ca4Slogin return Ok(n); 327cde5492fSlogin } 328cde5492fSlogin 329cde5492fSlogin /// @brief sys_listen系统调用的实际执行函数 330cde5492fSlogin /// 331cde5492fSlogin /// @param fd 文件描述符 332ab5c8ca4Slogin /// @param backlog 队列最大连接数 333cde5492fSlogin /// 334cde5492fSlogin /// @return 成功返回0,失败返回错误码 335ab5c8ca4Slogin pub fn listen(fd: usize, backlog: usize) -> Result<usize, SystemError> { 3361496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 337cde5492fSlogin .get_socket(fd as i32) 338cde5492fSlogin .ok_or(SystemError::EBADF)?; 3392f6f547aSGnoCiYeH let mut socket = unsafe { socket.inner_no_preempt() }; 340cde5492fSlogin socket.listen(backlog)?; 341cde5492fSlogin return Ok(0); 342cde5492fSlogin } 343cde5492fSlogin 344cde5492fSlogin /// @brief sys_shutdown系统调用的实际执行函数 345cde5492fSlogin /// 346cde5492fSlogin /// @param fd 文件描述符 347cde5492fSlogin /// @param how 关闭方式 348cde5492fSlogin /// 349cde5492fSlogin /// @return 成功返回0,失败返回错误码 350ab5c8ca4Slogin pub fn shutdown(fd: usize, how: usize) -> Result<usize, SystemError> { 3511496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 352cde5492fSlogin .get_socket(fd as i32) 353cde5492fSlogin .ok_or(SystemError::EBADF)?; 35440609970SGnoCiYeH let mut socket = unsafe { socket.inner_no_preempt() }; 35540609970SGnoCiYeH socket.shutdown(ShutdownType::from_bits_truncate(how as u8))?; 356cde5492fSlogin return Ok(0); 357cde5492fSlogin } 358cde5492fSlogin 359cde5492fSlogin /// @brief sys_accept系统调用的实际执行函数 360cde5492fSlogin /// 361cde5492fSlogin /// @param fd 文件描述符 362cde5492fSlogin /// @param addr SockAddr 363cde5492fSlogin /// @param addrlen 地址长度 364cde5492fSlogin /// 365cde5492fSlogin /// @return 成功返回新的文件描述符,失败返回错误码 366ab5c8ca4Slogin pub fn accept(fd: usize, addr: *mut SockAddr, addrlen: *mut u32) -> Result<usize, SystemError> { 367c47fe904SLoGin return Self::do_accept(fd, addr, addrlen, 0); 368c47fe904SLoGin } 369c47fe904SLoGin 370c47fe904SLoGin /// sys_accept4 - accept a connection on a socket 371c47fe904SLoGin /// 372c47fe904SLoGin /// 373c47fe904SLoGin /// If flags is 0, then accept4() is the same as accept(). The 374c47fe904SLoGin /// following values can be bitwise ORed in flags to obtain different 375c47fe904SLoGin /// behavior: 376c47fe904SLoGin /// 377c47fe904SLoGin /// - SOCK_NONBLOCK 378c47fe904SLoGin /// Set the O_NONBLOCK file status flag on the open file 379c47fe904SLoGin /// description (see open(2)) referred to by the new file 380c47fe904SLoGin /// descriptor. Using this flag saves extra calls to fcntl(2) 381c47fe904SLoGin /// to achieve the same result. 382c47fe904SLoGin /// 383c47fe904SLoGin /// - SOCK_CLOEXEC 384c47fe904SLoGin /// Set the close-on-exec (FD_CLOEXEC) flag on the new file 385c47fe904SLoGin /// descriptor. See the description of the O_CLOEXEC flag in 386c47fe904SLoGin /// open(2) for reasons why this may be useful. 387c47fe904SLoGin pub fn accept4( 388c47fe904SLoGin fd: usize, 389c47fe904SLoGin addr: *mut SockAddr, 390c47fe904SLoGin addrlen: *mut u32, 391c47fe904SLoGin mut flags: u32, 392c47fe904SLoGin ) -> Result<usize, SystemError> { 393c47fe904SLoGin // 如果flags不合法,返回错误 394c47fe904SLoGin if (flags & (!(SOCK_CLOEXEC | SOCK_NONBLOCK)).bits()) != 0 { 395c47fe904SLoGin return Err(SystemError::EINVAL); 396c47fe904SLoGin } 397c47fe904SLoGin 398c47fe904SLoGin if SOCK_NONBLOCK != FileMode::O_NONBLOCK && ((flags & SOCK_NONBLOCK.bits()) != 0) { 3994ad52e57S裕依2439 flags = (flags & !FileMode::O_NONBLOCK.bits()) | FileMode::O_NONBLOCK.bits(); 400c47fe904SLoGin } 401c47fe904SLoGin 402c47fe904SLoGin return Self::do_accept(fd, addr, addrlen, flags); 403c47fe904SLoGin } 404c47fe904SLoGin 405c47fe904SLoGin fn do_accept( 406c47fe904SLoGin fd: usize, 407c47fe904SLoGin addr: *mut SockAddr, 408c47fe904SLoGin addrlen: *mut u32, 409c47fe904SLoGin flags: u32, 410c47fe904SLoGin ) -> Result<usize, SystemError> { 4111496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 412cde5492fSlogin .get_socket(fd as i32) 413cde5492fSlogin .ok_or(SystemError::EBADF)?; 414cde5492fSlogin // kdebug!("accept: socket={:?}", socket); 4152f6f547aSGnoCiYeH let mut socket = unsafe { socket.inner_no_preempt() }; 416cde5492fSlogin // 从socket中接收连接 417cde5492fSlogin let (new_socket, remote_endpoint) = socket.accept()?; 418cde5492fSlogin drop(socket); 419cde5492fSlogin 420cde5492fSlogin // kdebug!("accept: new_socket={:?}", new_socket); 421cde5492fSlogin // Insert the new socket into the file descriptor vector 422cde5492fSlogin let new_socket: Arc<SocketInode> = SocketInode::new(new_socket); 423c47fe904SLoGin 424c47fe904SLoGin let mut file_mode = FileMode::O_RDWR; 4254ad52e57S裕依2439 if flags & SOCK_NONBLOCK.bits() != 0 { 426c47fe904SLoGin file_mode |= FileMode::O_NONBLOCK; 427c47fe904SLoGin } 4284ad52e57S裕依2439 if flags & SOCK_CLOEXEC.bits() != 0 { 429c47fe904SLoGin file_mode |= FileMode::O_CLOEXEC; 430c47fe904SLoGin } 431c47fe904SLoGin 4321496ba7bSLoGin let new_fd = ProcessManager::current_pcb() 4331496ba7bSLoGin .fd_table() 4341496ba7bSLoGin .write() 435c47fe904SLoGin .alloc_fd(File::new(new_socket, file_mode)?, None)?; 436cde5492fSlogin // kdebug!("accept: new_fd={}", new_fd); 437cde5492fSlogin if !addr.is_null() { 438cde5492fSlogin // kdebug!("accept: write remote_endpoint to user"); 439cde5492fSlogin // 将对端地址写入用户空间 440cde5492fSlogin let sockaddr_in = SockAddr::from(remote_endpoint); 441cde5492fSlogin unsafe { 442cde5492fSlogin sockaddr_in.write_to_user(addr, addrlen)?; 443cde5492fSlogin } 444cde5492fSlogin } 445ab5c8ca4Slogin return Ok(new_fd as usize); 446cde5492fSlogin } 447cde5492fSlogin 448cde5492fSlogin /// @brief sys_getsockname系统调用的实际执行函数 449cde5492fSlogin /// 450cde5492fSlogin /// Returns the current address to which the socket 451cde5492fSlogin /// sockfd is bound, in the buffer pointed to by addr. 452cde5492fSlogin /// 453cde5492fSlogin /// @param fd 文件描述符 454cde5492fSlogin /// @param addr SockAddr 455cde5492fSlogin /// @param addrlen 地址长度 456cde5492fSlogin /// 457cde5492fSlogin /// @return 成功返回0,失败返回错误码 458ab5c8ca4Slogin pub fn getsockname( 459cde5492fSlogin fd: usize, 460cde5492fSlogin addr: *mut SockAddr, 461cde5492fSlogin addrlen: *mut u32, 462ab5c8ca4Slogin ) -> Result<usize, SystemError> { 463cde5492fSlogin if addr.is_null() { 464cde5492fSlogin return Err(SystemError::EINVAL); 465cde5492fSlogin } 4661496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 467cde5492fSlogin .get_socket(fd as i32) 468cde5492fSlogin .ok_or(SystemError::EBADF)?; 469cde5492fSlogin let socket = socket.inner(); 470cde5492fSlogin let endpoint: Endpoint = socket.endpoint().ok_or(SystemError::EINVAL)?; 471cde5492fSlogin drop(socket); 472cde5492fSlogin 473cde5492fSlogin let sockaddr_in = SockAddr::from(endpoint); 474cde5492fSlogin unsafe { 475cde5492fSlogin sockaddr_in.write_to_user(addr, addrlen)?; 476cde5492fSlogin } 477cde5492fSlogin return Ok(0); 478cde5492fSlogin } 479cde5492fSlogin 480cde5492fSlogin /// @brief sys_getpeername系统调用的实际执行函数 481cde5492fSlogin /// 482cde5492fSlogin /// @param fd 文件描述符 483cde5492fSlogin /// @param addr SockAddr 484cde5492fSlogin /// @param addrlen 地址长度 485cde5492fSlogin /// 486cde5492fSlogin /// @return 成功返回0,失败返回错误码 487ab5c8ca4Slogin pub fn getpeername( 488cde5492fSlogin fd: usize, 489cde5492fSlogin addr: *mut SockAddr, 490cde5492fSlogin addrlen: *mut u32, 491ab5c8ca4Slogin ) -> Result<usize, SystemError> { 492cde5492fSlogin if addr.is_null() { 493cde5492fSlogin return Err(SystemError::EINVAL); 494cde5492fSlogin } 495cde5492fSlogin 4961496ba7bSLoGin let socket: Arc<SocketInode> = ProcessManager::current_pcb() 497cde5492fSlogin .get_socket(fd as i32) 498cde5492fSlogin .ok_or(SystemError::EBADF)?; 499cde5492fSlogin let socket = socket.inner(); 500cde5492fSlogin let endpoint: Endpoint = socket.peer_endpoint().ok_or(SystemError::EINVAL)?; 501cde5492fSlogin drop(socket); 502cde5492fSlogin 503cde5492fSlogin let sockaddr_in = SockAddr::from(endpoint); 504cde5492fSlogin unsafe { 505cde5492fSlogin sockaddr_in.write_to_user(addr, addrlen)?; 506cde5492fSlogin } 507cde5492fSlogin return Ok(0); 508cde5492fSlogin } 509ab5c8ca4Slogin } 510cde5492fSlogin 511cde5492fSlogin // 参考资料: https://pubs.opengroup.org/onlinepubs/9699919799/basedefs/netinet_in.h.html#tag_13_32 512cde5492fSlogin #[repr(C)] 513cde5492fSlogin #[derive(Debug, Clone, Copy)] 514cde5492fSlogin pub struct SockAddrIn { 515cde5492fSlogin pub sin_family: u16, 516cde5492fSlogin pub sin_port: u16, 517cde5492fSlogin pub sin_addr: u32, 518cde5492fSlogin pub sin_zero: [u8; 8], 519cde5492fSlogin } 520cde5492fSlogin 521cde5492fSlogin #[repr(C)] 522cde5492fSlogin #[derive(Debug, Clone, Copy)] 523cde5492fSlogin pub struct SockAddrUn { 524cde5492fSlogin pub sun_family: u16, 525cde5492fSlogin pub sun_path: [u8; 108], 526cde5492fSlogin } 527cde5492fSlogin 528cde5492fSlogin #[repr(C)] 529cde5492fSlogin #[derive(Debug, Clone, Copy)] 530cde5492fSlogin pub struct SockAddrLl { 531cde5492fSlogin pub sll_family: u16, 532cde5492fSlogin pub sll_protocol: u16, 533cde5492fSlogin pub sll_ifindex: u32, 534cde5492fSlogin pub sll_hatype: u16, 535cde5492fSlogin pub sll_pkttype: u8, 536cde5492fSlogin pub sll_halen: u8, 537cde5492fSlogin pub sll_addr: [u8; 8], 538cde5492fSlogin } 539cde5492fSlogin 540cde5492fSlogin #[repr(C)] 541cde5492fSlogin #[derive(Debug, Clone, Copy)] 542cde5492fSlogin pub struct SockAddrNl { 543cde5492fSlogin nl_family: u16, 544cde5492fSlogin nl_pad: u16, 545cde5492fSlogin nl_pid: u32, 546cde5492fSlogin nl_groups: u32, 547cde5492fSlogin } 548cde5492fSlogin 549cde5492fSlogin #[repr(C)] 550cde5492fSlogin #[derive(Debug, Clone, Copy)] 551cde5492fSlogin pub struct SockAddrPlaceholder { 552cde5492fSlogin pub family: u16, 553cde5492fSlogin pub data: [u8; 14], 554cde5492fSlogin } 555cde5492fSlogin 556cde5492fSlogin #[repr(C)] 557cde5492fSlogin #[derive(Clone, Copy)] 558cde5492fSlogin pub union SockAddr { 559cde5492fSlogin pub family: u16, 560cde5492fSlogin pub addr_in: SockAddrIn, 561cde5492fSlogin pub addr_un: SockAddrUn, 562cde5492fSlogin pub addr_ll: SockAddrLl, 563cde5492fSlogin pub addr_nl: SockAddrNl, 564cde5492fSlogin pub addr_ph: SockAddrPlaceholder, 565cde5492fSlogin } 566cde5492fSlogin 567cde5492fSlogin impl SockAddr { 568cde5492fSlogin /// @brief 把用户传入的SockAddr转换为Endpoint结构体 569cde5492fSlogin pub fn to_endpoint(addr: *const SockAddr, len: usize) -> Result<Endpoint, SystemError> { 570cde5492fSlogin verify_area( 5714fda81ceSLoGin VirtAddr::new(addr as usize), 5724fda81ceSLoGin core::mem::size_of::<SockAddr>(), 573cde5492fSlogin ) 5744fda81ceSLoGin .map_err(|_| SystemError::EFAULT)?; 575cde5492fSlogin 576cde5492fSlogin let addr = unsafe { addr.as_ref() }.ok_or(SystemError::EFAULT)?; 577cde5492fSlogin unsafe { 578cde5492fSlogin match AddressFamily::try_from(addr.family)? { 579cde5492fSlogin AddressFamily::INet => { 5806046f775S裕依 if len < addr.len()? { 5816046f775S裕依 return Err(SystemError::EINVAL); 5826046f775S裕依 } 5836046f775S裕依 584cde5492fSlogin let addr_in: SockAddrIn = addr.addr_in; 585cde5492fSlogin 586cde5492fSlogin let ip: wire::IpAddress = wire::IpAddress::from(wire::Ipv4Address::from_bytes( 587cde5492fSlogin &u32::from_be(addr_in.sin_addr).to_be_bytes()[..], 588cde5492fSlogin )); 589cde5492fSlogin let port = u16::from_be(addr_in.sin_port); 590cde5492fSlogin 591cde5492fSlogin return Ok(Endpoint::Ip(Some(wire::IpEndpoint::new(ip, port)))); 592cde5492fSlogin } 5936046f775S裕依 AddressFamily::Unix => { 5946046f775S裕依 let addr_un: SockAddrUn = addr.addr_un; 5956046f775S裕依 5966046f775S裕依 let path = CStr::from_bytes_until_nul(&addr_un.sun_path) 5976046f775S裕依 .map_err(|_| SystemError::EINVAL)? 5986046f775S裕依 .to_str() 5996046f775S裕依 .map_err(|_| SystemError::EINVAL)?; 6006046f775S裕依 6016046f775S裕依 let fd = Syscall::open(path.as_ptr(), FileMode::O_RDWR.bits(), 0o755, true)?; 6026046f775S裕依 6036046f775S裕依 let binding = ProcessManager::current_pcb().fd_table(); 6046046f775S裕依 let fd_table_guard = binding.read(); 6056046f775S裕依 606*dfe53cf0SGnoCiYeH let file = fd_table_guard.get_file_by_fd(fd as i32).unwrap(); 6076046f775S裕依 if file.file_type() != FileType::Socket { 6086046f775S裕依 return Err(SystemError::ENOTSOCK); 6096046f775S裕依 } 6106046f775S裕依 let inode = file.inode(); 6116046f775S裕依 let socketinode = inode.as_any_ref().downcast_ref::<Arc<SocketInode>>(); 6126046f775S裕依 6136046f775S裕依 return Ok(Endpoint::Inode(socketinode.cloned())); 6146046f775S裕依 } 615cde5492fSlogin AddressFamily::Packet => { 616cde5492fSlogin // TODO: support packet socket 617cde5492fSlogin return Err(SystemError::EINVAL); 618cde5492fSlogin } 619cde5492fSlogin AddressFamily::Netlink => { 620cde5492fSlogin // TODO: support netlink socket 621cde5492fSlogin return Err(SystemError::EINVAL); 622cde5492fSlogin } 623cde5492fSlogin _ => { 624cde5492fSlogin return Err(SystemError::EINVAL); 625cde5492fSlogin } 626cde5492fSlogin } 627cde5492fSlogin } 628cde5492fSlogin } 629cde5492fSlogin 630cde5492fSlogin /// @brief 获取地址长度 631cde5492fSlogin pub fn len(&self) -> Result<usize, SystemError> { 632cde5492fSlogin let ret = match AddressFamily::try_from(unsafe { self.family })? { 633cde5492fSlogin AddressFamily::INet => Ok(core::mem::size_of::<SockAddrIn>()), 634cde5492fSlogin AddressFamily::Packet => Ok(core::mem::size_of::<SockAddrLl>()), 635cde5492fSlogin AddressFamily::Netlink => Ok(core::mem::size_of::<SockAddrNl>()), 636cde5492fSlogin AddressFamily::Unix => Err(SystemError::EINVAL), 637cde5492fSlogin _ => Err(SystemError::EINVAL), 638cde5492fSlogin }; 639cde5492fSlogin 640cde5492fSlogin return ret; 641cde5492fSlogin } 642cde5492fSlogin 643cde5492fSlogin /// @brief 把SockAddr的数据写入用户空间 644cde5492fSlogin /// 645cde5492fSlogin /// @param addr 用户空间的SockAddr的地址 646cde5492fSlogin /// @param len 要写入的长度 647cde5492fSlogin /// 648cde5492fSlogin /// @return 成功返回写入的长度,失败返回错误码 649cde5492fSlogin pub unsafe fn write_to_user( 650cde5492fSlogin &self, 651cde5492fSlogin addr: *mut SockAddr, 652cde5492fSlogin addr_len: *mut u32, 653cde5492fSlogin ) -> Result<usize, SystemError> { 654cde5492fSlogin // 当用户传入的地址或者长度为空时,直接返回0 655cde5492fSlogin if addr.is_null() || addr_len.is_null() { 656cde5492fSlogin return Ok(0); 657cde5492fSlogin } 6584fda81ceSLoGin 659cde5492fSlogin // 检查用户传入的地址是否合法 6604fda81ceSLoGin verify_area( 6614fda81ceSLoGin VirtAddr::new(addr as usize), 6624fda81ceSLoGin core::mem::size_of::<SockAddr>(), 6634fda81ceSLoGin ) 6644fda81ceSLoGin .map_err(|_| SystemError::EFAULT)?; 6654fda81ceSLoGin 6664fda81ceSLoGin verify_area( 6674fda81ceSLoGin VirtAddr::new(addr_len as usize), 6684fda81ceSLoGin core::mem::size_of::<u32>(), 6694fda81ceSLoGin ) 6704fda81ceSLoGin .map_err(|_| SystemError::EFAULT)?; 671cde5492fSlogin 672cde5492fSlogin let to_write = min(self.len()?, *addr_len as usize); 673cde5492fSlogin if to_write > 0 { 674cde5492fSlogin let buf = core::slice::from_raw_parts_mut(addr as *mut u8, to_write); 675cde5492fSlogin buf.copy_from_slice(core::slice::from_raw_parts( 676cde5492fSlogin self as *const SockAddr as *const u8, 677cde5492fSlogin to_write, 678cde5492fSlogin )); 679cde5492fSlogin } 680cde5492fSlogin *addr_len = self.len()? as u32; 681cde5492fSlogin return Ok(to_write); 682cde5492fSlogin } 683cde5492fSlogin } 684cde5492fSlogin 685cde5492fSlogin impl From<Endpoint> for SockAddr { 686cde5492fSlogin fn from(value: Endpoint) -> Self { 687cde5492fSlogin match value { 688cde5492fSlogin Endpoint::Ip(ip_endpoint) => { 689cde5492fSlogin // 未指定地址 690b5b571e0SLoGin if ip_endpoint.is_none() { 691cde5492fSlogin return SockAddr { 692cde5492fSlogin addr_ph: SockAddrPlaceholder { 693cde5492fSlogin family: AddressFamily::Unspecified as u16, 694cde5492fSlogin data: [0; 14], 695cde5492fSlogin }, 696cde5492fSlogin }; 697cde5492fSlogin } 698cde5492fSlogin // 指定了地址 699cde5492fSlogin let ip_endpoint = ip_endpoint.unwrap(); 700cde5492fSlogin match ip_endpoint.addr { 701cde5492fSlogin wire::IpAddress::Ipv4(ipv4_addr) => { 702cde5492fSlogin let addr_in = SockAddrIn { 703cde5492fSlogin sin_family: AddressFamily::INet as u16, 704cde5492fSlogin sin_port: ip_endpoint.port.to_be(), 705cde5492fSlogin sin_addr: u32::from_be_bytes(ipv4_addr.0).to_be(), 706cde5492fSlogin sin_zero: [0; 8], 707cde5492fSlogin }; 708cde5492fSlogin 709cde5492fSlogin return SockAddr { addr_in }; 710cde5492fSlogin } 711cde5492fSlogin _ => { 712cde5492fSlogin unimplemented!("not support ipv6"); 713cde5492fSlogin } 714cde5492fSlogin } 715cde5492fSlogin } 716cde5492fSlogin 717cde5492fSlogin Endpoint::LinkLayer(link_endpoint) => { 718cde5492fSlogin let addr_ll = SockAddrLl { 719cde5492fSlogin sll_family: AddressFamily::Packet as u16, 720cde5492fSlogin sll_protocol: 0, 721cde5492fSlogin sll_ifindex: link_endpoint.interface as u32, 722cde5492fSlogin sll_hatype: 0, 723cde5492fSlogin sll_pkttype: 0, 724cde5492fSlogin sll_halen: 0, 725cde5492fSlogin sll_addr: [0; 8], 726cde5492fSlogin }; 727cde5492fSlogin 728cde5492fSlogin return SockAddr { addr_ll }; 7294ad52e57S裕依2439 } 7306046f775S裕依 7314ad52e57S裕依2439 _ => { 7324ad52e57S裕依2439 // todo: support other endpoint, like Netlink... 7334ad52e57S裕依2439 unimplemented!("not support {value:?}"); 7344ad52e57S裕依2439 } 735cde5492fSlogin } 736cde5492fSlogin } 737cde5492fSlogin } 738cde5492fSlogin 739cde5492fSlogin #[repr(C)] 740cde5492fSlogin #[derive(Debug, Clone, Copy)] 741cde5492fSlogin pub struct MsgHdr { 742cde5492fSlogin /// 指向一个SockAddr结构体的指针 743cde5492fSlogin pub msg_name: *mut SockAddr, 744cde5492fSlogin /// SockAddr结构体的大小 745cde5492fSlogin pub msg_namelen: u32, 746cde5492fSlogin /// scatter/gather array 747cde5492fSlogin pub msg_iov: *mut IoVec, 748cde5492fSlogin /// elements in msg_iov 749cde5492fSlogin pub msg_iovlen: usize, 750cde5492fSlogin /// 辅助数据 751cde5492fSlogin pub msg_control: *mut u8, 752cde5492fSlogin /// 辅助数据长度 753cde5492fSlogin pub msg_controllen: usize, 754cde5492fSlogin /// 接收到的消息的标志 755cde5492fSlogin pub msg_flags: u32, 756cde5492fSlogin } 757cde5492fSlogin 758cde5492fSlogin #[derive(Debug, Clone, Copy, FromPrimitive, ToPrimitive, PartialEq, Eq)] 759cde5492fSlogin pub enum PosixIpProtocol { 760cde5492fSlogin /// Dummy protocol for TCP. 761cde5492fSlogin IP = 0, 762cde5492fSlogin /// Internet Control Message Protocol. 763cde5492fSlogin ICMP = 1, 764cde5492fSlogin /// Internet Group Management Protocol. 765cde5492fSlogin IGMP = 2, 766cde5492fSlogin /// IPIP tunnels (older KA9Q tunnels use 94). 767cde5492fSlogin IPIP = 4, 768cde5492fSlogin /// Transmission Control Protocol. 769cde5492fSlogin TCP = 6, 770cde5492fSlogin /// Exterior Gateway Protocol. 771cde5492fSlogin EGP = 8, 772cde5492fSlogin /// PUP protocol. 773cde5492fSlogin PUP = 12, 774cde5492fSlogin /// User Datagram Protocol. 775cde5492fSlogin UDP = 17, 776cde5492fSlogin /// XNS IDP protocol. 777cde5492fSlogin IDP = 22, 778cde5492fSlogin /// SO Transport Protocol Class 4. 779cde5492fSlogin TP = 29, 780cde5492fSlogin /// Datagram Congestion Control Protocol. 781cde5492fSlogin DCCP = 33, 782cde5492fSlogin /// IPv6-in-IPv4 tunnelling. 783cde5492fSlogin IPv6 = 41, 784cde5492fSlogin /// RSVP Protocol. 785cde5492fSlogin RSVP = 46, 786cde5492fSlogin /// Generic Routing Encapsulation. (Cisco GRE) (rfc 1701, 1702) 787cde5492fSlogin GRE = 47, 788cde5492fSlogin /// Encapsulation Security Payload protocol 789cde5492fSlogin ESP = 50, 790cde5492fSlogin /// Authentication Header protocol 791cde5492fSlogin AH = 51, 792cde5492fSlogin /// Multicast Transport Protocol. 793cde5492fSlogin MTP = 92, 794cde5492fSlogin /// IP option pseudo header for BEET 795cde5492fSlogin BEETPH = 94, 796cde5492fSlogin /// Encapsulation Header. 797cde5492fSlogin ENCAP = 98, 798cde5492fSlogin /// Protocol Independent Multicast. 799cde5492fSlogin PIM = 103, 800cde5492fSlogin /// Compression Header Protocol. 801cde5492fSlogin COMP = 108, 802cde5492fSlogin /// Stream Control Transport Protocol 803cde5492fSlogin SCTP = 132, 804cde5492fSlogin /// UDP-Lite protocol (RFC 3828) 805cde5492fSlogin UDPLITE = 136, 806cde5492fSlogin /// MPLS in IP (RFC 4023) 807cde5492fSlogin MPLSINIP = 137, 808cde5492fSlogin /// Ethernet-within-IPv6 Encapsulation 809cde5492fSlogin ETHERNET = 143, 810cde5492fSlogin /// Raw IP packets 811cde5492fSlogin RAW = 255, 812cde5492fSlogin /// Multipath TCP connection 813cde5492fSlogin MPTCP = 262, 814cde5492fSlogin } 815cde5492fSlogin 816cde5492fSlogin impl TryFrom<u16> for PosixIpProtocol { 817cde5492fSlogin type Error = SystemError; 818cde5492fSlogin 819cde5492fSlogin fn try_from(value: u16) -> Result<Self, Self::Error> { 820cde5492fSlogin match <Self as FromPrimitive>::from_u16(value) { 821cde5492fSlogin Some(p) => Ok(p), 822cde5492fSlogin None => Err(SystemError::EPROTONOSUPPORT), 823cde5492fSlogin } 824cde5492fSlogin } 825cde5492fSlogin } 826cde5492fSlogin 827b5b571e0SLoGin impl From<PosixIpProtocol> for u16 { 828b5b571e0SLoGin fn from(value: PosixIpProtocol) -> Self { 829b5b571e0SLoGin <PosixIpProtocol as ToPrimitive>::to_u16(&value).unwrap() 830cde5492fSlogin } 831cde5492fSlogin } 832cde5492fSlogin 833cde5492fSlogin #[allow(non_camel_case_types)] 834cde5492fSlogin #[derive(Debug, Clone, Copy, FromPrimitive, ToPrimitive, PartialEq, Eq)] 835cde5492fSlogin pub enum PosixSocketOption { 836cde5492fSlogin SO_DEBUG = 1, 837cde5492fSlogin SO_REUSEADDR = 2, 838cde5492fSlogin SO_TYPE = 3, 839cde5492fSlogin SO_ERROR = 4, 840cde5492fSlogin SO_DONTROUTE = 5, 841cde5492fSlogin SO_BROADCAST = 6, 842cde5492fSlogin SO_SNDBUF = 7, 843cde5492fSlogin SO_RCVBUF = 8, 844cde5492fSlogin SO_SNDBUFFORCE = 32, 845cde5492fSlogin SO_RCVBUFFORCE = 33, 846cde5492fSlogin SO_KEEPALIVE = 9, 847cde5492fSlogin SO_OOBINLINE = 10, 848cde5492fSlogin SO_NO_CHECK = 11, 849cde5492fSlogin SO_PRIORITY = 12, 850cde5492fSlogin SO_LINGER = 13, 851cde5492fSlogin SO_BSDCOMPAT = 14, 852cde5492fSlogin SO_REUSEPORT = 15, 853cde5492fSlogin SO_PASSCRED = 16, 854cde5492fSlogin SO_PEERCRED = 17, 855cde5492fSlogin SO_RCVLOWAT = 18, 856cde5492fSlogin SO_SNDLOWAT = 19, 857cde5492fSlogin SO_RCVTIMEO_OLD = 20, 858cde5492fSlogin SO_SNDTIMEO_OLD = 21, 859cde5492fSlogin 860cde5492fSlogin SO_SECURITY_AUTHENTICATION = 22, 861cde5492fSlogin SO_SECURITY_ENCRYPTION_TRANSPORT = 23, 862cde5492fSlogin SO_SECURITY_ENCRYPTION_NETWORK = 24, 863cde5492fSlogin 864cde5492fSlogin SO_BINDTODEVICE = 25, 865cde5492fSlogin 866cde5492fSlogin /// 与SO_GET_FILTER相同 867cde5492fSlogin SO_ATTACH_FILTER = 26, 868cde5492fSlogin SO_DETACH_FILTER = 27, 869cde5492fSlogin 870cde5492fSlogin SO_PEERNAME = 28, 871cde5492fSlogin 872cde5492fSlogin SO_ACCEPTCONN = 30, 873cde5492fSlogin 874cde5492fSlogin SO_PEERSEC = 31, 875cde5492fSlogin SO_PASSSEC = 34, 876cde5492fSlogin 877cde5492fSlogin SO_MARK = 36, 878cde5492fSlogin 879cde5492fSlogin SO_PROTOCOL = 38, 880cde5492fSlogin SO_DOMAIN = 39, 881cde5492fSlogin 882cde5492fSlogin SO_RXQ_OVFL = 40, 883cde5492fSlogin 884cde5492fSlogin /// 与SCM_WIFI_STATUS相同 885cde5492fSlogin SO_WIFI_STATUS = 41, 886cde5492fSlogin SO_PEEK_OFF = 42, 887cde5492fSlogin 888cde5492fSlogin /* Instruct lower device to use last 4-bytes of skb data as FCS */ 889cde5492fSlogin SO_NOFCS = 43, 890cde5492fSlogin 891cde5492fSlogin SO_LOCK_FILTER = 44, 892cde5492fSlogin SO_SELECT_ERR_QUEUE = 45, 893cde5492fSlogin SO_BUSY_POLL = 46, 894cde5492fSlogin SO_MAX_PACING_RATE = 47, 895cde5492fSlogin SO_BPF_EXTENSIONS = 48, 896cde5492fSlogin SO_INCOMING_CPU = 49, 897cde5492fSlogin SO_ATTACH_BPF = 50, 898cde5492fSlogin // SO_DETACH_BPF = SO_DETACH_FILTER, 899cde5492fSlogin SO_ATTACH_REUSEPORT_CBPF = 51, 900cde5492fSlogin SO_ATTACH_REUSEPORT_EBPF = 52, 901cde5492fSlogin 902cde5492fSlogin SO_CNX_ADVICE = 53, 903cde5492fSlogin SCM_TIMESTAMPING_OPT_STATS = 54, 904cde5492fSlogin SO_MEMINFO = 55, 905cde5492fSlogin SO_INCOMING_NAPI_ID = 56, 906cde5492fSlogin SO_COOKIE = 57, 907cde5492fSlogin SCM_TIMESTAMPING_PKTINFO = 58, 908cde5492fSlogin SO_PEERGROUPS = 59, 909cde5492fSlogin SO_ZEROCOPY = 60, 910cde5492fSlogin /// 与SCM_TXTIME相同 911cde5492fSlogin SO_TXTIME = 61, 912cde5492fSlogin 913cde5492fSlogin SO_BINDTOIFINDEX = 62, 914cde5492fSlogin 915cde5492fSlogin SO_TIMESTAMP_OLD = 29, 916cde5492fSlogin SO_TIMESTAMPNS_OLD = 35, 917cde5492fSlogin SO_TIMESTAMPING_OLD = 37, 918cde5492fSlogin SO_TIMESTAMP_NEW = 63, 919cde5492fSlogin SO_TIMESTAMPNS_NEW = 64, 920cde5492fSlogin SO_TIMESTAMPING_NEW = 65, 921cde5492fSlogin 922cde5492fSlogin SO_RCVTIMEO_NEW = 66, 923cde5492fSlogin SO_SNDTIMEO_NEW = 67, 924cde5492fSlogin 925cde5492fSlogin SO_DETACH_REUSEPORT_BPF = 68, 926cde5492fSlogin 927cde5492fSlogin SO_PREFER_BUSY_POLL = 69, 928cde5492fSlogin SO_BUSY_POLL_BUDGET = 70, 929cde5492fSlogin 930cde5492fSlogin SO_NETNS_COOKIE = 71, 931cde5492fSlogin SO_BUF_LOCK = 72, 932cde5492fSlogin SO_RESERVE_MEM = 73, 933cde5492fSlogin SO_TXREHASH = 74, 934cde5492fSlogin SO_RCVMARK = 75, 935cde5492fSlogin } 936cde5492fSlogin 937cde5492fSlogin impl TryFrom<i32> for PosixSocketOption { 938cde5492fSlogin type Error = SystemError; 939cde5492fSlogin 940cde5492fSlogin fn try_from(value: i32) -> Result<Self, Self::Error> { 941cde5492fSlogin match <Self as FromPrimitive>::from_i32(value) { 942cde5492fSlogin Some(p) => Ok(p), 943cde5492fSlogin None => Err(SystemError::EINVAL), 944cde5492fSlogin } 945cde5492fSlogin } 946cde5492fSlogin } 947cde5492fSlogin 948b5b571e0SLoGin impl From<PosixSocketOption> for i32 { 949b5b571e0SLoGin fn from(value: PosixSocketOption) -> Self { 950b5b571e0SLoGin <PosixSocketOption as ToPrimitive>::to_i32(&value).unwrap() 951cde5492fSlogin } 952cde5492fSlogin } 953cde5492fSlogin 954cde5492fSlogin #[derive(Debug, Clone, Copy, PartialEq, Eq, FromPrimitive, ToPrimitive)] 955cde5492fSlogin pub enum PosixTcpSocketOptions { 956cde5492fSlogin /// Turn off Nagle's algorithm. 957cde5492fSlogin NoDelay = 1, 958cde5492fSlogin /// Limit MSS. 959cde5492fSlogin MaxSegment = 2, 960cde5492fSlogin /// Never send partially complete segments. 961cde5492fSlogin Cork = 3, 962cde5492fSlogin /// Start keeplives after this period. 963cde5492fSlogin KeepIdle = 4, 964cde5492fSlogin /// Interval between keepalives. 965cde5492fSlogin KeepIntvl = 5, 966cde5492fSlogin /// Number of keepalives before death. 967cde5492fSlogin KeepCnt = 6, 968cde5492fSlogin /// Number of SYN retransmits. 969cde5492fSlogin Syncnt = 7, 970cde5492fSlogin /// Lifetime for orphaned FIN-WAIT-2 state. 971cde5492fSlogin Linger2 = 8, 972cde5492fSlogin /// Wake up listener only when data arrive. 973cde5492fSlogin DeferAccept = 9, 974cde5492fSlogin /// Bound advertised window 975cde5492fSlogin WindowClamp = 10, 976cde5492fSlogin /// Information about this connection. 977cde5492fSlogin Info = 11, 978cde5492fSlogin /// Block/reenable quick acks. 979cde5492fSlogin QuickAck = 12, 980cde5492fSlogin /// Congestion control algorithm. 981cde5492fSlogin Congestion = 13, 982cde5492fSlogin /// TCP MD5 Signature (RFC2385). 983cde5492fSlogin Md5Sig = 14, 984cde5492fSlogin /// Use linear timeouts for thin streams 985cde5492fSlogin ThinLinearTimeouts = 16, 986cde5492fSlogin /// Fast retrans. after 1 dupack. 987cde5492fSlogin ThinDupack = 17, 988cde5492fSlogin /// How long for loss retry before timeout. 989cde5492fSlogin UserTimeout = 18, 990cde5492fSlogin /// TCP sock is under repair right now. 991cde5492fSlogin Repair = 19, 992cde5492fSlogin RepairQueue = 20, 993cde5492fSlogin QueueSeq = 21, 994cde5492fSlogin RepairOptions = 22, 995cde5492fSlogin /// Enable FastOpen on listeners 996cde5492fSlogin FastOpen = 23, 997cde5492fSlogin Timestamp = 24, 998cde5492fSlogin /// Limit number of unsent bytes in write queue. 999cde5492fSlogin NotSentLowat = 25, 1000cde5492fSlogin /// Get Congestion Control (optional) info. 1001cde5492fSlogin CCInfo = 26, 1002cde5492fSlogin /// Record SYN headers for new connections. 1003cde5492fSlogin SaveSyn = 27, 1004cde5492fSlogin /// Get SYN headers recorded for connection. 1005cde5492fSlogin SavedSyn = 28, 1006cde5492fSlogin /// Get/set window parameters. 1007cde5492fSlogin RepairWindow = 29, 1008cde5492fSlogin /// Attempt FastOpen with connect. 1009cde5492fSlogin FastOpenConnect = 30, 1010cde5492fSlogin /// Attach a ULP to a TCP connection. 1011cde5492fSlogin ULP = 31, 1012cde5492fSlogin /// TCP MD5 Signature with extensions. 1013cde5492fSlogin Md5SigExt = 32, 1014cde5492fSlogin /// Set the key for Fast Open(cookie). 1015cde5492fSlogin FastOpenKey = 33, 1016cde5492fSlogin /// Enable TFO without a TFO cookie. 1017cde5492fSlogin FastOpenNoCookie = 34, 1018cde5492fSlogin ZeroCopyReceive = 35, 1019cde5492fSlogin /// Notify bytes available to read as a cmsg on read. 1020cde5492fSlogin /// 与TCP_CM_INQ相同 1021cde5492fSlogin INQ = 36, 1022cde5492fSlogin /// delay outgoing packets by XX usec 1023cde5492fSlogin TxDelay = 37, 1024cde5492fSlogin } 1025cde5492fSlogin 1026cde5492fSlogin impl TryFrom<i32> for PosixTcpSocketOptions { 1027cde5492fSlogin type Error = SystemError; 1028cde5492fSlogin 1029cde5492fSlogin fn try_from(value: i32) -> Result<Self, Self::Error> { 1030cde5492fSlogin match <Self as FromPrimitive>::from_i32(value) { 1031cde5492fSlogin Some(p) => Ok(p), 1032cde5492fSlogin None => Err(SystemError::EINVAL), 1033cde5492fSlogin } 1034cde5492fSlogin } 1035cde5492fSlogin } 1036cde5492fSlogin 1037b5b571e0SLoGin impl From<PosixTcpSocketOptions> for i32 { 1038b5b571e0SLoGin fn from(val: PosixTcpSocketOptions) -> Self { 1039b5b571e0SLoGin <PosixTcpSocketOptions as ToPrimitive>::to_i32(&val).unwrap() 1040cde5492fSlogin } 1041cde5492fSlogin } 1042