Fix iOS Simulator Release test crashes.
[WebKit-https.git] / Source / WebCore / platform / network / ios / WebCoreURLResponseIOS.mm
1 /*
2  * Copyright (C) 2009 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 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 #import "config.h"
30 #import "WebCoreURLResponseIOS.h"
31
32 #if PLATFORM(IOS_FAMILY)
33
34 #import "QuickLook.h"
35 #import "UTIUtilities.h"
36 #import <MobileCoreServices/MobileCoreServices.h>
37
38 #import "QuickLookSoftLink.h"
39
40 namespace WebCore {
41
42 // <rdar://problem/46332893> Register .mjs files as whatever UTI indicates JavaScript
43 static CFDictionaryRef createExtensionToMIMETypeMap()
44 {
45     CFStringRef keys[] = {
46         CFSTR("mjs")
47     };
48
49     CFStringRef values[] = {
50         CFSTR("application/javascript")
51     };
52
53     ASSERT(sizeof(keys) == sizeof(values));
54     return CFDictionaryCreate(kCFAllocatorDefault, (const void**)&keys, (const void**)&values, sizeof(keys) / sizeof(CFStringRef), &kCFTypeDictionaryKeyCallBacks, &kCFTypeDictionaryValueCallBacks);
55 }
56
57 void adjustMIMETypeIfNecessary(CFURLResponseRef cfResponse, bool isMainResourceLoad)
58 {
59     RetainPtr<CFStringRef> mimeType = CFURLResponseGetMIMEType(cfResponse);
60     RetainPtr<CFStringRef> updatedMIMEType = mimeType;
61     if (!updatedMIMEType)
62         updatedMIMEType = defaultMIMEType().createCFString();
63
64     // <rdar://problem/46332893> Register .mjs files as whatever UTI indicates JavaScript
65     if (!mimeType) {
66         auto url = CFURLResponseGetURL(cfResponse);
67         if ([(__bridge NSURL *)url isFileURL]) {
68             RetainPtr<CFStringRef> extension = adoptCF(CFURLCopyPathExtension(url));
69             if (extension) {
70                 static CFDictionaryRef extensionMap = createExtensionToMIMETypeMap();
71                 CFMutableStringRef mutableExtension = CFStringCreateMutableCopy(kCFAllocatorDefault, 0, extension.get());
72                 CFStringLowercase(mutableExtension, NULL);
73                 extension = adoptCF(mutableExtension);
74                 if (auto newMIMEType = (CFStringRef)CFDictionaryGetValue(extensionMap, extension.get()))
75                     updatedMIMEType = newMIMEType;
76             }
77         }
78     }
79
80 #if USE(QUICK_LOOK)
81     // We must ensure that the MIME type is correct, so that QuickLook's web plugin is called when needed.
82     // We filter the basic MIME types so that we don't do unnecessary work in standard browsing situations.
83     if (isMainResourceLoad && shouldUseQuickLookForMIMEType((NSString *)updatedMIMEType.get())) {
84         RetainPtr<CFStringRef> suggestedFilename = adoptCF(CFURLResponseCopySuggestedFilename(cfResponse));
85         RetainPtr<CFStringRef> quickLookMIMEType = adoptCF((CFStringRef)QLTypeCopyBestMimeTypeForFileNameAndMimeType((NSString *)suggestedFilename.get(), (NSString *)mimeType.get()));
86         if (!quickLookMIMEType) {
87             auto url = CFURLResponseGetURL(cfResponse);
88             if ([(NSURL *)url isFileURL]) {
89                 RetainPtr<CFStringRef> extension = adoptCF(CFURLCopyPathExtension(url));
90                 if (extension) {
91                     RetainPtr<CFStringRef> uti = adoptCF(UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, extension.get(), nullptr));
92                     String MIMEType = MIMETypeFromUTITree(uti.get());
93                     if (!MIMEType.isEmpty())
94                         quickLookMIMEType = MIMEType.createCFString();
95                 }
96             }
97         }
98
99         if (quickLookMIMEType)
100             updatedMIMEType = quickLookMIMEType;
101     }
102 #else
103     UNUSED_PARAM(isMainResourceLoad);
104 #endif // USE(QUICK_LOOK)
105     if (!mimeType || CFStringCompare(mimeType.get(), updatedMIMEType.get(), kCFCompareCaseInsensitive) != kCFCompareEqualTo)
106         CFURLResponseSetMIMEType(cfResponse, updatedMIMEType.get());
107 }
108
109 } // namespace WebCore
110
111 #endif // PLATFORM(IOS_FAMILY)