feat(s2n-quic-transport) Adds packet storage feature

quic/s2n-quic-core/src/connection/limits.rs

@@ -39,6 +39,7 @@ const MAX_KEEP_ALIVE_PERIOD_DEFAULT: Duration = Duration::from_secs(30);
 pub const ANTI_AMPLIFICATION_MULTIPLIER: u8 = 3;
 
 pub const DEFAULT_STREAM_BATCH_SIZE: u8 = 1;
+pub const DEFAULT_STORED_PACKET_SIZE: usize = 0;
 
 #[non_exhaustive]
 #[derive(Debug)]
@@ -104,6 +105,7 @@ pub struct Limits {
     pub(crate) migration_support: MigrationSupport,
     pub(crate) anti_amplification_multiplier: u8,
     pub(crate) stream_batch_size: u8,
+    pub(crate) stored_packet_size: usize,
 }
 
 impl Default for Limits {
@@ -151,6 +153,7 @@ impl Limits {
             migration_support: MigrationSupport::RECOMMENDED,
             anti_amplification_multiplier: ANTI_AMPLIFICATION_MULTIPLIER,
             stream_batch_size: DEFAULT_STREAM_BATCH_SIZE,
+            stored_packet_size: DEFAULT_STORED_PACKET_SIZE,
         }
     }
 
@@ -254,6 +257,7 @@ impl Limits {
         u64
     );
     setter!(with_stream_batch_size, stream_batch_size, u8);
+    setter!(with_stored_packet_size, stored_packet_size, usize);
     setter!(with_ack_elicitation_interval, ack_elicitation_interval, u8);
     setter!(with_max_ack_ranges, ack_ranges_limit, u8);
     setter!(
@@ -422,6 +426,12 @@ impl Limits {
     pub fn stream_batch_size(&self) -> u8 {
         self.stream_batch_size
     }
+
+    #[doc(hidden)]
+    #[inline]
+    pub fn stored_packet_size(&self) -> usize {
+        self.stored_packet_size
+    }
 }
 
 #[must_use]

quic/s2n-quic-core/src/packet/handshake.rs

@@ -22,6 +22,8 @@ use crate::{
     transport,
     varint::VarInt,
 };
+#[cfg(feature = "alloc")]
+use alloc::vec::Vec;
 use s2n_codec::{CheckedRange, DecoderBufferMut, DecoderBufferMutResult, Encoder, EncoderValue};
 
 //= https://www.rfc-editor.org/rfc/rfc9000#section-17.2.4
@@ -65,6 +67,11 @@ pub type EncryptedHandshake<'a> =
 pub type CleartextHandshake<'a> = Handshake<&'a [u8], &'a [u8], PacketNumber, DecoderBufferMut<'a>>;
 
 impl<'a> ProtectedHandshake<'a> {
+    #[cfg(feature = "alloc")]
+    pub fn get_wire_bytes(&self) -> Vec<u8> {
+        self.payload.buffer.encode_to_vec()
+    }
+
     #[inline]
     pub(crate) fn decode(
         _tag: Tag,

quic/s2n-quic-core/src/packet/short.rs

@@ -16,6 +16,8 @@ use crate::{
     },
     transport,
 };
+#[cfg(feature = "alloc")]
+use alloc::vec::Vec;
 use s2n_codec::{CheckedRange, DecoderBufferMut, DecoderBufferMutResult, Encoder, EncoderValue};
 
 //= https://www.rfc-editor.org/rfc/rfc9000#section-17.3.1
@@ -194,6 +196,10 @@ impl<'a> ProtectedShort<'a> {
             .get_checked_range(&self.destination_connection_id)
             .into_less_safe_slice()
     }
+    #[cfg(feature = "alloc")]
+    pub fn get_wire_bytes(&self) -> Vec<u8> {
+        self.payload.buffer.encode_to_vec()
+    }
 }
 
 impl<'a> EncryptedShort<'a> {

quic/s2n-quic-transport/src/connection/connection_container/tests.rs

@@ -138,6 +138,9 @@ impl connection::Trait for TestConnection {
         _datagram: &mut <Self::Config as endpoint::Config>::DatagramEndpoint,
         _dc_endpoint: &mut <Self::Config as endpoint::Config>::DcEndpoint,
         _conn_limits_endpoint: &mut <Self::Config as endpoint::Config>::ConnectionLimits,
+        _random_generator: &mut <Self::Config as endpoint::Config>::RandomGenerator,
+        _packet_interceptor: &mut <Self::Config as endpoint::Config>::PacketInterceptor,
+        _connection_id_format: &mut <Self::Config as endpoint::Config>::ConnectionIdFormat,
     ) -> Result<(), connection::Error> {
         Ok(())
     }
@@ -153,6 +156,7 @@ impl connection::Trait for TestConnection {
         _datagram_endpoint: &mut <Self::Config as endpoint::Config>::DatagramEndpoint,
         _dc_endpoint: &mut <Self::Config as endpoint::Config>::DcEndpoint,
         _conn_limits_endpoint: &mut <Self::Config as endpoint::Config>::ConnectionLimits,
+        _connection_id_format: &<Self::Config as endpoint::Config>::ConnectionIdFormat,
     ) -> Result<(), ProcessingError> {
         Ok(())
     }
@@ -169,6 +173,7 @@ impl connection::Trait for TestConnection {
         _datagram_endpoint: &mut <Self::Config as endpoint::Config>::DatagramEndpoint,
         _dc_endpoint: &mut <Self::Config as endpoint::Config>::DcEndpoint,
         _conn_limits_endpoint: &mut <Self::Config as endpoint::Config>::ConnectionLimits,
+        _connection_id_format: &<Self::Config as endpoint::Config>::ConnectionIdFormat,
     ) -> Result<(), ProcessingError> {
         Ok(())
     }
@@ -185,6 +190,7 @@ impl connection::Trait for TestConnection {
         _datagram_endpoint: &mut <Self::Config as endpoint::Config>::DatagramEndpoint,
         _dc_endpoint: &mut <Self::Config as endpoint::Config>::DcEndpoint,
         _connection_limits_endpoint: &mut <Self::Config as endpoint::Config>::ConnectionLimits,
+        _connection_id_format: &<Self::Config as endpoint::Config>::ConnectionIdFormat,
     ) -> Result<(), ProcessingError> {
         Ok(())
     }