fix(scheduler): awaiting tasks

* Correctly set a sleep duration of 0 for tasks that do not have a
  deadline.
* Look in the `deadline_awaiting` queue if the task trying to wake up
  has a non-zero duration.

Signed-off-by: Anhad Singh <[email protected]>

Author: Andy-Python-Programmer

src/aero_kernel/src/userland/scheduler/round_robin.rs

@@ -90,6 +90,8 @@ impl TaskQueue {
         debug_assert!(!task.link.is_linked()); // Make sure the task is not already linked
 
         task.update_state(TaskState::AwaitingIo);
+        task.set_sleep_duration(0);
+
         self.awaiting.push_back(task);
     }
 }
@@ -213,7 +215,11 @@ impl SchedulerInterface for RoundRobin {
         let queue = self.queue.get_mut();
 
         if task.state() == TaskState::AwaitingIo {
-            let mut cursor = unsafe { queue.awaiting.cursor_mut_from_ptr(task.as_ref()) };
+            let mut cursor = if task.load_sleep_duration() > 0 {
+                unsafe { queue.deadline_awaiting.cursor_mut_from_ptr(task.as_ref()) }
+            } else {
+                unsafe { queue.awaiting.cursor_mut_from_ptr(task.as_ref()) }
+            };
 
             if let Some(task) = cursor.remove() {
                 queue.push_runnable(task);

src/aero_kernel/src/utils/sync.rs

@@ -178,7 +178,7 @@ pub struct BMutexGuard<'a, T: ?Sized + 'a> {
     mutex: &'a BMutex<T>,
 }
 
-impl<'a, T: ?Sized> core::ops::Deref for BMutexGuard<'a, T> {
+impl<T: ?Sized> core::ops::Deref for BMutexGuard<'_, T> {
     type Target = T;
 
     #[inline]
@@ -187,14 +187,14 @@ impl<'a, T: ?Sized> core::ops::Deref for BMutexGuard<'a, T> {
     }
 }
 
-impl<'a, T: ?Sized> core::ops::DerefMut for BMutexGuard<'a, T> {
+impl<T: ?Sized> core::ops::DerefMut for BMutexGuard<'_, T> {
     #[inline]
     fn deref_mut(&mut self) -> &mut T {
         self.guard.deref_mut()
     }
 }
 
-impl<'a, T: ?Sized> Drop for BMutexGuard<'a, T> {
+impl<T: ?Sized> Drop for BMutexGuard<'_, T> {
     fn drop(&mut self) {
         self.mutex.wq.notify();
     }
@@ -261,7 +261,7 @@ pub struct MutexGuard<'a, T: ?Sized + 'a> {
     irq_lock: bool,
 }
 
-impl<'a, T: ?Sized> core::ops::Deref for MutexGuard<'a, T> {
+impl<T: ?Sized> core::ops::Deref for MutexGuard<'_, T> {
     type Target = T;
 
     #[inline]
@@ -270,14 +270,14 @@ impl<'a, T: ?Sized> core::ops::Deref for MutexGuard<'a, T> {
     }
 }
 
-impl<'a, T: ?Sized> core::ops::DerefMut for MutexGuard<'a, T> {
+impl<T: ?Sized> core::ops::DerefMut for MutexGuard<'_, T> {
     #[inline]
     fn deref_mut(&mut self) -> &mut T {
         &mut self.guard
     }
 }
 
-impl<'a, T: ?Sized> Drop for MutexGuard<'a, T> {
+impl<T: ?Sized> Drop for MutexGuard<'_, T> {
     #[inline]
     fn drop(&mut self) {
         unsafe {