12fa35be8db85ea7a8477ee299f3db15cfa041e0
[WebKit-https.git] / WebKitTools / FindSafari / FindSafari.cpp
1 /*
2  * Copyright (C) 2007 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 <shlwapi.h>
30 #include <stdio.h>
31 #include <tchar.h>
32 #include <windows.h>
33
34 #define LOG(header, ...) \
35     do { \
36         _ftprintf(stderr, header); \
37         _ftprintf(stderr, __VA_ARGS__); \
38     } while (0)
39 #define LOG_WARNING(...) LOG(TEXT("WARNING: "), __VA_ARGS__)
40 #define LOG_ERROR(...) LOG(TEXT("ERROR: "), __VA_ARGS__)
41
42 static TCHAR* getStringValue(HKEY key, LPCTSTR valueName)
43 {
44     DWORD type = 0;
45     DWORD bufferSize = 0;
46     if (RegQueryValueEx(key, valueName, 0, &type, 0, &bufferSize) != ERROR_SUCCESS || type != REG_SZ)
47         return 0;
48
49     TCHAR* buffer = (TCHAR*)malloc(bufferSize);
50     if (RegQueryValueEx(key, 0, 0, &type, reinterpret_cast<LPBYTE>(buffer), &bufferSize) != ERROR_SUCCESS) {
51         free(buffer);
52         return 0;
53     }
54
55     return buffer;
56 }
57
58 static LPOLESTR getWebViewCLSID()
59 {
60     // FIXME <rdar://5505062>: Once WebKit switches to truly version-independent
61     // ProgIDs, this should just become "WebKit.WebView".
62     LPCTSTR webViewProgID = TEXT("WebKit.WebView.3");
63
64     CLSID clsid = CLSID_NULL;
65     HRESULT hr = CLSIDFromProgID(webViewProgID, &clsid);
66     if (FAILED(hr)) {
67         LOG_WARNING(TEXT("Failed to get CLSID for %s\n"), webViewProgID);
68         return 0;
69     }
70
71     LPOLESTR clsidString = 0;
72     if (FAILED(StringFromCLSID(clsid, &clsidString))) {
73         LOG_WARNING(TEXT("Failed to get string representation of CLSID for WebView\n"));
74         return 0;
75     }
76
77     return clsidString;
78 }
79
80 static TCHAR* getInstalledWebKitDirectory()
81 {
82     LPCTSTR keyPrefix = TEXT("SOFTWARE\\Classes\\CLSID\\");
83     LPCTSTR keySuffix = TEXT("\\InprocServer32");
84
85     LPOLESTR clsid = getWebViewCLSID();
86     if (!clsid)
87         return 0;
88
89     size_t keyBufferLength = _tcslen(keyPrefix) + _tcslen(clsid) + _tcslen(keySuffix) + 1;
90     TCHAR* keyString = (TCHAR*)malloc(keyBufferLength * sizeof(TCHAR));
91
92     int ret = _sntprintf_s(keyString, keyBufferLength, keyBufferLength - 1, TEXT("%s%s%s"), keyPrefix, clsid, keySuffix);
93     CoTaskMemFree(clsid);
94     if (ret == -1) {
95         LOG_WARNING(TEXT("Failed to construct InprocServer32 key\n"));
96         return 0;
97     }
98
99     HKEY serverKey = 0;
100     LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, keyString, 0, KEY_READ, &serverKey);
101     free(keyString);
102     if (error != ERROR_SUCCESS) {
103         LOG_WARNING(TEXT("Failed to open registry key %s\n"), keyString);
104         return 0;
105     }
106
107     TCHAR* webKitPath = getStringValue(serverKey, 0);
108     RegCloseKey(serverKey);
109     if (!webKitPath) {
110         LOG_WARNING(TEXT("Couldn't retrieve value for registry key %s\n"), keyString);
111         return 0;
112     }
113
114     TCHAR* startOfFileName = PathFindFileName(webKitPath);
115     if (startOfFileName == webKitPath) {
116         LOG_WARNING(TEXT("Couldn't find filename from path %s\n"), webKitPath);
117         free(webKitPath);
118         return 0;
119     }
120
121     *startOfFileName = '\0';
122     return webKitPath;
123 }
124
125 int _tmain(int argc, TCHAR* argv[])
126 {
127     TCHAR* path = getInstalledWebKitDirectory();
128     if (!path) {
129         LOG_ERROR(TEXT("Couldn't determine installed Safari path\n"));
130         return 1;
131     }
132
133     if (argc < 2 || _tcscmp(argv[1], TEXT("/printSafariLauncher"))) {
134         _tprintf(TEXT("%s"), path);
135         free(path);
136         return 0;
137     }
138
139     TCHAR* lines[] = {
140         TEXT("@echo off"),
141         TEXT("mkdir 2>NUL \"%%TMP%%\\WebKitNightly\\Safari.resources\""),
142         TEXT("xcopy /y /i /d \"%sSafari.exe\" \"%%TMP%%\\WebKitNightly\""),
143         TEXT("xcopy /y /i /d /e \"%sSafari.resources\" \"%%TMP%%\\WebKitNightly\\Safari.resources\""),
144         TEXT("set PATH=\"%%CD%%;%s;%%PATH%%\""),
145         TEXT("\"%%TMP%%\\WebKitNightly\\Safari.exe\" /customWebKit"),
146     };
147
148     for (int i = 0; i < ARRAYSIZE(lines); ++i) {
149         _tprintf(lines[i], path);
150         _tprintf(TEXT("\n"));
151     }
152
153     free(path);
154     return 0;
155 }