Kernel: Stop idling after handling an IRQ

If we receive an IRQ while the idle task is running, prevent it from
re-halting the CPU after the IRQ handler returns.

Instead have the idle task yield to the scheduler, so we can see if
the IRQ has unblocked something.

1 files changed, 21 insertions, 0 deletions

diff --git a/Kernel/Scheduler.cpp b/Kernel/Scheduler.cpp
index ed5ce0eb0c..310f9c2d14 100644
--- a/Kernel/Scheduler.cpp
+++ b/Kernel/Scheduler.cpp
@@ -575,3 +575,24 @@ void Scheduler::timer_tick(RegisterDump& regs)
         "orl $0x00004000, (%esp)\n"
         "popf\n");
 }
+
+static bool s_should_stop_idling = false;
+
+void Scheduler::stop_idling()
+{
+    if (current != &s_colonel_process->main_thread())
+        return;
+
+    s_should_stop_idling = true;
+}
+
+void Scheduler::idle_loop()
+{
+    for (;;) {
+        asm("hlt");
+        if (s_should_stop_idling) {
+            s_should_stop_idling = false;
+            yield();
+        }
+    }
+}