sark-h2 0.12.0

HTTP/2 protocol primitives + client/server bootstrap for sark stack
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

use o3::buffer::{Bytes, Retained};

use crate::conn::{Conn, ConnError};
use crate::role::ServerRole;

pub trait Transport {
    fn connection(&mut self) -> &mut Conn<ServerRole>;

    fn drain_events(&mut self) -> usize;
}

pub struct Driver<'a, T> {
    transport: &'a mut T,
}

impl<'a, T> Driver<'a, T>
where
    T: Transport,
{
    pub fn new(transport: &'a mut T) -> Self {
        Self { transport }
    }

    pub fn ingest(&mut self, bytes: &[u8]) -> Result<(), ConnError> {
        if self.transport.connection().goaway_sent()
            || self.transport.connection().goaway_received().is_some()
        {
            return Ok(());
        }

        let result = self.transport.connection().ingest(bytes);
        self.drain(result)
    }

    pub fn ingest_retained(&mut self, bytes: Bytes<Retained>) -> Result<(), ConnError> {
        if self.transport.connection().goaway_sent()
            || self.transport.connection().goaway_received().is_some()
        {
            return Ok(());
        }

        let result = self.transport.connection().ingest_retained(bytes);
        self.drain(result)
    }

    fn drain(&mut self, mut result: Result<(), ConnError>) -> Result<(), ConnError> {
        loop {
            let drained = self.transport.drain_events();
            match result {
                Ok(()) => return Ok(()),
                Err(ConnError::Overload) if drained != 0 => {
                    result = self.transport.connection().resume();
                }
                Err(error) => return Err(error),
            }
        }
    }
}