general/occlum
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix the return IoEvents of channel poll

Browse Source
This commit is contained in:
He Sun 2020-12-30 20:27:41 +08:00 committed by Tate, Hongliang Tian
parent 3101d8fa88
commit 0b829ca087
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
src/libos/src/fs/channel.rs
View File

@ -247,17 +247,14 @@ impl<I> Producer<I> {
let writable = {
let mut rb_producer = self.inner.lock().unwrap();
!rb_producer.is_full()
!rb_producer.is_full() || self.is_self_shutdown() || self.is_peer_shutdown()
};
if writable {
events |= IoEvents::OUT;
}
if self.is_self_shutdown() {
events |= IoEvents::HUP;
}
if self.is_peer_shutdown() {
events |= IoEvents::RDHUP;
events |= IoEvents::ERR;
}
events
@ -376,15 +373,12 @@ impl<I> Consumer<I> {
let readable = {
let mut rb_consumer = self.inner.lock().unwrap();
!rb_consumer.is_empty()
!rb_consumer.is_empty() || self.is_self_shutdown() || self.is_peer_shutdown()
};
if readable {
events |= IoEvents::IN;
}
if self.is_self_shutdown() {
events |= IoEvents::RDHUP;
}
if self.is_peer_shutdown() {
events |= IoEvents::HUP;
}
@ -417,7 +411,11 @@ impl<I> Consumer<I> {
}
pub fn items_to_consume(&self) -> usize {
self.inner.lock().unwrap().len()
if self.is_self_shutdown() {
0
} else {
self.inner.lock().unwrap().len()
}
}
pub fn capacity(&self) -> usize {