1 /*
2 * Copyright (C) 2008 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 *
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 3. Neither the name of Apple Computer, Inc. ("Apple") nor the names of
14 * its contributors may be used to endorse or promote products derived
15 * from this software without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
18 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
20 * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29 #include "config.h"
30 #include "ArchiveFactory.h"
31
32 #include "MIMETypeRegistry.h"
33 #include "PlatformString.h"
34
35 #if USE(CF) && !PLATFORM(QT)
36 #include "LegacyWebArchive.h"
37 #endif
38
39 #include <wtf/HashMap.h>
40 #include <wtf/HashSet.h>
41 #include <wtf/StdLibExtras.h>
42
43 namespace WebCore {
44
45 typedef PassRefPtr<Archive> RawDataCreationFunction(SharedBuffer*);
46 typedef HashMap<String, RawDataCreationFunction*, CaseFoldingHash> ArchiveMIMETypesMap;
47
48 // The create functions in the archive classes return PassRefPtr to concrete subclasses
49 // of Archive. This adaptor makes the functions have a uniform return type.
archiveFactoryCreate(SharedBuffer * buffer)50 template <typename ArchiveClass> static PassRefPtr<Archive> archiveFactoryCreate(SharedBuffer* buffer)
51 {
52 return ArchiveClass::create(buffer);
53 }
54
archiveMIMETypes()55 static ArchiveMIMETypesMap& archiveMIMETypes()
56 {
57 DEFINE_STATIC_LOCAL(ArchiveMIMETypesMap, mimeTypes, ());
58 static bool initialized = false;
59
60 if (initialized)
61 return mimeTypes;
62
63 #if USE(CF)
64 mimeTypes.set("application/x-webarchive", archiveFactoryCreate<LegacyWebArchive>);
65 #endif
66
67 initialized = true;
68 return mimeTypes;
69 }
70
isArchiveMimeType(const String & mimeType)71 bool ArchiveFactory::isArchiveMimeType(const String& mimeType)
72 {
73 return !mimeType.isEmpty() && archiveMIMETypes().contains(mimeType);
74 }
75
create(SharedBuffer * data,const String & mimeType)76 PassRefPtr<Archive> ArchiveFactory::create(SharedBuffer* data, const String& mimeType)
77 {
78 RawDataCreationFunction* function = mimeType.isEmpty() ? 0 : archiveMIMETypes().get(mimeType);
79 return function ? function(data) : PassRefPtr<Archive>(0);
80 }
81
registerKnownArchiveMIMETypes()82 void ArchiveFactory::registerKnownArchiveMIMETypes()
83 {
84 HashSet<String>& mimeTypes = MIMETypeRegistry::getSupportedNonImageMIMETypes();
85 ArchiveMIMETypesMap::iterator i = archiveMIMETypes().begin();
86 ArchiveMIMETypesMap::iterator end = archiveMIMETypes().end();
87
88 for (; i != end; ++i)
89 mimeTypes.add(i->first);
90 }
91
92 }
93