| #![allow(dead_code)] // stack_guard isn't used right now on all platforms |
| |
| use crate::cell::RefCell; |
| use crate::sys::thread::guard::Guard; |
| use crate::thread::Thread; |
| |
| struct ThreadInfo { |
| stack_guard: Option<Guard>, |
| thread: Thread, |
| } |
| |
| thread_local! { static THREAD_INFO: RefCell<Option<ThreadInfo>> = RefCell::new(None) } |
| |
| impl ThreadInfo { |
| fn with<R, F>(f: F) -> Option<R> where F: FnOnce(&mut ThreadInfo) -> R { |
| THREAD_INFO.try_with(move |c| { |
| if c.borrow().is_none() { |
| *c.borrow_mut() = Some(ThreadInfo { |
| stack_guard: None, |
| thread: Thread::new(None), |
| }) |
| } |
| f(c.borrow_mut().as_mut().unwrap()) |
| }).ok() |
| } |
| } |
| |
| pub fn current_thread() -> Option<Thread> { |
| ThreadInfo::with(|info| info.thread.clone()) |
| } |
| |
| pub fn stack_guard() -> Option<Guard> { |
| ThreadInfo::with(|info| info.stack_guard.clone()).and_then(|o| o) |
| } |
| |
| pub fn set(stack_guard: Option<Guard>, thread: Thread) { |
| THREAD_INFO.with(|c| assert!(c.borrow().is_none())); |
| THREAD_INFO.with(move |c| *c.borrow_mut() = Some(ThreadInfo{ |
| stack_guard, |
| thread, |
| })); |
| } |
| |
| pub fn reset_guard(stack_guard: Option<Guard>) { |
| THREAD_INFO.with(move |c| c.borrow_mut().as_mut().unwrap().stack_guard = stack_guard); |
| } |