src/libstd/sys/wasm/thread.rs

   1 // Copyright 2017 The Rust Project Developers. See the COPYRIGHT
   2 // file at the top-level directory of this distribution and at
   3 // http://rust-lang.org/COPYRIGHT.
   4 //
   5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
   6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
   7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
   8 // option. This file may not be copied, modified, or distributed
   9 // except according to those terms.
  10
  11 use boxed::FnBox;
  12 use ffi::CStr;
  13 use io;
  14 use sys::{unsupported, Void};
  15 use time::Duration;
  16
  17 pub struct Thread(Void);
  18
  19 pub const DEFAULT_MIN_STACK_SIZE: usize = 4096;
  20
  21 impl Thread {
  22     // unsafe: see thread::Builder::spawn_unchecked for safety requirements
  23     pub unsafe fn new(_stack: usize, _p: Box<dyn FnBox()>)
  24         -> io::Result<Thread>
  25     {
  26         unsupported()
  27     }
  28
  29     pub fn yield_now() {
  30         // do nothing
  31     }
  32
  33     pub fn set_name(_name: &CStr) {
  34         // nope
  35     }
  36
  37     #[cfg(not(target_feature = "atomics"))]
  38     pub fn sleep(_dur: Duration) {
  39         panic!("can't sleep");
  40     }
  41
  42     #[cfg(target_feature = "atomics")]
  43     pub fn sleep(dur: Duration) {
  44         use arch::wasm32::atomic;
  45         use cmp;
  46
  47         // Use an atomic wait to block the current thread artificially with a
  48         // timeout listed. Note that we should never be notified (return value
  49         // of 0) or our comparison should never fail (return value of 1) so we
  50         // should always only resume execution through a timeout (return value
  51         // 2).
  52         let mut nanos = dur.as_nanos();
  53         while nanos > 0 {
  54             let amt = cmp::min(i64::max_value() as u128, nanos);
  55             let mut x = 0;
  56             let val = unsafe { atomic::wait_i32(&mut x, 0, amt as i64) };
  57             debug_assert_eq!(val, 2);
  58             nanos -= amt;
  59         }
  60     }
  61
  62     pub fn join(self) {
  63         match self.0 {}
  64     }
  65 }
  66
  67 pub mod guard {
  68     pub type Guard = !;
  69     pub unsafe fn current() -> Option<Guard> { None }
  70     pub unsafe fn init() -> Option<Guard> { None }
  71     pub unsafe fn deinit() {}
  72 }
  73
  74 cfg_if! {
  75     if #[cfg(all(target_feature = "atomics", feature = "wasm-bindgen-threads"))] {
  76         #[link(wasm_import_module = "__wbindgen_thread_xform__")]
  77         extern {
  78             fn __wbindgen_current_id() -> u32;
  79             fn __wbindgen_tcb_get() -> u32;
  80             fn __wbindgen_tcb_set(ptr: u32);
  81         }
  82         pub fn my_id() -> u32 {
  83             unsafe { __wbindgen_current_id() }
  84         }
  85
  86         // These are currently only ever used in `thread_local_atomics.rs`, if
  87         // you'd like to use them be sure to update that and make sure everyone
  88         // agrees what's what.
  89         pub fn tcb_get() -> *mut u8 {
  90             use mem;
  91             assert_eq!(mem::size_of::<*mut u8>(), mem::size_of::<u32>());
  92             unsafe { __wbindgen_tcb_get() as *mut u8 }
  93         }
  94
  95         pub fn tcb_set(ptr: *mut u8) {
  96             unsafe { __wbindgen_tcb_set(ptr as u32); }
  97         }
  98
  99         // FIXME: still need something for hooking exiting a thread to free
 100         // data...
 101
 102     } else if #[cfg(target_feature = "atomics")] {
 103         pub fn my_id() -> u32 {
 104             panic!("thread ids not implemented on wasm with atomics yet")
 105         }
 106
 107         pub fn tcb_get() -> *mut u8 {
 108             panic!("thread local data not implemented on wasm with atomics yet")
 109         }
 110
 111         pub fn tcb_set(ptr: *mut u8) {
 112             panic!("thread local data not implemented on wasm with atomics yet")
 113         }
 114     } else {
 115         // stubbed out because no functions actually access these intrinsics
 116         // unless atomics are enabled
 117     }
 118 }