/* -*- 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/. */


#ifdef _WIN32
#ifndef UNICODE
#define UNICODE
#endif
#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;
            }
        }

        // TODO: moggi: do we need all that code?
        // 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;
#endif