Landing Recent QUIC changes until Sun Sep 4 03:41:00

net/tools/quic/stateless_rejector.h

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef NET_QUIC_STATELESS_REJECTOR_H_
 #define NET_QUIC_STATELESS_REJECTOR_H_
 
 #include "base/strings/string_piece.h"
 #include "net/quic/core/crypto/crypto_framer.h"
 #include "net/quic/core/crypto/quic_crypto_server_config.h"
 #include "net/quic/core/quic_protocol.h"
 
 namespace net {
 
 // The StatelessRejector receives CHLO messages and generates an SREJ
 // message in response, if the CHLO can be statelessly rejected.
 class StatelessRejector {
  public:
   enum State {
+    UNKNOWN,      // State has not yet been determined
     UNSUPPORTED,  // Stateless rejects are not supported
     FAILED,       // There was an error processing the CHLO.
     ACCEPTED,     // The CHLO was accepted
     REJECTED,     // The CHLO was rejected.
   };
 
   StatelessRejector(QuicVersion version,
                     const QuicVersionVector& versions,
                     const QuicCryptoServerConfig* crypto_config,
                     QuicCompressedCertsCache* compressed_certs_cache,
                     const QuicClock* clock,
                     QuicRandom* random,
                     QuicByteCount chlo_packet_size,
                     const IPEndPoint& client_address,
                     const IPEndPoint& server_address);
 
   ~StatelessRejector();
 
-  // Called when |chlo| is received for |connection_id| to determine
-  // if it should be statelessly rejected.
+  // Called when |chlo| is received for |connection_id|.
   void OnChlo(QuicVersion version,
               QuicConnectionId connection_id,
               QuicConnectionId server_designated_connection_id,
               const CryptoHandshakeMessage& chlo);
 
+  class ProcessDoneCallback {
+   public:
+    virtual ~ProcessDoneCallback() = default;
+    virtual void Run(std::unique_ptr<StatelessRejector> rejector) = 0;
+  };
+
+  // Perform processing to determine whether the CHLO received in OnChlo should
+  // be statelessly rejected, and invoke the callback once a decision has been
+  // made.
+  static void Process(std::unique_ptr<StatelessRejector> rejector,
+                      std::unique_ptr<ProcessDoneCallback> cb);
+
   // Returns the state of the rejector after OnChlo() has been called.
   State state() const { return state_; }
 
   // Returns the error code when state() returns FAILED.
   QuicErrorCode error() const { return error_; }
 
   // Returns the error details when state() returns FAILED.
   std::string error_details() const { return error_details_; }
 
+  // Returns the connection ID.
+  QuicConnectionId connection_id() const { return connection_id_; }
+
   // Returns the SREJ message when state() returns REJECTED.
   const CryptoHandshakeMessage& reply() const { return reply_; }
 
  private:
   // Helper class which is passed in to
   // QuicCryptoServerConfig::ValidateClientHello.
   class ValidateCallback;
   friend class ValidateCallback;
 
   void ProcessClientHello(
       const CryptoHandshakeMessage& client_hello,
-      const ValidateClientHelloResultCallback::Result& result);
+      const ValidateClientHelloResultCallback::Result& result,
+      std::unique_ptr<StatelessRejector> rejector,
+      std::unique_ptr<StatelessRejector::ProcessDoneCallback> cb);
 
   State state_;
   QuicErrorCode error_;
   std::string error_details_;
   QuicVersion version_;
   QuicVersionVector versions_;
   QuicConnectionId connection_id_;
   QuicConnectionId server_designated_connection_id_;
   QuicByteCount chlo_packet_size_;
   IPEndPoint client_address_;
   IPEndPoint server_address_;
   const QuicClock* clock_;
   QuicRandom* random_;
   const QuicCryptoServerConfig* crypto_config_;
   QuicCompressedCertsCache* compressed_certs_cache_;
-  const CryptoHandshakeMessage* chlo_;
+  CryptoHandshakeMessage chlo_;
   CryptoHandshakeMessage reply_;
   CryptoFramer crypto_framer_;
   QuicCryptoProof proof_;
 
   DISALLOW_COPY_AND_ASSIGN(StatelessRejector);
 };
 
 }  // namespace net
 
 #endif  // NET_QUIC_STATELESS_REJECTOR_H_