actix_http/
helpers.rs

1use std::io;
2
3use bytes::BufMut;
4use http::Version;
5
6const DIGITS_START: u8 = b'0';
7
8pub(crate) fn write_status_line<B: BufMut>(version: Version, n: u16, buf: &mut B) {
9    match version {
10        Version::HTTP_11 => buf.put_slice(b"HTTP/1.1 "),
11        Version::HTTP_10 => buf.put_slice(b"HTTP/1.0 "),
12        Version::HTTP_09 => buf.put_slice(b"HTTP/0.9 "),
13        _ => {
14            // other HTTP version handlers do not use this method
15        }
16    }
17
18    let d100 = (n / 100) as u8;
19    let d10 = ((n / 10) % 10) as u8;
20    let d1 = (n % 10) as u8;
21
22    buf.put_u8(DIGITS_START + d100);
23    buf.put_u8(DIGITS_START + d10);
24    buf.put_u8(DIGITS_START + d1);
25
26    // trailing space before reason
27    buf.put_u8(b' ');
28}
29
30/// Write out content length header.
31///
32/// Buffer must to contain enough space or be implicitly extendable.
33pub fn write_content_length<B: BufMut>(n: u64, buf: &mut B, camel_case: bool) {
34    if n == 0 {
35        if camel_case {
36            buf.put_slice(b"\r\nContent-Length: 0\r\n");
37        } else {
38            buf.put_slice(b"\r\ncontent-length: 0\r\n");
39        }
40
41        return;
42    }
43
44    let mut buffer = itoa::Buffer::new();
45
46    if camel_case {
47        buf.put_slice(b"\r\nContent-Length: ");
48    } else {
49        buf.put_slice(b"\r\ncontent-length: ");
50    }
51
52    buf.put_slice(buffer.format(n).as_bytes());
53    buf.put_slice(b"\r\n");
54}
55
56/// An `io::Write`r that only requires mutable reference and assumes that there is space available
57/// in the buffer for every write operation or that it can be extended implicitly (like
58/// `bytes::BytesMut`, for example).
59///
60/// This is slightly faster (~10%) than `bytes::buf::Writer` in such cases because it does not
61/// perform a remaining length check before writing.
62pub(crate) struct MutWriter<'a, B>(pub(crate) &'a mut B);
63
64impl<B> io::Write for MutWriter<'_, B>
65where
66    B: BufMut,
67{
68    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
69        self.0.put_slice(buf);
70        Ok(buf.len())
71    }
72
73    fn flush(&mut self) -> io::Result<()> {
74        Ok(())
75    }
76}
77
78#[cfg(test)]
79mod tests {
80    use std::str::from_utf8;
81
82    use bytes::BytesMut;
83
84    use super::*;
85
86    #[test]
87    fn test_status_line() {
88        let mut bytes = BytesMut::new();
89        bytes.reserve(50);
90        write_status_line(Version::HTTP_11, 200, &mut bytes);
91        assert_eq!(from_utf8(&bytes.split().freeze()).unwrap(), "HTTP/1.1 200 ");
92
93        let mut bytes = BytesMut::new();
94        bytes.reserve(50);
95        write_status_line(Version::HTTP_09, 404, &mut bytes);
96        assert_eq!(from_utf8(&bytes.split().freeze()).unwrap(), "HTTP/0.9 404 ");
97
98        let mut bytes = BytesMut::new();
99        bytes.reserve(50);
100        write_status_line(Version::HTTP_09, 515, &mut bytes);
101        assert_eq!(from_utf8(&bytes.split().freeze()).unwrap(), "HTTP/0.9 515 ");
102    }
103
104    #[test]
105    fn test_write_content_length() {
106        let mut bytes = BytesMut::new();
107        bytes.reserve(50);
108        write_content_length(0, &mut bytes, false);
109        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 0\r\n"[..]);
110        bytes.reserve(50);
111        write_content_length(9, &mut bytes, false);
112        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 9\r\n"[..]);
113        bytes.reserve(50);
114        write_content_length(10, &mut bytes, false);
115        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 10\r\n"[..]);
116        bytes.reserve(50);
117        write_content_length(99, &mut bytes, false);
118        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 99\r\n"[..]);
119        bytes.reserve(50);
120        write_content_length(100, &mut bytes, false);
121        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 100\r\n"[..]);
122        bytes.reserve(50);
123        write_content_length(101, &mut bytes, false);
124        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 101\r\n"[..]);
125        bytes.reserve(50);
126        write_content_length(998, &mut bytes, false);
127        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 998\r\n"[..]);
128        bytes.reserve(50);
129        write_content_length(1000, &mut bytes, false);
130        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 1000\r\n"[..]);
131        bytes.reserve(50);
132        write_content_length(1001, &mut bytes, false);
133        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 1001\r\n"[..]);
134        bytes.reserve(50);
135        write_content_length(5909, &mut bytes, false);
136        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 5909\r\n"[..]);
137        bytes.reserve(50);
138        write_content_length(9999, &mut bytes, false);
139        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 9999\r\n"[..]);
140        bytes.reserve(50);
141        write_content_length(10001, &mut bytes, false);
142        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 10001\r\n"[..]);
143        bytes.reserve(50);
144        write_content_length(59094, &mut bytes, false);
145        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 59094\r\n"[..]);
146        bytes.reserve(50);
147        write_content_length(99999, &mut bytes, false);
148        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 99999\r\n"[..]);
149
150        bytes.reserve(50);
151        write_content_length(590947, &mut bytes, false);
152        assert_eq!(
153            bytes.split().freeze(),
154            b"\r\ncontent-length: 590947\r\n"[..]
155        );
156        bytes.reserve(50);
157        write_content_length(999999, &mut bytes, false);
158        assert_eq!(
159            bytes.split().freeze(),
160            b"\r\ncontent-length: 999999\r\n"[..]
161        );
162        bytes.reserve(50);
163        write_content_length(5909471, &mut bytes, false);
164        assert_eq!(
165            bytes.split().freeze(),
166            b"\r\ncontent-length: 5909471\r\n"[..]
167        );
168        bytes.reserve(50);
169        write_content_length(59094718, &mut bytes, false);
170        assert_eq!(
171            bytes.split().freeze(),
172            b"\r\ncontent-length: 59094718\r\n"[..]
173        );
174        bytes.reserve(50);
175        write_content_length(4294973728, &mut bytes, false);
176        assert_eq!(
177            bytes.split().freeze(),
178            b"\r\ncontent-length: 4294973728\r\n"[..]
179        );
180    }
181
182    #[test]
183    fn write_content_length_camel_case() {
184        let mut bytes = BytesMut::new();
185        write_content_length(0, &mut bytes, false);
186        assert_eq!(bytes.split().freeze(), b"\r\ncontent-length: 0\r\n"[..]);
187
188        let mut bytes = BytesMut::new();
189        write_content_length(0, &mut bytes, true);
190        assert_eq!(bytes.split().freeze(), b"\r\nContent-Length: 0\r\n"[..]);
191    }
192}
actix_http/helpers.rs

actix_http/
helpers.rs
