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_reliable_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" |
(...skipping 15 matching lines...) Expand all Loading... |
26 | 26 |
27 QuicReliableClientStream::~QuicReliableClientStream() { | 27 QuicReliableClientStream::~QuicReliableClientStream() { |
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 QuicReliableClientStream::OnStreamHeadersComplete(bool fin, |
33 size_t frame_len) { | 33 size_t frame_len) { |
34 QuicDataStream::OnStreamHeadersComplete(fin, frame_len); | 34 QuicDataStream::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(); | 36 NotifyDelegateOfHeadersCompleteLater(frame_len); |
37 } | 37 } |
38 | 38 |
39 void QuicReliableClientStream::OnDataAvailable() { | 39 void QuicReliableClientStream::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 |
(...skipping 79 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
126 bool QuicReliableClientStream::CanWrite(const CompletionCallback& callback) { | 126 bool QuicReliableClientStream::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(), EffectivePriority()); | 129 session()->MarkConnectionLevelWriteBlocked(id(), EffectivePriority()); |
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 QuicReliableClientStream::NotifyDelegateOfHeadersCompleteLater( |
| 137 size_t frame_len) { |
137 DCHECK(delegate_); | 138 DCHECK(delegate_); |
138 base::ThreadTaskRunnerHandle::Get()->PostTask( | 139 base::ThreadTaskRunnerHandle::Get()->PostTask( |
139 FROM_HERE, | 140 FROM_HERE, |
140 base::Bind(&QuicReliableClientStream::NotifyDelegateOfHeadersComplete, | 141 base::Bind(&QuicReliableClientStream::NotifyDelegateOfHeadersComplete, |
141 weak_factory_.GetWeakPtr())); | 142 weak_factory_.GetWeakPtr(), frame_len)); |
142 } | 143 } |
143 | 144 |
144 void QuicReliableClientStream::NotifyDelegateOfHeadersComplete() { | 145 void QuicReliableClientStream::NotifyDelegateOfHeadersComplete( |
| 146 size_t frame_len) { |
145 if (!delegate_) | 147 if (!delegate_) |
146 return; | 148 return; |
147 | 149 |
148 size_t headers_len = decompressed_headers().length(); | 150 size_t headers_len = decompressed_headers().length(); |
149 SpdyHeaderBlock headers; | 151 SpdyHeaderBlock headers; |
150 SpdyFramer framer(SpdyUtils::GetSpdyVersionForQuicVersion(version())); | 152 SpdyFramer framer(SpdyUtils::GetSpdyVersionForQuicVersion(version())); |
151 size_t len = framer.ParseHeaderBlockInBuffer(decompressed_headers().data(), | 153 size_t len = framer.ParseHeaderBlockInBuffer(decompressed_headers().data(), |
152 headers_len, &headers); | 154 headers_len, &headers); |
153 MarkHeadersConsumed(headers_len); | 155 MarkHeadersConsumed(headers_len); |
154 headers_delivered_ = true; | 156 headers_delivered_ = true; |
155 if (len == 0 || len != headers_len) { | 157 if (len == 0 || len != headers_len) { |
156 DLOG(WARNING) << "Invalid headers"; | 158 DLOG(WARNING) << "Invalid headers"; |
157 Reset(QUIC_BAD_APPLICATION_PAYLOAD); | 159 Reset(QUIC_BAD_APPLICATION_PAYLOAD); |
158 return; | 160 return; |
159 } | 161 } |
160 | 162 |
161 delegate_->OnHeadersAvailable(headers); | 163 delegate_->OnHeadersAvailable(headers, frame_len); |
162 } | 164 } |
163 | 165 |
164 void QuicReliableClientStream::NotifyDelegateOfDataAvailableLater() { | 166 void QuicReliableClientStream::NotifyDelegateOfDataAvailableLater() { |
165 DCHECK(delegate_); | 167 DCHECK(delegate_); |
166 base::ThreadTaskRunnerHandle::Get()->PostTask( | 168 base::ThreadTaskRunnerHandle::Get()->PostTask( |
167 FROM_HERE, | 169 FROM_HERE, |
168 base::Bind(&QuicReliableClientStream::NotifyDelegateOfDataAvailable, | 170 base::Bind(&QuicReliableClientStream::NotifyDelegateOfDataAvailable, |
169 weak_factory_.GetWeakPtr())); | 171 weak_factory_.GetWeakPtr())); |
170 } | 172 } |
171 | 173 |
172 void QuicReliableClientStream::NotifyDelegateOfDataAvailable() { | 174 void QuicReliableClientStream::NotifyDelegateOfDataAvailable() { |
173 if (delegate_) | 175 if (delegate_) |
174 delegate_->OnDataAvailable(); | 176 delegate_->OnDataAvailable(); |
175 } | 177 } |
176 | 178 |
177 } // namespace net | 179 } // namespace net |
OLD | NEW |