Fixing sizing issue

Signed-off-by: Shivansh Vij <[email protected]>
loopholelabs · Oct 12, 2023 · ff0fe6e · ff0fe6e
1 parent 145beac
commit ff0fe6e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/buffer.go b/buffer.go
@@ -33,9 +33,9 @@ func (buf *Buffer) Reset() {
 func (buf *Buffer) Grow(n int) {
 	if cap(buf.b)-buf.offset < n {
 		if cap(buf.b) < n {
-			buf.b = append(buf.b[:buf.offset], make([]byte, n)...)
+			buf.b = append(buf.b[:buf.offset], make([]byte, n+cap(buf.b)-buf.offset)...)
 		} else {
-			buf.b = append(buf.b[:buf.offset], make([]byte, cap(buf.b))...)
+			buf.b = append(buf.b[:buf.offset], make([]byte, cap(buf.b)*2-buf.offset)...)
 		}
 	}
 }