ehttpd_range/
requestext.rs

1//! An extension trait for HTTP requests to work with range requests
2
3use crate::anyrange::AnyInclusiveRange;
4use ehttpd::{
5    bytes::DataParseExt,
6    error,
7    error::Error,
8    http::{Request, RequestExt},
9};
10use std::str;
11
12/// An extension trait for HTTP requests to work with range requests
13pub trait RequestRangeExt {
14    /// The request `Range` field if any
15    fn range(&self) -> Result<Option<AnyInclusiveRange<u64>>, Error>;
16}
17impl<'a, const HEADER_SIZE_MAX: usize> RequestRangeExt for Request<'a, HEADER_SIZE_MAX> {
18    fn range(&self) -> Result<Option<AnyInclusiveRange<u64>>, Error> {
19        // Get the range request
20        let Some(range) = self.field("Range") else {
21            return Ok(None);
22        };
23
24        // Parse the range header
25        let mut range = range.clone();
26        let kind = range.split_off(b"=").ok_or_else(|| error!("Invalid range header field"))?;
27        if !kind.eq(b"bytes") {
28            return Err(error!("Invalid range kind"))?;
29        }
30
31        // Read start and end values
32        let start = range.split_off(b"-").ok_or_else(|| error!("Invalid range"))?;
33        let end = range;
34
35        // Parse the start and end values
36        let start = str::from_utf8(&start)?;
37        let end = str::from_utf8(&end)?;
38        let range = match (start, end) {
39            ("", "") => AnyInclusiveRange::Full,
40            (start, "") => AnyInclusiveRange::From { start: start.parse()? },
41            ("", end) => AnyInclusiveRange::To { end: end.parse()? },
42            (start, end) => AnyInclusiveRange::FromTo { start: start.parse()?, end: end.parse()? },
43        };
44        Ok(Some(range))
45    }
46}
ehttpd_range/requestext.rs

ehttpd_range/
requestext.rs
