| 1 | #[ cfg(not(all(test, loom)))] | 
| 2 | pub(crate) mod sync { | 
|---|
| 3 | pub(crate) mod atomic { | 
|---|
| 4 | pub(crate) use core::sync::atomic::{AtomicPtr, AtomicUsize, Ordering}; | 
|---|
| 5 |  | 
|---|
| 6 | pub(crate) trait AtomicMut<T> { | 
|---|
| 7 | fn with_mut<F, R>(&mut self, f: F) -> R | 
|---|
| 8 | where | 
|---|
| 9 | F: FnOnce(&mut *mut T) -> R; | 
|---|
| 10 | } | 
|---|
| 11 |  | 
|---|
| 12 | impl<T> AtomicMut<T> for AtomicPtr<T> { | 
|---|
| 13 | fn with_mut<F, R>(&mut self, f: F) -> R | 
|---|
| 14 | where | 
|---|
| 15 | F: FnOnce(&mut *mut T) -> R, | 
|---|
| 16 | { | 
|---|
| 17 | f(self.get_mut()) | 
|---|
| 18 | } | 
|---|
| 19 | } | 
|---|
| 20 | } | 
|---|
| 21 | } | 
|---|
| 22 |  | 
|---|
| 23 | #[ cfg(all(test, loom))] | 
|---|
| 24 | pub(crate) mod sync { | 
|---|
| 25 | pub(crate) mod atomic { | 
|---|
| 26 | pub(crate) use loom::sync::atomic::{AtomicPtr, AtomicUsize, Ordering}; | 
|---|
| 27 |  | 
|---|
| 28 | pub(crate) trait AtomicMut<T> {} | 
|---|
| 29 | } | 
|---|
| 30 | } | 
|---|
| 31 |  | 
|---|