| Index: net/quic/quic_server_packet_writer.h
|
| diff --git a/net/quic/quic_server_packet_writer.h b/net/quic/quic_server_packet_writer.h
|
| index 63829935521019e5712e27410f70bbc38485ff40..678a18a1d530a2ad87fa0ab84035dbdedc18626e 100644
|
| --- a/net/quic/quic_server_packet_writer.h
|
| +++ b/net/quic/quic_server_packet_writer.h
|
| @@ -11,38 +11,42 @@
|
| #include "net/quic/quic_connection.h"
|
| #include "net/quic/quic_packet_writer.h"
|
| #include "net/quic/quic_protocol.h"
|
| +#include "net/quic/quic_types.h"
|
| #include "net/udp/udp_server_socket.h"
|
|
|
| namespace net {
|
|
|
| -struct WriteResult;
|
| +class QuicBlockedWriterInterface;
|
|
|
| // Chrome specific packet writer which uses a UDPServerSocket for writing
|
| // data.
|
| class QuicServerPacketWriter : public QuicPacketWriter {
|
| public:
|
| - QuicServerPacketWriter();
|
| - explicit QuicServerPacketWriter(UDPServerSocket* socket);
|
| + QuicServerPacketWriter(QuicBlockedWriterInterface* blocked_writer,
|
| + UDPServerSocket* socket);
|
| virtual ~QuicServerPacketWriter();
|
|
|
| // QuicPacketWriter
|
| virtual WriteResult WritePacket(const char* buffer,
|
| size_t buf_len,
|
| - const net::IPAddressNumber& self_address,
|
| - const net::IPEndPoint& peer_address) OVERRIDE;
|
| + const IPAddressNumber& self_address,
|
| + const IPEndPoint& peer_address) OVERRIDE;
|
| virtual bool IsWriteBlockedDataBuffered() const OVERRIDE;
|
| virtual bool IsWriteBlocked() const OVERRIDE;
|
| virtual void SetWritable() OVERRIDE;
|
|
|
| void OnWriteComplete(int rv);
|
| - void SetConnection(QuicConnection* connection) {
|
| - connection_ = connection;
|
| + void SetWriteCallback(base::Callback<void(WriteResult wr)> callback) {
|
| + callback_ = callback;
|
| }
|
|
|
| private:
|
| base::WeakPtrFactory<QuicServerPacketWriter> weak_factory_;
|
| UDPServerSocket* socket_;
|
| - QuicConnection* connection_;
|
| +
|
| + // To be notified after every successful asynchronous write.
|
| + QuicBlockedWriterInterface* blocked_writer_;
|
| + base::Callback<void(WriteResult wr)> callback_;
|
| bool write_blocked_;
|
|
|
| DISALLOW_COPY_AND_ASSIGN(QuicServerPacketWriter);
|
|
|