[zstd_test] Add a test for ZipBomb payload on Decompress

See #60
DataDog · Apr 6, 2022 · eaf4b06 · eaf4b06
1 parent 393c3c1
commit eaf4b06
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/zstd_test.go b/zstd_test.go
@@ -2,6 +2,7 @@ package zstd
 
 import (
 	"bytes"
+	b64 "encoding/base64"
 	"errors"
 	"fmt"
 	"io/ioutil"
@@ -284,6 +285,14 @@ func TestLegacy(t *testing.T) {
 	}
 }
 
+func TestBadPayloadZipBomb(t *testing.T) {
+	payload, _ := b64.StdEncoding.DecodeString("KLUv/dcwMDAwMDAwMDAwMAAA")
+	_, err := Decompress(nil, payload)
+	if err.Error() != "Src size is incorrect" {
+		t.Fatal("zstd should detect that the size is incorrect")
+	}
+}
+
 func BenchmarkCompression(b *testing.B) {
 	if raw == nil {
 		b.Fatal(ErrNoPayloadEnv)