From cfbdd9a608d564b492799ca1838759d156834953 Mon Sep 17 00:00:00 2001
From: Andy Wang <cbeuw.andy@gmail.com>
Date: Mon, 7 Sep 2020 14:47:37 +0100
Subject: [PATCH] Prevent nil pointer panic when server response is incorrect
 under CDN mode

---
 internal/client/websocket.go | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/internal/client/websocket.go b/internal/client/websocket.go
index 3e4cf05..e1b5ce7 100644
--- a/internal/client/websocket.go
+++ b/internal/client/websocket.go
@@ -62,3 +62,10 @@ func (ws *WSOverTLS) Handshake(rawConn net.Conn, authInfo AuthInfo) (sessionKey
 
 	return
 }
+
+func (ws *WSOverTLS) Close() error {
+	if ws.WebSocketConn != nil {
+		return ws.WebSocketConn.Close()
+	}
+	return nil
+}