OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ | 5 #ifndef NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ |
6 #define NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ | 6 #define NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ |
7 | 7 |
8 #include <list> | 8 #include <list> |
9 #include <map> | 9 #include <map> |
10 #include <set> | 10 #include <set> |
(...skipping 35 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
46 RequestPriority priority, | 46 RequestPriority priority, |
47 const scoped_refptr<TransportSocketParams>& params, | 47 const scoped_refptr<TransportSocketParams>& params, |
48 base::TimeDelta timeout_duration, | 48 base::TimeDelta timeout_duration, |
49 const CompletionCallback& callback, | 49 const CompletionCallback& callback, |
50 ClientSocketFactory* client_socket_factory, | 50 ClientSocketFactory* client_socket_factory, |
51 HostResolver* host_resolver, | 51 HostResolver* host_resolver, |
52 ClientSocketHandle* handle, | 52 ClientSocketHandle* handle, |
53 Delegate* delegate, | 53 Delegate* delegate, |
54 NetLog* pool_net_log, | 54 NetLog* pool_net_log, |
55 const BoundNetLog& request_net_log); | 55 const BoundNetLog& request_net_log); |
56 virtual ~WebSocketTransportConnectJob(); | 56 ~WebSocketTransportConnectJob() override; |
57 | 57 |
58 // Unlike normal socket pools, the WebSocketTransportClientPool uses | 58 // Unlike normal socket pools, the WebSocketTransportClientPool uses |
59 // early-binding of sockets. | 59 // early-binding of sockets. |
60 ClientSocketHandle* handle() const { return handle_; } | 60 ClientSocketHandle* handle() const { return handle_; } |
61 | 61 |
62 // Stash the callback from RequestSocket() here for convenience. | 62 // Stash the callback from RequestSocket() here for convenience. |
63 const CompletionCallback& callback() const { return callback_; } | 63 const CompletionCallback& callback() const { return callback_; } |
64 | 64 |
65 const BoundNetLog& request_net_log() const { return request_net_log_; } | 65 const BoundNetLog& request_net_log() const { return request_net_log_; } |
66 | 66 |
67 // ConnectJob methods. | 67 // ConnectJob methods. |
68 virtual LoadState GetLoadState() const override; | 68 LoadState GetLoadState() const override; |
69 | 69 |
70 private: | 70 private: |
71 friend class WebSocketTransportConnectSubJob; | 71 friend class WebSocketTransportConnectSubJob; |
72 friend class TransportConnectJobHelper; | 72 friend class TransportConnectJobHelper; |
73 friend class WebSocketEndpointLockManager; | 73 friend class WebSocketEndpointLockManager; |
74 | 74 |
75 // Although it is not strictly necessary, it makes the code simpler if each | 75 // Although it is not strictly necessary, it makes the code simpler if each |
76 // subjob knows what type it is. | 76 // subjob knows what type it is. |
77 enum SubJobType { SUB_JOB_IPV4, SUB_JOB_IPV6 }; | 77 enum SubJobType { SUB_JOB_IPV4, SUB_JOB_IPV6 }; |
78 | 78 |
79 int DoResolveHost(); | 79 int DoResolveHost(); |
80 int DoResolveHostComplete(int result); | 80 int DoResolveHostComplete(int result); |
81 int DoTransportConnect(); | 81 int DoTransportConnect(); |
82 int DoTransportConnectComplete(int result); | 82 int DoTransportConnectComplete(int result); |
83 | 83 |
84 // Called back from a SubJob when it completes. | 84 // Called back from a SubJob when it completes. |
85 void OnSubJobComplete(int result, WebSocketTransportConnectSubJob* job); | 85 void OnSubJobComplete(int result, WebSocketTransportConnectSubJob* job); |
86 | 86 |
87 // Called from |fallback_timer_|. | 87 // Called from |fallback_timer_|. |
88 void StartIPv4JobAsync(); | 88 void StartIPv4JobAsync(); |
89 | 89 |
90 // Begins the host resolution and the TCP connect. Returns OK on success | 90 // Begins the host resolution and the TCP connect. Returns OK on success |
91 // and ERR_IO_PENDING if it cannot immediately service the request. | 91 // and ERR_IO_PENDING if it cannot immediately service the request. |
92 // Otherwise, it returns a net error code. | 92 // Otherwise, it returns a net error code. |
93 virtual int ConnectInternal() override; | 93 int ConnectInternal() override; |
94 | 94 |
95 TransportConnectJobHelper helper_; | 95 TransportConnectJobHelper helper_; |
96 | 96 |
97 // The addresses are divided into IPv4 and IPv6, which are performed partially | 97 // The addresses are divided into IPv4 and IPv6, which are performed partially |
98 // in parallel. If the list of IPv6 addresses is non-empty, then the IPv6 jobs | 98 // in parallel. If the list of IPv6 addresses is non-empty, then the IPv6 jobs |
99 // go first, followed after |kIPv6FallbackTimerInMs| by the IPv4 | 99 // go first, followed after |kIPv6FallbackTimerInMs| by the IPv4 |
100 // addresses. First sub-job to establish a connection wins. | 100 // addresses. First sub-job to establish a connection wins. |
101 scoped_ptr<WebSocketTransportConnectSubJob> ipv4_job_; | 101 scoped_ptr<WebSocketTransportConnectSubJob> ipv4_job_; |
102 scoped_ptr<WebSocketTransportConnectSubJob> ipv6_job_; | 102 scoped_ptr<WebSocketTransportConnectSubJob> ipv6_job_; |
103 | 103 |
(...skipping 12 matching lines...) Expand all Loading... |
116 class NET_EXPORT_PRIVATE WebSocketTransportClientSocketPool | 116 class NET_EXPORT_PRIVATE WebSocketTransportClientSocketPool |
117 : public TransportClientSocketPool { | 117 : public TransportClientSocketPool { |
118 public: | 118 public: |
119 WebSocketTransportClientSocketPool(int max_sockets, | 119 WebSocketTransportClientSocketPool(int max_sockets, |
120 int max_sockets_per_group, | 120 int max_sockets_per_group, |
121 ClientSocketPoolHistograms* histograms, | 121 ClientSocketPoolHistograms* histograms, |
122 HostResolver* host_resolver, | 122 HostResolver* host_resolver, |
123 ClientSocketFactory* client_socket_factory, | 123 ClientSocketFactory* client_socket_factory, |
124 NetLog* net_log); | 124 NetLog* net_log); |
125 | 125 |
126 virtual ~WebSocketTransportClientSocketPool(); | 126 ~WebSocketTransportClientSocketPool() override; |
127 | 127 |
128 // Allow another connection to be started to the IPEndPoint that this |handle| | 128 // Allow another connection to be started to the IPEndPoint that this |handle| |
129 // is connected to. Used when the WebSocket handshake completes successfully. | 129 // is connected to. Used when the WebSocket handshake completes successfully. |
130 // This only works if the socket is connected, however the caller does not | 130 // This only works if the socket is connected, however the caller does not |
131 // need to explicitly check for this. Instead, ensure that dead sockets are | 131 // need to explicitly check for this. Instead, ensure that dead sockets are |
132 // returned to ReleaseSocket() in a timely fashion. | 132 // returned to ReleaseSocket() in a timely fashion. |
133 static void UnlockEndpoint(ClientSocketHandle* handle); | 133 static void UnlockEndpoint(ClientSocketHandle* handle); |
134 | 134 |
135 // ClientSocketPool implementation. | 135 // ClientSocketPool implementation. |
136 virtual int RequestSocket(const std::string& group_name, | 136 int RequestSocket(const std::string& group_name, |
137 const void* resolve_info, | 137 const void* resolve_info, |
138 RequestPriority priority, | 138 RequestPriority priority, |
139 ClientSocketHandle* handle, | 139 ClientSocketHandle* handle, |
140 const CompletionCallback& callback, | 140 const CompletionCallback& callback, |
141 const BoundNetLog& net_log) override; | 141 const BoundNetLog& net_log) override; |
142 virtual void RequestSockets(const std::string& group_name, | 142 void RequestSockets(const std::string& group_name, |
143 const void* params, | 143 const void* params, |
144 int num_sockets, | 144 int num_sockets, |
145 const BoundNetLog& net_log) override; | 145 const BoundNetLog& net_log) override; |
146 virtual void CancelRequest(const std::string& group_name, | 146 void CancelRequest(const std::string& group_name, |
147 ClientSocketHandle* handle) override; | 147 ClientSocketHandle* handle) override; |
148 virtual void ReleaseSocket(const std::string& group_name, | 148 void ReleaseSocket(const std::string& group_name, |
149 scoped_ptr<StreamSocket> socket, | 149 scoped_ptr<StreamSocket> socket, |
150 int id) override; | 150 int id) override; |
151 virtual void FlushWithError(int error) override; | 151 void FlushWithError(int error) override; |
152 virtual void CloseIdleSockets() override; | 152 void CloseIdleSockets() override; |
153 virtual int IdleSocketCount() const override; | 153 int IdleSocketCount() const override; |
154 virtual int IdleSocketCountInGroup( | 154 int IdleSocketCountInGroup(const std::string& group_name) const override; |
155 const std::string& group_name) const override; | 155 LoadState GetLoadState(const std::string& group_name, |
156 virtual LoadState GetLoadState( | 156 const ClientSocketHandle* handle) const override; |
157 const std::string& group_name, | 157 base::DictionaryValue* GetInfoAsValue( |
158 const ClientSocketHandle* handle) const override; | |
159 virtual base::DictionaryValue* GetInfoAsValue( | |
160 const std::string& name, | 158 const std::string& name, |
161 const std::string& type, | 159 const std::string& type, |
162 bool include_nested_pools) const override; | 160 bool include_nested_pools) const override; |
163 virtual base::TimeDelta ConnectionTimeout() const override; | 161 base::TimeDelta ConnectionTimeout() const override; |
164 virtual ClientSocketPoolHistograms* histograms() const override; | 162 ClientSocketPoolHistograms* histograms() const override; |
165 | 163 |
166 // HigherLayeredPool implementation. | 164 // HigherLayeredPool implementation. |
167 virtual bool IsStalled() const override; | 165 bool IsStalled() const override; |
168 | 166 |
169 private: | 167 private: |
170 class ConnectJobDelegate : public ConnectJob::Delegate { | 168 class ConnectJobDelegate : public ConnectJob::Delegate { |
171 public: | 169 public: |
172 explicit ConnectJobDelegate(WebSocketTransportClientSocketPool* owner); | 170 explicit ConnectJobDelegate(WebSocketTransportClientSocketPool* owner); |
173 virtual ~ConnectJobDelegate(); | 171 ~ConnectJobDelegate() override; |
174 | 172 |
175 virtual void OnConnectJobComplete(int result, ConnectJob* job) override; | 173 void OnConnectJobComplete(int result, ConnectJob* job) override; |
176 | 174 |
177 private: | 175 private: |
178 WebSocketTransportClientSocketPool* owner_; | 176 WebSocketTransportClientSocketPool* owner_; |
179 | 177 |
180 DISALLOW_COPY_AND_ASSIGN(ConnectJobDelegate); | 178 DISALLOW_COPY_AND_ASSIGN(ConnectJobDelegate); |
181 }; | 179 }; |
182 | 180 |
183 // Store the arguments from a call to RequestSocket() that has stalled so we | 181 // Store the arguments from a call to RequestSocket() that has stalled so we |
184 // can replay it when there are available socket slots. | 182 // can replay it when there are available socket slots. |
185 struct StalledRequest { | 183 struct StalledRequest { |
(...skipping 53 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
239 bool flushing_; | 237 bool flushing_; |
240 | 238 |
241 base::WeakPtrFactory<WebSocketTransportClientSocketPool> weak_factory_; | 239 base::WeakPtrFactory<WebSocketTransportClientSocketPool> weak_factory_; |
242 | 240 |
243 DISALLOW_COPY_AND_ASSIGN(WebSocketTransportClientSocketPool); | 241 DISALLOW_COPY_AND_ASSIGN(WebSocketTransportClientSocketPool); |
244 }; | 242 }; |
245 | 243 |
246 } // namespace net | 244 } // namespace net |
247 | 245 |
248 #endif // NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ | 246 #endif // NET_SOCKET_WEBSOCKET_TRANSPORT_CLIENT_SOCKET_POOL_H_ |
OLD | NEW |