1 /*

     2  * Copyright (C) 2008 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 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 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 "WebArchive.h"

    29 

    30 #include "DOMCoreClasses.h"

    31 #include "MemoryStream.h"

    32 #include <WebCore/LegacyWebArchive.h>

    33 

    34 using namespace WebCore;

    35 

    36 // WebArchive ----------------------------------------------------------------

    37 

    38 WebArchive* WebArchive::createInstance()

    39 {

    40     WebArchive* instance = new WebArchive(0);

    41     instance->AddRef();

    42     return instance;

    43 }

    44 

    45 WebArchive* WebArchive::createInstance(PassRefPtr<LegacyWebArchive> coreArchive)

    46 {

    47     WebArchive* instance = new WebArchive(coreArchive);

    48 

    49     instance->AddRef();

    50     return instance;

    51 }

    52 

    53 WebArchive::WebArchive(PassRefPtr<LegacyWebArchive> coreArchive)

    54     : m_refCount(0)

    55     , m_archive(coreArchive)

    56 {

    57     gClassCount++;

    58     gClassNameCount.add("WebArchive");

    59 }

    60 

    61 WebArchive::~WebArchive()

    62 {

    63     gClassCount--;

    64     gClassNameCount.remove("WebArchive");

    65 }

    66 

    67 HRESULT STDMETHODCALLTYPE WebArchive::QueryInterface(REFIID riid, void** ppvObject)

    68 {

    69     *ppvObject = 0;

    70     if (IsEqualGUID(riid, IID_IUnknown))

    71         *ppvObject = static_cast<IWebArchive*>(this);

    72     else if (IsEqualGUID(riid, __uuidof(IWebArchive)))

    73         *ppvObject = static_cast<IWebArchive*>(this);

    74     else

    75         return E_NOINTERFACE;

    76 

    77     AddRef();

    78     return S_OK;

    79 }

    80 

    81 ULONG STDMETHODCALLTYPE WebArchive::AddRef()

    82 {

    83     return ++m_refCount;

    84 }

    85 

    86 ULONG STDMETHODCALLTYPE WebArchive::Release()

    87 {

    88     ULONG newRef = --m_refCount;

    89     if (!newRef)

    90         delete(this);

    91 

    92     return newRef;

    93 }

    94 

    95 HRESULT STDMETHODCALLTYPE WebArchive::initWithMainResource(

    96         /* [in] */ IWebResource*, 

    97         /* [in, size_is(cSubResources)] */ IWebResource**, 

    98         /* [in] */ int, 

    99         /* in, size_is(cSubFrameArchives)] */ IWebArchive**, 

   100         /* [in] */ int)

   101 {

   102     return E_NOTIMPL;

   103 }

   104 

   105 HRESULT STDMETHODCALLTYPE WebArchive::initWithData(

   106         /* [in] */ IStream*)

   107 {

   108     return E_NOTIMPL;

   109 }

   110 

   111 HRESULT STDMETHODCALLTYPE WebArchive::initWithNode(

   112         /* [in] */ IDOMNode* node)

   113 {

   114     if (!node)

   115         return E_POINTER;

   116 

   117     COMPtr<DOMNode> domNode(Query, node);

   118     if (!domNode)

   119         return E_NOINTERFACE;

   120 

   121     m_archive = LegacyWebArchive::create(domNode->node());

   122     

   123     return S_OK;

   124 }

   125 

   126 HRESULT STDMETHODCALLTYPE WebArchive::mainResource(

   127         /* [out, retval] */ IWebResource**)

   128 {

   129     return E_NOTIMPL;

   130 }

   131 

   132 HRESULT STDMETHODCALLTYPE WebArchive::subResources(

   133         /* [out, retval] */ IEnumVARIANT**)

   134 {

   135     return E_NOTIMPL;

   136 }

   137 

   138 HRESULT STDMETHODCALLTYPE WebArchive::subframeArchives(

   139         /* [out, retval] */ IEnumVARIANT**)

   140 {

   141     return E_NOTIMPL;

   142 }

   143 

   144 HRESULT STDMETHODCALLTYPE WebArchive::data(

   145         /* [out, retval] */ IStream** stream)

   146 {

   147     RetainPtr<CFDataRef> cfData = m_archive->rawDataRepresentation();

   148     if (!cfData)

   149         return E_FAIL;

   150 

   151     RefPtr<SharedBuffer> buffer = SharedBuffer::create(CFDataGetBytePtr(cfData.get()), CFDataGetLength(cfData.get()));

   152 

   153     return MemoryStream::createInstance(buffer).copyRefTo(stream);

   154 }