sark-h2 0.28.2

Fiber-free HTTP/2 protocol primitives and manifold server for sark
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
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109

use core::mem;

use sark_http::field::metadata::section::{rules, storage};

use crate::{conn, conn::ingress::headers::Active, hpack::decode};

pub(super) enum Mode<P: rules::Policy> {
    Idle(decode::Decoder, storage::Scratch),
    Active(Active<P>),
    Failed,
}

pub(super) struct State<P: rules::Policy> {
    mode: Mode<P>,
}

impl<P: rules::Policy> State<P> {
    pub(super) fn new(decoder: decode::Decoder) -> Self {
        Self {
            mode: Mode::Idle(decoder, storage::Scratch::default()),
        }
    }

    pub(super) fn active(&self) -> Option<&Active<P>> {
        match &self.mode {
            Mode::Active(active) => Some(active),
            Mode::Idle(_, _) | Mode::Failed => None,
        }
    }

    pub(super) fn try_active(&self) -> Result<&Active<P>, conn::Error> {
        match &self.mode {
            Mode::Active(active) => Ok(active),
            Mode::Idle(_, _) => Err(conn::Error::HeaderState(conn::HeaderState {
                expected: conn::HeaderPhase::Active,
                actual: conn::HeaderPhase::Idle,
            })),
            Mode::Failed => Err(conn::Error::HeaderState(conn::HeaderState {
                expected: conn::HeaderPhase::Active,
                actual: conn::HeaderPhase::Failed,
            })),
        }
    }

    pub(super) fn try_mut(&mut self) -> Result<&mut Active<P>, conn::Error> {
        match &mut self.mode {
            Mode::Active(active) => Ok(active),
            Mode::Idle(_, _) => Err(conn::Error::HeaderState(conn::HeaderState {
                expected: conn::HeaderPhase::Active,
                actual: conn::HeaderPhase::Idle,
            })),
            Mode::Failed => Err(conn::Error::HeaderState(conn::HeaderState {
                expected: conn::HeaderPhase::Active,
                actual: conn::HeaderPhase::Failed,
            })),
        }
    }

    /// Both maps leave self.mode at Failed while change owns the extracted payload.
    pub(super) fn map_idle<R>(
        &mut self,
        change: impl FnOnce(decode::Decoder, storage::Scratch) -> (Mode<P>, Result<R, conn::Error>),
    ) -> Result<R, conn::Error> {
        let (mode, result) = match mem::replace(&mut self.mode, Mode::Failed) {
            Mode::Idle(decoder, scratch) => change(decoder, scratch),
            Mode::Active(active) => (
                Mode::Active(active),
                Err(conn::Error::HeaderState(conn::HeaderState {
                    expected: conn::HeaderPhase::Idle,
                    actual: conn::HeaderPhase::Active,
                })),
            ),
            Mode::Failed => (
                Mode::Failed,
                Err(conn::Error::HeaderState(conn::HeaderState {
                    expected: conn::HeaderPhase::Idle,
                    actual: conn::HeaderPhase::Failed,
                })),
            ),
        };
        mem::forget(mem::replace(&mut self.mode, mode));
        result
    }

    pub(super) fn map_active<R>(
        &mut self,
        change: impl FnOnce(Active<P>) -> (Mode<P>, Result<R, conn::Error>),
    ) -> Result<R, conn::Error> {
        let (mode, result) = match mem::replace(&mut self.mode, Mode::Failed) {
            Mode::Active(active) => change(active),
            Mode::Idle(decoder, scratch) => (
                Mode::Idle(decoder, scratch),
                Err(conn::Error::HeaderState(conn::HeaderState {
                    expected: conn::HeaderPhase::Active,
                    actual: conn::HeaderPhase::Idle,
                })),
            ),
            Mode::Failed => (
                Mode::Failed,
                Err(conn::Error::HeaderState(conn::HeaderState {
                    expected: conn::HeaderPhase::Active,
                    actual: conn::HeaderPhase::Failed,
                })),
            ),
        };
        mem::forget(mem::replace(&mut self.mode, mode));
        result
    }
}