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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
| // Copyright 2016 syzkaller project authors. All rights reserved.
| // Use of this source code is governed by Apache 2 LICENSE that can be found in the LICENSE file.
|
| package prog
|
| import (
| "bytes"
| "strings"
| "testing"
| )
|
| func TestAssignSizeRandom(t *testing.T) {
| target, rs, iters := initTest(t)
| for i := 0; i < iters; i++ {
| p := target.Generate(rs, 10, nil)
| data0 := p.Serialize()
| for _, call := range p.Calls {
| target.assignSizesCall(call)
| }
| if data1 := p.Serialize(); !bytes.Equal(data0, data1) {
| t.Fatalf("different lens assigned, initial:\n%s\nnew:\n%s\n", data0, data1)
| }
| p.Mutate(rs, 10, nil, nil)
| p.Serialize()
| for _, call := range p.Calls {
| target.assignSizesCall(call)
| }
| }
| }
|
| func TestAssignSize(t *testing.T) {
| target := initTargetTest(t, "test", "64")
| // nolint: lll
| tests := []struct {
| unsizedProg string
| sizedProg string
| }{
| {
| "test$length0(&(0x7f0000000000)={0xff, 0x0})",
| "test$length0(&(0x7f0000000000)={0xff, 0x2})",
| },
| {
| "test$length1(&(0x7f0000001000)={0xff, 0x0})",
| "test$length1(&(0x7f0000001000)={0xff, 0x4})",
| },
| {
| "test$length2(&(0x7f0000001000)={0xff, 0x0})",
| "test$length2(&(0x7f0000001000)={0xff, 0x8})",
| },
| {
| "test$length3(&(0x7f0000005000)={0xff, 0x0, 0x0})",
| "test$length3(&(0x7f0000005000)={0xff, 0x4, 0x2})",
| },
| {
| "test$length4(&(0x7f0000003000)={0x0, 0x0})",
| "test$length4(&(0x7f0000003000)={0x2, 0x2})",
| },
| {
| "test$length5(&(0x7f0000002000)={0xff, 0x0})",
| "test$length5(&(0x7f0000002000)={0xff, 0x4})",
| },
| {
| "test$length6(&(0x7f0000002000)={[0xff, 0xff, 0xff, 0xff], 0x0})",
| "test$length6(&(0x7f0000002000)={[0xff, 0xff, 0xff, 0xff], 0x4})",
| },
| {
| "test$length7(&(0x7f0000003000)={[0xff, 0xff, 0xff, 0xff], 0x0})",
| "test$length7(&(0x7f0000003000)={[0xff, 0xff, 0xff, 0xff], 0x8})",
| },
| {
| "test$length8(&(0x7f000001f000)={0x00, {0xff, 0x0, 0x00, [0xff, 0xff, 0xff]}, [{0xff, 0x0, 0x00, [0xff, 0xff, 0xff]}], 0x00, 0x0, [0xff, 0xff]})",
| "test$length8(&(0x7f000001f000)={0x32, {0xff, 0x1, 0x10, [0xff, 0xff, 0xff]}, [{0xff, 0x1, 0x10, [0xff, 0xff, 0xff]}], 0x10, 0x1, [0xff, 0xff]})",
| },
| {
| "test$length9(&(0x7f000001f000)={&(0x7f0000000000/0x5000)=nil, 0x0000})",
| "test$length9(&(0x7f000001f000)={&(0x7f0000000000/0x5000)=nil, 0x5000})",
| },
| {
| "test$length10(&(0x7f0000000000/0x5000)=nil, 0x0000, 0x0000, 0x0000, 0x0000)",
| "test$length10(&(0x7f0000000000/0x5000)=nil, 0x5000, 0x5000, 0x2800, 0x1400)",
| },
| {
| "test$length11(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x00)",
| "test$length11(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x30)",
| },
| {
| "test$length12(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x00)",
| "test$length12(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x30)",
| },
| {
| "test$length13(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x00)",
| "test$length13(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x30)",
| },
| {
| "test$length14(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x00)",
| "test$length14(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x30)",
| },
| {
| "test$length15(0xff, 0x0)",
| "test$length15(0xff, 0x2)",
| },
| {
| "test$length16(&(0x7f0000000000)={[0x42, 0x42], 0xff, 0xff, 0xff, 0xff, 0xff})",
| "test$length16(&(0x7f0000000000)={[0x42, 0x42], 0x2, 0x10, 0x8, 0x4, 0x2})",
| },
| {
| "test$length17(&(0x7f0000000000)={0x42, 0xff, 0xff, 0xff, 0xff})",
| "test$length17(&(0x7f0000000000)={0x42, 0x8, 0x4, 0x2, 0x1})",
| },
| {
| "test$length18(&(0x7f0000000000)={0x42, 0xff, 0xff, 0xff, 0xff})",
| "test$length18(&(0x7f0000000000)={0x42, 0x8, 0x4, 0x2, 0x1})",
| },
| {
| "test$length19(&(0x7f0000000000)={{0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0xff}, 0xff, 0xff, 0xff})",
| "test$length19(&(0x7f0000000000)={{0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0x14}, 0x14, 0x14, 0x5})",
| },
| {
| "test$length20(&(0x7f0000000000)={{{0xff, 0xff, 0xff, 0xff}, 0xff, 0xff, 0xff}, 0xff, 0xff})",
| "test$length20(&(0x7f0000000000)={{{0x4, 0x4, 0x7, 0x9}, 0x7, 0x7, 0x9}, 0x9, 0x9})",
| },
| {
| "test$length21(&(0x7f0000000000)=0x0, 0x0)",
| "test$length21(&(0x7f0000000000), 0x40)",
| },
| {
| "test$length22(&(0x7f0000000000)='12345', 0x0)",
| "test$length22(&(0x7f0000000000)='12345', 0x28)",
| },
| {
| "test$length23(&(0x7f0000000000)={0x1, {0x2, 0x0}})",
| "test$length23(&(0x7f0000000000)={0x1, {0x2, 0x6}})",
| },
| {
| "test$length24(&(0x7f0000000000)={{0x0, {0x0}}, {0x0, {0x0}}})",
| "test$length24(&(0x7f0000000000)={{0x0, {0x8}}, {0x0, {0x10}}})",
| },
| {
| "test$length26(&(0x7f0000000000), 0x0)",
| "test$length26(&(0x7f0000000000), 0x8)",
| },
| {
| "test$length27(&(0x7f0000000000), 0x0)",
| "test$length27(&(0x7f0000000000), 0x2a)",
| },
| {
| "test$length28(&(0x7f0000000000), 0x0)",
| "test$length28(&(0x7f0000000000), 0x2a)",
| },
| {
| "test$length29(&(0x7f0000000000)={'./a\\x00', './b/c\\x00', 0x0, 0x0, 0x0})",
| "test$length29(&(0x7f0000000000)={'./a\\x00', './b/c\\x00', 0xa, 0x14, 0x21})",
| },
| }
|
| for i, test := range tests {
| p, err := target.Deserialize([]byte(test.unsizedProg))
| if err != nil {
| t.Fatalf("failed to deserialize prog %v: %v", i, err)
| }
| for _, call := range p.Calls {
| target.assignSizesCall(call)
| }
| p1 := strings.TrimSpace(string(p.Serialize()))
| if p1 != test.sizedProg {
| t.Fatalf("failed to assign sizes in prog %v\ngot %v\nwant %v", i, p1, test.sizedProg)
| }
| }
| }
|
|
