ax-task 0.7.1

OS-independent IRQ-safe SMP task scheduling core
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

//! Current-task access to task-owned lockdep state.

use crate::sync::{
    IrqSaveGuard,
    lockdep::{HeldLock, HeldLockSnapshot, HeldLockStack},
};

fn with_current_held_locks(operation: impl FnOnce(&mut HeldLockStack)) {
    let _irq_guard = IrqSaveGuard::new();
    let Ok(current) = crate::thread::current::current_thread_handle() else {
        return;
    };
    // SAFETY: local IRQ exclusion prevents migration and scheduler replacement
    // of `current` until the complete task-owned stack operation has finished.
    unsafe { current.core.with_held_locks(operation) };
}

/// Copies the calling task's held-lock stack into a lockdep snapshot.
#[doc(hidden)]
pub(crate) fn collect_current_task_held_locks(snapshot: &mut HeldLockSnapshot) {
    with_current_held_locks(|stack| snapshot.extend(stack));
}

/// Records one lock acquisition in the calling task's lockdep state.
#[doc(hidden)]
pub(crate) fn push_current_task_held_lock(held: HeldLock) {
    with_current_held_locks(|stack| stack.push(held));
}

/// Records one lock release in the calling task's lockdep state.
#[doc(hidden)]
pub(crate) fn pop_current_task_held_lock(lock_addr: usize) {
    with_current_held_locks(|stack| stack.pop_checked(lock_addr));
}