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
|
// Copyright 2022 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_test
import (
"flag"
"fmt"
"io"
"os"
"path/filepath"
"reflect"
"sort"
"strings"
"testing"
"github.com/google/go-cmp/cmp"
"github.com/google/syzkaller/pkg/osutil"
. "github.com/google/syzkaller/prog"
"github.com/google/syzkaller/sys/targets"
)
var flagUpdate = flag.Bool("update", false, "update test files accordingly to current results")
func TestForEachAsset(t *testing.T) {
target, err := GetTarget(targets.Linux, targets.AMD64)
if err != nil {
t.Fatal(err)
}
files, err := filepath.Glob(filepath.Join("testdata", "fs_images", "*.in"))
if err != nil {
t.Fatalf("directory read failed: %v", err)
}
allOutFiles, err := filepath.Glob(filepath.Join("testdata", "fs_images", "*.out*"))
if err != nil {
t.Fatalf("directory read failed: %v", err)
}
testedOutFiles := []string{}
for _, file := range files {
sourceProg, err := os.ReadFile(file)
if err != nil {
t.Fatal(err)
}
p, err := target.Deserialize(sourceProg, NonStrict)
if err != nil {
t.Fatalf("failed to deserialize %s: %s", file, err)
}
base := strings.TrimSuffix(file, ".in")
p.ForEachAsset(func(name string, typ AssetType, r io.Reader, c *Call) {
if typ != MountInRepro {
t.Fatalf("unknown asset type %v", typ)
}
if !strings.HasPrefix(c.Meta.Name, "syz_mount_image$") {
t.Fatalf("unexpected syscall name %v", c.Meta.Name)
}
testResult, err := io.ReadAll(r)
if err != nil {
t.Fatal(err)
}
outFilePath := fmt.Sprintf("%v.out_%v", base, name)
if *flagUpdate {
if err := osutil.WriteFile(outFilePath, testResult); err != nil {
t.Fatal(err)
}
}
if !osutil.IsExist(outFilePath) {
t.Fatalf("asset %v does not exist", outFilePath)
}
testedOutFiles = append(testedOutFiles, outFilePath)
outFile, err := os.ReadFile(outFilePath)
if err != nil {
t.Fatal(err)
}
if !reflect.DeepEqual(testResult, outFile) {
t.Fatalf("output not equal:\nWant: %x\nGot: %x", outFile, testResult)
}
})
}
sort.Strings(testedOutFiles)
sort.Strings(allOutFiles)
if diff := cmp.Diff(allOutFiles, testedOutFiles); diff != "" {
t.Fatalf("not all output files used: %v", diff)
}
}
|
