winio-ui-winui 0.3.5

WinUI backend for winio.
Documentation 
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

//! Hook GetMessage to wait for custom objects.

use std::sync::Once;

use slim_detours_sys::SlimDetoursInlineHook;
use sync_unsafe_cell::SyncUnsafeCell;
use windows::core::Result;
use windows_sys::Win32::{
    Foundation::HWND,
    UI::WindowsAndMessaging::{GetMessageW, MSG},
};

type GetMessageWFn =
    unsafe extern "system" fn(msg: *mut MSG, hwnd: HWND, min: u32, max: u32) -> i32;

static TRUE_GET_MESSAGE_W: SyncUnsafeCell<GetMessageWFn> = SyncUnsafeCell::new(GetMessageW);

unsafe extern "system" fn get_message(msg: *mut MSG, hwnd: HWND, min: u32, max: u32) -> i32 {
    if let Some(res) = crate::runtime::run_runtime(msg, hwnd, min, max) {
        res
    } else {
        (*TRUE_GET_MESSAGE_W.get())(msg, hwnd, min, max)
    }
}

fn detour_attach() -> Result<()> {
    let res =
        unsafe { SlimDetoursInlineHook(1, TRUE_GET_MESSAGE_W.get().cast(), get_message as _) };
    windows::core::HRESULT(res).ok()
}

static DETOUR_GUARD: Once = Once::new();

pub fn init_hook() {
    DETOUR_GUARD.call_once(|| {
        detour_attach().ok();
    });
}