Get the raw connection

Author: dgrr

conn.go

@@ -14,6 +14,7 @@ import (
 // This handler is compatible with io.Writer.
 type Conn struct {
 	c  net.Conn
+	br *bufio.Reader
 	bw *bufio.Writer
 
 	input  chan *Frame
@@ -92,6 +93,7 @@ func (c *Conn) reset(conn net.Conn) {
 	c.MaxPayloadSize = DefaultPayloadSize
 	c.userValues = make(map[string]interface{})
 	c.c = conn
+	c.br = bufio.NewReader(conn)
 	c.bw = bufio.NewWriter(conn)
 }
 
@@ -105,7 +107,7 @@ func (c *Conn) readLoop() {
 		// if c.ReadTimeout != 0 {
 		// }
 
-		_, err := fr.ReadFrom(c.c)
+		_, err := fr.ReadFrom(c.br)
 		if err != nil {
 			select {
 			case c.errch <- closeError{err: err}:

server.go

@@ -194,6 +194,12 @@ func (s *Server) Upgrade(ctx *fasthttp.RequestCtx) {
 			})
 
 			ctx.Hijack(func(c net.Conn) {
+				if nc, ok := c.(interface {
+					UnsafeConn() net.Conn
+				}); ok {
+					c = nc.UnsafeConn()
+				}
+
 				conn := acquireConn(c)
 				conn.id = atomic.AddUint64(&s.nextID, 1)
 				// establishing default options