summaryrefslogtreecommitdiffstats
path: root/src/encoding/csv/fuzz.go
blob: a03fa83d8ce3a7a7c2075369e95c0ea6e577b71d (plain)
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
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
// Copyright 2019 The Go Authors. All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

//go:build gofuzz
// +build gofuzz

package csv

import (
	"bytes"
	"fmt"
	"reflect"
)

func Fuzz(data []byte) int {
	score := 0
	buf := new(bytes.Buffer)

	for _, tt := range []Reader{
		{},
		{Comma: ';'},
		{Comma: '\t'},
		{LazyQuotes: true},
		{TrimLeadingSpace: true},
		{Comment: '#'},
		{Comment: ';'},
	} {
		r := NewReader(bytes.NewReader(data))
		r.Comma = tt.Comma
		r.Comment = tt.Comment
		r.LazyQuotes = tt.LazyQuotes
		r.TrimLeadingSpace = tt.TrimLeadingSpace

		records, err := r.ReadAll()
		if err != nil {
			continue
		}
		score = 1

		buf.Reset()
		w := NewWriter(buf)
		w.Comma = tt.Comma
		err = w.WriteAll(records)
		if err != nil {
			fmt.Printf("writer  = %#v\n", w)
			fmt.Printf("records = %v\n", records)
			panic(err)
		}

		r = NewReader(buf)
		r.Comma = tt.Comma
		r.Comment = tt.Comment
		r.LazyQuotes = tt.LazyQuotes
		r.TrimLeadingSpace = tt.TrimLeadingSpace
		result, err := r.ReadAll()
		if err != nil {
			fmt.Printf("reader  = %#v\n", r)
			fmt.Printf("records = %v\n", records)
			panic(err)
		}

		if !reflect.DeepEqual(records, result) {
			fmt.Println("records = \n", records)
			fmt.Println("result  = \n", records)
			panic("not equal")
		}
	}

	return score
}