a840f1dee3170ec04b37f441ca723789894d47b3
[WebKit-https.git] / Source / WebKit2 / NetworkProcess / NetworkSession.h
1 /*
2  * Copyright (C) 2015 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 #pragma once
27
28 #if PLATFORM(COCOA)
29 OBJC_CLASS NSURLSession;
30 OBJC_CLASS NSOperationQueue;
31 OBJC_CLASS WKNetworkSessionDelegate;
32 #endif
33
34 #include "DownloadID.h"
35 #include "NetworkDataTask.h"
36 #include <WebCore/SessionID.h>
37 #include <wtf/HashMap.h>
38 #include <wtf/Ref.h>
39 #include <wtf/RefCounted.h>
40
41 namespace WebCore {
42 class NetworkStorageSession;
43 }
44
45 namespace WebKit {
46
47 class CustomProtocolManager;
48
49 class NetworkSession : public RefCounted<NetworkSession> {
50     friend class NetworkDataTask;
51 public:
52     enum class Type {
53         Normal,
54         Ephemeral
55     };
56
57     static Ref<NetworkSession> create(Type, WebCore::SessionID, CustomProtocolManager*);
58     static NetworkSession& defaultSession();
59     ~NetworkSession();
60
61     void invalidateAndCancel();
62
63     WebCore::SessionID sessionID() const { return m_sessionID; }
64
65     // Must be called before any NetworkSession has been created.
66     static void setCustomProtocolManager(CustomProtocolManager*);
67 #if PLATFORM(COCOA)
68     static void setSourceApplicationAuditTokenData(RetainPtr<CFDataRef>&&);
69     static void setSourceApplicationBundleIdentifier(const String&);
70     static void setSourceApplicationSecondaryIdentifier(const String&);
71 #if PLATFORM(IOS)
72     static void setCTDataConnectionServiceType(const String&);
73 #endif
74 #endif
75
76     void clearCredentials();
77
78     NetworkDataTask* dataTaskForIdentifier(NetworkDataTask::TaskIdentifier, WebCore::StoredCredentials);
79
80     void addDownloadID(NetworkDataTask::TaskIdentifier, DownloadID);
81     DownloadID downloadID(NetworkDataTask::TaskIdentifier);
82     DownloadID takeDownloadID(NetworkDataTask::TaskIdentifier);
83     
84 private:
85     NetworkSession(Type, WebCore::SessionID, CustomProtocolManager*);
86     WebCore::NetworkStorageSession& networkStorageSession();
87
88     WebCore::SessionID m_sessionID;
89
90     HashMap<NetworkDataTask::TaskIdentifier, NetworkDataTask*> m_dataTaskMapWithCredentials;
91     HashMap<NetworkDataTask::TaskIdentifier, NetworkDataTask*> m_dataTaskMapWithoutCredentials;
92     HashMap<NetworkDataTask::TaskIdentifier, DownloadID> m_downloadMap;
93
94 #if PLATFORM(COCOA)
95     RetainPtr<NSURLSession> m_sessionWithCredentialStorage;
96     RetainPtr<WKNetworkSessionDelegate> m_sessionWithCredentialStorageDelegate;
97     RetainPtr<NSURLSession> m_sessionWithoutCredentialStorage;
98     RetainPtr<WKNetworkSessionDelegate> m_sessionWithoutCredentialStorageDelegate;
99 #endif
100 };
101
102 } // namespace WebKit