2c33ca2480d309f87aac4abdc9c99e80793c2647
[WebKit-https.git] / WebKit / win / WebDataSource.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 "WebDataSource.h"
29
30 #include "IWebMutableURLRequest.h"
31 #include "MemoryStream.h"
32 #include "WebDocumentLoader.h"
33 #include "WebError.h"
34 #include "WebFrame.h"
35 #include "WebKit.h"
36 #include "WebHTMLRepresentation.h"
37 #include "WebKitStatisticsPrivate.h"
38 #include "WebMutableURLRequest.h"
39 #include "WebResource.h"
40 #include "WebURLResponse.h"
41
42 #pragma warning(push, 0)
43 #include <WebCore/BString.h>
44 #include <WebCore/DocLoader.h>
45 #include <WebCore/Document.h>
46 #include <WebCore/FrameLoader.h>
47 #include <WebCore/KURL.h>
48 #pragma warning(pop)
49
50 using namespace WebCore;
51
52 // WebDataSource ----------------------------------------------------------------
53
54 // {F230854D-7091-428a-8DB5-37CABA44C105}
55 const GUID IID_WebDataSource = 
56 { 0x5c2f9099, 0xe65e, 0x4a0f, { 0x9c, 0xa0, 0x6a, 0xd6, 0x92, 0x52, 0xa6, 0x2a } };
57
58 WebDataSource::WebDataSource(WebDocumentLoader* loader)
59     : m_refCount(0)
60     , m_loader(loader)
61 {
62     WebDataSourceCount++;
63     gClassCount++;
64 }
65
66 WebDataSource::~WebDataSource()
67 {
68     if (m_loader)
69         m_loader->detachDataSource();
70     WebDataSourceCount--;
71     gClassCount--;
72 }
73
74 WebDataSource* WebDataSource::createInstance(WebDocumentLoader* loader)
75 {
76     WebDataSource* instance = new WebDataSource(loader);
77     instance->AddRef();
78     return instance;
79 }
80
81 WebDocumentLoader* WebDataSource::documentLoader() const
82 {
83     return m_loader.get();
84 }
85
86 // IWebDataSourcePrivate ------------------------------------------------------
87
88 HRESULT STDMETHODCALLTYPE WebDataSource::overrideEncoding( 
89     /* [retval][out] */ BSTR* /*encoding*/)
90 {
91     ASSERT_NOT_REACHED();
92     return E_NOTIMPL;
93 }
94
95 HRESULT STDMETHODCALLTYPE WebDataSource::setOverrideEncoding( 
96     /* [in] */ BSTR /*encoding*/)
97 {
98     ASSERT_NOT_REACHED();
99     return E_NOTIMPL;
100 }
101
102 HRESULT STDMETHODCALLTYPE WebDataSource::mainDocumentError(
103         /* [retval][out] */ IWebError** error)
104 {
105     if (!error) {
106         ASSERT_NOT_REACHED();
107         return E_POINTER;
108     }
109
110     *error = 0;
111
112     if (!m_loader)
113         return E_FAIL;
114
115     if (m_loader->mainDocumentError().isNull())
116         return S_OK;
117
118     *error = WebError::createInstance(m_loader->mainDocumentError());
119     return S_OK;
120 }
121
122 // IUnknown -------------------------------------------------------------------
123
124 HRESULT STDMETHODCALLTYPE WebDataSource::QueryInterface(REFIID riid, void** ppvObject)
125 {
126     *ppvObject = 0;
127     if (IsEqualGUID(riid, IID_WebDataSource))
128         *ppvObject = this;
129     else if (IsEqualGUID(riid, IID_IUnknown))
130         *ppvObject = static_cast<IWebDataSource*>(this);
131     else if (IsEqualGUID(riid, IID_IWebDataSource))
132         *ppvObject = static_cast<IWebDataSource*>(this);
133     else if (IsEqualGUID(riid, IID_IWebDataSourcePrivate))
134         *ppvObject = static_cast<IWebDataSourcePrivate*>(this);
135     else
136         return E_NOINTERFACE;
137
138     AddRef();
139     return S_OK;
140 }
141
142 ULONG STDMETHODCALLTYPE WebDataSource::AddRef(void)
143 {
144     return ++m_refCount;
145 }
146
147 ULONG STDMETHODCALLTYPE WebDataSource::Release(void)
148 {
149     ULONG newRef = --m_refCount;
150     if (!newRef)
151         delete(this);
152
153     return newRef;
154 }
155
156 // IWebDataSource --------------------------------------------------------------
157
158 HRESULT STDMETHODCALLTYPE WebDataSource::initWithRequest( 
159     /* [in] */ IWebURLRequest* /*request*/)
160 {
161     ASSERT_NOT_REACHED();
162     return E_NOTIMPL;
163 }
164
165 HRESULT STDMETHODCALLTYPE WebDataSource::data( 
166     /* [retval][out] */ IStream** stream)
167 {
168     *stream = 0;
169     if (!m_loader)
170         return E_FAIL;
171
172     *stream = MemoryStream::createInstance(m_loader->mainResourceData());
173     return S_OK;
174 }
175
176 HRESULT STDMETHODCALLTYPE WebDataSource::representation( 
177     /* [retval][out] */ IWebDocumentRepresentation** rep)
178 {
179     HRESULT hr = S_OK;
180     if (!m_representation) {
181         WebHTMLRepresentation* htmlRep = WebHTMLRepresentation::createInstance(static_cast<WebFrame*>(m_loader->frameLoader()->client()));
182         hr = htmlRep->QueryInterface(IID_IWebDocumentRepresentation, (void**) &m_representation);
183         htmlRep->Release();
184     }
185
186     return m_representation.copyRefTo(rep);
187 }
188
189 HRESULT STDMETHODCALLTYPE WebDataSource::webFrame( 
190     /* [retval][out] */ IWebFrame** frame)
191 {
192     *frame = static_cast<WebFrame*>(m_loader->frameLoader()->client());
193     (*frame)->AddRef();
194     return S_OK;
195 }
196
197 HRESULT STDMETHODCALLTYPE WebDataSource::initialRequest( 
198     /* [retval][out] */ IWebURLRequest** request)
199 {
200     *request = WebMutableURLRequest::createInstance(m_loader->initialRequest());
201     return S_OK;
202 }
203
204 HRESULT STDMETHODCALLTYPE WebDataSource::request( 
205     /* [retval][out] */ IWebMutableURLRequest** request)
206 {
207     *request = WebMutableURLRequest::createInstance(m_loader->request());
208     return S_OK;
209 }
210
211 HRESULT STDMETHODCALLTYPE WebDataSource::response( 
212     /* [retval][out] */ IWebURLResponse** response)
213 {
214     *response = WebURLResponse::createInstance(m_loader->response());
215     return S_OK;
216 }
217
218 HRESULT STDMETHODCALLTYPE WebDataSource::textEncodingName( 
219     /* [retval][out] */ BSTR* name)
220 {
221     String encoding = m_loader->overrideEncoding();
222     if (encoding.isNull())
223         encoding = m_loader->response().textEncodingName();
224
225     *name = BString(encoding).release();
226
227     return S_OK;
228 }
229
230 HRESULT STDMETHODCALLTYPE WebDataSource::isLoading( 
231     /* [retval][out] */ BOOL* loading)
232 {
233     *loading = m_loader->isLoadingInAPISense();
234     return S_OK;
235 }
236
237 HRESULT STDMETHODCALLTYPE WebDataSource::pageTitle( 
238     /* [retval][out] */ BSTR* title)
239 {
240     *title = BString(m_loader->title()).release();
241     return S_OK;
242 }
243
244 HRESULT STDMETHODCALLTYPE WebDataSource::unreachableURL( 
245     /* [retval][out] */ BSTR* url)
246 {
247     KURL unreachableURL = m_loader->unreachableURL();
248     BString urlString((LPOLESTR)unreachableURL.url().unicode(), unreachableURL.url().length());
249
250     *url = urlString.release();
251     return S_OK;
252 }
253
254 HRESULT STDMETHODCALLTYPE WebDataSource::webArchive( 
255     /* [retval][out] */ IWebArchive** /*archive*/)
256 {
257     ASSERT_NOT_REACHED();
258     return E_NOTIMPL;
259 }
260
261 HRESULT STDMETHODCALLTYPE WebDataSource::mainResource( 
262     /* [retval][out] */ IWebResource** /*resource*/)
263 {
264     ASSERT_NOT_REACHED();
265     return E_NOTIMPL;
266 }
267
268 HRESULT STDMETHODCALLTYPE WebDataSource::subresources( 
269     /* [retval][out] */ IEnumVARIANT** /*result*/)
270 {
271     ASSERT_NOT_REACHED();
272     return E_NOTIMPL;
273 }
274
275 HRESULT STDMETHODCALLTYPE WebDataSource::subresourceForURL( 
276     /* [in] */ BSTR url,
277     /* [retval][out] */ IWebResource** resource)
278 {
279     if (!resource) {
280         ASSERT_NOT_REACHED();
281         return E_POINTER;
282     }
283
284     *resource = 0;
285
286     Document *doc = m_loader->frameLoader()->frame()->document();
287
288     if (!doc)
289         return E_FAIL;
290
291     CachedResource *cachedResource = doc->docLoader()->cachedResource(String(url));
292
293     if (!cachedResource)
294         return E_FAIL;
295
296     *resource = WebResource::createInstance(cachedResource->data(), cachedResource->response());
297     return S_OK;
298 }
299
300 HRESULT STDMETHODCALLTYPE WebDataSource::addSubresource( 
301     /* [in] */ IWebResource* /*subresource*/)
302 {
303     ASSERT_NOT_REACHED();
304     return E_NOTIMPL;
305 }