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/DocLoader.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()).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->docLoader()->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 }