Merge pull request apache#410 from Zariel/fix-fuzz-version

Fix fuzz version
jameshartig · Jun 12, 2015 · c384b1f · c384b1f
2 parents ff9b40c + 7c55328
commit c384b1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/frame_test.go b/frame_test.go
@@ -29,13 +29,12 @@ func TestFuzzBugs(t *testing.T) {
 		var bw bytes.Buffer
 
 		r := bytes.NewReader(test)
-
-		head, err := readHeader(r, make([]byte, 8))
+		head, err := readHeader(r, make([]byte, 9))
 		if err != nil {
 			continue
 		}
 
-		framer := newFramer(r, &bw, nil, 2)
+		framer := newFramer(r, &bw, nil, byte(head.version))
 		err = framer.readFrame(&head)
 		if err != nil {
 			continue

diff --git a/fuzz.go b/fuzz.go
@@ -14,7 +14,7 @@ func Fuzz(data []byte) int {
 		return 0
 	}
 
-	framer := newFramer(r, &bw, nil, 3)
+	framer := newFramer(r, &bw, nil, byte(head.version))
 	err = framer.readFrame(&head)
 	if err != nil {
 		return 0