RefPtr<LegacyWebArchive> LegacyWebArchive::create(RefPtr<ArchiveResource>&& mainResource, Vector<RefPtr<ArchiveResource>>&& subresources, Vector<RefPtr<LegacyWebArchive>>&& subframeArchives)
{
if (!mainResource)
return nullptr;
auto archive = create();
archive->setMainResource(WTFMove(mainResource));
for (auto& subresource : subresources)
archive->addSubresource(WTFMove(subresource));
for (auto& subframeArchive : subframeArchives)
archive->addSubframeArchive(WTFMove(subframeArchive));
return WTFMove(archive);
}
WebArchiveAndroid::WebArchiveAndroid(PassRefPtr<ArchiveResource> mainResource,
Vector<PassRefPtr<ArchiveResource> >& subresources,
Vector<PassRefPtr<Archive> >& subframeArchives)
{
setMainResource(mainResource);
for (Vector<PassRefPtr<ArchiveResource> >::iterator subresourcesIterator = subresources.begin();
subresourcesIterator != subresources.end();
subresourcesIterator++) {
addSubresource(*subresourcesIterator);
}
for (Vector<PassRefPtr<Archive> >::iterator subframesIterator = subframeArchives.begin();
subframesIterator != subframeArchives.end();
subframesIterator++) {
addSubframeArchive(*subframesIterator);
}
}
bool LegacyWebArchive::extract(CFDictionaryRef dictionary)
{
ASSERT(dictionary);
if (!dictionary) {
LOG(Archives, "LegacyWebArchive - Null root CFDictionary, aborting invalid WebArchive");
return false;
}
CFDictionaryRef mainResourceDict = static_cast<CFDictionaryRef>(CFDictionaryGetValue(dictionary, LegacyWebArchiveMainResourceKey));
if (!mainResourceDict) {
LOG(Archives, "LegacyWebArchive - No main resource in archive, aborting invalid WebArchive");
return false;
}
if (CFGetTypeID(mainResourceDict) != CFDictionaryGetTypeID()) {
LOG(Archives, "LegacyWebArchive - Main resource is not the expected CFDictionary, aborting invalid WebArchive");
return false;
}
setMainResource(createResource(mainResourceDict));
if (!mainResource()) {
LOG(Archives, "LegacyWebArchive - Failed to parse main resource from CFDictionary or main resource does not exist, aborting invalid WebArchive");
return false;
}
CFArrayRef subresourceArray = static_cast<CFArrayRef>(CFDictionaryGetValue(dictionary, LegacyWebArchiveSubresourcesKey));
if (subresourceArray && CFGetTypeID(subresourceArray) != CFArrayGetTypeID()) {
LOG(Archives, "LegacyWebArchive - Subresources is not the expected Array, aborting invalid WebArchive");
return false;
}
if (subresourceArray) {
CFIndex count = CFArrayGetCount(subresourceArray);
for (CFIndex i = 0; i < count; ++i) {
CFDictionaryRef subresourceDict = static_cast<CFDictionaryRef>(CFArrayGetValueAtIndex(subresourceArray, i));
if (CFGetTypeID(subresourceDict) != CFDictionaryGetTypeID()) {
LOG(Archives, "LegacyWebArchive - Subresource is not expected CFDictionary, aborting invalid WebArchive");
return false;
}
addSubresource(createResource(subresourceDict));
}
}
CFArrayRef subframeArray = static_cast<CFArrayRef>(CFDictionaryGetValue(dictionary, LegacyWebArchiveSubframeArchivesKey));
if (subframeArray && CFGetTypeID(subframeArray) != CFArrayGetTypeID()) {
LOG(Archives, "LegacyWebArchive - Subframe archives is not the expected Array, aborting invalid WebArchive");
return false;
}
if (subframeArray) {
CFIndex count = CFArrayGetCount(subframeArray);
for (CFIndex i = 0; i < count; ++i) {
CFDictionaryRef subframeDict = static_cast<CFDictionaryRef>(CFArrayGetValueAtIndex(subframeArray, i));
if (CFGetTypeID(subframeDict) != CFDictionaryGetTypeID()) {
LOG(Archives, "LegacyWebArchive - Subframe array is not expected CFDictionary, aborting invalid WebArchive");
return false;
}
RefPtr<LegacyWebArchive> subframeArchive = create();
if (subframeArchive->extract(subframeDict))
addSubframeArchive(subframeArchive.release());
else
LOG(Archives, "LegacyWebArchive - Invalid subframe archive skipped");
}
}
return true;
}
