OLD | NEW |
| (Empty) |
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 #include "net/socket/client_socket_handle.h" | |
6 | |
7 #include "base/bind.h" | |
8 #include "base/bind_helpers.h" | |
9 #include "base/compiler_specific.h" | |
10 #include "base/metrics/histogram.h" | |
11 #include "base/logging.h" | |
12 #include "net/base/net_errors.h" | |
13 #include "net/socket/client_socket_pool.h" | |
14 #include "net/socket/client_socket_pool_histograms.h" | |
15 | |
16 namespace net { | |
17 | |
18 ClientSocketHandle::ClientSocketHandle() | |
19 : is_initialized_(false), | |
20 pool_(NULL), | |
21 higher_pool_(NULL), | |
22 reuse_type_(ClientSocketHandle::UNUSED), | |
23 callback_(base::Bind(&ClientSocketHandle::OnIOComplete, | |
24 base::Unretained(this))), | |
25 is_ssl_error_(false) {} | |
26 | |
27 ClientSocketHandle::~ClientSocketHandle() { | |
28 Reset(); | |
29 } | |
30 | |
31 void ClientSocketHandle::Reset() { | |
32 ResetInternal(true); | |
33 ResetErrorState(); | |
34 } | |
35 | |
36 void ClientSocketHandle::ResetInternal(bool cancel) { | |
37 // Was Init called? | |
38 if (!group_name_.empty()) { | |
39 // If so, we must have a pool. | |
40 CHECK(pool_); | |
41 if (is_initialized()) { | |
42 if (socket_) { | |
43 socket_->NetLog().EndEvent(NetLog::TYPE_SOCKET_IN_USE); | |
44 // Release the socket back to the ClientSocketPool so it can be | |
45 // deleted or reused. | |
46 pool_->ReleaseSocket(group_name_, socket_.Pass(), pool_id_); | |
47 } else { | |
48 // If the handle has been initialized, we should still have a | |
49 // socket. | |
50 NOTREACHED(); | |
51 } | |
52 } else if (cancel) { | |
53 // If we did not get initialized yet and we have a socket | |
54 // request pending, cancel it. | |
55 pool_->CancelRequest(group_name_, this); | |
56 } | |
57 } | |
58 is_initialized_ = false; | |
59 socket_.reset(); | |
60 group_name_.clear(); | |
61 reuse_type_ = ClientSocketHandle::UNUSED; | |
62 user_callback_.Reset(); | |
63 if (higher_pool_) | |
64 RemoveHigherLayeredPool(higher_pool_); | |
65 pool_ = NULL; | |
66 idle_time_ = base::TimeDelta(); | |
67 init_time_ = base::TimeTicks(); | |
68 setup_time_ = base::TimeDelta(); | |
69 connect_timing_ = LoadTimingInfo::ConnectTiming(); | |
70 pool_id_ = -1; | |
71 } | |
72 | |
73 void ClientSocketHandle::ResetErrorState() { | |
74 is_ssl_error_ = false; | |
75 ssl_error_response_info_ = HttpResponseInfo(); | |
76 pending_http_proxy_connection_.reset(); | |
77 } | |
78 | |
79 LoadState ClientSocketHandle::GetLoadState() const { | |
80 CHECK(!is_initialized()); | |
81 CHECK(!group_name_.empty()); | |
82 // Because of http://crbug.com/37810 we may not have a pool, but have | |
83 // just a raw socket. | |
84 if (!pool_) | |
85 return LOAD_STATE_IDLE; | |
86 return pool_->GetLoadState(group_name_, this); | |
87 } | |
88 | |
89 bool ClientSocketHandle::IsPoolStalled() const { | |
90 if (!pool_) | |
91 return false; | |
92 return pool_->IsStalled(); | |
93 } | |
94 | |
95 void ClientSocketHandle::AddHigherLayeredPool(HigherLayeredPool* higher_pool) { | |
96 CHECK(higher_pool); | |
97 CHECK(!higher_pool_); | |
98 // TODO(mmenke): |pool_| should only be NULL in tests. Maybe stop doing that | |
99 // so this be be made into a DCHECK, and the same can be done in | |
100 // RemoveHigherLayeredPool? | |
101 if (pool_) { | |
102 pool_->AddHigherLayeredPool(higher_pool); | |
103 higher_pool_ = higher_pool; | |
104 } | |
105 } | |
106 | |
107 void ClientSocketHandle::RemoveHigherLayeredPool( | |
108 HigherLayeredPool* higher_pool) { | |
109 CHECK(higher_pool_); | |
110 CHECK_EQ(higher_pool_, higher_pool); | |
111 if (pool_) { | |
112 pool_->RemoveHigherLayeredPool(higher_pool); | |
113 higher_pool_ = NULL; | |
114 } | |
115 } | |
116 | |
117 bool ClientSocketHandle::GetLoadTimingInfo( | |
118 bool is_reused, | |
119 LoadTimingInfo* load_timing_info) const { | |
120 // Only return load timing information when there's a socket. | |
121 if (!socket_) | |
122 return false; | |
123 | |
124 load_timing_info->socket_log_id = socket_->NetLog().source().id; | |
125 load_timing_info->socket_reused = is_reused; | |
126 | |
127 // No times if the socket is reused. | |
128 if (is_reused) | |
129 return true; | |
130 | |
131 load_timing_info->connect_timing = connect_timing_; | |
132 return true; | |
133 } | |
134 | |
135 void ClientSocketHandle::SetSocket(scoped_ptr<StreamSocket> s) { | |
136 socket_ = s.Pass(); | |
137 } | |
138 | |
139 void ClientSocketHandle::OnIOComplete(int result) { | |
140 CompletionCallback callback = user_callback_; | |
141 user_callback_.Reset(); | |
142 HandleInitCompletion(result); | |
143 callback.Run(result); | |
144 } | |
145 | |
146 scoped_ptr<StreamSocket> ClientSocketHandle::PassSocket() { | |
147 return socket_.Pass(); | |
148 } | |
149 | |
150 void ClientSocketHandle::HandleInitCompletion(int result) { | |
151 CHECK_NE(ERR_IO_PENDING, result); | |
152 ClientSocketPoolHistograms* histograms = pool_->histograms(); | |
153 histograms->AddErrorCode(result); | |
154 if (result != OK) { | |
155 if (!socket_.get()) | |
156 ResetInternal(false); // Nothing to cancel since the request failed. | |
157 else | |
158 is_initialized_ = true; | |
159 return; | |
160 } | |
161 is_initialized_ = true; | |
162 CHECK_NE(-1, pool_id_) << "Pool should have set |pool_id_| to a valid value."; | |
163 setup_time_ = base::TimeTicks::Now() - init_time_; | |
164 | |
165 histograms->AddSocketType(reuse_type()); | |
166 switch (reuse_type()) { | |
167 case ClientSocketHandle::UNUSED: | |
168 histograms->AddRequestTime(setup_time()); | |
169 break; | |
170 case ClientSocketHandle::UNUSED_IDLE: | |
171 histograms->AddUnusedIdleTime(idle_time()); | |
172 break; | |
173 case ClientSocketHandle::REUSED_IDLE: | |
174 histograms->AddReusedIdleTime(idle_time()); | |
175 break; | |
176 default: | |
177 NOTREACHED(); | |
178 break; | |
179 } | |
180 | |
181 // Broadcast that the socket has been acquired. | |
182 // TODO(eroman): This logging is not complete, in particular set_socket() and | |
183 // release() socket. It ends up working though, since those methods are being | |
184 // used to layer sockets (and the destination sources are the same). | |
185 DCHECK(socket_.get()); | |
186 socket_->NetLog().BeginEvent( | |
187 NetLog::TYPE_SOCKET_IN_USE, | |
188 requesting_source_.ToEventParametersCallback()); | |
189 } | |
190 | |
191 } // namespace net | |
OLD | NEW |