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};
48
49static COMPtr<IClassFactory> classFactory(const CLSID& clsid)
50{
51    typedef HashMap<CLSID, COMPtr<IClassFactory>, CLSIDHash, CLSIDHashTraits> FactoryMap;
52    static FactoryMap& factories = *new FactoryMap;
53
54    pair<FactoryMap::iterator, bool> result = factories.add(clsid, 0);
55    COMPtr<IClassFactory>& factory = result.first->second;
56    bool added = result.second;
57    if (added && FAILED(DllGetClassObject(clsid, __uuidof(factory), reinterpret_cast<void**>(&factory))))
58        factory = 0;
59
60    return factory;
61}
62
63HRESULT WebKitCreateInstance(REFCLSID rclsid, IUnknown* pUnkOuter, REFIID riid, void** ppvObject)
64{
65    COMPtr<IClassFactory> factory = classFactory(rclsid);
66    if (!factory)
67        return REGDB_E_CLASSNOTREG;
68
69    return factory->CreateInstance(pUnkOuter, riid, ppvObject);
70}
71