fuzz: add .Delete fuzzing (#52)

* fuzz: add .Delete fuzzing * Update fuzz/delete/fuzz.go Remove spurious newline. Co-authored-by: Ismail Khoffi <[email protected]> Co-authored-by: Ismail Khoffi <[email protected]>
celestiaorg · Jul 29, 2021 · a8b51d5 · a8b51d5
1 parent 2e3797d
commit a8b51d5
Show file tree

Hide file tree

Showing 2 changed files with 38 additions and 1 deletion.
diff --git a/fuzz/delete/fuzz.go b/fuzz/delete/fuzz.go
@@ -0,0 +1,36 @@
+package delete
+
+import (
+	"bytes"
+	"crypto/sha256"
+
+	"github.com/celestiaorg/smt"
+)
+
+func Fuzz(data []byte) int {
+	if len(data) == 0 {
+		return -1
+	}
+
+	splits := bytes.Split(data, []byte("*"))
+	if len(splits) < 3 {
+		return -1
+	}
+
+	smn, smv := smt.NewSimpleMap(), smt.NewSimpleMap()
+	tree := smt.NewSparseMerkleTree(smn, smv, sha256.New())
+	for i := 0; i < len(splits)-1; i += 2 {
+		key, value := splits[i], splits[i+1]
+		tree.Update(key, value)
+	}
+
+	deleteKey := splits[len(splits)-1]
+	newRoot, err := tree.Delete(deleteKey)
+	if err != nil {
+		return 0
+	}
+	if len(newRoot) == 0 {
+		panic("newRoot is nil yet err==nil")
+	}
+	return 1
+}
diff --git a/oss-fuzz-build.sh b/oss-fuzz-build.sh
@@ -2,4 +2,5 @@
 
 export FUZZ_ROOT="github.com/celestiaorg/smt"
 
-compile_go_fuzzer "$FUZZ_ROOT"/fuzz Fuzz fuzz_basic_op fuzz
+compile_go_fuzzer "$FUZZ_ROOT"/fuzz Fuzz fuzz_basic_op fuzz
+compile_go_fuzzer "$FUZZ_ROOT"/fuzz/delete Fuzz fuzz_delete fuzz