Chromium Code Reviews| Index: net/spdy/spdy_stream.cc |
| diff --git a/net/spdy/spdy_stream.cc b/net/spdy/spdy_stream.cc |
| index 2698603fe78e1007cc7d9d19ecfd70d89d46c27a..a8ab51c63759bcbd9a8238cc7b44bb5effe8a015 100644 |
| --- a/net/spdy/spdy_stream.cc |
| +++ b/net/spdy/spdy_stream.cc |
| @@ -72,11 +72,10 @@ bool ContainsUpperAscii(const std::string& str) { |
| } // namespace |
| SpdyStream::SpdyStream(SpdySession* session, |
| - SpdyStreamId stream_id, |
| bool pushed, |
| const BoundNetLog& net_log) |
| : continue_buffering_data_(true), |
| - stream_id_(stream_id), |
| + stream_id_(0), |
| priority_(HIGHEST), |
| slot_(0), |
| stalled_by_flow_control_(false), |
| @@ -99,8 +98,54 @@ SpdyStream::SpdyStream(SpdySession* session, |
| domain_bound_cert_type_(CLIENT_CERT_INVALID_TYPE) { |
| } |
| +RequestPriority SpdyStream::GetPriority() const { |
| + return priority(); |
| +} |
| + |
| +SpdyFrame* SpdyStream::ProduceNextFrame() { |
| + if (io_state_ == STATE_SEND_DOMAIN_BOUND_CERT_COMPLETE) { |
| + CHECK(request_.get()); |
| + std::string origin = GetUrl().GetOrigin().spec(); |
| + origin.erase(origin.length() - 1); // trim trailing slash |
| + SpdyCredentialControlFrame* frame = session_->CreateCredentialFrame( |
| + origin, domain_bound_cert_type_, domain_bound_private_key_, |
| + domain_bound_cert_, priority_); |
|
ramant (doing other things)
2012/06/20 01:50:40
nit: consider adding a CHECK_GT(stream_id_, 0u)?
Ryan Hamilton
2012/06/21 18:14:58
Done.
|
| + return frame; |
| + } else if (io_state_ == STATE_SEND_HEADERS_COMPLETE) { |
| + SpdyControlFlags flags = |
| + has_upload_data_ ? CONTROL_FLAG_NONE : CONTROL_FLAG_FIN; |
| + |
| + CHECK(request_.get()); |
| + SpdySynStreamControlFrame* frame = session_->CreateSynStream( |
| + stream_id_, priority_, slot_, flags, request_); |
| + set_stream_id(frame->stream_id()); |
|
ramant (doing other things)
2012/06/20 01:50:40
nit: can we do CHECK_EQ(stream_id_, 0u)? Or do we
Ryan Hamilton
2012/06/21 18:14:58
Sorry, this line of code was a bug. We should not
|
| + send_time_ = base::TimeTicks::Now(); |
| + return frame; |
| + } else { |
| + CHECK(!cancelled()); |
| + // We must need to write stream data. |
| + // Until the headers have been completely sent, we can not be sure |
| + // that our stream_id is correct. |
| + DCHECK_GT(io_state_, STATE_SEND_HEADERS_COMPLETE); |
| + DCHECK_GT(stream_id_, 0u); |
| + DCHECK(!pending_data_frames_.empty()); |
| + SpdyFrame* frame = pending_data_frames_.front(); |
| + pending_data_frames_.pop_front(); |
| + return frame; |
| + } |
| +} |
| + |
| +SpdyStream* SpdyStream::GetSpdyStream() const { |
| + return const_cast<SpdyStream*>(this); |
| +} |
| + |
| SpdyStream::~SpdyStream() { |
| UpdateHistograms(); |
| + while (!pending_data_frames_.empty()) { |
| + SpdyFrame* frame = pending_data_frames_.back(); |
| + pending_data_frames_.pop_back(); |
| + delete frame; |
| + } |
| } |
| void SpdyStream::SetDelegate(Delegate* delegate) { |
| @@ -495,7 +540,10 @@ void SpdyStream::Cancel() { |
| } |
| void SpdyStream::Close() { |
| - session_->CloseStream(stream_id_, net::OK); |
| + if (stream_id_ != 0) |
| + session_->CloseStream(stream_id_, net::OK); |
| + else |
| + session_->CloseCreatedStream(this, OK); |
| } |
| int SpdyStream::SendRequest(bool has_upload_data) { |
| @@ -522,7 +570,13 @@ int SpdyStream::WriteStreamData(IOBuffer* data, int length, |
| // Until the headers have been completely sent, we can not be sure |
| // that our stream_id is correct. |
| DCHECK_GT(io_state_, STATE_SEND_HEADERS_COMPLETE); |
| - return session_->WriteStreamData(stream_id_, data, length, flags); |
| + DCHECK_GT(stream_id_, 0u); |
|
ramant (doing other things)
2012/06/20 01:50:40
nit: can we change it to CHECK_GT to assert that s
Ryan Hamilton
2012/06/21 18:14:58
Done, but I feel like I don't have a good sense fo
|
| + |
| + pending_data_frames_.push_back( |
| + session_->CreateDataFrame(stream_id_, data, length, flags)); |
| + |
| + session_->SetStreamHasWriteAvailable(this); |
| + return ERR_IO_PENDING; |
| } |
| bool SpdyStream::GetSSLInfo(SSLInfo* ssl_info, |
| @@ -668,12 +722,12 @@ int SpdyStream::DoSendDomainBoundCert() { |
| CHECK(request_.get()); |
| std::string origin = GetUrl().GetOrigin().spec(); |
| origin.erase(origin.length() - 1); // trim trailing slash |
| - int rv = session_->WriteCredentialFrame( |
| + SpdyCredentialControlFrame* frame = session_->CreateCredentialFrame( |
| origin, domain_bound_cert_type_, domain_bound_private_key_, |
|
ramant (doing other things)
2012/06/20 01:50:40
nit: SpdyStream::ProduceNextFrame also seems to be
Ryan Hamilton
2012/06/21 18:14:58
Done.
|
| domain_bound_cert_, priority_); |
| - if (rv != ERR_IO_PENDING) |
| - return rv; |
| - return OK; |
| + frame->length(); |
| + session_->SetStreamHasWriteAvailable(this); |
| + return ERR_IO_PENDING; |
| } |
| int SpdyStream::DoSendDomainBoundCertComplete(int result) { |
| @@ -687,18 +741,7 @@ int SpdyStream::DoSendDomainBoundCertComplete(int result) { |
| int SpdyStream::DoSendHeaders() { |
| CHECK(!cancelled_); |
| - SpdyControlFlags flags = CONTROL_FLAG_NONE; |
| - if (!has_upload_data_) |
| - flags = CONTROL_FLAG_FIN; |
| - |
| - CHECK(request_.get()); |
| - int result = session_->WriteSynStream( |
| - stream_id_, priority_, slot_, flags, |
| - request_); |
| - if (result != ERR_IO_PENDING) |
| - return result; |
| - |
| - send_time_ = base::TimeTicks::Now(); |
| + session_->SetStreamHasWriteAvailable(this); |
| io_state_ = STATE_SEND_HEADERS_COMPLETE; |
| return ERR_IO_PENDING; |
| } |