summaryrefslogtreecommitdiffstats
path: root/third_party/rust/wio/src/apc.rs
blob: 21dcef2974edfecdc4cc04dc499d1605dad88709 (plain)
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
// 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 std::os::windows::io::{AsRawHandle};
use thread::{Thread};

pub fn queue<T>(func: T, thread: &Thread) -> Result<()> where T: FnOnce() + 'static {
    unsafe extern "system" fn helper<T: FnOnce() + 'static>(thing: w::ULONG_PTR) {
        let func = Box::from_raw(thing as *mut T);
        func()
    }
    let thing = Box::into_raw(Box::new(func)) as w::ULONG_PTR;
    match unsafe { k32::QueueUserAPC(Some(helper::<T>), thread.as_raw_handle(), thing) } {
        0 => {
            // If it fails we still need to deallocate the function
            unsafe { Box::from_raw(thing as *mut T) };
            last_error()
        },
        _ => Ok(()),
    }
}
pub fn queue_current<T>(func: T) -> Result<()> where T: FnOnce() + 'static {
    unsafe extern "system" fn helper<T: FnOnce() + 'static>(thing: w::ULONG_PTR) {
        let func = Box::from_raw(thing as *mut T);
        func()
    }
    let thing = Box::into_raw(Box::new(func)) as w::ULONG_PTR;
    match unsafe { k32::QueueUserAPC(Some(helper::<T>), k32::GetCurrentThread(), thing) } {
        0 => {
            // If it fails we still need to deallocate the function
            unsafe { Box::from_raw(thing as *mut T) };
            last_error()
        },
        _ => Ok(()),
    }
}