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

#![no_std]

use preemption::{hold_preemption, PreemptionGuard};

pub type Mutex<T> = sync::Mutex<T, DisablePreemption>;
pub type MutexGuard<'a, T> = sync::MutexGuard<'a, T, DisablePreemption>;

pub type PreemptionSafeMutex<T> = Mutex<T>;
pub type PreemptionSafeMutexGuard<'a, T> = MutexGuard<'a, T>;

pub type RwLock<T> = sync::RwLock<T, DisablePreemption>;
pub type RwLockReadGuard<'a, T> = sync::RwLockReadGuard<'a, T, DisablePreemption>;
pub type RwLockWriteGuard<'a, T> = sync::RwLockWriteGuard<'a, T, DisablePreemption>;

pub type PreemptionSafeRwLock<T> = RwLock<T>;
pub type PreemptionSafeRwLockReadGuard<'a, T> = RwLockReadGuard<'a, T>;
pub type PreemptionSafeRwLockWriteGuard<'a, T> = RwLockWriteGuard<'a, T>;

/// A deadlock prevention method that disables preemption.
#[derive(Copy, Clone, Debug, Default, Eq, Hash, Ord, PartialEq, PartialOrd)]
pub struct DisablePreemption {}

impl sync::DeadlockPrevention for DisablePreemption {
    type Guard = PreemptionGuard;

    const EXPENSIVE: bool = true;

    #[inline]
    fn enter() -> Self::Guard {
        hold_preemption()
    }
}