14ad52e57S裕依2439 use core::{any::Any, fmt::Debug, sync::atomic::AtomicUsize}; 24ad52e57S裕依2439 34ad52e57S裕依2439 use alloc::{ 44ad52e57S裕依2439 boxed::Box, 54ad52e57S裕依2439 collections::LinkedList, 64ad52e57S裕依2439 string::String, 74ad52e57S裕依2439 sync::{Arc, Weak}, 84ad52e57S裕依2439 vec::Vec, 94ad52e57S裕依2439 }; 104ad52e57S裕依2439 use hashbrown::HashMap; 11*2eab6dd7S曾俊 use log::warn; 124ad52e57S裕依2439 use smoltcp::{ 13d623e902SGnoCiYeH iface::SocketSet, 1493c37970Ssun5etop socket::{self, raw, tcp, udp}, 154ad52e57S裕依2439 }; 164ad52e57S裕依2439 use system_error::SystemError; 174ad52e57S裕依2439 184ad52e57S裕依2439 use crate::{ 19f0c87a89SGnoCiYeH arch::rand::rand, 204ad52e57S裕依2439 filesystem::vfs::{ 214ad52e57S裕依2439 file::FileMode, syscall::ModeType, FilePrivateData, FileSystem, FileType, IndexNode, 224ad52e57S裕依2439 Metadata, 234ad52e57S裕依2439 }, 244ad52e57S裕依2439 libs::{ 254ad52e57S裕依2439 rwlock::{RwLock, RwLockReadGuard, RwLockWriteGuard}, 264ad52e57S裕依2439 spinlock::{SpinLock, SpinLockGuard}, 274ad52e57S裕依2439 wait_queue::EventWaitQueue, 284ad52e57S裕依2439 }, 2937cef00bSSamuel Dai process::{Pid, ProcessManager}, 30f0c87a89SGnoCiYeH sched::{schedule, SchedMode}, 314ad52e57S裕依2439 }; 324ad52e57S裕依2439 336046f775S裕依 use self::{ 34d623e902SGnoCiYeH handle::GlobalSocketHandle, 356046f775S裕依 inet::{RawSocket, TcpSocket, UdpSocket}, 366046f775S裕依 unix::{SeqpacketSocket, StreamSocket}, 376046f775S裕依 }; 384ad52e57S裕依2439 394ad52e57S裕依2439 use super::{ 404ad52e57S裕依2439 event_poll::{EPollEventType, EPollItem, EventPoll}, 414ad52e57S裕依2439 Endpoint, Protocol, ShutdownType, 424ad52e57S裕依2439 }; 434ad52e57S裕依2439 44d623e902SGnoCiYeH pub mod handle; 456046f775S裕依 pub mod inet; 466046f775S裕依 pub mod unix; 474ad52e57S裕依2439 484ad52e57S裕依2439 lazy_static! { 494ad52e57S裕依2439 /// 所有socket的集合 504ad52e57S裕依2439 /// TODO: 优化这里,自己实现SocketSet!!!现在这样的话,不管全局有多少个网卡,每个时间点都只会有1个进程能够访问socket 514ad52e57S裕依2439 pub static ref SOCKET_SET: SpinLock<SocketSet<'static >> = SpinLock::new(SocketSet::new(vec![])); 524ad52e57S裕依2439 /// SocketHandle表,每个SocketHandle对应一个SocketHandleItem, 534ad52e57S裕依2439 /// 注意!:在网卡中断中需要拿到这张表的,在获取读锁时应该确保关中断避免死锁 54d623e902SGnoCiYeH pub static ref HANDLE_MAP: RwLock<HashMap<GlobalSocketHandle, SocketHandleItem>> = RwLock::new(HashMap::new()); 554ad52e57S裕依2439 /// 端口管理器 564ad52e57S裕依2439 pub static ref PORT_MANAGER: PortManager = PortManager::new(); 574ad52e57S裕依2439 } 584ad52e57S裕依2439 594ad52e57S裕依2439 /* For setsockopt(2) */ 604ad52e57S裕依2439 // See: linux-5.19.10/include/uapi/asm-generic/socket.h#9 614ad52e57S裕依2439 pub const SOL_SOCKET: u8 = 1; 624ad52e57S裕依2439 634ad52e57S裕依2439 /// 根据地址族、socket类型和协议创建socket 644ad52e57S裕依2439 pub(super) fn new_socket( 654ad52e57S裕依2439 address_family: AddressFamily, 664ad52e57S裕依2439 socket_type: PosixSocketType, 674ad52e57S裕依2439 protocol: Protocol, 684ad52e57S裕依2439 ) -> Result<Box<dyn Socket>, SystemError> { 694ad52e57S裕依2439 let socket: Box<dyn Socket> = match address_family { 704ad52e57S裕依2439 AddressFamily::Unix => match socket_type { 716046f775S裕依 PosixSocketType::Stream => Box::new(StreamSocket::new(SocketOptions::default())), 724ad52e57S裕依2439 PosixSocketType::SeqPacket => Box::new(SeqpacketSocket::new(SocketOptions::default())), 734ad52e57S裕依2439 _ => { 744ad52e57S裕依2439 return Err(SystemError::EINVAL); 754ad52e57S裕依2439 } 764ad52e57S裕依2439 }, 774ad52e57S裕依2439 AddressFamily::INet => match socket_type { 784ad52e57S裕依2439 PosixSocketType::Stream => Box::new(TcpSocket::new(SocketOptions::default())), 794ad52e57S裕依2439 PosixSocketType::Datagram => Box::new(UdpSocket::new(SocketOptions::default())), 804ad52e57S裕依2439 PosixSocketType::Raw => Box::new(RawSocket::new(protocol, SocketOptions::default())), 814ad52e57S裕依2439 _ => { 824ad52e57S裕依2439 return Err(SystemError::EINVAL); 834ad52e57S裕依2439 } 844ad52e57S裕依2439 }, 854ad52e57S裕依2439 _ => { 864ad52e57S裕依2439 return Err(SystemError::EAFNOSUPPORT); 874ad52e57S裕依2439 } 884ad52e57S裕依2439 }; 89d623e902SGnoCiYeH 9037cef00bSSamuel Dai let handle_item = SocketHandleItem::new(None); 91d623e902SGnoCiYeH HANDLE_MAP 92d623e902SGnoCiYeH .write_irqsave() 93d623e902SGnoCiYeH .insert(socket.socket_handle(), handle_item); 944ad52e57S裕依2439 Ok(socket) 954ad52e57S裕依2439 } 964ad52e57S裕依2439 974ad52e57S裕依2439 pub trait Socket: Sync + Send + Debug + Any { 984ad52e57S裕依2439 /// @brief 从socket中读取数据,如果socket是阻塞的,那么直到读取到数据才返回 994ad52e57S裕依2439 /// 1004ad52e57S裕依2439 /// @param buf 读取到的数据存放的缓冲区 1014ad52e57S裕依2439 /// 1024ad52e57S裕依2439 /// @return - 成功:(返回读取的数据的长度,读取数据的端点). 1034ad52e57S裕依2439 /// - 失败:错误码 1046046f775S裕依 fn read(&self, buf: &mut [u8]) -> (Result<usize, SystemError>, Endpoint); 1054ad52e57S裕依2439 1064ad52e57S裕依2439 /// @brief 向socket中写入数据。如果socket是阻塞的,那么直到写入的数据全部写入socket中才返回 1074ad52e57S裕依2439 /// 1084ad52e57S裕依2439 /// @param buf 要写入的数据 1094ad52e57S裕依2439 /// @param to 要写入的目的端点,如果是None,那么写入的数据将会被丢弃 1104ad52e57S裕依2439 /// 1114ad52e57S裕依2439 /// @return 返回写入的数据的长度 1124ad52e57S裕依2439 fn write(&self, buf: &[u8], to: Option<Endpoint>) -> Result<usize, SystemError>; 1134ad52e57S裕依2439 1144ad52e57S裕依2439 /// @brief 对应于POSIX的connect函数,用于连接到指定的远程服务器端点 1154ad52e57S裕依2439 /// 1164ad52e57S裕依2439 /// It is used to establish a connection to a remote server. 1174ad52e57S裕依2439 /// When a socket is connected to a remote server, 1184ad52e57S裕依2439 /// the operating system will establish a network connection with the server 1194ad52e57S裕依2439 /// and allow data to be sent and received between the local socket and the remote server. 1204ad52e57S裕依2439 /// 1214ad52e57S裕依2439 /// @param endpoint 要连接的端点 1224ad52e57S裕依2439 /// 1234ad52e57S裕依2439 /// @return 返回连接是否成功 1246046f775S裕依 fn connect(&mut self, _endpoint: Endpoint) -> Result<(), SystemError>; 1254ad52e57S裕依2439 1264ad52e57S裕依2439 /// @brief 对应于POSIX的bind函数,用于绑定到本机指定的端点 1274ad52e57S裕依2439 /// 1284ad52e57S裕依2439 /// The bind() function is used to associate a socket with a particular IP address and port number on the local machine. 1294ad52e57S裕依2439 /// 1304ad52e57S裕依2439 /// @param endpoint 要绑定的端点 1314ad52e57S裕依2439 /// 1324ad52e57S裕依2439 /// @return 返回绑定是否成功 1334ad52e57S裕依2439 fn bind(&mut self, _endpoint: Endpoint) -> Result<(), SystemError> { 1346046f775S裕依 Err(SystemError::ENOSYS) 1354ad52e57S裕依2439 } 1364ad52e57S裕依2439 1374ad52e57S裕依2439 /// @brief 对应于 POSIX 的 shutdown 函数,用于关闭socket。 1384ad52e57S裕依2439 /// 1394ad52e57S裕依2439 /// shutdown() 函数用于启动网络连接的正常关闭。 1404ad52e57S裕依2439 /// 当在两个端点之间建立网络连接时,任一端点都可以通过调用其端点对象上的 shutdown() 函数来启动关闭序列。 1414ad52e57S裕依2439 /// 此函数向远程端点发送关闭消息以指示本地端点不再接受新数据。 1424ad52e57S裕依2439 /// 1434ad52e57S裕依2439 /// @return 返回是否成功关闭 1444ad52e57S裕依2439 fn shutdown(&mut self, _type: ShutdownType) -> Result<(), SystemError> { 1456046f775S裕依 Err(SystemError::ENOSYS) 1464ad52e57S裕依2439 } 1474ad52e57S裕依2439 1484ad52e57S裕依2439 /// @brief 对应于POSIX的listen函数,用于监听端点 1494ad52e57S裕依2439 /// 1504ad52e57S裕依2439 /// @param backlog 最大的等待连接数 1514ad52e57S裕依2439 /// 1524ad52e57S裕依2439 /// @return 返回监听是否成功 1534ad52e57S裕依2439 fn listen(&mut self, _backlog: usize) -> Result<(), SystemError> { 1546046f775S裕依 Err(SystemError::ENOSYS) 1554ad52e57S裕依2439 } 1564ad52e57S裕依2439 1574ad52e57S裕依2439 /// @brief 对应于POSIX的accept函数,用于接受连接 1584ad52e57S裕依2439 /// 1594ad52e57S裕依2439 /// @param endpoint 对端的端点 1604ad52e57S裕依2439 /// 1614ad52e57S裕依2439 /// @return 返回接受连接是否成功 1624ad52e57S裕依2439 fn accept(&mut self) -> Result<(Box<dyn Socket>, Endpoint), SystemError> { 1636046f775S裕依 Err(SystemError::ENOSYS) 1644ad52e57S裕依2439 } 1654ad52e57S裕依2439 1664ad52e57S裕依2439 /// @brief 获取socket的端点 1674ad52e57S裕依2439 /// 1684ad52e57S裕依2439 /// @return 返回socket的端点 1694ad52e57S裕依2439 fn endpoint(&self) -> Option<Endpoint> { 1706046f775S裕依 None 1714ad52e57S裕依2439 } 1724ad52e57S裕依2439 1734ad52e57S裕依2439 /// @brief 获取socket的对端端点 1744ad52e57S裕依2439 /// 1754ad52e57S裕依2439 /// @return 返回socket的对端端点 1764ad52e57S裕依2439 fn peer_endpoint(&self) -> Option<Endpoint> { 1774ad52e57S裕依2439 None 1784ad52e57S裕依2439 } 1794ad52e57S裕依2439 1804ad52e57S裕依2439 /// @brief 1814ad52e57S裕依2439 /// The purpose of the poll function is to provide 1824ad52e57S裕依2439 /// a non-blocking way to check if a socket is ready for reading or writing, 1834ad52e57S裕依2439 /// so that you can efficiently handle multiple sockets in a single thread or event loop. 1844ad52e57S裕依2439 /// 1854ad52e57S裕依2439 /// @return (in, out, err) 1864ad52e57S裕依2439 /// 1874ad52e57S裕依2439 /// The first boolean value indicates whether the socket is ready for reading. If it is true, then there is data available to be read from the socket without blocking. 1884ad52e57S裕依2439 /// The second boolean value indicates whether the socket is ready for writing. If it is true, then data can be written to the socket without blocking. 1894ad52e57S裕依2439 /// The third boolean value indicates whether the socket has encountered an error condition. If it is true, then the socket is in an error state and should be closed or reset 1904ad52e57S裕依2439 /// 1914ad52e57S裕依2439 fn poll(&self) -> EPollEventType { 1926046f775S裕依 EPollEventType::empty() 1934ad52e57S裕依2439 } 1944ad52e57S裕依2439 1954ad52e57S裕依2439 /// @brief socket的ioctl函数 1964ad52e57S裕依2439 /// 1974ad52e57S裕依2439 /// @param cmd ioctl命令 1984ad52e57S裕依2439 /// @param arg0 ioctl命令的第一个参数 1994ad52e57S裕依2439 /// @param arg1 ioctl命令的第二个参数 2004ad52e57S裕依2439 /// @param arg2 ioctl命令的第三个参数 2014ad52e57S裕依2439 /// 2024ad52e57S裕依2439 /// @return 返回ioctl命令的返回值 2034ad52e57S裕依2439 fn ioctl( 2044ad52e57S裕依2439 &self, 2054ad52e57S裕依2439 _cmd: usize, 2064ad52e57S裕依2439 _arg0: usize, 2074ad52e57S裕依2439 _arg1: usize, 2084ad52e57S裕依2439 _arg2: usize, 2094ad52e57S裕依2439 ) -> Result<usize, SystemError> { 2106046f775S裕依 Ok(0) 2114ad52e57S裕依2439 } 2124ad52e57S裕依2439 2134ad52e57S裕依2439 /// @brief 获取socket的元数据 2146046f775S裕依 fn metadata(&self) -> SocketMetadata; 2154ad52e57S裕依2439 2164ad52e57S裕依2439 fn box_clone(&self) -> Box<dyn Socket>; 2174ad52e57S裕依2439 2184ad52e57S裕依2439 /// @brief 设置socket的选项 2194ad52e57S裕依2439 /// 2204ad52e57S裕依2439 /// @param level 选项的层次 2214ad52e57S裕依2439 /// @param optname 选项的名称 2224ad52e57S裕依2439 /// @param optval 选项的值 2234ad52e57S裕依2439 /// 2244ad52e57S裕依2439 /// @return 返回设置是否成功, 如果不支持该选项,返回ENOSYS 2254ad52e57S裕依2439 fn setsockopt( 2264ad52e57S裕依2439 &self, 2274ad52e57S裕依2439 _level: usize, 2284ad52e57S裕依2439 _optname: usize, 2294ad52e57S裕依2439 _optval: &[u8], 2304ad52e57S裕依2439 ) -> Result<(), SystemError> { 231*2eab6dd7S曾俊 warn!("setsockopt is not implemented"); 2326046f775S裕依 Ok(()) 2334ad52e57S裕依2439 } 2344ad52e57S裕依2439 235d623e902SGnoCiYeH fn socket_handle(&self) -> GlobalSocketHandle; 2364ad52e57S裕依2439 2376046f775S裕依 fn write_buffer(&self, _buf: &[u8]) -> Result<usize, SystemError> { 2386046f775S裕依 todo!() 2396046f775S裕依 } 2406046f775S裕依 2416046f775S裕依 fn as_any_ref(&self) -> &dyn Any; 2426046f775S裕依 2436046f775S裕依 fn as_any_mut(&mut self) -> &mut dyn Any; 2446046f775S裕依 2454ad52e57S裕依2439 fn add_epoll(&mut self, epitem: Arc<EPollItem>) -> Result<(), SystemError> { 2464ad52e57S裕依2439 HANDLE_MAP 2474ad52e57S裕依2439 .write_irqsave() 2484ad52e57S裕依2439 .get_mut(&self.socket_handle()) 2494ad52e57S裕依2439 .unwrap() 2504ad52e57S裕依2439 .add_epoll(epitem); 2514ad52e57S裕依2439 Ok(()) 2524ad52e57S裕依2439 } 2534ad52e57S裕依2439 2544ad52e57S裕依2439 fn remove_epoll(&mut self, epoll: &Weak<SpinLock<EventPoll>>) -> Result<(), SystemError> { 2554ad52e57S裕依2439 HANDLE_MAP 2564ad52e57S裕依2439 .write_irqsave() 2574ad52e57S裕依2439 .get_mut(&self.socket_handle()) 2584ad52e57S裕依2439 .unwrap() 2594ad52e57S裕依2439 .remove_epoll(epoll)?; 2604ad52e57S裕依2439 2614ad52e57S裕依2439 Ok(()) 2624ad52e57S裕依2439 } 2634ad52e57S裕依2439 2644ad52e57S裕依2439 fn clear_epoll(&mut self) -> Result<(), SystemError> { 2654ad52e57S裕依2439 let mut handle_map_guard = HANDLE_MAP.write_irqsave(); 2664ad52e57S裕依2439 let handle_item = handle_map_guard.get_mut(&self.socket_handle()).unwrap(); 2674ad52e57S裕依2439 2684ad52e57S裕依2439 for epitem in handle_item.epitems.lock_irqsave().iter() { 2694ad52e57S裕依2439 let epoll = epitem.epoll(); 2704ad52e57S裕依2439 if epoll.upgrade().is_some() { 2714ad52e57S裕依2439 EventPoll::ep_remove( 2724ad52e57S裕依2439 &mut epoll.upgrade().unwrap().lock_irqsave(), 2734ad52e57S裕依2439 epitem.fd(), 2744ad52e57S裕依2439 None, 2754ad52e57S裕依2439 )?; 2764ad52e57S裕依2439 } 2774ad52e57S裕依2439 } 2784ad52e57S裕依2439 2794ad52e57S裕依2439 Ok(()) 2804ad52e57S裕依2439 } 281d623e902SGnoCiYeH 282d623e902SGnoCiYeH fn close(&mut self); 2834ad52e57S裕依2439 } 2844ad52e57S裕依2439 2854ad52e57S裕依2439 impl Clone for Box<dyn Socket> { 2864ad52e57S裕依2439 fn clone(&self) -> Box<dyn Socket> { 2874ad52e57S裕依2439 self.box_clone() 2884ad52e57S裕依2439 } 2894ad52e57S裕依2439 } 2904ad52e57S裕依2439 2914ad52e57S裕依2439 /// # Socket在文件系统中的inode封装 2924ad52e57S裕依2439 #[derive(Debug)] 2934ad52e57S裕依2439 pub struct SocketInode(SpinLock<Box<dyn Socket>>, AtomicUsize); 2944ad52e57S裕依2439 2954ad52e57S裕依2439 impl SocketInode { 2964ad52e57S裕依2439 pub fn new(socket: Box<dyn Socket>) -> Arc<Self> { 2974ad52e57S裕依2439 Arc::new(Self(SpinLock::new(socket), AtomicUsize::new(0))) 2984ad52e57S裕依2439 } 2994ad52e57S裕依2439 3004ad52e57S裕依2439 #[inline] 3014ad52e57S裕依2439 pub fn inner(&self) -> SpinLockGuard<Box<dyn Socket>> { 3026046f775S裕依 self.0.lock() 3034ad52e57S裕依2439 } 3044ad52e57S裕依2439 3054ad52e57S裕依2439 pub unsafe fn inner_no_preempt(&self) -> SpinLockGuard<Box<dyn Socket>> { 3066046f775S裕依 self.0.lock_no_preempt() 3074ad52e57S裕依2439 } 30837cef00bSSamuel Dai 30937cef00bSSamuel Dai fn do_close(&self) -> Result<(), SystemError> { 31037cef00bSSamuel Dai let prev_ref_count = self.1.fetch_sub(1, core::sync::atomic::Ordering::SeqCst); 31137cef00bSSamuel Dai if prev_ref_count == 1 { 31237cef00bSSamuel Dai // 最后一次关闭,需要释放 31337cef00bSSamuel Dai let mut socket = self.0.lock_irqsave(); 31437cef00bSSamuel Dai 31537cef00bSSamuel Dai if socket.metadata().socket_type == SocketType::Unix { 31637cef00bSSamuel Dai return Ok(()); 31737cef00bSSamuel Dai } 31837cef00bSSamuel Dai 31937cef00bSSamuel Dai if let Some(Endpoint::Ip(Some(ip))) = socket.endpoint() { 32037cef00bSSamuel Dai PORT_MANAGER.unbind_port(socket.metadata().socket_type, ip.port); 32137cef00bSSamuel Dai } 32237cef00bSSamuel Dai 32337cef00bSSamuel Dai socket.clear_epoll()?; 32437cef00bSSamuel Dai 32537cef00bSSamuel Dai HANDLE_MAP 32637cef00bSSamuel Dai .write_irqsave() 32737cef00bSSamuel Dai .remove(&socket.socket_handle()) 32837cef00bSSamuel Dai .unwrap(); 32937cef00bSSamuel Dai socket.close(); 33037cef00bSSamuel Dai } 33137cef00bSSamuel Dai 33237cef00bSSamuel Dai Ok(()) 33337cef00bSSamuel Dai } 33437cef00bSSamuel Dai } 33537cef00bSSamuel Dai 33637cef00bSSamuel Dai impl Drop for SocketInode { 33737cef00bSSamuel Dai fn drop(&mut self) { 33837cef00bSSamuel Dai for _ in 0..self.1.load(core::sync::atomic::Ordering::SeqCst) { 33937cef00bSSamuel Dai let _ = self.do_close(); 34037cef00bSSamuel Dai } 34137cef00bSSamuel Dai } 3424ad52e57S裕依2439 } 3434ad52e57S裕依2439 3444ad52e57S裕依2439 impl IndexNode for SocketInode { 345dfe53cf0SGnoCiYeH fn open( 346dfe53cf0SGnoCiYeH &self, 347dfe53cf0SGnoCiYeH _data: SpinLockGuard<FilePrivateData>, 348dfe53cf0SGnoCiYeH _mode: &FileMode, 349dfe53cf0SGnoCiYeH ) -> Result<(), SystemError> { 3504ad52e57S裕依2439 self.1.fetch_add(1, core::sync::atomic::Ordering::SeqCst); 3514ad52e57S裕依2439 Ok(()) 3524ad52e57S裕依2439 } 3534ad52e57S裕依2439 354dfe53cf0SGnoCiYeH fn close(&self, _data: SpinLockGuard<FilePrivateData>) -> Result<(), SystemError> { 35537cef00bSSamuel Dai self.do_close() 3564ad52e57S裕依2439 } 3574ad52e57S裕依2439 3584ad52e57S裕依2439 fn read_at( 3594ad52e57S裕依2439 &self, 3604ad52e57S裕依2439 _offset: usize, 3614ad52e57S裕依2439 len: usize, 3624ad52e57S裕依2439 buf: &mut [u8], 363f0c87a89SGnoCiYeH data: SpinLockGuard<FilePrivateData>, 3644ad52e57S裕依2439 ) -> Result<usize, SystemError> { 365f0c87a89SGnoCiYeH drop(data); 3666046f775S裕依 self.0.lock_no_preempt().read(&mut buf[0..len]).0 3674ad52e57S裕依2439 } 3684ad52e57S裕依2439 3694ad52e57S裕依2439 fn write_at( 3704ad52e57S裕依2439 &self, 3714ad52e57S裕依2439 _offset: usize, 3724ad52e57S裕依2439 len: usize, 3734ad52e57S裕依2439 buf: &[u8], 374f0c87a89SGnoCiYeH data: SpinLockGuard<FilePrivateData>, 3754ad52e57S裕依2439 ) -> Result<usize, SystemError> { 376f0c87a89SGnoCiYeH drop(data); 3776046f775S裕依 self.0.lock_no_preempt().write(&buf[0..len], None) 3784ad52e57S裕依2439 } 3794ad52e57S裕依2439 3804ad52e57S裕依2439 fn poll(&self, _private_data: &FilePrivateData) -> Result<usize, SystemError> { 3814ad52e57S裕依2439 let events = self.0.lock_irqsave().poll(); 3824ad52e57S裕依2439 return Ok(events.bits() as usize); 3834ad52e57S裕依2439 } 3844ad52e57S裕依2439 3854ad52e57S裕依2439 fn fs(&self) -> Arc<dyn FileSystem> { 3864ad52e57S裕依2439 todo!() 3874ad52e57S裕依2439 } 3884ad52e57S裕依2439 3894ad52e57S裕依2439 fn as_any_ref(&self) -> &dyn Any { 3904ad52e57S裕依2439 self 3914ad52e57S裕依2439 } 3924ad52e57S裕依2439 3934ad52e57S裕依2439 fn list(&self) -> Result<Vec<String>, SystemError> { 3944ad52e57S裕依2439 return Err(SystemError::ENOTDIR); 3954ad52e57S裕依2439 } 3964ad52e57S裕依2439 3974ad52e57S裕依2439 fn metadata(&self) -> Result<Metadata, SystemError> { 3984ad52e57S裕依2439 let meta = Metadata { 3994ad52e57S裕依2439 mode: ModeType::from_bits_truncate(0o755), 4004ad52e57S裕依2439 file_type: FileType::Socket, 4014ad52e57S裕依2439 ..Default::default() 4024ad52e57S裕依2439 }; 4034ad52e57S裕依2439 4044ad52e57S裕依2439 return Ok(meta); 4054ad52e57S裕依2439 } 4064ad52e57S裕依2439 4074ad52e57S裕依2439 fn resize(&self, _len: usize) -> Result<(), SystemError> { 4084ad52e57S裕依2439 return Ok(()); 4094ad52e57S裕依2439 } 4104ad52e57S裕依2439 } 4114ad52e57S裕依2439 4124ad52e57S裕依2439 #[derive(Debug)] 4134ad52e57S裕依2439 pub struct SocketHandleItem { 4144ad52e57S裕依2439 /// shutdown状态 4154ad52e57S裕依2439 pub shutdown_type: RwLock<ShutdownType>, 4164ad52e57S裕依2439 /// socket的waitqueue 41737cef00bSSamuel Dai pub wait_queue: Arc<EventWaitQueue>, 4184ad52e57S裕依2439 /// epitems,考虑写在这是否是最优解? 4194ad52e57S裕依2439 pub epitems: SpinLock<LinkedList<Arc<EPollItem>>>, 4204ad52e57S裕依2439 } 4214ad52e57S裕依2439 4224ad52e57S裕依2439 impl SocketHandleItem { 42337cef00bSSamuel Dai pub fn new(wait_queue: Option<Arc<EventWaitQueue>>) -> Self { 4244ad52e57S裕依2439 Self { 4254ad52e57S裕依2439 shutdown_type: RwLock::new(ShutdownType::empty()), 42637cef00bSSamuel Dai wait_queue: wait_queue.unwrap_or(Arc::new(EventWaitQueue::new())), 4274ad52e57S裕依2439 epitems: SpinLock::new(LinkedList::new()), 4284ad52e57S裕依2439 } 4294ad52e57S裕依2439 } 4304ad52e57S裕依2439 4316046f775S裕依 /// ## 在socket的等待队列上睡眠 4324ad52e57S裕依2439 pub fn sleep( 433d623e902SGnoCiYeH socket_handle: GlobalSocketHandle, 4344ad52e57S裕依2439 events: u64, 435d623e902SGnoCiYeH handle_map_guard: RwLockReadGuard<'_, HashMap<GlobalSocketHandle, SocketHandleItem>>, 4364ad52e57S裕依2439 ) { 4374ad52e57S裕依2439 unsafe { 4384ad52e57S裕依2439 handle_map_guard 4394ad52e57S裕依2439 .get(&socket_handle) 4404ad52e57S裕依2439 .unwrap() 4414ad52e57S裕依2439 .wait_queue 4424ad52e57S裕依2439 .sleep_without_schedule(events) 4434ad52e57S裕依2439 }; 4444ad52e57S裕依2439 drop(handle_map_guard); 445f0c87a89SGnoCiYeH schedule(SchedMode::SM_NONE); 4464ad52e57S裕依2439 } 4474ad52e57S裕依2439 4484ad52e57S裕依2439 pub fn shutdown_type(&self) -> ShutdownType { 449b5b571e0SLoGin *self.shutdown_type.read() 4504ad52e57S裕依2439 } 4514ad52e57S裕依2439 4524ad52e57S裕依2439 pub fn shutdown_type_writer(&mut self) -> RwLockWriteGuard<ShutdownType> { 4534ad52e57S裕依2439 self.shutdown_type.write_irqsave() 4544ad52e57S裕依2439 } 4554ad52e57S裕依2439 4564ad52e57S裕依2439 pub fn add_epoll(&mut self, epitem: Arc<EPollItem>) { 4574ad52e57S裕依2439 self.epitems.lock_irqsave().push_back(epitem) 4584ad52e57S裕依2439 } 4594ad52e57S裕依2439 4604ad52e57S裕依2439 pub fn remove_epoll(&mut self, epoll: &Weak<SpinLock<EventPoll>>) -> Result<(), SystemError> { 4614ad52e57S裕依2439 let is_remove = !self 4624ad52e57S裕依2439 .epitems 4634ad52e57S裕依2439 .lock_irqsave() 4644ad52e57S裕依2439 .extract_if(|x| x.epoll().ptr_eq(epoll)) 4654ad52e57S裕依2439 .collect::<Vec<_>>() 4664ad52e57S裕依2439 .is_empty(); 4674ad52e57S裕依2439 4684ad52e57S裕依2439 if is_remove { 4694ad52e57S裕依2439 return Ok(()); 4704ad52e57S裕依2439 } 4714ad52e57S裕依2439 4724ad52e57S裕依2439 Err(SystemError::ENOENT) 4734ad52e57S裕依2439 } 4744ad52e57S裕依2439 } 4754ad52e57S裕依2439 4764ad52e57S裕依2439 /// # TCP 和 UDP 的端口管理器。 4774ad52e57S裕依2439 /// 如果 TCP/UDP 的 socket 绑定了某个端口,它会在对应的表中记录,以检测端口冲突。 4784ad52e57S裕依2439 pub struct PortManager { 4794ad52e57S裕依2439 // TCP 端口记录表 48037cef00bSSamuel Dai tcp_port_table: SpinLock<HashMap<u16, Pid>>, 4814ad52e57S裕依2439 // UDP 端口记录表 48237cef00bSSamuel Dai udp_port_table: SpinLock<HashMap<u16, Pid>>, 4834ad52e57S裕依2439 } 4844ad52e57S裕依2439 4854ad52e57S裕依2439 impl PortManager { 4864ad52e57S裕依2439 pub fn new() -> Self { 4874ad52e57S裕依2439 return Self { 4884ad52e57S裕依2439 tcp_port_table: SpinLock::new(HashMap::new()), 4894ad52e57S裕依2439 udp_port_table: SpinLock::new(HashMap::new()), 4904ad52e57S裕依2439 }; 4914ad52e57S裕依2439 } 4924ad52e57S裕依2439 4934ad52e57S裕依2439 /// @brief 自动分配一个相对应协议中未被使用的PORT,如果动态端口均已被占用,返回错误码 EADDRINUSE 4944ad52e57S裕依2439 pub fn get_ephemeral_port(&self, socket_type: SocketType) -> Result<u16, SystemError> { 4954ad52e57S裕依2439 // TODO: selects non-conflict high port 4964ad52e57S裕依2439 4974ad52e57S裕依2439 static mut EPHEMERAL_PORT: u16 = 0; 4984ad52e57S裕依2439 unsafe { 4994ad52e57S裕依2439 if EPHEMERAL_PORT == 0 { 5004ad52e57S裕依2439 EPHEMERAL_PORT = (49152 + rand() % (65536 - 49152)) as u16; 5014ad52e57S裕依2439 } 5024ad52e57S裕依2439 } 5034ad52e57S裕依2439 5044ad52e57S裕依2439 let mut remaining = 65536 - 49152; // 剩余尝试分配端口次数 5054ad52e57S裕依2439 let mut port: u16; 5064ad52e57S裕依2439 while remaining > 0 { 5074ad52e57S裕依2439 unsafe { 5084ad52e57S裕依2439 if EPHEMERAL_PORT == 65535 { 5094ad52e57S裕依2439 EPHEMERAL_PORT = 49152; 5104ad52e57S裕依2439 } else { 511b5b571e0SLoGin EPHEMERAL_PORT += 1; 5124ad52e57S裕依2439 } 5134ad52e57S裕依2439 port = EPHEMERAL_PORT; 5144ad52e57S裕依2439 } 5154ad52e57S裕依2439 5164ad52e57S裕依2439 // 使用 ListenTable 检查端口是否被占用 5174ad52e57S裕依2439 let listen_table_guard = match socket_type { 518b5b571e0SLoGin SocketType::Udp => self.udp_port_table.lock(), 519b5b571e0SLoGin SocketType::Tcp => self.tcp_port_table.lock(), 5204ad52e57S裕依2439 _ => panic!("{:?} cann't get a port", socket_type), 5214ad52e57S裕依2439 }; 522b5b571e0SLoGin if listen_table_guard.get(&port).is_none() { 5234ad52e57S裕依2439 drop(listen_table_guard); 5244ad52e57S裕依2439 return Ok(port); 5254ad52e57S裕依2439 } 5264ad52e57S裕依2439 remaining -= 1; 5274ad52e57S裕依2439 } 5284ad52e57S裕依2439 return Err(SystemError::EADDRINUSE); 5294ad52e57S裕依2439 } 5304ad52e57S裕依2439 5314ad52e57S裕依2439 /// @brief 检测给定端口是否已被占用,如果未被占用则在 TCP/UDP 对应的表中记录 5324ad52e57S裕依2439 /// 5334ad52e57S裕依2439 /// TODO: 增加支持端口复用的逻辑 53437cef00bSSamuel Dai pub fn bind_port(&self, socket_type: SocketType, port: u16) -> Result<(), SystemError> { 5354ad52e57S裕依2439 if port > 0 { 5364ad52e57S裕依2439 let mut listen_table_guard = match socket_type { 537b5b571e0SLoGin SocketType::Udp => self.udp_port_table.lock(), 538b5b571e0SLoGin SocketType::Tcp => self.tcp_port_table.lock(), 5394ad52e57S裕依2439 _ => panic!("{:?} cann't bind a port", socket_type), 5404ad52e57S裕依2439 }; 5414ad52e57S裕依2439 match listen_table_guard.get(&port) { 5424ad52e57S裕依2439 Some(_) => return Err(SystemError::EADDRINUSE), 54337cef00bSSamuel Dai None => listen_table_guard.insert(port, ProcessManager::current_pid()), 5444ad52e57S裕依2439 }; 5454ad52e57S裕依2439 drop(listen_table_guard); 5464ad52e57S裕依2439 } 5474ad52e57S裕依2439 return Ok(()); 5484ad52e57S裕依2439 } 5494ad52e57S裕依2439 5504ad52e57S裕依2439 /// @brief 在对应的端口记录表中将端口和 socket 解绑 55137cef00bSSamuel Dai /// should call this function when socket is closed or aborted 55237cef00bSSamuel Dai pub fn unbind_port(&self, socket_type: SocketType, port: u16) { 5534ad52e57S裕依2439 let mut listen_table_guard = match socket_type { 554b5b571e0SLoGin SocketType::Udp => self.udp_port_table.lock(), 555b5b571e0SLoGin SocketType::Tcp => self.tcp_port_table.lock(), 55637cef00bSSamuel Dai _ => { 55737cef00bSSamuel Dai return; 55837cef00bSSamuel Dai } 5594ad52e57S裕依2439 }; 5604ad52e57S裕依2439 listen_table_guard.remove(&port); 5614ad52e57S裕依2439 drop(listen_table_guard); 5624ad52e57S裕依2439 } 5634ad52e57S裕依2439 } 5644ad52e57S裕依2439 5654ad52e57S裕依2439 /// @brief socket的类型 5664ad52e57S裕依2439 #[derive(Debug, Clone, Copy, PartialEq)] 5674ad52e57S裕依2439 pub enum SocketType { 5684ad52e57S裕依2439 /// 原始的socket 569b5b571e0SLoGin Raw, 5704ad52e57S裕依2439 /// 用于Tcp通信的 Socket 571b5b571e0SLoGin Tcp, 5724ad52e57S裕依2439 /// 用于Udp通信的 Socket 573b5b571e0SLoGin Udp, 5746046f775S裕依 /// unix域的 Socket 5756046f775S裕依 Unix, 5764ad52e57S裕依2439 } 5774ad52e57S裕依2439 5784ad52e57S裕依2439 bitflags! { 5794ad52e57S裕依2439 /// @brief socket的选项 5804ad52e57S裕依2439 #[derive(Default)] 5814ad52e57S裕依2439 pub struct SocketOptions: u32 { 5824ad52e57S裕依2439 /// 是否阻塞 5834ad52e57S裕依2439 const BLOCK = 1 << 0; 5844ad52e57S裕依2439 /// 是否允许广播 5854ad52e57S裕依2439 const BROADCAST = 1 << 1; 5864ad52e57S裕依2439 /// 是否允许多播 5874ad52e57S裕依2439 const MULTICAST = 1 << 2; 5884ad52e57S裕依2439 /// 是否允许重用地址 5894ad52e57S裕依2439 const REUSEADDR = 1 << 3; 5904ad52e57S裕依2439 /// 是否允许重用端口 5914ad52e57S裕依2439 const REUSEPORT = 1 << 4; 5924ad52e57S裕依2439 } 5934ad52e57S裕依2439 } 5944ad52e57S裕依2439 5954ad52e57S裕依2439 #[derive(Debug, Clone)] 5964ad52e57S裕依2439 /// @brief 在trait Socket的metadata函数中返回该结构体供外部使用 5974ad52e57S裕依2439 pub struct SocketMetadata { 5984ad52e57S裕依2439 /// socket的类型 5994ad52e57S裕依2439 pub socket_type: SocketType, 6004ad52e57S裕依2439 /// 接收缓冲区的大小 6014ad52e57S裕依2439 pub rx_buf_size: usize, 6024ad52e57S裕依2439 /// 发送缓冲区的大小 6034ad52e57S裕依2439 pub tx_buf_size: usize, 6044ad52e57S裕依2439 /// 元数据的缓冲区的大小 6054ad52e57S裕依2439 pub metadata_buf_size: usize, 6064ad52e57S裕依2439 /// socket的选项 6074ad52e57S裕依2439 pub options: SocketOptions, 6084ad52e57S裕依2439 } 6094ad52e57S裕依2439 6104ad52e57S裕依2439 impl SocketMetadata { 6114ad52e57S裕依2439 fn new( 6124ad52e57S裕依2439 socket_type: SocketType, 6134ad52e57S裕依2439 rx_buf_size: usize, 6144ad52e57S裕依2439 tx_buf_size: usize, 6154ad52e57S裕依2439 metadata_buf_size: usize, 6164ad52e57S裕依2439 options: SocketOptions, 6174ad52e57S裕依2439 ) -> Self { 6184ad52e57S裕依2439 Self { 6194ad52e57S裕依2439 socket_type, 6204ad52e57S裕依2439 rx_buf_size, 6214ad52e57S裕依2439 tx_buf_size, 6224ad52e57S裕依2439 metadata_buf_size, 6234ad52e57S裕依2439 options, 6244ad52e57S裕依2439 } 6254ad52e57S裕依2439 } 6264ad52e57S裕依2439 } 6274ad52e57S裕依2439 6284ad52e57S裕依2439 /// @brief 地址族的枚举 6294ad52e57S裕依2439 /// 6304ad52e57S裕依2439 /// 参考:https://code.dragonos.org.cn/xref/linux-5.19.10/include/linux/socket.h#180 6314ad52e57S裕依2439 #[derive(Debug, Clone, Copy, PartialEq, Eq, FromPrimitive, ToPrimitive)] 6324ad52e57S裕依2439 pub enum AddressFamily { 6334ad52e57S裕依2439 /// AF_UNSPEC 表示地址族未指定 6344ad52e57S裕依2439 Unspecified = 0, 6354ad52e57S裕依2439 /// AF_UNIX 表示Unix域的socket (与AF_LOCAL相同) 6364ad52e57S裕依2439 Unix = 1, 6374ad52e57S裕依2439 /// AF_INET 表示IPv4的socket 6384ad52e57S裕依2439 INet = 2, 6394ad52e57S裕依2439 /// AF_AX25 表示AMPR AX.25的socket 6404ad52e57S裕依2439 AX25 = 3, 6414ad52e57S裕依2439 /// AF_IPX 表示IPX的socket 6424ad52e57S裕依2439 IPX = 4, 6434ad52e57S裕依2439 /// AF_APPLETALK 表示Appletalk的socket 6444ad52e57S裕依2439 Appletalk = 5, 6454ad52e57S裕依2439 /// AF_NETROM 表示AMPR NET/ROM的socket 6464ad52e57S裕依2439 Netrom = 6, 6474ad52e57S裕依2439 /// AF_BRIDGE 表示多协议桥接的socket 6484ad52e57S裕依2439 Bridge = 7, 6494ad52e57S裕依2439 /// AF_ATMPVC 表示ATM PVCs的socket 6504ad52e57S裕依2439 Atmpvc = 8, 6514ad52e57S裕依2439 /// AF_X25 表示X.25的socket 6524ad52e57S裕依2439 X25 = 9, 6534ad52e57S裕依2439 /// AF_INET6 表示IPv6的socket 6544ad52e57S裕依2439 INet6 = 10, 6554ad52e57S裕依2439 /// AF_ROSE 表示AMPR ROSE的socket 6564ad52e57S裕依2439 Rose = 11, 6574ad52e57S裕依2439 /// AF_DECnet Reserved for DECnet project 6584ad52e57S裕依2439 Decnet = 12, 6594ad52e57S裕依2439 /// AF_NETBEUI Reserved for 802.2LLC project 6604ad52e57S裕依2439 Netbeui = 13, 6614ad52e57S裕依2439 /// AF_SECURITY 表示Security callback的伪AF 6624ad52e57S裕依2439 Security = 14, 6634ad52e57S裕依2439 /// AF_KEY 表示Key management API 6644ad52e57S裕依2439 Key = 15, 6654ad52e57S裕依2439 /// AF_NETLINK 表示Netlink的socket 6664ad52e57S裕依2439 Netlink = 16, 6674ad52e57S裕依2439 /// AF_PACKET 表示Low level packet interface 6684ad52e57S裕依2439 Packet = 17, 6694ad52e57S裕依2439 /// AF_ASH 表示Ash 6704ad52e57S裕依2439 Ash = 18, 6714ad52e57S裕依2439 /// AF_ECONET 表示Acorn Econet 6724ad52e57S裕依2439 Econet = 19, 6734ad52e57S裕依2439 /// AF_ATMSVC 表示ATM SVCs 6744ad52e57S裕依2439 Atmsvc = 20, 6754ad52e57S裕依2439 /// AF_RDS 表示Reliable Datagram Sockets 6764ad52e57S裕依2439 Rds = 21, 6774ad52e57S裕依2439 /// AF_SNA 表示Linux SNA Project 6784ad52e57S裕依2439 Sna = 22, 6794ad52e57S裕依2439 /// AF_IRDA 表示IRDA sockets 6804ad52e57S裕依2439 Irda = 23, 6814ad52e57S裕依2439 /// AF_PPPOX 表示PPPoX sockets 6824ad52e57S裕依2439 Pppox = 24, 6834ad52e57S裕依2439 /// AF_WANPIPE 表示WANPIPE API sockets 6844ad52e57S裕依2439 WanPipe = 25, 6854ad52e57S裕依2439 /// AF_LLC 表示Linux LLC 6864ad52e57S裕依2439 Llc = 26, 6874ad52e57S裕依2439 /// AF_IB 表示Native InfiniBand address 6884ad52e57S裕依2439 /// 介绍:https://access.redhat.com/documentation/en-us/red_hat_enterprise_linux/9/html-single/configuring_infiniband_and_rdma_networks/index#understanding-infiniband-and-rdma_configuring-infiniband-and-rdma-networks 6894ad52e57S裕依2439 Ib = 27, 6904ad52e57S裕依2439 /// AF_MPLS 表示MPLS 6914ad52e57S裕依2439 Mpls = 28, 6924ad52e57S裕依2439 /// AF_CAN 表示Controller Area Network 6934ad52e57S裕依2439 Can = 29, 6944ad52e57S裕依2439 /// AF_TIPC 表示TIPC sockets 6954ad52e57S裕依2439 Tipc = 30, 6964ad52e57S裕依2439 /// AF_BLUETOOTH 表示Bluetooth sockets 6974ad52e57S裕依2439 Bluetooth = 31, 6984ad52e57S裕依2439 /// AF_IUCV 表示IUCV sockets 6994ad52e57S裕依2439 Iucv = 32, 7004ad52e57S裕依2439 /// AF_RXRPC 表示RxRPC sockets 7014ad52e57S裕依2439 Rxrpc = 33, 7024ad52e57S裕依2439 /// AF_ISDN 表示mISDN sockets 7034ad52e57S裕依2439 Isdn = 34, 7044ad52e57S裕依2439 /// AF_PHONET 表示Phonet sockets 7054ad52e57S裕依2439 Phonet = 35, 7064ad52e57S裕依2439 /// AF_IEEE802154 表示IEEE 802.15.4 sockets 7074ad52e57S裕依2439 Ieee802154 = 36, 7084ad52e57S裕依2439 /// AF_CAIF 表示CAIF sockets 7094ad52e57S裕依2439 Caif = 37, 7104ad52e57S裕依2439 /// AF_ALG 表示Algorithm sockets 7114ad52e57S裕依2439 Alg = 38, 7124ad52e57S裕依2439 /// AF_NFC 表示NFC sockets 7134ad52e57S裕依2439 Nfc = 39, 7144ad52e57S裕依2439 /// AF_VSOCK 表示vSockets 7154ad52e57S裕依2439 Vsock = 40, 7164ad52e57S裕依2439 /// AF_KCM 表示Kernel Connection Multiplexor 7174ad52e57S裕依2439 Kcm = 41, 7184ad52e57S裕依2439 /// AF_QIPCRTR 表示Qualcomm IPC Router 7194ad52e57S裕依2439 Qipcrtr = 42, 7204ad52e57S裕依2439 /// AF_SMC 表示SMC-R sockets. 7214ad52e57S裕依2439 /// reserve number for PF_SMC protocol family that reuses AF_INET address family 7224ad52e57S裕依2439 Smc = 43, 7234ad52e57S裕依2439 /// AF_XDP 表示XDP sockets 7244ad52e57S裕依2439 Xdp = 44, 7254ad52e57S裕依2439 /// AF_MCTP 表示Management Component Transport Protocol 7264ad52e57S裕依2439 Mctp = 45, 7274ad52e57S裕依2439 /// AF_MAX 表示最大的地址族 7284ad52e57S裕依2439 Max = 46, 7294ad52e57S裕依2439 } 7304ad52e57S裕依2439 7314ad52e57S裕依2439 impl TryFrom<u16> for AddressFamily { 7324ad52e57S裕依2439 type Error = SystemError; 7334ad52e57S裕依2439 fn try_from(x: u16) -> Result<Self, Self::Error> { 7344ad52e57S裕依2439 use num_traits::FromPrimitive; 735b5b571e0SLoGin return <Self as FromPrimitive>::from_u16(x).ok_or(SystemError::EINVAL); 7364ad52e57S裕依2439 } 7374ad52e57S裕依2439 } 7384ad52e57S裕依2439 7394ad52e57S裕依2439 /// @brief posix套接字类型的枚举(这些值与linux内核中的值一致) 7404ad52e57S裕依2439 #[derive(Debug, Clone, Copy, PartialEq, Eq, FromPrimitive, ToPrimitive)] 7414ad52e57S裕依2439 pub enum PosixSocketType { 7424ad52e57S裕依2439 Stream = 1, 7434ad52e57S裕依2439 Datagram = 2, 7444ad52e57S裕依2439 Raw = 3, 7454ad52e57S裕依2439 Rdm = 4, 7464ad52e57S裕依2439 SeqPacket = 5, 7474ad52e57S裕依2439 Dccp = 6, 7484ad52e57S裕依2439 Packet = 10, 7494ad52e57S裕依2439 } 7504ad52e57S裕依2439 7514ad52e57S裕依2439 impl TryFrom<u8> for PosixSocketType { 7524ad52e57S裕依2439 type Error = SystemError; 7534ad52e57S裕依2439 fn try_from(x: u8) -> Result<Self, Self::Error> { 7544ad52e57S裕依2439 use num_traits::FromPrimitive; 755b5b571e0SLoGin return <Self as FromPrimitive>::from_u8(x).ok_or(SystemError::EINVAL); 7564ad52e57S裕依2439 } 7574ad52e57S裕依2439 } 7584ad52e57S裕依2439 7594ad52e57S裕依2439 /// ### 为socket提供无锁的poll方法 7604ad52e57S裕依2439 /// 7614ad52e57S裕依2439 /// 因为在网卡中断中,需要轮询socket的状态,如果使用socket文件或者其inode来poll 7624ad52e57S裕依2439 /// 在当前的设计,会必然死锁,所以引用这一个设计来解决,提供无的poll 7634ad52e57S裕依2439 pub struct SocketPollMethod; 7644ad52e57S裕依2439 7654ad52e57S裕依2439 impl SocketPollMethod { 7664ad52e57S裕依2439 pub fn poll(socket: &socket::Socket, shutdown: ShutdownType) -> EPollEventType { 7674ad52e57S裕依2439 match socket { 7684ad52e57S裕依2439 socket::Socket::Udp(udp) => Self::udp_poll(udp, shutdown), 7694ad52e57S裕依2439 socket::Socket::Tcp(tcp) => Self::tcp_poll(tcp, shutdown), 77093c37970Ssun5etop socket::Socket::Raw(raw) => Self::raw_poll(raw, shutdown), 7714ad52e57S裕依2439 _ => todo!(), 7724ad52e57S裕依2439 } 7734ad52e57S裕依2439 } 7744ad52e57S裕依2439 7754ad52e57S裕依2439 pub fn tcp_poll(socket: &tcp::Socket, shutdown: ShutdownType) -> EPollEventType { 7764ad52e57S裕依2439 let mut events = EPollEventType::empty(); 7774ad52e57S裕依2439 if socket.is_listening() && socket.is_active() { 7784ad52e57S裕依2439 events.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM); 7794ad52e57S裕依2439 return events; 7804ad52e57S裕依2439 } 7814ad52e57S裕依2439 7824ad52e57S裕依2439 // socket已经关闭 7834ad52e57S裕依2439 if !socket.is_open() { 7844ad52e57S裕依2439 events.insert(EPollEventType::EPOLLHUP) 7854ad52e57S裕依2439 } 7864ad52e57S裕依2439 if shutdown.contains(ShutdownType::RCV_SHUTDOWN) { 7874ad52e57S裕依2439 events.insert( 7884ad52e57S裕依2439 EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM | EPollEventType::EPOLLRDHUP, 7894ad52e57S裕依2439 ); 7904ad52e57S裕依2439 } 7914ad52e57S裕依2439 7924ad52e57S裕依2439 let state = socket.state(); 7934ad52e57S裕依2439 if state != tcp::State::SynSent && state != tcp::State::SynReceived { 7944ad52e57S裕依2439 // socket有可读数据 7954ad52e57S裕依2439 if socket.can_recv() { 7964ad52e57S裕依2439 events.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM); 7974ad52e57S裕依2439 } 7984ad52e57S裕依2439 7994ad52e57S裕依2439 if !(shutdown.contains(ShutdownType::SEND_SHUTDOWN)) { 8004ad52e57S裕依2439 // 缓冲区可写 8014ad52e57S裕依2439 if socket.send_queue() < socket.send_capacity() { 8024ad52e57S裕依2439 events.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM); 8034ad52e57S裕依2439 } else { 8044ad52e57S裕依2439 // TODO:触发缓冲区已满的信号 8054ad52e57S裕依2439 todo!("A signal that the buffer is full needs to be sent"); 8064ad52e57S裕依2439 } 8074ad52e57S裕依2439 } else { 8084ad52e57S裕依2439 // 如果我们的socket关闭了SEND_SHUTDOWN,epoll事件就是EPOLLOUT 8094ad52e57S裕依2439 events.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM); 8104ad52e57S裕依2439 } 8114ad52e57S裕依2439 } else if state == tcp::State::SynSent { 8124ad52e57S裕依2439 events.insert(EPollEventType::EPOLLOUT | EPollEventType::EPOLLWRNORM); 8134ad52e57S裕依2439 } 8144ad52e57S裕依2439 8154ad52e57S裕依2439 // socket发生错误 8164ad52e57S裕依2439 if !socket.is_active() { 8174ad52e57S裕依2439 events.insert(EPollEventType::EPOLLERR); 8184ad52e57S裕依2439 } 8194ad52e57S裕依2439 8204ad52e57S裕依2439 events 8214ad52e57S裕依2439 } 8224ad52e57S裕依2439 8234ad52e57S裕依2439 pub fn udp_poll(socket: &udp::Socket, shutdown: ShutdownType) -> EPollEventType { 8244ad52e57S裕依2439 let mut event = EPollEventType::empty(); 8254ad52e57S裕依2439 8264ad52e57S裕依2439 if shutdown.contains(ShutdownType::RCV_SHUTDOWN) { 8274ad52e57S裕依2439 event.insert( 8284ad52e57S裕依2439 EPollEventType::EPOLLRDHUP | EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM, 8294ad52e57S裕依2439 ); 8304ad52e57S裕依2439 } 8314ad52e57S裕依2439 if shutdown.contains(ShutdownType::SHUTDOWN_MASK) { 8324ad52e57S裕依2439 event.insert(EPollEventType::EPOLLHUP); 8334ad52e57S裕依2439 } 8344ad52e57S裕依2439 8354ad52e57S裕依2439 if socket.can_recv() { 8364ad52e57S裕依2439 event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM); 8374ad52e57S裕依2439 } 8384ad52e57S裕依2439 8394ad52e57S裕依2439 if socket.can_send() { 8404ad52e57S裕依2439 event.insert( 8414ad52e57S裕依2439 EPollEventType::EPOLLOUT 8424ad52e57S裕依2439 | EPollEventType::EPOLLWRNORM 8434ad52e57S裕依2439 | EPollEventType::EPOLLWRBAND, 8444ad52e57S裕依2439 ); 8454ad52e57S裕依2439 } else { 8464ad52e57S裕依2439 // TODO: 缓冲区空间不够,需要使用信号处理 8474ad52e57S裕依2439 todo!() 8484ad52e57S裕依2439 } 8494ad52e57S裕依2439 8504ad52e57S裕依2439 return event; 8514ad52e57S裕依2439 } 85293c37970Ssun5etop 85393c37970Ssun5etop pub fn raw_poll(socket: &raw::Socket, shutdown: ShutdownType) -> EPollEventType { 854*2eab6dd7S曾俊 //debug!("enter raw_poll!"); 85593c37970Ssun5etop let mut event = EPollEventType::empty(); 85693c37970Ssun5etop 85793c37970Ssun5etop if shutdown.contains(ShutdownType::RCV_SHUTDOWN) { 85893c37970Ssun5etop event.insert( 85993c37970Ssun5etop EPollEventType::EPOLLRDHUP | EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM, 86093c37970Ssun5etop ); 86193c37970Ssun5etop } 86293c37970Ssun5etop if shutdown.contains(ShutdownType::SHUTDOWN_MASK) { 86393c37970Ssun5etop event.insert(EPollEventType::EPOLLHUP); 86493c37970Ssun5etop } 86593c37970Ssun5etop 86693c37970Ssun5etop if socket.can_recv() { 867*2eab6dd7S曾俊 //debug!("poll can recv!"); 86893c37970Ssun5etop event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM); 86993c37970Ssun5etop } else { 870*2eab6dd7S曾俊 //debug!("poll can not recv!"); 87193c37970Ssun5etop } 87293c37970Ssun5etop 87393c37970Ssun5etop if socket.can_send() { 874*2eab6dd7S曾俊 //debug!("poll can send!"); 87593c37970Ssun5etop event.insert( 87693c37970Ssun5etop EPollEventType::EPOLLOUT 87793c37970Ssun5etop | EPollEventType::EPOLLWRNORM 87893c37970Ssun5etop | EPollEventType::EPOLLWRBAND, 87993c37970Ssun5etop ); 88093c37970Ssun5etop } else { 881*2eab6dd7S曾俊 //debug!("poll can not send!"); 88293c37970Ssun5etop // TODO: 缓冲区空间不够,需要使用信号处理 88393c37970Ssun5etop todo!() 88493c37970Ssun5etop } 88593c37970Ssun5etop return event; 88693c37970Ssun5etop } 8874ad52e57S裕依2439 } 888