From 5d11a6c09b32c8c621cec144055ee80114e5719c Mon Sep 17 00:00:00 2001
From: Tryggvi Larusson <tryggvil@greenqloud.com>
Date: Wed, 24 Feb 2016 12:57:15 +0000
Subject: [PATCH] Adding support for running behind a ssl proxy with
 X-Forwarded-Proto header

---
 wsgate/wsgateEHS.cpp | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/wsgate/wsgateEHS.cpp b/wsgate/wsgateEHS.cpp
index f1139454..873c1753 100644
--- a/wsgate/wsgateEHS.cpp
+++ b/wsgate/wsgateEHS.cpp
@@ -536,8 +536,11 @@ namespace wsgate{
         MimeType mt = simpleMime(to_lower_copy(basename));
         if (HTML == mt) {
             ostringstream oss;
-
-            oss << (request->Secure() ? "wss://" : "ws://") << thisHost << "/wsgate";
+            bool is_secure = request->Secure();
+            if (request->Headers("X-Forwarded-Proto").compare("https") == 0) {
+              is_secure=true;
+            }
+            oss << (is_secure ? "wss://" : "ws://") << thisHost << "/wsgate";
 
             replace_all(body, "%WSURI%", oss.str());
             replace_all(body, "%JSDEBUG%", (bDynDebug ? "-debug" : ""));