1 /*
2 * Copyright (C) 2007, 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 *
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14 * its contributors may be used to endorse or promote products derived
15 * from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #include "resource.h"
30
31 #include <shlwapi.h>
32 #include <stdio.h>
33 #include <tchar.h>
34 #include <windows.h>
35
36 #define LOG(header, ...) \
37 do { \
38 _ftprintf(stderr, header); \
39 _ftprintf(stderr, __VA_ARGS__); \
40 } while (0)
41 #define LOG_WARNING(...) LOG(TEXT("WARNING: "), __VA_ARGS__)
42 #define LOG_ERROR(...) LOG(TEXT("ERROR: "), __VA_ARGS__)
43
getStringValue(HKEY key,LPCTSTR valueName)44 static TCHAR* getStringValue(HKEY key, LPCTSTR valueName)
45 {
46 DWORD type = 0;
47 DWORD bufferSize = 0;
48 if (RegQueryValueEx(key, valueName, 0, &type, 0, &bufferSize) != ERROR_SUCCESS || type != REG_SZ)
49 return 0;
50
51 TCHAR* buffer = (TCHAR*)malloc(bufferSize);
52 if (RegQueryValueEx(key, valueName, 0, &type, reinterpret_cast<LPBYTE>(buffer), &bufferSize) != ERROR_SUCCESS) {
53 free(buffer);
54 return 0;
55 }
56
57 return buffer;
58 }
59
getInstalledWebKitDirectory()60 static TCHAR* getInstalledWebKitDirectory()
61 {
62 LPCTSTR installPathKeyString = TEXT("SOFTWARE\\Apple Computer, Inc.\\Safari");
63 LPCTSTR installPathWin64KeyString = TEXT("SOFTWARE\\Wow6432Node\\Apple Computer, Inc.\\Safari");
64 HKEY installPathKey = 0;
65 LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, installPathKeyString, 0, KEY_READ, &installPathKey);
66 if (error != ERROR_SUCCESS)
67 error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, installPathWin64KeyString, 0, KEY_READ, &installPathKey);
68 if (error != ERROR_SUCCESS) {
69 LOG_WARNING(TEXT("Failed to open registry key %s\n"), installPathKeyString);
70 return 0;
71 }
72 LPTSTR webKitPath = getStringValue(installPathKey, TEXT("InstallDir"));
73 RegCloseKey(installPathKey);
74 if (!webKitPath) {
75 LOG_WARNING(TEXT("Couldn't retrieve value for registry key %s\n"), installPathKeyString);
76 return 0;
77 }
78 return webKitPath;
79 }
80
_tmain(int argc,TCHAR * argv[])81 int _tmain(int argc, TCHAR* argv[])
82 {
83 TCHAR* path = getInstalledWebKitDirectory();
84 if (!path) {
85 LOG_ERROR(TEXT("Couldn't determine installed Safari path\n"));
86 return 1;
87 }
88
89 bool printLauncher = false;
90 bool printEnvironment = false;
91 bool debugger = false;
92
93 for (int i = 1; i < argc; ++i) {
94 if (!_tcscmp(argv[i], TEXT("/printSafariLauncher"))) {
95 printLauncher = true;
96 continue;
97 }
98 if (!_tcscmp(argv[i], TEXT("/printSafariEnvironment"))) {
99 printEnvironment = true;
100 continue;
101 }
102 if (!_tcscmp(argv[i], TEXT("/debugger"))) {
103 debugger = true;
104 continue;
105 }
106 }
107
108 // printLauncher is inclusive of printEnvironment, so do not
109 // leave both enabled:
110 if (printLauncher && printEnvironment)
111 printEnvironment = false;
112
113 if (!printLauncher && !printEnvironment) {
114 _tprintf(TEXT("%s\n"), path);
115 free(path);
116 return 0;
117 }
118
119 LPCTSTR lines[] = {
120 TEXT("@echo off"),
121 TEXT("del /s /q \"%%TMP%%\\WebKitNightly\""),
122 TEXT("mkdir 2>NUL \"%%TMP%%\\WebKitNightly\\Safari.resources\""),
123 TEXT("mkdir 2>NUL \"%%TMP%%\\WebKitNightly\\WebKit.resources\""),
124 TEXT("mkdir 2>NUL \"%%TMP%%\\WebKitNightly\\JavaScriptCore.resources\""),
125 TEXT("xcopy /y /i /d \"%sSafari.exe\" \"%%TMP%%\\WebKitNightly\""),
126 TEXT("if exist \"%sSafari.dll\" xcopy /y /i /d \"%sSafari.dll\" \"%%TMP%%\\WebKitNightly\""),
127 TEXT("xcopy /y /i /d /e \"%sSafari.resources\" \"%%TMP%%\\WebKitNightly\\Safari.resources\""),
128 TEXT("xcopy /y /i /d /e \"%splugins\" \"%%TMP%%\\WebKitNightly\\plugins\""),
129 TEXT("xcopy /y /i /d WebKit.dll \"%%TMP%%\\WebKitNightly\""),
130 TEXT("xcopy /y /i /d WebKit.pdb \"%%TMP%%\\WebKitNightly\""),
131 TEXT("xcopy /y /i /d /e WebKit.resources \"%%TMP%%\\WebKitNightly\\WebKit.resources\""),
132 TEXT("xcopy /y /i /d JavaScriptCore.dll \"%%TMP%%\\WebKitNightly\""),
133 TEXT("xcopy /y /i /d JavaScriptCore.pdb \"%%TMP%%\\WebKitNightly\""),
134 TEXT("xcopy /y /i /d /e JavaScriptCore.resources \"%%TMP%%\\WebKitNightly\\JavaScriptCore.resources\""),
135 TEXT("set PATH=%%CD%%;%s;%%PATH%%"),
136 };
137
138 LPCTSTR command = TEXT("\"%TMP%\\WebKitNightly\\Safari.exe\"");
139
140 LPCTSTR launchLines[] = {
141 TEXT("%s"),
142 };
143
144 LPCTSTR debuggerLines[] = {
145 TEXT("if exist \"%%DevEnvDir%%\\VCExpress.exe\" ("),
146 TEXT("\"%%DevEnvDir%%\\VCExpress.exe\" /debugExe %s"),
147 TEXT(") else ("),
148 TEXT("\"%%DevEnvDir%%\\devenv.exe\" /debugExe %s"),
149 TEXT(")"),
150 };
151
152 for (int i = 0; i < ARRAYSIZE(lines); ++i) {
153 _tprintf(lines[i], path, path);
154 _tprintf(TEXT("\n"));
155 }
156
157 LPCTSTR* endLines = debugger ? debuggerLines : launchLines;
158
159 // Don't print launch command if we just want the environment set up...
160 if (!printEnvironment) {
161 for (unsigned i = 0; i < (debugger ? ARRAYSIZE(debuggerLines) : ARRAYSIZE(launchLines)); ++i) {
162 _tprintf(endLines[i], command);
163 _tprintf(TEXT("\n"));
164 }
165 }
166
167 free(path);
168 return 0;
169 }
170