Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(692)

Side by Side Diff: net/quic/quic_server_packet_writer.h

Issue 340433002: Port QuicServer to Chrome network stack (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Indentation fix Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
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_QUIC_SERVER_PACKET_WRITER_H_ 5 #ifndef NET_QUIC_QUIC_SERVER_PACKET_WRITER_H_
6 #define NET_QUIC_QUIC_SERVER_PACKET_WRITER_H_ 6 #define NET_QUIC_QUIC_SERVER_PACKET_WRITER_H_
7 7
8 #include "base/basictypes.h" 8 #include "base/basictypes.h"
9 #include "base/memory/weak_ptr.h" 9 #include "base/memory/weak_ptr.h"
10 #include "net/base/ip_endpoint.h" 10 #include "net/base/ip_endpoint.h"
11 #include "net/quic/quic_connection.h" 11 #include "net/quic/quic_connection.h"
12 #include "net/quic/quic_packet_writer.h" 12 #include "net/quic/quic_packet_writer.h"
13 #include "net/quic/quic_protocol.h" 13 #include "net/quic/quic_protocol.h"
14 #include "net/udp/udp_server_socket.h"
15 14
16 namespace net { 15 namespace net {
17 16
17 class QuicBlockedWriterInterface;
18 class UDPServerSocket;
18 struct WriteResult; 19 struct WriteResult;
19 20
20 // Chrome specific packet writer which uses a UDPServerSocket for writing 21 // Chrome specific packet writer which uses a UDPServerSocket for writing
21 // data. 22 // data.
22 class QuicServerPacketWriter : public QuicPacketWriter { 23 class QuicServerPacketWriter : public QuicPacketWriter {
23 public: 24 public:
24 QuicServerPacketWriter(); 25 typedef base::Callback<void(WriteResult)> WriteCallback;
25 explicit QuicServerPacketWriter(UDPServerSocket* socket); 26
27 QuicServerPacketWriter(UDPServerSocket* socket,
28 QuicBlockedWriterInterface* blocked_writer);
26 virtual ~QuicServerPacketWriter(); 29 virtual ~QuicServerPacketWriter();
27 30
28 // QuicPacketWriter 31 // Use this method to write packets rather than WritePacket:
29 virtual WriteResult WritePacket(const char* buffer, 32 // QuicServerPacketWriter requires a callback to exist for every write, which
30 size_t buf_len, 33 // will be called once the write completes.
31 const net::IPAddressNumber& self_address, 34 virtual WriteResult WritePacketWithCallback(
32 const net::IPEndPoint& peer_address) OVERRIDE; 35 const char* buffer,
36 size_t buf_len,
37 const IPAddressNumber& self_address,
38 const IPEndPoint& peer_address,
39 WriteCallback callback);
40
41 void OnWriteComplete(int rv);
wtc 2014/06/24 21:47:21 Does OnWriteComplete need to be public? It seems t
dmziegler 2014/06/24 22:36:48 Good point. I'll make it private in a future CL to
42
43 // QuicPacketWriter implementation:
33 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE; 44 virtual bool IsWriteBlockedDataBuffered() const OVERRIDE;
34 virtual bool IsWriteBlocked() const OVERRIDE; 45 virtual bool IsWriteBlocked() const OVERRIDE;
35 virtual void SetWritable() OVERRIDE; 46 virtual void SetWritable() OVERRIDE;
36 47
37 void OnWriteComplete(int rv); 48 protected:
38 void SetConnection(QuicConnection* connection) { 49 // Do not call WritePacket on its own -- use WritePacketWithCallback
39 connection_ = connection; 50 virtual WriteResult WritePacket(const char* buffer,
40 } 51 size_t buf_len,
41 52 const IPAddressNumber& self_address,
53 const IPEndPoint& peer_address) OVERRIDE;
42 private: 54 private:
43 base::WeakPtrFactory<QuicServerPacketWriter> weak_factory_; 55 base::WeakPtrFactory<QuicServerPacketWriter> weak_factory_;
44 UDPServerSocket* socket_; 56 UDPServerSocket* socket_;
45 QuicConnection* connection_; 57
58 // To be notified after every successful asynchronous write.
59 QuicBlockedWriterInterface* blocked_writer_;
60
61 // To call once the write completes.
62 WriteCallback callback_;
63
64 // Whether a write is currently in flight.
46 bool write_blocked_; 65 bool write_blocked_;
47 66
48 DISALLOW_COPY_AND_ASSIGN(QuicServerPacketWriter); 67 DISALLOW_COPY_AND_ASSIGN(QuicServerPacketWriter);
49 }; 68 };
50 69
51 } // namespace net 70 } // namespace net
52 71
53 #endif // NET_QUIC_QUIC_SERVER_PACKET_WRITER_H_ 72 #endif // NET_QUIC_QUIC_SERVER_PACKET_WRITER_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698