drew/acadia
1
0
Fork 0
Code Issues 6 Pull requests Projects Releases Packages Wiki Activity Actions

[Zion] Add a framework for better process exit.

Browse source
This commit is contained in:
Drew Galbraith 2023-11-16 23:03:27 -08:00
parent aa2d80b557
commit 308dd6a203
9 changed files with 80 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

18
zion/scheduler/scheduler.cpp
View file

@ -43,6 +43,8 @@ void Scheduler::Preempt() {
return;
}
ClearDeadThreadsFromFront();
asm volatile("cli");
if (current_thread_ == sleep_thread_) {
// Sleep should never be preempted. (We should yield it if another thread
@ -66,9 +68,14 @@ void Scheduler::Preempt() {
void Scheduler::Yield() {
if (!enabled_) {
// This is expected to fire once at the start when we enqueue the first
// thread before the scheduler is enabled. Maybe we should get rid of it?
dbgln("WARN Scheduler skipped yield.");
return;
}
ClearDeadThreadsFromFront();
asm volatile("cli");
glcr::RefPtr<Thread> prev = current_thread_;
@ -78,7 +85,9 @@ void Scheduler::Yield() {
return;
} else {
current_thread_ = runnable_threads_.PopFront();
prev->SetState(Thread::RUNNABLE);
if (!prev->IsDying()) {
prev->SetState(Thread::RUNNABLE);
}
}
} else {
if (runnable_threads_.size() == 0) {
@ -90,3 +99,10 @@ void Scheduler::Yield() {
SwapToCurrent(*prev);
}
void Scheduler::ClearDeadThreadsFromFront() {
while (runnable_threads_.size() > 0 &&
runnable_threads_.PeekFront()->IsDying()) {
runnable_threads_.PopFront();
}
}