mirror of
https://github.com/asterinas/asterinas.git
synced 2025-06-28 03:43:23 +00:00
38 lines
1.0 KiB
Rust
38 lines
1.0 KiB
Rust
// SPDX-License-Identifier: MPL-2.0
|
|
|
|
pub(super) mod cpu_local;
|
|
mod guard;
|
|
|
|
pub use self::guard::{disable_preempt, DisabledPreemptGuard};
|
|
|
|
/// Halts the CPU until interrupts if no preemption is required.
|
|
///
|
|
/// This function will return if:
|
|
/// - preemption is required when calling this function,
|
|
/// - preemption is required during halting the CPU, or
|
|
/// - interrupts occur during halting the CPU.
|
|
///
|
|
/// This function will perform preemption before returning if
|
|
/// preemption is required.
|
|
///
|
|
/// # Panics
|
|
///
|
|
/// This function will panic if it is called in the atomic mode
|
|
/// ([`crate::task::atomic_mode`]).
|
|
#[track_caller]
|
|
pub fn halt_cpu() {
|
|
crate::task::atomic_mode::might_sleep();
|
|
|
|
let irq_guard = crate::trap::irq::disable_local();
|
|
|
|
if cpu_local::need_preempt() {
|
|
drop(irq_guard);
|
|
} else {
|
|
core::mem::forget(irq_guard);
|
|
// IRQs were previously enabled (checked by `might_sleep`). So we can re-enable them now.
|
|
crate::arch::irq::enable_local_and_halt();
|
|
}
|
|
|
|
super::scheduler::might_preempt();
|
|
}
|