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     _tprintf(TEXT("%s"), path);
 134     free(path);
 135     return 0;
 136 }