Jan 9, 2021 · Jul 4, 2020 · Jan 9, 2021 · Jan 9, 2021
diff --git a/netconn.go b/netconn.go
 //
 // A received StatusNormalClosure or StatusGoingAway close frame will be translated to
 // io.EOF when reading.
 //
 // Furthermore, the ReadLimit is set to -1 to disable it.
 func NetConn(ctx context.Context, c *Conn, msgType MessageType) net.Conn {
 c.SetReadLimit(-1)

 nc := &netConn{
 c:       c,
 msgType: msgType,
diff --git a/read.go b/read.go
 // By default, the connection has a message read limit of 32768 bytes.
 //
 // When the limit is hit, the connection will be closed with StatusMessageTooBig.
 //
 // Set to -1 to disable.
 func (c *Conn) SetReadLimit(n int64) {
 // We add read one more byte than the limit in case
 // there is a fin frame that needs to be read.
 c.msgReader.limitReader.limit.Store(n + 1)
 if n >= 0 {
 // We read one more byte than the limit in case
 // there is a fin frame that needs to be read.
 n++
 }

 c.msgReader.limitReader.limit.Store(n)
 }

 const defaultReadLimit = 32768
 }

 func (lr *limitReader) Read(p []byte) (int, error) {
 if lr.n <= 0 {
 if lr.n < 0 {
 return lr.r.Read(p)
 }

 if lr.n == 0 {
 err := fmt.Errorf("read limited at %v bytes", lr.limit.Load())
 lr.c.writeError(StatusMessageTooBig, err)
 return 0, err
 }
 n, err := lr.r.Read(p)
 lr.n -= int64(n)
 if lr.n < 0 {
 lr.n = 0
 }
 return n, err
 }
Original file line number	Diff line number	Diff line change
Expand Up		@@ -38,7 +38,11 @@ import (
		//
		// A received StatusNormalClosure or StatusGoingAway close frame will be translated to
		// io.EOF when reading.
		//
		// Furthermore, the ReadLimit is set to -1 to disable it.
		func NetConn(ctx context.Context, c *Conn, msgType MessageType) net.Conn {
		c.SetReadLimit(-1)

		nc := &netConn{
		c: c,
		msgType: msgType,
Expand Down
Original file line number	Diff line number	Diff line change
Expand Up		@@ -74,10 +74,16 @@ func (c *Conn) CloseRead(ctx context.Context) context.Context {
		// By default, the connection has a message read limit of 32768 bytes.
		//
		// When the limit is hit, the connection will be closed with StatusMessageTooBig.
		//
		// Set to -1 to disable.
		func (c *Conn) SetReadLimit(n int64) {
		// We add read one more byte than the limit in case
		// there is a fin frame that needs to be read.
		c.msgReader.limitReader.limit.Store(n + 1)
		if n >= 0 {
		// We read one more byte than the limit in case
		// there is a fin frame that needs to be read.
		n++
		}

		c.msgReader.limitReader.limit.Store(n)
		}

		const defaultReadLimit = 32768
Expand DownExpand Up		@@ -455,7 +461,11 @@ func (lr *limitReader) reset(r io.Reader) {
		}

		func (lr *limitReader) Read(p []byte) (int, error) {
		if lr.n <= 0 {
		if lr.n < 0 {
		return lr.r.Read(p)
		}

		if lr.n == 0 {
		err := fmt.Errorf("read limited at %v bytes", lr.limit.Load())
		lr.c.writeError(StatusMessageTooBig, err)
		return 0, err
Expand All		@@ -466,6 +476,9 @@ func (lr *limitReader) Read(p []byte) (int, error) {
		}
		n, err := lr.r.Read(p)
		lr.n -= int64(n)
		if lr.n < 0 {
		lr.n = 0
		}
		return n, err
		}

Expand Down