|
@@ -8,7 +8,6 @@ import (
|
|
|
"bufio"
|
|
"bufio"
|
|
|
"errors"
|
|
"errors"
|
|
|
"io"
|
|
"io"
|
|
|
- "net"
|
|
|
|
|
"net/http"
|
|
"net/http"
|
|
|
"net/url"
|
|
"net/url"
|
|
|
"strings"
|
|
"strings"
|
|
@@ -171,17 +170,12 @@ func (u *Upgrader) Upgrade(w http.ResponseWriter, r *http.Request, responseHeade
|
|
|
}
|
|
}
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
- var (
|
|
|
|
|
- netConn net.Conn
|
|
|
|
|
- err error
|
|
|
|
|
- )
|
|
|
|
|
-
|
|
|
|
|
h, ok := w.(http.Hijacker)
|
|
h, ok := w.(http.Hijacker)
|
|
|
if !ok {
|
|
if !ok {
|
|
|
return u.returnError(w, r, http.StatusInternalServerError, "websocket: response does not implement http.Hijacker")
|
|
return u.returnError(w, r, http.StatusInternalServerError, "websocket: response does not implement http.Hijacker")
|
|
|
}
|
|
}
|
|
|
var brw *bufio.ReadWriter
|
|
var brw *bufio.ReadWriter
|
|
|
- netConn, brw, err = h.Hijack()
|
|
|
|
|
|
|
+ netConn, brw, err := h.Hijack()
|
|
|
if err != nil {
|
|
if err != nil {
|
|
|
return u.returnError(w, r, http.StatusInternalServerError, err.Error())
|
|
return u.returnError(w, r, http.StatusInternalServerError, err.Error())
|
|
|
}
|
|
}
|
|
@@ -331,7 +325,7 @@ func IsWebSocketUpgrade(r *http.Request) bool {
|
|
|
tokenListContainsValue(r.Header, "Upgrade", "websocket")
|
|
tokenListContainsValue(r.Header, "Upgrade", "websocket")
|
|
|
}
|
|
}
|
|
|
|
|
|
|
|
-// bufioReader size returns the size of a bufio.Reader.
|
|
|
|
|
|
|
+// bufioReaderSize size returns the size of a bufio.Reader.
|
|
|
func bufioReaderSize(originalReader io.Reader, br *bufio.Reader) int {
|
|
func bufioReaderSize(originalReader io.Reader, br *bufio.Reader) int {
|
|
|
// This code assumes that peek on a reset reader returns
|
|
// This code assumes that peek on a reset reader returns
|
|
|
// bufio.Reader.buf[:0].
|
|
// bufio.Reader.buf[:0].
|