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

net/quic/core/quic_server_session_base.h

 // Copyright (c) 2012 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.
 //
 // A server specific QuicSession subclass.
 
 #ifndef NET_QUIC_QUIC_SERVER_SESSION_BASE_H_
 #define NET_QUIC_QUIC_SERVER_SESSION_BASE_H_
 
 #include <stdint.h>
@@ skipping 36 matching lines @@
     virtual void OnConnectionClosed(QuicConnectionId connection_id,
                                     QuicErrorCode error,
                                     const std::string& error_details) = 0;
 
     // Called when the session has become write blocked.
     virtual void OnWriteBlocked(QuicBlockedWriterInterface* blocked_writer) = 0;
 
     // Called after the given connection is added to the time-wait std::list.
     virtual void OnConnectionAddedToTimeWaitList(
         QuicConnectionId connection_id) = 0;
-  };
 
-  // Provides helper functions for the session.
-  class Helper {
-   public:
-    virtual ~Helper() {}
-
-    // Given the current connection_id, generates a new ConnectionId to
-    // be returned with a stateless reject.
-    virtual QuicConnectionId GenerateConnectionIdForReject(
-        QuicConnectionId connection_id) const = 0;
-
-    // Returns true if |message|, which was received on |self_address| is
-    // acceptable according to the visitor's policy. Otherwise, returns false
-    // and populates |error_details|.
-    virtual bool CanAcceptClientHello(const CryptoHandshakeMessage& message,
-                                      const IPEndPoint& self_address,
-                                      std::string* error_details) const = 0;
+    // Called before a packet is going to be processed by |session|.
+    virtual void OnPacketBeingDispatchedToSession(
+        QuicServerSessionBase* session) = 0;
   };
 
   // Does not take ownership of |connection|. |crypto_config| must outlive the
-  // session.
+  // session. |helper| must outlive any created crypto streams.
   QuicServerSessionBase(const QuicConfig& config,
                         QuicConnection* connection,
                         Visitor* visitor,
-                        Helper* helper,
+                        QuicCryptoServerStream::Helper* helper,
                         const QuicCryptoServerConfig* crypto_config,
                         QuicCompressedCertsCache* compressed_certs_cache);
 
   // Override the base class to notify the owner of the connection close.
   void OnConnectionClosed(QuicErrorCode error,
                           const std::string& error_details,
                           ConnectionCloseSource source) override;
   void OnWriteBlocked() override;
 
   // Sends a server config update to the client, containing new bandwidth
@@ skipping 11 matching lines @@
   // Override base class to process bandwidth related config received from
   // client.
   void OnConfigNegotiated() override;
 
   void set_serving_region(const std::string& serving_region) {
     serving_region_ = serving_region;
   }
 
   bool server_push_enabled() const { return server_push_enabled_; }
 
-  // Delegates to the helper's GenerateConnectionIdForReject method.
-  QuicConnectionId GenerateConnectionIdForReject(
-      QuicConnectionId connection_id);
-
-  // Delegates to the helper's CanAcceptClientHello method.
-  bool CanAcceptClientHello(const CryptoHandshakeMessage& message,
-                            std::string* error_details);
-
  protected:
   // QuicSession methods(override them with return type of QuicSpdyStream*):
   QuicCryptoServerStreamBase* GetCryptoStream() override;
 
   // If an outgoing stream can be created, return true.
   // Return false when connection is closed or forward secure encryption hasn't
   // established yet or number of server initiated streams already reaches the
   // upper limit.
   bool ShouldCreateOutgoingDynamicStream() override;
 
   // If we should create an incoming stream, returns true. Otherwise
   // does error handling, including communicating the error to the client and
   // possibly closing the connection, and returns false.
   bool ShouldCreateIncomingDynamicStream(QuicStreamId id) override;
 
   virtual QuicCryptoServerStreamBase* CreateQuicCryptoServerStream(
       const QuicCryptoServerConfig* crypto_config,
       QuicCompressedCertsCache* compressed_certs_cache) = 0;
 
   const QuicCryptoServerConfig* crypto_config() { return crypto_config_; }
 
   void set_server_push_enabled(bool enable) { server_push_enabled_ = enable; }
 
+  Visitor* visitor() { return visitor_; }
+
+  QuicCryptoServerStream::Helper* stream_helper() { return helper_; }
+
  private:
   friend class test::QuicServerSessionBasePeer;
   friend class test::QuicSimpleServerSessionPeer;
 
   const QuicCryptoServerConfig* crypto_config_;
 
   // The cache which contains most recently compressed certs.
   // Owned by QuicDispatcher.
   QuicCompressedCertsCache* compressed_certs_cache_;
 
   std::unique_ptr<QuicCryptoServerStreamBase> crypto_stream_;
   Visitor* visitor_;
-  Helper* helper_;
+
+  // Pointer to the helper used to create crypto server streams. Must outlive
+  // streams created via CreateQuicCryptoServerStream.
+  QuicCryptoServerStream::Helper* helper_;
 
   // Whether bandwidth resumption is enabled for this connection.
   bool bandwidth_resumption_enabled_;
 
   // The most recent bandwidth estimate sent to the client.
   QuicBandwidth bandwidth_estimate_sent_to_client_;
 
   // Text describing server location. Sent to the client as part of the bandwith
   // estimate in the source-address token. Optional, can be left empty.
   std::string serving_region_;
@@ skipping 13 matching lines @@
   // Set during handshake. If true, resources in x-associated-content and link
   // headers will be pushed. see: go/gfe_server_push.
   bool server_push_enabled_;
 
   DISALLOW_COPY_AND_ASSIGN(QuicServerSessionBase);
 };
 
 }  // namespace net
 
 #endif  // NET_QUIC_QUIC_SERVER_SESSION_BASE_H_