/*!
 * win_delay_load_hook.cc - notify hook for windows
 * Copyright (c) 2020, Christopher Jeffrey (MIT License).
 * https://github.com/chjj/cmake-node
 *
 * Parts of this software are based on nodejs/node-gyp:
 *   Copyright (c) 2012, Nathan Rajlich (MIT License).
 *   https://github.com/nodejs/node-gyp
 *
 * Resources:
 *   https://github.com/nodejs/node-gyp/blob/master/src/win_delay_load_hook.cc
 *   https://docs.microsoft.com/en-us/cpp/build/reference/notification-hooks
 *
 * Explanation:
 *
 *   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.
 */

#if defined(_MSC_VER) && !defined(__clang__)
#  pragma managed(push, off)
#endif

#ifndef WIN32_LEAN_AND_MEAN
#  define WIN32_LEAN_AND_MEAN
#endif

#include <windows.h>
#include <delayimp.h>
#include <string.h>

#ifndef __MINGW32__
#  pragma comment(lib, "kernel32.lib")
#  pragma comment(lib, "delayimp.lib")
#endif

static FARPROC WINAPI
load_exe_hook(unsigned int event, DelayLoadInfo *info) {
  HMODULE m;

  if (event != dliNotePreLoadLibrary)
    return NULL;

  if (_stricmp(info->szDll, NODE_HOST_BINARY) != 0)
    return NULL;

  m = GetModuleHandle(NULL);

  return (FARPROC)m;
}

decltype(__pfnDliNotifyHook2) __pfnDliNotifyHook2 = load_exe_hook;

#if defined(_MSC_VER) && !defined(__clang__)
#  pragma managed(pop)
#endif