فهرست منبع

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

Chris Bannister 11 سال پیش
والد
کامیت
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
 		var bw bytes.Buffer
 
 
 		r := bytes.NewReader(test)
 		r := bytes.NewReader(test)
-
-		head, err := readHeader(r, make([]byte, 8))
+		head, err := readHeader(r, make([]byte, 9))
 		if err != nil {
 		if err != nil {
 			continue
 			continue
 		}
 		}
 
 
-		framer := newFramer(r, &bw, nil, 2)
+		framer := newFramer(r, &bw, nil, byte(head.version))
 		err = framer.readFrame(&head)
 		err = framer.readFrame(&head)
 		if err != nil {
 		if err != nil {
 			continue
 			continue

+ 1 - 1
fuzz.go

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