Add a knob (QuicSentPacketManager::ReduceMemoryUsage) to reduce the memory used by unacked_packet_map and bandwidth_sampler owned by QuicSentPacketManager.

In follow-up, this knob can be used if the session is idled long enough.

PiperOrigin-RevId: 875952422

Author: wang178c

quiche/quic/core/congestion_control/bandwidth_sampler.h

@@ -447,6 +447,8 @@ class QUICHE_EXPORT BandwidthSampler : public BandwidthSamplerInterface {
     return overestimate_avoidance_;
   }
 
+  void ReduceMemoryUsage() { connection_state_map_.shrink_to_fit(); }
+
  private:
   friend class test::BandwidthSamplerPeer;
 

quiche/quic/core/congestion_control/bbr2_misc.h

@@ -546,6 +546,8 @@ class QUICHE_EXPORT Bbr2NetworkModel {
     return rounds_with_queueing_;
   }
 
+  void ReduceMemoryUsage() { bandwidth_sampler_.ReduceMemoryUsage(); }
+
  private:
   // Called when a new round trip starts.
   void OnNewRound();

quiche/quic/core/congestion_control/bbr2_sender.h

@@ -100,6 +100,7 @@ class QUICHE_EXPORT Bbr2Sender final : public SendAlgorithmInterface {
 
   bool EnableECT0() override { return false; }
   bool EnableECT1() override { return false; }
+  void ReduceMemoryUsage() override { model_.ReduceMemoryUsage(); }
   // End implementation of SendAlgorithmInterface.
 
   const Bbr2Params& Params() const { return params_; }

quiche/quic/core/congestion_control/bbr_sender.h

@@ -136,6 +136,7 @@ class QUICHE_EXPORT BbrSender : public SendAlgorithmInterface {
   void PopulateConnectionStats(QuicConnectionStats* stats) const override;
   bool EnableECT0() override { return false; }
   bool EnableECT1() override { return false; }
+  void ReduceMemoryUsage() override { sampler_.ReduceMemoryUsage(); }
   // End implementation of SendAlgorithmInterface.
 
   // Gets the number of RTTs BBR remains in STARTUP phase.

quiche/quic/core/congestion_control/send_algorithm_interface.h

@@ -194,6 +194,9 @@ class QUICHE_EXPORT SendAlgorithmInterface {
   // ECN mode is enabled, it is an error to call either of these methods.
   virtual bool EnableECT0() = 0;
   virtual bool EnableECT1() = 0;
+
+  // Called to reduce the memory usage of the send algorithm.
+  virtual void ReduceMemoryUsage() = 0;
 };
 
 }  // namespace quic

quiche/quic/core/congestion_control/tcp_cubic_sender_bytes.h

@@ -78,6 +78,7 @@ class QUICHE_EXPORT TcpCubicSenderBytes : public SendAlgorithmInterface {
   void PopulateConnectionStats(QuicConnectionStats* /*stats*/) const override {}
   bool EnableECT0() override { return false; }
   bool EnableECT1() override { return false; }
+  void ReduceMemoryUsage() override {}
   // End implementation of SendAlgorithmInterface.
 
   QuicByteCount min_congestion_window() const { return min_congestion_window_; }

quiche/quic/core/packet_number_indexed_queue.h

@@ -96,6 +96,9 @@ class QUICHE_NO_EXPORT PacketNumberIndexedQueue {
   // Reserves the specified memory capacity in the underlying deque.
   void Reserve(size_t capacity) { entries_.reserve(capacity); }
 
+  // Reduces the capacity of the underlying deque.
+  void shrink_to_fit() { entries_.shrink_to_fit(); }
+
  private:
   // Wrapper around T used to mark whether the entry is actually in the map.
   struct QUICHE_NO_EXPORT EntryWrapper : T {

quiche/quic/core/quic_sent_packet_manager.h

@@ -417,6 +417,11 @@ class QUICHE_EXPORT QuicSentPacketManager {
     return send_algorithm_.get();
   }
 
+  void ReduceMemoryUsage() {
+    unacked_packets_.ReduceMemoryUsage();
+    send_algorithm_->ReduceMemoryUsage();
+  }
+
   // Wrapper for SendAlgorithmInterface functions, since these functions are
   // not const.
   bool EnableECT0() {

quiche/quic/core/quic_unacked_packet_map.h

@@ -269,6 +269,8 @@ class QUICHE_EXPORT QuicUnackedPacketMap {
         ", packets_in_flight: ", packets_in_flight_, "}");
   }
 
+  void ReduceMemoryUsage() { unacked_packets_.shrink_to_fit(); }
+
  private:
   friend class test::QuicUnackedPacketMapPeer;
 

quiche/quic/test_tools/quic_test_utils.h

@@ -1255,6 +1255,7 @@ class MockSendAlgorithm : public SendAlgorithmInterface {
               (const, override));
   MOCK_METHOD(bool, EnableECT0, (), (override));
   MOCK_METHOD(bool, EnableECT1, (), (override));
+  MOCK_METHOD(void, ReduceMemoryUsage, (), (override));
 };
 
 class MockLossAlgorithm : public LossDetectionInterface {