You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.8 KiB
62 lines
1.8 KiB
8 years ago
|
// Copyright 2014 The Rust Project Developers. See the COPYRIGHT
|
||
|
// file at the top-level directory of this distribution and at
|
||
|
// http://rust-lang.org/COPYRIGHT.
|
||
|
//
|
||
|
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
|
||
|
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
|
||
|
// option. This file may not be copied, modified, or distributed
|
||
|
// except according to those terms.
|
||
|
|
||
|
#![allow(dead_code)] // stack_guard isn't used right now on all platforms
|
||
|
|
||
|
use cell::RefCell;
|
||
|
use thread::Thread;
|
||
|
use thread::LocalKeyState;
|
||
|
|
||
|
struct ThreadInfo {
|
||
|
stack_guard: Option<usize>,
|
||
|
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 {
|
||
|
if THREAD_INFO.state() == LocalKeyState::Destroyed {
|
||
|
return None
|
||
|
}
|
||
|
|
||
|
THREAD_INFO.with(move |c| {
|
||
|
if c.borrow().is_none() {
|
||
|
*c.borrow_mut() = Some(ThreadInfo {
|
||
|
stack_guard: None,
|
||
|
thread: NewThread::new(None),
|
||
|
})
|
||
|
}
|
||
|
Some(f(c.borrow_mut().as_mut().unwrap()))
|
||
|
})
|
||
|
}
|
||
|
}
|
||
|
|
||
|
pub fn current_thread() -> Option<Thread> {
|
||
|
ThreadInfo::with(|info| info.thread.clone())
|
||
|
}
|
||
|
|
||
|
pub fn stack_guard() -> Option<usize> {
|
||
|
ThreadInfo::with(|info| info.stack_guard).and_then(|o| o)
|
||
|
}
|
||
|
|
||
|
pub fn set(stack_guard: Option<usize>, thread: Thread) {
|
||
|
THREAD_INFO.with(|c| assert!(c.borrow().is_none()));
|
||
|
THREAD_INFO.with(move |c| *c.borrow_mut() = Some(ThreadInfo{
|
||
|
stack_guard: stack_guard,
|
||
|
thread: thread,
|
||
|
}));
|
||
|
}
|
||
|
|
||
|
// a hack to get around privacy restrictions; implemented by `std::thread`
|
||
|
pub trait NewThread {
|
||
|
fn new(name: Option<String>) -> Self;
|
||
|
}
|