Merge pull request #54 from yoshuawuyts/client-host

feat(client): insert host header

Author: yoshuawuyts

src/client.rs

@@ -49,7 +49,7 @@ impl Encoder {
 }
 
 /// Encode an HTTP request on the client.
-pub async fn encode(req: Request) -> Result<Encoder, std::io::Error> {
+pub async fn encode(req: Request) -> Result<Encoder, Error> {
     let mut buf: Vec<u8> = vec![];
 
     let mut url = req.url().path().to_owned();
@@ -66,6 +66,20 @@ pub async fn encode(req: Request) -> Result<Encoder, std::io::Error> {
     log::trace!("> {}", &val);
     buf.write_all(val.as_bytes()).await?;
 
+    // Insert Host header
+    // Insert host
+    let host = req.url().host_str().ok_or_else(|| {
+        Error::from_str(
+            ErrorKind::InvalidInput,
+            "missing hostname",
+            StatusCode::BadRequest,
+        )
+    })?;
+
+    let val = format!("host: {}\r\n", host);
+    log::trace!("> {}", &val);
+    buf.write_all(val.as_bytes()).await?;
+
     // If the body isn't streaming, we can set the content-length ahead of time. Else we need to
     // send all items in chunks.
     if let Some(len) = req.len() {

tests/fixtures/client-request1.txt

@@ -1,4 +1,5 @@
 GET / HTTP/1.1
+host: example.com
 content-length: 0
 date: {DATE}