blob: dd5813d1d7f84e62676342293344afda388953c8 [file] [log] [blame]
/*
* 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 "WebHTMLRepresentation.h"
#include "WebKit.h"
#include "WebFrame.h"
#include "WebKitStatisticsPrivate.h"
#include <WebCore/BString.h>
#include <WebCore/Frame.h>
#include <WebCore/HTMLInputElement.h>
#include <WebCore/TextResourceDecoder.h>
using namespace WebCore;
// WebHTMLRepresentation ------------------------------------------------------
WebHTMLRepresentation::WebHTMLRepresentation()
: m_refCount(0)
, m_frame(0)
{
WebHTMLRepresentationCount++;
gClassCount++;
gClassNameCount.add("WebHTMLRepresentation");
}
WebHTMLRepresentation::~WebHTMLRepresentation()
{
if (m_frame) {
m_frame->Release();
m_frame = 0;
}
WebHTMLRepresentationCount--;
gClassCount--;
gClassNameCount.remove("WebHTMLRepresentation");
}
WebHTMLRepresentation* WebHTMLRepresentation::createInstance(WebFrame* frame)
{
WebHTMLRepresentation* instance = new WebHTMLRepresentation();
instance->m_frame = frame;
frame->AddRef();
instance->AddRef();
return instance;
}
// IUnknown -------------------------------------------------------------------
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::QueryInterface(REFIID riid, void** ppvObject)
{
*ppvObject = 0;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IWebHTMLRepresentation*>(this);
else if (IsEqualGUID(riid, IID_IWebHTMLRepresentation))
*ppvObject = static_cast<IWebHTMLRepresentation*>(this);
else if (IsEqualGUID(riid, IID_IWebDocumentRepresentation))
*ppvObject = static_cast<IWebDocumentRepresentation*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG STDMETHODCALLTYPE WebHTMLRepresentation::AddRef()
{
return ++m_refCount;
}
ULONG STDMETHODCALLTYPE WebHTMLRepresentation::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
// IWebHTMLRepresentation --------------------------------------------------------------------
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedMIMETypes(
/* [out][in] */ BSTR* /*types*/,
/* [out][in] */ int* /*cTypes*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedNonImageMIMETypes(
/* [out][in] */ BSTR* /*types*/,
/* [out][in] */ int* /*cTypes*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedImageMIMETypes(
/* [out][in] */ BSTR* /*types*/,
/* [out][in] */ int* /*cTypes*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::attributedStringFromDOMNodes(
/* [in] */ IDOMNode* /*startNode*/,
/* [in] */ int /*startOffset*/,
/* [in] */ IDOMNode* /*endNode*/,
/* [in] */ int /*endOffset*/,
/* [retval][out] */ IDataObject** /*attributedString*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementWithName(
/* [in] */ BSTR name,
/* [in] */ IDOMElement* form,
/* [retval][out] */ IDOMElement** element)
{
if (!m_frame)
return E_FAIL;
return m_frame->elementWithName(name, form, element);
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementDoesAutoComplete(
/* [in] */ IDOMElement* element,
/* [retval][out] */ BOOL* result)
{
BOOL doesAutoComplete;
HRESULT hr = m_frame->elementDoesAutoComplete(element, &doesAutoComplete);
*result = doesAutoComplete ? TRUE : FALSE;
return hr;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementIsPassword(
/* [in] */ IDOMElement* element,
/* [retval][out] */ BOOL* result)
{
bool isPassword;
HRESULT hr = m_frame->elementIsPassword(element, &isPassword);
*result = isPassword ? TRUE : FALSE;
return hr;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::formForElement(
/* [in] */ IDOMElement* element,
/* [retval][out] */ IDOMElement** form)
{
if (!m_frame)
return E_FAIL;
return m_frame->formForElement(element, form);
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::currentForm(
/* [retval][out] */ IDOMElement** form)
{
if (!m_frame)
return E_FAIL;
return m_frame->currentForm(form);
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::controlsInForm(
/* [in] */ IDOMElement* form,
/* [out][in] */ IDOMElement** controls,
/* [out][in] */ int* cControls)
{
return m_frame->controlsInForm(form, controls, cControls);
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::deprecatedSearchForLabels(
/* [size_is][in] */ BSTR* labels,
/* [in] */ int cLabels,
/* [in] */ IDOMElement* beforeElement,
/* [retval][out] */ BSTR* result)
{
return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, 0, 0, result);
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::matchLabels(
/* [size_is][in] */ BSTR* labels,
/* [in] */ int cLabels,
/* [in] */ IDOMElement* againstElement,
/* [retval][out] */ BSTR* result)
{
return m_frame->matchLabelsAgainstElement(labels, cLabels, againstElement, result);
}
HRESULT WebHTMLRepresentation::searchForLabels(BSTR* labels, unsigned cLabels, IDOMElement* beforeElement, unsigned* resultDistance, BOOL* resultIsInCellAbove, BSTR* result)
{
return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, resultDistance, resultIsInCellAbove, result);
}
// IWebDocumentRepresentation ----------------------------------------------------------------
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::setDataSource(
/* [in] */ IWebDataSource* /*dataSource*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedData(
/* [in] */ IStream* /*data*/,
/* [in] */ IWebDataSource* /*dataSource*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedError(
/* [in] */ IWebError* /*error*/,
/* [in] */ IWebDataSource* /*dataSource*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::finishedLoadingWithDataSource(
/* [in] */ IWebDataSource* /*dataSource*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::canProvideDocumentSource(
/* [retval][out] */ BOOL* result)
{
bool canProvideSource;
HRESULT hr = this->m_frame->canProvideDocumentSource(&canProvideSource);
*result = canProvideSource ? TRUE : FALSE;
return hr;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::documentSource(
/* [retval][out] */ BSTR* source)
{
if (!source)
return E_FAIL;
*source = 0;
HRESULT hr = S_OK;
COMPtr<IWebDataSource> dataSource;
hr = m_frame->dataSource(&dataSource);
if (FAILED(hr))
return hr;
COMPtr<IStream> data;
hr = dataSource->data(&data);
if (FAILED(hr))
return hr;
STATSTG stat;
hr = data->Stat(&stat, STATFLAG_NONAME);
if (FAILED(hr))
return hr;
if (stat.cbSize.HighPart || !stat.cbSize.LowPart)
return E_FAIL;
Vector<char> dataBuffer(stat.cbSize.LowPart);
ULONG read;
hr = data->Read(dataBuffer.data(), static_cast<ULONG>(dataBuffer.size()), &read);
if (FAILED(hr))
return hr;
WebCore::Frame* frame = core(m_frame);
if (!frame)
return E_FAIL;
WebCore::Document* doc = frame->document();
if (!doc)
return E_FAIL;
WebCore::TextResourceDecoder* decoder = doc->decoder();
if (!decoder)
return E_FAIL;
*source = WebCore::BString(decoder->encoding().decode(dataBuffer.data(), dataBuffer.size())).release();
return S_OK;
}
HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::title(
/* [retval][out] */ BSTR* /*docTitle*/)
{
ASSERT_NOT_REACHED();
return E_NOTIMPL;
}