This commit is contained in:
vinibuzzacaro 2025-11-11 21:59:30 -03:00 committed by GitHub
commit 09fe594fcb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 22 additions and 14 deletions

View File

@ -2,6 +2,10 @@
## Unreleased ## Unreleased
## 3.11.3
- Update `HttpMessage`s `content_type` method to return `Option<&str>` instead of `&str`.
## 3.11.2 ## 3.11.2
- Properly wake Payload receivers when feeding errors or EOF. - Properly wake Payload receivers when feeding errors or EOF.

View File

@ -46,13 +46,11 @@ pub trait HttpMessage: Sized {
/// Read the request content type. If request did not contain a *Content-Type* header, an empty /// Read the request content type. If request did not contain a *Content-Type* header, an empty
/// string is returned. /// string is returned.
fn content_type(&self) -> &str { fn content_type(&self) -> Option<&str> {
if let Some(content_type) = self.headers().get(header::CONTENT_TYPE) { self.headers()
if let Ok(content_type) = content_type.to_str() { .get(header::CONTENT_TYPE)
return content_type.split(';').next().unwrap().trim(); .and_then(|ct| ct.to_str().ok())
} .map(|ct| ct.split(';').next().unwrap().trim())
}
""
} }
/// Get content type encoding. /// Get content type encoding.
@ -142,13 +140,13 @@ mod tests {
let req = TestRequest::default() let req = TestRequest::default()
.insert_header(("content-type", "text/plain")) .insert_header(("content-type", "text/plain"))
.finish(); .finish();
assert_eq!(req.content_type(), "text/plain"); assert_eq!(req.content_type(), Some("text/plain"));
let req = TestRequest::default() let req = TestRequest::default()
.insert_header(("content-type", "application/json; charset=utf-8")) .insert_header(("content-type", "application/json; charset=utf-8"))
.finish(); .finish();
assert_eq!(req.content_type(), "application/json"); assert_eq!(req.content_type(), Some("application/json"));
let req = TestRequest::default().finish(); let req = TestRequest::default().finish();
assert_eq!(req.content_type(), ""); assert_eq!(req.content_type(), None);
} }
#[test] #[test]

View File

@ -505,7 +505,10 @@ mod tests {
.set_form(&payload) .set_form(&payload)
.to_request(); .to_request();
assert_eq!(req.content_type(), "application/x-www-form-urlencoded"); assert_eq!(
req.content_type().unwrap(),
"application/x-www-form-urlencoded"
);
let result: Person = call_and_read_body_json(&app, req).await; let result: Person = call_and_read_body_json(&app, req).await;
assert_eq!(&result.id, "12345"); assert_eq!(&result.id, "12345");
@ -549,7 +552,7 @@ mod tests {
.set_json(&payload) .set_json(&payload)
.to_request(); .to_request();
assert_eq!(req.content_type(), "application/json"); assert_eq!(req.content_type(), Some("application/json"));
let result: Person = call_and_read_body_json(&app, req).await; let result: Person = call_and_read_body_json(&app, req).await;
assert_eq!(&result.id, "12345"); assert_eq!(&result.id, "12345");

View File

@ -289,14 +289,17 @@ impl<T> UrlEncoded<T> {
/// Create a new future to decode a URL encoded request payload. /// Create a new future to decode a URL encoded request payload.
pub fn new(req: &HttpRequest, payload: &mut Payload) -> Self { pub fn new(req: &HttpRequest, payload: &mut Payload) -> Self {
// check content type // check content type
if req.content_type().to_lowercase() != "application/x-www-form-urlencoded" { let encoding_error = req
.content_type()
.map(|ct| !ct.eq_ignore_ascii_case("application/x-www-form-urlencoded"))
.unwrap_or(true);
if encoding_error {
return Self::err(UrlencodedError::ContentType); return Self::err(UrlencodedError::ContentType);
} }
let encoding = match req.encoding() { let encoding = match req.encoding() {
Ok(enc) => enc, Ok(enc) => enc,
Err(_) => return Self::err(UrlencodedError::ContentType), Err(_) => return Self::err(UrlencodedError::ContentType),
}; };
let mut len = None; let mut len = None;
if let Some(l) = req.headers().get(&CONTENT_LENGTH) { if let Some(l) = req.headers().get(&CONTENT_LENGTH) {
if let Ok(s) = l.to_str() { if let Ok(s) = l.to_str() {