From 993b7495baa2f3cc89eb2af1bba5537449efa82e Mon Sep 17 00:00:00 2001
From: Timon Kruiper <timonkruiper@gmail.com>
Date: Thu, 22 Dec 2022 13:34:58 +0100
Subject: [PATCH] Kernel/aarch64: Implement Processor::check_invoke_scheduler()

---
 Kernel/Arch/aarch64/Processor.cpp | 10 +++++++++-
 Kernel/Arch/aarch64/Processor.h   |  1 +
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/Kernel/Arch/aarch64/Processor.cpp b/Kernel/Arch/aarch64/Processor.cpp
index 23a0cf6f08..4639717317 100644
--- a/Kernel/Arch/aarch64/Processor.cpp
+++ b/Kernel/Arch/aarch64/Processor.cpp
@@ -11,6 +11,7 @@
 #include <Kernel/Arch/TrapFrame.h>
 #include <Kernel/Arch/aarch64/ASM_wrapper.h>
 #include <Kernel/Arch/aarch64/CPU.h>
+#include <Kernel/Scheduler.h>
 #include <Kernel/Thread.h>
 #include <Kernel/Time/TimeManagement.h>
 
@@ -171,7 +172,14 @@ ErrorOr<Vector<FlatPtr, 32>> Processor::capture_stack_trace(Thread& thread, size
 
 void Processor::check_invoke_scheduler()
 {
-    TODO_AARCH64();
+    VERIFY_INTERRUPTS_DISABLED();
+    VERIFY(!m_in_irq);
+    VERIFY(!m_in_critical);
+    VERIFY(&Processor::current() == this);
+    if (m_invoke_scheduler_async && m_scheduler_initialized) {
+        m_invoke_scheduler_async = false;
+        Scheduler::invoke_async();
+    }
 }
 
 }
diff --git a/Kernel/Arch/aarch64/Processor.h b/Kernel/Arch/aarch64/Processor.h
index 1603196ccf..26940b1d6e 100644
--- a/Kernel/Arch/aarch64/Processor.h
+++ b/Kernel/Arch/aarch64/Processor.h
@@ -279,6 +279,7 @@ private:
     FlatPtr m_in_irq { 0 };
     bool m_in_scheduler { false };
     bool m_invoke_scheduler_async { false };
+    bool m_scheduler_initialized { false };
 };
 
 }