blob: 18637487bf8a04796aff2a4a8b68956224ca504c [file] [log] [blame]
/*
* Copyright (C) 2008, 2015 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 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 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 "WebKitDLL.h"
#include "WebArchive.h"
#include "DOMCoreClasses.h"
#include "MemoryStream.h"
#include <WebCore/LegacyWebArchive.h>
using namespace WebCore;
// WebArchive ----------------------------------------------------------------
WebArchive* WebArchive::createInstance()
{
WebArchive* instance = new WebArchive(0);
instance->AddRef();
return instance;
}
WebArchive* WebArchive::createInstance(RefPtr<LegacyWebArchive>&& coreArchive)
{
WebArchive* instance = new WebArchive(WTFMove(coreArchive));
instance->AddRef();
return instance;
}
WebArchive::WebArchive(RefPtr<LegacyWebArchive>&& coreArchive)
: m_archive(WTFMove(coreArchive))
{
gClassCount++;
gClassNameCount().add("WebArchive");
}
WebArchive::~WebArchive()
{
gClassCount--;
gClassNameCount().remove("WebArchive");
}
HRESULT WebArchive::QueryInterface(_In_ REFIID riid, _COM_Outptr_ void** ppvObject)
{
if (!ppvObject)
return E_POINTER;
*ppvObject = nullptr;
if (IsEqualGUID(riid, IID_IUnknown))
*ppvObject = static_cast<IWebArchive*>(this);
else if (IsEqualGUID(riid, __uuidof(IWebArchive)))
*ppvObject = static_cast<IWebArchive*>(this);
else
return E_NOINTERFACE;
AddRef();
return S_OK;
}
ULONG WebArchive::AddRef()
{
return ++m_refCount;
}
ULONG WebArchive::Release()
{
ULONG newRef = --m_refCount;
if (!newRef)
delete(this);
return newRef;
}
HRESULT WebArchive::initWithMainResource(_In_opt_ IWebResource*,
__inout_ecount_full(cSubResources) IWebResource**, int cSubResources,
__inout_ecount_full(cSubFrameArchives) IWebArchive**, int cSubFrameArchives)
{
return E_NOTIMPL;
}
HRESULT WebArchive::initWithData(_In_opt_ IStream*)
{
return E_NOTIMPL;
}
HRESULT WebArchive::initWithNode(_In_opt_ IDOMNode* node)
{
if (!node)
return E_POINTER;
COMPtr<DOMNode> domNode(Query, node);
if (!domNode)
return E_NOINTERFACE;
m_archive = LegacyWebArchive::create(*domNode->node());
return S_OK;
}
HRESULT WebArchive::mainResource(_COM_Outptr_opt_ IWebResource** result)
{
if (!result)
return E_POINTER;
*result = nullptr;
return E_NOTIMPL;
}
HRESULT WebArchive::subResources(_COM_Outptr_opt_ IEnumVARIANT** result)
{
if (!result)
return E_POINTER;
*result = nullptr;
return E_NOTIMPL;
}
HRESULT WebArchive::subframeArchives(_COM_Outptr_opt_ IEnumVARIANT** result)
{
if (!result)
return E_POINTER;
*result = nullptr;
return E_NOTIMPL;
}
HRESULT WebArchive::data(_COM_Outptr_opt_ IStream** stream)
{
if (!stream)
return E_POINTER;
*stream = nullptr;
RetainPtr<CFDataRef> cfData = m_archive->rawDataRepresentation();
if (!cfData)
return E_FAIL;
auto buffer = SharedBuffer::create(CFDataGetBytePtr(cfData.get()), CFDataGetLength(cfData.get()));
return MemoryStream::createInstance(WTFMove(buffer)).copyRefTo(stream);
}