DLLLauncherMain.cpp [plain text]
#include <shlwapi.h>
#include <string>
#include <vector>
#include <windows.h>
#if defined _M_IX86
#define PROCESSORARCHITECTURE "x86"
#elif defined _M_IA64
#define PROCESSORARCHITECTURE "ia64"
#elif defined _M_X64
#define PROCESSORARCHITECTURE "amd64"
#else
#define PROCESSORARCHITECTURE "*"
#endif
#pragma comment(linker, "/manifestdependency:\"type='win32' name='Microsoft.Windows.Common-Controls' version='6.0.0.0' processorArchitecture='" PROCESSORARCHITECTURE "' publicKeyToken='6595b64144ccf1df' language='*'\"")
#if defined(_MSC_VER) && (_MSC_VER >= 1600) && !defined(WIN_CAIRO)
#pragma comment(linker, "/manifestdependency:\"type='win32' name='Microsoft.VC80.CRT' version='8.0.50727.6195' processorArchitecture='" PROCESSORARCHITECTURE "' publicKeyToken='1fc8b3b9a1e18e3b' language='*'\"")
#endif
static void enableTerminationOnHeapCorruption()
{
HEAP_INFORMATION_CLASS heapEnableTerminationOnCorruption = static_cast<HEAP_INFORMATION_CLASS>(1);
HeapSetInformation(0, heapEnableTerminationOnCorruption, 0, 0);
}
static std::wstring getStringValue(HKEY key, const std::wstring& valueName)
{
DWORD type = 0;
DWORD bufferSize = 0;
if (::RegQueryValueExW(key, valueName.c_str(), 0, &type, 0, &bufferSize) != ERROR_SUCCESS || type != REG_SZ)
return std::wstring();
std::vector<wchar_t> buffer(bufferSize / sizeof(wchar_t));
if (::RegQueryValueExW(key, valueName.c_str(), 0, &type, reinterpret_cast<LPBYTE>(&buffer[0]), &bufferSize) != ERROR_SUCCESS)
return std::wstring();
return &buffer[0];
}
static std::wstring applePathFromRegistry(const std::wstring& key, const std::wstring& value)
{
HKEY applePathKey = 0;
if (::RegOpenKeyExW(HKEY_LOCAL_MACHINE, key.c_str(), 0, KEY_READ, &applePathKey) != ERROR_SUCCESS)
return std::wstring();
std::wstring path = getStringValue(applePathKey, value);
::RegCloseKey(applePathKey);
return path;
}
static std::wstring appleApplicationSupportDirectory()
{
return applePathFromRegistry(L"SOFTWARE\\Apple Inc.\\Apple Application Support", L"InstallDir");
}
static std::wstring copyEnvironmentVariable(const std::wstring& variable)
{
DWORD length = ::GetEnvironmentVariableW(variable.c_str(), 0, 0);
if (!length)
return std::wstring();
std::vector<wchar_t> buffer(length);
if (!GetEnvironmentVariable(variable.c_str(), &buffer[0], buffer.size()) || !buffer[0])
return std::wstring();
return &buffer[0];
}
static bool prependPath(const std::wstring& directoryToPrepend)
{
std::wstring pathVariable = L"PATH";
std::wstring oldPath = copyEnvironmentVariable(pathVariable);
std::wstring newPath = directoryToPrepend + L';' + oldPath;
return ::SetEnvironmentVariableW(pathVariable.c_str(), newPath.c_str());
}
static int fatalError(const std::wstring& programName, const std::wstring& message)
{
std::wstring caption = programName + L" can't open.";
::MessageBoxW(0, message.c_str(), caption.c_str(), MB_ICONERROR);
return 1;
}
static bool directoryExists(const std::wstring& path)
{
DWORD attrib = ::GetFileAttributes(path.c_str());
return ((attrib != INVALID_FILE_ATTRIBUTES) && (attrib & FILE_ATTRIBUTE_DIRECTORY));
}
static bool modifyPath(const std::wstring& programName)
{
#ifdef WIN_CAIRO
std::wstring pathWinCairo = copyEnvironmentVariable(L"WEBKIT_LIBRARIES");
if (!directoryExists(pathWinCairo))
return true;
#if defined(_M_X64)
pathWinCairo += L"\\bin64";
#else
pathWinCairo += L"\\bin32";
#endif
if (!SetDllDirectory(pathWinCairo.c_str())) {
fatalError(programName, L"Failed to SetDllDirectory");
return false;
}
return true;
#else
const std::wstring& pathPrefix = appleApplicationSupportDirectory();
if (!directoryExists(pathPrefix)) {
fatalError(programName, L"Failed to determine path to AAS directory.");
return false;
}
if (prependPath(pathPrefix))
return true;
fatalError(programName, L"Failed to modify PATH environment variable.");
return false;
#endif
}
static std::wstring getLastErrorString(HRESULT hr)
{
static const DWORD kFlags = FORMAT_MESSAGE_FROM_SYSTEM | FORMAT_MESSAGE_IGNORE_INSERTS;
static const size_t bufSize = 4096;
wchar_t errorMessage[bufSize];
DWORD len = ::FormatMessageW(kFlags, 0, hr, 0, errorMessage, bufSize, 0);
if (len >= bufSize)
len = bufSize - 1;
errorMessage[len] = 0;
return errorMessage;
}
#if USE_CONSOLE_ENTRY_POINT
int main(int argc, const char* argv[])
#else
int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPWSTR lpstrCmdLine, int nCmdShow)
#endif
{
enableTerminationOnHeapCorruption();
wchar_t exePath[MAX_PATH];
if (!::GetModuleFileNameW(0, exePath, _countof(exePath)))
return fatalError(L"Unknown Program", L"Failed to determine name of executable: " + getLastErrorString(::GetLastError()));
::PathRemoveExtensionW(exePath);
std::wstring programName = ::PathFindFileNameW(exePath);
if (!modifyPath(programName))
return 1;
std::wstring dllName = programName + L"Lib.dll";
if (!::PathRemoveFileSpecW(exePath))
return fatalError(programName, L"::PathRemoveFileSpecW failed: " + getLastErrorString(::GetLastError()));
if (!::PathAppendW(exePath, dllName.c_str()))
return fatalError(programName, L"::PathAppendW failed: " + getLastErrorString(::GetLastError()));
HMODULE module = ::LoadLibraryW(exePath);
if (!module)
return fatalError(programName, L"::LoadLibraryW failed: \npath=" + std::wstring(exePath) + L"\n" + getLastErrorString(::GetLastError()));
#if USE_CONSOLE_ENTRY_POINT
typedef int (WINAPI*EntryPoint)(int, const char*[]);
#if defined _M_AMD64 || defined _WIN64
const char* entryPointName = "dllLauncherEntryPoint";
#else
const char* entryPointName = "_dllLauncherEntryPoint@8";
#endif
#else
typedef int (WINAPI*EntryPoint)(HINSTANCE, HINSTANCE, LPWSTR, int);
#if defined _M_AMD64 || defined _WIN64
const char* entryPointName = "dllLauncherEntryPoint";
#else
const char* entryPointName = "_dllLauncherEntryPoint@16";
#endif
#endif
EntryPoint entryPoint = reinterpret_cast<EntryPoint>(::GetProcAddress(module, entryPointName));
if (!entryPoint)
return fatalError(programName, L"Failed to find dllLauncherEntryPoint function: " + getLastErrorString(::GetLastError()));
#if USE_CONSOLE_ENTRY_POINT
return entryPoint(argc, argv);
#else
return entryPoint(hInstance, hPrevInstance, lpstrCmdLine, nCmdShow);
#endif
}