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 3 // found in the LICENSE file. 4 5 #ifndef NET_SOCKET_CLIENT_SOCKET_POOL_MANAGER_IMPL_H_ 6 #define NET_SOCKET_CLIENT_SOCKET_POOL_MANAGER_IMPL_H_ 7 8 #include <map> 9 #include "base/basictypes.h" 10 #include "base/compiler_specific.h" 11 #include "base/memory/ref_counted.h" 12 #include "base/memory/scoped_ptr.h" 13 #include "base/stl_util.h" 14 #include "base/template_util.h" 15 #include "base/threading/non_thread_safe.h" 16 #include "net/cert/cert_database.h" 17 #include "net/http/http_network_session.h" 18 #include "net/socket/client_socket_pool_histograms.h" 19 #include "net/socket/client_socket_pool_manager.h" 20 21 namespace net { 22 23 class CertVerifier; 24 class ClientSocketFactory; 25 class ClientSocketPoolHistograms; 26 class CTVerifier; 27 class HttpProxyClientSocketPool; 28 class HostResolver; 29 class NetLog; 30 class ServerBoundCertService; 31 class ProxyService; 32 class SOCKSClientSocketPool; 33 class SSLClientSocketPool; 34 class SSLConfigService; 35 class TransportClientSocketPool; 36 class TransportSecurityState; 37 38 namespace internal { 39 40 // A helper class for auto-deleting Values in the destructor. 41 template <typename Key, typename Value> 42 class OwnedPoolMap : public std::map<Key, Value> { 43 public: OwnedPoolMap()44 OwnedPoolMap() { 45 COMPILE_ASSERT(base::is_pointer<Value>::value, 46 value_must_be_a_pointer); 47 } 48 ~OwnedPoolMap()49 ~OwnedPoolMap() { 50 STLDeleteValues(this); 51 } 52 }; 53 54 } // namespace internal 55 56 class ClientSocketPoolManagerImpl : public base::NonThreadSafe, 57 public ClientSocketPoolManager, 58 public CertDatabase::Observer { 59 public: 60 ClientSocketPoolManagerImpl(NetLog* net_log, 61 ClientSocketFactory* socket_factory, 62 HostResolver* host_resolver, 63 CertVerifier* cert_verifier, 64 ServerBoundCertService* server_bound_cert_service, 65 TransportSecurityState* transport_security_state, 66 CTVerifier* cert_transparency_verifier, 67 const std::string& ssl_session_cache_shard, 68 ProxyService* proxy_service, 69 SSLConfigService* ssl_config_service, 70 HttpNetworkSession::SocketPoolType pool_type); 71 virtual ~ClientSocketPoolManagerImpl(); 72 73 virtual void FlushSocketPoolsWithError(int error) OVERRIDE; 74 virtual void CloseIdleSockets() OVERRIDE; 75 76 virtual TransportClientSocketPool* GetTransportSocketPool() OVERRIDE; 77 78 virtual SSLClientSocketPool* GetSSLSocketPool() OVERRIDE; 79 80 virtual SOCKSClientSocketPool* GetSocketPoolForSOCKSProxy( 81 const HostPortPair& socks_proxy) OVERRIDE; 82 83 virtual HttpProxyClientSocketPool* GetSocketPoolForHTTPProxy( 84 const HostPortPair& http_proxy) OVERRIDE; 85 86 virtual SSLClientSocketPool* GetSocketPoolForSSLWithProxy( 87 const HostPortPair& proxy_server) OVERRIDE; 88 89 // Creates a Value summary of the state of the socket pools. The caller is 90 // responsible for deleting the returned value. 91 virtual base::Value* SocketPoolInfoToValue() const OVERRIDE; 92 93 // CertDatabase::Observer methods: 94 virtual void OnCertAdded(const X509Certificate* cert) OVERRIDE; 95 virtual void OnCACertChanged(const X509Certificate* cert) OVERRIDE; 96 97 private: 98 typedef internal::OwnedPoolMap<HostPortPair, TransportClientSocketPool*> 99 TransportSocketPoolMap; 100 typedef internal::OwnedPoolMap<HostPortPair, SOCKSClientSocketPool*> 101 SOCKSSocketPoolMap; 102 typedef internal::OwnedPoolMap<HostPortPair, HttpProxyClientSocketPool*> 103 HTTPProxySocketPoolMap; 104 typedef internal::OwnedPoolMap<HostPortPair, SSLClientSocketPool*> 105 SSLSocketPoolMap; 106 107 NetLog* const net_log_; 108 ClientSocketFactory* const socket_factory_; 109 HostResolver* const host_resolver_; 110 CertVerifier* const cert_verifier_; 111 ServerBoundCertService* const server_bound_cert_service_; 112 TransportSecurityState* const transport_security_state_; 113 CTVerifier* const cert_transparency_verifier_; 114 const std::string ssl_session_cache_shard_; 115 ProxyService* const proxy_service_; 116 const scoped_refptr<SSLConfigService> ssl_config_service_; 117 const HttpNetworkSession::SocketPoolType pool_type_; 118 119 // Note: this ordering is important. 120 121 ClientSocketPoolHistograms transport_pool_histograms_; 122 scoped_ptr<TransportClientSocketPool> transport_socket_pool_; 123 124 ClientSocketPoolHistograms ssl_pool_histograms_; 125 scoped_ptr<SSLClientSocketPool> ssl_socket_pool_; 126 127 ClientSocketPoolHistograms transport_for_socks_pool_histograms_; 128 TransportSocketPoolMap transport_socket_pools_for_socks_proxies_; 129 130 ClientSocketPoolHistograms socks_pool_histograms_; 131 SOCKSSocketPoolMap socks_socket_pools_; 132 133 ClientSocketPoolHistograms transport_for_http_proxy_pool_histograms_; 134 TransportSocketPoolMap transport_socket_pools_for_http_proxies_; 135 136 ClientSocketPoolHistograms transport_for_https_proxy_pool_histograms_; 137 TransportSocketPoolMap transport_socket_pools_for_https_proxies_; 138 139 ClientSocketPoolHistograms ssl_for_https_proxy_pool_histograms_; 140 SSLSocketPoolMap ssl_socket_pools_for_https_proxies_; 141 142 ClientSocketPoolHistograms http_proxy_pool_histograms_; 143 HTTPProxySocketPoolMap http_proxy_socket_pools_; 144 145 ClientSocketPoolHistograms ssl_socket_pool_for_proxies_histograms_; 146 SSLSocketPoolMap ssl_socket_pools_for_proxies_; 147 148 DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolManagerImpl); 149 }; 150 151 } // namespace net 152 153 #endif // NET_SOCKET_CLIENT_SOCKET_POOL_MANAGER_IMPL_H_ 154