Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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_SSL_CLIENT_SOCKET_POOL_H_ | 5 #ifndef NET_SOCKET_SSL_CLIENT_SOCKET_POOL_H_ |
| 6 #define NET_SOCKET_SSL_CLIENT_SOCKET_POOL_H_ | 6 #define NET_SOCKET_SSL_CLIENT_SOCKET_POOL_H_ |
| 7 | 7 |
| 8 #include <map> | |
| 8 #include <string> | 9 #include <string> |
| 10 #include <vector> | |
| 9 | 11 |
| 10 #include "base/memory/ref_counted.h" | 12 #include "base/memory/ref_counted.h" |
| 11 #include "base/memory/scoped_ptr.h" | 13 #include "base/memory/scoped_ptr.h" |
| 12 #include "base/time/time.h" | 14 #include "base/time/time.h" |
| 13 #include "net/base/privacy_mode.h" | 15 #include "net/base/privacy_mode.h" |
| 14 #include "net/dns/host_resolver.h" | 16 #include "net/dns/host_resolver.h" |
| 15 #include "net/http/http_response_info.h" | 17 #include "net/http/http_response_info.h" |
| 16 #include "net/socket/client_socket_pool.h" | 18 #include "net/socket/client_socket_pool.h" |
| 17 #include "net/socket/client_socket_pool_base.h" | 19 #include "net/socket/client_socket_pool_base.h" |
| 18 #include "net/socket/client_socket_pool_histograms.h" | 20 #include "net/socket/client_socket_pool_histograms.h" |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 91 const bool want_spdy_over_npn_; | 93 const bool want_spdy_over_npn_; |
| 92 bool ignore_limits_; | 94 bool ignore_limits_; |
| 93 | 95 |
| 94 DISALLOW_COPY_AND_ASSIGN(SSLSocketParams); | 96 DISALLOW_COPY_AND_ASSIGN(SSLSocketParams); |
| 95 }; | 97 }; |
| 96 | 98 |
| 97 // SSLConnectJob handles the SSL handshake after setting up the underlying | 99 // SSLConnectJob handles the SSL handshake after setting up the underlying |
| 98 // connection as specified in the params. | 100 // connection as specified in the params. |
| 99 class SSLConnectJob : public ConnectJob { | 101 class SSLConnectJob : public ConnectJob { |
| 100 public: | 102 public: |
| 101 SSLConnectJob( | 103 // List of SSLConnectJobs that are either in the process of connecting |
| 102 const std::string& group_name, | 104 // to or waiting to connect to the same server. |
| 103 RequestPriority priority, | 105 typedef std::vector<SSLConnectJob*> PendingJobList; |
| 104 const scoped_refptr<SSLSocketParams>& params, | 106 |
| 105 const base::TimeDelta& timeout_duration, | 107 SSLConnectJob(const std::string& group_name, |
|
wtc
2014/06/26 17:48:17
Please move the comment here:
// Note: the SSLC
| |
| 106 TransportClientSocketPool* transport_pool, | 108 RequestPriority priority, |
| 107 SOCKSClientSocketPool* socks_pool, | 109 const scoped_refptr<SSLSocketParams>& params, |
| 108 HttpProxyClientSocketPool* http_proxy_pool, | 110 const base::TimeDelta& timeout_duration, |
| 109 ClientSocketFactory* client_socket_factory, | 111 TransportClientSocketPool* transport_pool, |
| 110 HostResolver* host_resolver, | 112 SOCKSClientSocketPool* socks_pool, |
| 111 const SSLClientSocketContext& context, | 113 HttpProxyClientSocketPool* http_proxy_pool, |
| 112 Delegate* delegate, | 114 ClientSocketFactory* client_socket_factory, |
| 113 NetLog* net_log); | 115 HostResolver* host_resolver, |
| 116 const SSLClientSocketContext& context, | |
| 117 PendingJobList* pending_jobs_list, | |
| 118 Delegate* delegate, | |
| 119 NetLog* net_log); | |
| 114 virtual ~SSLConnectJob(); | 120 virtual ~SSLConnectJob(); |
| 115 | 121 |
| 116 // ConnectJob methods. | 122 // ConnectJob methods. |
| 117 virtual LoadState GetLoadState() const OVERRIDE; | 123 virtual LoadState GetLoadState() const OVERRIDE; |
| 118 | 124 |
| 119 virtual void GetAdditionalErrorState(ClientSocketHandle * handle) OVERRIDE; | 125 virtual void GetAdditionalErrorState(ClientSocketHandle * handle) OVERRIDE; |
| 120 | 126 |
| 127 // Enables or disables SSLConnectJob waiting. | |
| 128 // If enabled, simultaneous attempts to connect to the same server | |
| 129 // will be serialized until an SSL session is cached, allowing those | |
| 130 // handshakes to perform a resumption handshake. | |
| 131 static NET_EXPORT void EnableJobWaiting(bool enable); | |
| 132 | |
| 133 static bool GetEnableJobWaiting(); | |
|
wtc
2014/06/26 17:48:17
Nit: this function can be named "JobWaitingEnabled
| |
| 134 | |
| 121 private: | 135 private: |
| 122 enum State { | 136 enum State { |
| 123 STATE_TRANSPORT_CONNECT, | 137 STATE_TRANSPORT_CONNECT, |
| 124 STATE_TRANSPORT_CONNECT_COMPLETE, | 138 STATE_TRANSPORT_CONNECT_COMPLETE, |
| 125 STATE_SOCKS_CONNECT, | 139 STATE_SOCKS_CONNECT, |
| 126 STATE_SOCKS_CONNECT_COMPLETE, | 140 STATE_SOCKS_CONNECT_COMPLETE, |
| 127 STATE_TUNNEL_CONNECT, | 141 STATE_TUNNEL_CONNECT, |
| 128 STATE_TUNNEL_CONNECT_COMPLETE, | 142 STATE_TUNNEL_CONNECT_COMPLETE, |
| 143 STATE_CREATE_SSL_SOCKET, | |
| 144 STATE_CHECK_FOR_RESUME, | |
| 129 STATE_SSL_CONNECT, | 145 STATE_SSL_CONNECT, |
| 130 STATE_SSL_CONNECT_COMPLETE, | 146 STATE_SSL_CONNECT_COMPLETE, |
| 147 STATE_PROCESS_PENDING_JOBS, | |
| 131 STATE_NONE, | 148 STATE_NONE, |
| 132 }; | 149 }; |
| 133 | 150 |
| 134 void OnIOComplete(int result); | 151 void OnIOComplete(int result); |
| 135 | 152 |
| 136 // Runs the state transition loop. | 153 // Runs the state transition loop. |
| 137 int DoLoop(int result); | 154 int DoLoop(int result); |
| 138 | 155 |
| 139 int DoTransportConnect(); | 156 int DoTransportConnect(); |
| 140 int DoTransportConnectComplete(int result); | 157 int DoTransportConnectComplete(int result); |
| 141 int DoSOCKSConnect(); | 158 int DoSOCKSConnect(); |
| 142 int DoSOCKSConnectComplete(int result); | 159 int DoSOCKSConnectComplete(int result); |
| 143 int DoTunnelConnect(); | 160 int DoTunnelConnect(); |
| 144 int DoTunnelConnectComplete(int result); | 161 int DoTunnelConnectComplete(int result); |
| 162 int DoCreateSSLSocket(); | |
| 163 int DoCheckForResume(); | |
| 145 int DoSSLConnect(); | 164 int DoSSLConnect(); |
| 146 int DoSSLConnectComplete(int result); | 165 int DoSSLConnectComplete(int result); |
| 166 int DoProcessPendingJobs(int result); | |
| 167 | |
| 168 // Decides how pending jobs should continue their connection | |
| 169 // based upon the success or failure of the leading job. | |
| 170 void ProcessPendingJobs(int result); | |
| 171 | |
| 172 // Tells a waiting SSLConnectJob to resume its SSL connection. | |
| 173 void ResumeSSLConnection(); | |
| 147 | 174 |
| 148 // Returns the initial state for the state machine based on the | 175 // Returns the initial state for the state machine based on the |
| 149 // |connection_type|. | 176 // |connection_type|. |
| 150 static State GetInitialState(SSLSocketParams::ConnectionType connection_type); | 177 static State GetInitialState(SSLSocketParams::ConnectionType connection_type); |
| 151 | 178 |
| 152 // Starts the SSL connection process. Returns OK on success and | 179 // Starts the SSL connection process. Returns OK on success and |
| 153 // ERR_IO_PENDING if it cannot immediately service the request. | 180 // ERR_IO_PENDING if it cannot immediately service the request. |
| 154 // Otherwise, it returns a net error code. | 181 // Otherwise, it returns a net error code. |
| 155 virtual int ConnectInternal() OVERRIDE; | 182 virtual int ConnectInternal() OVERRIDE; |
| 156 | 183 |
| 184 static bool enable_job_waiting_; | |
| 157 scoped_refptr<SSLSocketParams> params_; | 185 scoped_refptr<SSLSocketParams> params_; |
| 158 TransportClientSocketPool* const transport_pool_; | 186 TransportClientSocketPool* const transport_pool_; |
| 159 SOCKSClientSocketPool* const socks_pool_; | 187 SOCKSClientSocketPool* const socks_pool_; |
| 160 HttpProxyClientSocketPool* const http_proxy_pool_; | 188 HttpProxyClientSocketPool* const http_proxy_pool_; |
| 161 ClientSocketFactory* const client_socket_factory_; | 189 ClientSocketFactory* const client_socket_factory_; |
| 162 HostResolver* const host_resolver_; | 190 HostResolver* const host_resolver_; |
| 163 | 191 |
| 164 const SSLClientSocketContext context_; | 192 const SSLClientSocketContext context_; |
| 165 | 193 |
| 166 State next_state_; | 194 State next_state_; |
| 167 CompletionCallback callback_; | 195 CompletionCallback io_callback_; |
| 168 scoped_ptr<ClientSocketHandle> transport_socket_handle_; | 196 scoped_ptr<ClientSocketHandle> transport_socket_handle_; |
| 169 scoped_ptr<SSLClientSocket> ssl_socket_; | 197 scoped_ptr<SSLClientSocket> ssl_socket_; |
| 170 | 198 |
| 199 // TODO(mshelley) ensure that these pointers are deleted. | |
| 200 PendingJobList* pending_jobs_; | |
| 171 HttpResponseInfo error_response_info_; | 201 HttpResponseInfo error_response_info_; |
| 172 | 202 |
| 173 DISALLOW_COPY_AND_ASSIGN(SSLConnectJob); | 203 DISALLOW_COPY_AND_ASSIGN(SSLConnectJob); |
| 174 }; | 204 }; |
| 175 | 205 |
| 176 class NET_EXPORT_PRIVATE SSLClientSocketPool | 206 class NET_EXPORT_PRIVATE SSLClientSocketPool |
| 177 : public ClientSocketPool, | 207 : public ClientSocketPool, |
| 178 public HigherLayeredPool, | 208 public HigherLayeredPool, |
| 179 public SSLConfigService::Observer { | 209 public SSLConfigService::Observer { |
| 180 public: | 210 public: |
| (...skipping 96 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 277 | 307 |
| 278 // ClientSocketPoolBase::ConnectJobFactory methods. | 308 // ClientSocketPoolBase::ConnectJobFactory methods. |
| 279 virtual scoped_ptr<ConnectJob> NewConnectJob( | 309 virtual scoped_ptr<ConnectJob> NewConnectJob( |
| 280 const std::string& group_name, | 310 const std::string& group_name, |
| 281 const PoolBase::Request& request, | 311 const PoolBase::Request& request, |
| 282 ConnectJob::Delegate* delegate) const OVERRIDE; | 312 ConnectJob::Delegate* delegate) const OVERRIDE; |
| 283 | 313 |
| 284 virtual base::TimeDelta ConnectionTimeout() const OVERRIDE; | 314 virtual base::TimeDelta ConnectionTimeout() const OVERRIDE; |
| 285 | 315 |
| 286 private: | 316 private: |
| 317 // Maps SSLConnectJob group names to lists of pending connect jobs to | |
| 318 // that group name. | |
| 319 typedef std::map<std::string, SSLConnectJob::PendingJobList*> PendingJobMap; | |
| 320 | |
| 287 TransportClientSocketPool* const transport_pool_; | 321 TransportClientSocketPool* const transport_pool_; |
| 288 SOCKSClientSocketPool* const socks_pool_; | 322 SOCKSClientSocketPool* const socks_pool_; |
| 289 HttpProxyClientSocketPool* const http_proxy_pool_; | 323 HttpProxyClientSocketPool* const http_proxy_pool_; |
| 290 ClientSocketFactory* const client_socket_factory_; | 324 ClientSocketFactory* const client_socket_factory_; |
| 291 HostResolver* const host_resolver_; | 325 HostResolver* const host_resolver_; |
| 292 const SSLClientSocketContext context_; | 326 const SSLClientSocketContext context_; |
| 293 base::TimeDelta timeout_; | 327 base::TimeDelta timeout_; |
| 294 NetLog* net_log_; | 328 NetLog* net_log_; |
| 329 scoped_ptr<PendingJobMap> pending_jobs_map_; | |
|
wtc
2014/06/26 17:48:17
Please declare this member as a non-pointer, if th
| |
| 295 | 330 |
| 296 DISALLOW_COPY_AND_ASSIGN(SSLConnectJobFactory); | 331 DISALLOW_COPY_AND_ASSIGN(SSLConnectJobFactory); |
| 297 }; | 332 }; |
| 298 | 333 |
| 299 TransportClientSocketPool* const transport_pool_; | 334 TransportClientSocketPool* const transport_pool_; |
| 300 SOCKSClientSocketPool* const socks_pool_; | 335 SOCKSClientSocketPool* const socks_pool_; |
| 301 HttpProxyClientSocketPool* const http_proxy_pool_; | 336 HttpProxyClientSocketPool* const http_proxy_pool_; |
| 302 PoolBase base_; | 337 PoolBase base_; |
| 303 const scoped_refptr<SSLConfigService> ssl_config_service_; | 338 const scoped_refptr<SSLConfigService> ssl_config_service_; |
| 304 | 339 |
| 305 DISALLOW_COPY_AND_ASSIGN(SSLClientSocketPool); | 340 DISALLOW_COPY_AND_ASSIGN(SSLClientSocketPool); |
| 306 }; | 341 }; |
| 307 | 342 |
| 308 } // namespace net | 343 } // namespace net |
| 309 | 344 |
| 310 #endif // NET_SOCKET_SSL_CLIENT_SOCKET_POOL_H_ | 345 #endif // NET_SOCKET_SSL_CLIENT_SOCKET_POOL_H_ |
| OLD | NEW |