<commit_before>/*
 * Copyright (C) 2007 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.
 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
 *     its contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "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 OR ITS 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 "WebKitInitializer.h"

// Needed to get SetDllDirectory
#define _WIN32_WINNT 0x0502

#include <shlwapi.h>
#include <tchar.h>
#include <windows.h>

static TCHAR* 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;

    TCHAR* buffer = new TCHAR[bufferSize / sizeof(TCHAR)];
    if (RegQueryValueEx(key, 0, 0, &type, reinterpret_cast<LPBYTE>(buffer), &bufferSize) != ERROR_SUCCESS) {
        delete [] buffer;
        return 0;
    }

    return buffer;
}

static LPOLESTR getWebViewCLSID()
{
    // FIXME <rdar://5505062>: Once WebKit switches to truly version-independent
    // ProgIDs, this should just become "WebKit.WebView".
    LPCTSTR webViewProgID = TEXT("WebKit.WebView.3");

    CLSID clsid = CLSID_NULL;
    HRESULT hr = CLSIDFromProgID(webViewProgID, &clsid);
    if (FAILED(hr)) {
        _ftprintf(stderr, TEXT("Failed to get CLSID for %s\n"), webViewProgID);
        return 0;
    }

    LPOLESTR clsidString = 0;
    if (FAILED(StringFromCLSID(clsid, &clsidString))) {
        _ftprintf(stderr, TEXT("Failed to get string representation of CLSID for WebView\n"));
        return 0;
    }

    return clsidString;
}

static TCHAR* getInstalledWebKitDirectory()
{
    LPCTSTR keyPrefix = TEXT("SOFTWARE\\Classes\\CLSID\\");
    LPCTSTR keySuffix = TEXT("\\InprocServer32");

    LPOLESTR clsid = getWebViewCLSID();
    if (!clsid)
        return 0;

    size_t keyBufferLength = _tcslen(keyPrefix) + _tcslen(clsid) + _tcslen(keySuffix) + 1;
    TCHAR* keyString = new TCHAR[keyBufferLength];

    int ret = _sntprintf_s(keyString, keyBufferLength, keyBufferLength - 1, TEXT("%s%s%s"), keyPrefix, clsid, keySuffix);
    CoTaskMemFree(clsid);
    if (ret == -1) {
        _ftprintf(stderr, TEXT("Failed to construct InprocServer32 key\n"));
        return 0;
    }

    HKEY serverKey = 0;
    LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, keyString, 0, KEY_READ, &serverKey);
    delete [] keyString;
    if (error != ERROR_SUCCESS) {
        _ftprintf(stderr, TEXT("Failed to open registry key %s\n"), keyString);
        return 0;
    }

    TCHAR* webKitPath = getStringValue(serverKey, 0);
    RegCloseKey(serverKey);
    if (!webKitPath) {
        _ftprintf(stderr, TEXT("Couldn't retrieve value for registry key %s\n"), keyString);
        return 0;
    }

    TCHAR* startOfFileName = PathFindFileName(webKitPath);
    if (startOfFileName == webKitPath) {
        _ftprintf(stderr, TEXT("Couldn't find filename from path %s\n"), webKitPath);
        delete [] webKitPath;
        return 0;
    }

    *startOfFileName = '\0';
    return webKitPath;
}

bool initializeWebKit()
{
    static bool haveInitialized;
    static bool success;
    if (haveInitialized)
        return success;
    haveInitialized = true;

#ifdef NDEBUG
    LPCTSTR webKitDLL = TEXT("WebKit.dll");
#else
    LPCTSTR webKitDLL = TEXT("WebKit_debug.dll");
#endif

    HMODULE webKitModule = LoadLibrary(webKitDLL);
    if (!webKitModule) {
        _ftprintf(stderr, TEXT("LoadLibrary(%s) failed\n"), webKitDLL);
        return false;
    }

    FARPROC dllRegisterServer = GetProcAddress(webKitModule, "DllRegisterServer");
    if (!dllRegisterServer) {
        _ftprintf(stderr, TEXT("GetProcAddress(\"DllRegisterServer\") failed\n"));
        return false;
    }

    dllRegisterServer();

    TCHAR* directory = getInstalledWebKitDirectory();
    if (!directory) {
        _ftprintf(stderr, TEXT("Couldn't determine installed WebKit directory\n"));
        return false;<commit_msg>Touch this file so it will rebuild<commit_after>/*
 * Copyright (C) 2007 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.
 * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
 *     its contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "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 OR ITS 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 "WebKitInitializer.h"

// Needed to get SetDllDirectory
#define _WIN32_WINNT 0x0502

#include <shlwapi.h>
#include <tchar.h>
#include <windows.h>

static TCHAR* 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;

    TCHAR* buffer = new TCHAR[bufferSize / sizeof(TCHAR)];
    if (RegQueryValueEx(key, 0, 0, &type, reinterpret_cast<LPBYTE>(buffer), &bufferSize) != ERROR_SUCCESS) {
        delete [] buffer;
        return 0;
    }

    return buffer;
}

static LPOLESTR getWebViewCLSID()
{
    // FIXME <rdar://5505062>: Once WebKit switches to truly version-independent
    // ProgIDs, this should just become "WebKit.WebView".
    LPCTSTR webViewProgID = TEXT("WebKit.WebView.3");

    CLSID clsid = CLSID_NULL;
    HRESULT hr = CLSIDFromProgID(webViewProgID, &clsid);
    if (FAILED(hr)) {
        _ftprintf(stderr, TEXT("Failed to get CLSID for %s\n"), webViewProgID);
        return 0;
    }

    LPOLESTR clsidString = 0;
    if (FAILED(StringFromCLSID(clsid, &clsidString))) {
        _ftprintf(stderr, TEXT("Failed to get string representation of CLSID for WebView\n"));
        return 0;
    }

    return clsidString;
}

static TCHAR* getInstalledWebKitDirectory()
{
    LPCTSTR keyPrefix = TEXT("SOFTWARE\\Classes\\CLSID\\");
    LPCTSTR keySuffix = TEXT("\\InprocServer32");

    LPOLESTR clsid = getWebViewCLSID();
    if (!clsid)
        return 0;

    size_t keyBufferLength = _tcslen(keyPrefix) + _tcslen(clsid) + _tcslen(keySuffix) + 1;
    TCHAR* keyString = new TCHAR[keyBufferLength];

    int ret = _sntprintf_s(keyString, keyBufferLength, keyBufferLength - 1, TEXT("%s%s%s"), keyPrefix, clsid, keySuffix);
    CoTaskMemFree(clsid);
    if (ret == -1) {
        _ftprintf(stderr, TEXT("Failed to construct InprocServer32 key\n"));
        return 0;
    }

    HKEY serverKey = 0;
    LONG error = RegOpenKeyEx(HKEY_LOCAL_MACHINE, keyString, 0, KEY_READ, &serverKey);
    delete [] keyString;
    if (error != ERROR_SUCCESS) {
        _ftprintf(stderr, TEXT("Failed to open registry key %s\n"), keyString);
        return 0;
    }

    TCHAR* webKitPath = getStringValue(serverKey, 0);
    RegCloseKey(serverKey);
    if (!webKitPath) {
        _ftprintf(stderr, TEXT("Couldn't retrieve value for registry key %s\n"), keyString);
        return 0;
    }

    TCHAR* startOfFileName = PathFindFileName(webKitPath);
    if (startOfFileName == webKitPath) {
        _ftprintf(stderr, TEXT("Couldn't find filename from path %s\n"), webKitPath);
        delete [] webKitPath;
        return 0;
    }

    *startOfFileName = '\0';
    return webKitPath;
}

bool initializeWebKit()
{
    static bool haveInitialized;
    static bool success;
    if (haveInitialized)
        return success;
    haveInitialized = true;

#ifdef NDEBUG
    LPCTSTR webKitDLL = TEXT("WebKit.dll");
#else
    LPCTSTR webKitDLL = TEXT("WebKit_debug.dll");
#endif

    HMODULE webKitModule = LoadLibrary(webKitDLL);
    if (!webKitModule) {
        _ftprintf(stderr, TEXT("LoadLibrary(%s) failed\n"), webKitDLL);
        return false;
    }

    FARPROC dllRegisterServer = GetProcAddress(webKitModule, "DllRegisterServer");
    if (!dllRegisterServer) {
        _ftprintf(stderr, TEXT("GetProcAddress(webKitModule, \"DllRegisterServer\") failed\n"));
        return false;
    }

    dllRegisterServer();

    TCHAR* directory = getInstalledWebKitDirectory();
    if (!directory) {
        _ftprintf(stderr, TEXT("Couldn't determine installed WebKit directory\n"));
        return false;
{'fraction_non_alphanumeric': 0.0986371866958925, 'fraction_numerical': 0.009339559706470981, 'mean_word_length': 3.5232758620689655, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 13, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '3515677', 'n_tokens_mistral': 3354, 'n_tokens_neox': 3119, 'n_words': 1034}