Move URL from WebCore to WTF
[WebKit-https.git] / Source / WebKit / NetworkProcess / CustomProtocols / Cocoa / LegacyCustomProtocolManagerCocoa.mm
1 /*
2  * Copyright (C) 2012, 2013 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. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #import "config.h"
27 #import "LegacyCustomProtocolManager.h"
28
29 #import "DataReference.h"
30 #import "LegacyCustomProtocolManagerMessages.h"
31 #import "NetworkProcess.h"
32 #import <Foundation/NSURLSession.h>
33 #import <WebCore/ResourceError.h>
34 #import <WebCore/ResourceRequest.h>
35 #import <WebCore/ResourceResponse.h>
36 #import <WebCore/TextEncoding.h>
37 #import <pal/spi/cocoa/NSURLConnectionSPI.h>
38 #import <wtf/URL.h>
39
40 using namespace WebKit;
41
42 @interface WKCustomProtocol : NSURLProtocol {
43 @private
44     uint64_t _customProtocolID;
45     RetainPtr<CFRunLoopRef> _initializationRunLoop;
46 }
47 @property (nonatomic, readonly) uint64_t customProtocolID;
48 @property (nonatomic, readonly) CFRunLoopRef initializationRunLoop;
49 @end
50
51 @implementation WKCustomProtocol
52
53 @synthesize customProtocolID=_customProtocolID;
54
55 + (BOOL)canInitWithRequest:(NSURLRequest *)request
56 {
57     if (auto* customProtocolManager = NetworkProcess::singleton().supplement<LegacyCustomProtocolManager>())
58         return customProtocolManager->supportsScheme([[[request URL] scheme] lowercaseString]);
59     return NO;
60 }
61
62 + (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request
63 {
64     return request;
65 }
66
67 + (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b
68 {
69     return NO;
70 }
71
72 - (id)initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)cachedResponse client:(id<NSURLProtocolClient>)client
73 {
74     self = [super initWithRequest:request cachedResponse:cachedResponse client:client];
75     if (!self)
76         return nil;
77
78     if (auto* customProtocolManager = NetworkProcess::singleton().supplement<LegacyCustomProtocolManager>())
79         _customProtocolID = customProtocolManager->addCustomProtocol(self);
80     _initializationRunLoop = CFRunLoopGetCurrent();
81
82     return self;
83 }
84
85 - (CFRunLoopRef)initializationRunLoop
86 {
87     return _initializationRunLoop.get();
88 }
89
90 - (void)startLoading
91 {
92     if (auto* customProtocolManager = NetworkProcess::singleton().supplement<LegacyCustomProtocolManager>())
93         customProtocolManager->startLoading(self.customProtocolID, [self request]);
94 }
95
96 - (void)stopLoading
97 {
98     if (auto* customProtocolManager = NetworkProcess::singleton().supplement<LegacyCustomProtocolManager>()) {
99         customProtocolManager->stopLoading(self.customProtocolID);
100         customProtocolManager->removeCustomProtocol(self.customProtocolID);
101     }
102 }
103
104 @end
105
106 namespace WebKit {
107
108 void LegacyCustomProtocolManager::registerProtocolClass()
109 {
110 }
111
112 void LegacyCustomProtocolManager::registerProtocolClass(NSURLSessionConfiguration *configuration)
113 {
114     configuration.protocolClasses = @[[WKCustomProtocol class]];
115 }
116
117 void LegacyCustomProtocolManager::registerScheme(const String& scheme)
118 {
119     ASSERT(!scheme.isNull());
120     LockHolder locker(m_registeredSchemesMutex);
121     m_registeredSchemes.add(scheme);
122 }
123
124 void LegacyCustomProtocolManager::unregisterScheme(const String& scheme)
125 {
126     ASSERT(!scheme.isNull());
127     LockHolder locker(m_registeredSchemesMutex);
128     m_registeredSchemes.remove(scheme);
129 }
130
131 bool LegacyCustomProtocolManager::supportsScheme(const String& scheme)
132 {
133     if (scheme.isNull())
134         return false;
135
136     LockHolder locker(m_registeredSchemesMutex);
137     return m_registeredSchemes.contains(scheme);
138 }
139
140 static inline void dispatchOnInitializationRunLoop(WKCustomProtocol* protocol, void (^block)())
141 {
142     CFRunLoopRef runloop = protocol.initializationRunLoop;
143     CFRunLoopPerformBlock(runloop, kCFRunLoopDefaultMode, block);
144     CFRunLoopWakeUp(runloop);
145 }
146
147 void LegacyCustomProtocolManager::didFailWithError(uint64_t customProtocolID, const WebCore::ResourceError& error)
148 {
149     RetainPtr<WKCustomProtocol> protocol = protocolForID(customProtocolID);
150     if (!protocol)
151         return;
152
153     RetainPtr<NSError> nsError = error.nsError();
154
155     dispatchOnInitializationRunLoop(protocol.get(), ^ {
156         [[protocol client] URLProtocol:protocol.get() didFailWithError:nsError.get()];
157     });
158
159     removeCustomProtocol(customProtocolID);
160 }
161
162 void LegacyCustomProtocolManager::didLoadData(uint64_t customProtocolID, const IPC::DataReference& data)
163 {
164     RetainPtr<WKCustomProtocol> protocol = protocolForID(customProtocolID);
165     if (!protocol)
166         return;
167
168     RetainPtr<NSData> nsData = adoptNS([[NSData alloc] initWithBytes:data.data() length:data.size()]);
169
170     dispatchOnInitializationRunLoop(protocol.get(), ^ {
171         [[protocol client] URLProtocol:protocol.get() didLoadData:nsData.get()];
172     });
173 }
174
175 void LegacyCustomProtocolManager::didReceiveResponse(uint64_t customProtocolID, const WebCore::ResourceResponse& response, uint32_t cacheStoragePolicy)
176 {
177     RetainPtr<WKCustomProtocol> protocol = protocolForID(customProtocolID);
178     if (!protocol)
179         return;
180
181     RetainPtr<NSURLResponse> nsResponse = response.nsURLResponse();
182
183     dispatchOnInitializationRunLoop(protocol.get(), ^ {
184         [[protocol client] URLProtocol:protocol.get() didReceiveResponse:nsResponse.get() cacheStoragePolicy:static_cast<NSURLCacheStoragePolicy>(cacheStoragePolicy)];
185     });
186 }
187
188 void LegacyCustomProtocolManager::didFinishLoading(uint64_t customProtocolID)
189 {
190     RetainPtr<WKCustomProtocol> protocol = protocolForID(customProtocolID);
191     if (!protocol)
192         return;
193
194     dispatchOnInitializationRunLoop(protocol.get(), ^ {
195         [[protocol client] URLProtocolDidFinishLoading:protocol.get()];
196     });
197
198     removeCustomProtocol(customProtocolID);
199 }
200
201 void LegacyCustomProtocolManager::wasRedirectedToRequest(uint64_t customProtocolID, const WebCore::ResourceRequest& request, const WebCore::ResourceResponse& redirectResponse)
202 {
203     RetainPtr<WKCustomProtocol> protocol = protocolForID(customProtocolID);
204     if (!protocol)
205         return;
206
207     RetainPtr<NSURLRequest> nsRequest = request.nsURLRequest(WebCore::HTTPBodyUpdatePolicy::DoNotUpdateHTTPBody);
208     RetainPtr<NSURLResponse> nsRedirectResponse = redirectResponse.nsURLResponse();
209
210     dispatchOnInitializationRunLoop(protocol.get(), [protocol, nsRequest, nsRedirectResponse]() {
211         [[protocol client] URLProtocol:protocol.get() wasRedirectedToRequest:nsRequest.get() redirectResponse:nsRedirectResponse.get()];
212     });
213 }
214
215 RetainPtr<WKCustomProtocol> LegacyCustomProtocolManager::protocolForID(uint64_t customProtocolID)
216 {
217     LockHolder locker(m_customProtocolMapMutex);
218
219     CustomProtocolMap::const_iterator it = m_customProtocolMap.find(customProtocolID);
220     if (it == m_customProtocolMap.end())
221         return nil;
222     
223     ASSERT(it->value);
224     return it->value;
225 }
226
227 } // namespace WebKit