Ver código fonte

use the version from the read header version when parsing the frames for fuzzing

Chris Bannister 10 anos atrás
pai
commit
7c553284a1
2 arquivos alterados com 3 adições e 4 exclusões
  1. 2 3
      frame_test.go
  2. 1 1
      fuzz.go

+ 2 - 3
frame_test.go

@@ -28,13 +28,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

+ 1 - 1
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