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 |