aranya-util 6.0.0

Utilities needed by other Aranya crates
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

use buggy::{Bug, BugExt as _};
use bytes::{Bytes, BytesMut};
use s2n_quic::{stream, Connection};

/// Read all of a stream until it has finished.
///
/// A bit more efficient than going through the `AsyncRead`-based impl,
/// especially if there was only one chunk of data. Also avoids needing to
/// convert/handle an `io::Error`.
pub async fn read_to_end(stream: &mut stream::ReceiveStream) -> Result<Bytes, stream::Error> {
    let Some(first) = stream.receive().await? else {
        return Ok(Bytes::new());
    };
    let Some(mut more) = stream.receive().await? else {
        return Ok(first);
    };
    let mut buf = BytesMut::from(first);
    loop {
        buf.extend_from_slice(&more);
        if let Some(even_more) = stream.receive().await? {
            more = even_more;
        } else {
            break;
        }
    }
    Ok(buf.freeze())
}

/// Indicates whether the stream error is "connection closed without error".
pub fn is_close_error(err: stream::Error) -> bool {
    matches!(
        err,
        stream::Error::ConnectionError {
            error: s2n_quic::connection::Error::Closed { .. },
            ..
        },
    )
}

// Extract the chosen PSK identity from the connection context.
pub fn get_conn_identity(conn: &mut Connection) -> Result<Vec<u8>, Bug> {
    Ok(*conn
        .take_tls_context()
        .assume("connection has tls context")?
        .downcast::<Vec<u8>>()
        .ok()
        .assume("can downcast identity")?)
}