Samuka007/asterinas
1
0
Fork 0
mirror of https://github.com/asterinas/asterinas.git synced 2025-06-10 13:56:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add user preemption functionality

Browse Source
This commit is contained in:
徐启航 2024-04-22 13:29:32 +00:00 committed by Tate, Hongliang Tian
parent efa75a2f31
commit 3b47143875
1 changed files with 26 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
framework/aster-frame/src/arch/x86/cpu.rs
View File

@ -212,6 +212,28 @@ impl TdxTrapFrame for GeneralRegs {
}
}
pub struct UserPreemption {
count: u32,
}
impl UserPreemption {
const PREEMPTION_INTERVAL: u32 = 100;
pub const fn new() -> Self {
UserPreemption { count: 0 }
}
pub fn might_preempt(&mut self) {
self.count = (self.count + 1) % Self::PREEMPTION_INTERVAL;
if self.count == 0 {
crate::arch::irq::enable_local();
crate::task::schedule();
crate::arch::irq::disable_local();
}
}
}
impl UserContext {
pub fn general_regs(&self) -> &GeneralRegs {
&self.user_context.general
@ -241,6 +263,8 @@ impl UserContextApiInternal for UserContext {
self.user_context.general.rflags |= (RFlags::INTERRUPT_FLAG | RFlags::ID).bits() as usize;
const SYSCALL_TRAPNUM: u16 = 0x100;
let mut user_preemption = UserPreemption::new();
// return when it is syscall or cpu exception type is Fault or Trap.
loop {
self.user_context.run();
@ -267,6 +291,8 @@ impl UserContextApiInternal for UserContext {
}
};
call_irq_callback_functions(&self.as_trap_frame());
user_preemption.might_preempt();
}
crate::arch::irq::enable_local();