From 8a0b35e95ae8d2e073e3eaf1906549fbeb706aa6 Mon Sep 17 00:00:00 2001
From: Rohith <gambol99@gmail.com>
Date: Sun, 7 Jan 2018 12:02:40 +0000
Subject: [PATCH] Method Check

Ensuring the http.Hijacker is implemented by the ResponseWriter
---
 utils.go | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/utils.go b/utils.go
index 7acc7a9..ac1b5ab 100644
--- a/utils.go
+++ b/utils.go
@@ -268,7 +268,10 @@ func tryUpdateConnection(req *http.Request, writer http.ResponseWriter, endpoint
 	defer tlsConn.Close()
 
 	// step: we need to hijack the underlining client connection
-	clientConn, _, err := writer.(http.Hijacker).Hijack()
+	clientConn, ok, err := writer.(http.Hijacker).Hijack()
+	if !ok {
+		return errors.New("writer does not implement http.Hijacker method")
+	}
 	if err != nil {
 		return err
 	}
-- 
GitLab