pkg/tailfile/fuzz_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

package tailfile

import (
	"os"
	"path/filepath"
	"testing"

	fuzz "github.com/AdaLogics/go-fuzz-headers"
)

func FuzzTailfile(f *testing.F) {
	f.Fuzz(func(t *testing.T, data []byte) {
		if len(data) < 5 {
			return
		}
		ff := fuzz.NewConsumer(data)
		n, err := ff.GetUint64()
		if err != nil {
			return
		}
		fileBytes, err := ff.GetBytes()
		if err != nil {
			return
		}
		tempDir := t.TempDir()
		fil, err := os.Create(filepath.Join(tempDir, "tailFile"))
		if err != nil {
			return
		}
		defer fil.Close()

		_, err = fil.Write(fileBytes)
		if err != nil {
			return
		}
		fil.Seek(0, 0)
		_, _ = TailFile(fil, int(n))
	})
}