Chromium Code Reviews| Index: net/spdy/spdy_http_stream.cc |
| diff --git a/net/spdy/spdy_http_stream.cc b/net/spdy/spdy_http_stream.cc |
| index 96995b8429b6f7055d44705efe31bc1b1be62214..31294124919151287f4fb6e19ac8f6692012a84e 100644 |
| --- a/net/spdy/spdy_http_stream.cc |
| +++ b/net/spdy/spdy_http_stream.cc |
| @@ -8,7 +8,6 @@ |
| #include <list> |
| #include "base/bind.h" |
| -#include "base/bind_helpers.h" |
| #include "base/logging.h" |
| #include "base/message_loop.h" |
| #include "base/stringprintf.h" |
| @@ -281,56 +280,21 @@ void SpdyHttpStream::Cancel() { |
| } |
| } |
| -void SpdyHttpStream::OnStreamCreated( |
| - const CompletionCallback& callback, |
| - int rv) { |
| - if (rv == OK) { |
| - stream_ = stream_request_.ReleaseStream(); |
| - stream_->SetDelegate(this); |
| - } |
| - callback.Run(rv); |
| -} |
| - |
| -int SpdyHttpStream::SendData() { |
| - CHECK(request_info_ && request_info_->upload_data_stream); |
| - CHECK_EQ(0, request_body_buf_->BytesRemaining()); |
| - |
| - // Read the data from the request body stream. |
| - const int bytes_read = request_info_->upload_data_stream->Read( |
| - raw_request_body_buf_, raw_request_body_buf_->size(), |
| - base::Bind( |
| - base::IgnoreResult(&SpdyHttpStream::OnRequestBodyReadCompleted), |
| - weak_factory_.GetWeakPtr())); |
| - |
| - if (bytes_read == ERR_IO_PENDING) |
| - return ERR_IO_PENDING; |
| - // ERR_IO_PENDING is the only possible error. |
| - DCHECK_GE(bytes_read, 0); |
| - return OnRequestBodyReadCompleted(bytes_read); |
| -} |
| - |
| SpdySendStatus SpdyHttpStream::OnSendHeadersComplete() { |
| if (!callback_.is_null()) |
| DoCallback(OK); |
| return has_upload_data_ ? MORE_DATA_TO_SEND : NO_MORE_DATA_TO_SEND; |
| } |
| -int SpdyHttpStream::OnSendBody() { |
| +void SpdyHttpStream::OnSendBody() { |
| CHECK(request_info_ && request_info_->upload_data_stream); |
| - const bool eof = request_info_->upload_data_stream->IsEOF(); |
| if (request_body_buf_->BytesRemaining() > 0) { |
| - stream_->QueueStreamData( |
| - request_body_buf_, |
| - request_body_buf_->BytesRemaining(), |
| - eof ? DATA_FLAG_FIN : DATA_FLAG_NONE); |
| - return ERR_IO_PENDING; |
| + SendRequestBodyData(); |
| + } else { |
| + // We shouldn't be called if there's no more data to read. |
| + CHECK(!request_info_->upload_data_stream->IsEOF()); |
| + ReadAndSendRequestBodyData(); |
| } |
| - |
| - // The entire body data has been sent. |
| - if (eof) |
| - return OK; |
| - |
| - return SendData(); |
| } |
| SpdySendStatus SpdyHttpStream::OnSendBodyComplete(size_t bytes_sent) { |
| @@ -461,6 +425,52 @@ void SpdyHttpStream::OnClose(int status) { |
| DoCallback(status); |
| } |
| +void SpdyHttpStream::OnStreamCreated( |
| + const CompletionCallback& callback, |
| + int rv) { |
| + if (rv == OK) { |
| + stream_ = stream_request_.ReleaseStream(); |
| + stream_->SetDelegate(this); |
| + } |
| + callback.Run(rv); |
| +} |
| + |
| +void SpdyHttpStream::ReadAndSendRequestBodyData() { |
| + CHECK(request_info_ && request_info_->upload_data_stream); |
| + CHECK_EQ(0, request_body_buf_->BytesRemaining()); |
| + |
| + // Read the data from the request body stream. |
| + const int rv = request_info_->upload_data_stream->Read( |
| + raw_request_body_buf_, raw_request_body_buf_->size(), |
| + base::Bind( |
| + &SpdyHttpStream::OnRequestBodyReadCompleted, |
| + weak_factory_.GetWeakPtr())); |
| + |
| + if (rv != ERR_IO_PENDING) { |
| + // ERR_IO_PENDING is the only possible error. |
| + CHECK_GE(rv, 0); |
| + OnRequestBodyReadCompleted(rv); |
| + } |
| +} |
| + |
| +void SpdyHttpStream::OnRequestBodyReadCompleted(int status) { |
| + CHECK_GE(status, 0); |
| + request_body_buf_ = new DrainableIOBuffer(raw_request_body_buf_, status); |
| + SendRequestBodyData(); |
| +} |
| + |
| +void SpdyHttpStream::SendRequestBodyData() { |
| + const bool eof = request_info_->upload_data_stream->IsEOF(); |
| + if (eof) { |
| + CHECK_GE(request_body_buf_->BytesRemaining(), 0); |
|
Ryan Hamilton
2013/05/22 03:17:30
Alternatively:
CHECK_GE(request_body_buf_->BytesRe
akalin
2013/05/22 08:40:05
Hmm, I prefer the split-out CHECK, simply because
|
| + } else { |
| + CHECK_GT(request_body_buf_->BytesRemaining(), 0); |
| + } |
| + stream_->QueueStreamData(request_body_buf_, |
| + request_body_buf_->BytesRemaining(), |
| + eof ? DATA_FLAG_FIN : DATA_FLAG_NONE); |
| +} |
| + |
| void SpdyHttpStream::ScheduleBufferedReadCallback() { |
| // If there is already a scheduled DoBufferedReadCallback, don't issue |
| // another one. Mark that we have received more data and return. |
| @@ -534,18 +544,6 @@ void SpdyHttpStream::DoCallback(int rv) { |
| c.Run(rv); |
| } |
| -int SpdyHttpStream::OnRequestBodyReadCompleted(int status) { |
| - DCHECK_GE(status, 0); |
| - |
| - request_body_buf_ = new DrainableIOBuffer(raw_request_body_buf_, status); |
| - |
| - const bool eof = request_info_->upload_data_stream->IsEOF(); |
| - stream_->QueueStreamData(request_body_buf_, |
| - request_body_buf_->BytesRemaining(), |
| - eof ? DATA_FLAG_FIN : DATA_FLAG_NONE); |
| - return ERR_IO_PENDING; |
| -} |
| - |
| void SpdyHttpStream::GetSSLInfo(SSLInfo* ssl_info) { |
| DCHECK(stream_); |
| bool using_npn; |