pulse_http 0.2.0

Async HTTP/1.1 framework on Tokio — raw TCP, routing, middleware, no Hyper
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

use crate::helpers::read_with_timeout::read_with_timeout;
use tokio::net::TcpStream;

pub async fn read_content_length_body(
    stream: &mut TcpStream,
    buffer: &mut Vec<u8>,
    body_start: usize,
    content_length: usize,
    max_body_size: usize,
    read_timeout_sec: u64,
) -> Result<(Vec<u8>, Vec<u8>), ()> {
    if content_length > max_body_size {
        return Err(());
    }

    let mut chunk = [0u8; 1024];

    while buffer.len().saturating_sub(body_start) < content_length {
        match read_with_timeout(stream, &mut chunk, read_timeout_sec).await {
            Ok(0) => return Err(()),
            Ok(n) => buffer.extend_from_slice(&chunk[..n]),
            Err(()) => return Err(()),
        }
    }

    let body_end = body_start + content_length;
    Ok((
        buffer[body_start..body_end].to_vec(),
        buffer[body_end..].to_vec(),
    ))
}