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 #import "QuickLook.h"
  33 #import "UTIUtilities.h"
  34 #import "WebCoreSystemInterface.h"
  35 #import <MobileCoreServices/MobileCoreServices.h>
  36
  37 #import "QuickLookSoftLink.h"
  38
  39 namespace WebCore {
  40
  41 void adjustMIMETypeIfNecessary(CFURLResponseRef cfResponse, bool isMainResourceLoad)
  42 {
  43     RetainPtr<CFStringRef> mimeType = CFURLResponseGetMIMEType(cfResponse);
  44     RetainPtr<CFStringRef> updatedMIMEType = mimeType;
  45     if (!updatedMIMEType)
  46         updatedMIMEType = defaultMIMEType().createCFString();
  47
  48 #if USE(QUICK_LOOK)
  49     // We must ensure that the MIME type is correct, so that QuickLook's web plugin is called when needed.
  50     // We filter the basic MIME types so that we don't do unnecessary work in standard browsing situations.
  51     if (isMainResourceLoad && shouldUseQuickLookForMIMEType((NSString *)updatedMIMEType.get())) {
  52         RetainPtr<CFStringRef> suggestedFilename = adoptCF(CFURLResponseCopySuggestedFilename(cfResponse));
  53         RetainPtr<CFStringRef> quickLookMIMEType = adoptCF((CFStringRef)QLTypeCopyBestMimeTypeForFileNameAndMimeType((NSString *)suggestedFilename.get(), (NSString *)mimeType.get()));
  54         if (!quickLookMIMEType) {
  55             auto url = CFURLResponseGetURL(cfResponse);
  56             if ([(NSURL *)url isFileURL]) {
  57                 RetainPtr<CFStringRef> extension = adoptCF(CFURLCopyPathExtension(url));
  58                 if (extension) {
  59                     RetainPtr<CFStringRef> uti = adoptCF(UTTypeCreatePreferredIdentifierForTag(kUTTagClassFilenameExtension, extension.get(), nullptr));
  60                     quickLookMIMEType = mimeTypeFromUTITree(uti.get());
  61                 }
  62             }
  63         }
  64
  65         if (quickLookMIMEType)
  66             updatedMIMEType = quickLookMIMEType;
  67     }
  68 #else
  69     UNUSED_PARAM(isMainResourceLoad);
  70 #endif // USE(QUICK_LOOK)
  71     if (!mimeType || CFStringCompare(mimeType.get(), updatedMIMEType.get(), kCFCompareCaseInsensitive) != kCFCompareEqualTo)
  72         CFURLResponseSetMIMEType(cfResponse, updatedMIMEType.get());
  73 }
  74
  75 } // namespace WebCore