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