马宇豪
2024-07-16 f591c27b57e2418c9495bc02ae8cfff84d35bc18
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
/*
 * When this file is linked to a DLL, it sets up a delay-load hook that
 * intervenes when the DLL is trying to load the host executable
 * dynamically. Instead of trying to locate the .exe file it'll just
 * return a handle to the process image.
 *
 * This allows compiled addons to work when the host executable is renamed.
 */
 
#ifdef _MSC_VER
 
#pragma managed(push, off)
 
#ifndef WIN32_LEAN_AND_MEAN
#define WIN32_LEAN_AND_MEAN
#endif
 
#include <windows.h>
 
#include <delayimp.h>
#include <string.h>
 
static FARPROC WINAPI load_exe_hook(unsigned int event, DelayLoadInfo* info) {
  HMODULE m;
  if (event != dliNotePreLoadLibrary)
    return NULL;
 
  if (_stricmp(info->szDll, HOST_BINARY) != 0)
    return NULL;
 
  m = GetModuleHandle(NULL);
  return (FARPROC) m;
}
 
decltype(__pfnDliNotifyHook2) __pfnDliNotifyHook2 = load_exe_hook;
 
#pragma managed(pop)
 
#endif