Added X-Forwarded-Proto

dvs-fork · May 14, 2019 · d4bd84c · d4bd84c
1 parent ebe7603
commit d4bd84c
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/http.go b/http.go
@@ -52,6 +52,14 @@ func startHTTPHandler(state *State) {
 			return
 		}
 
+		requestedScheme := "http"
+
+		if c.Request.TLS != nil {
+			requestedScheme = "https"
+		}
+
+		c.Request.Header.Set("X-Forwarded-Proto", requestedScheme)
+
 		proxyHolder := loc.(*ProxyHolder)
 
 		url := *c.Request.URL
@@ -65,7 +73,7 @@ func startHTTPHandler(state *State) {
 			return net.Dial("unix", proxyHolder.ProxyTo)
 		}
 
-		if websocket.IsWebSocketUpgrade(c.Request) {
+		if c.IsWebsocket() {
 			scheme := "ws"
 			if url.Scheme == "https" {
 				scheme = "wss"