Source/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 "WebKit.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 #include <WebCore/BString.h>
  42 #include <WebCore/CachedResourceLoader.h>
  43 #include <WebCore/Document.h>
  44 #include <WebCore/Frame.h>
  45 #include <WebCore/FrameLoader.h>
  46 #include <WebCore/KURL.h>
  47
  48 using namespace WebCore;
  49
  50 // WebDataSource ----------------------------------------------------------------
  51
  52 // {F230854D-7091-428a-8DB5-37CABA44C105}
  53 const GUID IID_WebDataSource =
  54 { 0x5c2f9099, 0xe65e, 0x4a0f, { 0x9c, 0xa0, 0x6a, 0xd6, 0x92, 0x52, 0xa6, 0x2a } };
  55
  56 WebDataSource::WebDataSource(WebDocumentLoader* loader)
  57     : m_refCount(0)
  58     , m_loader(loader)
  59 {
  60     WebDataSourceCount++;
  61     gClassCount++;
  62     gClassNameCount.add("WebDataSource");
  63 }
  64
  65 WebDataSource::~WebDataSource()
  66 {
  67     if (m_loader)
  68         m_loader->detachDataSource();
  69     WebDataSourceCount--;
  70     gClassCount--;
  71     gClassNameCount.remove("WebDataSource");
  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 HRESULT STDMETHODCALLTYPE WebDataSource::setDeferMainResourceDataLoad(
 123     /* [in] */ BOOL flag)
 124 {
 125     if (!m_loader)
 126         return E_FAIL;
 127
 128     m_loader->setDeferMainResourceDataLoad(flag);
 129     return S_OK;
 130 }
 131
 132 // IUnknown -------------------------------------------------------------------
 133
 134 HRESULT STDMETHODCALLTYPE WebDataSource::QueryInterface(REFIID riid, void** ppvObject)
 135 {
 136     *ppvObject = 0;
 137     if (IsEqualGUID(riid, IID_WebDataSource))
 138         *ppvObject = this;
 139     else if (IsEqualGUID(riid, IID_IUnknown))
 140         *ppvObject = static_cast<IWebDataSource*>(this);
 141     else if (IsEqualGUID(riid, IID_IWebDataSource))
 142         *ppvObject = static_cast<IWebDataSource*>(this);
 143     else if (IsEqualGUID(riid, IID_IWebDataSourcePrivate))
 144         *ppvObject = static_cast<IWebDataSourcePrivate*>(this);
 145     else
 146         return E_NOINTERFACE;
 147
 148     AddRef();
 149     return S_OK;
 150 }
 151
 152 ULONG STDMETHODCALLTYPE WebDataSource::AddRef(void)
 153 {
 154     return ++m_refCount;
 155 }
 156
 157 ULONG STDMETHODCALLTYPE WebDataSource::Release(void)
 158 {
 159     ULONG newRef = --m_refCount;
 160     if (!newRef)
 161         delete(this);
 162
 163     return newRef;
 164 }
 165
 166 // IWebDataSource --------------------------------------------------------------
 167
 168 HRESULT STDMETHODCALLTYPE WebDataSource::initWithRequest(
 169     /* [in] */ IWebURLRequest* /*request*/)
 170 {
 171     ASSERT_NOT_REACHED();
 172     return E_NOTIMPL;
 173 }
 174
 175 HRESULT STDMETHODCALLTYPE WebDataSource::data(
 176     /* [retval][out] */ IStream** stream)
 177 {
 178     *stream = 0;
 179     if (!m_loader)
 180         return E_FAIL;
 181
 182     return MemoryStream::createInstance(m_loader->mainResourceData()).copyRefTo(stream);
 183 }
 184
 185 HRESULT STDMETHODCALLTYPE WebDataSource::representation(
 186     /* [retval][out] */ IWebDocumentRepresentation** rep)
 187 {
 188     HRESULT hr = S_OK;
 189     if (!m_representation) {
 190         WebHTMLRepresentation* htmlRep = WebHTMLRepresentation::createInstance(static_cast<WebFrame*>(m_loader->frameLoader()->client()));
 191         hr = htmlRep->QueryInterface(IID_IWebDocumentRepresentation, (void**) &m_representation);
 192         htmlRep->Release();
 193     }
 194
 195     return m_representation.copyRefTo(rep);
 196 }
 197
 198 HRESULT STDMETHODCALLTYPE WebDataSource::webFrame(
 199     /* [retval][out] */ IWebFrame** frame)
 200 {
 201     *frame = static_cast<WebFrame*>(m_loader->frameLoader()->client());
 202     (*frame)->AddRef();
 203     return S_OK;
 204 }
 205
 206 HRESULT STDMETHODCALLTYPE WebDataSource::initialRequest(
 207     /* [retval][out] */ IWebURLRequest** request)
 208 {
 209     *request = WebMutableURLRequest::createInstance(m_loader->originalRequest());
 210     return S_OK;
 211 }
 212
 213 HRESULT STDMETHODCALLTYPE WebDataSource::request(
 214     /* [retval][out] */ IWebMutableURLRequest** request)
 215 {
 216     *request = WebMutableURLRequest::createInstance(m_loader->request());
 217     return S_OK;
 218 }
 219
 220 HRESULT STDMETHODCALLTYPE WebDataSource::response(
 221     /* [retval][out] */ IWebURLResponse** response)
 222 {
 223     *response = WebURLResponse::createInstance(m_loader->response());
 224     return S_OK;
 225 }
 226
 227 HRESULT STDMETHODCALLTYPE WebDataSource::textEncodingName(
 228     /* [retval][out] */ BSTR* name)
 229 {
 230     String encoding = m_loader->overrideEncoding();
 231     if (encoding.isNull())
 232         encoding = m_loader->response().textEncodingName();
 233
 234     *name = BString(encoding).release();
 235
 236     return S_OK;
 237 }
 238
 239 HRESULT STDMETHODCALLTYPE WebDataSource::isLoading(
 240     /* [retval][out] */ BOOL* loading)
 241 {
 242     *loading = m_loader->isLoadingInAPISense();
 243     return S_OK;
 244 }
 245
 246 HRESULT STDMETHODCALLTYPE WebDataSource::pageTitle(
 247     /* [retval][out] */ BSTR* title)
 248 {
 249     *title = BString(m_loader->title().string()).release();
 250     return S_OK;
 251 }
 252
 253 HRESULT STDMETHODCALLTYPE WebDataSource::unreachableURL(
 254     /* [retval][out] */ BSTR* url)
 255 {
 256     KURL unreachableURL = m_loader->unreachableURL();
 257     BString urlString((LPOLESTR)unreachableURL.string().characters(), unreachableURL.string().length());
 258
 259     *url = urlString.release();
 260     return S_OK;
 261 }
 262
 263 HRESULT STDMETHODCALLTYPE WebDataSource::webArchive(
 264     /* [retval][out] */ IWebArchive** /*archive*/)
 265 {
 266     ASSERT_NOT_REACHED();
 267     return E_NOTIMPL;
 268 }
 269
 270 HRESULT STDMETHODCALLTYPE WebDataSource::mainResource(
 271     /* [retval][out] */ IWebResource** /*resource*/)
 272 {
 273     ASSERT_NOT_REACHED();
 274     return E_NOTIMPL;
 275 }
 276
 277 HRESULT STDMETHODCALLTYPE WebDataSource::subresources(
 278     /* [retval][out] */ IEnumVARIANT** /*result*/)
 279 {
 280     ASSERT_NOT_REACHED();
 281     return E_NOTIMPL;
 282 }
 283
 284 HRESULT STDMETHODCALLTYPE WebDataSource::subresourceForURL(
 285     /* [in] */ BSTR url,
 286     /* [retval][out] */ IWebResource** resource)
 287 {
 288     if (!resource) {
 289         ASSERT_NOT_REACHED();
 290         return E_POINTER;
 291     }
 292
 293     *resource = 0;
 294
 295     Document *doc = m_loader->frameLoader()->frame()->document();
 296
 297     if (!doc)
 298         return E_FAIL;
 299
 300     CachedResource *cachedResource = doc->cachedResourceLoader()->cachedResource(String(url));
 301
 302     if (!cachedResource)
 303         return E_FAIL;
 304
 305     *resource = WebResource::createInstance(cachedResource->data(), cachedResource->response());
 306     return S_OK;
 307 }
 308
 309 HRESULT STDMETHODCALLTYPE WebDataSource::addSubresource(
 310     /* [in] */ IWebResource* /*subresource*/)
 311 {
 312     ASSERT_NOT_REACHED();
 313     return E_NOTIMPL;
 314 }