| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 // A server side dispatcher which dispatches a given client's data to their | 5 // A server side dispatcher which dispatches a given client's data to their |
| 6 // stream. | 6 // stream. |
| 7 | 7 |
| 8 #ifndef NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ | 8 #ifndef NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ |
| 9 #define NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ | 9 #define NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ |
| 10 | 10 |
| (...skipping 23 matching lines...) Expand all Loading... |
| 34 | 34 |
| 35 class ProcessPacketInterface { | 35 class ProcessPacketInterface { |
| 36 public: | 36 public: |
| 37 virtual ~ProcessPacketInterface() {} | 37 virtual ~ProcessPacketInterface() {} |
| 38 virtual void ProcessPacket(const IPEndPoint& server_address, | 38 virtual void ProcessPacket(const IPEndPoint& server_address, |
| 39 const IPEndPoint& client_address, | 39 const IPEndPoint& client_address, |
| 40 const QuicEncryptedPacket& packet) = 0; | 40 const QuicEncryptedPacket& packet) = 0; |
| 41 }; | 41 }; |
| 42 | 42 |
| 43 class QuicDispatcher : public QuicServerSessionVisitor, | 43 class QuicDispatcher : public QuicServerSessionVisitor, |
| 44 public ProcessPacketInterface { | 44 public ProcessPacketInterface, |
| 45 public QuicBlockedWriterInterface { |
| 45 public: | 46 public: |
| 46 // Creates per-connection packet writers out of the QuicDispatcher's shared | 47 // Creates per-connection packet writers out of the QuicDispatcher's shared |
| 47 // QuicPacketWriter. The per-connection writers' IsWriteBlocked() state must | 48 // QuicPacketWriter. The per-connection writers' IsWriteBlocked() state must |
| 48 // always be the same as the shared writer's IsWriteBlocked(), or else the | 49 // always be the same as the shared writer's IsWriteBlocked(), or else the |
| 49 // QuicDispatcher::OnCanWrite logic will not work. (This will hopefully be | 50 // QuicDispatcher::OnCanWrite logic will not work. (This will hopefully be |
| 50 // cleaned up for bug 16950226.) | 51 // cleaned up for bug 16950226.) |
| 51 class PacketWriterFactory { | 52 class PacketWriterFactory { |
| 52 public: | 53 public: |
| 53 virtual ~PacketWriterFactory() {} | 54 virtual ~PacketWriterFactory() {} |
| 54 | 55 |
| (...skipping 18 matching lines...) Expand all Loading... |
| 73 // of supported QUIC versions. Takes ownership of |packet_writer_factory|, | 74 // of supported QUIC versions. Takes ownership of |packet_writer_factory|, |
| 74 // which is used to create per-connection writers. | 75 // which is used to create per-connection writers. |
| 75 QuicDispatcher(const QuicConfig& config, | 76 QuicDispatcher(const QuicConfig& config, |
| 76 const QuicCryptoServerConfig& crypto_config, | 77 const QuicCryptoServerConfig& crypto_config, |
| 77 const QuicVersionVector& supported_versions, | 78 const QuicVersionVector& supported_versions, |
| 78 PacketWriterFactory* packet_writer_factory, | 79 PacketWriterFactory* packet_writer_factory, |
| 79 QuicConnectionHelperInterface* helper); | 80 QuicConnectionHelperInterface* helper); |
| 80 | 81 |
| 81 ~QuicDispatcher() override; | 82 ~QuicDispatcher() override; |
| 82 | 83 |
| 83 virtual void Initialize(int fd); | 84 // Takes ownership of |writer|. |
| 85 void InitializeWithWriter(QuicPacketWriter* writer); |
| 84 | 86 |
| 85 // Process the incoming packet by creating a new session, passing it to | 87 // Process the incoming packet by creating a new session, passing it to |
| 86 // an existing session, or passing it to the TimeWaitListManager. | 88 // an existing session, or passing it to the TimeWaitListManager. |
| 87 void ProcessPacket(const IPEndPoint& server_address, | 89 void ProcessPacket(const IPEndPoint& server_address, |
| 88 const IPEndPoint& client_address, | 90 const IPEndPoint& client_address, |
| 89 const QuicEncryptedPacket& packet) override; | 91 const QuicEncryptedPacket& packet) override; |
| 90 | 92 |
| 91 // Called when the socket becomes writable to allow queued writes to happen. | 93 // Called when the socket becomes writable to allow queued writes to happen. |
| 92 virtual void OnCanWrite(); | 94 void OnCanWrite() override; |
| 93 | 95 |
| 94 // Returns true if there's anything in the blocked writer list. | 96 // Returns true if there's anything in the blocked writer list. |
| 95 virtual bool HasPendingWrites() const; | 97 virtual bool HasPendingWrites() const; |
| 96 | 98 |
| 97 // Sends ConnectionClose frames to all connected clients. | 99 // Sends ConnectionClose frames to all connected clients. |
| 98 void Shutdown(); | 100 void Shutdown(); |
| 99 | 101 |
| 100 // QuicServerSessionVisitor interface implementation: | 102 // QuicServerSessionVisitor interface implementation: |
| 101 // Ensure that the closed connection is cleaned up asynchronously. | 103 // Ensure that the closed connection is cleaned up asynchronously. |
| 102 void OnConnectionClosed(QuicConnectionId connection_id, | 104 void OnConnectionClosed(QuicConnectionId connection_id, |
| (...skipping 12 matching lines...) Expand all Loading... |
| 115 QuicConnectionId connection_id) override; | 117 QuicConnectionId connection_id) override; |
| 116 | 118 |
| 117 typedef base::hash_map<QuicConnectionId, QuicServerSession*> SessionMap; | 119 typedef base::hash_map<QuicConnectionId, QuicServerSession*> SessionMap; |
| 118 | 120 |
| 119 const SessionMap& session_map() const { return session_map_; } | 121 const SessionMap& session_map() const { return session_map_; } |
| 120 | 122 |
| 121 // Deletes all sessions on the closed session list and clears the list. | 123 // Deletes all sessions on the closed session list and clears the list. |
| 122 void DeleteSessions(); | 124 void DeleteSessions(); |
| 123 | 125 |
| 124 protected: | 126 protected: |
| 125 // Instantiates a new low-level packet writer. Caller takes ownership of the | |
| 126 // returned object. | |
| 127 virtual QuicPacketWriter* CreateWriter(int fd); | |
| 128 | |
| 129 virtual QuicServerSession* CreateQuicSession( | 127 virtual QuicServerSession* CreateQuicSession( |
| 130 QuicConnectionId connection_id, | 128 QuicConnectionId connection_id, |
| 131 const IPEndPoint& server_address, | 129 const IPEndPoint& server_address, |
| 132 const IPEndPoint& client_address); | 130 const IPEndPoint& client_address); |
| 133 | 131 |
| 134 // Called by |framer_visitor_| when the public header has been parsed. | 132 // Called by |framer_visitor_| when the public header has been parsed. |
| 135 virtual bool OnUnauthenticatedPublicHeader( | 133 virtual bool OnUnauthenticatedPublicHeader( |
| 136 const QuicPacketPublicHeader& header); | 134 const QuicPacketPublicHeader& header); |
| 137 | 135 |
| 138 // Called by OnUnauthenticatedPublicHeader when the packet is not for a | 136 // Called by OnUnauthenticatedPublicHeader when the packet is not for a |
| (...skipping 116 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 255 QuicFramer framer_; | 253 QuicFramer framer_; |
| 256 scoped_ptr<QuicFramerVisitor> framer_visitor_; | 254 scoped_ptr<QuicFramerVisitor> framer_visitor_; |
| 257 | 255 |
| 258 DISALLOW_COPY_AND_ASSIGN(QuicDispatcher); | 256 DISALLOW_COPY_AND_ASSIGN(QuicDispatcher); |
| 259 }; | 257 }; |
| 260 | 258 |
| 261 } // namespace tools | 259 } // namespace tools |
| 262 } // namespace net | 260 } // namespace net |
| 263 | 261 |
| 264 #endif // NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ | 262 #endif // NET_TOOLS_QUIC_QUIC_DISPATCHER_H_ |
| OLD | NEW |