s2n-quic-dc 0.77.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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88

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

use crate::{
    event,
    stream::{
        environment::{tokio::Environment, Peer, SetupResult, SocketSet},
        recv::shared::RecvBuffer,
        server::tokio::tcp::LazyBoundStream,
        TransportFeatures,
    },
};
use s2n_quic_core::inet::SocketAddress;
use tokio::net::TcpStream;

/// A socket that is already registered with the application runtime
pub struct Registered {
    pub socket: TcpStream,
    pub peer_addr: SocketAddress,
    pub local_port: u16,
    pub recv_buffer: RecvBuffer,
}

impl<Sub> Peer<Environment<Sub>> for Registered
where
    Sub: event::Subscriber + Clone,
{
    type ReadWorkerSocket = ();
    type WriteWorkerSocket = ();

    fn features(&self) -> TransportFeatures {
        TransportFeatures::TCP
    }

    #[inline]
    fn setup(
        self,
        _env: &Environment<Sub>,
    ) -> SetupResult<Self::ReadWorkerSocket, Self::WriteWorkerSocket> {
        let remote_addr = self.peer_addr;
        let application = Box::new(self.socket);
        let socket = SocketSet {
            application,
            read_worker: None,
            write_worker: None,
            remote_addr,
            source_queue_id: None,
        };
        Ok((socket, self.recv_buffer))
    }
}

/// A socket that should be reregistered with the application runtime
pub struct Reregistered {
    pub socket: LazyBoundStream,
    pub peer_addr: SocketAddress,
    pub local_port: u16,
    pub recv_buffer: RecvBuffer,
}

impl<Sub> Peer<Environment<Sub>> for Reregistered
where
    Sub: event::Subscriber + Clone,
{
    type ReadWorkerSocket = ();
    type WriteWorkerSocket = ();

    fn features(&self) -> TransportFeatures {
        TransportFeatures::TCP
    }

    #[inline]
    fn setup(
        self,
        _env: &Environment<Sub>,
    ) -> SetupResult<Self::ReadWorkerSocket, Self::WriteWorkerSocket> {
        let remote_addr = self.peer_addr;
        let application = Box::new(self.socket.into_std()?);
        let socket = SocketSet {
            application,
            read_worker: None,
            write_worker: None,
            remote_addr,
            source_queue_id: None,
        };
        Ok((socket, self.recv_buffer))
    }
}