diff options
author | Ben Murdoch <benm@google.com> | 2011-05-13 16:23:25 +0100 |
---|---|---|
committer | Ben Murdoch <benm@google.com> | 2011-05-16 11:35:02 +0100 |
commit | 65f03d4f644ce73618e5f4f50dd694b26f55ae12 (patch) | |
tree | f478babb801e720de7bfaee23443ffe029f58731 /Source/WebKit2/WebProcess/WebKitMain.cpp | |
parent | 47de4a2fb7262c7ebdb9cd133ad2c54c187454d0 (diff) | |
download | external_webkit-65f03d4f644ce73618e5f4f50dd694b26f55ae12.zip external_webkit-65f03d4f644ce73618e5f4f50dd694b26f55ae12.tar.gz external_webkit-65f03d4f644ce73618e5f4f50dd694b26f55ae12.tar.bz2 |
Merge WebKit at r75993: Initial merge by git.
Change-Id: I602bbdc3974787a3b0450456a30a7868286921c3
Diffstat (limited to 'Source/WebKit2/WebProcess/WebKitMain.cpp')
-rw-r--r-- | Source/WebKit2/WebProcess/WebKitMain.cpp | 174 |
1 files changed, 174 insertions, 0 deletions
diff --git a/Source/WebKit2/WebProcess/WebKitMain.cpp b/Source/WebKit2/WebProcess/WebKitMain.cpp new file mode 100644 index 0000000..53084a6 --- /dev/null +++ b/Source/WebKit2/WebProcess/WebKitMain.cpp @@ -0,0 +1,174 @@ +/* + * Copyright (C) 2010 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF + * THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "CommandLine.h" + +#include "PluginProcessMain.h" +#include "ProcessLauncher.h" +#include "WebProcessMain.h" +#include <wtf/text/CString.h> + +#if PLATFORM(MAC) +#include <objc/objc-auto.h> +#endif + +using namespace WebKit; + +static int WebKitMain(const CommandLine& commandLine) +{ + ProcessLauncher::ProcessType processType; + if (!ProcessLauncher::getProcessTypeFromString(commandLine["type"].utf8().data(), processType)) + return EXIT_FAILURE; + + switch (processType) { + case ProcessLauncher::WebProcess: + return WebProcessMain(commandLine); + case ProcessLauncher::PluginProcess: +#if ENABLE(PLUGIN_PROCESS) + return PluginProcessMain(commandLine); +#else + break; +#endif + } + + return EXIT_FAILURE; +} + +#if PLATFORM(MAC) + +extern "C" WK_EXPORT int WebKitMain(int argc, char** argv); + +int WebKitMain(int argc, char** argv) +{ + ASSERT(!objc_collectingEnabled()); + + CommandLine commandLine; + if (!commandLine.parse(argc, argv)) + return EXIT_FAILURE; + + return WebKitMain(commandLine); +} + +#elif PLATFORM(WIN) + +#ifndef DEBUG_ALL +#define PROCESS_NAME L"WebKit2WebKitProcess.exe" +#else +#define PROCESS_NAME L"WebKit2WebProcess_debug.exe" +#endif + +static void enableDataExecutionPrevention() +{ + // Enable Data Execution prevention at runtime rather than via /NXCOMPAT + // http://blogs.msdn.com/michael_howard/archive/2008/01/29/new-nx-apis-added-to-windows-vista-sp1-windows-xp-sp3-and-windows-server-2008.aspx + + const DWORD enableDEP = 0x00000001; + + HMODULE hMod = ::GetModuleHandleW(L"Kernel32.dll"); + if (!hMod) + return; + + typedef BOOL (WINAPI *PSETDEP)(DWORD); + + PSETDEP procSet = reinterpret_cast<PSETDEP>(::GetProcAddress(hMod, "SetProcessDEPPolicy")); + if (!procSet) + return; + + // Enable Data Execution Prevention, but allow ATL thunks (for compatibility with the version of ATL that ships with the Platform SDK). + procSet(enableDEP); +} + +static void enableTerminationOnHeapCorruption() +{ + // Enable termination on heap corruption on OSes that support it (Vista and XPSP3). + // http://msdn.microsoft.com/en-us/library/aa366705(VS.85).aspx + + const HEAP_INFORMATION_CLASS heapEnableTerminationOnCorruption = static_cast<HEAP_INFORMATION_CLASS>(1); + + HMODULE hMod = ::GetModuleHandleW(L"kernel32.dll"); + if (!hMod) + return; + + typedef BOOL (WINAPI*HSI)(HANDLE, HEAP_INFORMATION_CLASS, PVOID, SIZE_T); + HSI heapSetInformation = reinterpret_cast<HSI>(::GetProcAddress(hMod, "HeapSetInformation")); + if (!heapSetInformation) + return; + + heapSetInformation(0, heapEnableTerminationOnCorruption, 0, 0); +} + +static void disableUserModeCallbackExceptionFilter() +{ + const DWORD PROCESS_CALLBACK_FILTER_ENABLED = 0x1; + typedef BOOL (NTAPI *getProcessUserModeExceptionPolicyPtr)(LPDWORD lpFlags); + typedef BOOL (NTAPI *setProcessUserModeExceptionPolicyPtr)(DWORD dwFlags); + + HMODULE lib = LoadLibrary(TEXT("kernel32.dll")); + ASSERT(lib); + + getProcessUserModeExceptionPolicyPtr getPolicyPtr = (getProcessUserModeExceptionPolicyPtr)GetProcAddress(lib, "GetProcessUserModeExceptionPolicy"); + setProcessUserModeExceptionPolicyPtr setPolicyPtr = (setProcessUserModeExceptionPolicyPtr)GetProcAddress(lib, "SetProcessUserModeExceptionPolicy"); + + DWORD dwFlags; + if (!getPolicyPtr || !setPolicyPtr || !getPolicyPtr(&dwFlags)) { + FreeLibrary(lib); + return; + } + + // If this flag isn't cleared, exceptions that are thrown when running in a 64-bit version of + // Windows are ignored, possibly leaving Safari in an inconsistent state that could cause an + // unrelated exception to be thrown. + // http://support.microsoft.com/kb/976038 + // http://blog.paulbetts.org/index.php/2010/07/20/the-case-of-the-disappearing-onload-exception-user-mode-callback-exceptions-in-x64/ + setPolicyPtr(dwFlags & ~PROCESS_CALLBACK_FILTER_ENABLED); + + FreeLibrary(lib); +} + +extern "C" __declspec(dllexport) +int WebKitMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPTSTR lpstrCmdLine, int nCmdShow) +{ +#ifndef NDEBUG + // Show an alert when Ctrl-Alt-Shift is held down during launch to give the user time to attach a + // debugger. This is useful for debugging problems that happen early in the web process's lifetime. + const unsigned short highBitMaskShort = 0x8000; + if (getenv("WEBKIT2_PAUSE_WEB_PROCESS_ON_LAUNCH") || (::GetKeyState(VK_CONTROL) & highBitMaskShort) && (::GetKeyState(VK_MENU) & highBitMaskShort) && (::GetKeyState(VK_SHIFT) & highBitMaskShort)) + ::MessageBoxW(0, L"You can now attach a debugger to " PROCESS_NAME L". You can use\nthe same debugger for WebKit2WebProcessand the UI process, if desired.\nClick OK when you are ready for WebKit2WebProcess to continue.", L"WebKit2WebProcess has launched", MB_OK | MB_ICONINFORMATION); +#endif + + enableDataExecutionPrevention(); + + enableTerminationOnHeapCorruption(); + + disableUserModeCallbackExceptionFilter(); + + CommandLine commandLine; + if (!commandLine.parse(lpstrCmdLine)) + return EXIT_FAILURE; + + return WebKitMain(commandLine); +} + +#endif |