| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/chromium/bidirectional_stream_quic_impl.h" | 5 #include "net/quic/chromium/bidirectional_stream_quic_impl.h" |
| 6 | 6 |
| 7 #include <utility> | 7 #include <utility> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/location.h" | 10 #include "base/location.h" |
| (...skipping 150 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 161 return rv; | 161 return rv; |
| 162 | 162 |
| 163 if (stream_->IsDoneReading()) { | 163 if (stream_->IsDoneReading()) { |
| 164 // If the write side is closed, OnFinRead() will call | 164 // If the write side is closed, OnFinRead() will call |
| 165 // BidirectionalStreamQuicImpl::OnClose(). | 165 // BidirectionalStreamQuicImpl::OnClose(). |
| 166 stream_->OnFinRead(); | 166 stream_->OnFinRead(); |
| 167 } | 167 } |
| 168 return rv; | 168 return rv; |
| 169 } | 169 } |
| 170 | 170 |
| 171 void BidirectionalStreamQuicImpl::SendData(const scoped_refptr<IOBuffer>& data, | |
| 172 int length, | |
| 173 bool end_stream) { | |
| 174 ScopedBoolSaver saver(&may_invoke_callbacks_, false); | |
| 175 DCHECK(length > 0 || (length == 0 && end_stream)); | |
| 176 if (!stream_) { | |
| 177 LOG(ERROR) << "Trying to send data after stream has been destroyed."; | |
| 178 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
| 179 FROM_HERE, base::Bind(&BidirectionalStreamQuicImpl::NotifyError, | |
| 180 weak_factory_.GetWeakPtr(), ERR_UNEXPECTED)); | |
| 181 return; | |
| 182 } | |
| 183 | |
| 184 std::unique_ptr<QuicConnection::ScopedPacketBundler> bundler; | |
| 185 if (!has_sent_headers_) { | |
| 186 DCHECK(!send_request_headers_automatically_); | |
| 187 // Creates a bundler only if there are headers to be sent along with the | |
| 188 // single data buffer. | |
| 189 bundler = | |
| 190 session_->CreatePacketBundler(QuicConnection::SEND_ACK_IF_PENDING); | |
| 191 // Sending the request might result in the stream being closed. | |
| 192 if (!WriteHeaders()) | |
| 193 return; | |
| 194 } | |
| 195 | |
| 196 QuicStringPiece string_data(data->data(), length); | |
| 197 int rv = stream_->WriteStreamData( | |
| 198 string_data, end_stream, | |
| 199 base::Bind(&BidirectionalStreamQuicImpl::OnSendDataComplete, | |
| 200 weak_factory_.GetWeakPtr())); | |
| 201 DCHECK(rv == OK || rv == ERR_IO_PENDING); | |
| 202 if (rv == OK) { | |
| 203 base::ThreadTaskRunnerHandle::Get()->PostTask( | |
| 204 FROM_HERE, base::Bind(&BidirectionalStreamQuicImpl::OnSendDataComplete, | |
| 205 weak_factory_.GetWeakPtr(), OK)); | |
| 206 } | |
| 207 } | |
| 208 | |
| 209 void BidirectionalStreamQuicImpl::SendvData( | 171 void BidirectionalStreamQuicImpl::SendvData( |
| 210 const std::vector<scoped_refptr<IOBuffer>>& buffers, | 172 const std::vector<scoped_refptr<IOBuffer>>& buffers, |
| 211 const std::vector<int>& lengths, | 173 const std::vector<int>& lengths, |
| 212 bool end_stream) { | 174 bool end_stream) { |
| 213 ScopedBoolSaver saver(&may_invoke_callbacks_, false); | 175 ScopedBoolSaver saver(&may_invoke_callbacks_, false); |
| 214 DCHECK_EQ(buffers.size(), lengths.size()); | 176 DCHECK_EQ(buffers.size(), lengths.size()); |
| 215 | 177 |
| 216 if (!stream_) { | 178 if (!stream_) { |
| 217 LOG(ERROR) << "Trying to send data after stream has been destroyed."; | 179 LOG(ERROR) << "Trying to send data after stream has been destroyed."; |
| 218 base::ThreadTaskRunnerHandle::Get()->PostTask( | 180 base::ThreadTaskRunnerHandle::Get()->PostTask( |
| (...skipping 227 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 446 if (!stream_) | 408 if (!stream_) |
| 447 return; | 409 return; |
| 448 closed_stream_received_bytes_ = stream_->stream_bytes_read(); | 410 closed_stream_received_bytes_ = stream_->stream_bytes_read(); |
| 449 closed_stream_sent_bytes_ = stream_->stream_bytes_written(); | 411 closed_stream_sent_bytes_ = stream_->stream_bytes_written(); |
| 450 closed_is_first_stream_ = stream_->IsFirstStream(); | 412 closed_is_first_stream_ = stream_->IsFirstStream(); |
| 451 stream_->ClearDelegate(); | 413 stream_->ClearDelegate(); |
| 452 stream_ = nullptr; | 414 stream_ = nullptr; |
| 453 } | 415 } |
| 454 | 416 |
| 455 } // namespace net | 417 } // namespace net |
| OLD | NEW |