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 "WebHTMLRepresentation.h"

    29 

    30 #include "WebKit.h"

    31 #include "WebFrame.h"

    32 #include "WebKitStatisticsPrivate.h"

    33 #include <WebCore/BString.h>

    34 #include <WebCore/Frame.h>

    35 #include <WebCore/HTMLInputElement.h>

    36 #include <WebCore/TextResourceDecoder.h>

    37 

    38 using namespace WebCore;

    39 

    40 // WebHTMLRepresentation ------------------------------------------------------

    41 

    42 WebHTMLRepresentation::WebHTMLRepresentation()

    43     : m_refCount(0)

    44     , m_frame(0)

    45 {

    46     WebHTMLRepresentationCount++;

    47     gClassCount++;

    48     gClassNameCount.add("WebHTMLRepresentation");

    49 }

    50 

    51 WebHTMLRepresentation::~WebHTMLRepresentation()

    52 {

    53     if (m_frame) {

    54         m_frame->Release();

    55         m_frame = 0;

    56     }

    57 

    58     WebHTMLRepresentationCount--;

    59     gClassCount--;

    60     gClassNameCount.remove("WebHTMLRepresentation");

    61 }

    62 

    63 WebHTMLRepresentation* WebHTMLRepresentation::createInstance(WebFrame* frame)

    64 {

    65     WebHTMLRepresentation* instance = new WebHTMLRepresentation();

    66     instance->m_frame = frame;

    67     frame->AddRef();

    68     instance->AddRef();

    69     return instance;

    70 }

    71 

    72 // IUnknown -------------------------------------------------------------------

    73 

    74 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::QueryInterface(REFIID riid, void** ppvObject)

    75 {

    76     *ppvObject = 0;

    77     if (IsEqualGUID(riid, IID_IUnknown))

    78         *ppvObject = static_cast<IWebHTMLRepresentation*>(this);

    79     else if (IsEqualGUID(riid, IID_IWebHTMLRepresentation))

    80         *ppvObject = static_cast<IWebHTMLRepresentation*>(this);

    81     else if (IsEqualGUID(riid, IID_IWebDocumentRepresentation))

    82         *ppvObject = static_cast<IWebDocumentRepresentation*>(this);

    83     else

    84         return E_NOINTERFACE;

    85 

    86     AddRef();

    87     return S_OK;

    88 }

    89 

    90 ULONG STDMETHODCALLTYPE WebHTMLRepresentation::AddRef()

    91 {

    92     return ++m_refCount;

    93 }

    94 

    95 ULONG STDMETHODCALLTYPE WebHTMLRepresentation::Release()

    96 {

    97     ULONG newRef = --m_refCount;

    98     if (!newRef)

    99         delete(this);

   100 

   101     return newRef;

   102 }

   103 

   104 // IWebHTMLRepresentation --------------------------------------------------------------------

   105 

   106 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedMIMETypes(

   107         /* [out][in] */ BSTR* /*types*/,

   108         /* [out][in] */ int* /*cTypes*/)

   109 {

   110     ASSERT_NOT_REACHED();

   111     return E_NOTIMPL;

   112 }

   113     

   114 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedNonImageMIMETypes(

   115         /* [out][in] */ BSTR* /*types*/,

   116         /* [out][in] */ int* /*cTypes*/)

   117 {

   118     ASSERT_NOT_REACHED();

   119     return E_NOTIMPL;

   120 }

   121     

   122 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::supportedImageMIMETypes(

   123         /* [out][in] */ BSTR* /*types*/,

   124         /* [out][in] */ int* /*cTypes*/)

   125 {

   126     ASSERT_NOT_REACHED();

   127     return E_NOTIMPL;

   128 }

   129     

   130 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::attributedStringFromDOMNodes(

   131         /* [in] */ IDOMNode* /*startNode*/,

   132         /* [in] */ int /*startOffset*/,

   133         /* [in] */ IDOMNode* /*endNode*/,

   134         /* [in] */ int /*endOffset*/,

   135         /* [retval][out] */ IDataObject** /*attributedString*/)

   136 {

   137     ASSERT_NOT_REACHED();

   138     return E_NOTIMPL;

   139 }

   140     

   141 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementWithName(

   142         /* [in] */ BSTR name,

   143         /* [in] */ IDOMElement* form,

   144         /* [retval][out] */ IDOMElement** element)

   145 {

   146     if (!m_frame)

   147         return E_FAIL;

   148 

   149     return m_frame->elementWithName(name, form, element);

   150 }

   151     

   152 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementDoesAutoComplete(

   153         /* [in] */ IDOMElement* element,

   154         /* [retval][out] */ BOOL* result)

   155 {

   156     BOOL doesAutoComplete;

   157     HRESULT hr = m_frame->elementDoesAutoComplete(element, &doesAutoComplete);

   158     *result = doesAutoComplete ? TRUE : FALSE;

   159     return hr;

   160 }

   161     

   162 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::elementIsPassword(

   163         /* [in] */ IDOMElement* element,

   164         /* [retval][out] */ BOOL* result)

   165 {

   166     bool isPassword;

   167     HRESULT hr = m_frame->elementIsPassword(element, &isPassword);

   168     *result = isPassword ?  TRUE : FALSE;

   169     return hr;

   170 }

   171     

   172 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::formForElement(

   173         /* [in] */ IDOMElement* element,

   174         /* [retval][out] */ IDOMElement** form)

   175 {

   176     if (!m_frame)

   177         return E_FAIL;

   178 

   179     return m_frame->formForElement(element, form);

   180 }

   181     

   182 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::currentForm(

   183         /* [retval][out] */ IDOMElement** form)

   184 {

   185     if (!m_frame)

   186         return E_FAIL;

   187 

   188     return m_frame->currentForm(form);

   189 }

   190     

   191 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::controlsInForm(

   192         /* [in] */ IDOMElement* form,

   193         /* [out][in] */ IDOMElement** controls,

   194         /* [out][in] */ int* cControls)

   195 {

   196     return m_frame->controlsInForm(form, controls, cControls);

   197 }

   198     

   199 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::deprecatedSearchForLabels(

   200         /* [size_is][in] */ BSTR* labels,

   201         /* [in] */ int cLabels,

   202         /* [in] */ IDOMElement* beforeElement,

   203         /* [retval][out] */ BSTR* result)

   204 {

   205     return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, 0, 0, result);

   206 }

   207     

   208 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::matchLabels(

   209         /* [size_is][in] */ BSTR* labels,

   210         /* [in] */ int cLabels,

   211         /* [in] */ IDOMElement* againstElement,

   212         /* [retval][out] */ BSTR* result)

   213 {

   214     return m_frame->matchLabelsAgainstElement(labels, cLabels, againstElement, result);

   215 }

   216 

   217 HRESULT WebHTMLRepresentation::searchForLabels(BSTR* labels, unsigned cLabels, IDOMElement* beforeElement, unsigned* resultDistance, BOOL* resultIsInCellAbove, BSTR* result)

   218 {

   219     return m_frame->searchForLabelsBeforeElement(labels, cLabels, beforeElement, resultDistance, resultIsInCellAbove, result);

   220 }

   221 

   222 // IWebDocumentRepresentation ----------------------------------------------------------------

   223 

   224 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::setDataSource(

   225         /* [in] */ IWebDataSource* /*dataSource*/)

   226 {

   227     ASSERT_NOT_REACHED();

   228     return E_NOTIMPL;

   229 }

   230     

   231 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedData(

   232         /* [in] */ IStream* /*data*/,

   233         /* [in] */ IWebDataSource* /*dataSource*/)

   234 {

   235     ASSERT_NOT_REACHED();

   236     return E_NOTIMPL;

   237 }

   238     

   239 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::receivedError(

   240         /* [in] */ IWebError* /*error*/,

   241         /* [in] */ IWebDataSource* /*dataSource*/)

   242 {

   243     ASSERT_NOT_REACHED();

   244     return E_NOTIMPL;

   245 }

   246     

   247 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::finishedLoadingWithDataSource(

   248         /* [in] */ IWebDataSource* /*dataSource*/)

   249 {

   250     ASSERT_NOT_REACHED();

   251     return E_NOTIMPL;

   252 }

   253     

   254 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::canProvideDocumentSource(

   255         /* [retval][out] */ BOOL* result)

   256 {

   257     bool canProvideSource;

   258     HRESULT hr = this->m_frame->canProvideDocumentSource(&canProvideSource);

   259     *result = canProvideSource ? TRUE : FALSE;

   260     return hr;

   261 }

   262     

   263 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::documentSource(

   264         /* [retval][out] */ BSTR* source)

   265 {

   266     if (!source)

   267         return E_FAIL;

   268 

   269     *source = 0;

   270 

   271     HRESULT hr = S_OK;

   272 

   273     COMPtr<IWebDataSource> dataSource;

   274     hr = m_frame->dataSource(&dataSource);

   275     if (FAILED(hr))

   276         return hr;

   277 

   278     COMPtr<IStream> data;

   279     hr = dataSource->data(&data);

   280     if (FAILED(hr))

   281         return hr;

   282 

   283     STATSTG stat;

   284     hr = data->Stat(&stat, STATFLAG_NONAME);

   285     if (FAILED(hr))

   286         return hr;

   287 

   288     if (stat.cbSize.HighPart || !stat.cbSize.LowPart)

   289         return E_FAIL;

   290 

   291     Vector<char> dataBuffer(stat.cbSize.LowPart);

   292     ULONG read;

   293     

   294     hr = data->Read(dataBuffer.data(), static_cast<ULONG>(dataBuffer.size()), &read);

   295     if (FAILED(hr))

   296         return hr;

   297 

   298     WebCore::Frame* frame = core(m_frame);

   299     if (!frame)

   300         return E_FAIL;

   301 

   302     WebCore::Document* doc = frame->document();

   303     if (!doc)

   304         return E_FAIL;

   305 

   306     WebCore::TextResourceDecoder* decoder = doc->decoder();

   307     if (!decoder)

   308         return E_FAIL;

   309 

   310     *source = WebCore::BString(decoder->encoding().decode(dataBuffer.data(), dataBuffer.size())).release();

   311     return S_OK;

   312 }

   313     

   314 HRESULT STDMETHODCALLTYPE WebHTMLRepresentation::title(

   315         /* [retval][out] */ BSTR* /*docTitle*/)

   316 {

   317     ASSERT_NOT_REACHED();

   318     return E_NOTIMPL;

   319 }