Rativel/BurritOS
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

♻️ Refactored lock_acquire

Browse Source
This commit is contained in:
François Autin 2023-04-19 23:13:27 +02:00
parent f55189f1fe
commit 5734e02b30
No known key found for this signature in database
GPG Key ID: 343F5D382E1DD77C
1 changed files with 17 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
src/kernel/thread_manager.rs
View File

@ -257,29 +257,24 @@ impl ThreadManager {
/// - **id** id of the lock, stored in [`ObjAddr`], id given by user program thought exceptions
/// - **machine** the machine where the threads are executed
pub fn lock_acquire(&mut self, id: i32, machine: &mut Machine) -> Result<MachineOk, MachineError> {
match self.get_g_current_thread() {
Some(thread) => {
let rc_thread = Rc::clone(thread);
let lock = self.get_obj_addrs().search_lock(id);
match lock {
Some(lock) => {
let old_status = machine.interrupt.set_status(InterruptStatus::InterruptOff);
if lock.free {
lock.free = false;
lock.owner = Some(Rc::clone(&rc_thread))
} else {
let thread = Rc::clone(&rc_thread);
lock.waiting_queue.push(Rc::clone(&thread));
self.thread_sleep(machine, Rc::clone(&thread));
}
machine.interrupt.set_status(old_status);
Ok(MachineOk::Ok)
},
None => Err("Cannot find lock")?
}
},
None => unreachable!("Current thread should not be None")
let current_thread = match self.get_g_current_thread() {
Some(thread) => Rc::clone(thread),
None => Err("lock_acquire error: current_thread should not be None.")?
};
let old_status = machine.interrupt.set_status(InterruptStatus::InterruptOff);
if let Some(lock) = self.get_obj_addrs().search_lock(id) {
if lock.free {
lock.free = false;
lock.owner = Some(current_thread)
} else {
lock.waiting_queue.push(current_thread.clone());
self.thread_sleep(machine, current_thread);
}
} else {
Err("lock_acquire error: cannot find Lock.")?
}
machine.interrupt.set_status(old_status);
Ok(MachineOk::Ok)
}
/// Wake up a waiter if necessary, or release it if no thread is waiting.