rwf2
/
Rocket
mirror of https://github.com/rwf2/Rocket.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove unsafe 'from_utf8_unchecked'.

This commit is contained in:
Sergio Benitez 2018-02-25 19:47:06 -08:00
parent 15bce5907e
commit d0f002c3d7
1 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lib/src/rocket.rs
View File

@ -1,5 +1,5 @@
use std::collections::HashMap;
use std::str::from_utf8_unchecked;
use std::str::from_utf8;
use std::cmp::min;
use std::net::SocketAddr;
use std::io::{self, Write};
@ -193,17 +193,14 @@ impl Rocket {
let (min_len, max_len) = ("_method=get".len(), "_method=delete".len());
let is_form = req.content_type().map_or(false, |ct| ct.is_form());
if is_form && req.method() == Method::Post && data_len >= min_len {
// We're only using this for comparison and throwing it away
// afterwards, so it doesn't matter if we have invalid UTF8.
let form = unsafe {
from_utf8_unchecked(&data.peek()[..min(data_len, max_len)])
};
if let Ok(form) = from_utf8(&data.peek()[..min(data_len, max_len)]) {
let method: Option<Result<Method, _>> = FormItems::from(form)
.filter(|&(key, _)| key.as_str() == "_method")
.map(|(_, value)| value.parse())
.next();
if let Some((key, value)) = FormItems::from(form).next() {
if key == "_method" {
if let Ok(method) = value.parse() {
req.set_method(method);
}
if let Some(Ok(method)) = method {
req.set_method(method);
}
}
}