alpine: add max metadata size to fuzzer (#1571)

Signed-off-by: AdamKorcz <adam@adalogics.com>
sigstore · Jun 30, 2023 · 7616da1 · 7616da1
1 parent 9c13e97
commit 7616da1
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/pkg/types/alpine/apk_test.go b/pkg/types/alpine/apk_test.go
@@ -53,8 +53,7 @@ func TestAlpinePackage(t *testing.T) {
 }
 
 func TestAlpineMetadataSize(t *testing.T) {
-	os.Setenv("MAX_APK_METADATA_SIZE", "10")
-	viper.AutomaticEnv()
+	viper.Set("max_apk_metadata_size", 10)
 
 	inputArchive, err := os.Open("tests/test_alpine.apk")
 	if err != nil {

diff --git a/pkg/types/alpine/fuzz_test.go b/pkg/types/alpine/fuzz_test.go
@@ -17,12 +17,22 @@ package alpine
 
 import (
 	"bytes"
+	"fmt"
 	"testing"
 
+	"github.com/spf13/viper"
+
 	fuzz "github.com/AdamKorcz/go-fuzz-headers-1"
 	utils "github.com/sigstore/rekor/pkg/fuzz"
 )
 
+func init() {
+	viper.Set("max_apk_metadata_size", 60000)
+	if viper.GetUint64("max_apk_metadata_size") != 60000 {
+		panic(fmt.Sprintf("max metadata size is not defined: %d", viper.GetUint64("max_apk_metadata_size")))
+	}
+}
+
 // FuzzPackageUnmarshal implements the fuzz test
 func FuzzPackageUnmarshal(f *testing.F) {
 	f.Fuzz(func(t *testing.T, data []byte) {