From 5871328be2ddfeb05d8077454e01da59c9012aed Mon Sep 17 00:00:00 2001
From: Rotem Yaari <vmalloc@gmail.com>
Date: Sat, 17 Nov 2018 20:27:29 +0200
Subject: [PATCH] Fix proxy to handle both chunked and non-chunked responses
 (#59)

---
 http-full-proxy/src/main.rs | 32 ++++++++++++++++++++++----------
 1 file changed, 22 insertions(+), 10 deletions(-)

diff --git a/http-full-proxy/src/main.rs b/http-full-proxy/src/main.rs
index 73a53b34..4d691870 100644
--- a/http-full-proxy/src/main.rs
+++ b/http-full-proxy/src/main.rs
@@ -11,7 +11,7 @@ use actix_web::{
     HttpResponse,
 };
 use clap::{value_t, Arg};
-use futures::Future;
+use futures::{future, Future};
 use std::net::ToSocketAddrs;
 use url::Url;
 
@@ -55,16 +55,28 @@ fn forward(
     forwarded_req
         .send()
         .map_err(Error::from)
-        .and_then(move |resp| {
-            let mut client_resp = HttpResponse::build(resp.status());
-            for (header_name, header_value) in
-                resp.headers().iter().filter(|(h, _)| *h != "connection")
-            {
-                client_resp.header(header_name.clone(), header_value.clone());
-            }
+        .and_then(construct_response)
+        .responder()
+}
 
-            Ok(client_resp.streaming(resp.payload()))
-        }).responder()
+fn construct_response(
+    resp: client::ClientResponse,
+) -> Box<dyn Future<Item = HttpResponse, Error = Error>> {
+    let mut client_resp = HttpResponse::build(resp.status());
+    for (header_name, header_value) in
+        resp.headers().iter().filter(|(h, _)| *h != "connection")
+    {
+        client_resp.header(header_name.clone(), header_value.clone());
+    }
+    if resp.chunked().unwrap_or(false) {
+        Box::new(future::ok(client_resp.streaming(resp.payload())))
+    } else {
+        Box::new(
+            resp.body()
+                .from_err()
+                .and_then(move |body| Ok(client_resp.body(body))),
+        )
+    }
 }
 
 fn main() {