| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/quic/quic_reliable_client_stream.h" | 5 #include "net/quic/quic_chromium_client_stream.h" |
| 6 | 6 |
| 7 #include "base/callback_helpers.h" | 7 #include "base/callback_helpers.h" |
| 8 #include "base/location.h" | 8 #include "base/location.h" |
| 9 #include "base/thread_task_runner_handle.h" | 9 #include "base/thread_task_runner_handle.h" |
| 10 #include "net/base/io_buffer.h" | 10 #include "net/base/io_buffer.h" |
| 11 #include "net/base/net_errors.h" | 11 #include "net/base/net_errors.h" |
| 12 #include "net/quic/quic_spdy_session.h" | 12 #include "net/quic/quic_spdy_session.h" |
| 13 #include "net/quic/quic_write_blocked_list.h" | 13 #include "net/quic/quic_write_blocked_list.h" |
| 14 #include "net/quic/spdy_utils.h" | 14 #include "net/quic/spdy_utils.h" |
| 15 | 15 |
| 16 namespace net { | 16 namespace net { |
| 17 | 17 |
| 18 QuicReliableClientStream::QuicReliableClientStream(QuicStreamId id, | 18 QuicChromiumClientStream::QuicChromiumClientStream(QuicStreamId id, |
| 19 QuicSpdySession* session, | 19 QuicSpdySession* session, |
| 20 const BoundNetLog& net_log) | 20 const BoundNetLog& net_log) |
| 21 : QuicSpdyStream(id, session), | 21 : QuicSpdyStream(id, session), |
| 22 net_log_(net_log), | 22 net_log_(net_log), |
| 23 delegate_(nullptr), | 23 delegate_(nullptr), |
| 24 headers_delivered_(false), | 24 headers_delivered_(false), |
| 25 weak_factory_(this) {} | 25 weak_factory_(this) {} |
| 26 | 26 |
| 27 QuicReliableClientStream::~QuicReliableClientStream() { | 27 QuicChromiumClientStream::~QuicChromiumClientStream() { |
| 28 if (delegate_) | 28 if (delegate_) |
| 29 delegate_->OnClose(connection_error()); | 29 delegate_->OnClose(connection_error()); |
| 30 } | 30 } |
| 31 | 31 |
| 32 void QuicReliableClientStream::OnStreamHeadersComplete(bool fin, | 32 void QuicChromiumClientStream::OnStreamHeadersComplete(bool fin, |
| 33 size_t frame_len) { | 33 size_t frame_len) { |
| 34 QuicSpdyStream::OnStreamHeadersComplete(fin, frame_len); | 34 QuicSpdyStream::OnStreamHeadersComplete(fin, frame_len); |
| 35 // The delegate will read the headers via a posted task. | 35 // The delegate will read the headers via a posted task. |
| 36 NotifyDelegateOfHeadersCompleteLater(frame_len); | 36 NotifyDelegateOfHeadersCompleteLater(frame_len); |
| 37 } | 37 } |
| 38 | 38 |
| 39 void QuicReliableClientStream::OnDataAvailable() { | 39 void QuicChromiumClientStream::OnDataAvailable() { |
| 40 // TODO(rch): buffer data if we don't have a delegate. | 40 // TODO(rch): buffer data if we don't have a delegate. |
| 41 if (!delegate_) { | 41 if (!delegate_) { |
| 42 DLOG(ERROR) << "Missing delegate"; | 42 DLOG(ERROR) << "Missing delegate"; |
| 43 Reset(QUIC_STREAM_CANCELLED); | 43 Reset(QUIC_STREAM_CANCELLED); |
| 44 return; | 44 return; |
| 45 } | 45 } |
| 46 | 46 |
| 47 if (!FinishedReadingHeaders() || !headers_delivered_) { | 47 if (!FinishedReadingHeaders() || !headers_delivered_) { |
| 48 // Buffer the data in the sequencer until the headers have been read. | 48 // Buffer the data in the sequencer until the headers have been read. |
| 49 return; | 49 return; |
| 50 } | 50 } |
| 51 | 51 |
| 52 // The delegate will read the data via a posted task, and | 52 // The delegate will read the data via a posted task, and |
| 53 // will be able to, potentially, read all data which has queued up. | 53 // will be able to, potentially, read all data which has queued up. |
| 54 NotifyDelegateOfDataAvailableLater(); | 54 NotifyDelegateOfDataAvailableLater(); |
| 55 } | 55 } |
| 56 | 56 |
| 57 void QuicReliableClientStream::OnClose() { | 57 void QuicChromiumClientStream::OnClose() { |
| 58 if (delegate_) { | 58 if (delegate_) { |
| 59 delegate_->OnClose(connection_error()); | 59 delegate_->OnClose(connection_error()); |
| 60 delegate_ = nullptr; | 60 delegate_ = nullptr; |
| 61 } | 61 } |
| 62 ReliableQuicStream::OnClose(); | 62 ReliableQuicStream::OnClose(); |
| 63 } | 63 } |
| 64 | 64 |
| 65 void QuicReliableClientStream::OnCanWrite() { | 65 void QuicChromiumClientStream::OnCanWrite() { |
| 66 ReliableQuicStream::OnCanWrite(); | 66 ReliableQuicStream::OnCanWrite(); |
| 67 | 67 |
| 68 if (!HasBufferedData() && !callback_.is_null()) { | 68 if (!HasBufferedData() && !callback_.is_null()) { |
| 69 base::ResetAndReturn(&callback_).Run(OK); | 69 base::ResetAndReturn(&callback_).Run(OK); |
| 70 } | 70 } |
| 71 } | 71 } |
| 72 | 72 |
| 73 SpdyPriority QuicReliableClientStream::Priority() const { | 73 SpdyPriority QuicChromiumClientStream::Priority() const { |
| 74 if (delegate_ && delegate_->HasSendHeadersComplete()) { | 74 if (delegate_ && delegate_->HasSendHeadersComplete()) { |
| 75 return QuicSpdyStream::Priority(); | 75 return QuicSpdyStream::Priority(); |
| 76 } | 76 } |
| 77 return net::kV3HighestPriority; | 77 return net::kV3HighestPriority; |
| 78 } | 78 } |
| 79 | 79 |
| 80 int QuicReliableClientStream::WriteStreamData( | 80 int QuicChromiumClientStream::WriteStreamData( |
| 81 base::StringPiece data, | 81 base::StringPiece data, |
| 82 bool fin, | 82 bool fin, |
| 83 const CompletionCallback& callback) { | 83 const CompletionCallback& callback) { |
| 84 // We should not have data buffered. | 84 // We should not have data buffered. |
| 85 DCHECK(!HasBufferedData()); | 85 DCHECK(!HasBufferedData()); |
| 86 // Writes the data, or buffers it. | 86 // Writes the data, or buffers it. |
| 87 WriteOrBufferData(data, fin, nullptr); | 87 WriteOrBufferData(data, fin, nullptr); |
| 88 if (!HasBufferedData()) { | 88 if (!HasBufferedData()) { |
| 89 return OK; | 89 return OK; |
| 90 } | 90 } |
| 91 | 91 |
| 92 callback_ = callback; | 92 callback_ = callback; |
| 93 return ERR_IO_PENDING; | 93 return ERR_IO_PENDING; |
| 94 } | 94 } |
| 95 | 95 |
| 96 void QuicReliableClientStream::SetDelegate( | 96 void QuicChromiumClientStream::SetDelegate( |
| 97 QuicReliableClientStream::Delegate* delegate) { | 97 QuicChromiumClientStream::Delegate* delegate) { |
| 98 DCHECK(!(delegate_ && delegate)); | 98 DCHECK(!(delegate_ && delegate)); |
| 99 delegate_ = delegate; | 99 delegate_ = delegate; |
| 100 if (delegate == nullptr && sequencer()->IsClosed()) { | 100 if (delegate == nullptr && sequencer()->IsClosed()) { |
| 101 OnFinRead(); | 101 OnFinRead(); |
| 102 } | 102 } |
| 103 } | 103 } |
| 104 | 104 |
| 105 void QuicReliableClientStream::OnError(int error) { | 105 void QuicChromiumClientStream::OnError(int error) { |
| 106 if (delegate_) { | 106 if (delegate_) { |
| 107 QuicReliableClientStream::Delegate* delegate = delegate_; | 107 QuicChromiumClientStream::Delegate* delegate = delegate_; |
| 108 delegate_ = nullptr; | 108 delegate_ = nullptr; |
| 109 delegate->OnError(error); | 109 delegate->OnError(error); |
| 110 } | 110 } |
| 111 } | 111 } |
| 112 | 112 |
| 113 int QuicReliableClientStream::Read(IOBuffer* buf, int buf_len) { | 113 int QuicChromiumClientStream::Read(IOBuffer* buf, int buf_len) { |
| 114 if (sequencer()->IsClosed()) | 114 if (sequencer()->IsClosed()) |
| 115 return 0; // EOF | 115 return 0; // EOF |
| 116 | 116 |
| 117 if (!HasBytesToRead()) | 117 if (!HasBytesToRead()) |
| 118 return ERR_IO_PENDING; | 118 return ERR_IO_PENDING; |
| 119 | 119 |
| 120 iovec iov; | 120 iovec iov; |
| 121 iov.iov_base = buf->data(); | 121 iov.iov_base = buf->data(); |
| 122 iov.iov_len = buf_len; | 122 iov.iov_len = buf_len; |
| 123 return Readv(&iov, 1); | 123 return Readv(&iov, 1); |
| 124 } | 124 } |
| 125 | 125 |
| 126 bool QuicReliableClientStream::CanWrite(const CompletionCallback& callback) { | 126 bool QuicChromiumClientStream::CanWrite(const CompletionCallback& callback) { |
| 127 bool can_write = session()->connection()->CanWrite(HAS_RETRANSMITTABLE_DATA); | 127 bool can_write = session()->connection()->CanWrite(HAS_RETRANSMITTABLE_DATA); |
| 128 if (!can_write) { | 128 if (!can_write) { |
| 129 session()->MarkConnectionLevelWriteBlocked(id(), Priority()); | 129 session()->MarkConnectionLevelWriteBlocked(id(), Priority()); |
| 130 DCHECK(callback_.is_null()); | 130 DCHECK(callback_.is_null()); |
| 131 callback_ = callback; | 131 callback_ = callback; |
| 132 } | 132 } |
| 133 return can_write; | 133 return can_write; |
| 134 } | 134 } |
| 135 | 135 |
| 136 void QuicReliableClientStream::NotifyDelegateOfHeadersCompleteLater( | 136 void QuicChromiumClientStream::NotifyDelegateOfHeadersCompleteLater( |
| 137 size_t frame_len) { | 137 size_t frame_len) { |
| 138 DCHECK(delegate_); | 138 DCHECK(delegate_); |
| 139 base::ThreadTaskRunnerHandle::Get()->PostTask( | 139 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 140 FROM_HERE, | 140 FROM_HERE, |
| 141 base::Bind(&QuicReliableClientStream::NotifyDelegateOfHeadersComplete, | 141 base::Bind(&QuicChromiumClientStream::NotifyDelegateOfHeadersComplete, |
| 142 weak_factory_.GetWeakPtr(), frame_len)); | 142 weak_factory_.GetWeakPtr(), frame_len)); |
| 143 } | 143 } |
| 144 | 144 |
| 145 void QuicReliableClientStream::NotifyDelegateOfHeadersComplete( | 145 void QuicChromiumClientStream::NotifyDelegateOfHeadersComplete( |
| 146 size_t frame_len) { | 146 size_t frame_len) { |
| 147 if (!delegate_) | 147 if (!delegate_) |
| 148 return; | 148 return; |
| 149 | 149 |
| 150 size_t headers_len = decompressed_headers().length(); | 150 size_t headers_len = decompressed_headers().length(); |
| 151 SpdyHeaderBlock headers; | 151 SpdyHeaderBlock headers; |
| 152 SpdyFramer framer(HTTP2); | 152 SpdyFramer framer(HTTP2); |
| 153 if (!framer.ParseHeaderBlockInBuffer(decompressed_headers().data(), | 153 if (!framer.ParseHeaderBlockInBuffer(decompressed_headers().data(), |
| 154 headers_len, &headers)) { | 154 headers_len, &headers)) { |
| 155 DLOG(WARNING) << "Invalid headers"; | 155 DLOG(WARNING) << "Invalid headers"; |
| 156 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 156 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
| 157 return; | 157 return; |
| 158 } | 158 } |
| 159 MarkHeadersConsumed(headers_len); | 159 MarkHeadersConsumed(headers_len); |
| 160 headers_delivered_ = true; | 160 headers_delivered_ = true; |
| 161 | 161 |
| 162 delegate_->OnHeadersAvailable(headers, frame_len); | 162 delegate_->OnHeadersAvailable(headers, frame_len); |
| 163 } | 163 } |
| 164 | 164 |
| 165 void QuicReliableClientStream::NotifyDelegateOfDataAvailableLater() { | 165 void QuicChromiumClientStream::NotifyDelegateOfDataAvailableLater() { |
| 166 DCHECK(delegate_); | 166 DCHECK(delegate_); |
| 167 base::ThreadTaskRunnerHandle::Get()->PostTask( | 167 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| 168 FROM_HERE, | 168 FROM_HERE, |
| 169 base::Bind(&QuicReliableClientStream::NotifyDelegateOfDataAvailable, | 169 base::Bind(&QuicChromiumClientStream::NotifyDelegateOfDataAvailable, |
| 170 weak_factory_.GetWeakPtr())); | 170 weak_factory_.GetWeakPtr())); |
| 171 } | 171 } |
| 172 | 172 |
| 173 void QuicReliableClientStream::NotifyDelegateOfDataAvailable() { | 173 void QuicChromiumClientStream::NotifyDelegateOfDataAvailable() { |
| 174 if (delegate_) | 174 if (delegate_) |
| 175 delegate_->OnDataAvailable(); | 175 delegate_->OnDataAvailable(); |
| 176 } | 176 } |
| 177 | 177 |
| 178 } // namespace net | 178 } // namespace net |
| OLD | NEW |