net/tools/quic/quic_simple_server.h

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