e5ae80758010b5cb2cbf1b99a5ed6ce10731bbac
[WebKit-https.git] / Source / WebKitLegacy / WebCoreSupport / NetworkStorageSessionMap.cpp
1 /*
2  * Copyright (C) 2019 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 "NetworkStorageSessionMap.h"
27
28 #include <WebCore/NetworkStorageSession.h>
29 #include <pal/SessionID.h>
30 #include <wtf/MainThread.h>
31 #include <wtf/ProcessID.h>
32 #include <wtf/ProcessPrivilege.h>
33 #include <wtf/text/StringConcatenateNumbers.h>
34
35 static std::unique_ptr<WebCore::NetworkStorageSession>& defaultNetworkStorageSession()
36 {
37     ASSERT(isMainThread());
38     static NeverDestroyed<std::unique_ptr<WebCore::NetworkStorageSession>> session;
39     return session;
40 }
41
42 static HashMap<PAL::SessionID, std::unique_ptr<WebCore::NetworkStorageSession>>& globalSessionMap()
43 {
44     static NeverDestroyed<HashMap<PAL::SessionID, std::unique_ptr<WebCore::NetworkStorageSession>>> map;
45     return map;
46 }
47
48 WebCore::NetworkStorageSession* NetworkStorageSessionMap::storageSession(const PAL::SessionID& sessionID)
49 {
50     if (sessionID == PAL::SessionID::defaultSessionID())
51         return &defaultStorageSession();
52     return globalSessionMap().get(sessionID);
53 }
54
55 WebCore::NetworkStorageSession& NetworkStorageSessionMap::defaultStorageSession()
56 {
57     if (!defaultNetworkStorageSession()) {
58 #if USE(CURL)
59         defaultNetworkStorageSession() = std::make_unique<WebCore::NetworkStorageSession>(PAL::SessionID::defaultSessionID(), nullptr);
60 #else
61         defaultNetworkStorageSession() = std::make_unique<WebCore::NetworkStorageSession>(PAL::SessionID::defaultSessionID());
62 #endif
63     }
64     return *defaultNetworkStorageSession();
65 }
66
67 void NetworkStorageSessionMap::switchToNewTestingSession()
68 {
69 #if PLATFORM(COCOA) || USE(CFURLCONNECTION)
70     // Session name should be short enough for shared memory region name to be under the limit, otehrwise sandbox rules won't work (see <rdar://problem/13642852>).
71     String sessionName = makeString("WebKit Test-", getCurrentProcessID());
72
73     auto session = adoptCF(WebCore::createPrivateStorageSession(sessionName.createCFString().get()));
74
75     RetainPtr<CFHTTPCookieStorageRef> cookieStorage;
76     if (WebCore::NetworkStorageSession::processMayUseCookieAPI()) {
77         ASSERT(hasProcessPrivilege(ProcessPrivilege::CanAccessRawCookies));
78         if (session)
79             cookieStorage = adoptCF(_CFURLStorageSessionCopyCookieStorage(kCFAllocatorDefault, session.get()));
80     }
81
82     defaultNetworkStorageSession() = std::make_unique<WebCore::NetworkStorageSession>(PAL::SessionID::defaultSessionID(), WTFMove(session), WTFMove(cookieStorage));
83 #endif
84 }
85
86 void NetworkStorageSessionMap::ensureSession(const PAL::SessionID& sessionID, const String& identifierBase)
87 {
88 #if PLATFORM(COCOA) || USE(CFURLCONNECTION)
89     auto addResult = globalSessionMap().add(sessionID, nullptr);
90     if (!addResult.isNewEntry)
91         return;
92
93     RetainPtr<CFStringRef> cfIdentifier = String(identifierBase + ".PrivateBrowsing").createCFString();
94
95     RetainPtr<CFURLStorageSessionRef> storageSession;
96     if (sessionID.isEphemeral())
97         storageSession = adoptCF(WebCore::createPrivateStorageSession(cfIdentifier.get()));
98     else
99         storageSession = WebCore::NetworkStorageSession::createCFStorageSessionForIdentifier(cfIdentifier.get());
100
101     RetainPtr<CFHTTPCookieStorageRef> cookieStorage;
102     if (WebCore::NetworkStorageSession::processMayUseCookieAPI()) {
103         ASSERT(hasProcessPrivilege(ProcessPrivilege::CanAccessRawCookies));
104         if (storageSession)
105             cookieStorage = adoptCF(_CFURLStorageSessionCopyCookieStorage(kCFAllocatorDefault, storageSession.get()));
106     }
107
108     addResult.iterator->value = std::make_unique<WebCore::NetworkStorageSession>(sessionID, WTFMove(storageSession), WTFMove(cookieStorage));
109
110 #elif USE(CURL)
111     globalSessionMap().ensure(sessionID, [sessionID] {
112         return std::make_unique<WebCore::NetworkStorageSession>(sessionID, nullptr);
113     });
114 #endif
115 }
116
117 void NetworkStorageSessionMap::destroySession(const PAL::SessionID& sessionID)
118 {
119     globalSessionMap().remove(sessionID);
120 }