Landing Recent QUIC changes until 5:55 PM, Aug 25, 2016 UTC-4

net/quic/core/quic_buffered_packet_store.h

 // Copyright (c) 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_QUIC_BUFFERED_PACKET_STORE_H_
 #define NET_QUIC_QUIC_BUFFERED_PACKET_STORE_H_
 
 #include "net/base/ip_address.h"
 #include "net/base/linked_hash_map.h"
 #include "net/quic/core/quic_alarm.h"
 #include "net/quic/core/quic_alarm_factory.h"
 #include "net/quic/core/quic_clock.h"
 #include "net/quic/core/quic_protocol.h"
 #include "net/quic/core/quic_time.h"
 
 namespace net {
 
 namespace test {
 class QuicBufferedPacketStorePeer;
 }  // namespace test
 
-// This class buffers undeliverable packets for each connection until either
-// 1) They are requested to be delivered via DeliverPacket(), or
+// This class buffers packets for each connection until either
+// 1) They are requested to be delivered via
+//    DeliverPacket()/DeliverPacketsForNextConnection(), or
 // 2) They expire after exceeding their lifetime in the store.
+//
+// It can only buffer packets on certain number of connections. It has two pools
+// of connections: connections with CHLO buffered and those without CHLO. The
+// latter has its own upper limit along with the max number of connections this
+// store can hold. The former pool can grow till this store is full.
 class NET_EXPORT_PRIVATE QuicBufferedPacketStore {
  public:
   enum EnqueuePacketResult {
     SUCCESS = 0,
     TOO_MANY_PACKETS,  // Too many packets stored up for a certain connection.
     TOO_MANY_CONNECTIONS  // Too many connections stored up in the store.
   };
 
   // A packets with client/server address.
   struct NET_EXPORT_PRIVATE BufferedPacket {
@@ skipping 43 matching lines @@
   QuicBufferedPacketStore(const QuicBufferedPacketStore&) = delete;
 
   ~QuicBufferedPacketStore();
 
   QuicBufferedPacketStore& operator=(const QuicBufferedPacketStore&) = delete;
 
   // Adds a copy of packet into packet queue for given connection.
   EnqueuePacketResult EnqueuePacket(QuicConnectionId connection_id,
                                     const QuicReceivedPacket& packet,
                                     IPEndPoint server_address,
-                                    IPEndPoint client_address);
+                                    IPEndPoint client_address,
+                                    bool is_chlo);
 
   // Returns true if there are any packets buffered for |connection_id|.
   bool HasBufferedPackets(QuicConnectionId connection_id) const;
 
   // Returns the list of buffered packets for |connection_id| and removes them
   // from the store. Returns an empty list if no early arrived packets for this
   // connection are present.
   std::list<BufferedPacket> DeliverPackets(QuicConnectionId connection_id);
 
   // Examines how long packets have been buffered in the store for each
   // connection. If they stay too long, removes them for new coming packets and
   // calls |visitor_|'s OnPotentialConnectionExpire().
   // Resets the alarm at the end.
   void OnExpirationTimeout();
 
+  // Delivers buffered packets for next connection with CHLO to open.
+  // Return connection id for next connection in |connection_id|
+  // and all buffered packets including CHLO.
+  // The returned std::list should at least has one packet(CHLO) if
+  // store does have any connection to open. If no connection in the store has
+  // received CHLO yet, empty std::list will be returned.
+  std::list<BufferedPacket> DeliverPacketsForNextConnection(
+      QuicConnectionId* connection_id);
+
+  // Is given connection already buffered in the store?
+  bool HasChloForConnection(QuicConnectionId connection_id);
+
+  // Is there any CHLO buffered in the store?
+  bool HasChlosBuffered() const;
+
  private:
   friend class test::QuicBufferedPacketStorePeer;
 
   // Set expiration alarm if it hasn't been set.
   void MaybeSetExpirationAlarm();
 
-  // Return true if number of connections in the store reaches maximum.
-  bool IsFull();
+  // Return true if add an extra packet will go beyond allowed max connection
+  // limit. The limit for non-CHLO packet and CHLO packet is different.
+  bool ShouldBufferPacket(bool is_chlo);
 
   // A map to store packet queues with creation time for each connection.
   BufferedPacketMap undecryptable_packets_;
 
   // The max time the packets of a connection can be buffer in the store.
   QuicTime::Delta connection_life_span_;
 
   VisitorInterface* visitor_;  // Unowned.
 
   const QuicClock* clock_;  // Unowned.
 
   // This alarm fires every |connection_life_span_| to clean up
   // packets staying in the store for too long.
   std::unique_ptr<QuicAlarm> expiration_alarm_;
+
+  // Keeps track of connection with CHLO buffered up already and the order they
+  // arrive.
+  linked_hash_map<QuicConnectionId, bool> connections_with_chlo_;
 };
 
 }  // namespace net
 
 #endif  // NET_QUIC_QUIC_BUFFERED_PACKET_STORE_H_