| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 #ifndef NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ | 5 #ifndef NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ |
| 6 #define NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ | 6 #define NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ |
| 7 | 7 |
| 8 #include "base/basictypes.h" | 8 #include "base/basictypes.h" |
| 9 #include "net/base/ip_endpoint.h" | 9 #include "net/base/ip_endpoint.h" |
| 10 #include "net/quic/quic_packet_writer.h" | 10 #include "net/quic/quic_packet_writer.h" |
| 11 | 11 |
| 12 namespace net { | 12 namespace net { |
| 13 | 13 |
| 14 struct WriteResult; | 14 struct WriteResult; |
| 15 | 15 |
| 16 namespace tools { | 16 namespace tools { |
| 17 | 17 |
| 18 // Default packet writer which wraps QuicSocketUtils WritePacket. | 18 // Default packet writer which wraps QuicSocketUtils WritePacket. |
| 19 class QuicDefaultPacketWriter : public QuicPacketWriter { | 19 class QuicDefaultPacketWriter : public QuicPacketWriter { |
| 20 public: | 20 public: |
| 21 explicit QuicDefaultPacketWriter(int fd); | 21 explicit QuicDefaultPacketWriter(int fd); |
| 22 virtual ~QuicDefaultPacketWriter(); | 22 virtual ~QuicDefaultPacketWriter(); |
| 23 | 23 |
| 24 // QuicPacketWriter | 24 // QuicPacketWriter |
| 25 virtual WriteResult WritePacket( | 25 virtual WriteResult WritePacket( |
| 26 const char* buffer, size_t buf_len, | 26 const char* buffer, size_t buf_len, |
| 27 const net::IPAddressNumber& self_address, | 27 const net::IPAddressNumber& self_address, |
| 28 const net::IPEndPoint& peer_address) OVERRIDE; | 28 const net::IPEndPoint& peer_address, |
| 29 base::Callback<void(WriteResult wr)> callback) OVERRIDE; |
| 29 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE; | 30 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE; |
| 30 virtual bool IsWriteBlocked() const OVERRIDE; | 31 virtual bool IsWriteBlocked() const OVERRIDE; |
| 31 virtual void SetWritable() OVERRIDE; | 32 virtual void SetWritable() OVERRIDE; |
| 32 | 33 |
| 33 void set_fd(int fd) { fd_ = fd; } | 34 void set_fd(int fd) { fd_ = fd; } |
| 34 | 35 |
| 35 protected: | 36 protected: |
| 36 void set_write_blocked(bool is_blocked) { | 37 void set_write_blocked(bool is_blocked) { |
| 37 write_blocked_ = is_blocked; | 38 write_blocked_ = is_blocked; |
| 38 } | 39 } |
| 39 int fd() { return fd_; } | 40 int fd() { return fd_; } |
| 40 | 41 |
| 41 private: | 42 private: |
| 42 int fd_; | 43 int fd_; |
| 43 bool write_blocked_; | 44 bool write_blocked_; |
| 44 | 45 |
| 45 DISALLOW_COPY_AND_ASSIGN(QuicDefaultPacketWriter); | 46 DISALLOW_COPY_AND_ASSIGN(QuicDefaultPacketWriter); |
| 46 }; | 47 }; |
| 47 | 48 |
| 48 } // namespace tools | 49 } // namespace tools |
| 49 } // namespace net | 50 } // namespace net |
| 50 | 51 |
| 51 #endif // NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ | 52 #endif // NET_TOOLS_QUIC_QUIC_DEFAULT_PACKET_WRITER_H_ |
| OLD | NEW |