mirror of
https://github.com/fafhrd91/actix-web
synced 2024-11-24 00:21:08 +01:00
HRS security fixes (#2363)
This commit is contained in:
parent
f6e69919ed
commit
e965d8298f
@ -3,6 +3,11 @@
|
|||||||
## Unreleased - 2021-xx-xx
|
## Unreleased - 2021-xx-xx
|
||||||
|
|
||||||
|
|
||||||
|
## 3.0.0-beta.8 - 2021-08-09
|
||||||
|
### Fixed
|
||||||
|
* Potential HTTP request smuggling vulnerabilities. [RUSTSEC-2021-0081](https://github.com/rustsec/advisory-db/pull/977)
|
||||||
|
|
||||||
|
|
||||||
## 3.0.0-beta.8 - 2021-06-26
|
## 3.0.0-beta.8 - 2021-06-26
|
||||||
### Changed
|
### Changed
|
||||||
* Change compression algorithm features flags. [#2250]
|
* Change compression algorithm features flags. [#2250]
|
||||||
@ -210,6 +215,11 @@
|
|||||||
[#1878]: https://github.com/actix/actix-web/pull/1878
|
[#1878]: https://github.com/actix/actix-web/pull/1878
|
||||||
|
|
||||||
|
|
||||||
|
## 2.2.1 - 2021-08-09
|
||||||
|
### Fixed
|
||||||
|
* Potential HTTP request smuggling vulnerabilities. [RUSTSEC-2021-0081](https://github.com/rustsec/advisory-db/pull/977)
|
||||||
|
|
||||||
|
|
||||||
## 2.2.0 - 2020-11-25
|
## 2.2.0 - 2020-11-25
|
||||||
### Added
|
### Added
|
||||||
* HttpResponse builders for 1xx status codes. [#1768]
|
* HttpResponse builders for 1xx status codes. [#1768]
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "actix-http"
|
name = "actix-http"
|
||||||
version = "3.0.0-beta.8"
|
version = "3.0.0-beta.9"
|
||||||
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
authors = ["Nikolay Kim <fafhrd91@gmail.com>"]
|
||||||
description = "HTTP primitives for the Actix ecosystem"
|
description = "HTTP primitives for the Actix ecosystem"
|
||||||
keywords = ["actix", "http", "framework", "async", "futures"]
|
keywords = ["actix", "http", "framework", "async", "futures"]
|
||||||
|
@ -3,11 +3,11 @@
|
|||||||
> HTTP primitives for the Actix ecosystem.
|
> HTTP primitives for the Actix ecosystem.
|
||||||
|
|
||||||
[![crates.io](https://img.shields.io/crates/v/actix-http?label=latest)](https://crates.io/crates/actix-http)
|
[![crates.io](https://img.shields.io/crates/v/actix-http?label=latest)](https://crates.io/crates/actix-http)
|
||||||
[![Documentation](https://docs.rs/actix-http/badge.svg?version=3.0.0-beta.8)](https://docs.rs/actix-http/3.0.0-beta.8)
|
[![Documentation](https://docs.rs/actix-http/badge.svg?version=3.0.0-beta.9)](https://docs.rs/actix-http/3.0.0-beta.9)
|
||||||
[![Version](https://img.shields.io/badge/rustc-1.46+-ab6000.svg)](https://blog.rust-lang.org/2020/03/12/Rust-1.46.html)
|
[![Version](https://img.shields.io/badge/rustc-1.46+-ab6000.svg)](https://blog.rust-lang.org/2020/03/12/Rust-1.46.html)
|
||||||
![MIT or Apache 2.0 licensed](https://img.shields.io/crates/l/actix-http.svg)
|
![MIT or Apache 2.0 licensed](https://img.shields.io/crates/l/actix-http.svg)
|
||||||
<br />
|
<br />
|
||||||
[![dependency status](https://deps.rs/crate/actix-http/3.0.0-beta.8/status.svg)](https://deps.rs/crate/actix-http/3.0.0-beta.8)
|
[![dependency status](https://deps.rs/crate/actix-http/3.0.0-beta.9/status.svg)](https://deps.rs/crate/actix-http/3.0.0-beta.9)
|
||||||
[![Download](https://img.shields.io/crates/d/actix-http.svg)](https://crates.io/crates/actix-http)
|
[![Download](https://img.shields.io/crates/d/actix-http.svg)](https://crates.io/crates/actix-http)
|
||||||
[![Chat on Discord](https://img.shields.io/discord/771444961383153695?label=chat&logo=discord)](https://discord.gg/NWpN5mmg3x)
|
[![Chat on Discord](https://img.shields.io/discord/771444961383153695?label=chat&logo=discord)](https://discord.gg/NWpN5mmg3x)
|
||||||
|
|
||||||
|
@ -196,7 +196,7 @@ pub enum ParseError {
|
|||||||
#[display(fmt = "IO error: {}", _0)]
|
#[display(fmt = "IO error: {}", _0)]
|
||||||
Io(io::Error),
|
Io(io::Error),
|
||||||
|
|
||||||
/// Parsing a field as string failed
|
/// Parsing a field as string failed.
|
||||||
#[display(fmt = "UTF8 error: {}", _0)]
|
#[display(fmt = "UTF8 error: {}", _0)]
|
||||||
Utf8(Utf8Error),
|
Utf8(Utf8Error),
|
||||||
}
|
}
|
||||||
|
432
actix-http/src/h1/chunked.rs
Normal file
432
actix-http/src/h1/chunked.rs
Normal file
@ -0,0 +1,432 @@
|
|||||||
|
use std::{io, task::Poll};
|
||||||
|
|
||||||
|
use bytes::{Buf as _, Bytes, BytesMut};
|
||||||
|
|
||||||
|
macro_rules! byte (
|
||||||
|
($rdr:ident) => ({
|
||||||
|
if $rdr.len() > 0 {
|
||||||
|
let b = $rdr[0];
|
||||||
|
$rdr.advance(1);
|
||||||
|
b
|
||||||
|
} else {
|
||||||
|
return Poll::Pending
|
||||||
|
}
|
||||||
|
})
|
||||||
|
);
|
||||||
|
|
||||||
|
#[derive(Debug, PartialEq, Clone)]
|
||||||
|
pub(super) enum ChunkedState {
|
||||||
|
Size,
|
||||||
|
SizeLws,
|
||||||
|
Extension,
|
||||||
|
SizeLf,
|
||||||
|
Body,
|
||||||
|
BodyCr,
|
||||||
|
BodyLf,
|
||||||
|
EndCr,
|
||||||
|
EndLf,
|
||||||
|
End,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl ChunkedState {
|
||||||
|
pub(super) fn step(
|
||||||
|
&self,
|
||||||
|
body: &mut BytesMut,
|
||||||
|
size: &mut u64,
|
||||||
|
buf: &mut Option<Bytes>,
|
||||||
|
) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
use self::ChunkedState::*;
|
||||||
|
match *self {
|
||||||
|
Size => ChunkedState::read_size(body, size),
|
||||||
|
SizeLws => ChunkedState::read_size_lws(body),
|
||||||
|
Extension => ChunkedState::read_extension(body),
|
||||||
|
SizeLf => ChunkedState::read_size_lf(body, size),
|
||||||
|
Body => ChunkedState::read_body(body, size, buf),
|
||||||
|
BodyCr => ChunkedState::read_body_cr(body),
|
||||||
|
BodyLf => ChunkedState::read_body_lf(body),
|
||||||
|
EndCr => ChunkedState::read_end_cr(body),
|
||||||
|
EndLf => ChunkedState::read_end_lf(body),
|
||||||
|
End => Poll::Ready(Ok(ChunkedState::End)),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn read_size(
|
||||||
|
rdr: &mut BytesMut,
|
||||||
|
size: &mut u64,
|
||||||
|
) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
let radix = 16;
|
||||||
|
|
||||||
|
let rem = match byte!(rdr) {
|
||||||
|
b @ b'0'..=b'9' => b - b'0',
|
||||||
|
b @ b'a'..=b'f' => b + 10 - b'a',
|
||||||
|
b @ b'A'..=b'F' => b + 10 - b'A',
|
||||||
|
b'\t' | b' ' => return Poll::Ready(Ok(ChunkedState::SizeLws)),
|
||||||
|
b';' => return Poll::Ready(Ok(ChunkedState::Extension)),
|
||||||
|
b'\r' => return Poll::Ready(Ok(ChunkedState::SizeLf)),
|
||||||
|
_ => {
|
||||||
|
return Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk size line: Invalid Size",
|
||||||
|
)));
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
match size.checked_mul(radix) {
|
||||||
|
Some(n) => {
|
||||||
|
*size = n as u64;
|
||||||
|
*size += rem as u64;
|
||||||
|
|
||||||
|
Poll::Ready(Ok(ChunkedState::Size))
|
||||||
|
}
|
||||||
|
None => {
|
||||||
|
log::debug!("chunk size would overflow u64");
|
||||||
|
Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk size line: Size is too big",
|
||||||
|
)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn read_size_lws(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
// LWS can follow the chunk size, but no more digits can come
|
||||||
|
b'\t' | b' ' => Poll::Ready(Ok(ChunkedState::SizeLws)),
|
||||||
|
b';' => Poll::Ready(Ok(ChunkedState::Extension)),
|
||||||
|
b'\r' => Poll::Ready(Ok(ChunkedState::SizeLf)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk size linear white space",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn read_extension(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\r' => Poll::Ready(Ok(ChunkedState::SizeLf)),
|
||||||
|
// strictly 0x20 (space) should be disallowed but we don't parse quoted strings here
|
||||||
|
0x00..=0x08 | 0x0a..=0x1f | 0x7f => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid character in chunk extension",
|
||||||
|
))),
|
||||||
|
_ => Poll::Ready(Ok(ChunkedState::Extension)), // no supported extensions
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn read_size_lf(
|
||||||
|
rdr: &mut BytesMut,
|
||||||
|
size: &mut u64,
|
||||||
|
) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\n' if *size > 0 => Poll::Ready(Ok(ChunkedState::Body)),
|
||||||
|
b'\n' if *size == 0 => Poll::Ready(Ok(ChunkedState::EndCr)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk size LF",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn read_body(
|
||||||
|
rdr: &mut BytesMut,
|
||||||
|
rem: &mut u64,
|
||||||
|
buf: &mut Option<Bytes>,
|
||||||
|
) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
log::trace!("Chunked read, remaining={:?}", rem);
|
||||||
|
|
||||||
|
let len = rdr.len() as u64;
|
||||||
|
if len == 0 {
|
||||||
|
Poll::Ready(Ok(ChunkedState::Body))
|
||||||
|
} else {
|
||||||
|
let slice;
|
||||||
|
if *rem > len {
|
||||||
|
slice = rdr.split().freeze();
|
||||||
|
*rem -= len;
|
||||||
|
} else {
|
||||||
|
slice = rdr.split_to(*rem as usize).freeze();
|
||||||
|
*rem = 0;
|
||||||
|
}
|
||||||
|
*buf = Some(slice);
|
||||||
|
if *rem > 0 {
|
||||||
|
Poll::Ready(Ok(ChunkedState::Body))
|
||||||
|
} else {
|
||||||
|
Poll::Ready(Ok(ChunkedState::BodyCr))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn read_body_cr(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\r' => Poll::Ready(Ok(ChunkedState::BodyLf)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk body CR",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn read_body_lf(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\n' => Poll::Ready(Ok(ChunkedState::Size)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk body LF",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn read_end_cr(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\r' => Poll::Ready(Ok(ChunkedState::EndLf)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk end CR",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
fn read_end_lf(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
||||||
|
match byte!(rdr) {
|
||||||
|
b'\n' => Poll::Ready(Ok(ChunkedState::End)),
|
||||||
|
_ => Poll::Ready(Err(io::Error::new(
|
||||||
|
io::ErrorKind::InvalidInput,
|
||||||
|
"Invalid chunk end LF",
|
||||||
|
))),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod tests {
|
||||||
|
use actix_codec::Decoder as _;
|
||||||
|
use bytes::{Bytes, BytesMut};
|
||||||
|
use http::Method;
|
||||||
|
|
||||||
|
use crate::{
|
||||||
|
error::ParseError,
|
||||||
|
h1::decoder::{MessageDecoder, PayloadItem},
|
||||||
|
HttpMessage as _, Request,
|
||||||
|
};
|
||||||
|
|
||||||
|
macro_rules! parse_ready {
|
||||||
|
($e:expr) => {{
|
||||||
|
match MessageDecoder::<Request>::default().decode($e) {
|
||||||
|
Ok(Some((msg, _))) => msg,
|
||||||
|
Ok(_) => unreachable!("Eof during parsing http request"),
|
||||||
|
Err(err) => unreachable!("Error during parsing http request: {:?}", err),
|
||||||
|
}
|
||||||
|
}};
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! expect_parse_err {
|
||||||
|
($e:expr) => {{
|
||||||
|
match MessageDecoder::<Request>::default().decode($e) {
|
||||||
|
Err(err) => match err {
|
||||||
|
ParseError::Io(_) => unreachable!("Parse error expected"),
|
||||||
|
_ => {}
|
||||||
|
},
|
||||||
|
_ => unreachable!("Error expected"),
|
||||||
|
}
|
||||||
|
}};
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_parse_chunked_payload_chunk_extension() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\
|
||||||
|
\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
assert!(msg.chunked().unwrap());
|
||||||
|
|
||||||
|
buf.extend(b"4;test\r\ndata\r\n4\r\nline\r\n0\r\n\r\n"); // test: test\r\n\r\n")
|
||||||
|
let chunk = pl.decode(&mut buf).unwrap().unwrap().chunk();
|
||||||
|
assert_eq!(chunk, Bytes::from_static(b"data"));
|
||||||
|
let chunk = pl.decode(&mut buf).unwrap().unwrap().chunk();
|
||||||
|
assert_eq!(chunk, Bytes::from_static(b"line"));
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert!(msg.eof());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_request_chunked() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\r\n",
|
||||||
|
);
|
||||||
|
let req = parse_ready!(&mut buf);
|
||||||
|
|
||||||
|
if let Ok(val) = req.chunked() {
|
||||||
|
assert!(val);
|
||||||
|
} else {
|
||||||
|
unreachable!("Error");
|
||||||
|
}
|
||||||
|
|
||||||
|
// intentional typo in "chunked"
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chnked\r\n\r\n",
|
||||||
|
);
|
||||||
|
expect_parse_err!(&mut buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_http_request_chunked_payload() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\r\n",
|
||||||
|
);
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
assert!(req.chunked().unwrap());
|
||||||
|
|
||||||
|
buf.extend(b"4\r\ndata\r\n4\r\nline\r\n0\r\n\r\n");
|
||||||
|
assert_eq!(
|
||||||
|
pl.decode(&mut buf).unwrap().unwrap().chunk().as_ref(),
|
||||||
|
b"data"
|
||||||
|
);
|
||||||
|
assert_eq!(
|
||||||
|
pl.decode(&mut buf).unwrap().unwrap().chunk().as_ref(),
|
||||||
|
b"line"
|
||||||
|
);
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().unwrap().eof());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_http_request_chunked_payload_and_next_message() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\r\n",
|
||||||
|
);
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
assert!(req.chunked().unwrap());
|
||||||
|
|
||||||
|
buf.extend(
|
||||||
|
b"4\r\ndata\r\n4\r\nline\r\n0\r\n\r\n\
|
||||||
|
POST /test2 HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\r\n"
|
||||||
|
.iter(),
|
||||||
|
);
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"data");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"line");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert!(msg.eof());
|
||||||
|
|
||||||
|
let (req, _) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert!(req.chunked().unwrap());
|
||||||
|
assert_eq!(*req.method(), Method::POST);
|
||||||
|
assert!(req.chunked().unwrap());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_http_request_chunked_payload_chunks() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
transfer-encoding: chunked\r\n\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
assert!(req.chunked().unwrap());
|
||||||
|
|
||||||
|
buf.extend(b"4\r\n1111\r\n");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"1111");
|
||||||
|
|
||||||
|
buf.extend(b"4\r\ndata\r");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"data");
|
||||||
|
|
||||||
|
buf.extend(b"\n4");
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().is_none());
|
||||||
|
|
||||||
|
buf.extend(b"\r");
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().is_none());
|
||||||
|
buf.extend(b"\n");
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().is_none());
|
||||||
|
|
||||||
|
buf.extend(b"li");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"li");
|
||||||
|
|
||||||
|
//trailers
|
||||||
|
//buf.feed_data("test: test\r\n");
|
||||||
|
//not_ready!(reader.parse(&mut buf, &mut readbuf));
|
||||||
|
|
||||||
|
buf.extend(b"ne\r\n0\r\n");
|
||||||
|
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(msg.chunk().as_ref(), b"ne");
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().is_none());
|
||||||
|
|
||||||
|
buf.extend(b"\r\n");
|
||||||
|
assert!(pl.decode(&mut buf).unwrap().unwrap().eof());
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn chunk_extension_quoted() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
Host: localhost:8080\r\n\
|
||||||
|
Transfer-Encoding: chunked\r\n\
|
||||||
|
\r\n\
|
||||||
|
2;hello=b;one=\"1 2 3\"\r\n\
|
||||||
|
xx",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (_msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
|
||||||
|
let chunk = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(chunk, PayloadItem::Chunk(Bytes::from_static(b"xx")));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_chunk_extension_invalid() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: localhost:8080\r\n\
|
||||||
|
Transfer-Encoding: chunked\r\n\
|
||||||
|
\r\n\
|
||||||
|
2;x\nx\r\n\
|
||||||
|
4c\r\n\
|
||||||
|
0\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (_msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
|
||||||
|
let err = pl.decode(&mut buf).unwrap_err();
|
||||||
|
assert!(err
|
||||||
|
.to_string()
|
||||||
|
.contains("Invalid character in chunk extension"));
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_chunk_size_overflow() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Transfer-Encoding: chunked\r\n\
|
||||||
|
\r\n\
|
||||||
|
f0000000000000003\r\n\
|
||||||
|
abc\r\n\
|
||||||
|
0\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (_msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
|
||||||
|
let err = pl.decode(&mut buf).unwrap_err();
|
||||||
|
assert!(err
|
||||||
|
.to_string()
|
||||||
|
.contains("Invalid chunk size line: Size is too big"));
|
||||||
|
}
|
||||||
|
}
|
@ -1,18 +1,18 @@
|
|||||||
use std::convert::TryFrom;
|
use std::{convert::TryFrom, io, marker::PhantomData, task::Poll};
|
||||||
use std::io;
|
|
||||||
use std::marker::PhantomData;
|
|
||||||
use std::task::Poll;
|
|
||||||
|
|
||||||
use actix_codec::Decoder;
|
use actix_codec::Decoder;
|
||||||
use bytes::{Buf, Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
use http::header::{HeaderName, HeaderValue};
|
use http::header::{HeaderName, HeaderValue};
|
||||||
use http::{header, Method, StatusCode, Uri, Version};
|
use http::{header, Method, StatusCode, Uri, Version};
|
||||||
use log::{debug, error, trace};
|
use log::{debug, error, trace};
|
||||||
|
|
||||||
use crate::error::ParseError;
|
use super::chunked::ChunkedState;
|
||||||
use crate::header::HeaderMap;
|
use crate::{
|
||||||
use crate::message::{ConnectionType, ResponseHead};
|
error::ParseError,
|
||||||
use crate::request::Request;
|
header::HeaderMap,
|
||||||
|
message::{ConnectionType, ResponseHead},
|
||||||
|
request::Request,
|
||||||
|
};
|
||||||
|
|
||||||
pub(crate) const MAX_BUFFER_SIZE: usize = 131_072;
|
pub(crate) const MAX_BUFFER_SIZE: usize = 131_072;
|
||||||
const MAX_HEADERS: usize = 96;
|
const MAX_HEADERS: usize = 96;
|
||||||
@ -67,6 +67,7 @@ pub(crate) trait MessageType: Sized {
|
|||||||
let mut has_upgrade_websocket = false;
|
let mut has_upgrade_websocket = false;
|
||||||
let mut expect = false;
|
let mut expect = false;
|
||||||
let mut chunked = false;
|
let mut chunked = false;
|
||||||
|
let mut seen_te = false;
|
||||||
let mut content_length = None;
|
let mut content_length = None;
|
||||||
|
|
||||||
{
|
{
|
||||||
@ -85,8 +86,17 @@ pub(crate) trait MessageType: Sized {
|
|||||||
};
|
};
|
||||||
|
|
||||||
match name {
|
match name {
|
||||||
header::CONTENT_LENGTH => {
|
header::CONTENT_LENGTH if content_length.is_some() => {
|
||||||
if let Ok(s) = value.to_str() {
|
debug!("multiple Content-Length");
|
||||||
|
return Err(ParseError::Header);
|
||||||
|
}
|
||||||
|
|
||||||
|
header::CONTENT_LENGTH => match value.to_str() {
|
||||||
|
Ok(s) if s.trim().starts_with('+') => {
|
||||||
|
debug!("illegal Content-Length: {:?}", s);
|
||||||
|
return Err(ParseError::Header);
|
||||||
|
}
|
||||||
|
Ok(s) => {
|
||||||
if let Ok(len) = s.parse::<u64>() {
|
if let Ok(len) = s.parse::<u64>() {
|
||||||
if len != 0 {
|
if len != 0 {
|
||||||
content_length = Some(len);
|
content_length = Some(len);
|
||||||
@ -95,15 +105,31 @@ pub(crate) trait MessageType: Sized {
|
|||||||
debug!("illegal Content-Length: {:?}", s);
|
debug!("illegal Content-Length: {:?}", s);
|
||||||
return Err(ParseError::Header);
|
return Err(ParseError::Header);
|
||||||
}
|
}
|
||||||
} else {
|
}
|
||||||
|
Err(_) => {
|
||||||
debug!("illegal Content-Length: {:?}", value);
|
debug!("illegal Content-Length: {:?}", value);
|
||||||
return Err(ParseError::Header);
|
return Err(ParseError::Header);
|
||||||
}
|
}
|
||||||
}
|
},
|
||||||
|
|
||||||
// transfer-encoding
|
// transfer-encoding
|
||||||
|
header::TRANSFER_ENCODING if seen_te => {
|
||||||
|
debug!("multiple Transfer-Encoding not allowed");
|
||||||
|
return Err(ParseError::Header);
|
||||||
|
}
|
||||||
|
|
||||||
header::TRANSFER_ENCODING => {
|
header::TRANSFER_ENCODING => {
|
||||||
|
seen_te = true;
|
||||||
|
|
||||||
if let Ok(s) = value.to_str().map(str::trim) {
|
if let Ok(s) = value.to_str().map(str::trim) {
|
||||||
chunked = s.eq_ignore_ascii_case("chunked");
|
if s.eq_ignore_ascii_case("chunked") {
|
||||||
|
chunked = true;
|
||||||
|
} else if s.eq_ignore_ascii_case("identity") {
|
||||||
|
// allow silently since multiple TE headers are already checked
|
||||||
|
} else {
|
||||||
|
debug!("illegal Transfer-Encoding: {:?}", s);
|
||||||
|
return Err(ParseError::Header);
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
return Err(ParseError::Header);
|
return Err(ParseError::Header);
|
||||||
}
|
}
|
||||||
@ -408,20 +434,6 @@ enum Kind {
|
|||||||
Eof,
|
Eof,
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, PartialEq, Clone)]
|
|
||||||
enum ChunkedState {
|
|
||||||
Size,
|
|
||||||
SizeLws,
|
|
||||||
Extension,
|
|
||||||
SizeLf,
|
|
||||||
Body,
|
|
||||||
BodyCr,
|
|
||||||
BodyLf,
|
|
||||||
EndCr,
|
|
||||||
EndLf,
|
|
||||||
End,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Decoder for PayloadDecoder {
|
impl Decoder for PayloadDecoder {
|
||||||
type Item = PayloadItem;
|
type Item = PayloadItem;
|
||||||
type Error = io::Error;
|
type Error = io::Error;
|
||||||
@ -451,19 +463,23 @@ impl Decoder for PayloadDecoder {
|
|||||||
Kind::Chunked(ref mut state, ref mut size) => {
|
Kind::Chunked(ref mut state, ref mut size) => {
|
||||||
loop {
|
loop {
|
||||||
let mut buf = None;
|
let mut buf = None;
|
||||||
|
|
||||||
// advances the chunked state
|
// advances the chunked state
|
||||||
*state = match state.step(src, size, &mut buf) {
|
*state = match state.step(src, size, &mut buf) {
|
||||||
Poll::Pending => return Ok(None),
|
Poll::Pending => return Ok(None),
|
||||||
Poll::Ready(Ok(state)) => state,
|
Poll::Ready(Ok(state)) => state,
|
||||||
Poll::Ready(Err(e)) => return Err(e),
|
Poll::Ready(Err(e)) => return Err(e),
|
||||||
};
|
};
|
||||||
|
|
||||||
if *state == ChunkedState::End {
|
if *state == ChunkedState::End {
|
||||||
trace!("End of chunked stream");
|
trace!("End of chunked stream");
|
||||||
return Ok(Some(PayloadItem::Eof));
|
return Ok(Some(PayloadItem::Eof));
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(buf) = buf {
|
if let Some(buf) = buf {
|
||||||
return Ok(Some(PayloadItem::Chunk(buf)));
|
return Ok(Some(PayloadItem::Chunk(buf)));
|
||||||
}
|
}
|
||||||
|
|
||||||
if src.is_empty() {
|
if src.is_empty() {
|
||||||
return Ok(None);
|
return Ok(None);
|
||||||
}
|
}
|
||||||
@ -480,201 +496,40 @@ impl Decoder for PayloadDecoder {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
macro_rules! byte (
|
|
||||||
($rdr:ident) => ({
|
|
||||||
if $rdr.len() > 0 {
|
|
||||||
let b = $rdr[0];
|
|
||||||
$rdr.advance(1);
|
|
||||||
b
|
|
||||||
} else {
|
|
||||||
return Poll::Pending
|
|
||||||
}
|
|
||||||
})
|
|
||||||
);
|
|
||||||
|
|
||||||
impl ChunkedState {
|
|
||||||
fn step(
|
|
||||||
&self,
|
|
||||||
body: &mut BytesMut,
|
|
||||||
size: &mut u64,
|
|
||||||
buf: &mut Option<Bytes>,
|
|
||||||
) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
use self::ChunkedState::*;
|
|
||||||
match *self {
|
|
||||||
Size => ChunkedState::read_size(body, size),
|
|
||||||
SizeLws => ChunkedState::read_size_lws(body),
|
|
||||||
Extension => ChunkedState::read_extension(body),
|
|
||||||
SizeLf => ChunkedState::read_size_lf(body, size),
|
|
||||||
Body => ChunkedState::read_body(body, size, buf),
|
|
||||||
BodyCr => ChunkedState::read_body_cr(body),
|
|
||||||
BodyLf => ChunkedState::read_body_lf(body),
|
|
||||||
EndCr => ChunkedState::read_end_cr(body),
|
|
||||||
EndLf => ChunkedState::read_end_lf(body),
|
|
||||||
End => Poll::Ready(Ok(ChunkedState::End)),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_size(
|
|
||||||
rdr: &mut BytesMut,
|
|
||||||
size: &mut u64,
|
|
||||||
) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
let radix = 16;
|
|
||||||
match byte!(rdr) {
|
|
||||||
b @ b'0'..=b'9' => {
|
|
||||||
*size *= radix;
|
|
||||||
*size += u64::from(b - b'0');
|
|
||||||
}
|
|
||||||
b @ b'a'..=b'f' => {
|
|
||||||
*size *= radix;
|
|
||||||
*size += u64::from(b + 10 - b'a');
|
|
||||||
}
|
|
||||||
b @ b'A'..=b'F' => {
|
|
||||||
*size *= radix;
|
|
||||||
*size += u64::from(b + 10 - b'A');
|
|
||||||
}
|
|
||||||
b'\t' | b' ' => return Poll::Ready(Ok(ChunkedState::SizeLws)),
|
|
||||||
b';' => return Poll::Ready(Ok(ChunkedState::Extension)),
|
|
||||||
b'\r' => return Poll::Ready(Ok(ChunkedState::SizeLf)),
|
|
||||||
_ => {
|
|
||||||
return Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk size line: Invalid Size",
|
|
||||||
)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Poll::Ready(Ok(ChunkedState::Size))
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_size_lws(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
trace!("read_size_lws");
|
|
||||||
match byte!(rdr) {
|
|
||||||
// LWS can follow the chunk size, but no more digits can come
|
|
||||||
b'\t' | b' ' => Poll::Ready(Ok(ChunkedState::SizeLws)),
|
|
||||||
b';' => Poll::Ready(Ok(ChunkedState::Extension)),
|
|
||||||
b'\r' => Poll::Ready(Ok(ChunkedState::SizeLf)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk size linear white space",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn read_extension(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\r' => Poll::Ready(Ok(ChunkedState::SizeLf)),
|
|
||||||
_ => Poll::Ready(Ok(ChunkedState::Extension)), // no supported extensions
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn read_size_lf(
|
|
||||||
rdr: &mut BytesMut,
|
|
||||||
size: &mut u64,
|
|
||||||
) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\n' if *size > 0 => Poll::Ready(Ok(ChunkedState::Body)),
|
|
||||||
b'\n' if *size == 0 => Poll::Ready(Ok(ChunkedState::EndCr)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk size LF",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_body(
|
|
||||||
rdr: &mut BytesMut,
|
|
||||||
rem: &mut u64,
|
|
||||||
buf: &mut Option<Bytes>,
|
|
||||||
) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
trace!("Chunked read, remaining={:?}", rem);
|
|
||||||
|
|
||||||
let len = rdr.len() as u64;
|
|
||||||
if len == 0 {
|
|
||||||
Poll::Ready(Ok(ChunkedState::Body))
|
|
||||||
} else {
|
|
||||||
let slice;
|
|
||||||
if *rem > len {
|
|
||||||
slice = rdr.split().freeze();
|
|
||||||
*rem -= len;
|
|
||||||
} else {
|
|
||||||
slice = rdr.split_to(*rem as usize).freeze();
|
|
||||||
*rem = 0;
|
|
||||||
}
|
|
||||||
*buf = Some(slice);
|
|
||||||
if *rem > 0 {
|
|
||||||
Poll::Ready(Ok(ChunkedState::Body))
|
|
||||||
} else {
|
|
||||||
Poll::Ready(Ok(ChunkedState::BodyCr))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
fn read_body_cr(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\r' => Poll::Ready(Ok(ChunkedState::BodyLf)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk body CR",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn read_body_lf(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\n' => Poll::Ready(Ok(ChunkedState::Size)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk body LF",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn read_end_cr(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\r' => Poll::Ready(Ok(ChunkedState::EndLf)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk end CR",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
fn read_end_lf(rdr: &mut BytesMut) -> Poll<Result<ChunkedState, io::Error>> {
|
|
||||||
match byte!(rdr) {
|
|
||||||
b'\n' => Poll::Ready(Ok(ChunkedState::End)),
|
|
||||||
_ => Poll::Ready(Err(io::Error::new(
|
|
||||||
io::ErrorKind::InvalidInput,
|
|
||||||
"Invalid chunk end LF",
|
|
||||||
))),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
mod tests {
|
mod tests {
|
||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
use http::{Method, Version};
|
use http::{Method, Version};
|
||||||
|
|
||||||
use super::*;
|
use super::*;
|
||||||
use crate::error::ParseError;
|
use crate::{
|
||||||
use crate::http::header::{HeaderName, SET_COOKIE};
|
error::ParseError,
|
||||||
use crate::HttpMessage;
|
http::header::{HeaderName, SET_COOKIE},
|
||||||
|
HttpMessage as _,
|
||||||
|
};
|
||||||
|
|
||||||
impl PayloadType {
|
impl PayloadType {
|
||||||
fn unwrap(self) -> PayloadDecoder {
|
pub(crate) fn unwrap(self) -> PayloadDecoder {
|
||||||
match self {
|
match self {
|
||||||
PayloadType::Payload(pl) => pl,
|
PayloadType::Payload(pl) => pl,
|
||||||
_ => panic!(),
|
_ => panic!(),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn is_unhandled(&self) -> bool {
|
pub(crate) fn is_unhandled(&self) -> bool {
|
||||||
matches!(self, PayloadType::Stream(_))
|
matches!(self, PayloadType::Stream(_))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl PayloadItem {
|
impl PayloadItem {
|
||||||
fn chunk(self) -> Bytes {
|
pub(crate) fn chunk(self) -> Bytes {
|
||||||
match self {
|
match self {
|
||||||
PayloadItem::Chunk(chunk) => chunk,
|
PayloadItem::Chunk(chunk) => chunk,
|
||||||
_ => panic!("error"),
|
_ => panic!("error"),
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
fn eof(&self) -> bool {
|
|
||||||
|
pub(crate) fn eof(&self) -> bool {
|
||||||
matches!(*self, PayloadItem::Eof)
|
matches!(*self, PayloadItem::Eof)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -967,34 +822,6 @@ mod tests {
|
|||||||
assert!(req.upgrade());
|
assert!(req.upgrade());
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_request_chunked() {
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\r\n",
|
|
||||||
);
|
|
||||||
let req = parse_ready!(&mut buf);
|
|
||||||
|
|
||||||
if let Ok(val) = req.chunked() {
|
|
||||||
assert!(val);
|
|
||||||
} else {
|
|
||||||
unreachable!("Error");
|
|
||||||
}
|
|
||||||
|
|
||||||
// intentional typo in "chunked"
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chnked\r\n\r\n",
|
|
||||||
);
|
|
||||||
let req = parse_ready!(&mut buf);
|
|
||||||
|
|
||||||
if let Ok(val) = req.chunked() {
|
|
||||||
assert!(!val);
|
|
||||||
} else {
|
|
||||||
unreachable!("Error");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_headers_content_length_err_1() {
|
fn test_headers_content_length_err_1() {
|
||||||
let mut buf = BytesMut::from(
|
let mut buf = BytesMut::from(
|
||||||
@ -1112,126 +939,6 @@ mod tests {
|
|||||||
expect_parse_err!(&mut buf);
|
expect_parse_err!(&mut buf);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_http_request_chunked_payload() {
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\r\n",
|
|
||||||
);
|
|
||||||
let mut reader = MessageDecoder::<Request>::default();
|
|
||||||
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
|
||||||
let mut pl = pl.unwrap();
|
|
||||||
assert!(req.chunked().unwrap());
|
|
||||||
|
|
||||||
buf.extend(b"4\r\ndata\r\n4\r\nline\r\n0\r\n\r\n");
|
|
||||||
assert_eq!(
|
|
||||||
pl.decode(&mut buf).unwrap().unwrap().chunk().as_ref(),
|
|
||||||
b"data"
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
pl.decode(&mut buf).unwrap().unwrap().chunk().as_ref(),
|
|
||||||
b"line"
|
|
||||||
);
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().unwrap().eof());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_http_request_chunked_payload_and_next_message() {
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\r\n",
|
|
||||||
);
|
|
||||||
let mut reader = MessageDecoder::<Request>::default();
|
|
||||||
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
|
||||||
let mut pl = pl.unwrap();
|
|
||||||
assert!(req.chunked().unwrap());
|
|
||||||
|
|
||||||
buf.extend(
|
|
||||||
b"4\r\ndata\r\n4\r\nline\r\n0\r\n\r\n\
|
|
||||||
POST /test2 HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\r\n"
|
|
||||||
.iter(),
|
|
||||||
);
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"data");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"line");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert!(msg.eof());
|
|
||||||
|
|
||||||
let (req, _) = reader.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert!(req.chunked().unwrap());
|
|
||||||
assert_eq!(*req.method(), Method::POST);
|
|
||||||
assert!(req.chunked().unwrap());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_http_request_chunked_payload_chunks() {
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\r\n",
|
|
||||||
);
|
|
||||||
|
|
||||||
let mut reader = MessageDecoder::<Request>::default();
|
|
||||||
let (req, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
|
||||||
let mut pl = pl.unwrap();
|
|
||||||
assert!(req.chunked().unwrap());
|
|
||||||
|
|
||||||
buf.extend(b"4\r\n1111\r\n");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"1111");
|
|
||||||
|
|
||||||
buf.extend(b"4\r\ndata\r");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"data");
|
|
||||||
|
|
||||||
buf.extend(b"\n4");
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().is_none());
|
|
||||||
|
|
||||||
buf.extend(b"\r");
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().is_none());
|
|
||||||
buf.extend(b"\n");
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().is_none());
|
|
||||||
|
|
||||||
buf.extend(b"li");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"li");
|
|
||||||
|
|
||||||
//trailers
|
|
||||||
//buf.feed_data("test: test\r\n");
|
|
||||||
//not_ready!(reader.parse(&mut buf, &mut readbuf));
|
|
||||||
|
|
||||||
buf.extend(b"ne\r\n0\r\n");
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert_eq!(msg.chunk().as_ref(), b"ne");
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().is_none());
|
|
||||||
|
|
||||||
buf.extend(b"\r\n");
|
|
||||||
assert!(pl.decode(&mut buf).unwrap().unwrap().eof());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn test_parse_chunked_payload_chunk_extension() {
|
|
||||||
let mut buf = BytesMut::from(
|
|
||||||
"GET /test HTTP/1.1\r\n\
|
|
||||||
transfer-encoding: chunked\r\n\
|
|
||||||
\r\n",
|
|
||||||
);
|
|
||||||
|
|
||||||
let mut reader = MessageDecoder::<Request>::default();
|
|
||||||
let (msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
|
||||||
let mut pl = pl.unwrap();
|
|
||||||
assert!(msg.chunked().unwrap());
|
|
||||||
|
|
||||||
buf.extend(b"4;test\r\ndata\r\n4\r\nline\r\n0\r\n\r\n"); // test: test\r\n\r\n")
|
|
||||||
let chunk = pl.decode(&mut buf).unwrap().unwrap().chunk();
|
|
||||||
assert_eq!(chunk, Bytes::from_static(b"data"));
|
|
||||||
let chunk = pl.decode(&mut buf).unwrap().unwrap().chunk();
|
|
||||||
assert_eq!(chunk, Bytes::from_static(b"line"));
|
|
||||||
let msg = pl.decode(&mut buf).unwrap().unwrap();
|
|
||||||
assert!(msg.eof());
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn test_response_http10_read_until_eof() {
|
fn test_response_http10_read_until_eof() {
|
||||||
let mut buf = BytesMut::from("HTTP/1.0 200 Ok\r\n\r\ntest data");
|
let mut buf = BytesMut::from("HTTP/1.0 200 Ok\r\n\r\ntest data");
|
||||||
@ -1243,4 +950,84 @@ mod tests {
|
|||||||
let chunk = pl.decode(&mut buf).unwrap().unwrap();
|
let chunk = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
assert_eq!(chunk, PayloadItem::Chunk(Bytes::from_static(b"test data")));
|
assert_eq!(chunk, PayloadItem::Chunk(Bytes::from_static(b"test data")));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_multiple_content_length() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Content-Length: 4\r\n\
|
||||||
|
Content-Length: 2\r\n\
|
||||||
|
\r\n\
|
||||||
|
abcd",
|
||||||
|
);
|
||||||
|
|
||||||
|
expect_parse_err!(&mut buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_content_length_plus() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Content-Length: +3\r\n\
|
||||||
|
\r\n\
|
||||||
|
000",
|
||||||
|
);
|
||||||
|
|
||||||
|
expect_parse_err!(&mut buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_unknown_transfer_encoding() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Transfer-Encoding: JUNK\r\n\
|
||||||
|
Transfer-Encoding: chunked\r\n\
|
||||||
|
\r\n\
|
||||||
|
5\r\n\
|
||||||
|
hello\r\n\
|
||||||
|
0",
|
||||||
|
);
|
||||||
|
|
||||||
|
expect_parse_err!(&mut buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn hrs_multiple_transfer_encoding() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET / HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Content-Length: 51\r\n\
|
||||||
|
Transfer-Encoding: identity\r\n\
|
||||||
|
Transfer-Encoding: chunked\r\n\
|
||||||
|
\r\n\
|
||||||
|
0\r\n\
|
||||||
|
\r\n\
|
||||||
|
GET /forbidden HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
expect_parse_err!(&mut buf);
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn transfer_encoding_agrees() {
|
||||||
|
let mut buf = BytesMut::from(
|
||||||
|
"GET /test HTTP/1.1\r\n\
|
||||||
|
Host: example.com\r\n\
|
||||||
|
Content-Length: 3\r\n\
|
||||||
|
Transfer-Encoding: identity\r\n\
|
||||||
|
\r\n\
|
||||||
|
0\r\n",
|
||||||
|
);
|
||||||
|
|
||||||
|
let mut reader = MessageDecoder::<Request>::default();
|
||||||
|
let (_msg, pl) = reader.decode(&mut buf).unwrap().unwrap();
|
||||||
|
let mut pl = pl.unwrap();
|
||||||
|
|
||||||
|
let chunk = pl.decode(&mut buf).unwrap().unwrap();
|
||||||
|
assert_eq!(chunk, PayloadItem::Chunk(Bytes::from_static(b"0\r\n")));
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
@ -81,6 +81,7 @@ pub(crate) trait MessageType: Sized {
|
|||||||
match length {
|
match length {
|
||||||
BodySize::Stream => {
|
BodySize::Stream => {
|
||||||
if chunked {
|
if chunked {
|
||||||
|
skip_len = true;
|
||||||
if camel_case {
|
if camel_case {
|
||||||
dst.put_slice(b"\r\nTransfer-Encoding: chunked\r\n")
|
dst.put_slice(b"\r\nTransfer-Encoding: chunked\r\n")
|
||||||
} else {
|
} else {
|
||||||
|
@ -1,6 +1,8 @@
|
|||||||
//! HTTP/1 protocol implementation.
|
//! HTTP/1 protocol implementation.
|
||||||
|
|
||||||
use bytes::{Bytes, BytesMut};
|
use bytes::{Bytes, BytesMut};
|
||||||
|
|
||||||
|
mod chunked;
|
||||||
mod client;
|
mod client;
|
||||||
mod codec;
|
mod codec;
|
||||||
mod decoder;
|
mod decoder;
|
||||||
|
Loading…
Reference in New Issue
Block a user