From 96f11de7ab15425d7a118fec5599c8fe87d280c0 Mon Sep 17 00:00:00 2001
From: Areg Harutyunyan <areg@cloudflare.com>
Date: Thu, 19 Mar 2020 01:39:59 +0000
Subject: [PATCH] TUN-2820: Serialized headers for Websockets

---
 h2mux/header.go | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/h2mux/header.go b/h2mux/header.go
index 62e71164..4a853c09 100644
--- a/h2mux/header.go
+++ b/h2mux/header.go
@@ -103,7 +103,8 @@ func ParseUserHeaders(headerNameToParseFrom string, headers []Header) ([]Header,
 func IsControlHeader(headerName string) bool {
 	headerName = strings.ToLower(headerName)
 
-	return strings.ToLower(headerName) == "content-length" ||
+	return headerName == "content-length" ||
+		headerName == "connection" || headerName == "upgrade" || // Websocket headers
 		strings.HasPrefix(headerName, ":") ||
 		strings.HasPrefix(headerName, "cf-")
 }