Explorar o código

Check and handle error return from hijack.

Gary Burd %!s(int64=11) %!d(string=hai) anos
pai
achega
4292df70df
Modificáronse 1 ficheiros con 3 adicións e 0 borrados
  1. 3 0
      server.go

+ 3 - 0
server.go

@@ -137,6 +137,9 @@ func (u *Upgrader) Upgrade(w http.ResponseWriter, r *http.Request, responseHeade
 	}
 	var rw *bufio.ReadWriter
 	netConn, rw, err = h.Hijack()
+	if err != nil {
+		return u.returnError(w, r, http.StatusInternalServerError, err.Error())
+	}
 	br = rw.Reader
 
 	if br.Buffered() > 0 {