diff options
Diffstat (limited to 'prog/encodingexec_test.go')
| -rw-r--r-- | prog/encodingexec_test.go | 280 |
1 files changed, 153 insertions, 127 deletions
diff --git a/prog/encodingexec_test.go b/prog/encodingexec_test.go index 29715e654..8a0325f0a 100644 --- a/prog/encodingexec_test.go +++ b/prog/encodingexec_test.go @@ -7,6 +7,7 @@ import ( "bytes" "encoding/binary" "fmt" + "reflect" "testing" ) @@ -15,33 +16,18 @@ func TestSerializeForExecRandom(t *testing.T) { buf := make([]byte, ExecBufferSize) for i := 0; i < iters; i++ { p := target.Generate(rs, 10, nil) - if _, err := p.SerializeForExec(buf, i%16); err != nil { + n, err := p.SerializeForExec(buf, i%16) + if err != nil { t.Fatalf("failed to serialize: %v", err) } + _, err = target.DeserializeExec(buf[:n]) + if err != nil { + t.Fatal(err) + } } } func TestSerializeForExec(t *testing.T) { - // A brief recap of exec format. - // Exec format is an sequence of uint64's which encodes a sequence of calls. - // The sequence is terminated by a speciall call ExecInstrEOF. - // Each call is (call ID, number of arguments, arguments...). - // Each argument is (type, size, value). - // There are 3 types of arguments: - // - ExecArgConst: value is const value - // - ExecArgResult: value is index of a call whose result we want to reference - // - ExecArgData: value is a binary blob (represented as ]size/8[ uint64's) - // There are 2 other special call: - // - ExecInstrCopyin: copies its second argument into address specified by first argument - // - ExecInstrCopyout: reads value at address specified by first argument (result can be referenced by ExecArgResult) - const ( - instrEOF = uint64(ExecInstrEOF) - instrCopyin = uint64(ExecInstrCopyin) - instrCopyout = uint64(ExecInstrCopyout) - argConst = uint64(ExecArgConst) - argResult = uint64(ExecArgResult) - argData = uint64(ExecArgData) - ) target := initTargetTest(t, "test", "64") var ( dataOffset = target.DataOffset @@ -57,207 +43,241 @@ func TestSerializeForExec(t *testing.T) { tests := []struct { prog string serialized []uint64 + decoded *ExecProg }{ { "syz_test()", []uint64{ callID("syz_test"), 0, - instrEOF, + execInstrEOF, + }, + &ExecProg{ + Calls: []ExecCall{ + { + Meta: target.SyscallMap["syz_test"], + Index: 0, + }, + }, + NumVars: 1, }, }, { "syz_test$int(0x1, 0x2, 0x3, 0x4, 0x5)", []uint64{ - callID("syz_test$int"), 5, argConst, 8, 1, 0, 0, argConst, 1, 2, 0, 0, argConst, 2, 3, 0, 0, argConst, 4, 4, 0, 0, argConst, 8, 5, 0, 0, - instrEOF, + callID("syz_test$int"), 5, + execArgConst, 8, 1, 0, 0, + execArgConst, 1, 2, 0, 0, + execArgConst, 2, 3, 0, 0, + execArgConst, 4, 4, 0, 0, + execArgConst, 8, 5, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align0(&(0x7f0000000000)={0x1, 0x2, 0x3, 0x4, 0x5})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 2, 1, 0, 0, - instrCopyin, dataOffset + 4, argConst, 4, 2, 0, 0, - instrCopyin, dataOffset + 8, argConst, 1, 3, 0, 0, - instrCopyin, dataOffset + 10, argConst, 2, 4, 0, 0, - instrCopyin, dataOffset + 16, argConst, 8, 5, 0, 0, - callID("syz_test$align0"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 2, 1, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 4, 2, 0, 0, + execInstrCopyin, dataOffset + 8, execArgConst, 1, 3, 0, 0, + execInstrCopyin, dataOffset + 10, execArgConst, 2, 4, 0, 0, + execInstrCopyin, dataOffset + 16, execArgConst, 8, 5, 0, 0, + callID("syz_test$align0"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align1(&(0x7f0000000000)={0x1, 0x2, 0x3, 0x4, 0x5})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 2, 1, 0, 0, - instrCopyin, dataOffset + 2, argConst, 4, 2, 0, 0, - instrCopyin, dataOffset + 6, argConst, 1, 3, 0, 0, - instrCopyin, dataOffset + 7, argConst, 2, 4, 0, 0, - instrCopyin, dataOffset + 9, argConst, 8, 5, 0, 0, - callID("syz_test$align1"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 2, 1, 0, 0, + execInstrCopyin, dataOffset + 2, execArgConst, 4, 2, 0, 0, + execInstrCopyin, dataOffset + 6, execArgConst, 1, 3, 0, 0, + execInstrCopyin, dataOffset + 7, execArgConst, 2, 4, 0, 0, + execInstrCopyin, dataOffset + 9, execArgConst, 8, 5, 0, 0, + callID("syz_test$align1"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align2(&(0x7f0000000000)={0x42, {[0x43]}, {[0x44]}})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 1, argConst, 2, 0x43, 0, 0, - instrCopyin, dataOffset + 4, argConst, 2, 0x44, 0, 0, - callID("syz_test$align2"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 1, execArgConst, 2, 0x43, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 2, 0x44, 0, 0, + callID("syz_test$align2"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align3(&(0x7f0000000000)={0x42, {0x43}, {0x44}})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 1, argConst, 1, 0x43, 0, 0, - instrCopyin, dataOffset + 4, argConst, 1, 0x44, 0, 0, - callID("syz_test$align3"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 1, execArgConst, 1, 0x43, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 1, 0x44, 0, 0, + callID("syz_test$align3"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align4(&(0x7f0000000000)={{0x42, 0x43}, 0x44})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 1, argConst, 2, 0x43, 0, 0, - instrCopyin, dataOffset + 4, argConst, 1, 0x44, 0, 0, - callID("syz_test$align4"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 1, execArgConst, 2, 0x43, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 1, 0x44, 0, 0, + callID("syz_test$align4"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align5(&(0x7f0000000000)={{0x42, []}, {0x43, [0x44, 0x45, 0x46]}, 0x47})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 8, 0x42, 0, 0, - instrCopyin, dataOffset + 8, argConst, 8, 0x43, 0, 0, - instrCopyin, dataOffset + 16, argConst, 2, 0x44, 0, 0, - instrCopyin, dataOffset + 18, argConst, 2, 0x45, 0, 0, - instrCopyin, dataOffset + 20, argConst, 2, 0x46, 0, 0, - instrCopyin, dataOffset + 22, argConst, 1, 0x47, 0, 0, - callID("syz_test$align5"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 8, 0x42, 0, 0, + execInstrCopyin, dataOffset + 8, execArgConst, 8, 0x43, 0, 0, + execInstrCopyin, dataOffset + 16, execArgConst, 2, 0x44, 0, 0, + execInstrCopyin, dataOffset + 18, execArgConst, 2, 0x45, 0, 0, + execInstrCopyin, dataOffset + 20, execArgConst, 2, 0x46, 0, 0, + execInstrCopyin, dataOffset + 22, execArgConst, 1, 0x47, 0, 0, + callID("syz_test$align5"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$align6(&(0x7f0000000000)={0x42, [0x43]})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 4, argConst, 4, 0x43, 0, 0, - callID("syz_test$align6"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 4, 0x43, 0, 0, + callID("syz_test$align6"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$union0(&(0x7f0000000000)={0x1, @f2=0x2})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 8, 1, 0, 0, - instrCopyin, dataOffset + 8, argConst, 1, 2, 0, 0, - callID("syz_test$union0"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 8, 1, 0, 0, + execInstrCopyin, dataOffset + 8, execArgConst, 1, 2, 0, 0, + callID("syz_test$union0"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$union1(&(0x7f0000000000)={@f1=0x42, 0x43})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 4, 0x42, 0, 0, - instrCopyin, dataOffset + 8, argConst, 1, 0x43, 0, 0, - callID("syz_test$union1"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 4, 0x42, 0, 0, + execInstrCopyin, dataOffset + 8, execArgConst, 1, 0x43, 0, 0, + callID("syz_test$union1"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$union2(&(0x7f0000000000)={@f1=0x42, 0x43})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 4, 0x42, 0, 0, - instrCopyin, dataOffset + 4, argConst, 1, 0x43, 0, 0, - callID("syz_test$union2"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 4, 0x42, 0, 0, + execInstrCopyin, dataOffset + 4, execArgConst, 1, 0x43, 0, 0, + callID("syz_test$union2"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$array0(&(0x7f0000000000)={0x1, [@f0=0x2, @f1=0x3], 0x4})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 1, 0, 0, - instrCopyin, dataOffset + 1, argConst, 2, 2, 0, 0, - instrCopyin, dataOffset + 3, argConst, 8, 3, 0, 0, - instrCopyin, dataOffset + 11, argConst, 8, 4, 0, 0, - callID("syz_test$array0"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 1, 0, 0, + execInstrCopyin, dataOffset + 1, execArgConst, 2, 2, 0, 0, + execInstrCopyin, dataOffset + 3, execArgConst, 8, 3, 0, 0, + execInstrCopyin, dataOffset + 11, execArgConst, 8, 4, 0, 0, + callID("syz_test$array0"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$array1(&(0x7f0000000000)={0x42, \"0102030405\"})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 1, argData, 5, 0x0504030201, - callID("syz_test$array1"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 1, execArgData, 5, 0x0504030201, + callID("syz_test$array1"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$array2(&(0x7f0000000000)={0x42, \"aaaaaaaabbbbbbbbccccccccdddddddd\", 0x43})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 2, 0x42, 0, 0, - instrCopyin, dataOffset + 2, argData, 16, 0xbbbbbbbbaaaaaaaa, 0xddddddddcccccccc, - instrCopyin, dataOffset + 18, argConst, 2, 0x43, 0, 0, - callID("syz_test$array2"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 2, 0x42, 0, 0, + execInstrCopyin, dataOffset + 2, execArgData, 16, 0xbbbbbbbbaaaaaaaa, 0xddddddddcccccccc, + execInstrCopyin, dataOffset + 18, execArgConst, 2, 0x43, 0, 0, + callID("syz_test$array2"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$end0(&(0x7f0000000000)={0x42, 0x42, 0x42, 0x42})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 1, 0x42, 0, 0, - instrCopyin, dataOffset + 1, argConst, 2, 0x4200, 0, 0, - instrCopyin, dataOffset + 3, argConst, 4, 0x42000000, 0, 0, - instrCopyin, dataOffset + 7, argConst, 8, 0x4200000000000000, 0, 0, - callID("syz_test$end0"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 1, 0x42, 0, 0, + execInstrCopyin, dataOffset + 1, execArgConst, 2, 0x4200, 0, 0, + execInstrCopyin, dataOffset + 3, execArgConst, 4, 0x42000000, 0, 0, + execInstrCopyin, dataOffset + 7, execArgConst, 8, 0x4200000000000000, 0, 0, + callID("syz_test$end0"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$end1(&(0x7f0000000000)={0xe, 0x42, 0x1})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 2, 0x0e00, 0, 0, - instrCopyin, dataOffset + 2, argConst, 4, 0x42000000, 0, 0, - instrCopyin, dataOffset + 6, argConst, 8, 0x0100000000000000, 0, 0, - callID("syz_test$end1"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 2, 0x0e00, 0, 0, + execInstrCopyin, dataOffset + 2, execArgConst, 4, 0x42000000, 0, 0, + execInstrCopyin, dataOffset + 6, execArgConst, 8, 0x0100000000000000, 0, 0, + callID("syz_test$end1"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$bf0(&(0x7f0000000000)={0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0x42, 0x42})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 2, 0x42, 0, 10, - instrCopyin, dataOffset + 8, argConst, 8, 0x42, 0, 0, - instrCopyin, dataOffset + 16, argConst, 2, 0x42, 0, 5, - instrCopyin, dataOffset + 16, argConst, 2, 0x42, 5, 6, - instrCopyin, dataOffset + 20, argConst, 4, 0x42, 0, 15, - instrCopyin, dataOffset + 24, argConst, 2, 0x42, 0, 11, - instrCopyin, dataOffset + 26, argConst, 2, 0x4200, 0, 11, - instrCopyin, dataOffset + 28, argConst, 1, 0x42, 0, 0, - callID("syz_test$bf0"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 2, 0x42, 0, 10, + execInstrCopyin, dataOffset + 8, execArgConst, 8, 0x42, 0, 0, + execInstrCopyin, dataOffset + 16, execArgConst, 2, 0x42, 0, 5, + execInstrCopyin, dataOffset + 16, execArgConst, 2, 0x42, 5, 6, + execInstrCopyin, dataOffset + 20, execArgConst, 4, 0x42, 0, 15, + execInstrCopyin, dataOffset + 24, execArgConst, 2, 0x42, 0, 11, + execInstrCopyin, dataOffset + 26, execArgConst, 2, 0x4200, 0, 11, + execInstrCopyin, dataOffset + 28, execArgConst, 1, 0x42, 0, 0, + callID("syz_test$bf0"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$bf1(&(0x7f0000000000)={{0x42, 0x42, 0x42}, 0x42})", []uint64{ - instrCopyin, dataOffset + 0, argConst, 4, 0x42, 0, 10, - instrCopyin, dataOffset + 0, argConst, 4, 0x42, 10, 10, - instrCopyin, dataOffset + 0, argConst, 4, 0x42, 20, 10, - instrCopyin, dataOffset + 4, argConst, 1, 0x42, 0, 0, - callID("syz_test$bf1"), 1, argConst, ptrSize, dataOffset, 0, 0, - instrEOF, + execInstrCopyin, dataOffset + 0, execArgConst, 4, 0x42, 0, 10, + execInstrCopyin, dataOffset + 0, execArgConst, 4, 0x42, 10, 10, + execInstrCopyin, dataOffset + 0, execArgConst, 4, 0x42, 20, 10, + execInstrCopyin, dataOffset + 4, execArgConst, 1, 0x42, 0, 0, + callID("syz_test$bf1"), 1, execArgConst, ptrSize, dataOffset, 0, 0, + execInstrEOF, }, + nil, }, { "syz_test$res1(0xffff)", []uint64{ - callID("syz_test$res1"), 1, argConst, 4, 0xffff, 0, 0, - instrEOF, + callID("syz_test$res1"), 1, execArgConst, 4, 0xffff, 0, 0, + execInstrEOF, }, + nil, }, } @@ -269,15 +289,13 @@ func TestSerializeForExec(t *testing.T) { if err != nil { t.Fatalf("failed to deserialize prog %v: %v", i, err) } - if _, err := p.SerializeForExec(buf, i%16); err != nil { + n, err := p.SerializeForExec(buf, i%16) + if err != nil { t.Fatalf("failed to serialize: %v", err) } w := new(bytes.Buffer) binary.Write(w, binary.LittleEndian, test.serialized) - data := buf - if len(data) > len(w.Bytes()) { - data = data[:len(w.Bytes())] - } + data := buf[:n] if !bytes.Equal(data, w.Bytes()) { got := make([]uint64, len(data)/8) binary.Read(bytes.NewReader(data), binary.LittleEndian, &got) @@ -285,7 +303,15 @@ func TestSerializeForExec(t *testing.T) { t.Logf("got: %v", got) t.Fatalf("mismatch") } - + decoded, err := target.DeserializeExec(data) + if err != nil { + t.Fatal(err) + } + if test.decoded != nil && !reflect.DeepEqual(decoded, *test.decoded) { + t.Logf("want: %#v", *test.decoded) + t.Logf("got: %#v", decoded) + t.Fatalf("decoded mismatch") + } }) } } |