diff options
Diffstat (limited to 'prog/size_test.go')
| -rw-r--r-- | prog/size_test.go | 123 |
1 files changed, 123 insertions, 0 deletions
diff --git a/prog/size_test.go b/prog/size_test.go new file mode 100644 index 000000000..41068e713 --- /dev/null +++ b/prog/size_test.go @@ -0,0 +1,123 @@ +// 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) { + rs, iters := initTest(t) + for i := 0; i < iters; i++ { + p := Generate(rs, 10, nil) + data0 := p.Serialize() + for _, call := range p.Calls { + assignSizesCall(call) + assignTypeAndDir(call) + } + if data1 := p.Serialize(); !bytes.Equal(data0, data1) { + t.Fatalf("different lens assigned, initial: %v, new: %v", data0, data1) + } + for try := 0; try <= 10; try++ { + p.Mutate(rs, 10, nil) + data0 := p.Serialize() + for _, call := range p.Calls { + assignSizesCall(call) + assignTypeAndDir(call) + } + if data1 := p.Serialize(); !bytes.Equal(data0, data1) { + t.Fatalf("different lens assigned, initial: %v, new: %v", data0, data1) + } + } + } +} + +func TestAssignSize(t *testing.T) { + tests := []struct { + unsizedProg string + sizedProg string + }{ + { + "syz_test$length0(&(0x7f0000000000)={0xff, 0x0})", + "syz_test$length0(&(0x7f0000000000)={0xff, 0x2})", + }, + { + "syz_test$length1(&(0x7f0000001000)={0xff, 0x0})", + "syz_test$length1(&(0x7f0000001000)={0xff, 0x4})", + }, + { + "syz_test$length2(&(0x7f0000001000)={0xff, 0x0})", + "syz_test$length2(&(0x7f0000001000)={0xff, 0x8})", + }, + { + "syz_test$length3(&(0x7f0000005000)={0xff, 0x0, 0x0})", + "syz_test$length3(&(0x7f0000005000)={0xff, 0x4, 0x2})", + }, + { + "syz_test$length4(&(0x7f0000003000)={0x0, 0x0})", + "syz_test$length4(&(0x7f0000003000)={0x2, 0x2})", + }, + { + "syz_test$length5(&(0x7f0000002000)={0xff, 0x0})", + "syz_test$length5(&(0x7f0000002000)={0xff, 0x4})", + }, + { + "syz_test$length6(&(0x7f0000002000)={[0xff, 0xff, 0xff, 0xff], 0x0})", + "syz_test$length6(&(0x7f0000002000)={[0xff, 0xff, 0xff, 0xff], 0x4})", + }, + { + "syz_test$length7(&(0x7f0000003000)={[0xff, 0xff, 0xff, 0xff], 0x0})", + "syz_test$length7(&(0x7f0000003000)={[0xff, 0xff, 0xff, 0xff], 0x8})", + }, + { + "syz_test$length8(&(0x7f000001f000)={0x00, {0xff, 0x0, 0x00, [0xff, 0xff, 0xff]}, [{0xff, 0x0, 0x00, [0xff, 0xff, 0xff]}], 0x00, 0x0, [0xff, 0xff]})", + "syz_test$length8(&(0x7f000001f000)={0x32, {0xff, 0x1, 0x10, [0xff, 0xff, 0xff]}, [{0xff, 0x1, 0x10, [0xff, 0xff, 0xff]}], 0x10, 0x1, [0xff, 0xff]})", + }, + { + "syz_test$length9(&(0x7f000001f000)={&(0x7f0000000000/0x5000)=nil, (0x0000)})", + "syz_test$length9(&(0x7f000001f000)={&(0x7f0000000000/0x5000)=nil, (0x5000)})", + }, + { + "syz_test$length10(&(0x7f0000000000/0x5000)=nil, (0x0000))", + "syz_test$length10(&(0x7f0000000000/0x5000)=nil, (0x5000))", + }, + { + "syz_test$length11(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x00)", + "syz_test$length11(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x30)", + }, + { + "syz_test$length12(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x00)", + "syz_test$length12(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, 0x30)", + }, + { + "syz_test$length13(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x00)", + "syz_test$length13(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x30)", + }, + { + "syz_test$length14(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x00)", + "syz_test$length14(&(0x7f0000000000)={0xff, 0xff, [0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff, 0xff]}, &(0x7f0000001000)=0x30)", + }, + { + "syz_test$length15(0xff, 0x0)", + "syz_test$length15(0xff, 0x2)", + }, + } + + for i, test := range tests { + p, err := Deserialize([]byte(test.unsizedProg)) + if err != nil { + t.Fatalf("failed to deserialize prog %v: %v", i, err) + } + for _, call := range p.Calls { + assignSizesCall(call) + assignTypeAndDir(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) + } + } +} |