| OLD | NEW |
| (Empty) |
| 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 | |
| 3 // found in the LICENSE file. | |
| 4 // | |
| 5 // A toy server, which listens on a specified address for QUIC traffic and | |
| 6 // handles incoming responses. | |
| 7 | |
| 8 #ifndef NET_QUIC_QUIC_SERVER_H_ | |
| 9 #define NET_QUIC_QUIC_SERVER_H_ | |
| 10 | |
| 11 #include "base/basictypes.h" | |
| 12 #include "base/memory/scoped_ptr.h" | |
| 13 #include "net/base/io_buffer.h" | |
| 14 #include "net/base/ip_endpoint.h" | |
| 15 #include "net/base/net_log.h" | |
| 16 #include "net/quic/crypto/quic_crypto_server_config.h" | |
| 17 #include "net/quic/quic_clock.h" | |
| 18 #include "net/quic/quic_config.h" | |
| 19 #include "net/quic/quic_connection_helper.h" | |
| 20 | |
| 21 namespace net { | |
| 22 | |
| 23 | |
| 24 namespace test { | |
| 25 class QuicServerPeer; | |
| 26 } // namespace test | |
| 27 | |
| 28 class QuicDispatcher; | |
| 29 class UDPServerSocket; | |
| 30 | |
| 31 class QuicServer { | |
| 32 public: | |
| 33 QuicServer(const QuicConfig& config, | |
| 34 const QuicVersionVector& supported_versions); | |
| 35 | |
| 36 virtual ~QuicServer(); | |
| 37 | |
| 38 // Start listening on the specified address. Returns an error code. | |
| 39 int Listen(const IPEndPoint& address); | |
| 40 | |
| 41 // Server deletion is imminent. Start cleaning up. | |
| 42 void Shutdown(); | |
| 43 | |
| 44 // Start reading on the socket. On asynchronous reads, this registers | |
| 45 // OnReadComplete as the callback, which will then call StartReading again. | |
| 46 void StartReading(); | |
| 47 | |
| 48 // Called on reads that complete asynchronously. Dispatches the packet and | |
| 49 // continues the read loop. | |
| 50 void OnReadComplete(int result); | |
| 51 | |
| 52 void SetStrikeRegisterNoStartupPeriod() { | |
| 53 crypto_config_.set_strike_register_no_startup_period(); | |
| 54 } | |
| 55 | |
| 56 // SetProofSource sets the ProofSource that will be used to verify the | |
| 57 // server's certificate, and takes ownership of |source|. | |
| 58 void SetProofSource(ProofSource* source) { | |
| 59 crypto_config_.SetProofSource(source); | |
| 60 } | |
| 61 | |
| 62 private: | |
| 63 friend class net::test::QuicServerPeer; | |
| 64 | |
| 65 // Initialize the internal state of the server. | |
| 66 void Initialize(); | |
| 67 | |
| 68 // Accepts data from the framer and demuxes clients to sessions. | |
| 69 scoped_ptr<QuicDispatcher> dispatcher_; | |
| 70 | |
| 71 // Used by the helper_ to time alarms. | |
| 72 QuicClock clock_; | |
| 73 | |
| 74 // Used to manage the message loop. | |
| 75 QuicConnectionHelper helper_; | |
| 76 | |
| 77 // Listening socket. Also used for outbound client communication. | |
| 78 scoped_ptr<UDPServerSocket> socket_; | |
| 79 | |
| 80 // config_ contains non-crypto parameters that are negotiated in the crypto | |
| 81 // handshake. | |
| 82 QuicConfig config_; | |
| 83 // crypto_config_ contains crypto parameters for the handshake. | |
| 84 QuicCryptoServerConfig crypto_config_; | |
| 85 | |
| 86 // This vector contains QUIC versions which we currently support. | |
| 87 // This should be ordered such that the highest supported version is the first | |
| 88 // element, with subsequent elements in descending order (versions can be | |
| 89 // skipped as necessary). | |
| 90 QuicVersionVector supported_versions_; | |
| 91 | |
| 92 // The address that the server listens on. | |
| 93 IPEndPoint server_address_; | |
| 94 | |
| 95 // Keeps track of whether a read is currently in flight, after which | |
| 96 // OnReadComplete will be called. | |
| 97 bool read_pending_; | |
| 98 | |
| 99 // The number of iterations of the read loop that have completed synchronously | |
| 100 // and without posting a new task to the message loop. | |
| 101 int synchronous_read_count_; | |
| 102 | |
| 103 // The target buffer of the current read. | |
| 104 scoped_refptr<IOBufferWithSize> read_buffer_; | |
| 105 | |
| 106 // The source address of the current read. | |
| 107 IPEndPoint client_address_; | |
| 108 | |
| 109 // The log to use for the socket. | |
| 110 NetLog net_log_; | |
| 111 | |
| 112 base::WeakPtrFactory<QuicServer> weak_factory_; | |
| 113 | |
| 114 DISALLOW_COPY_AND_ASSIGN(QuicServer); | |
| 115 }; | |
| 116 | |
| 117 } // namespace net | |
| 118 | |
| 119 #endif // NET_QUIC_QUIC_SERVER_H_ | |
| OLD | NEW |