Merge pull request #16 from yoshuawuyts/normalize_headers

Issue 8: Normalize headers and make use of content-length value

Author: rylev

src/body.rs

@@ -31,6 +31,14 @@ impl<R: AsyncRead> Body<R> {
         }
     }
 
+    /// Create a new instance from a reader with a known size.
+    pub fn new_with_size(reader: R, size: usize) -> Self {
+        Self {
+            reader,
+            length: Some(size),
+        }
+    }
+
     /// Create a new empty body.
     pub fn empty(reader: R) -> Self {
         Self {

src/client.rs

@@ -126,9 +126,20 @@ where
     let body = match httparse_res
         .headers
         .iter()
-        .find(|h| h.name == "Content-Length")
+        .find(|h| h.name.eq_ignore_ascii_case("Content-Length"))
     {
-        Some(_header) => Body::new(reader), // TODO: use the header value
+        Some(content_length) => {
+            let length = std::str::from_utf8(content_length.value)
+                .ok()
+                .map(|s| s.parse::<usize>().ok())
+                .flatten();
+
+            if let Some(len) = length {
+                Body::new_with_size(reader, len)
+            } else {
+                return Err("Invalid value for Content-Length".into());
+            }
+        }
         None => Body::empty(reader),
     };
 

src/server.rs

@@ -210,9 +210,20 @@ where
     let body = match httparse_req
         .headers
         .iter()
-        .find(|h| h.name == "Content-Length")
+        .find(|h| h.name.eq_ignore_ascii_case("Content-Length"))
     {
-        Some(_header) => Body::new(reader), // TODO: use the header value
+        Some(content_length) => {
+            let length = std::str::from_utf8(content_length.value)
+                .ok()
+                .map(|s| s.parse::<usize>().ok())
+                .flatten();
+
+            if let Some(len) = length {
+                Body::new_with_size(reader, len)
+            } else {
+                return Err("Invalid value for Content-Length".into());
+            }
+        }
         None => Body::empty(reader),
     };