111 lines
2.7 KiB
Rust
111 lines
2.7 KiB
Rust
//@compile-flags: -Zmiri-disable-weak-memory-emulation -Zmiri-preemption-rate=0
|
|
|
|
use std::sync::atomic::{fence, AtomicUsize, Ordering};
|
|
use std::thread::spawn;
|
|
|
|
#[derive(Copy, Clone)]
|
|
struct EvilSend<T>(pub T);
|
|
|
|
unsafe impl<T> Send for EvilSend<T> {}
|
|
unsafe impl<T> Sync for EvilSend<T> {}
|
|
|
|
fn test_fence_sync() {
|
|
static SYNC: AtomicUsize = AtomicUsize::new(0);
|
|
|
|
let mut var = 0u32;
|
|
let ptr = &mut var as *mut u32;
|
|
let evil_ptr = EvilSend(ptr);
|
|
|
|
let j1 = spawn(move || {
|
|
unsafe { *evil_ptr.0 = 1 };
|
|
fence(Ordering::Release);
|
|
SYNC.store(1, Ordering::Relaxed)
|
|
});
|
|
|
|
let j2 = spawn(move || {
|
|
if SYNC.load(Ordering::Relaxed) == 1 {
|
|
fence(Ordering::Acquire);
|
|
unsafe { *evil_ptr.0 }
|
|
} else {
|
|
panic!(); // relies on thread 2 going last
|
|
}
|
|
});
|
|
|
|
j1.join().unwrap();
|
|
j2.join().unwrap();
|
|
}
|
|
|
|
fn test_multiple_reads() {
|
|
let mut var = 42u32;
|
|
let ptr = &mut var as *mut u32;
|
|
let evil_ptr = EvilSend(ptr);
|
|
|
|
let j1 = spawn(move || unsafe { *evil_ptr.0 });
|
|
let j2 = spawn(move || unsafe { *evil_ptr.0 });
|
|
let j3 = spawn(move || unsafe { *evil_ptr.0 });
|
|
let j4 = spawn(move || unsafe { *evil_ptr.0 });
|
|
|
|
assert_eq!(j1.join().unwrap(), 42);
|
|
assert_eq!(j2.join().unwrap(), 42);
|
|
assert_eq!(j3.join().unwrap(), 42);
|
|
assert_eq!(j4.join().unwrap(), 42);
|
|
|
|
var = 10;
|
|
assert_eq!(var, 10);
|
|
}
|
|
|
|
pub fn test_rmw_no_block() {
|
|
static SYNC: AtomicUsize = AtomicUsize::new(0);
|
|
|
|
let mut a = 0u32;
|
|
let b = &mut a as *mut u32;
|
|
let c = EvilSend(b);
|
|
|
|
unsafe {
|
|
let j1 = spawn(move || {
|
|
*c.0 = 1;
|
|
SYNC.store(1, Ordering::Release);
|
|
});
|
|
|
|
let j2 = spawn(move || {
|
|
if SYNC.swap(2, Ordering::Relaxed) == 1 {
|
|
//No op, blocking store removed
|
|
}
|
|
});
|
|
|
|
let j3 = spawn(move || if SYNC.load(Ordering::Acquire) == 2 { *c.0 } else { 0 });
|
|
|
|
j1.join().unwrap();
|
|
j2.join().unwrap();
|
|
let v = j3.join().unwrap();
|
|
assert!(v == 1 || v == 2); // relies on thread 3 going last
|
|
}
|
|
}
|
|
|
|
pub fn test_simple_release() {
|
|
static SYNC: AtomicUsize = AtomicUsize::new(0);
|
|
|
|
let mut a = 0u32;
|
|
let b = &mut a as *mut u32;
|
|
let c = EvilSend(b);
|
|
|
|
unsafe {
|
|
let j1 = spawn(move || {
|
|
*c.0 = 1;
|
|
SYNC.store(1, Ordering::Release);
|
|
});
|
|
|
|
let j2 = spawn(move || if SYNC.load(Ordering::Acquire) == 1 { *c.0 } else { 0 });
|
|
|
|
j1.join().unwrap();
|
|
assert_eq!(j2.join().unwrap(), 1); // relies on thread 2 going last
|
|
}
|
|
}
|
|
|
|
pub fn main() {
|
|
test_fence_sync();
|
|
test_multiple_reads();
|
|
test_rmw_no_block();
|
|
test_simple_release();
|
|
}
|