#include "WebKitDLL.h"
#include "WebArchive.h"
#include "DOMCoreClasses.h"
#include "MemoryStream.h"
#include <WebCore/LegacyWebArchive.h>
using namespace WebCore;
WebArchive* WebArchive::createInstance()
{
WebArchive* instance = new WebArchive(0);
instance->AddRef();
return instance;
}
WebArchive* WebArchive::createInstance(PassRefPtr<LegacyWebArchive> coreArchive)
{
WebArchive* instance = new WebArchive(coreArchive);
instance->AddRef();
return instance;
}
WebArchive::WebArchive(PassRefPtr<LegacyWebArchive> coreArchive)
: m_archive(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;
RefPtr<SharedBuffer> buffer = SharedBuffer::create(CFDataGetBytePtr(cfData.get()), CFDataGetLength(cfData.get()));
return MemoryStream::createInstance(buffer).copyRefTo(stream);
}