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

//! A standalone DLL that hosts the extension functionality
//! of all [me3_game_support_fromsoft] supported games.
#![feature(assert_matches)]

use windows::Win32::{
    Foundation::HINSTANCE,
    System::{
        LibraryLoader::{DisableThreadLibraryCalls, FreeLibraryAndExitThread},
        SystemServices::{DLL_PROCESS_ATTACH, DLL_PROCESS_DETACH},
    },
};

mod bootstrap;
mod script_api;
mod widgets;

#[no_mangle]
pub extern "stdcall" fn DllMain(hinst_dll: HINSTANCE, fdw_reason: u32, _: *const ()) -> i32 {
    let success = match fdw_reason {
        DLL_PROCESS_ATTACH => {
            unsafe {
                DisableThreadLibraryCalls(hinst_dll);
                let _ = std::thread::spawn(move || {
                    let exit_code = match std::panic::catch_unwind(bootstrap::setup_and_run) {
                        Err(e) => {
                            log::error!("me3_host panicked in bootstrap: {:#?}", e);
                            0
                        }
                        Ok(Err(e)) => {
                            log::error!("encountered an error during setup: {:#?}", e);
                            0
                        }
                        Ok(_) => 1,
                    };
                    FreeLibraryAndExitThread(hinst_dll, exit_code)
                });
            }

            true
        }
        DLL_PROCESS_DETACH => true,
        _ => true,
    };

    success as i32
}