Start of support for multiple WebsiteDataStore/SessionIDs per process
[WebKit-https.git] / Source / WebKit2 / WebProcess / WebCoreSupport / mac / WebFrameNetworkingContext.mm
1 /*
2  * Copyright (C) 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 #include "config.h"
27 #include "WebFrameNetworkingContext.h"
28
29 #include "NetworkSession.h"
30 #include "SessionTracker.h"
31 #include "WebCookieManager.h"
32 #include "WebPage.h"
33 #include "WebProcess.h"
34 #include "WebsiteDataStoreParameters.h"
35 #include <WebCore/CFNetworkSPI.h>
36 #include <WebCore/Frame.h>
37 #include <WebCore/FrameLoader.h>
38 #include <WebCore/FrameLoaderClient.h>
39 #include <WebCore/NetworkStorageSession.h>
40 #include <WebCore/Page.h>
41 #include <WebCore/ResourceError.h>
42 #include <WebCore/Settings.h>
43 #include <WebKitSystemInterface.h>
44
45 using namespace WebCore;
46
47 namespace WebKit {
48     
49 void WebFrameNetworkingContext::ensurePrivateBrowsingSession(SessionID sessionID)
50 {
51     ASSERT(sessionID.isEphemeral());
52
53     if (WebCore::NetworkStorageSession::storageSession(sessionID))
54         return;
55
56     String base;
57     if (SessionTracker::getIdentifierBase().isNull())
58         base = [[NSBundle mainBundle] bundleIdentifier];
59     else
60         base = SessionTracker::getIdentifierBase();
61
62     NetworkStorageSession::ensurePrivateBrowsingSession(sessionID, base + '.' + String::number(sessionID.sessionID()));
63 #if USE(NETWORK_SESSION)
64     auto networkSession = NetworkSession::create(sessionID);
65     SessionTracker::setSession(sessionID, WTFMove(networkSession));
66 #endif
67 }
68
69 void WebFrameNetworkingContext::ensureWebsiteDataStoreSession(WebsiteDataStoreParameters&& parameters)
70 {
71     if (NetworkStorageSession::storageSession(parameters.sessionID))
72         return;
73
74     String base;
75     if (SessionTracker::getIdentifierBase().isNull())
76         base = [[NSBundle mainBundle] bundleIdentifier];
77     else
78         base = SessionTracker::getIdentifierBase();
79
80 #if PLATFORM(IOS)
81     SandboxExtension::consumePermanently(parameters.cookieStorageDirectoryExtensionHandle);
82 #endif
83
84     RetainPtr<CFHTTPCookieStorageRef> uiProcessCookieStorage;
85
86 #if PLATFORM(MAC) && __MAC_OS_X_VERSION_MIN_REQUIRED >= 101100
87     RetainPtr<CFDataRef> cookieStorageData = adoptCF(CFDataCreate(kCFAllocatorDefault, parameters.uiProcessCookieStorageIdentifier.data(), parameters.uiProcessCookieStorageIdentifier.size()));
88     uiProcessCookieStorage = adoptCF(CFHTTPCookieStorageCreateFromIdentifyingData(kCFAllocatorDefault, cookieStorageData.get()));
89 #endif
90
91     NetworkStorageSession::ensureSession(parameters.sessionID, base + '.' + String::number(parameters.sessionID.sessionID()), WTFMove(uiProcessCookieStorage));
92
93 #if USE(NETWORK_SESSION)
94     auto networkSession = NetworkSession::create(parameters.sessionID);
95     SessionTracker::setSession(parameters.sessionID, WTFMove(networkSession));
96 #endif
97 }
98
99 void WebFrameNetworkingContext::setCookieAcceptPolicyForAllContexts(HTTPCookieAcceptPolicy policy)
100 {
101     [[NSHTTPCookieStorage sharedHTTPCookieStorage] setCookieAcceptPolicy:static_cast<NSHTTPCookieAcceptPolicy>(policy)];
102
103     NetworkStorageSession::forEach([&] (const NetworkStorageSession& networkStorageSession) {
104         if (auto cookieStorage = networkStorageSession.cookieStorage())
105             CFHTTPCookieStorageSetCookieAcceptPolicy(cookieStorage.get(), policy);
106     });
107 }
108     
109 bool WebFrameNetworkingContext::localFileContentSniffingEnabled() const
110 {
111     return frame() && frame()->settings().localFileContentSniffingEnabled();
112 }
113
114 SchedulePairHashSet* WebFrameNetworkingContext::scheduledRunLoopPairs() const
115 {
116     if (!frame() || !frame()->page())
117         return nullptr;
118     return frame()->page()->scheduledRunLoopPairs();
119 }
120
121 RetainPtr<CFDataRef> WebFrameNetworkingContext::sourceApplicationAuditData() const
122 {
123     return WebProcess::singleton().sourceApplicationAuditData();
124 }
125
126 String WebFrameNetworkingContext::sourceApplicationIdentifier() const
127 {
128     return SessionTracker::getIdentifierBase();
129 }
130
131 ResourceError WebFrameNetworkingContext::blockedError(const ResourceRequest& request) const
132 {
133     return frame()->loader().client().blockedError(request);
134 }
135
136 NetworkStorageSession& WebFrameNetworkingContext::storageSession() const
137 {
138     ASSERT(RunLoop::isMain());
139     if (frame()) {
140         if (auto* storageSession = WebCore::NetworkStorageSession::storageSession(frame()->page()->sessionID()))
141             return *storageSession;
142         // Some requests may still be coming shortly after WebProcess was told to destroy its session.
143         LOG_ERROR("WEB Invalid session ID. Please file a bug unless you just disabled private browsing, in which case it's an expected race.");
144     }
145     return NetworkStorageSession::defaultStorageSession();
146 }
147
148 WebFrameLoaderClient* WebFrameNetworkingContext::webFrameLoaderClient() const
149 {
150     if (!frame())
151         return 0;
152
153     return toWebFrameLoaderClient(frame()->loader().client());
154 }
155
156 }