liuxiaolong
2020-09-16 4501e38ca66f09b35aaaf43fa5a316554930fcf4
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
// +build fuzz
 
package dns
 
import "strings"
 
func Fuzz(data []byte) int {
    msg := new(Msg)
 
    if err := msg.Unpack(data); err != nil {
        return 0
    }
    if _, err := msg.Pack(); err != nil {
        return 0
    }
 
    return 1
}
 
func FuzzNewRR(data []byte) int {
    str := string(data)
    // Do not fuzz lines that include the $INCLUDE keyword and hint the fuzzer
    // at avoiding them.
    // See GH#1025 for context.
    if strings.Contains(strings.ToUpper(str), "$INCLUDE") {
        return -1
    }
    if _, err := NewRR(str); err != nil {
        return 0
    }
    return 1
}