sark_core/http/codec/decode/
headers.rs

1use std::ops::ControlFlow;
2
3use crate::error::{Error, Result};
4use crate::http::codec::Header;
5use crate::http::codec::header_utils::CsvScanMode;
6
7#[derive(Clone, Copy)]
8pub(super) struct TransferEncodingInfo {
9    pub(super) has_transfer_encoding: bool,
10    pub(super) is_chunked: bool,
11}
12
13#[derive(Clone, Copy, Debug, Default)]
14pub struct HeaderScan {
15    pub content_length: Option<usize>,
16    pub has_transfer_encoding: bool,
17    pub is_chunked_transfer: bool,
18    pub has_expect: bool,
19    pub expect_continue: bool,
20    pub duplicate_content_length: bool,
21    pub accept_encoding_gzip: bool,
22}
23
24impl crate::http::codec::Parse {
25    pub(super) fn parse_transfer_encoding_value(value: &[u8]) -> Result<TransferEncodingInfo> {
26        let mut saw_any = false;
27        let mut saw_chunked = false;
28        let mut last_is_chunked = false;
29        let mut invalid_encoding = false;
30        let mut invalid_ordering = false;
31
32        Header::scan_csv_tokens(
33            value,
34            CsvScanMode::Strict,
35            |token: &[u8]| -> ControlFlow<()> {
36                if !token.is_ascii() {
37                    invalid_encoding = true;
38                    return ControlFlow::Break(());
39                }
40                saw_any = true;
41                let is_chunked = token.eq_ignore_ascii_case(b"chunked");
42                if saw_chunked && !last_is_chunked {
43                    invalid_ordering = true;
44                    return ControlFlow::Break(());
45                }
46                if saw_chunked && !is_chunked {
47                    invalid_ordering = true;
48                    return ControlFlow::Break(());
49                }
50                if is_chunked {
51                    saw_chunked = true;
52                }
53                last_is_chunked = is_chunked;
54                ControlFlow::Continue(())
55            },
56        )
57        .and_then(|_| {
58            if invalid_encoding {
59                return Err(Error::BadRequest("Invalid Transfer-Encoding".into()));
60            }
61            if invalid_ordering {
62                return Err(Error::BadRequest(
63                    "Invalid Transfer-Encoding ordering".into(),
64                ));
65            }
66            if saw_any {
67                Ok(())
68            } else {
69                Err(Error::BadRequest("Invalid Transfer-Encoding".into()))
70            }
71        })?;
72        if saw_chunked && !last_is_chunked {
73            return Err(Error::BadRequest(
74                "Invalid Transfer-Encoding ordering".into(),
75            ));
76        }
77
78        Ok(TransferEncodingInfo {
79            has_transfer_encoding: saw_any,
80            is_chunked: saw_chunked && last_is_chunked,
81        })
82    }
83
84    pub fn header_scan(headers: &[httparse::Header<'_>]) -> Result<HeaderScan> {
85        let mut content_length: Option<usize> = None;
86        let mut saw_content_length = false;
87        let mut has_transfer_encoding = false;
88        let mut is_chunked_transfer = false;
89        let mut has_expect = false;
90        let mut expect_continue = false;
91        let mut duplicate_content_length = false;
92
93        for h in headers.iter().filter(|h| !h.name.is_empty()) {
94            if h.name.eq_ignore_ascii_case("content-length") {
95                if saw_content_length {
96                    duplicate_content_length = true;
97                }
98                saw_content_length = true;
99                let len = Header::content_length(h.value)?;
100                content_length = Some(len);
101                continue;
102            }
103
104            if h.name.eq_ignore_ascii_case("transfer-encoding") {
105                let te = Self::parse_transfer_encoding_value(h.value)?;
106                has_transfer_encoding = has_transfer_encoding || te.has_transfer_encoding;
107                if te.is_chunked {
108                    is_chunked_transfer = true;
109                }
110                continue;
111            }
112
113            if h.name.eq_ignore_ascii_case("expect") {
114                has_expect = true;
115                if Header::trimmed_eq_ascii_case(h.value, b"100-continue") {
116                    expect_continue = true;
117                }
118            }
119        }
120
121        Ok(HeaderScan {
122            content_length,
123            has_transfer_encoding,
124            is_chunked_transfer,
125            has_expect,
126            expect_continue,
127            duplicate_content_length,
128            accept_encoding_gzip: false,
129        })
130    }
131
132    pub fn content_length(headers: &[httparse::Header<'_>]) -> Result<Option<usize>> {
133        for h in headers {
134            if h.name.eq_ignore_ascii_case("content-length") {
135                return Ok(Some(Header::content_length(h.value)?));
136            }
137        }
138        Ok(None)
139    }
140
141    pub fn is_chunked(headers: &[httparse::Header<'_>]) -> bool {
142        Header::has_name(headers, "transfer-encoding")
143            && headers.iter().any(|h| {
144                h.name.eq_ignore_ascii_case("transfer-encoding")
145                    && Header::has_token(h.value, b"chunked")
146            })
147    }
148}
sark_core/http/codec/decode/headers.rs

sark_core/http/codec/decode/
headers.rs
