s2: Expand destination slice if big enough. (#204)

klauspost · Jan 11, 2020 · da94045 · da94045
1 parent 5c1f19d
commit da94045
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/s2/decode.go b/s2/decode.go
@@ -60,7 +60,7 @@ func Decode(dst, src []byte) ([]byte, error) {
 	if err != nil {
 		return nil, err
 	}
-	if dLen <= len(dst) {
+	if dLen <= cap(dst) {
 		dst = dst[:dLen]
 	} else {
 		dst = make([]byte, dLen)

diff --git a/s2/encode.go b/s2/encode.go
@@ -32,8 +32,10 @@ import (
 func Encode(dst, src []byte) []byte {
 	if n := MaxEncodedLen(len(src)); n < 0 {
 		panic(ErrTooLarge)
-	} else if len(dst) < n {
+	} else if cap(dst) < n {
 		dst = make([]byte, n)
+	} else {
+		dst = dst[:n]
 	}
 
 	// The block starts with the varint-encoded length of the decompressed bytes.