mirror of
https://github.com/classilla/tenfourfox.git
synced 2024-10-22 02:25:05 +00:00
104 lines
4.5 KiB
C++
104 lines
4.5 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include <windows.h>
|
|
|
|
// Delayed load libraries are loaded when the first symbol is used.
|
|
// The following ensures that we load the delayed loaded libraries from the
|
|
// system directory.
|
|
struct AutoLoadSystemDependencies
|
|
{
|
|
AutoLoadSystemDependencies()
|
|
{
|
|
// Remove the current directory from the search path for dynamically loaded
|
|
// DLLs as a precaution. This call has no effect for delay load DLLs.
|
|
SetDllDirectory(L"");
|
|
|
|
HMODULE module = ::GetModuleHandleW(L"kernel32.dll");
|
|
if (module) {
|
|
// SetDefaultDllDirectories is always available on Windows 8 and above. It
|
|
// is also available on Windows Vista, Windows Server 2008, and
|
|
// Windows 7 when MS KB2533623 has been applied.
|
|
decltype(SetDefaultDllDirectories)* setDefaultDllDirectories =
|
|
(decltype(SetDefaultDllDirectories)*) GetProcAddress(module, "SetDefaultDllDirectories");
|
|
if (setDefaultDllDirectories) {
|
|
setDefaultDllDirectories(LOAD_LIBRARY_SEARCH_SYSTEM32);
|
|
return;
|
|
}
|
|
}
|
|
|
|
// When SetDefaultDllDirectories is not available, fallback to preloading
|
|
// dlls. The order that these are loaded does not matter since they are
|
|
// loaded using the LOAD_WITH_ALTERED_SEARCH_PATH flag.
|
|
#ifdef HAVE_64BIT_BUILD
|
|
// DLLs for Firefox x64 on Windows 7 (x64).
|
|
// Note: dwmapi.dll is preloaded since a crash will try to load it from the
|
|
// application's directory.
|
|
static LPCWSTR delayDLLs[] = { L"apphelp.dll",
|
|
L"cryptbase.dll",
|
|
L"cryptsp.dll",
|
|
L"dwmapi.dll",
|
|
L"mpr.dll",
|
|
L"ntmarta.dll",
|
|
L"profapi.dll",
|
|
L"propsys.dll",
|
|
L"sspicli.dll",
|
|
L"wsock32.dll" };
|
|
|
|
#else
|
|
// DLLs for Firefox x86 on Windows XP through Windows 7 (x86 and x64).
|
|
// Note: dwmapi.dll is preloaded since a crash will try to load it from the
|
|
// application's directory.
|
|
static LPCWSTR delayDLLs[] = { L"apphelp.dll",
|
|
L"crypt32.dll",
|
|
L"cryptbase.dll",
|
|
L"cryptsp.dll",
|
|
L"dwmapi.dll",
|
|
L"mpr.dll",
|
|
L"msasn1.dll",
|
|
L"ntmarta.dll",
|
|
L"profapi.dll",
|
|
L"propsys.dll",
|
|
L"psapi.dll",
|
|
L"secur32.dll",
|
|
L"sspicli.dll",
|
|
L"userenv.dll",
|
|
L"uxtheme.dll",
|
|
L"ws2_32.dll",
|
|
L"ws2help.dll",
|
|
L"wsock32.dll" };
|
|
#endif
|
|
|
|
WCHAR systemDirectory[MAX_PATH + 1] = { L'\0' };
|
|
// If GetSystemDirectory fails we accept that we'll load the DLLs from the
|
|
// normal search path.
|
|
GetSystemDirectoryW(systemDirectory, MAX_PATH + 1);
|
|
size_t systemDirLen = wcslen(systemDirectory);
|
|
|
|
// Make the system directory path terminate with a slash
|
|
if (systemDirectory[systemDirLen - 1] != L'\\' && systemDirLen) {
|
|
systemDirectory[systemDirLen] = L'\\';
|
|
++systemDirLen;
|
|
// No need to re-null terminate
|
|
}
|
|
|
|
// For each known DLL ensure it is loaded from the system32 directory
|
|
for (size_t i = 0; i < sizeof(delayDLLs) / sizeof(delayDLLs[0]); ++i) {
|
|
size_t fileLen = wcslen(delayDLLs[i]);
|
|
wcsncpy(systemDirectory + systemDirLen, delayDLLs[i],
|
|
MAX_PATH - systemDirLen);
|
|
if (systemDirLen + fileLen <= MAX_PATH) {
|
|
systemDirectory[systemDirLen + fileLen] = L'\0';
|
|
} else {
|
|
systemDirectory[MAX_PATH] = L'\0';
|
|
}
|
|
LPCWSTR fullModulePath = systemDirectory; // just for code readability
|
|
// LOAD_WITH_ALTERED_SEARCH_PATH makes a dll look in its own directory for
|
|
// dependencies and is only available on Win 7 and below.
|
|
LoadLibraryExW(fullModulePath, nullptr, LOAD_WITH_ALTERED_SEARCH_PATH);
|
|
}
|
|
}
|
|
} loadDLLs;
|