2019-11-07 11:06:18 +00:00
|
|
|
/**
|
|
|
|
|
* Copyright (c) 2019 Kylart <kylart.dev@gmail.com>
|
|
|
|
|
*
|
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
|
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
|
|
|
* in the Software without restriction, including without limitation the rights
|
|
|
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
|
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
|
|
|
* furnished to do so, subject to the following conditions:
|
|
|
|
|
*
|
|
|
|
|
* The above copyright notice and this permission notice shall be included in
|
|
|
|
|
* all copies or substantial portions of the Software.
|
|
|
|
|
*
|
|
|
|
|
* Coming right from https://github.com/nodejs/node-addon-api/issues/269#issuecomment-455580129
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
#ifdef _MSC_VER
|
|
|
|
|
|
|
|
|
|
#define WIN32_LEAN_AND_MEAN
|
|
|
|
|
#include <windows.h>
|
|
|
|
|
#include <Shlwapi.h>
|
|
|
|
|
#include <delayimp.h>
|
|
|
|
|
#include <string.h>
|
|
|
|
|
|
|
|
|
|
#pragma comment(lib, "Shlwapi.lib")
|
|
|
|
|
|
|
|
|
|
static FARPROC WINAPI load_exe_hook(unsigned int event, DelayLoadInfo* info) {
|
|
|
|
|
if (event != dliNotePreLoadLibrary)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
if (_stricmp(info->szDll, "iojs.exe") != 0 &&
|
|
|
|
|
_stricmp(info->szDll, "node.exe") != 0 &&
|
|
|
|
|
_stricmp(info->szDll, "node.dll") != 0)
|
|
|
|
|
return NULL;
|
|
|
|
|
|
|
|
|
|
// Get a handle to the current process executable.
|
|
|
|
|
HMODULE processModule = GetModuleHandle(NULL);
|
|
|
|
|
|
|
|
|
|
// Get the path to the executable.
|
2019-11-07 13:22:50 +00:00
|
|
|
TCHAR processPath[_MAX_PATH]; // NOLINT
|
2019-11-07 11:06:18 +00:00
|
|
|
GetModuleFileName(processModule, processPath, _MAX_PATH);
|
|
|
|
|
|
|
|
|
|
// Get the name of the current executable.
|
|
|
|
|
LPSTR processName = PathFindFileName(processPath);
|
|
|
|
|
|
|
|
|
|
// If the current process is node or iojs, then just return the proccess module.
|
|
|
|
|
if (_stricmp(processName, "node.exe") == 0 ||
|
|
|
|
|
_stricmp(processName, "iojs.exe") == 0) {
|
|
|
|
|
return (FARPROC)processModule;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// If it is another process, attempt to load 'node.dll' from the same directory.
|
|
|
|
|
PathRemoveFileSpec(processPath);
|
|
|
|
|
PathAppend(processPath, "node.dll");
|
|
|
|
|
|
|
|
|
|
HMODULE nodeDllModule = GetModuleHandle(processPath);
|
|
|
|
|
if (nodeDllModule != NULL) {
|
|
|
|
|
// This application has a node.dll in the same directory as the executable, use that.
|
|
|
|
|
return (FARPROC)nodeDllModule;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Fallback to the current executable, which must statically link to node.lib.
|
|
|
|
|
return (FARPROC)processModule;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// See https://docs.microsoft.com/en-us/cpp/build/reference/notification-hooks
|
|
|
|
|
decltype(__pfnDliNotifyHook2) __pfnDliNotifyHook2 = load_exe_hook;
|
|
|
|
|
|
|
|
|
|
#endif
|
