Add assign len fields tests

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)
+		}
+	}
+}