blob: 6c626f7c3e81ddd90b1184fce5634e64f4709bb5 [file] [log] [blame]
[email protected]a796bcec2010-03-22 17:17:261// Copyright (c) 2010 The Chromium Authors. All rights reserved.
[email protected]ff579d42009-06-24 15:47:022// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
[email protected]d80a4322009-08-14 07:07:494//
5// A ClientSocketPoolBase is used to restrict the number of sockets open at
6// a time. It also maintains a list of idle persistent sockets for reuse.
7// Subclasses of ClientSocketPool should compose ClientSocketPoolBase to handle
8// the core logic of (1) restricting the number of active (connected or
9// connecting) sockets per "group" (generally speaking, the hostname), (2)
10// maintaining a per-group list of idle, persistent sockets for reuse, and (3)
11// limiting the total number of active sockets in the system.
12//
13// ClientSocketPoolBase abstracts socket connection details behind ConnectJob,
14// ConnectJobFactory, and SocketParams. When a socket "slot" becomes available,
15// the ClientSocketPoolBase will ask the ConnectJobFactory to create a
16// ConnectJob with a SocketParams. Subclasses of ClientSocketPool should
17// implement their socket specific connection by subclassing ConnectJob and
18// implementing ConnectJob::ConnectInternal(). They can control the parameters
19// passed to each new ConnectJob instance via their ConnectJobFactory subclass
20// and templated SocketParams parameter.
21//
[email protected]ff579d42009-06-24 15:47:0222#ifndef NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_
23#define NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_
[email protected]32b76ef2010-07-26 23:08:2424#pragma once
[email protected]ff579d42009-06-24 15:47:0225
26#include <deque>
[email protected]e1b54dc2010-10-06 21:27:2227#include <list>
[email protected]ff579d42009-06-24 15:47:0228#include <map>
[email protected]5fc08e32009-07-15 17:09:5729#include <set>
[email protected]ff579d42009-06-24 15:47:0230#include <string>
31
32#include "base/basictypes.h"
[email protected]100d5fb92009-12-21 21:08:3533#include "base/ref_counted.h"
[email protected]ff579d42009-06-24 15:47:0234#include "base/scoped_ptr.h"
[email protected]6b624c62010-03-14 08:37:3235#include "base/task.h"
[email protected]ff579d42009-06-24 15:47:0236#include "base/time.h"
37#include "base/timer.h"
38#include "net/base/address_list.h"
39#include "net/base/completion_callback.h"
[email protected]ff579d42009-06-24 15:47:0240#include "net/base/load_states.h"
[email protected]d80a4322009-08-14 07:07:4941#include "net/base/net_errors.h"
[email protected]9e743cd2010-03-16 07:03:5342#include "net/base/net_log.h"
[email protected]a554a8262010-05-20 00:13:5243#include "net/base/network_change_notifier.h"
[email protected]ac790b42009-12-02 04:31:3144#include "net/base/request_priority.h"
[email protected]2ab05b52009-07-01 23:57:5845#include "net/socket/client_socket.h"
[email protected]ff579d42009-06-24 15:47:0246#include "net/socket/client_socket_pool.h"
47
48namespace net {
49
[email protected]ff579d42009-06-24 15:47:0250class ClientSocketHandle;
[email protected]ff579d42009-06-24 15:47:0251
52// ConnectJob provides an abstract interface for "connecting" a socket.
53// The connection may involve host resolution, tcp connection, ssl connection,
54// etc.
55class ConnectJob {
56 public:
[email protected]ab838892009-06-30 18:49:0557 class Delegate {
58 public:
59 Delegate() {}
60 virtual ~Delegate() {}
61
[email protected]2ab05b52009-07-01 23:57:5862 // Alerts the delegate that the connection completed.
63 virtual void OnConnectJobComplete(int result, ConnectJob* job) = 0;
[email protected]ab838892009-06-30 18:49:0564
65 private:
66 DISALLOW_COPY_AND_ASSIGN(Delegate);
67 };
68
[email protected]974ebd62009-08-03 23:14:3469 // A |timeout_duration| of 0 corresponds to no timeout.
[email protected]2ab05b52009-07-01 23:57:5870 ConnectJob(const std::string& group_name,
[email protected]974ebd62009-08-03 23:14:3471 base::TimeDelta timeout_duration,
[email protected]fd7b7c92009-08-20 19:38:3072 Delegate* delegate,
[email protected]9e743cd2010-03-16 07:03:5373 const BoundNetLog& net_log);
[email protected]2ab05b52009-07-01 23:57:5874 virtual ~ConnectJob();
[email protected]ff579d42009-06-24 15:47:0275
[email protected]2ab05b52009-07-01 23:57:5876 // Accessors
77 const std::string& group_name() const { return group_name_; }
[email protected]9e743cd2010-03-16 07:03:5378 const BoundNetLog& net_log() { return net_log_; }
[email protected]2c2bef152010-10-13 00:55:0379 bool is_preconnect() const { return preconnect_state_ != NOT_PRECONNECT; }
80 bool is_unused_preconnect() const {
81 return preconnect_state_ == UNUSED_PRECONNECT;
82 }
83
84 // Initialized by the ClientSocketPoolBaseHelper.
85 // TODO(willchan): Move most of the constructor arguments over here. We
86 // shouldn't give the ConnectJobFactory (subclasses) the ability to screw up
87 // the initialization.
88 void Initialize(bool is_preconnect);
[email protected]2ab05b52009-07-01 23:57:5889
[email protected]8e12ae02009-07-02 16:15:0490 // Releases |socket_| to the client. On connection error, this should return
91 // NULL.
[email protected]2ab05b52009-07-01 23:57:5892 ClientSocket* ReleaseSocket() { return socket_.release(); }
[email protected]ab838892009-06-30 18:49:0593
[email protected]ff579d42009-06-24 15:47:0294 // Begins connecting the socket. Returns OK on success, ERR_IO_PENDING if it
95 // cannot complete synchronously without blocking, or another net error code
[email protected]2ab05b52009-07-01 23:57:5896 // on error. In asynchronous completion, the ConnectJob will notify
97 // |delegate_| via OnConnectJobComplete. In both asynchronous and synchronous
98 // completion, ReleaseSocket() can be called to acquire the connected socket
99 // if it succeeded.
[email protected]974ebd62009-08-03 23:14:34100 int Connect();
[email protected]ff579d42009-06-24 15:47:02101
[email protected]2c2bef152010-10-13 00:55:03102 // Precondition: is_unused_preconnect() must be true. Marks the job as a
103 // used preconnect job.
104 void UseForNormalRequest();
105
[email protected]46451352009-09-01 14:54:21106 virtual LoadState GetLoadState() const = 0;
[email protected]fd7b7c92009-08-20 19:38:30107
[email protected]e60e47a2010-07-14 03:37:18108 // If Connect returns an error (or OnConnectJobComplete reports an error
109 // result) this method will be called, allowing the pool to add
110 // additional error state to the ClientSocketHandle (post late-binding).
111 virtual void GetAdditionalErrorState(ClientSocketHandle* handle) {}
112
[email protected]0496f9a32010-09-30 16:08:07113 const BoundNetLog& net_log() const { return net_log_; }
114
[email protected]ab838892009-06-30 18:49:05115 protected:
[email protected]06650c52010-06-03 00:49:17116 void set_socket(ClientSocket* socket);
[email protected]2ab05b52009-07-01 23:57:58117 ClientSocket* socket() { return socket_.get(); }
[email protected]fd7b7c92009-08-20 19:38:30118 void NotifyDelegateOfCompletion(int rv);
[email protected]a796bcec2010-03-22 17:17:26119 void ResetTimer(base::TimeDelta remainingTime);
[email protected]ab838892009-06-30 18:49:05120
[email protected]ff579d42009-06-24 15:47:02121 private:
[email protected]2c2bef152010-10-13 00:55:03122 enum PreconnectState {
123 NOT_PRECONNECT,
124 UNUSED_PRECONNECT,
125 USED_PRECONNECT,
126 };
127
[email protected]974ebd62009-08-03 23:14:34128 virtual int ConnectInternal() = 0;
129
[email protected]06650c52010-06-03 00:49:17130 void LogConnectStart();
131 void LogConnectCompletion(int net_error);
132
[email protected]974ebd62009-08-03 23:14:34133 // Alerts the delegate that the ConnectJob has timed out.
134 void OnTimeout();
135
[email protected]2ab05b52009-07-01 23:57:58136 const std::string group_name_;
[email protected]974ebd62009-08-03 23:14:34137 const base::TimeDelta timeout_duration_;
138 // Timer to abort jobs that take too long.
139 base::OneShotTimer<ConnectJob> timer_;
140 Delegate* delegate_;
[email protected]2ab05b52009-07-01 23:57:58141 scoped_ptr<ClientSocket> socket_;
[email protected]9e743cd2010-03-16 07:03:53142 BoundNetLog net_log_;
[email protected]a2006ece2010-04-23 16:44:02143 // A ConnectJob is idle until Connect() has been called.
144 bool idle_;
[email protected]2c2bef152010-10-13 00:55:03145 PreconnectState preconnect_state_;
[email protected]ab838892009-06-30 18:49:05146
[email protected]ff579d42009-06-24 15:47:02147 DISALLOW_COPY_AND_ASSIGN(ConnectJob);
148};
149
[email protected]d80a4322009-08-14 07:07:49150namespace internal {
151
152// ClientSocketPoolBaseHelper is an internal class that implements almost all
153// the functionality from ClientSocketPoolBase without using templates.
154// ClientSocketPoolBase adds templated definitions built on top of
155// ClientSocketPoolBaseHelper. This class is not for external use, please use
156// ClientSocketPoolBase instead.
157class ClientSocketPoolBaseHelper
[email protected]2431756e2010-09-29 20:26:13158 : public ConnectJob::Delegate,
[email protected]a554a8262010-05-20 00:13:52159 public NetworkChangeNotifier::Observer {
[email protected]ff579d42009-06-24 15:47:02160 public:
[email protected]2c2bef152010-10-13 00:55:03161 // Used to specify specific behavior for the ClientSocketPool.
162 enum Flag {
163 NORMAL = 0, // Normal behavior.
164 NO_IDLE_SOCKETS = 0x1, // Do not return an idle socket. Create a new one.
165 };
166
167 typedef uint32 Flags;
168
[email protected]d80a4322009-08-14 07:07:49169 class Request {
170 public:
[email protected]684970b2009-08-14 04:54:46171 Request(ClientSocketHandle* handle,
[email protected]ff579d42009-06-24 15:47:02172 CompletionCallback* callback,
[email protected]ac790b42009-12-02 04:31:31173 RequestPriority priority,
[email protected]2c2bef152010-10-13 00:55:03174 Flags flags,
[email protected]9e743cd2010-03-16 07:03:53175 const BoundNetLog& net_log);
[email protected]ff579d42009-06-24 15:47:02176
[email protected]fd4fe0b2010-02-08 23:02:15177 virtual ~Request();
[email protected]d80a4322009-08-14 07:07:49178
179 ClientSocketHandle* handle() const { return handle_; }
180 CompletionCallback* callback() const { return callback_; }
[email protected]ac790b42009-12-02 04:31:31181 RequestPriority priority() const { return priority_; }
[email protected]2c2bef152010-10-13 00:55:03182 const Flags flags() const { return flags_; }
[email protected]9e743cd2010-03-16 07:03:53183 const BoundNetLog& net_log() const { return net_log_; }
[email protected]d80a4322009-08-14 07:07:49184
185 private:
186 ClientSocketHandle* const handle_;
187 CompletionCallback* const callback_;
[email protected]ac790b42009-12-02 04:31:31188 const RequestPriority priority_;
[email protected]2c2bef152010-10-13 00:55:03189 const Flags flags_;
[email protected]9e743cd2010-03-16 07:03:53190 BoundNetLog net_log_;
[email protected]d80a4322009-08-14 07:07:49191
192 DISALLOW_COPY_AND_ASSIGN(Request);
[email protected]ff579d42009-06-24 15:47:02193 };
194
195 class ConnectJobFactory {
196 public:
197 ConnectJobFactory() {}
198 virtual ~ConnectJobFactory() {}
199
200 virtual ConnectJob* NewConnectJob(
201 const std::string& group_name,
202 const Request& request,
[email protected]06650c52010-06-03 00:49:17203 ConnectJob::Delegate* delegate) const = 0;
[email protected]ff579d42009-06-24 15:47:02204
[email protected]a796bcec2010-03-22 17:17:26205 virtual base::TimeDelta ConnectionTimeout() const = 0;
206
[email protected]ff579d42009-06-24 15:47:02207 private:
208 DISALLOW_COPY_AND_ASSIGN(ConnectJobFactory);
209 };
210
[email protected]100d5fb92009-12-21 21:08:35211 ClientSocketPoolBaseHelper(
212 int max_sockets,
213 int max_sockets_per_group,
214 base::TimeDelta unused_idle_socket_timeout,
215 base::TimeDelta used_idle_socket_timeout,
[email protected]66761b952010-06-25 21:30:38216 ConnectJobFactory* connect_job_factory);
[email protected]ff579d42009-06-24 15:47:02217
[email protected]2431756e2010-09-29 20:26:13218 ~ClientSocketPoolBaseHelper();
219
[email protected]d80a4322009-08-14 07:07:49220 // See ClientSocketPool::RequestSocket for documentation on this function.
[email protected]e7e99322010-05-04 23:30:17221 // ClientSocketPoolBaseHelper takes ownership of |request|, which must be
222 // heap allocated.
[email protected]d80a4322009-08-14 07:07:49223 int RequestSocket(const std::string& group_name, const Request* request);
[email protected]ff579d42009-06-24 15:47:02224
[email protected]2c2bef152010-10-13 00:55:03225 // See ClientSocketPool::RequestSocket for documentation on this function.
226 void RequestSockets(const std::string& group_name,
227 const Request& request,
228 int num_sockets);
229
[email protected]d80a4322009-08-14 07:07:49230 // See ClientSocketPool::CancelRequest for documentation on this function.
[email protected]ff579d42009-06-24 15:47:02231 void CancelRequest(const std::string& group_name,
[email protected]05ea9ff2010-07-15 19:08:21232 ClientSocketHandle* handle);
[email protected]ff579d42009-06-24 15:47:02233
[email protected]d80a4322009-08-14 07:07:49234 // See ClientSocketPool::ReleaseSocket for documentation on this function.
[email protected]ff579d42009-06-24 15:47:02235 void ReleaseSocket(const std::string& group_name,
[email protected]a7e38572010-06-07 18:22:24236 ClientSocket* socket,
237 int id);
[email protected]ff579d42009-06-24 15:47:02238
[email protected]a7e38572010-06-07 18:22:24239 // See ClientSocketPool::Flush for documentation on this function.
240 void Flush();
[email protected]241c5c2c2010-06-21 18:46:00241
[email protected]d80a4322009-08-14 07:07:49242 // See ClientSocketPool::CloseIdleSockets for documentation on this function.
[email protected]ff579d42009-06-24 15:47:02243 void CloseIdleSockets();
244
[email protected]d80a4322009-08-14 07:07:49245 // See ClientSocketPool::IdleSocketCount() for documentation on this function.
[email protected]ff579d42009-06-24 15:47:02246 int idle_socket_count() const {
247 return idle_socket_count_;
248 }
249
[email protected]d80a4322009-08-14 07:07:49250 // See ClientSocketPool::IdleSocketCountInGroup() for documentation on this
251 // function.
[email protected]ff579d42009-06-24 15:47:02252 int IdleSocketCountInGroup(const std::string& group_name) const;
253
[email protected]d80a4322009-08-14 07:07:49254 // See ClientSocketPool::GetLoadState() for documentation on this function.
[email protected]ff579d42009-06-24 15:47:02255 LoadState GetLoadState(const std::string& group_name,
256 const ClientSocketHandle* handle) const;
257
[email protected]6b624c62010-03-14 08:37:32258 int ConnectRetryIntervalMs() const {
259 // TODO(mbelshe): Make this tuned dynamically based on measured RTT.
260 // For now, just use the max retry interval.
261 return ClientSocketPool::kMaxConnectRetryIntervalMs;
262 }
263
[email protected]d80a4322009-08-14 07:07:49264 // ConnectJob::Delegate methods:
[email protected]2ab05b52009-07-01 23:57:58265 virtual void OnConnectJobComplete(int result, ConnectJob* job);
[email protected]ff579d42009-06-24 15:47:02266
[email protected]a554a8262010-05-20 00:13:52267 // NetworkChangeNotifier::Observer methods:
[email protected]66761b952010-06-25 21:30:38268 virtual void OnIPAddressChanged();
[email protected]a554a8262010-05-20 00:13:52269
[email protected]974ebd62009-08-03 23:14:34270 int NumConnectJobsInGroup(const std::string& group_name) const {
[email protected]aed99ef02010-08-26 14:04:32271 return group_map_.find(group_name)->second->jobs().size();
[email protected]974ebd62009-08-03 23:14:34272 }
[email protected]211d2172009-07-22 15:48:53273
[email protected]2c2bef152010-10-13 00:55:03274 int NumActiveSocketsInGroup(const std::string& group_name) const {
275 return group_map_.find(group_name)->second->active_socket_count();
276 }
277
[email protected]2abfe90a2010-08-25 17:49:51278 bool HasGroup(const std::string& group_name) const;
279
[email protected]9bf28db2009-08-29 01:35:16280 // Closes all idle sockets if |force| is true. Else, only closes idle
281 // sockets that timed out or can't be reused. Made public for testing.
282 void CleanupIdleSockets(bool force);
283
[email protected]59d7a5a2010-08-30 16:44:27284 // See ClientSocketPool::GetInfoAsValue for documentation on this function.
[email protected]ba00b492010-09-08 14:53:38285 DictionaryValue* GetInfoAsValue(const std::string& name,
286 const std::string& type) const;
[email protected]59d7a5a2010-08-30 16:44:27287
[email protected]a796bcec2010-03-22 17:17:26288 base::TimeDelta ConnectionTimeout() const {
289 return connect_job_factory_->ConnectionTimeout();
290 }
291
[email protected]06d94042010-08-25 01:45:22292 static void set_connect_backup_jobs_enabled(bool enabled);
293 void EnableConnectBackupJobs();
[email protected]7c28e9a2010-03-20 01:16:13294
[email protected]ff579d42009-06-24 15:47:02295 private:
[email protected]5389bc72009-11-05 23:34:24296 friend class base::RefCounted<ClientSocketPoolBaseHelper>;
297
[email protected]ff579d42009-06-24 15:47:02298 // Entry for a persistent socket which became idle at time |start_time|.
299 struct IdleSocket {
[email protected]0f873e82010-09-02 16:09:01300 IdleSocket() : socket(NULL) {}
[email protected]ff579d42009-06-24 15:47:02301 ClientSocket* socket;
302 base::TimeTicks start_time;
303
304 // An idle socket should be removed if it can't be reused, or has been idle
305 // for too long. |now| is the current time value (TimeTicks::Now()).
[email protected]9bf28db2009-08-29 01:35:16306 // |timeout| is the length of time to wait before timing out an idle socket.
[email protected]ff579d42009-06-24 15:47:02307 //
308 // An idle socket can't be reused if it is disconnected or has received
309 // data unexpectedly (hence no longer idle). The unread data would be
310 // mistaken for the beginning of the next response if we were to reuse the
311 // socket for a new request.
[email protected]9bf28db2009-08-29 01:35:16312 bool ShouldCleanup(base::TimeTicks now, base::TimeDelta timeout) const;
[email protected]ff579d42009-06-24 15:47:02313 };
314
[email protected]2c2bef152010-10-13 00:55:03315 typedef std::deque<const Request* > RequestQueue;
[email protected]d80a4322009-08-14 07:07:49316 typedef std::map<const ClientSocketHandle*, const Request*> RequestMap;
[email protected]ff579d42009-06-24 15:47:02317
318 // A Group is allocated per group_name when there are idle sockets or pending
319 // requests. Otherwise, the Group object is removed from the map.
[email protected]eb5a99382010-07-11 03:18:26320 // |active_socket_count| tracks the number of sockets held by clients.
[email protected]aed99ef02010-08-26 14:04:32321 class Group {
322 public:
323 Group();
324 ~Group();
[email protected]2ab05b52009-07-01 23:57:58325
326 bool IsEmpty() const {
[email protected]aed99ef02010-08-26 14:04:32327 return active_socket_count_ == 0 && idle_sockets_.empty() &&
328 jobs_.empty() && pending_requests_.empty();
[email protected]2ab05b52009-07-01 23:57:58329 }
330
331 bool HasAvailableSocketSlot(int max_sockets_per_group) const {
[email protected]2c2bef152010-10-13 00:55:03332 return NumActiveSocketSlots() < max_sockets_per_group;
333 }
334
335 int NumActiveSocketSlots() const {
336 return active_socket_count_ + static_cast<int>(jobs_.size()) +
337 static_cast<int>(idle_sockets_.size());
[email protected]2ab05b52009-07-01 23:57:58338 }
339
[email protected]05ea9ff2010-07-15 19:08:21340 bool IsStalled(int max_sockets_per_group) const {
341 return HasAvailableSocketSlot(max_sockets_per_group) &&
[email protected]aed99ef02010-08-26 14:04:32342 pending_requests_.size() > jobs_.size();
[email protected]05ea9ff2010-07-15 19:08:21343 }
344
[email protected]ac790b42009-12-02 04:31:31345 RequestPriority TopPendingPriority() const {
[email protected]aed99ef02010-08-26 14:04:32346 return pending_requests_.front()->priority();
[email protected]211d2172009-07-22 15:48:53347 }
348
[email protected]aed99ef02010-08-26 14:04:32349 bool HasBackupJob() const { return !method_factory_.empty(); }
350
[email protected]6b624c62010-03-14 08:37:32351 void CleanupBackupJob() {
[email protected]aed99ef02010-08-26 14:04:32352 method_factory_.RevokeAll();
[email protected]6b624c62010-03-14 08:37:32353 }
354
[email protected]aed99ef02010-08-26 14:04:32355 // Set a timer to create a backup socket if it takes too long to create one.
356 void StartBackupSocketTimer(const std::string& group_name,
357 ClientSocketPoolBaseHelper* pool);
358
[email protected]2c2bef152010-10-13 00:55:03359 // Searches |jobs_| to see if there's a preconnect ConnectJob, and if so,
360 // uses it. Returns true on success. Otherwise, returns false.
361 bool TryToUsePreconnectConnectJob();
[email protected]aed99ef02010-08-26 14:04:32362
[email protected]2c2bef152010-10-13 00:55:03363 void AddJob(ConnectJob* job) { jobs_.insert(job); }
364 void RemoveJob(ConnectJob* job) { jobs_.erase(job); }
[email protected]aed99ef02010-08-26 14:04:32365 void RemoveAllJobs();
366
367 void IncrementActiveSocketCount() { active_socket_count_++; }
368 void DecrementActiveSocketCount() { active_socket_count_--; }
369
[email protected]2c2bef152010-10-13 00:55:03370 const std::set<ConnectJob*>& jobs() const { return jobs_; }
[email protected]e1b54dc2010-10-06 21:27:22371 const std::list<IdleSocket>& idle_sockets() const { return idle_sockets_; }
[email protected]aed99ef02010-08-26 14:04:32372 const RequestQueue& pending_requests() const { return pending_requests_; }
373 int active_socket_count() const { return active_socket_count_; }
374 RequestQueue* mutable_pending_requests() { return &pending_requests_; }
[email protected]e1b54dc2010-10-06 21:27:22375 std::list<IdleSocket>* mutable_idle_sockets() { return &idle_sockets_; }
[email protected]aed99ef02010-08-26 14:04:32376
377 private:
[email protected]2c2bef152010-10-13 00:55:03378 // Called when the backup socket timer fires.
379 void OnBackupSocketTimerFired(
380 std::string group_name,
381 ClientSocketPoolBaseHelper* pool);
382
[email protected]e1b54dc2010-10-06 21:27:22383 std::list<IdleSocket> idle_sockets_;
[email protected]2c2bef152010-10-13 00:55:03384 std::set<ConnectJob*> jobs_;
[email protected]aed99ef02010-08-26 14:04:32385 RequestQueue pending_requests_;
386 int active_socket_count_; // number of active sockets used by clients
387 // A factory to pin the backup_job tasks.
388 ScopedRunnableMethodFactory<Group> method_factory_;
[email protected]ff579d42009-06-24 15:47:02389 };
390
[email protected]aed99ef02010-08-26 14:04:32391 typedef std::map<std::string, Group*> GroupMap;
[email protected]ff579d42009-06-24 15:47:02392
[email protected]2c2bef152010-10-13 00:55:03393 typedef std::set<ConnectJob*> ConnectJobSet;
[email protected]ff579d42009-06-24 15:47:02394
[email protected]05ea9ff2010-07-15 19:08:21395 struct CallbackResultPair {
396 CallbackResultPair() : callback(NULL), result(OK) {}
397 CallbackResultPair(CompletionCallback* callback_in, int result_in)
398 : callback(callback_in), result(result_in) {}
399
400 CompletionCallback* callback;
401 int result;
402 };
403
404 typedef std::map<const ClientSocketHandle*, CallbackResultPair>
405 PendingCallbackMap;
406
[email protected]d80a4322009-08-14 07:07:49407 static void InsertRequestIntoQueue(const Request* r,
[email protected]ff579d42009-06-24 15:47:02408 RequestQueue* pending_requests);
[email protected]fd7b7c92009-08-20 19:38:30409 static const Request* RemoveRequestFromQueue(RequestQueue::iterator it,
[email protected]3f00be82010-09-27 19:50:02410 Group* group);
[email protected]ff579d42009-06-24 15:47:02411
[email protected]aed99ef02010-08-26 14:04:32412 Group* GetOrCreateGroup(const std::string& group_name);
413 void RemoveGroup(const std::string& group_name);
414 void RemoveGroup(GroupMap::iterator it);
415
[email protected]ff579d42009-06-24 15:47:02416 // Called when the number of idle sockets changes.
417 void IncrementIdleCount();
418 void DecrementIdleCount();
419
[email protected]211d2172009-07-22 15:48:53420 // Scans the group map for groups which have an available socket slot and
[email protected]05ea9ff2010-07-15 19:08:21421 // at least one pending request. Returns true if any groups are stalled, and
422 // if so, fills |group| and |group_name| with data of the stalled group
[email protected]211d2172009-07-22 15:48:53423 // having highest priority.
[email protected]05ea9ff2010-07-15 19:08:21424 bool FindTopStalledGroup(Group** group, std::string* group_name);
[email protected]211d2172009-07-22 15:48:53425
[email protected]ff579d42009-06-24 15:47:02426 // Called when timer_ fires. This method scans the idle sockets removing
427 // sockets that timed out or can't be reused.
428 void OnCleanupTimerFired() {
429 CleanupIdleSockets(false);
430 }
431
[email protected]4d3b05d2010-01-27 21:27:29432 // Removes |job| from |connect_job_set_|. Also updates |group| if non-NULL.
[email protected]2c2bef152010-10-13 00:55:03433 void RemoveConnectJob(ConnectJob* job, Group* group);
[email protected]ff579d42009-06-24 15:47:02434
[email protected]05ea9ff2010-07-15 19:08:21435 // Tries to see if we can handle any more requests for |group|.
436 void OnAvailableSocketSlot(const std::string& group_name, Group* group);
[email protected]8ae03f42010-07-07 19:08:10437
[email protected]eb5a99382010-07-11 03:18:26438 // Process a pending socket request for a group.
[email protected]05ea9ff2010-07-15 19:08:21439 void ProcessPendingRequest(const std::string& group_name, Group* group);
[email protected]ff579d42009-06-24 15:47:02440
[email protected]2ab05b52009-07-01 23:57:58441 // Assigns |socket| to |handle| and updates |group|'s counters appropriately.
442 void HandOutSocket(ClientSocket* socket,
443 bool reused,
444 ClientSocketHandle* handle,
[email protected]f9d285c2009-08-17 19:54:29445 base::TimeDelta time_idle,
[email protected]fd4fe0b2010-02-08 23:02:15446 Group* group,
[email protected]9e743cd2010-03-16 07:03:53447 const BoundNetLog& net_log);
[email protected]2ab05b52009-07-01 23:57:58448
[email protected]0f873e82010-09-02 16:09:01449 // Adds |socket| to the list of idle sockets for |group|.
450 void AddIdleSocket(ClientSocket* socket, Group* group);
[email protected]5fc08e32009-07-15 17:09:57451
[email protected]06f92462010-08-31 19:24:14452 // Iterates through |group_map_|, canceling all ConnectJobs and deleting
453 // groups if they are no longer needed.
[email protected]5fc08e32009-07-15 17:09:57454 void CancelAllConnectJobs();
455
[email protected]06f92462010-08-31 19:24:14456 // Iterates through |group_map_|, posting ERR_ABORTED callbacks for all
457 // requests, and then deleting groups if they are no longer needed.
458 void AbortAllRequests();
459
[email protected]211d2172009-07-22 15:48:53460 // Returns true if we can't create any more sockets due to the total limit.
[email protected]211d2172009-07-22 15:48:53461 bool ReachedMaxSocketsLimit() const;
462
[email protected]fd4fe0b2010-02-08 23:02:15463 // This is the internal implementation of RequestSocket(). It differs in that
[email protected]9e743cd2010-03-16 07:03:53464 // it does not handle logging into NetLog of the queueing status of
[email protected]fd4fe0b2010-02-08 23:02:15465 // |request|.
466 int RequestSocketInternal(const std::string& group_name,
467 const Request* request);
468
[email protected]eb5a99382010-07-11 03:18:26469 // Assigns an idle socket for the group to the request.
470 // Returns |true| if an idle socket is available, false otherwise.
[email protected]aed99ef02010-08-26 14:04:32471 bool AssignIdleSocketToGroup(const Request* request, Group* group);
[email protected]eb5a99382010-07-11 03:18:26472
[email protected]06650c52010-06-03 00:49:17473 static void LogBoundConnectJobToRequest(
474 const NetLog::Source& connect_job_source, const Request* request);
475
[email protected]43a21b82010-06-10 21:30:54476 // Closes one idle socket. Picks the first one encountered.
477 // TODO(willchan): Consider a better algorithm for doing this. Perhaps we
478 // should keep an ordered list of idle sockets, and close them in order.
479 // Requires maintaining more state. It's not clear if it's worth it since
480 // I'm not sure if we hit this situation often.
481 void CloseOneIdleSocket();
482
[email protected]eb5a99382010-07-11 03:18:26483 // Checks if there are stalled socket groups that should be notified
484 // for possible wakeup.
485 void CheckForStalledSocketGroups();
486
[email protected]05ea9ff2010-07-15 19:08:21487 // Posts a task to call InvokeUserCallback() on the next iteration through the
488 // current message loop. Inserts |callback| into |pending_callback_map_|,
489 // keyed by |handle|.
490 void InvokeUserCallbackLater(
491 ClientSocketHandle* handle, CompletionCallback* callback, int rv);
492
493 // Invokes the user callback for |handle|. By the time this task has run,
494 // it's possible that the request has been cancelled, so |handle| may not
495 // exist in |pending_callback_map_|. We look up the callback and result code
496 // in |pending_callback_map_|.
497 void InvokeUserCallback(ClientSocketHandle* handle);
[email protected]eb5a99382010-07-11 03:18:26498
[email protected]ff579d42009-06-24 15:47:02499 GroupMap group_map_;
500
[email protected]05ea9ff2010-07-15 19:08:21501 // Map of the ClientSocketHandles for which we have a pending Task to invoke a
502 // callback. This is necessary since, before we invoke said callback, it's
503 // possible that the request is cancelled.
504 PendingCallbackMap pending_callback_map_;
505
[email protected]ff579d42009-06-24 15:47:02506 // Timer used to periodically prune idle sockets that timed out or can't be
507 // reused.
[email protected]d80a4322009-08-14 07:07:49508 base::RepeatingTimer<ClientSocketPoolBaseHelper> timer_;
[email protected]ff579d42009-06-24 15:47:02509
510 // The total number of idle sockets in the system.
511 int idle_socket_count_;
512
[email protected]211d2172009-07-22 15:48:53513 // Number of connecting sockets across all groups.
514 int connecting_socket_count_;
515
516 // Number of connected sockets we handed out across all groups.
517 int handed_out_socket_count_;
518
519 // The maximum total number of sockets. See ReachedMaxSocketsLimit.
520 const int max_sockets_;
521
[email protected]ff579d42009-06-24 15:47:02522 // The maximum number of sockets kept per group.
523 const int max_sockets_per_group_;
524
[email protected]9bf28db2009-08-29 01:35:16525 // The time to wait until closing idle sockets.
526 const base::TimeDelta unused_idle_socket_timeout_;
527 const base::TimeDelta used_idle_socket_timeout_;
528
[email protected]ab838892009-06-30 18:49:05529 const scoped_ptr<ConnectJobFactory> connect_job_factory_;
[email protected]ff579d42009-06-24 15:47:02530
[email protected]7c28e9a2010-03-20 01:16:13531 // TODO(vandebo) Remove when backup jobs move to TCPClientSocketPool
[email protected]06d94042010-08-25 01:45:22532 bool connect_backup_jobs_enabled_;
[email protected]7c28e9a2010-03-20 01:16:13533
[email protected]a7e38572010-06-07 18:22:24534 // A unique id for the pool. It gets incremented every time we Flush() the
535 // pool. This is so that when sockets get released back to the pool, we can
536 // make sure that they are discarded rather than reused.
537 int pool_generation_number_;
[email protected]09d6ecb02010-07-22 20:10:45538
[email protected]2431756e2010-09-29 20:26:13539 ScopedRunnableMethodFactory<ClientSocketPoolBaseHelper> method_factory_;
540
[email protected]3598c6022010-09-17 23:13:09541 DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolBaseHelper);
[email protected]d80a4322009-08-14 07:07:49542};
543
544} // namespace internal
545
[email protected]9bf28db2009-08-29 01:35:16546// The maximum duration, in seconds, to keep used idle persistent sockets alive.
547static const int kUsedIdleSocketTimeout = 300; // 5 minutes
548
[email protected]d80a4322009-08-14 07:07:49549template <typename SocketParams>
550class ClientSocketPoolBase {
551 public:
552 class Request : public internal::ClientSocketPoolBaseHelper::Request {
553 public:
554 Request(ClientSocketHandle* handle,
555 CompletionCallback* callback,
[email protected]ac790b42009-12-02 04:31:31556 RequestPriority priority,
[email protected]2c2bef152010-10-13 00:55:03557 internal::ClientSocketPoolBaseHelper::Flags flags,
[email protected]df4b4ef2010-07-12 18:25:21558 const scoped_refptr<SocketParams>& params,
[email protected]9e743cd2010-03-16 07:03:53559 const BoundNetLog& net_log)
[email protected]d80a4322009-08-14 07:07:49560 : internal::ClientSocketPoolBaseHelper::Request(
[email protected]2c2bef152010-10-13 00:55:03561 handle, callback, priority, flags, net_log),
[email protected]d80a4322009-08-14 07:07:49562 params_(params) {}
563
[email protected]df4b4ef2010-07-12 18:25:21564 const scoped_refptr<SocketParams>& params() const { return params_; }
[email protected]d80a4322009-08-14 07:07:49565
566 private:
[email protected]2c2bef152010-10-13 00:55:03567 const scoped_refptr<SocketParams> params_;
[email protected]d80a4322009-08-14 07:07:49568 };
569
570 class ConnectJobFactory {
571 public:
572 ConnectJobFactory() {}
573 virtual ~ConnectJobFactory() {}
574
575 virtual ConnectJob* NewConnectJob(
576 const std::string& group_name,
577 const Request& request,
[email protected]06650c52010-06-03 00:49:17578 ConnectJob::Delegate* delegate) const = 0;
[email protected]d80a4322009-08-14 07:07:49579
[email protected]a796bcec2010-03-22 17:17:26580 virtual base::TimeDelta ConnectionTimeout() const = 0;
581
[email protected]d80a4322009-08-14 07:07:49582 private:
583 DISALLOW_COPY_AND_ASSIGN(ConnectJobFactory);
584 };
585
[email protected]9bf28db2009-08-29 01:35:16586 // |max_sockets| is the maximum number of sockets to be maintained by this
587 // ClientSocketPool. |max_sockets_per_group| specifies the maximum number of
588 // sockets a "group" can have. |unused_idle_socket_timeout| specifies how
589 // long to leave an unused idle socket open before closing it.
590 // |used_idle_socket_timeout| specifies how long to leave a previously used
591 // idle socket open before closing it.
[email protected]100d5fb92009-12-21 21:08:35592 ClientSocketPoolBase(
593 int max_sockets,
594 int max_sockets_per_group,
[email protected]2431756e2010-09-29 20:26:13595 ClientSocketPoolHistograms* histograms,
[email protected]100d5fb92009-12-21 21:08:35596 base::TimeDelta unused_idle_socket_timeout,
597 base::TimeDelta used_idle_socket_timeout,
[email protected]66761b952010-06-25 21:30:38598 ConnectJobFactory* connect_job_factory)
[email protected]b89f7e42010-05-20 20:37:00599 : histograms_(histograms),
[email protected]2431756e2010-09-29 20:26:13600 helper_(max_sockets, max_sockets_per_group,
601 unused_idle_socket_timeout, used_idle_socket_timeout,
602 new ConnectJobFactoryAdaptor(connect_job_factory)) {}
[email protected]d80a4322009-08-14 07:07:49603
[email protected]20cb5f482009-12-16 01:01:25604 virtual ~ClientSocketPoolBase() {}
[email protected]d80a4322009-08-14 07:07:49605
606 // These member functions simply forward to ClientSocketPoolBaseHelper.
607
608 // RequestSocket bundles up the parameters into a Request and then forwards to
[email protected]2c2bef152010-10-13 00:55:03609 // ClientSocketPoolBaseHelper::RequestSocket().
[email protected]d80a4322009-08-14 07:07:49610 int RequestSocket(const std::string& group_name,
[email protected]df4b4ef2010-07-12 18:25:21611 const scoped_refptr<SocketParams>& params,
[email protected]ac790b42009-12-02 04:31:31612 RequestPriority priority,
[email protected]d80a4322009-08-14 07:07:49613 ClientSocketHandle* handle,
614 CompletionCallback* callback,
[email protected]9e743cd2010-03-16 07:03:53615 const BoundNetLog& net_log) {
[email protected]2c2bef152010-10-13 00:55:03616 Request* request =
617 new Request(handle, callback, priority,
618 internal::ClientSocketPoolBaseHelper::NORMAL,
619 params, net_log);
[email protected]2431756e2010-09-29 20:26:13620 return helper_.RequestSocket(group_name, request);
[email protected]d80a4322009-08-14 07:07:49621 }
622
[email protected]2c2bef152010-10-13 00:55:03623 // RequestSockets bundles up the parameters into a Request and then forwards
624 // to ClientSocketPoolBaseHelper::RequestSockets(). Note that it assigns the
625 // priority to LOWEST and specifies the NO_IDLE_SOCKETS flag.
626 void RequestSockets(const std::string& group_name,
627 const scoped_refptr<SocketParams>& params,
628 int num_sockets,
629 const BoundNetLog& net_log) {
630 const Request request(NULL /* no handle */,
631 NULL /* no callback */,
632 LOWEST,
633 internal::ClientSocketPoolBaseHelper::NO_IDLE_SOCKETS,
634 params,
635 net_log);
636 helper_.RequestSockets(group_name, request, num_sockets);
637 }
638
[email protected]d80a4322009-08-14 07:07:49639 void CancelRequest(const std::string& group_name,
[email protected]05ea9ff2010-07-15 19:08:21640 ClientSocketHandle* handle) {
[email protected]2431756e2010-09-29 20:26:13641 return helper_.CancelRequest(group_name, handle);
[email protected]d80a4322009-08-14 07:07:49642 }
643
[email protected]241c5c2c2010-06-21 18:46:00644 void ReleaseSocket(const std::string& group_name, ClientSocket* socket,
645 int id) {
[email protected]2431756e2010-09-29 20:26:13646 return helper_.ReleaseSocket(group_name, socket, id);
[email protected]d80a4322009-08-14 07:07:49647 }
648
[email protected]2431756e2010-09-29 20:26:13649 void CloseIdleSockets() { return helper_.CloseIdleSockets(); }
[email protected]d80a4322009-08-14 07:07:49650
[email protected]2431756e2010-09-29 20:26:13651 int idle_socket_count() const { return helper_.idle_socket_count(); }
[email protected]d80a4322009-08-14 07:07:49652
653 int IdleSocketCountInGroup(const std::string& group_name) const {
[email protected]2431756e2010-09-29 20:26:13654 return helper_.IdleSocketCountInGroup(group_name);
[email protected]d80a4322009-08-14 07:07:49655 }
656
657 LoadState GetLoadState(const std::string& group_name,
658 const ClientSocketHandle* handle) const {
[email protected]2431756e2010-09-29 20:26:13659 return helper_.GetLoadState(group_name, handle);
[email protected]d80a4322009-08-14 07:07:49660 }
661
662 virtual void OnConnectJobComplete(int result, ConnectJob* job) {
[email protected]2431756e2010-09-29 20:26:13663 return helper_.OnConnectJobComplete(result, job);
[email protected]d80a4322009-08-14 07:07:49664 }
665
[email protected]d80a4322009-08-14 07:07:49666 int NumConnectJobsInGroup(const std::string& group_name) const {
[email protected]2431756e2010-09-29 20:26:13667 return helper_.NumConnectJobsInGroup(group_name);
[email protected]d80a4322009-08-14 07:07:49668 }
669
[email protected]2c2bef152010-10-13 00:55:03670 int NumActiveSocketsInGroup(const std::string& group_name) const {
671 return helper_.NumActiveSocketsInGroup(group_name);
672 }
673
[email protected]2abfe90a2010-08-25 17:49:51674 bool HasGroup(const std::string& group_name) const {
[email protected]2431756e2010-09-29 20:26:13675 return helper_.HasGroup(group_name);
[email protected]2abfe90a2010-08-25 17:49:51676 }
677
[email protected]9bf28db2009-08-29 01:35:16678 void CleanupIdleSockets(bool force) {
[email protected]2431756e2010-09-29 20:26:13679 return helper_.CleanupIdleSockets(force);
[email protected]9bf28db2009-08-29 01:35:16680 }
681
[email protected]ba00b492010-09-08 14:53:38682 DictionaryValue* GetInfoAsValue(const std::string& name,
683 const std::string& type) const {
[email protected]2431756e2010-09-29 20:26:13684 return helper_.GetInfoAsValue(name, type);
[email protected]59d7a5a2010-08-30 16:44:27685 }
686
[email protected]a796bcec2010-03-22 17:17:26687 base::TimeDelta ConnectionTimeout() const {
[email protected]2431756e2010-09-29 20:26:13688 return helper_.ConnectionTimeout();
[email protected]a796bcec2010-03-22 17:17:26689 }
690
[email protected]2431756e2010-09-29 20:26:13691 ClientSocketPoolHistograms* histograms() const {
[email protected]b89f7e42010-05-20 20:37:00692 return histograms_;
693 }
[email protected]a796bcec2010-03-22 17:17:26694
[email protected]2431756e2010-09-29 20:26:13695 void EnableConnectBackupJobs() { helper_.EnableConnectBackupJobs(); }
[email protected]7c28e9a2010-03-20 01:16:13696
[email protected]2431756e2010-09-29 20:26:13697 void Flush() { helper_.Flush(); }
[email protected]a7e38572010-06-07 18:22:24698
[email protected]d80a4322009-08-14 07:07:49699 private:
700 // This adaptor class exists to bridge the
701 // internal::ClientSocketPoolBaseHelper::ConnectJobFactory and
702 // ClientSocketPoolBase::ConnectJobFactory types, allowing clients to use the
703 // typesafe ClientSocketPoolBase::ConnectJobFactory, rather than having to
704 // static_cast themselves.
705 class ConnectJobFactoryAdaptor
706 : public internal::ClientSocketPoolBaseHelper::ConnectJobFactory {
707 public:
708 typedef typename ClientSocketPoolBase<SocketParams>::ConnectJobFactory
709 ConnectJobFactory;
710
[email protected]9008c86f2010-08-06 07:10:24711 explicit ConnectJobFactoryAdaptor(ConnectJobFactory* connect_job_factory)
[email protected]d80a4322009-08-14 07:07:49712 : connect_job_factory_(connect_job_factory) {}
713 virtual ~ConnectJobFactoryAdaptor() {}
714
715 virtual ConnectJob* NewConnectJob(
716 const std::string& group_name,
717 const internal::ClientSocketPoolBaseHelper::Request& request,
[email protected]06650c52010-06-03 00:49:17718 ConnectJob::Delegate* delegate) const {
[email protected]d80a4322009-08-14 07:07:49719 const Request* casted_request = static_cast<const Request*>(&request);
720 return connect_job_factory_->NewConnectJob(
[email protected]06650c52010-06-03 00:49:17721 group_name, *casted_request, delegate);
[email protected]d80a4322009-08-14 07:07:49722 }
723
[email protected]a796bcec2010-03-22 17:17:26724 virtual base::TimeDelta ConnectionTimeout() const {
725 return connect_job_factory_->ConnectionTimeout();
726 }
727
[email protected]d80a4322009-08-14 07:07:49728 const scoped_ptr<ConnectJobFactory> connect_job_factory_;
729 };
730
[email protected]b89f7e42010-05-20 20:37:00731 // Histograms for the pool
[email protected]2431756e2010-09-29 20:26:13732 ClientSocketPoolHistograms* const histograms_;
733 internal::ClientSocketPoolBaseHelper helper_;
[email protected]d80a4322009-08-14 07:07:49734
[email protected]ff579d42009-06-24 15:47:02735 DISALLOW_COPY_AND_ASSIGN(ClientSocketPoolBase);
736};
737
[email protected]ff579d42009-06-24 15:47:02738} // namespace net
739
740#endif // NET_SOCKET_CLIENT_SOCKET_POOL_BASE_H_