/* * Copyright (C) 2013 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. AND ITS CONTRIBUTORS ``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 ITS 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 "APIWebArchive.h" #if PLATFORM(COCOA) #include "APIArray.h" #include "APIData.h" #include "APIWebArchiveResource.h" #include <WebCore/LegacyWebArchive.h> #include <wtf/RetainPtr.h> using namespace WebCore; namespace API { Ref<WebArchive> WebArchive::create(WebArchiveResource* mainResource, RefPtr<API::Array>&& subresources, RefPtr<API::Array>&& subframeArchives) { return adoptRef(*new WebArchive(mainResource, WTFMove(subresources), WTFMove(subframeArchives))); } Ref<WebArchive> WebArchive::create(API::Data* data) { return adoptRef(*new WebArchive(data)); } Ref<WebArchive> WebArchive::create(RefPtr<LegacyWebArchive>&& legacyWebArchive) { return adoptRef(*new WebArchive(legacyWebArchive.releaseNonNull())); } Ref<WebArchive> WebArchive::create(Range& range) { return adoptRef(*new WebArchive(LegacyWebArchive::create(&range))); } WebArchive::WebArchive(WebArchiveResource* mainResource, RefPtr<API::Array>&& subresources, RefPtr<API::Array>&& subframeArchives) : m_cachedMainResource(mainResource) , m_cachedSubresources(subresources) , m_cachedSubframeArchives(subframeArchives) { auto coreMainResource = m_cachedMainResource->coreArchiveResource(); Vector<Ref<ArchiveResource>> coreArchiveResources; coreArchiveResources.reserveInitialCapacity(m_cachedSubresources->size()); for (size_t i = 0; i < m_cachedSubresources->size(); ++i) { auto resource = m_cachedSubresources->at<WebArchiveResource>(i); ASSERT(resource); ASSERT(resource->coreArchiveResource()); coreArchiveResources.uncheckedAppend(*resource->coreArchiveResource()); } Vector<Ref<LegacyWebArchive>> coreSubframeLegacyWebArchives; coreSubframeLegacyWebArchives.reserveInitialCapacity(m_cachedSubframeArchives->size()); for (size_t i = 0; i < m_cachedSubframeArchives->size(); ++i) { auto subframeWebArchive = m_cachedSubframeArchives->at<WebArchive>(i); ASSERT(subframeWebArchive); ASSERT(subframeWebArchive->coreLegacyWebArchive()); coreSubframeLegacyWebArchives.uncheckedAppend(*subframeWebArchive->coreLegacyWebArchive()); } m_legacyWebArchive = LegacyWebArchive::create(*coreMainResource, WTFMove(coreArchiveResources), WTFMove(coreSubframeLegacyWebArchives)); } WebArchive::WebArchive(API::Data* data) { m_legacyWebArchive = LegacyWebArchive::create(SharedBuffer::create(data->bytes(), data->size()).get()); } WebArchive::WebArchive(RefPtr<LegacyWebArchive>&& legacyWebArchive) : m_legacyWebArchive(legacyWebArchive) { } WebArchive::~WebArchive() { } WebArchiveResource* WebArchive::mainResource() { if (!m_cachedMainResource) m_cachedMainResource = WebArchiveResource::create(m_legacyWebArchive->mainResource()); return m_cachedMainResource.get(); } API::Array* WebArchive::subresources() { if (!m_cachedSubresources) { Vector<RefPtr<API::Object>> subresources; subresources.reserveInitialCapacity(m_legacyWebArchive->subresources().size()); for (auto& subresource : m_legacyWebArchive->subresources()) subresources.uncheckedAppend(WebArchiveResource::create(subresource.ptr())); m_cachedSubresources = API::Array::create(WTFMove(subresources)); } return m_cachedSubresources.get(); } API::Array* WebArchive::subframeArchives() { if (!m_cachedSubframeArchives) { Vector<RefPtr<API::Object>> subframeWebArchives; subframeWebArchives.reserveInitialCapacity(m_legacyWebArchive->subframeArchives().size()); for (auto& subframeArchive : m_legacyWebArchive->subframeArchives()) subframeWebArchives.uncheckedAppend(WebArchive::create(static_cast<LegacyWebArchive*>(subframeArchive.ptr()))); m_cachedSubframeArchives = API::Array::create(WTFMove(subframeWebArchives)); } return m_cachedSubframeArchives.get(); } static void releaseCFData(unsigned char*, const void* data) { // Balanced by CFRetain in WebArchive::data(). CFRelease(data); } Ref<API::Data> WebArchive::data() { RetainPtr<CFDataRef> rawDataRepresentation = m_legacyWebArchive->rawDataRepresentation(); // Balanced by CFRelease in releaseCFData. CFRetain(rawDataRepresentation.get()); return API::Data::createWithoutCopying(CFDataGetBytePtr(rawDataRepresentation.get()), CFDataGetLength(rawDataRepresentation.get()), releaseCFData, rawDataRepresentation.get()); } LegacyWebArchive* WebArchive::coreLegacyWebArchive() { return m_legacyWebArchive.get(); } } // namespace WebKit #endif // PLATFORM(COCOA)