zishuo/rtsp-simple-server
1
0
Fork 0
mirror of https://github.com/aler9/rtsp-simple-server synced 2025-11-02 20:04:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

improve fuzz tests (#3596)

Browse Source
This commit is contained in:
Alessandro Ros
2024-08-01 17:01:56 +02:00
committed by GitHub
parent 59ae3add7e
commit c9a938a501
5 changed files with 43 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
internal/protocols/rtmp/message/reader_test.go
View File

@@ -297,9 +297,18 @@ func FuzzReader(f *testing.F) {
0x01, 0x00, 0x00, 0x00, 0x88, 0x68, 0x76, 0x63,
0x31, 0x01, 0x02, 0x03,
})
f.Fuzz(func(_ *testing.T, b []byte) {
bc := bytecounter.NewReader(bytes.NewReader(b))
r := NewReader(bc, bc, nil)
r.Read() //nolint:errcheck
bcr := bytecounter.NewReader(bytes.NewReader(b))
r := NewReader(bcr, bcr, nil)
var buf bytes.Buffer
bcw := bytecounter.NewWriter(&buf)
w := NewWriter(bcw, bcw, true)
msg, err := r.Read()
if err == nil {
w.Write(msg) //nolint:errcheck
}
})
}