1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
|
// 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.
// All files in the project carrying such notice may not be copied, modified, or distributed
// except according to those terms.
use {Result, k32, last_error, w};
use handle::{Handle};
use std::os::windows::io::{AsRawHandle, FromRawHandle, IntoRawHandle};
use std::thread::{JoinHandle};
pub struct Thread(Handle);
impl Thread {
pub fn current() -> Result<Thread> {
unsafe { Handle::duplicate_from(k32::GetCurrentThread()).map(Thread) }
}
/// Returns the old affinity mask on success
pub fn set_affinity_mask(&self, mask: usize) -> Result<usize> {
let res = unsafe {
k32::SetThreadAffinityMask(*self.0, mask as w::ULONG_PTR)
};
match res {
0 => last_error(),
prev => Ok(prev as usize),
}
}
}
impl<T> From<JoinHandle<T>> for Thread {
fn from(o: JoinHandle<T>) -> Thread {
unsafe { Thread::from_raw_handle(o.into_raw_handle()) }
}
}
impl<'a, T> From<&'a JoinHandle<T>> for Thread {
fn from(o: &'a JoinHandle<T>) -> Thread {
unsafe { Thread::from_raw_handle(o.as_raw_handle()) }
}
}
impl AsRawHandle for Thread {
fn as_raw_handle(&self) -> w::HANDLE {
self.0.as_raw_handle()
}
}
impl IntoRawHandle for Thread {
fn into_raw_handle(self) -> w::HANDLE {
self.0.into_raw_handle()
}
}
impl FromRawHandle for Thread {
unsafe fn from_raw_handle(handle: w::HANDLE) -> Thread {
Thread(Handle::from_raw_handle(handle))
}
}
|
