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

Fix the dequeuing order of items in a FutexBucket

Browse Source
This commit is contained in:
He Sun 2020-08-11 20:30:09 +08:00 committed by Tate, Hongliang Tian
parent 6d39587c40
commit b04aa2d7ea
1 changed files with 26 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/libos/src/process/do_futex.rs
View File

@ -75,7 +75,7 @@ pub fn futex_wait(
futex_val: i32,
timeout: &Option<timespec_t>,
) -> Result<()> {
info!(
debug!(
"futex_wait addr: {:#x}, val: {}, timeout: {:?}",
futex_addr as usize, futex_val, timeout
);
@ -130,6 +130,11 @@ pub fn futex_wait(
/// Do futex wake
pub fn futex_wake(futex_addr: *const i32, max_count: usize) -> Result<usize> {
debug!(
"futex_wake addr: {:#x}, max_count: {}",
futex_addr as usize, max_count
);
// Get and lock the futex bucket
let futex_key = FutexKey::new(futex_addr);
let (_, futex_bucket_ref) = FUTEX_BUCKETS.get_bucket(futex_key);
@ -266,29 +271,29 @@ impl FutexBucket {
self.queue.push_back(item);
}
// TODO: this is an O(N) operation. Try to make it more efficient
pub fn dequeue_item(&mut self, futex_item: &FutexItem) -> Option<FutexItem> {
let item_i = self.queue.iter().position(|item| *item == *futex_item);
if item_i.is_none() {
return None;
}
self.queue.swap_remove_back(item_i.unwrap())
self.queue.remove(item_i.unwrap())
}
// TODO: consider using std::future to improve the readability
pub fn dequeue_and_wake_items(&mut self, key: FutexKey, max_count: usize) -> usize {
let mut count = 0;
let mut idx = 0;
let mut items_to_wake = Vec::new();
while count < max_count && idx < self.queue.len() {
if key == self.queue[idx].key {
if let Some(item) = self.queue.swap_remove_back(idx) {
items_to_wake.push(item);
count += 1;
}
self.queue.retain(|item| {
if count >= max_count || key != item.key {
true
} else {
idx += 1;
items_to_wake.push(item.clone());
count += 1;
false
}
}
});
FutexItem::batch_wake(&items_to_wake);
count
@ -315,18 +320,18 @@ impl FutexBucket {
max_nrequeues: usize,
) -> () {
let mut count = 0;
let mut idx = 0;
while count < max_nrequeues && idx < self.queue.len() {
if key == self.queue[idx].key {
if let Some(mut item) = self.queue.swap_remove_back(idx) {
item.key = new_key;
another.enqueue_item(item);
count += 1;
}
self.queue.retain(|item| {
if count >= max_nrequeues || key != item.key {
true
} else {
idx += 1;
let mut new_item = item.clone();
new_item.key = new_key;
another.enqueue_item(new_item);
count += 1;
false
}
}
});
}
}