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 #include "net/dns/dns_socket_pool.h" | 5 #include "net/dns/dns_socket_pool.h" |
6 | 6 |
7 #include "base/logging.h" | 7 #include "base/logging.h" |
8 #include "base/macros.h" | 8 #include "base/macros.h" |
| 9 #include "base/memory/ptr_util.h" |
9 #include "base/rand_util.h" | 10 #include "base/rand_util.h" |
10 #include "base/stl_util.h" | |
11 #include "net/base/address_list.h" | 11 #include "net/base/address_list.h" |
12 #include "net/base/ip_endpoint.h" | 12 #include "net/base/ip_endpoint.h" |
13 #include "net/base/net_errors.h" | 13 #include "net/base/net_errors.h" |
14 #include "net/log/net_log_source.h" | 14 #include "net/log/net_log_source.h" |
15 #include "net/socket/client_socket_factory.h" | 15 #include "net/socket/client_socket_factory.h" |
16 #include "net/socket/stream_socket.h" | 16 #include "net/socket/stream_socket.h" |
17 #include "net/udp/datagram_client_socket.h" | 17 #include "net/udp/datagram_client_socket.h" |
18 | 18 |
19 namespace net { | 19 namespace net { |
20 | 20 |
(...skipping 118 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
139 | 139 |
140 std::unique_ptr<DatagramClientSocket> AllocateSocket( | 140 std::unique_ptr<DatagramClientSocket> AllocateSocket( |
141 unsigned server_index) override; | 141 unsigned server_index) override; |
142 | 142 |
143 void FreeSocket(unsigned server_index, | 143 void FreeSocket(unsigned server_index, |
144 std::unique_ptr<DatagramClientSocket> socket) override; | 144 std::unique_ptr<DatagramClientSocket> socket) override; |
145 | 145 |
146 private: | 146 private: |
147 void FillPool(unsigned server_index, unsigned size); | 147 void FillPool(unsigned server_index, unsigned size); |
148 | 148 |
149 typedef std::vector<DatagramClientSocket*> SocketVector; | 149 typedef std::vector<std::unique_ptr<DatagramClientSocket>> SocketVector; |
150 | 150 |
151 std::vector<SocketVector> pools_; | 151 std::vector<SocketVector> pools_; |
152 | 152 |
153 DISALLOW_COPY_AND_ASSIGN(DefaultDnsSocketPool); | 153 DISALLOW_COPY_AND_ASSIGN(DefaultDnsSocketPool); |
154 }; | 154 }; |
155 | 155 |
156 // static | 156 // static |
157 std::unique_ptr<DnsSocketPool> DnsSocketPool::CreateDefault( | 157 std::unique_ptr<DnsSocketPool> DnsSocketPool::CreateDefault( |
158 ClientSocketFactory* factory, | 158 ClientSocketFactory* factory, |
159 const RandIntCallback& rand_int_callback) { | 159 const RandIntCallback& rand_int_callback) { |
160 return std::unique_ptr<DnsSocketPool>( | 160 return std::unique_ptr<DnsSocketPool>( |
161 new DefaultDnsSocketPool(factory, rand_int_callback)); | 161 new DefaultDnsSocketPool(factory, rand_int_callback)); |
162 } | 162 } |
163 | 163 |
164 void DefaultDnsSocketPool::Initialize( | 164 void DefaultDnsSocketPool::Initialize( |
165 const std::vector<IPEndPoint>* nameservers, | 165 const std::vector<IPEndPoint>* nameservers, |
166 NetLog* net_log) { | 166 NetLog* net_log) { |
167 InitializeInternal(nameservers, net_log); | 167 InitializeInternal(nameservers, net_log); |
168 | 168 |
169 DCHECK(pools_.empty()); | 169 DCHECK(pools_.empty()); |
170 const unsigned num_servers = nameservers->size(); | 170 const unsigned num_servers = nameservers->size(); |
171 pools_.resize(num_servers); | 171 pools_.resize(num_servers); |
172 for (unsigned server_index = 0; server_index < num_servers; ++server_index) | 172 for (unsigned server_index = 0; server_index < num_servers; ++server_index) |
173 FillPool(server_index, kInitialPoolSize); | 173 FillPool(server_index, kInitialPoolSize); |
174 } | 174 } |
175 | 175 |
176 DefaultDnsSocketPool::~DefaultDnsSocketPool() { | 176 DefaultDnsSocketPool::~DefaultDnsSocketPool() { |
177 unsigned num_servers = pools_.size(); | |
178 for (unsigned server_index = 0; server_index < num_servers; ++server_index) { | |
179 SocketVector& pool = pools_[server_index]; | |
180 base::STLDeleteElements(&pool); | |
181 } | |
182 } | 177 } |
183 | 178 |
184 std::unique_ptr<DatagramClientSocket> DefaultDnsSocketPool::AllocateSocket( | 179 std::unique_ptr<DatagramClientSocket> DefaultDnsSocketPool::AllocateSocket( |
185 unsigned server_index) { | 180 unsigned server_index) { |
186 DCHECK_LT(server_index, pools_.size()); | 181 DCHECK_LT(server_index, pools_.size()); |
187 SocketVector& pool = pools_[server_index]; | 182 SocketVector& pool = pools_[server_index]; |
188 | 183 |
189 FillPool(server_index, kAllocateMinSize); | 184 FillPool(server_index, kAllocateMinSize); |
190 if (pool.size() == 0) { | 185 if (pool.size() == 0) { |
191 DVLOG(1) << "No DNS sockets available in pool " << server_index << "!"; | 186 DVLOG(1) << "No DNS sockets available in pool " << server_index << "!"; |
192 return std::unique_ptr<DatagramClientSocket>(); | 187 return std::unique_ptr<DatagramClientSocket>(); |
193 } | 188 } |
194 | 189 |
195 if (pool.size() < kAllocateMinSize) { | 190 if (pool.size() < kAllocateMinSize) { |
196 DVLOG(1) << "Low DNS port entropy: wanted " << kAllocateMinSize | 191 DVLOG(1) << "Low DNS port entropy: wanted " << kAllocateMinSize |
197 << " sockets to choose from, but only have " << pool.size() | 192 << " sockets to choose from, but only have " << pool.size() |
198 << " in pool " << server_index << "."; | 193 << " in pool " << server_index << "."; |
199 } | 194 } |
200 | 195 |
201 unsigned socket_index = GetRandomInt(0, pool.size() - 1); | 196 unsigned socket_index = GetRandomInt(0, pool.size() - 1); |
202 DatagramClientSocket* socket = pool[socket_index]; | 197 std::unique_ptr<DatagramClientSocket> socket = std::move(pool[socket_index]); |
203 pool[socket_index] = pool.back(); | 198 pool[socket_index] = std::move(pool.back()); |
204 pool.pop_back(); | 199 pool.pop_back(); |
205 | 200 |
206 return std::unique_ptr<DatagramClientSocket>(socket); | 201 return socket; |
207 } | 202 } |
208 | 203 |
209 void DefaultDnsSocketPool::FreeSocket( | 204 void DefaultDnsSocketPool::FreeSocket( |
210 unsigned server_index, | 205 unsigned server_index, |
211 std::unique_ptr<DatagramClientSocket> socket) { | 206 std::unique_ptr<DatagramClientSocket> socket) { |
212 DCHECK_LT(server_index, pools_.size()); | 207 DCHECK_LT(server_index, pools_.size()); |
213 } | 208 } |
214 | 209 |
215 void DefaultDnsSocketPool::FillPool(unsigned server_index, unsigned size) { | 210 void DefaultDnsSocketPool::FillPool(unsigned server_index, unsigned size) { |
216 SocketVector& pool = pools_[server_index]; | 211 SocketVector& pool = pools_[server_index]; |
217 | 212 |
218 for (unsigned pool_index = pool.size(); pool_index < size; ++pool_index) { | 213 for (unsigned pool_index = pool.size(); pool_index < size; ++pool_index) { |
219 DatagramClientSocket* socket = | 214 std::unique_ptr<DatagramClientSocket> socket = |
220 CreateConnectedSocket(server_index).release(); | 215 CreateConnectedSocket(server_index); |
221 if (!socket) | 216 if (!socket) |
222 break; | 217 break; |
223 pool.push_back(socket); | 218 pool.push_back(std::move(socket)); |
224 } | 219 } |
225 } | 220 } |
226 | 221 |
227 } // namespace net | 222 } // namespace net |
OLD | NEW |