68a378060840b008e13cbec5ff02cd5eb9a986e2
[WebKit-https.git] / 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(CFURLCONNECTION)
30
31 #include <pal/spi/cf/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             m_httpStatusCode = CFHTTPMessageGetResponseStatusCode(httpResponse);
90             
91             if (initLevel < AllFields) {
92                 RetainPtr<CFDictionaryRef> headers = adoptCF(CFHTTPMessageCopyAllHeaderFields(httpResponse));
93                 for (auto& commonHeader : commonHeaderFields) {
94                     CFStringRef value;
95                     if (CFDictionaryGetValueIfPresent(headers.get(), commonHeader, (const void **)&value))
96                         m_httpHeaderFields.set(commonHeader, value);
97                 }
98             }
99         } else
100             m_httpStatusCode = 0;
101     }
102
103     if (m_initLevel < AllFields && initLevel == AllFields) {
104         CFHTTPMessageRef httpResponse = CFURLResponseGetHTTPResponse(m_cfResponse.get());
105         if (httpResponse) {
106             m_httpVersion = String(adoptCF(CFHTTPMessageCopyVersion(httpResponse)).get()).convertToASCIIUppercase();
107
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(CFURLCONNECTION)