OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ | 5 #ifndef NET_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ |
6 #define NET_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ | 6 #define NET_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ |
7 | 7 |
8 #include <stddef.h> | 8 #include <stddef.h> |
9 | 9 |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
(...skipping 15 matching lines...) Expand all Loading... |
26 // Chrome specific packet writer which uses a UDPServerSocket for writing | 26 // Chrome specific packet writer which uses a UDPServerSocket for writing |
27 // data. | 27 // data. |
28 class QuicSimpleServerPacketWriter : public QuicPacketWriter { | 28 class QuicSimpleServerPacketWriter : public QuicPacketWriter { |
29 public: | 29 public: |
30 typedef base::Callback<void(WriteResult)> WriteCallback; | 30 typedef base::Callback<void(WriteResult)> WriteCallback; |
31 | 31 |
32 QuicSimpleServerPacketWriter(UDPServerSocket* socket, | 32 QuicSimpleServerPacketWriter(UDPServerSocket* socket, |
33 QuicBlockedWriterInterface* blocked_writer); | 33 QuicBlockedWriterInterface* blocked_writer); |
34 ~QuicSimpleServerPacketWriter() override; | 34 ~QuicSimpleServerPacketWriter() override; |
35 | 35 |
36 // Use this method to write packets rather than WritePacket: | 36 // Wraps WritePacket, and ensures that |callback| is run on successful write. |
37 // QuicSimpleServerPacketWriter requires a callback to exist for every | 37 WriteResult WritePacketWithCallback(const char* buffer, |
38 // write, which will be called once the write completes. | 38 size_t buf_len, |
39 virtual WriteResult WritePacketWithCallback(const char* buffer, | 39 const IPAddress& self_address, |
40 size_t buf_len, | 40 const IPEndPoint& peer_address, |
41 const IPAddress& self_address, | 41 PerPacketOptions* options, |
42 const IPEndPoint& peer_address, | 42 WriteCallback callback); |
43 PerPacketOptions* options, | 43 |
44 WriteCallback callback); | 44 WriteResult WritePacket(const char* buffer, |
| 45 size_t buf_len, |
| 46 const IPAddress& self_address, |
| 47 const IPEndPoint& peer_address, |
| 48 PerPacketOptions* options) override; |
45 | 49 |
46 void OnWriteComplete(int rv); | 50 void OnWriteComplete(int rv); |
47 | 51 |
48 // QuicPacketWriter implementation: | 52 // QuicPacketWriter implementation: |
49 bool IsWriteBlockedDataBuffered() const override; | 53 bool IsWriteBlockedDataBuffered() const override; |
50 bool IsWriteBlocked() const override; | 54 bool IsWriteBlocked() const override; |
51 void SetWritable() override; | 55 void SetWritable() override; |
52 QuicByteCount GetMaxPacketSize(const IPEndPoint& peer_address) const override; | 56 QuicByteCount GetMaxPacketSize(const IPEndPoint& peer_address) const override; |
53 | 57 |
54 protected: | |
55 // Do not call WritePacket on its own -- use WritePacketWithCallback | |
56 WriteResult WritePacket(const char* buffer, | |
57 size_t buf_len, | |
58 const IPAddress& self_address, | |
59 const IPEndPoint& peer_address, | |
60 PerPacketOptions* options) override; | |
61 | |
62 private: | 58 private: |
63 UDPServerSocket* socket_; | 59 UDPServerSocket* socket_; |
64 | 60 |
65 // To be notified after every successful asynchronous write. | 61 // To be notified after every successful asynchronous write. |
66 QuicBlockedWriterInterface* blocked_writer_; | 62 QuicBlockedWriterInterface* blocked_writer_; |
67 | 63 |
68 // To call once the write completes. | 64 // To call once the write completes. |
69 WriteCallback callback_; | 65 WriteCallback callback_; |
70 | 66 |
71 // Whether a write is currently in flight. | 67 // Whether a write is currently in flight. |
72 bool write_blocked_; | 68 bool write_blocked_; |
73 | 69 |
74 base::WeakPtrFactory<QuicSimpleServerPacketWriter> weak_factory_; | 70 base::WeakPtrFactory<QuicSimpleServerPacketWriter> weak_factory_; |
75 | 71 |
76 DISALLOW_COPY_AND_ASSIGN(QuicSimpleServerPacketWriter); | 72 DISALLOW_COPY_AND_ASSIGN(QuicSimpleServerPacketWriter); |
77 }; | 73 }; |
78 | 74 |
79 } // namespace net | 75 } // namespace net |
80 | 76 |
81 #endif // NET_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ | 77 #endif // NET_QUIC_TOOLS_QUIC_SIMPLE_SERVER_PACKET_WRITER_H_ |
OLD | NEW |