use core::sync::atomic::Ordering::SeqCst; use core::sync::atomic::*; #[test] fn bool_() { let a = AtomicBool::new(false); assert_eq!(a.compare_exchange(false, true, SeqCst, SeqCst), Ok(false)); assert_eq!(a.compare_exchange(false, true, SeqCst, SeqCst), Err(true)); a.store(false, SeqCst); assert_eq!(a.compare_exchange(false, true, SeqCst, SeqCst), Ok(false)); } #[test] #[should_panic = "there is no such thing as an acquire store"] fn store_illegal_rt_store_acquire_ordering() { let a = AtomicBool::new(false); let ord = Ordering::Acquire; a.store(true, ord); } #[test] #[should_panic = "there is no such thing as an acquire-release store"] fn store_illegal_rt_store_acq_rel_ordering() { let a = AtomicBool::new(false); let ord = Ordering::AcqRel; a.store(true, ord); } #[test] #[should_panic = "there is no such thing as a release load"] fn store_illegal_rt_load_release_ordering() { let a = AtomicBool::new(false); let ord = Ordering::Release; a.load(ord); } #[test] #[should_panic = "there is no such thing as an acquire-release load"] fn store_illegal_rt_load_acq_rel_ordering() { let a = AtomicBool::new(false); let ord = Ordering::AcqRel; a.load(ord); } #[test] fn bool_and() { let a = AtomicBool::new(true); assert_eq!(a.fetch_and(false, SeqCst), true); assert_eq!(a.load(SeqCst), false); } #[test] fn bool_nand() { let a = AtomicBool::new(false); assert_eq!(a.fetch_nand(false, SeqCst), false); assert_eq!(a.load(SeqCst), true); assert_eq!(a.fetch_nand(false, SeqCst), true); assert_eq!(a.load(SeqCst), true); assert_eq!(a.fetch_nand(true, SeqCst), true); assert_eq!(a.load(SeqCst), false); assert_eq!(a.fetch_nand(true, SeqCst), false); assert_eq!(a.load(SeqCst), true); } #[test] fn uint_and() { let x = AtomicUsize::new(0xf731); assert_eq!(x.fetch_and(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 & 0x137f); } #[test] fn uint_nand() { let x = AtomicUsize::new(0xf731); assert_eq!(x.fetch_nand(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), !(0xf731 & 0x137f)); } #[test] fn uint_or() { let x = AtomicUsize::new(0xf731); assert_eq!(x.fetch_or(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 | 0x137f); } #[test] fn uint_xor() { let x = AtomicUsize::new(0xf731); assert_eq!(x.fetch_xor(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 ^ 0x137f); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn uint_min() { let x = AtomicUsize::new(0xf731); assert_eq!(x.fetch_min(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0x137f); assert_eq!(x.fetch_min(0xf731, SeqCst), 0x137f); assert_eq!(x.load(SeqCst), 0x137f); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn uint_max() { let x = AtomicUsize::new(0x137f); assert_eq!(x.fetch_max(0xf731, SeqCst), 0x137f); assert_eq!(x.load(SeqCst), 0xf731); assert_eq!(x.fetch_max(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731); } #[test] fn int_and() { let x = AtomicIsize::new(0xf731); assert_eq!(x.fetch_and(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 & 0x137f); } #[test] fn int_nand() { let x = AtomicIsize::new(0xf731); assert_eq!(x.fetch_nand(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), !(0xf731 & 0x137f)); } #[test] fn int_or() { let x = AtomicIsize::new(0xf731); assert_eq!(x.fetch_or(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 | 0x137f); } #[test] fn int_xor() { let x = AtomicIsize::new(0xf731); assert_eq!(x.fetch_xor(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731 ^ 0x137f); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn int_min() { let x = AtomicIsize::new(0xf731); assert_eq!(x.fetch_min(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0x137f); assert_eq!(x.fetch_min(0xf731, SeqCst), 0x137f); assert_eq!(x.load(SeqCst), 0x137f); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn int_max() { let x = AtomicIsize::new(0x137f); assert_eq!(x.fetch_max(0xf731, SeqCst), 0x137f); assert_eq!(x.load(SeqCst), 0xf731); assert_eq!(x.fetch_max(0x137f, SeqCst), 0xf731); assert_eq!(x.load(SeqCst), 0xf731); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn ptr_add_null() { let atom = AtomicPtr::::new(core::ptr::null_mut()); assert_eq!(atom.fetch_ptr_add(1, SeqCst).addr(), 0); assert_eq!(atom.load(SeqCst).addr(), 8); assert_eq!(atom.fetch_byte_add(1, SeqCst).addr(), 8); assert_eq!(atom.load(SeqCst).addr(), 9); assert_eq!(atom.fetch_ptr_sub(1, SeqCst).addr(), 9); assert_eq!(atom.load(SeqCst).addr(), 1); assert_eq!(atom.fetch_byte_sub(1, SeqCst).addr(), 1); assert_eq!(atom.load(SeqCst).addr(), 0); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn ptr_add_data() { let num = 0i64; let n = &num as *const i64 as *mut _; let atom = AtomicPtr::::new(n); assert_eq!(atom.fetch_ptr_add(1, SeqCst), n); assert_eq!(atom.load(SeqCst), n.wrapping_add(1)); assert_eq!(atom.fetch_ptr_sub(1, SeqCst), n.wrapping_add(1)); assert_eq!(atom.load(SeqCst), n); let bytes_from_n = |b| n.wrapping_byte_add(b); assert_eq!(atom.fetch_byte_add(1, SeqCst), n); assert_eq!(atom.load(SeqCst), bytes_from_n(1)); assert_eq!(atom.fetch_byte_add(5, SeqCst), bytes_from_n(1)); assert_eq!(atom.load(SeqCst), bytes_from_n(6)); assert_eq!(atom.fetch_byte_sub(1, SeqCst), bytes_from_n(6)); assert_eq!(atom.load(SeqCst), bytes_from_n(5)); assert_eq!(atom.fetch_byte_sub(5, SeqCst), bytes_from_n(5)); assert_eq!(atom.load(SeqCst), n); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn ptr_bitops() { let atom = AtomicPtr::::new(core::ptr::null_mut()); assert_eq!(atom.fetch_or(0b0111, SeqCst).addr(), 0); assert_eq!(atom.load(SeqCst).addr(), 0b0111); assert_eq!(atom.fetch_and(0b1101, SeqCst).addr(), 0b0111); assert_eq!(atom.load(SeqCst).addr(), 0b0101); assert_eq!(atom.fetch_xor(0b1111, SeqCst).addr(), 0b0101); assert_eq!(atom.load(SeqCst).addr(), 0b1010); } #[test] #[cfg(any(not(target_arch = "arm"), target_os = "linux"))] // Missing intrinsic in compiler-builtins fn ptr_bitops_tagging() { #[repr(align(16))] struct Tagme(#[allow(dead_code)] u128); let tagme = Tagme(1000); let ptr = &tagme as *const Tagme as *mut Tagme; let atom: AtomicPtr = AtomicPtr::new(ptr); const MASK_TAG: usize = 0b1111; const MASK_PTR: usize = !MASK_TAG; assert_eq!(ptr.addr() & MASK_TAG, 0); assert_eq!(atom.fetch_or(0b0111, SeqCst), ptr); assert_eq!(atom.load(SeqCst), ptr.map_addr(|a| a | 0b111)); assert_eq!(atom.fetch_and(MASK_PTR | 0b0010, SeqCst), ptr.map_addr(|a| a | 0b111)); assert_eq!(atom.load(SeqCst), ptr.map_addr(|a| a | 0b0010)); assert_eq!(atom.fetch_xor(0b1011, SeqCst), ptr.map_addr(|a| a | 0b0010)); assert_eq!(atom.load(SeqCst), ptr.map_addr(|a| a | 0b1001)); assert_eq!(atom.fetch_and(MASK_PTR, SeqCst), ptr.map_addr(|a| a | 0b1001)); assert_eq!(atom.load(SeqCst), ptr); } static S_FALSE: AtomicBool = AtomicBool::new(false); static S_TRUE: AtomicBool = AtomicBool::new(true); static S_INT: AtomicIsize = AtomicIsize::new(0); static S_UINT: AtomicUsize = AtomicUsize::new(0); #[test] fn static_init() { // Note that we're not really testing the mutability here but it's important // on Android at the moment (#49775) assert!(!S_FALSE.swap(true, SeqCst)); assert!(S_TRUE.swap(false, SeqCst)); assert!(S_INT.fetch_add(1, SeqCst) == 0); assert!(S_UINT.fetch_add(1, SeqCst) == 0); } #[test] fn atomic_access_bool() { let mut atom = AtomicBool::new(false); assert_eq!(*atom.get_mut(), false); atom.store(true, SeqCst); assert_eq!(*atom.get_mut(), true); atom.fetch_or(false, SeqCst); assert_eq!(*atom.get_mut(), true); atom.fetch_and(false, SeqCst); assert_eq!(*atom.get_mut(), false); atom.fetch_nand(true, SeqCst); assert_eq!(*atom.get_mut(), true); atom.fetch_xor(true, SeqCst); assert_eq!(*atom.get_mut(), false); } #[test] fn atomic_alignment() { #[cfg(target_has_atomic = "8")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "ptr")] assert_eq!(align_of::>(), size_of::>()); #[cfg(target_has_atomic = "8")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "8")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "16")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "16")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "32")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "32")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "64")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "64")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "128")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "128")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "ptr")] assert_eq!(align_of::(), size_of::()); #[cfg(target_has_atomic = "ptr")] assert_eq!(align_of::(), size_of::()); } #[test] fn atomic_compare_exchange() { use Ordering::*; static ATOMIC: AtomicIsize = AtomicIsize::new(0); ATOMIC.compare_exchange(0, 1, Relaxed, Relaxed).ok(); ATOMIC.compare_exchange(0, 1, Relaxed, Acquire).ok(); ATOMIC.compare_exchange(0, 1, Relaxed, SeqCst).ok(); ATOMIC.compare_exchange(0, 1, Acquire, Relaxed).ok(); ATOMIC.compare_exchange(0, 1, Acquire, Acquire).ok(); ATOMIC.compare_exchange(0, 1, Acquire, SeqCst).ok(); ATOMIC.compare_exchange(0, 1, Release, Relaxed).ok(); ATOMIC.compare_exchange(0, 1, Release, Acquire).ok(); ATOMIC.compare_exchange(0, 1, Release, SeqCst).ok(); ATOMIC.compare_exchange(0, 1, AcqRel, Relaxed).ok(); ATOMIC.compare_exchange(0, 1, AcqRel, Acquire).ok(); ATOMIC.compare_exchange(0, 1, AcqRel, SeqCst).ok(); ATOMIC.compare_exchange(0, 1, SeqCst, Relaxed).ok(); ATOMIC.compare_exchange(0, 1, SeqCst, Acquire).ok(); ATOMIC.compare_exchange(0, 1, SeqCst, SeqCst).ok(); ATOMIC.compare_exchange_weak(0, 1, Relaxed, Relaxed).ok(); ATOMIC.compare_exchange_weak(0, 1, Relaxed, Acquire).ok(); ATOMIC.compare_exchange_weak(0, 1, Relaxed, SeqCst).ok(); ATOMIC.compare_exchange_weak(0, 1, Acquire, Relaxed).ok(); ATOMIC.compare_exchange_weak(0, 1, Acquire, Acquire).ok(); ATOMIC.compare_exchange_weak(0, 1, Acquire, SeqCst).ok(); ATOMIC.compare_exchange_weak(0, 1, Release, Relaxed).ok(); ATOMIC.compare_exchange_weak(0, 1, Release, Acquire).ok(); ATOMIC.compare_exchange_weak(0, 1, Release, SeqCst).ok(); ATOMIC.compare_exchange_weak(0, 1, AcqRel, Relaxed).ok(); ATOMIC.compare_exchange_weak(0, 1, AcqRel, Acquire).ok(); ATOMIC.compare_exchange_weak(0, 1, AcqRel, SeqCst).ok(); ATOMIC.compare_exchange_weak(0, 1, SeqCst, Relaxed).ok(); ATOMIC.compare_exchange_weak(0, 1, SeqCst, Acquire).ok(); ATOMIC.compare_exchange_weak(0, 1, SeqCst, SeqCst).ok(); } #[test] #[should_panic = "there is no such thing as an acquire-release failure ordering"] fn atomic_compare_exchange_illegal_acq_rel() { use Ordering::*; static ATOMIC: AtomicIsize = AtomicIsize::new(0); let failure = AcqRel; ATOMIC.compare_exchange(0, 1, Relaxed, failure).ok(); } #[test] #[should_panic = "there is no such thing as a release failure ordering"] fn atomic_compare_exchange_illegal_release() { use Ordering::*; static ATOMIC: AtomicIsize = AtomicIsize::new(0); let failure = Release; ATOMIC.compare_exchange(0, 1, Relaxed, failure).ok(); } #[test] #[should_panic = "there is no such thing as an acquire-release failure ordering"] fn atomic_compare_exchange_weak_illegal_acq_rel() { use Ordering::*; static ATOMIC: AtomicIsize = AtomicIsize::new(0); let failure = AcqRel; ATOMIC.compare_exchange_weak(0, 1, Relaxed, failure).ok(); } #[test] #[should_panic = "there is no such thing as a release failure ordering"] fn atomic_compare_exchange_weak_illegal_release() { use Ordering::*; static ATOMIC: AtomicIsize = AtomicIsize::new(0); let failure = Release; ATOMIC.compare_exchange_weak(0, 1, Relaxed, failure).ok(); } #[test] fn atomic_swap() { use Ordering::*; static ATOMIC: AtomicBool = AtomicBool::new(false); assert_eq!(ATOMIC.swap(true, Relaxed), false); assert_eq!(ATOMIC.swap(false, Acquire), true); assert_eq!(ATOMIC.swap(true, Release), false); assert_eq!(ATOMIC.swap(false, AcqRel), true); assert_eq!(ATOMIC.swap(true, SeqCst), false); } #[test] fn atomic_add() { use Ordering::*; static ATOMIC: AtomicU8 = AtomicU8::new(0); assert_eq!(ATOMIC.fetch_add(1, Relaxed), 0); assert_eq!(ATOMIC.fetch_add(1, Acquire), 1); assert_eq!(ATOMIC.fetch_add(1, Release), 2); assert_eq!(ATOMIC.fetch_add(1, AcqRel), 3); assert_eq!(ATOMIC.fetch_add(1, SeqCst), 4); assert_eq!(ATOMIC.load(Relaxed), 5); } #[test] fn atomic_sub() { use Ordering::*; static ATOMIC: AtomicU8 = AtomicU8::new(5); assert_eq!(ATOMIC.fetch_sub(1, Relaxed), 5); assert_eq!(ATOMIC.fetch_sub(1, Acquire), 4); assert_eq!(ATOMIC.fetch_sub(1, Release), 3); assert_eq!(ATOMIC.fetch_sub(1, AcqRel), 2); assert_eq!(ATOMIC.fetch_sub(1, SeqCst), 1); assert_eq!(ATOMIC.load(Relaxed), 0); } #[test] fn atomic_and_or() { use Ordering::*; static ATOMIC: AtomicBool = AtomicBool::new(false); assert_eq!(ATOMIC.fetch_or(true, Relaxed), false); assert_eq!(ATOMIC.fetch_and(false, Relaxed), true); assert_eq!(ATOMIC.fetch_or(true, Acquire), false); assert_eq!(ATOMIC.fetch_and(false, Acquire), true); assert_eq!(ATOMIC.fetch_or(true, Release), false); assert_eq!(ATOMIC.fetch_and(false, Release), true); assert_eq!(ATOMIC.fetch_or(true, AcqRel), false); assert_eq!(ATOMIC.fetch_and(false, AcqRel), true); assert_eq!(ATOMIC.fetch_or(true, SeqCst), false); assert_eq!(ATOMIC.fetch_and(false, SeqCst), true); assert_eq!(ATOMIC.load(Relaxed), false); } #[test] fn atomic_nand() { use Ordering::*; static ATOMIC: AtomicU8 = AtomicU8::new(0x13); assert_eq!(ATOMIC.fetch_nand(0x13, Relaxed), 0x13); assert_eq!(ATOMIC.fetch_nand(0xec, Acquire), 0xec); assert_eq!(ATOMIC.fetch_nand(0x13, Release), 0x13); assert_eq!(ATOMIC.fetch_nand(0xec, AcqRel), 0xec); assert_eq!(ATOMIC.fetch_nand(0x13, SeqCst), 0x13); assert_eq!(ATOMIC.load(Relaxed), 0xec); } #[test] fn atomic_xor() { use Ordering::*; static ATOMIC: AtomicBool = AtomicBool::new(false); assert_eq!(ATOMIC.fetch_xor(true, Relaxed), false); assert_eq!(ATOMIC.fetch_xor(true, Acquire), true); assert_eq!(ATOMIC.fetch_xor(true, Release), false); assert_eq!(ATOMIC.fetch_xor(true, AcqRel), true); assert_eq!(ATOMIC.fetch_xor(true, SeqCst), false); assert_eq!(ATOMIC.load(Relaxed), true); } #[test] fn atomic_max() { use Ordering::*; static ATOMIC: AtomicI8 = AtomicI8::new(0); assert_eq!(ATOMIC.fetch_max(1, Relaxed), 0); assert_eq!(ATOMIC.fetch_max(2, Acquire), 1); assert_eq!(ATOMIC.fetch_max(3, Release), 2); assert_eq!(ATOMIC.fetch_max(4, AcqRel), 3); assert_eq!(ATOMIC.fetch_max(5, SeqCst), 4); assert_eq!(ATOMIC.load(Relaxed), 5); } #[test] fn atomic_umax() { use Ordering::*; static ATOMIC: AtomicU8 = AtomicU8::new(0); assert_eq!(ATOMIC.fetch_max(1, Relaxed), 0); assert_eq!(ATOMIC.fetch_max(2, Acquire), 1); assert_eq!(ATOMIC.fetch_max(3, Release), 2); assert_eq!(ATOMIC.fetch_max(4, AcqRel), 3); assert_eq!(ATOMIC.fetch_max(5, SeqCst), 4); assert_eq!(ATOMIC.load(Relaxed), 5); } #[test] fn atomic_min() { use Ordering::*; static ATOMIC: AtomicI8 = AtomicI8::new(5); assert_eq!(ATOMIC.fetch_min(4, Relaxed), 5); assert_eq!(ATOMIC.fetch_min(3, Acquire), 4); assert_eq!(ATOMIC.fetch_min(2, Release), 3); assert_eq!(ATOMIC.fetch_min(1, AcqRel), 2); assert_eq!(ATOMIC.fetch_min(0, SeqCst), 1); assert_eq!(ATOMIC.load(Relaxed), 0); } #[test] fn atomic_umin() { use Ordering::*; static ATOMIC: AtomicU8 = AtomicU8::new(5); assert_eq!(ATOMIC.fetch_min(4, Relaxed), 5); assert_eq!(ATOMIC.fetch_min(3, Acquire), 4); assert_eq!(ATOMIC.fetch_min(2, Release), 3); assert_eq!(ATOMIC.fetch_min(1, AcqRel), 2); assert_eq!(ATOMIC.fetch_min(0, SeqCst), 1); assert_eq!(ATOMIC.load(Relaxed), 0); } /* FIXME(#110395) #[test] fn atomic_const_from() { const _ATOMIC_U8: AtomicU8 = AtomicU8::from(1); const _ATOMIC_BOOL: AtomicBool = AtomicBool::from(true); const _ATOMIC_PTR: AtomicPtr = AtomicPtr::from(core::ptr::null_mut()); } */