diff --git a/server/server.go b/server/server.go index cdf1788..f3d8e83 100644 --- a/server/server.go +++ b/server/server.go @@ -157,12 +157,20 @@ func (server *TTYServer) handleTTYWebsocket(w http.ResponseWriter, r *http.Reque w.WriteHeader(http.StatusForbidden) return } - upgrader := websocket.Upgrader{ ReadBufferSize: 1024, WriteBufferSize: 1024, - CrossOrigin: crossOrigin, } + if crossOrigin { + upgrader = websocket.Upgrader{ + ReadBufferSize: 1024, + WriteBufferSize: 1024, + CheckOrigin: func(r *http.Request) bool { + return true + }, + } + } + conn, err := upgrader.Upgrade(w, r, nil) if err != nil { diff --git a/vendor/github.com/gorilla/websocket/server.go b/vendor/github.com/gorilla/websocket/server.go index 932ef3b..24d53b3 100644 --- a/vendor/github.com/gorilla/websocket/server.go +++ b/vendor/github.com/gorilla/websocket/server.go @@ -72,8 +72,6 @@ type Upgrader struct { // guarantee that compression will be supported. Currently only "no context // takeover" modes are supported. EnableCompression bool - - CrossOrigin bool } func (u *Upgrader) returnError(w http.ResponseWriter, r *http.Request, status int, reason string) (*Conn, error) { @@ -151,7 +149,7 @@ func (u *Upgrader) Upgrade(w http.ResponseWriter, r *http.Request, responseHeade if checkOrigin == nil { checkOrigin = checkSameOrigin } - if !checkOrigin(r) && !u.CrossOrigin { + if !checkOrigin(r) { return u.returnError(w, r, http.StatusForbidden, "websocket: request origin not allowed by Upgrader.CheckOrigin") }