mirror of
https://github.com/actix/actix-extras.git
synced 2024-11-24 16:02:59 +01:00
If buffer is empty, read more data before calling parser.
This commit is contained in:
parent
a42a8a2321
commit
66881d7dd1
@ -39,19 +39,8 @@ impl HttpResponseParser {
|
|||||||
T: IoStream,
|
T: IoStream,
|
||||||
{
|
{
|
||||||
loop {
|
loop {
|
||||||
// Read some more data into the buffer for the parser.
|
// Don't call parser until we have data to parse.
|
||||||
match io.read_available(buf) {
|
if !buf.is_empty() {
|
||||||
Ok(Async::Ready((false, true))) => {
|
|
||||||
return Err(HttpResponseParserError::Disconnect)
|
|
||||||
}
|
|
||||||
Ok(Async::Ready(_)) => (),
|
|
||||||
Ok(Async::NotReady) => return Ok(Async::NotReady),
|
|
||||||
Err(err) => {
|
|
||||||
return Err(HttpResponseParserError::Error(err.into()))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Call HTTP response parser.
|
|
||||||
match HttpResponseParser::parse_message(buf)
|
match HttpResponseParser::parse_message(buf)
|
||||||
.map_err(HttpResponseParserError::Error)?
|
.map_err(HttpResponseParserError::Error)?
|
||||||
{
|
{
|
||||||
@ -63,7 +52,19 @@ impl HttpResponseParser {
|
|||||||
if buf.capacity() >= MAX_BUFFER_SIZE {
|
if buf.capacity() >= MAX_BUFFER_SIZE {
|
||||||
return Err(HttpResponseParserError::Error(ParseError::TooLarge));
|
return Err(HttpResponseParserError::Error(ParseError::TooLarge));
|
||||||
}
|
}
|
||||||
// Parser needs more data. Loop and read more data.
|
// Parser needs more data.
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Read some more data into the buffer for the parser.
|
||||||
|
match io.read_available(buf) {
|
||||||
|
Ok(Async::Ready((false, true))) => {
|
||||||
|
return Err(HttpResponseParserError::Disconnect)
|
||||||
|
}
|
||||||
|
Ok(Async::Ready(_)) => (),
|
||||||
|
Ok(Async::NotReady) => return Ok(Async::NotReady),
|
||||||
|
Err(err) => {
|
||||||
|
return Err(HttpResponseParserError::Error(err.into()))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user