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 COMPUTER, 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 COMPUTER, 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 "config.h"
27#include "WebKitCOMAPI.h"
28
29#include "WebKit.h"
30#include "WebKitDLL.h"
31
32#include <WebCore/COMPtr.h>
33
34struct CLSIDHash {
35    static unsigned hash(const CLSID& clsid)
36    {
37        RPC_STATUS status;
38        return ::UuidHash(const_cast<CLSID*>(&clsid), &status);
39    }
40    static bool equal(const CLSID& a, const CLSID& b) { return ::IsEqualCLSID(a, b); }
41    static const bool safeToCompareToEmptyOrDeleted = true;
42};
43
44struct CLSIDHashTraits : WTF::GenericHashTraits<CLSID> {
45    static void constructDeletedValue(CLSID& slot) { slot = CLSID_NULL; }
46    static bool isDeletedValue(const CLSID& value) { return value == CLSID_NULL; }
47    // FIXME: This is a work around for the regression introducing in r130643 when running DRT in debug mode on Windows.
48    // Expanding the size of the hash table causes an assertion failure ASSERT(!isDeletedBucket) when reinserting items into the new table,
49    // presumably due to the collision issues described in r132302.
50    // This work around avoids the issue entirely simply by making sure that the table will not have to resize in running DRT.
51    static const int minimumTableSize = 64;
52};
53
54static COMPtr<IClassFactory> classFactory(const CLSID& clsid)
55{
56    typedef HashMap<CLSID, COMPtr<IClassFactory>, CLSIDHash, CLSIDHashTraits> FactoryMap;
57    static FactoryMap& factories = *new FactoryMap;
58
59    FactoryMap::AddResult result = factories.add(clsid, 0);
60    COMPtr<IClassFactory>& factory = result.iterator->value;
61    if (result.isNewEntry && FAILED(DllGetClassObject(clsid, __uuidof(factory), reinterpret_cast<void**>(&factory))))
62        factory = 0;
63
64    return factory;
65}
66
67HRESULT WebKitCreateInstance(REFCLSID rclsid, IUnknown* pUnkOuter, REFIID riid, void** ppvObject)
68{
69    COMPtr<IClassFactory> factory = classFactory(rclsid);
70    if (!factory)
71        return REGDB_E_CLASSNOTREG;
72
73    return factory->CreateInstance(pUnkOuter, riid, ppvObject);
74}
75