Add WTF::move()
[WebKit-https.git] / Source / WebKit2 / Shared / APIWebArchive.cpp
1 /*
2  * Copyright (C) 2013 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. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "config.h"
27 #include "APIWebArchive.h"
28
29 #if PLATFORM(COCOA)
30
31 #include "APIArray.h"
32 #include "APIData.h"
33 #include "APIWebArchiveResource.h"
34 #include <WebCore/LegacyWebArchive.h>
35 #include <wtf/RetainPtr.h>
36
37 using namespace WebCore;
38
39 namespace API {
40
41 PassRefPtr<WebArchive> WebArchive::create(WebArchiveResource* mainResource, PassRefPtr<API::Array> subresources, PassRefPtr<API::Array> subframeArchives)
42 {
43     return adoptRef(new WebArchive(mainResource, subresources, subframeArchives));
44 }
45
46 PassRefPtr<WebArchive> WebArchive::create(API::Data* data)
47 {
48     return adoptRef(new WebArchive(data));
49 }
50
51 PassRefPtr<WebArchive> WebArchive::create(PassRefPtr<LegacyWebArchive> legacyWebArchive)
52 {
53     return adoptRef(new WebArchive(legacyWebArchive));
54 }
55
56 PassRefPtr<WebArchive> WebArchive::create(Range* range)
57 {
58     return adoptRef(new WebArchive(LegacyWebArchive::create(range)));
59 }
60
61 WebArchive::WebArchive(WebArchiveResource* mainResource, PassRefPtr<API::Array> subresources, PassRefPtr<API::Array> subframeArchives)
62     : m_cachedMainResource(mainResource)
63     , m_cachedSubresources(subresources)
64     , m_cachedSubframeArchives(subframeArchives)
65 {
66     RefPtr<ArchiveResource> coreMainResource = m_cachedMainResource->coreArchiveResource();
67
68     Vector<RefPtr<ArchiveResource>> coreArchiveResources;
69     coreArchiveResources.reserveInitialCapacity(m_cachedSubresources->size());
70     for (size_t i = 0; i < m_cachedSubresources->size(); ++i) {
71         RefPtr<WebArchiveResource> resource = m_cachedSubresources->at<WebArchiveResource>(i);
72         ASSERT(resource);
73         coreArchiveResources.uncheckedAppend(resource->coreArchiveResource());
74     }
75
76     Vector<RefPtr<LegacyWebArchive>> coreSubframeLegacyWebArchives;
77     coreSubframeLegacyWebArchives.reserveInitialCapacity(m_cachedSubframeArchives->size());
78
79     for (size_t i = 0; i < m_cachedSubframeArchives->size(); ++i) {
80         RefPtr<WebArchive> subframeWebArchive = m_cachedSubframeArchives->at<WebArchive>(i);
81         ASSERT(subframeWebArchive);
82         coreSubframeLegacyWebArchives.uncheckedAppend(subframeWebArchive->coreLegacyWebArchive());
83     }
84
85     m_legacyWebArchive = LegacyWebArchive::create(coreMainResource.release(), WTF::move(coreArchiveResources), WTF::move(coreSubframeLegacyWebArchives));
86 }
87
88 WebArchive::WebArchive(API::Data* data)
89 {
90     RefPtr<SharedBuffer> buffer = SharedBuffer::create(data->bytes(), data->size());
91     m_legacyWebArchive = LegacyWebArchive::create(buffer.get());
92 }
93
94 WebArchive::WebArchive(PassRefPtr<LegacyWebArchive> legacyWebArchive)
95     : m_legacyWebArchive(legacyWebArchive)
96 {
97 }
98
99 WebArchive::~WebArchive()
100 {
101 }
102
103 WebArchiveResource* WebArchive::mainResource()
104 {
105     if (!m_cachedMainResource)
106         m_cachedMainResource = WebArchiveResource::create(m_legacyWebArchive->mainResource());
107     return m_cachedMainResource.get();
108 }
109
110 API::Array* WebArchive::subresources()
111 {
112     if (!m_cachedSubresources) {
113         Vector<RefPtr<API::Object>> subresources;
114         subresources.reserveInitialCapacity(m_legacyWebArchive->subresources().size());
115
116         for (const auto& subresource : m_legacyWebArchive->subresources())
117             subresources.uncheckedAppend(WebArchiveResource::create(subresource));
118
119         m_cachedSubresources = API::Array::create(WTF::move(subresources));
120     }
121
122     return m_cachedSubresources.get();
123 }
124
125 API::Array* WebArchive::subframeArchives()
126 {
127     if (!m_cachedSubframeArchives) {
128         Vector<RefPtr<API::Object>> subframeWebArchives;
129         subframeWebArchives.reserveInitialCapacity(m_legacyWebArchive->subframeArchives().size());
130
131         for (const auto& subframeArchive : m_legacyWebArchive->subframeArchives())
132             subframeWebArchives.uncheckedAppend(WebArchive::create(static_cast<LegacyWebArchive*>(subframeArchive.get())));
133
134         m_cachedSubframeArchives = API::Array::create(WTF::move(subframeWebArchives));
135     }
136
137     return m_cachedSubframeArchives.get();
138 }
139
140 static void releaseCFData(unsigned char*, const void* data)
141 {
142     // Balanced by CFRetain in WebArchive::data().
143     CFRelease(data);
144 }
145
146 PassRefPtr<API::Data> WebArchive::data()
147 {
148     RetainPtr<CFDataRef> rawDataRepresentation = m_legacyWebArchive->rawDataRepresentation();
149
150     // Balanced by CFRelease in releaseCFData.
151     CFRetain(rawDataRepresentation.get());
152
153     return API::Data::createWithoutCopying(CFDataGetBytePtr(rawDataRepresentation.get()), CFDataGetLength(rawDataRepresentation.get()), releaseCFData, rawDataRepresentation.get());
154 }
155
156 LegacyWebArchive* WebArchive::coreLegacyWebArchive()
157 {
158     return m_legacyWebArchive.get();
159 }
160
161 } // namespace WebKit
162
163 #endif // PLATFORM(COCOA)