Reviewed by Adam.
[WebKit-https.git] / WebKit / win / WebKitClassFactory.cpp
1 /*
2  * Copyright (C) 2006, 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  * 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 "WebKitDLL.h"
28 #include "WebKitClassFactory.h"
29
30 #include "ForEachCoClass.h"
31 #include "CFDictionaryPropertyBag.h"
32 #include "WebCache.h"
33 #include "WebDatabaseManager.h"
34 #include "WebDownload.h"
35 #include "WebError.h"
36 #include "WebFrame.h"
37 #include "WebHistory.h"
38 #include "WebHistoryItem.h"
39 #include "WebIconDatabase.h"
40 #include "WebJavaScriptCollector.h"
41 #include "WebKit.h"
42 #include "WebScrollBar.h"
43 #include "WebKitStatistics.h"
44 #include "WebMutableURLRequest.h"
45 #include "WebNotificationCenter.h"
46 #include "WebPreferences.h"
47 #include "WebTextRenderer.h"
48 #include "WebURLCredential.h"
49 #include "WebURLProtectionSpace.h"
50 #include "WebURLResponse.h"
51 #include "WebScriptDebugServer.h"
52 #include "WebView.h"
53 #include <SafariTheme/SafariTheme.h>
54 #pragma warning(push, 0)
55 #include <WebCore/FontDatabase.h>
56 #pragma warning(pop)
57
58 // WebKitClassFactory ---------------------------------------------------------
59
60 typedef void (APIENTRY*STInitializePtr)();
61
62 WebKitClassFactory::WebKitClassFactory(CLSID targetClass)
63 : m_targetClass(targetClass)
64 , m_refCount(0)
65 {
66     static bool didInitializeSafariTheme;
67     if (!didInitializeSafariTheme) {
68         if (HMODULE module = LoadLibrary(SAFARITHEMEDLL))
69             if (STInitializePtr stInit = (STInitializePtr)GetProcAddress(module, "STInitialize"))
70                 stInit();
71         didInitializeSafariTheme = true;
72     }
73
74     WebCore::populateFontDatabase();
75
76     gClassCount++;
77 }
78
79 WebKitClassFactory::~WebKitClassFactory()
80 {
81     gClassCount--;
82 }
83
84 // IUnknown -------------------------------------------------------------------
85
86 HRESULT STDMETHODCALLTYPE WebKitClassFactory::QueryInterface(REFIID riid, void** ppvObject)
87 {
88     *ppvObject = 0;
89     if (IsEqualGUID(riid, IID_IUnknown))
90         *ppvObject = static_cast<IUnknown*>(this);
91     else if (IsEqualGUID(riid, IID_IClassFactory))
92         *ppvObject = static_cast<IClassFactory*>(this);
93     else
94         return E_NOINTERFACE;
95
96     AddRef();
97     return S_OK;
98 }
99
100 ULONG STDMETHODCALLTYPE WebKitClassFactory::AddRef(void)
101 {
102     return ++m_refCount;
103 }
104
105 ULONG STDMETHODCALLTYPE WebKitClassFactory::Release(void)
106 {
107     ULONG newRef = --m_refCount;
108     if (!newRef && !gLockCount)
109         delete(this);
110
111     return newRef;
112 }
113
114 // IClassFactory --------------------------------------------------------------
115
116 HRESULT STDMETHODCALLTYPE WebKitClassFactory::CreateInstance(IUnknown* pUnkOuter, REFIID riid, void** ppvObject)
117 {
118     IUnknown* unknown = 0;
119     *ppvObject = 0;
120     
121     if (pUnkOuter)
122         return CLASS_E_NOAGGREGATION;
123
124 #define INITIALIZE_IF_CLASS(cls) \
125     if (IsEqualGUID(m_targetClass, CLSID_##cls)) \
126         unknown = static_cast<I##cls*>(cls::createInstance()); \
127     else \
128     // end of macro
129
130     // These #defines are needed to appease the INITIALIZE_IF_CLASS macro.
131     // There is no ICFDictionaryPropertyBag, we use IPropertyBag instead.
132 #define ICFDictionaryPropertyBag IPropertyBag
133     // There is no IWebScrollBar, we only have IWebScrollBarPrivate.
134 #define IWebScrollBar IWebScrollBarPrivate
135     // There is no class called WebURLRequest -- WebMutableURLRequest implements it for us.
136 #define WebURLRequest WebMutableURLRequest
137
138     FOR_EACH_COCLASS(INITIALIZE_IF_CLASS)
139         // This is the final else case
140         return CLASS_E_CLASSNOTAVAILABLE;
141
142 #undef ICFDictionaryPropertyBag
143 #undef IWebScrollBar
144 #undef WebURLRequest
145 #undef INITIALIZE_IF_CLASS
146
147     if (!unknown)
148         return E_OUTOFMEMORY;
149
150     HRESULT hr = unknown->QueryInterface(riid, ppvObject);
151     if (FAILED(hr))
152         delete unknown;
153     else
154         unknown->Release(); // both createInstance() and QueryInterface() added refs
155
156     return hr;
157 }
158
159 HRESULT STDMETHODCALLTYPE WebKitClassFactory::LockServer(BOOL fLock)
160 {
161     if (fLock)
162         gLockCount++;
163     else
164         gLockCount--;
165
166     return S_OK;
167 }