berkus
/
actix-web
mirror of https://github.com/fafhrd91/actix-web
1
0
Fork 0
Code Issues Releases Wiki Activity

add comment for None variant output of MessageType::decode for Request

This commit is contained in:
fakeshadow 2021-01-27 02:37:38 -08:00
parent a3fd266629
commit 3eca681cc2
2 changed files with 13 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
actix-http/src/h1/decoder.rs
View File

@ -208,8 +208,9 @@ impl MessageType for Request {
trace!("MAX_BUFFER_SIZE unprocessed data reached, closing"); trace!("MAX_BUFFER_SIZE unprocessed data reached, closing");
Err(ParseError::TooLarge) Err(ParseError::TooLarge)
} else { } else {
// Return None to notify more read are needed for parsing request
Ok(None) Ok(None)
} };
} }
} }
}; };
@ -277,14 +278,16 @@ impl MessageType for ResponseHead {
(len, version, status, res.headers.len()) (len, version, status, res.headers.len())
} }
httparse::Status::Partial => return { httparse::Status::Partial => {
if src.len() >= MAX_BUFFER_SIZE { return {
error!("MAX_BUFFER_SIZE unprocessed data reached, closing"); if src.len() >= MAX_BUFFER_SIZE {
Err(ParseError::TooLarge) error!("MAX_BUFFER_SIZE unprocessed data reached, closing");
} else { Err(ParseError::TooLarge)
Ok(None) } else {
Ok(None)
}
} }
}, }
} }
}; };

2
actix-http/src/h1/dispatcher.rs
View File

@ -609,6 +609,8 @@ where
} }
} }
} }
// decode is partial and buffer is not full yet.
// break and wait for more read.
Ok(None) => break, Ok(None) => break,
Err(ParseError::Io(e)) => { Err(ParseError::Io(e)) => {
self.as_mut().client_disconnected(); self.as_mut().client_disconnected();