s2n-quic-dc 0.85.0

Internal crate used by s2n-quic
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55

// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

use crate::{packet, stream::recv};
use s2n_codec::DecoderBufferMut;
use s2n_quic_core::inet::{ExplicitCongestionNotification, SocketAddress};

pub trait Dispatch {
    #[inline(always)]
    fn tag_len(&self) -> usize {
        16
    }

    fn on_packet(
        &mut self,
        remote_addr: &SocketAddress,
        ecn: ExplicitCongestionNotification,
        packet: packet::Packet,
    ) -> Result<(), recv::Error>;

    #[inline]
    fn on_datagram_segment(
        &mut self,
        remote_addr: &SocketAddress,
        ecn: ExplicitCongestionNotification,
        segment: &mut [u8],
    ) -> Result<(), recv::Error> {
        let tag_len = self.tag_len();
        let segment_len = segment.len();
        let mut decoder = DecoderBufferMut::new(segment);

        while !decoder.is_empty() {
            let packet = match decoder.decode_parameterized(tag_len) {
                Ok((packet, remaining)) => {
                    decoder = remaining;
                    packet
                }
                Err(decoder_error) => {
                    // the packet was likely corrupted so log it and move on to the
                    // next segment
                    tracing::warn!(
                        %decoder_error,
                        segment_len
                    );

                    break;
                }
            };

            self.on_packet(remote_addr, ecn, packet)?;
        }

        Ok(())
    }
}