| Index: net/quic/quic_default_packet_writer.cc
|
| diff --git a/net/quic/quic_default_packet_writer.cc b/net/quic/quic_default_packet_writer.cc
|
| index 5dd80c6fd3dfd25c72e37aff5f5cac309d82c618..73f9dea826fc61e3f5f4d587690ea84caeba66ad 100644
|
| --- a/net/quic/quic_default_packet_writer.cc
|
| +++ b/net/quic/quic_default_packet_writer.cc
|
| @@ -26,7 +26,8 @@ QuicDefaultPacketWriter::~QuicDefaultPacketWriter() {}
|
| WriteResult QuicDefaultPacketWriter::WritePacket(
|
| const char* buffer, size_t buf_len,
|
| const net::IPAddressNumber& self_address,
|
| - const net::IPEndPoint& peer_address) {
|
| + const net::IPEndPoint& peer_address,
|
| + base::Callback<void(WriteResult wr)> callback) {
|
| scoped_refptr<StringIOBuffer> buf(
|
| new StringIOBuffer(std::string(buffer, buf_len)));
|
| DCHECK(!IsWriteBlocked());
|
| @@ -42,6 +43,7 @@ WriteResult QuicDefaultPacketWriter::WritePacket(
|
| } else {
|
| status = WRITE_STATUS_BLOCKED;
|
| write_blocked_ = true;
|
| + callback_ = callback;
|
| }
|
| }
|
|
|
| @@ -66,8 +68,10 @@ void QuicDefaultPacketWriter::OnWriteComplete(int rv) {
|
| DCHECK_NE(rv, ERR_IO_PENDING);
|
| write_blocked_ = false;
|
| WriteResult result(rv < 0 ? WRITE_STATUS_ERROR : WRITE_STATUS_OK, rv);
|
| - connection_->OnPacketSent(result);
|
| - connection_->OnCanWrite();
|
| + if (!callback_.is_null()) {
|
| + callback_.Run(result);
|
| + }
|
| + blocked_writer_->OnCanWrite();
|
| }
|
|
|
| } // namespace net
|
|
|