1 // Copyright 2012 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/socket/client_socket_pool_manager_impl.h"
6
7 #include <algorithm>
8 #include <utility>
9
10 #include "base/check_op.h"
11 #include "base/values.h"
12 #include "net/base/proxy_server.h"
13 #include "net/base/proxy_string_util.h"
14 #include "net/http/http_network_session.h"
15 #include "net/socket/socks_connect_job.h"
16 #include "net/socket/ssl_connect_job.h"
17 #include "net/socket/transport_client_socket_pool.h"
18 #include "net/socket/transport_connect_job.h"
19 #include "net/socket/websocket_transport_client_socket_pool.h"
20
21 namespace net {
22
23 class SocketPerformanceWatcherFactory;
24
ClientSocketPoolManagerImpl(const CommonConnectJobParams & common_connect_job_params,const CommonConnectJobParams & websocket_common_connect_job_params,HttpNetworkSession::SocketPoolType pool_type,bool cleanup_on_ip_address_change)25 ClientSocketPoolManagerImpl::ClientSocketPoolManagerImpl(
26 const CommonConnectJobParams& common_connect_job_params,
27 const CommonConnectJobParams& websocket_common_connect_job_params,
28 HttpNetworkSession::SocketPoolType pool_type,
29 bool cleanup_on_ip_address_change)
30 : common_connect_job_params_(common_connect_job_params),
31 websocket_common_connect_job_params_(websocket_common_connect_job_params),
32 pool_type_(pool_type),
33 cleanup_on_ip_address_change_(cleanup_on_ip_address_change) {
34 // |websocket_endpoint_lock_manager| must only be set for websocket
35 // connections.
36 DCHECK(!common_connect_job_params_.websocket_endpoint_lock_manager);
37 DCHECK(websocket_common_connect_job_params.websocket_endpoint_lock_manager);
38 }
39
~ClientSocketPoolManagerImpl()40 ClientSocketPoolManagerImpl::~ClientSocketPoolManagerImpl() {
41 DCHECK_CALLED_ON_VALID_THREAD(thread_checker_);
42 }
43
FlushSocketPoolsWithError(int net_error,const char * net_log_reason_utf8)44 void ClientSocketPoolManagerImpl::FlushSocketPoolsWithError(
45 int net_error,
46 const char* net_log_reason_utf8) {
47 for (const auto& it : socket_pools_) {
48 it.second->FlushWithError(net_error, net_log_reason_utf8);
49 }
50 }
51
CloseIdleSockets(const char * net_log_reason_utf8)52 void ClientSocketPoolManagerImpl::CloseIdleSockets(
53 const char* net_log_reason_utf8) {
54 for (const auto& it : socket_pools_) {
55 it.second->CloseIdleSockets(net_log_reason_utf8);
56 }
57 }
58
GetSocketPool(const ProxyServer & proxy_server)59 ClientSocketPool* ClientSocketPoolManagerImpl::GetSocketPool(
60 const ProxyServer& proxy_server) {
61 SocketPoolMap::const_iterator it = socket_pools_.find(proxy_server);
62 if (it != socket_pools_.end())
63 return it->second.get();
64
65 int sockets_per_proxy_server;
66 int sockets_per_group;
67 if (proxy_server.is_direct()) {
68 sockets_per_proxy_server = max_sockets_per_pool(pool_type_);
69 sockets_per_group = max_sockets_per_group(pool_type_);
70 } else {
71 sockets_per_proxy_server = max_sockets_per_proxy_server(pool_type_);
72 sockets_per_group =
73 std::min(sockets_per_proxy_server, max_sockets_per_group(pool_type_));
74 }
75
76 std::unique_ptr<ClientSocketPool> new_pool;
77
78 // Use specialized WebSockets pool for WebSockets when no proxy is in use.
79 if (pool_type_ == HttpNetworkSession::WEBSOCKET_SOCKET_POOL &&
80 proxy_server.is_direct()) {
81 new_pool = std::make_unique<WebSocketTransportClientSocketPool>(
82 sockets_per_proxy_server, sockets_per_group, proxy_server,
83 &websocket_common_connect_job_params_);
84 } else {
85 new_pool = std::make_unique<TransportClientSocketPool>(
86 sockets_per_proxy_server, sockets_per_group,
87 unused_idle_socket_timeout(pool_type_), proxy_server,
88 pool_type_ == HttpNetworkSession::WEBSOCKET_SOCKET_POOL,
89 &common_connect_job_params_, cleanup_on_ip_address_change_);
90 }
91
92 std::pair<SocketPoolMap::iterator, bool> ret =
93 socket_pools_.insert(std::make_pair(proxy_server, std::move(new_pool)));
94 return ret.first->second.get();
95 }
96
SocketPoolInfoToValue() const97 base::Value ClientSocketPoolManagerImpl::SocketPoolInfoToValue() const {
98 base::Value::List list;
99 for (const auto& socket_pool : socket_pools_) {
100 // TODO(menke): Is this really needed?
101 const char* type;
102 if (socket_pool.first.is_direct()) {
103 type = "transport_socket_pool";
104 } else if (socket_pool.first.is_socks()) {
105 type = "socks_socket_pool";
106 } else {
107 type = "http_proxy_socket_pool";
108 }
109 list.Append(socket_pool.second->GetInfoAsValue(
110 ProxyServerToProxyUri(socket_pool.first), type));
111 }
112
113 return base::Value(std::move(list));
114 }
115
116 } // namespace net
117