| 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/spdy/spdy_stream.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/compiler_specific.h" | |
| 9 #include "base/logging.h" | |
| 10 #include "base/message_loop/message_loop.h" | |
| 11 #include "base/profiler/scoped_tracker.h" | |
| 12 #include "base/strings/string_number_conversions.h" | |
| 13 #include "base/strings/stringprintf.h" | |
| 14 #include "base/values.h" | |
| 15 #include "net/spdy/spdy_buffer_producer.h" | |
| 16 #include "net/spdy/spdy_http_utils.h" | |
| 17 #include "net/spdy/spdy_session.h" | |
| 18 | |
| 19 namespace net { | |
| 20 | |
| 21 namespace { | |
| 22 | |
| 23 base::Value* NetLogSpdyStreamErrorCallback(SpdyStreamId stream_id, | |
| 24 int status, | |
| 25 const std::string* description, | |
| 26 NetLog::LogLevel /* log_level */) { | |
| 27 base::DictionaryValue* dict = new base::DictionaryValue(); | |
| 28 dict->SetInteger("stream_id", static_cast<int>(stream_id)); | |
| 29 dict->SetInteger("status", status); | |
| 30 dict->SetString("description", *description); | |
| 31 return dict; | |
| 32 } | |
| 33 | |
| 34 base::Value* NetLogSpdyStreamWindowUpdateCallback( | |
| 35 SpdyStreamId stream_id, | |
| 36 int32 delta, | |
| 37 int32 window_size, | |
| 38 NetLog::LogLevel /* log_level */) { | |
| 39 base::DictionaryValue* dict = new base::DictionaryValue(); | |
| 40 dict->SetInteger("stream_id", stream_id); | |
| 41 dict->SetInteger("delta", delta); | |
| 42 dict->SetInteger("window_size", window_size); | |
| 43 return dict; | |
| 44 } | |
| 45 | |
| 46 bool ContainsUppercaseAscii(const std::string& str) { | |
| 47 for (std::string::const_iterator i(str.begin()); i != str.end(); ++i) { | |
| 48 if (*i >= 'A' && *i <= 'Z') { | |
| 49 return true; | |
| 50 } | |
| 51 } | |
| 52 return false; | |
| 53 } | |
| 54 | |
| 55 } // namespace | |
| 56 | |
| 57 // A wrapper around a stream that calls into ProduceSynStreamFrame(). | |
| 58 class SpdyStream::SynStreamBufferProducer : public SpdyBufferProducer { | |
| 59 public: | |
| 60 SynStreamBufferProducer(const base::WeakPtr<SpdyStream>& stream) | |
| 61 : stream_(stream) { | |
| 62 DCHECK(stream_.get()); | |
| 63 } | |
| 64 | |
| 65 ~SynStreamBufferProducer() override {} | |
| 66 | |
| 67 scoped_ptr<SpdyBuffer> ProduceBuffer() override { | |
| 68 if (!stream_.get()) { | |
| 69 NOTREACHED(); | |
| 70 return scoped_ptr<SpdyBuffer>(); | |
| 71 } | |
| 72 DCHECK_GT(stream_->stream_id(), 0u); | |
| 73 return scoped_ptr<SpdyBuffer>( | |
| 74 new SpdyBuffer(stream_->ProduceSynStreamFrame())); | |
| 75 } | |
| 76 | |
| 77 private: | |
| 78 const base::WeakPtr<SpdyStream> stream_; | |
| 79 }; | |
| 80 | |
| 81 SpdyStream::SpdyStream(SpdyStreamType type, | |
| 82 const base::WeakPtr<SpdySession>& session, | |
| 83 const GURL& url, | |
| 84 RequestPriority priority, | |
| 85 int32 initial_send_window_size, | |
| 86 int32 initial_recv_window_size, | |
| 87 const BoundNetLog& net_log) | |
| 88 : type_(type), | |
| 89 stream_id_(0), | |
| 90 url_(url), | |
| 91 priority_(priority), | |
| 92 send_stalled_by_flow_control_(false), | |
| 93 send_window_size_(initial_send_window_size), | |
| 94 recv_window_size_(initial_recv_window_size), | |
| 95 unacked_recv_window_bytes_(0), | |
| 96 session_(session), | |
| 97 delegate_(NULL), | |
| 98 pending_send_status_(MORE_DATA_TO_SEND), | |
| 99 request_time_(base::Time::Now()), | |
| 100 response_headers_status_(RESPONSE_HEADERS_ARE_INCOMPLETE), | |
| 101 io_state_(STATE_IDLE), | |
| 102 response_status_(OK), | |
| 103 net_log_(net_log), | |
| 104 raw_received_bytes_(0), | |
| 105 send_bytes_(0), | |
| 106 recv_bytes_(0), | |
| 107 write_handler_guard_(false), | |
| 108 weak_ptr_factory_(this) { | |
| 109 CHECK(type_ == SPDY_BIDIRECTIONAL_STREAM || | |
| 110 type_ == SPDY_REQUEST_RESPONSE_STREAM || | |
| 111 type_ == SPDY_PUSH_STREAM); | |
| 112 CHECK_GE(priority_, MINIMUM_PRIORITY); | |
| 113 CHECK_LE(priority_, MAXIMUM_PRIORITY); | |
| 114 } | |
| 115 | |
| 116 SpdyStream::~SpdyStream() { | |
| 117 CHECK(!write_handler_guard_); | |
| 118 UpdateHistograms(); | |
| 119 } | |
| 120 | |
| 121 void SpdyStream::SetDelegate(Delegate* delegate) { | |
| 122 CHECK(!delegate_); | |
| 123 CHECK(delegate); | |
| 124 delegate_ = delegate; | |
| 125 | |
| 126 CHECK(io_state_ == STATE_IDLE || | |
| 127 io_state_ == STATE_HALF_CLOSED_LOCAL_UNCLAIMED || | |
| 128 io_state_ == STATE_RESERVED_REMOTE); | |
| 129 | |
| 130 if (io_state_ == STATE_HALF_CLOSED_LOCAL_UNCLAIMED) { | |
| 131 DCHECK_EQ(type_, SPDY_PUSH_STREAM); | |
| 132 base::MessageLoop::current()->PostTask( | |
| 133 FROM_HERE, | |
| 134 base::Bind(&SpdyStream::PushedStreamReplay, GetWeakPtr())); | |
| 135 } | |
| 136 } | |
| 137 | |
| 138 void SpdyStream::PushedStreamReplay() { | |
| 139 DCHECK_EQ(type_, SPDY_PUSH_STREAM); | |
| 140 DCHECK_NE(stream_id_, 0u); | |
| 141 CHECK_EQ(stream_id_ % 2, 0u); | |
| 142 | |
| 143 CHECK_EQ(io_state_, STATE_HALF_CLOSED_LOCAL_UNCLAIMED); | |
| 144 io_state_ = STATE_HALF_CLOSED_LOCAL; | |
| 145 | |
| 146 // The delegate methods called below may delete |this|, so use | |
| 147 // |weak_this| to detect that. | |
| 148 base::WeakPtr<SpdyStream> weak_this = GetWeakPtr(); | |
| 149 | |
| 150 CHECK(delegate_); | |
| 151 SpdyResponseHeadersStatus status = | |
| 152 delegate_->OnResponseHeadersUpdated(response_headers_); | |
| 153 if (status == RESPONSE_HEADERS_ARE_INCOMPLETE) { | |
| 154 // Since RESPONSE_HEADERS_ARE_INCOMPLETE was returned, we must not | |
| 155 // have been closed. Since we don't have complete headers, assume | |
| 156 // we're waiting for another HEADERS frame, and we had better not | |
| 157 // have any pending data frames. | |
| 158 CHECK(weak_this); | |
| 159 if (!pending_recv_data_.empty()) { | |
| 160 LogStreamError(ERR_SPDY_PROTOCOL_ERROR, | |
| 161 "Data received with incomplete headers."); | |
| 162 session_->CloseActiveStream(stream_id_, ERR_SPDY_PROTOCOL_ERROR); | |
| 163 } | |
| 164 return; | |
| 165 } | |
| 166 | |
| 167 // OnResponseHeadersUpdated() may have closed |this|. | |
| 168 if (!weak_this) | |
| 169 return; | |
| 170 | |
| 171 response_headers_status_ = RESPONSE_HEADERS_ARE_COMPLETE; | |
| 172 | |
| 173 while (!pending_recv_data_.empty()) { | |
| 174 // Take ownership of the first element of |pending_recv_data_|. | |
| 175 scoped_ptr<SpdyBuffer> buffer(pending_recv_data_.front()); | |
| 176 pending_recv_data_.weak_erase(pending_recv_data_.begin()); | |
| 177 | |
| 178 bool eof = (buffer == NULL); | |
| 179 | |
| 180 CHECK(delegate_); | |
| 181 delegate_->OnDataReceived(buffer.Pass()); | |
| 182 | |
| 183 // OnDataReceived() may have closed |this|. | |
| 184 if (!weak_this) | |
| 185 return; | |
| 186 | |
| 187 if (eof) { | |
| 188 DCHECK(pending_recv_data_.empty()); | |
| 189 session_->CloseActiveStream(stream_id_, OK); | |
| 190 DCHECK(!weak_this); | |
| 191 // |pending_recv_data_| is invalid at this point. | |
| 192 break; | |
| 193 } | |
| 194 } | |
| 195 } | |
| 196 | |
| 197 scoped_ptr<SpdyFrame> SpdyStream::ProduceSynStreamFrame() { | |
| 198 CHECK_EQ(io_state_, STATE_IDLE); | |
| 199 CHECK(request_headers_); | |
| 200 CHECK_GT(stream_id_, 0u); | |
| 201 | |
| 202 SpdyControlFlags flags = | |
| 203 (pending_send_status_ == NO_MORE_DATA_TO_SEND) ? | |
| 204 CONTROL_FLAG_FIN : CONTROL_FLAG_NONE; | |
| 205 scoped_ptr<SpdyFrame> frame(session_->CreateSynStream( | |
| 206 stream_id_, priority_, flags, *request_headers_)); | |
| 207 send_time_ = base::TimeTicks::Now(); | |
| 208 return frame.Pass(); | |
| 209 } | |
| 210 | |
| 211 void SpdyStream::DetachDelegate() { | |
| 212 DCHECK(!IsClosed()); | |
| 213 delegate_ = NULL; | |
| 214 Cancel(); | |
| 215 } | |
| 216 | |
| 217 void SpdyStream::AdjustSendWindowSize(int32 delta_window_size) { | |
| 218 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 219 | |
| 220 if (IsClosed()) | |
| 221 return; | |
| 222 | |
| 223 // Check for wraparound. | |
| 224 if (send_window_size_ > 0) { | |
| 225 DCHECK_LE(delta_window_size, kint32max - send_window_size_); | |
| 226 } | |
| 227 if (send_window_size_ < 0) { | |
| 228 DCHECK_GE(delta_window_size, kint32min - send_window_size_); | |
| 229 } | |
| 230 send_window_size_ += delta_window_size; | |
| 231 PossiblyResumeIfSendStalled(); | |
| 232 } | |
| 233 | |
| 234 void SpdyStream::OnWriteBufferConsumed( | |
| 235 size_t frame_payload_size, | |
| 236 size_t consume_size, | |
| 237 SpdyBuffer::ConsumeSource consume_source) { | |
| 238 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 239 if (consume_source == SpdyBuffer::DISCARD) { | |
| 240 // If we're discarding a frame or part of it, increase the send | |
| 241 // window by the number of discarded bytes. (Although if we're | |
| 242 // discarding part of a frame, it's probably because of a write | |
| 243 // error and we'll be tearing down the stream soon.) | |
| 244 size_t remaining_payload_bytes = std::min(consume_size, frame_payload_size); | |
| 245 DCHECK_GT(remaining_payload_bytes, 0u); | |
| 246 IncreaseSendWindowSize(static_cast<int32>(remaining_payload_bytes)); | |
| 247 } | |
| 248 // For consumed bytes, the send window is increased when we receive | |
| 249 // a WINDOW_UPDATE frame. | |
| 250 } | |
| 251 | |
| 252 void SpdyStream::IncreaseSendWindowSize(int32 delta_window_size) { | |
| 253 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 254 DCHECK_GE(delta_window_size, 1); | |
| 255 | |
| 256 // Ignore late WINDOW_UPDATEs. | |
| 257 if (IsClosed()) | |
| 258 return; | |
| 259 | |
| 260 if (send_window_size_ > 0) { | |
| 261 // Check for overflow. | |
| 262 int32 max_delta_window_size = kint32max - send_window_size_; | |
| 263 if (delta_window_size > max_delta_window_size) { | |
| 264 std::string desc = base::StringPrintf( | |
| 265 "Received WINDOW_UPDATE [delta: %d] for stream %d overflows " | |
| 266 "send_window_size_ [current: %d]", delta_window_size, stream_id_, | |
| 267 send_window_size_); | |
| 268 session_->ResetStream(stream_id_, RST_STREAM_FLOW_CONTROL_ERROR, desc); | |
| 269 return; | |
| 270 } | |
| 271 } | |
| 272 | |
| 273 send_window_size_ += delta_window_size; | |
| 274 | |
| 275 net_log_.AddEvent( | |
| 276 NetLog::TYPE_SPDY_STREAM_UPDATE_SEND_WINDOW, | |
| 277 base::Bind(&NetLogSpdyStreamWindowUpdateCallback, | |
| 278 stream_id_, delta_window_size, send_window_size_)); | |
| 279 | |
| 280 PossiblyResumeIfSendStalled(); | |
| 281 } | |
| 282 | |
| 283 void SpdyStream::DecreaseSendWindowSize(int32 delta_window_size) { | |
| 284 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 285 | |
| 286 if (IsClosed()) | |
| 287 return; | |
| 288 | |
| 289 // We only call this method when sending a frame. Therefore, | |
| 290 // |delta_window_size| should be within the valid frame size range. | |
| 291 DCHECK_GE(delta_window_size, 1); | |
| 292 DCHECK_LE(delta_window_size, kMaxSpdyFrameChunkSize); | |
| 293 | |
| 294 // |send_window_size_| should have been at least |delta_window_size| for | |
| 295 // this call to happen. | |
| 296 DCHECK_GE(send_window_size_, delta_window_size); | |
| 297 | |
| 298 send_window_size_ -= delta_window_size; | |
| 299 | |
| 300 net_log_.AddEvent( | |
| 301 NetLog::TYPE_SPDY_STREAM_UPDATE_SEND_WINDOW, | |
| 302 base::Bind(&NetLogSpdyStreamWindowUpdateCallback, | |
| 303 stream_id_, -delta_window_size, send_window_size_)); | |
| 304 } | |
| 305 | |
| 306 void SpdyStream::OnReadBufferConsumed( | |
| 307 size_t consume_size, | |
| 308 SpdyBuffer::ConsumeSource consume_source) { | |
| 309 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 310 DCHECK_GE(consume_size, 1u); | |
| 311 DCHECK_LE(consume_size, static_cast<size_t>(kint32max)); | |
| 312 IncreaseRecvWindowSize(static_cast<int32>(consume_size)); | |
| 313 } | |
| 314 | |
| 315 void SpdyStream::IncreaseRecvWindowSize(int32 delta_window_size) { | |
| 316 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 317 | |
| 318 // By the time a read is processed by the delegate, this stream may | |
| 319 // already be inactive. | |
| 320 if (!session_->IsStreamActive(stream_id_)) | |
| 321 return; | |
| 322 | |
| 323 DCHECK_GE(unacked_recv_window_bytes_, 0); | |
| 324 DCHECK_GE(recv_window_size_, unacked_recv_window_bytes_); | |
| 325 DCHECK_GE(delta_window_size, 1); | |
| 326 // Check for overflow. | |
| 327 DCHECK_LE(delta_window_size, kint32max - recv_window_size_); | |
| 328 | |
| 329 recv_window_size_ += delta_window_size; | |
| 330 net_log_.AddEvent( | |
| 331 NetLog::TYPE_SPDY_STREAM_UPDATE_RECV_WINDOW, | |
| 332 base::Bind(&NetLogSpdyStreamWindowUpdateCallback, | |
| 333 stream_id_, delta_window_size, recv_window_size_)); | |
| 334 | |
| 335 unacked_recv_window_bytes_ += delta_window_size; | |
| 336 if (unacked_recv_window_bytes_ > | |
| 337 session_->stream_initial_recv_window_size() / 2) { | |
| 338 session_->SendStreamWindowUpdate( | |
| 339 stream_id_, static_cast<uint32>(unacked_recv_window_bytes_)); | |
| 340 unacked_recv_window_bytes_ = 0; | |
| 341 } | |
| 342 } | |
| 343 | |
| 344 void SpdyStream::DecreaseRecvWindowSize(int32 delta_window_size) { | |
| 345 DCHECK(session_->IsStreamActive(stream_id_)); | |
| 346 DCHECK_GE(session_->flow_control_state(), SpdySession::FLOW_CONTROL_STREAM); | |
| 347 DCHECK_GE(delta_window_size, 1); | |
| 348 | |
| 349 // Since we never decrease the initial receive window size, | |
| 350 // |delta_window_size| should never cause |recv_window_size_| to go | |
| 351 // negative. If we do, the receive window isn't being respected. | |
| 352 if (delta_window_size > recv_window_size_) { | |
| 353 session_->ResetStream( | |
| 354 stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 355 "delta_window_size is " + base::IntToString(delta_window_size) + | |
| 356 " in DecreaseRecvWindowSize, which is larger than the receive " + | |
| 357 "window size of " + base::IntToString(recv_window_size_)); | |
| 358 return; | |
| 359 } | |
| 360 | |
| 361 recv_window_size_ -= delta_window_size; | |
| 362 net_log_.AddEvent( | |
| 363 NetLog::TYPE_SPDY_STREAM_UPDATE_RECV_WINDOW, | |
| 364 base::Bind(&NetLogSpdyStreamWindowUpdateCallback, | |
| 365 stream_id_, -delta_window_size, recv_window_size_)); | |
| 366 } | |
| 367 | |
| 368 int SpdyStream::GetPeerAddress(IPEndPoint* address) const { | |
| 369 return session_->GetPeerAddress(address); | |
| 370 } | |
| 371 | |
| 372 int SpdyStream::GetLocalAddress(IPEndPoint* address) const { | |
| 373 return session_->GetLocalAddress(address); | |
| 374 } | |
| 375 | |
| 376 bool SpdyStream::WasEverUsed() const { | |
| 377 return session_->WasEverUsed(); | |
| 378 } | |
| 379 | |
| 380 base::Time SpdyStream::GetRequestTime() const { | |
| 381 return request_time_; | |
| 382 } | |
| 383 | |
| 384 void SpdyStream::SetRequestTime(base::Time t) { | |
| 385 request_time_ = t; | |
| 386 } | |
| 387 | |
| 388 int SpdyStream::OnInitialResponseHeadersReceived( | |
| 389 const SpdyHeaderBlock& initial_response_headers, | |
| 390 base::Time response_time, | |
| 391 base::TimeTicks recv_first_byte_time) { | |
| 392 // SpdySession guarantees that this is called at most once. | |
| 393 CHECK(response_headers_.empty()); | |
| 394 | |
| 395 // Check to make sure that we don't receive the response headers | |
| 396 // before we're ready for it. | |
| 397 switch (type_) { | |
| 398 case SPDY_BIDIRECTIONAL_STREAM: | |
| 399 // For a bidirectional stream, we're ready for the response | |
| 400 // headers once we've finished sending the request headers. | |
| 401 if (io_state_ == STATE_IDLE) { | |
| 402 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 403 "Response received before request sent"); | |
| 404 return ERR_SPDY_PROTOCOL_ERROR; | |
| 405 } | |
| 406 break; | |
| 407 | |
| 408 case SPDY_REQUEST_RESPONSE_STREAM: | |
| 409 // For a request/response stream, we're ready for the response | |
| 410 // headers once we've finished sending the request headers. | |
| 411 if (io_state_ == STATE_IDLE) { | |
| 412 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 413 "Response received before request sent"); | |
| 414 return ERR_SPDY_PROTOCOL_ERROR; | |
| 415 } | |
| 416 break; | |
| 417 | |
| 418 case SPDY_PUSH_STREAM: | |
| 419 // Push streams transition to a locally half-closed state upon headers. | |
| 420 // We must continue to buffer data while waiting for a call to | |
| 421 // SetDelegate() (which may not ever happen). | |
| 422 CHECK_EQ(io_state_, STATE_RESERVED_REMOTE); | |
| 423 if (!delegate_) { | |
| 424 io_state_ = STATE_HALF_CLOSED_LOCAL_UNCLAIMED; | |
| 425 } else { | |
| 426 io_state_ = STATE_HALF_CLOSED_LOCAL; | |
| 427 } | |
| 428 break; | |
| 429 } | |
| 430 | |
| 431 metrics_.StartStream(); | |
| 432 | |
| 433 DCHECK_NE(io_state_, STATE_IDLE); | |
| 434 | |
| 435 response_time_ = response_time; | |
| 436 recv_first_byte_time_ = recv_first_byte_time; | |
| 437 return MergeWithResponseHeaders(initial_response_headers); | |
| 438 } | |
| 439 | |
| 440 int SpdyStream::OnAdditionalResponseHeadersReceived( | |
| 441 const SpdyHeaderBlock& additional_response_headers) { | |
| 442 if (type_ == SPDY_REQUEST_RESPONSE_STREAM) { | |
| 443 session_->ResetStream( | |
| 444 stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 445 "Additional headers received for request/response stream"); | |
| 446 return ERR_SPDY_PROTOCOL_ERROR; | |
| 447 } else if (type_ == SPDY_PUSH_STREAM && | |
| 448 response_headers_status_ == RESPONSE_HEADERS_ARE_COMPLETE) { | |
| 449 session_->ResetStream( | |
| 450 stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 451 "Additional headers received for push stream"); | |
| 452 return ERR_SPDY_PROTOCOL_ERROR; | |
| 453 } | |
| 454 return MergeWithResponseHeaders(additional_response_headers); | |
| 455 } | |
| 456 | |
| 457 void SpdyStream::OnPushPromiseHeadersReceived(const SpdyHeaderBlock& headers) { | |
| 458 CHECK(!request_headers_.get()); | |
| 459 CHECK_EQ(io_state_, STATE_IDLE); | |
| 460 CHECK_EQ(type_, SPDY_PUSH_STREAM); | |
| 461 DCHECK(!delegate_); | |
| 462 | |
| 463 io_state_ = STATE_RESERVED_REMOTE; | |
| 464 request_headers_.reset(new SpdyHeaderBlock(headers)); | |
| 465 } | |
| 466 | |
| 467 void SpdyStream::OnDataReceived(scoped_ptr<SpdyBuffer> buffer) { | |
| 468 DCHECK(session_->IsStreamActive(stream_id_)); | |
| 469 | |
| 470 // If we're still buffering data for a push stream, we will do the | |
| 471 // check for data received with incomplete headers in | |
| 472 // PushedStreamReplayData(). | |
| 473 if (io_state_ == STATE_HALF_CLOSED_LOCAL_UNCLAIMED) { | |
| 474 DCHECK_EQ(type_, SPDY_PUSH_STREAM); | |
| 475 // It should be valid for this to happen in the server push case. | |
| 476 // We'll return received data when delegate gets attached to the stream. | |
| 477 if (buffer) { | |
| 478 pending_recv_data_.push_back(buffer.release()); | |
| 479 } else { | |
| 480 pending_recv_data_.push_back(NULL); | |
| 481 metrics_.StopStream(); | |
| 482 // Note: we leave the stream open in the session until the stream | |
| 483 // is claimed. | |
| 484 } | |
| 485 return; | |
| 486 } | |
| 487 | |
| 488 // If we have response headers but the delegate has indicated that | |
| 489 // it's still incomplete, then that's a protocol error. | |
| 490 if (response_headers_status_ == RESPONSE_HEADERS_ARE_INCOMPLETE) { | |
| 491 LogStreamError(ERR_SPDY_PROTOCOL_ERROR, | |
| 492 "Data received with incomplete headers."); | |
| 493 session_->CloseActiveStream(stream_id_, ERR_SPDY_PROTOCOL_ERROR); | |
| 494 return; | |
| 495 } | |
| 496 | |
| 497 CHECK(!IsClosed()); | |
| 498 | |
| 499 if (!buffer) { | |
| 500 metrics_.StopStream(); | |
| 501 if (io_state_ == STATE_OPEN) { | |
| 502 io_state_ = STATE_HALF_CLOSED_REMOTE; | |
| 503 } else if (io_state_ == STATE_HALF_CLOSED_LOCAL) { | |
| 504 io_state_ = STATE_CLOSED; | |
| 505 // Deletes |this|. | |
| 506 session_->CloseActiveStream(stream_id_, OK); | |
| 507 } else { | |
| 508 NOTREACHED() << io_state_; | |
| 509 } | |
| 510 return; | |
| 511 } | |
| 512 | |
| 513 size_t length = buffer->GetRemainingSize(); | |
| 514 DCHECK_LE(length, session_->GetDataFrameMaximumPayload()); | |
| 515 if (session_->flow_control_state() >= SpdySession::FLOW_CONTROL_STREAM) { | |
| 516 DecreaseRecvWindowSize(static_cast<int32>(length)); | |
| 517 buffer->AddConsumeCallback( | |
| 518 base::Bind(&SpdyStream::OnReadBufferConsumed, GetWeakPtr())); | |
| 519 } | |
| 520 | |
| 521 // Track our bandwidth. | |
| 522 metrics_.RecordBytes(length); | |
| 523 recv_bytes_ += length; | |
| 524 recv_last_byte_time_ = base::TimeTicks::Now(); | |
| 525 | |
| 526 // May close |this|. | |
| 527 delegate_->OnDataReceived(buffer.Pass()); | |
| 528 } | |
| 529 | |
| 530 void SpdyStream::OnFrameWriteComplete(SpdyFrameType frame_type, | |
| 531 size_t frame_size) { | |
| 532 // TODO(pkasting): Remove ScopedTracker below once crbug.com/457517 is fixed. | |
| 533 tracked_objects::ScopedTracker tracking_profile( | |
| 534 FROM_HERE_WITH_EXPLICIT_FUNCTION( | |
| 535 "457517 SpdyStream::OnFrameWriteComplete")); | |
| 536 DCHECK_NE(type_, SPDY_PUSH_STREAM); | |
| 537 | |
| 538 if (frame_size < session_->GetFrameMinimumSize() || | |
| 539 frame_size > session_->GetFrameMaximumSize()) { | |
| 540 NOTREACHED(); | |
| 541 return; | |
| 542 } | |
| 543 CHECK(frame_type == SYN_STREAM || | |
| 544 frame_type == DATA) << frame_type; | |
| 545 | |
| 546 int result = (frame_type == SYN_STREAM) ? | |
| 547 OnRequestHeadersSent() : OnDataSent(frame_size); | |
| 548 if (result == ERR_IO_PENDING) { | |
| 549 // The write operation hasn't completed yet. | |
| 550 return; | |
| 551 } | |
| 552 | |
| 553 if (pending_send_status_ == NO_MORE_DATA_TO_SEND) { | |
| 554 if(io_state_ == STATE_OPEN) { | |
| 555 io_state_ = STATE_HALF_CLOSED_LOCAL; | |
| 556 } else if(io_state_ == STATE_HALF_CLOSED_REMOTE) { | |
| 557 io_state_ = STATE_CLOSED; | |
| 558 } else { | |
| 559 NOTREACHED() << io_state_; | |
| 560 } | |
| 561 } | |
| 562 // Notify delegate of write completion. Must not destroy |this|. | |
| 563 CHECK(delegate_); | |
| 564 { | |
| 565 base::WeakPtr<SpdyStream> weak_this = GetWeakPtr(); | |
| 566 write_handler_guard_ = true; | |
| 567 if (frame_type == SYN_STREAM) { | |
| 568 delegate_->OnRequestHeadersSent(); | |
| 569 } else { | |
| 570 delegate_->OnDataSent(); | |
| 571 } | |
| 572 CHECK(weak_this); | |
| 573 write_handler_guard_ = false; | |
| 574 } | |
| 575 | |
| 576 if (io_state_ == STATE_CLOSED) { | |
| 577 // Deletes |this|. | |
| 578 session_->CloseActiveStream(stream_id_, OK); | |
| 579 } | |
| 580 } | |
| 581 | |
| 582 int SpdyStream::OnRequestHeadersSent() { | |
| 583 CHECK_EQ(io_state_, STATE_IDLE); | |
| 584 CHECK_NE(stream_id_, 0u); | |
| 585 | |
| 586 io_state_ = STATE_OPEN; | |
| 587 return OK; | |
| 588 } | |
| 589 | |
| 590 int SpdyStream::OnDataSent(size_t frame_size) { | |
| 591 CHECK(io_state_ == STATE_OPEN || | |
| 592 io_state_ == STATE_HALF_CLOSED_REMOTE) << io_state_; | |
| 593 | |
| 594 size_t frame_payload_size = frame_size - session_->GetDataFrameMinimumSize(); | |
| 595 | |
| 596 CHECK_GE(frame_size, session_->GetDataFrameMinimumSize()); | |
| 597 CHECK_LE(frame_payload_size, session_->GetDataFrameMaximumPayload()); | |
| 598 | |
| 599 send_bytes_ += frame_payload_size; | |
| 600 | |
| 601 // If more data is available to send, dispatch it and | |
| 602 // return that the write operation is still ongoing. | |
| 603 pending_send_data_->DidConsume(frame_payload_size); | |
| 604 if (pending_send_data_->BytesRemaining() > 0) { | |
| 605 QueueNextDataFrame(); | |
| 606 return ERR_IO_PENDING; | |
| 607 } else { | |
| 608 pending_send_data_ = NULL; | |
| 609 return OK; | |
| 610 } | |
| 611 } | |
| 612 | |
| 613 SpdyMajorVersion SpdyStream::GetProtocolVersion() const { | |
| 614 return session_->GetProtocolVersion(); | |
| 615 } | |
| 616 | |
| 617 void SpdyStream::LogStreamError(int status, const std::string& description) { | |
| 618 net_log_.AddEvent(NetLog::TYPE_SPDY_STREAM_ERROR, | |
| 619 base::Bind(&NetLogSpdyStreamErrorCallback, | |
| 620 stream_id_, status, &description)); | |
| 621 } | |
| 622 | |
| 623 void SpdyStream::OnClose(int status) { | |
| 624 // In most cases, the stream should already be CLOSED. The exception is when a | |
| 625 // SpdySession is shutting down while the stream is in an intermediate state. | |
| 626 io_state_ = STATE_CLOSED; | |
| 627 response_status_ = status; | |
| 628 Delegate* delegate = delegate_; | |
| 629 delegate_ = NULL; | |
| 630 if (delegate) | |
| 631 delegate->OnClose(status); | |
| 632 // Unset |stream_id_| last so that the delegate can look it up. | |
| 633 stream_id_ = 0; | |
| 634 } | |
| 635 | |
| 636 void SpdyStream::Cancel() { | |
| 637 // We may be called again from a delegate's OnClose(). | |
| 638 if (io_state_ == STATE_CLOSED) | |
| 639 return; | |
| 640 | |
| 641 if (stream_id_ != 0) { | |
| 642 session_->ResetStream(stream_id_, RST_STREAM_CANCEL, std::string()); | |
| 643 } else { | |
| 644 session_->CloseCreatedStream(GetWeakPtr(), RST_STREAM_CANCEL); | |
| 645 } | |
| 646 // |this| is invalid at this point. | |
| 647 } | |
| 648 | |
| 649 void SpdyStream::Close() { | |
| 650 // We may be called again from a delegate's OnClose(). | |
| 651 if (io_state_ == STATE_CLOSED) | |
| 652 return; | |
| 653 | |
| 654 if (stream_id_ != 0) { | |
| 655 session_->CloseActiveStream(stream_id_, OK); | |
| 656 } else { | |
| 657 session_->CloseCreatedStream(GetWeakPtr(), OK); | |
| 658 } | |
| 659 // |this| is invalid at this point. | |
| 660 } | |
| 661 | |
| 662 base::WeakPtr<SpdyStream> SpdyStream::GetWeakPtr() { | |
| 663 return weak_ptr_factory_.GetWeakPtr(); | |
| 664 } | |
| 665 | |
| 666 int SpdyStream::SendRequestHeaders(scoped_ptr<SpdyHeaderBlock> request_headers, | |
| 667 SpdySendStatus send_status) { | |
| 668 CHECK_NE(type_, SPDY_PUSH_STREAM); | |
| 669 CHECK_EQ(pending_send_status_, MORE_DATA_TO_SEND); | |
| 670 CHECK(!request_headers_); | |
| 671 CHECK(!pending_send_data_.get()); | |
| 672 CHECK_EQ(io_state_, STATE_IDLE); | |
| 673 request_headers_ = request_headers.Pass(); | |
| 674 pending_send_status_ = send_status; | |
| 675 session_->EnqueueStreamWrite( | |
| 676 GetWeakPtr(), SYN_STREAM, | |
| 677 scoped_ptr<SpdyBufferProducer>( | |
| 678 new SynStreamBufferProducer(GetWeakPtr()))); | |
| 679 return ERR_IO_PENDING; | |
| 680 } | |
| 681 | |
| 682 void SpdyStream::SendData(IOBuffer* data, | |
| 683 int length, | |
| 684 SpdySendStatus send_status) { | |
| 685 CHECK_NE(type_, SPDY_PUSH_STREAM); | |
| 686 CHECK_EQ(pending_send_status_, MORE_DATA_TO_SEND); | |
| 687 CHECK(io_state_ == STATE_OPEN || | |
| 688 io_state_ == STATE_HALF_CLOSED_REMOTE) << io_state_; | |
| 689 CHECK(!pending_send_data_.get()); | |
| 690 pending_send_data_ = new DrainableIOBuffer(data, length); | |
| 691 pending_send_status_ = send_status; | |
| 692 QueueNextDataFrame(); | |
| 693 } | |
| 694 | |
| 695 bool SpdyStream::GetSSLInfo(SSLInfo* ssl_info, | |
| 696 bool* was_npn_negotiated, | |
| 697 NextProto* protocol_negotiated) { | |
| 698 return session_->GetSSLInfo( | |
| 699 ssl_info, was_npn_negotiated, protocol_negotiated); | |
| 700 } | |
| 701 | |
| 702 bool SpdyStream::GetSSLCertRequestInfo(SSLCertRequestInfo* cert_request_info) { | |
| 703 return session_->GetSSLCertRequestInfo(cert_request_info); | |
| 704 } | |
| 705 | |
| 706 void SpdyStream::PossiblyResumeIfSendStalled() { | |
| 707 if (IsLocallyClosed()) { | |
| 708 return; | |
| 709 } | |
| 710 if (send_stalled_by_flow_control_ && !session_->IsSendStalled() && | |
| 711 send_window_size_ > 0) { | |
| 712 net_log_.AddEvent( | |
| 713 NetLog::TYPE_SPDY_STREAM_FLOW_CONTROL_UNSTALLED, | |
| 714 NetLog::IntegerCallback("stream_id", stream_id_)); | |
| 715 send_stalled_by_flow_control_ = false; | |
| 716 QueueNextDataFrame(); | |
| 717 } | |
| 718 } | |
| 719 | |
| 720 bool SpdyStream::IsClosed() const { | |
| 721 return io_state_ == STATE_CLOSED; | |
| 722 } | |
| 723 | |
| 724 bool SpdyStream::IsLocallyClosed() const { | |
| 725 return io_state_ == STATE_HALF_CLOSED_LOCAL_UNCLAIMED || | |
| 726 io_state_ == STATE_HALF_CLOSED_LOCAL || | |
| 727 io_state_ == STATE_CLOSED; | |
| 728 } | |
| 729 | |
| 730 bool SpdyStream::IsIdle() const { | |
| 731 return io_state_ == STATE_IDLE; | |
| 732 } | |
| 733 | |
| 734 bool SpdyStream::IsOpen() const { | |
| 735 return io_state_ == STATE_OPEN; | |
| 736 } | |
| 737 | |
| 738 bool SpdyStream::IsReservedRemote() const { | |
| 739 return io_state_ == STATE_RESERVED_REMOTE; | |
| 740 } | |
| 741 | |
| 742 NextProto SpdyStream::GetProtocol() const { | |
| 743 return session_->protocol(); | |
| 744 } | |
| 745 | |
| 746 bool SpdyStream::GetLoadTimingInfo(LoadTimingInfo* load_timing_info) const { | |
| 747 if (stream_id_ == 0) | |
| 748 return false; | |
| 749 | |
| 750 return session_->GetLoadTimingInfo(stream_id_, load_timing_info); | |
| 751 } | |
| 752 | |
| 753 GURL SpdyStream::GetUrlFromHeaders() const { | |
| 754 if (!request_headers_) | |
| 755 return GURL(); | |
| 756 | |
| 757 return GetUrlFromHeaderBlock( | |
| 758 *request_headers_, GetProtocolVersion(), type_ == SPDY_PUSH_STREAM); | |
| 759 } | |
| 760 | |
| 761 bool SpdyStream::HasUrlFromHeaders() const { | |
| 762 return !GetUrlFromHeaders().is_empty(); | |
| 763 } | |
| 764 | |
| 765 void SpdyStream::UpdateHistograms() { | |
| 766 // We need at least the receive timers to be filled in, as otherwise | |
| 767 // metrics can be bogus. | |
| 768 if (recv_first_byte_time_.is_null() || recv_last_byte_time_.is_null()) | |
| 769 return; | |
| 770 | |
| 771 base::TimeTicks effective_send_time; | |
| 772 if (type_ == SPDY_PUSH_STREAM) { | |
| 773 // Push streams shouldn't have |send_time_| filled in. | |
| 774 DCHECK(send_time_.is_null()); | |
| 775 effective_send_time = recv_first_byte_time_; | |
| 776 } else { | |
| 777 // For non-push streams, we also need |send_time_| to be filled | |
| 778 // in. | |
| 779 if (send_time_.is_null()) | |
| 780 return; | |
| 781 effective_send_time = send_time_; | |
| 782 } | |
| 783 | |
| 784 UMA_HISTOGRAM_TIMES("Net.SpdyStreamTimeToFirstByte", | |
| 785 recv_first_byte_time_ - effective_send_time); | |
| 786 UMA_HISTOGRAM_TIMES("Net.SpdyStreamDownloadTime", | |
| 787 recv_last_byte_time_ - recv_first_byte_time_); | |
| 788 UMA_HISTOGRAM_TIMES("Net.SpdyStreamTime", | |
| 789 recv_last_byte_time_ - effective_send_time); | |
| 790 | |
| 791 UMA_HISTOGRAM_COUNTS("Net.SpdySendBytes", send_bytes_); | |
| 792 UMA_HISTOGRAM_COUNTS("Net.SpdyRecvBytes", recv_bytes_); | |
| 793 } | |
| 794 | |
| 795 void SpdyStream::QueueNextDataFrame() { | |
| 796 // Until the request has been completely sent, we cannot be sure | |
| 797 // that our stream_id is correct. | |
| 798 CHECK(io_state_ == STATE_OPEN || | |
| 799 io_state_ == STATE_HALF_CLOSED_REMOTE) << io_state_; | |
| 800 CHECK_GT(stream_id_, 0u); | |
| 801 CHECK(pending_send_data_.get()); | |
| 802 // Only the final fame may have a length of 0. | |
| 803 if (pending_send_status_ == NO_MORE_DATA_TO_SEND) { | |
| 804 CHECK_GE(pending_send_data_->BytesRemaining(), 0); | |
| 805 } else { | |
| 806 CHECK_GT(pending_send_data_->BytesRemaining(), 0); | |
| 807 } | |
| 808 | |
| 809 SpdyDataFlags flags = | |
| 810 (pending_send_status_ == NO_MORE_DATA_TO_SEND) ? | |
| 811 DATA_FLAG_FIN : DATA_FLAG_NONE; | |
| 812 scoped_ptr<SpdyBuffer> data_buffer( | |
| 813 session_->CreateDataBuffer(stream_id_, | |
| 814 pending_send_data_.get(), | |
| 815 pending_send_data_->BytesRemaining(), | |
| 816 flags)); | |
| 817 // We'll get called again by PossiblyResumeIfSendStalled(). | |
| 818 if (!data_buffer) | |
| 819 return; | |
| 820 | |
| 821 if (session_->flow_control_state() >= SpdySession::FLOW_CONTROL_STREAM) { | |
| 822 DCHECK_GE(data_buffer->GetRemainingSize(), | |
| 823 session_->GetDataFrameMinimumSize()); | |
| 824 size_t payload_size = | |
| 825 data_buffer->GetRemainingSize() - session_->GetDataFrameMinimumSize(); | |
| 826 DCHECK_LE(payload_size, session_->GetDataFrameMaximumPayload()); | |
| 827 | |
| 828 // Send window size is based on payload size, so nothing to do if this is | |
| 829 // just a FIN with no payload. | |
| 830 if (payload_size != 0) { | |
| 831 DecreaseSendWindowSize(static_cast<int32>(payload_size)); | |
| 832 // This currently isn't strictly needed, since write frames are | |
| 833 // discarded only if the stream is about to be closed. But have it | |
| 834 // here anyway just in case this changes. | |
| 835 data_buffer->AddConsumeCallback( | |
| 836 base::Bind(&SpdyStream::OnWriteBufferConsumed, | |
| 837 GetWeakPtr(), payload_size)); | |
| 838 } | |
| 839 } | |
| 840 | |
| 841 session_->EnqueueStreamWrite( | |
| 842 GetWeakPtr(), DATA, | |
| 843 scoped_ptr<SpdyBufferProducer>( | |
| 844 new SimpleBufferProducer(data_buffer.Pass()))); | |
| 845 } | |
| 846 | |
| 847 int SpdyStream::MergeWithResponseHeaders( | |
| 848 const SpdyHeaderBlock& new_response_headers) { | |
| 849 if (new_response_headers.find("transfer-encoding") != | |
| 850 new_response_headers.end()) { | |
| 851 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 852 "Received transfer-encoding header"); | |
| 853 return ERR_SPDY_PROTOCOL_ERROR; | |
| 854 } | |
| 855 | |
| 856 for (SpdyHeaderBlock::const_iterator it = new_response_headers.begin(); | |
| 857 it != new_response_headers.end(); ++it) { | |
| 858 // Disallow uppercase headers. | |
| 859 if (ContainsUppercaseAscii(it->first)) { | |
| 860 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 861 "Upper case characters in header: " + it->first); | |
| 862 return ERR_SPDY_PROTOCOL_ERROR; | |
| 863 } | |
| 864 | |
| 865 SpdyHeaderBlock::iterator it2 = response_headers_.lower_bound(it->first); | |
| 866 // Disallow duplicate headers. This is just to be conservative. | |
| 867 if (it2 != response_headers_.end() && it2->first == it->first) { | |
| 868 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 869 "Duplicate header: " + it->first); | |
| 870 return ERR_SPDY_PROTOCOL_ERROR; | |
| 871 } | |
| 872 | |
| 873 response_headers_.insert(it2, *it); | |
| 874 } | |
| 875 | |
| 876 // If delegate_ is not yet attached, we'll call | |
| 877 // OnResponseHeadersUpdated() after the delegate gets attached to | |
| 878 // the stream. | |
| 879 if (delegate_) { | |
| 880 // The call to OnResponseHeadersUpdated() below may delete |this|, | |
| 881 // so use |weak_this| to detect that. | |
| 882 base::WeakPtr<SpdyStream> weak_this = GetWeakPtr(); | |
| 883 | |
| 884 SpdyResponseHeadersStatus status = | |
| 885 delegate_->OnResponseHeadersUpdated(response_headers_); | |
| 886 if (status == RESPONSE_HEADERS_ARE_INCOMPLETE) { | |
| 887 // Since RESPONSE_HEADERS_ARE_INCOMPLETE was returned, we must not | |
| 888 // have been closed. | |
| 889 CHECK(weak_this); | |
| 890 // Incomplete headers are OK only for push streams. | |
| 891 if (type_ != SPDY_PUSH_STREAM) { | |
| 892 session_->ResetStream(stream_id_, RST_STREAM_PROTOCOL_ERROR, | |
| 893 "Incomplete headers"); | |
| 894 return ERR_INCOMPLETE_SPDY_HEADERS; | |
| 895 } | |
| 896 } else if (weak_this) { | |
| 897 response_headers_status_ = RESPONSE_HEADERS_ARE_COMPLETE; | |
| 898 } | |
| 899 } | |
| 900 | |
| 901 return OK; | |
| 902 } | |
| 903 | |
| 904 #define STATE_CASE(s) \ | |
| 905 case s: \ | |
| 906 description = base::StringPrintf("%s (0x%08X)", #s, s); \ | |
| 907 break | |
| 908 | |
| 909 std::string SpdyStream::DescribeState(State state) { | |
| 910 std::string description; | |
| 911 switch (state) { | |
| 912 STATE_CASE(STATE_IDLE); | |
| 913 STATE_CASE(STATE_OPEN); | |
| 914 STATE_CASE(STATE_HALF_CLOSED_LOCAL_UNCLAIMED); | |
| 915 STATE_CASE(STATE_HALF_CLOSED_LOCAL); | |
| 916 STATE_CASE(STATE_CLOSED); | |
| 917 default: | |
| 918 description = base::StringPrintf("Unknown state 0x%08X (%u)", state, | |
| 919 state); | |
| 920 break; | |
| 921 } | |
| 922 return description; | |
| 923 } | |
| 924 | |
| 925 #undef STATE_CASE | |
| 926 | |
| 927 } // namespace net | |
| OLD | NEW |