s2: Fix crash on small better buffers (#308)

klauspost · Jan 8, 2021 · b45090e · b45090e
1 parent 39a5f72
commit b45090e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/s2/encode_better.go b/s2/encode_better.go
@@ -63,6 +63,9 @@ func encodeBlockBetter(dst, src []byte) (d int) {
 	// lets us use a fast path for emitLiteral in the main loop, while we are
 	// looking for copies.
 	sLimit := len(src) - inputMargin
+	if len(src) < minNonLiteralBlockSize {
+		return 0
+	}
 
 	// Bail if we can't compress to at least this.
 	dstLimit := len(src) - len(src)>>5 - 5