Current process pointer getter

djphoenix · Apr 24, 2017 · 5c31e0b · 5c31e0b
1 parent 8383b61
commit 5c31e0b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/src/process/include/processmanager.hpp b/src/process/include/processmanager.hpp
@@ -45,6 +45,7 @@ class ProcessManager {
   uint64_t RegisterProcess(Process *process);
   void queueThread(Process *process, Thread *thread);
   void dequeueThread(Thread *thread);
+  Process *currentProcess();
   static ProcessManager* getManager();
   static void NORETURN process_loop();
 };
diff --git a/src/process/processmanager.cpp b/src/process/processmanager.cpp
@@ -194,3 +194,13 @@ void ProcessManager::dequeueThread(Thread *thread) {
   processSwitchMutex.release();
   LeaveCritical(t);
 }
+
+Process *ProcessManager::currentProcess() {
+  uint64_t cpuid = ACPI::getController()->getCPUID();
+  uint64_t t = EnterCritical();
+  processSwitchMutex.lock();
+  QueuedThread *curr = cpuThreads[cpuid];
+  processSwitchMutex.release();
+  LeaveCritical(t);
+  return curr ? curr->process : 0;
+}