From 2fad63c189e58311318d6c19c56d8342c95d065f Mon Sep 17 00:00:00 2001 From: Jack Christensen Date: Sat, 26 Feb 2022 09:37:14 -0600 Subject: [PATCH] Set cap when returning slice from chunkReader --- pgproto3/chunkreader.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pgproto3/chunkreader.go b/pgproto3/chunkreader.go index 598aaa85..d17b93ea 100644 --- a/pgproto3/chunkreader.go +++ b/pgproto3/chunkreader.go @@ -41,7 +41,7 @@ func newChunkReader(r io.Reader, minBufSize int) *chunkReader { func (r *chunkReader) Next(n int) (buf []byte, err error) { // n bytes already in buf if (r.wp - r.rp) >= n { - buf = r.buf[r.rp : r.rp+n] + buf = r.buf[r.rp : r.rp+n : r.rp+n] r.rp += n r.resetBufIfEmpty() return buf, err @@ -63,7 +63,7 @@ func (r *chunkReader) Next(n int) (buf []byte, err error) { return nil, err } - buf = r.buf[r.rp : r.rp+n] + buf = r.buf[r.rp : r.rp+n : r.rp+n] r.rp += n r.resetBufIfEmpty() return buf, nil