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 

    34 struct 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 

    44 struct 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 

    49 static 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 

    63 HRESULT 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 }