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