Tools/WebKitLauncherWin/WebKitLauncherWin.cpp

   1 /*
   2  * Copyright (C) 2009 Apple Inc. All Rights Reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  * 1. Redistributions of source code must retain the above copyright
   8  *    notice, this list of conditions and the following disclaimer.
   9  * 2. Redistributions in binary form must reproduce the above copyright
  10  *    notice, this list of conditions and the following disclaimer in the
  11  *    documentation and/or other materials provided with the distribution.
  12  *
  13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "resource.h"
  27 #include <shlwapi.h>
  28 #include <string>
  29 #include <tchar.h>
  30 #include <vector>
  31 #include <windows.h>
  32
  33 using namespace std;
  34
  35 typedef basic_string<TCHAR> tstring;
  36
  37 static tstring getStringValue(HKEY key, const tstring& valueName)
  38 {
  39     DWORD type = 0;
  40     DWORD bufferSize = 0;
  41     if (RegQueryValueEx(key, valueName.c_str(), 0, &type, 0, &bufferSize) != ERROR_SUCCESS || type != REG_SZ)
  42         return tstring();
  43
  44     vector<TCHAR> buffer(bufferSize / sizeof(TCHAR));
  45     if (RegQueryValueEx(key, valueName.c_str(), 0, &type, reinterpret_cast<LPBYTE>(&buffer[0]), &bufferSize) != ERROR_SUCCESS)
  46         return tstring();
  47
  48     return &buffer[0];
  49 }
  50
  51 static tstring applePathFromRegistry(const tstring& key, const tstring& value)
  52 {
  53     HKEY applePathKey = 0;
  54     LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, key.c_str(), 0, KEY_READ, &applePathKey);
  55     if (error != ERROR_SUCCESS)
  56         return tstring();
  57     tstring path = getStringValue(applePathKey, value);
  58     RegCloseKey(applePathKey);
  59     return path;
  60 }
  61
  62 static tstring safariInstallDir()
  63 {
  64     return applePathFromRegistry(TEXT("SOFTWARE\\Apple Computer, Inc.\\Safari"), TEXT("InstallDir"));
  65 }
  66
  67 static tstring safariBrowserExe()
  68 {
  69     return applePathFromRegistry(TEXT("SOFTWARE\\Apple Computer, Inc.\\Safari"), TEXT("BrowserExe"));
  70 }
  71
  72 int APIENTRY _tWinMain(HINSTANCE instance, HINSTANCE, LPTSTR commandLine, int)
  73 {
  74     tstring path = safariInstallDir();
  75     tstring browserExe = safariBrowserExe();
  76     if (path.empty() || browserExe.empty()) {
  77         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);
  78         return 1;
  79     }
  80
  81     // Set WEBKITNIGHTLY environment variable to point to the nightly bits
  82     TCHAR exePath[MAX_PATH];
  83     if (!GetModuleFileName(0, exePath, ARRAYSIZE(exePath)))
  84         return 1;
  85     if (!PathRemoveFileSpec(exePath))
  86         return 1;
  87     SetEnvironmentVariable(TEXT("WEBKITNIGHTLY"), exePath);
  88
  89     tstring finalCommandLine = TEXT('"') + browserExe + TEXT("\" ") + commandLine;
  90
  91     // Launch Safari as a child process
  92     STARTUPINFO startupInfo = {0};
  93     startupInfo.cb = sizeof(startupInfo);
  94     PROCESS_INFORMATION processInfo = {0};
  95     if (!CreateProcess(browserExe.c_str(), const_cast<LPTSTR>(finalCommandLine.c_str()), 0, 0, FALSE, NORMAL_PRIORITY_CLASS | CREATE_UNICODE_ENVIRONMENT, 0, path.c_str(), &startupInfo, &processInfo))
  96         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);
  97
  98     return 0;
  99 }