diff -r 000000000000 -r 4f2f89ce4247 WebKitTools/WebKitLauncherWin/WebKitLauncherWin.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/WebKitTools/WebKitLauncherWin/WebKitLauncherWin.cpp Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,95 @@ +/* + * Copyright (C) 2009 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. ``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 + * 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 "resource.h" +#include +#include +#include + +static LPTSTR getStringValue(HKEY key, LPCTSTR valueName) +{ + DWORD type = 0; + DWORD bufferSize = 0; + if (RegQueryValueEx(key, valueName, 0, &type, 0, &bufferSize) != ERROR_SUCCESS || type != REG_SZ) + return 0; + + LPTSTR buffer = static_cast(malloc(bufferSize)); + if (RegQueryValueEx(key, valueName, 0, &type, reinterpret_cast(buffer), &bufferSize) != ERROR_SUCCESS) { + free(buffer); + return 0; + } + + return buffer; +} + +static LPTSTR applePathFromRegistry(LPCTSTR key, LPCTSTR value) +{ + HKEY applePathKey = 0; + LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, key, 0, KEY_READ, &applePathKey); + if (error != ERROR_SUCCESS) + return 0; + LPTSTR path = getStringValue(applePathKey, value); + RegCloseKey(applePathKey); + return path; +} + +static LPTSTR safariInstallDir() +{ + return applePathFromRegistry(TEXT("SOFTWARE\\Apple Computer, Inc.\\Safari"), TEXT("InstallDir")); +} + +static LPTSTR safariBrowserExe() +{ + return applePathFromRegistry(TEXT("SOFTWARE\\Apple Computer, Inc.\\Safari"), TEXT("BrowserExe")); +} + +int APIENTRY _tWinMain(HINSTANCE instance, HINSTANCE, LPTSTR commandLine, int) +{ + LPTSTR path = safariInstallDir(); + LPTSTR browserExe = safariBrowserExe(); + if (!path || !browserExe) { + MessageBox(0, TEXT("Safari must be installed to run a WebKit nightly. You can download Safari from http://www.apple.com/safari/download"), TEXT("Safari not found"), MB_ICONSTOP); + return 1; + } + + // Set WEBKITNIGHTLY environment variable to point to the nightly bits + TCHAR exePath[MAX_PATH]; + if (!GetModuleFileName(0, exePath, ARRAYSIZE(exePath))) + return 1; + if (!PathRemoveFileSpec(exePath)) + return 1; + SetEnvironmentVariable(TEXT("WEBKITNIGHTLY"), exePath); + + // Launch Safari as a child process + STARTUPINFO startupInfo = {0}; + startupInfo.cb = sizeof(startupInfo); + PROCESS_INFORMATION processInfo = {0}; + if (!CreateProcess(browserExe, commandLine, 0, 0, FALSE, NORMAL_PRIORITY_CLASS | CREATE_UNICODE_ENVIRONMENT, 0, path, &startupInfo, &processInfo)) + MessageBox(0, TEXT("Safari could not be launched. Please make sure you have the latest version of Safari installed and try again. You can download Safari from http://www.apple.com/safari/download"), TEXT("Safari launch failed"), MB_ICONSTOP); + + free(browserExe); + free(path); + return 0; +}