From 84b1742f4eda25b0ef83e228024adef2f64d8cb2 Mon Sep 17 00:00:00 2001
From: dencoded <33698537+dencoded@users.noreply.github.com>
Date: Mon, 23 Mar 2020 17:20:21 -0400
Subject: [PATCH] permessage-deflate WS-extension support added

---
 apiserver/server.go | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/apiserver/server.go b/apiserver/server.go
index 7850c96..18aff00 100644
--- a/apiserver/server.go
+++ b/apiserver/server.go
@@ -61,11 +61,13 @@ func NewAPIServer(token string, indiServerAddr string, phd2ServerAddr string, po
 		port:           port,
 		isTLS:          isTLS,
 		e:              echo.New(),
-		upgrader:       websocket.Upgrader{},
-		connList:       []net.Conn{},
-		indiProfile:    indiProfile,
-		currMode:       currMode,
-		agentModes:     agentModes,
+		upgrader: websocket.Upgrader{
+			EnableCompression: true,
+		},
+		connList:    []net.Conn{},
+		indiProfile: indiProfile,
+		currMode:    currMode,
+		agentModes:  agentModes,
 	}
 
 	if logutil.IsDev {
@@ -91,7 +93,9 @@ func NewAPIServer(token string, indiServerAddr string, phd2ServerAddr string, po
 		if err != nil {
 			return false
 		}
-		return allowedOrigins[host]
+
+		// check both host and host:port from --api-origins param values
+		return allowedOrigins[host] || allowedOrigins[u.Host]
 	}
 
 	return apiServer