DragonOS-Community/DragonOS
1
1
Fork 0
mirror of https://github.com/DragonOS-Community/DragonOS.git synced 2025-06-08 18:26:48 +00:00
Code Issues Releases Wiki Activity

feat(net): 实现raw socket的poll (#739)

Browse Source 
feat(net): 实现raw socket的poll
This commit is contained in:
sun5etop 2024-04-21 13:36:44 +08:00 committed by GitHub
parent b502fbf0b9
commit 93c379703e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 37 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
kernel/src/net/socket/mod.rs
View File

@ -10,7 +10,7 @@ use alloc::{
use hashbrown::HashMap;
use smoltcp::{
iface::SocketSet,
socket::{self, tcp, udp},
socket::{self, raw, tcp, udp},
};
use system_error::SystemError;
@ -756,6 +756,7 @@ impl SocketPollMethod {
match socket {
socket::Socket::Udp(udp) => Self::udp_poll(udp, shutdown),
socket::Socket::Tcp(tcp) => Self::tcp_poll(tcp, shutdown),
socket::Socket::Raw(raw) => Self::raw_poll(raw, shutdown),
_ => todo!(),
}
}
@ -837,4 +838,39 @@ impl SocketPollMethod {
return event;
}
pub fn raw_poll(socket: &raw::Socket, shutdown: ShutdownType) -> EPollEventType {
//kdebug!("enter raw_poll!");
let mut event = EPollEventType::empty();
if shutdown.contains(ShutdownType::RCV_SHUTDOWN) {
event.insert(
EPollEventType::EPOLLRDHUP | EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM,
);
}
if shutdown.contains(ShutdownType::SHUTDOWN_MASK) {
event.insert(EPollEventType::EPOLLHUP);
}
if socket.can_recv() {
//kdebug!("poll can recv!");
event.insert(EPollEventType::EPOLLIN | EPollEventType::EPOLLRDNORM);
} else {
//kdebug!("poll can not recv!");
}
if socket.can_send() {
//kdebug!("poll can send!");
event.insert(
EPollEventType::EPOLLOUT
| EPollEventType::EPOLLWRNORM
| EPollEventType::EPOLLWRBAND,
);
} else {
//kdebug!("poll can not send!");
// TODO: 缓冲区空间不够，需要使用信号处理
todo!()
}
return event;
}
}