--- /dev/null Thu Jan 01 00:00:00 1970 +0000
+++ b/WebKit/win/WebDataSource.cpp Fri Sep 17 09:02:29 2010 +0300
@@ -0,0 +1,314 @@
+/*
+ * Copyright (C) 2006, 2007 Apple Inc. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ * notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ * notice, this list of conditions and the following disclaimer in the
+ * documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY APPLE COMPUTER, INC. ``AS IS'' AND ANY
+ * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE COMPUTER, INC. OR
+ * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+ * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+ * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+ * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
+ * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
+ * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "config.h"
+#include "WebKitDLL.h"
+#include "WebDataSource.h"
+
+#include "WebKit.h"
+#include "MemoryStream.h"
+#include "WebDocumentLoader.h"
+#include "WebError.h"
+#include "WebFrame.h"
+#include "WebKit.h"
+#include "WebHTMLRepresentation.h"
+#include "WebKitStatisticsPrivate.h"
+#include "WebMutableURLRequest.h"
+#include "WebResource.h"
+#include "WebURLResponse.h"
+#include <WebCore/BString.h>
+#include <WebCore/DocLoader.h>
+#include <WebCore/Document.h>
+#include <WebCore/Frame.h>
+#include <WebCore/FrameLoader.h>
+#include <WebCore/KURL.h>
+
+using namespace WebCore;
+
+// WebDataSource ----------------------------------------------------------------
+
+// {F230854D-7091-428a-8DB5-37CABA44C105}
+const GUID IID_WebDataSource =
+{ 0x5c2f9099, 0xe65e, 0x4a0f, { 0x9c, 0xa0, 0x6a, 0xd6, 0x92, 0x52, 0xa6, 0x2a } };
+
+WebDataSource::WebDataSource(WebDocumentLoader* loader)
+ : m_refCount(0)
+ , m_loader(loader)
+{
+ WebDataSourceCount++;
+ gClassCount++;
+ gClassNameCount.add("WebDataSource");
+}
+
+WebDataSource::~WebDataSource()
+{
+ if (m_loader)
+ m_loader->detachDataSource();
+ WebDataSourceCount--;
+ gClassCount--;
+ gClassNameCount.remove("WebDataSource");
+}
+
+WebDataSource* WebDataSource::createInstance(WebDocumentLoader* loader)
+{
+ WebDataSource* instance = new WebDataSource(loader);
+ instance->AddRef();
+ return instance;
+}
+
+WebDocumentLoader* WebDataSource::documentLoader() const
+{
+ return m_loader.get();
+}
+
+// IWebDataSourcePrivate ------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebDataSource::overrideEncoding(
+ /* [retval][out] */ BSTR* /*encoding*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::setOverrideEncoding(
+ /* [in] */ BSTR /*encoding*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::mainDocumentError(
+ /* [retval][out] */ IWebError** error)
+{
+ if (!error) {
+ ASSERT_NOT_REACHED();
+ return E_POINTER;
+ }
+
+ *error = 0;
+
+ if (!m_loader)
+ return E_FAIL;
+
+ if (m_loader->mainDocumentError().isNull())
+ return S_OK;
+
+ *error = WebError::createInstance(m_loader->mainDocumentError());
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::setDeferMainResourceDataLoad(
+ /* [in] */ BOOL flag)
+{
+ if (!m_loader)
+ return E_FAIL;
+
+ m_loader->setDeferMainResourceDataLoad(flag);
+ return S_OK;
+}
+
+// IUnknown -------------------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebDataSource::QueryInterface(REFIID riid, void** ppvObject)
+{
+ *ppvObject = 0;
+ if (IsEqualGUID(riid, IID_WebDataSource))
+ *ppvObject = this;
+ else if (IsEqualGUID(riid, IID_IUnknown))
+ *ppvObject = static_cast<IWebDataSource*>(this);
+ else if (IsEqualGUID(riid, IID_IWebDataSource))
+ *ppvObject = static_cast<IWebDataSource*>(this);
+ else if (IsEqualGUID(riid, IID_IWebDataSourcePrivate))
+ *ppvObject = static_cast<IWebDataSourcePrivate*>(this);
+ else
+ return E_NOINTERFACE;
+
+ AddRef();
+ return S_OK;
+}
+
+ULONG STDMETHODCALLTYPE WebDataSource::AddRef(void)
+{
+ return ++m_refCount;
+}
+
+ULONG STDMETHODCALLTYPE WebDataSource::Release(void)
+{
+ ULONG newRef = --m_refCount;
+ if (!newRef)
+ delete(this);
+
+ return newRef;
+}
+
+// IWebDataSource --------------------------------------------------------------
+
+HRESULT STDMETHODCALLTYPE WebDataSource::initWithRequest(
+ /* [in] */ IWebURLRequest* /*request*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::data(
+ /* [retval][out] */ IStream** stream)
+{
+ *stream = 0;
+ if (!m_loader)
+ return E_FAIL;
+
+ return MemoryStream::createInstance(m_loader->mainResourceData()).copyRefTo(stream);
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::representation(
+ /* [retval][out] */ IWebDocumentRepresentation** rep)
+{
+ HRESULT hr = S_OK;
+ if (!m_representation) {
+ WebHTMLRepresentation* htmlRep = WebHTMLRepresentation::createInstance(static_cast<WebFrame*>(m_loader->frameLoader()->client()));
+ hr = htmlRep->QueryInterface(IID_IWebDocumentRepresentation, (void**) &m_representation);
+ htmlRep->Release();
+ }
+
+ return m_representation.copyRefTo(rep);
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::webFrame(
+ /* [retval][out] */ IWebFrame** frame)
+{
+ *frame = static_cast<WebFrame*>(m_loader->frameLoader()->client());
+ (*frame)->AddRef();
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::initialRequest(
+ /* [retval][out] */ IWebURLRequest** request)
+{
+ *request = WebMutableURLRequest::createInstance(m_loader->originalRequest());
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::request(
+ /* [retval][out] */ IWebMutableURLRequest** request)
+{
+ *request = WebMutableURLRequest::createInstance(m_loader->request());
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::response(
+ /* [retval][out] */ IWebURLResponse** response)
+{
+ *response = WebURLResponse::createInstance(m_loader->response());
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::textEncodingName(
+ /* [retval][out] */ BSTR* name)
+{
+ String encoding = m_loader->overrideEncoding();
+ if (encoding.isNull())
+ encoding = m_loader->response().textEncodingName();
+
+ *name = BString(encoding).release();
+
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::isLoading(
+ /* [retval][out] */ BOOL* loading)
+{
+ *loading = m_loader->isLoadingInAPISense();
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::pageTitle(
+ /* [retval][out] */ BSTR* title)
+{
+ *title = BString(m_loader->title()).release();
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::unreachableURL(
+ /* [retval][out] */ BSTR* url)
+{
+ KURL unreachableURL = m_loader->unreachableURL();
+ BString urlString((LPOLESTR)unreachableURL.string().characters(), unreachableURL.string().length());
+
+ *url = urlString.release();
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::webArchive(
+ /* [retval][out] */ IWebArchive** /*archive*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::mainResource(
+ /* [retval][out] */ IWebResource** /*resource*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::subresources(
+ /* [retval][out] */ IEnumVARIANT** /*result*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::subresourceForURL(
+ /* [in] */ BSTR url,
+ /* [retval][out] */ IWebResource** resource)
+{
+ if (!resource) {
+ ASSERT_NOT_REACHED();
+ return E_POINTER;
+ }
+
+ *resource = 0;
+
+ Document *doc = m_loader->frameLoader()->frame()->document();
+
+ if (!doc)
+ return E_FAIL;
+
+ CachedResource *cachedResource = doc->docLoader()->cachedResource(String(url));
+
+ if (!cachedResource)
+ return E_FAIL;
+
+ *resource = WebResource::createInstance(cachedResource->data(), cachedResource->response());
+ return S_OK;
+}
+
+HRESULT STDMETHODCALLTYPE WebDataSource::addSubresource(
+ /* [in] */ IWebResource* /*subresource*/)
+{
+ ASSERT_NOT_REACHED();
+ return E_NOTIMPL;
+}