Source/WebCore/platform/network/cf/ResourceResponseCFNet.cpp

   1 /*
   2  * Copyright (C) 2006-2007, 2016 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. ``AS IS'' AND ANY
  14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
  17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
  21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "ResourceResponse.h"
  28
  29 #if USE(CFNETWORK)
  30
  31 #include "CFNetworkSPI.h"
  32
  33 #include "HTTPParsers.h"
  34 #include "MIMETypeRegistry.h"
  35 #include <wtf/RetainPtr.h>
  36
  37 #if PLATFORM(COCOA)
  38 #include "WebCoreSystemInterface.h"
  39 #endif
  40
  41 namespace WebCore {
  42
  43 static CFStringRef const commonHeaderFields[] = {
  44     CFSTR("Age"), CFSTR("Cache-Control"), CFSTR("Content-Type"), CFSTR("Date"), CFSTR("Etag"), CFSTR("Expires"), CFSTR("Last-Modified"), CFSTR("Pragma")
  45 };
  46
  47 CFURLResponseRef ResourceResponse::cfURLResponse() const
  48 {
  49     if (!m_cfResponse && !m_isNull) {
  50 #if PLATFORM(COCOA)
  51         nsURLResponse();
  52 #else
  53         RetainPtr<CFURLRef> url = m_url.createCFURL();
  54         // FIXME: This creates a very incomplete CFURLResponse, which does not even have a status code.
  55         m_cfResponse = adoptCF(CFURLResponseCreate(0, url.get(), m_mimeType.string().createCFString().get(), m_expectedContentLength, m_textEncodingName.string().createCFString().get(), kCFURLCacheStorageAllowed));
  56 #endif
  57     }
  58
  59     return m_cfResponse.get();
  60 }
  61
  62 static void addToHTTPHeaderMap(const void* key, const void* value, void* context)
  63 {
  64     HTTPHeaderMap* httpHeaderMap = (HTTPHeaderMap*)context;
  65     httpHeaderMap->set((CFStringRef)key, (CFStringRef)value);
  66 }
  67
  68 void ResourceResponse::platformLazyInit(InitLevel initLevel)
  69 {
  70     if (m_initLevel > initLevel)
  71         return;
  72
  73     if (m_isNull || !m_cfResponse.get())
  74         return;
  75
  76     if (m_initLevel < CommonFieldsOnly && initLevel >= CommonFieldsOnly) {
  77         m_url = CFURLResponseGetURL(m_cfResponse.get());
  78         m_mimeType = CFURLResponseGetMIMEType(m_cfResponse.get());
  79         m_expectedContentLength = CFURLResponseGetExpectedContentLength(m_cfResponse.get());
  80         m_textEncodingName = CFURLResponseGetTextEncodingName(m_cfResponse.get());
  81
  82         // Workaround for <rdar://problem/8757088>, can be removed once that is fixed.
  83         unsigned textEncodingNameLength = m_textEncodingName.length();
  84         if (textEncodingNameLength >= 2 && m_textEncodingName[0U] == '"' && m_textEncodingName[textEncodingNameLength - 1] == '"')
  85             m_textEncodingName = m_textEncodingName.string().substring(1, textEncodingNameLength - 2);
  86
  87         CFHTTPMessageRef httpResponse = CFURLResponseGetHTTPResponse(m_cfResponse.get());
  88         if (httpResponse) {
  89             RetainPtr<CFStringRef> messageString = adoptCF(CFHTTPMessageCopyVersion(httpResponse));
  90             m_httpVersion = String(messageString.get()).upper();
  91             m_httpStatusCode = CFHTTPMessageGetResponseStatusCode(httpResponse);
  92
  93             if (initLevel < AllFields) {
  94                 RetainPtr<CFDictionaryRef> headers = adoptCF(CFHTTPMessageCopyAllHeaderFields(httpResponse));
  95                 for (auto& commonHeader : commonHeaderFields) {
  96                     CFStringRef value;
  97                     if (CFDictionaryGetValueIfPresent(headers.get(), commonHeader, (const void **)&value))
  98                         m_httpHeaderFields.set(commonHeader, value);
  99                 }
 100             }
 101         } else
 102             m_httpStatusCode = 0;
 103     }
 104
 105     if (m_initLevel < AllFields && initLevel == AllFields) {
 106         CFHTTPMessageRef httpResponse = CFURLResponseGetHTTPResponse(m_cfResponse.get());
 107         if (httpResponse) {
 108             RetainPtr<CFStringRef> statusLine = adoptCF(CFHTTPMessageCopyResponseStatusLine(httpResponse));
 109             m_httpStatusText = extractReasonPhraseFromHTTPStatusLine(statusLine.get());
 110
 111             RetainPtr<CFDictionaryRef> headers = adoptCF(CFHTTPMessageCopyAllHeaderFields(httpResponse));
 112             CFDictionaryApplyFunction(headers.get(), addToHTTPHeaderMap, &m_httpHeaderFields);
 113         }
 114     }
 115
 116     m_initLevel = initLevel;
 117 }
 118
 119 #if !PLATFORM(COCOA)
 120 CertificateInfo ResourceResponse::platformCertificateInfo() const
 121 {
 122     return { };
 123 }
 124 #endif
 125
 126 String ResourceResponse::platformSuggestedFilename() const
 127 {
 128     if (!cfURLResponse())
 129         return String();
 130     RetainPtr<CFStringRef> suggestedFilename = adoptCF(CFURLResponseCopySuggestedFilename(cfURLResponse()));
 131     return suggestedFilename.get();
 132 }
 133
 134 bool ResourceResponse::platformCompare(const ResourceResponse& a, const ResourceResponse& b)
 135 {
 136     // CFEqual crashes if you pass it 0 so do an early check before calling it.
 137     if (!a.cfURLResponse() || !b.cfURLResponse())
 138         return a.cfURLResponse() == b.cfURLResponse();
 139     return CFEqual(a.cfURLResponse(), b.cfURLResponse());
 140 }
 141
 142
 143 } // namespace WebCore
 144
 145 #endif // USE(CFNETWORK)