فهرست منبع

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

Chris Bannister 10 سال پیش
والد
کامیت
7c553284a1
2فایلهای تغییر یافته به همراه3 افزوده شده و 4 حذف شده
  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