src/tools/miri/tests/fail/data_race/rmw_race.rs

   1 //@compile-flags: -Zmiri-disable-weak-memory-emulation -Zmiri-preemption-rate=0
   2
   3 use std::sync::atomic::{AtomicUsize, Ordering};
   4 use std::thread::spawn;
   5
   6 #[derive(Copy, Clone)]
   7 struct EvilSend<T>(pub T);
   8
   9 unsafe impl<T> Send for EvilSend<T> {}
  10 unsafe impl<T> Sync for EvilSend<T> {}
  11
  12 static SYNC: AtomicUsize = AtomicUsize::new(0);
  13
  14 pub fn main() {
  15     let mut a = 0u32;
  16     let b = &mut a as *mut u32;
  17     let c = EvilSend(b);
  18
  19     // Note: this is scheduler-dependent
  20     // the operations need to occur in
  21     // order:
  22     //  1. store release : 1
  23     //  2. RMW relaxed : 1 -> 2
  24     //  3. store relaxed : 3
  25     //  4. load acquire : 3
  26     unsafe {
  27         let j1 = spawn(move || {
  28             *c.0 = 1;
  29             SYNC.store(1, Ordering::Release);
  30         });
  31
  32         let j2 = spawn(move || {
  33             if SYNC.swap(2, Ordering::Relaxed) == 1 {
  34                 // Blocks the acquire-release sequence
  35                 SYNC.store(3, Ordering::Relaxed);
  36             }
  37         });
  38
  39         let j3 = spawn(move || {
  40             if SYNC.load(Ordering::Acquire) == 3 {
  41                 *c.0 //~ ERROR: Data race detected between Read on thread `<unnamed>` and Write on thread `<unnamed>`
  42             } else {
  43                 0
  44             }
  45         });
  46
  47         j1.join().unwrap();
  48         j2.join().unwrap();
  49         j3.join().unwrap();
  50     }
  51 }