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
|
// Copyright 2015 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 ipc
import (
"fmt"
"math/rand"
"os"
"path/filepath"
"runtime"
"testing"
"time"
"github.com/google/syzkaller/pkg/csource"
"github.com/google/syzkaller/pkg/osutil"
"github.com/google/syzkaller/prog"
_ "github.com/google/syzkaller/sys"
)
const timeout = 10 * time.Second
func buildExecutor(t *testing.T, target *prog.Target) string {
src := fmt.Sprintf("../../executor/executor_%v.cc", target.OS)
return buildProgram(t, target, filepath.FromSlash(src))
}
func buildSource(t *testing.T, target *prog.Target, src []byte) string {
tmp, err := osutil.WriteTempFile(src)
if err != nil {
t.Fatalf("%v", err)
}
defer os.Remove(tmp)
return buildProgram(t, target, tmp)
}
func buildProgram(t *testing.T, target *prog.Target, src string) string {
bin, err := csource.Build(target, "c++", src)
if err != nil {
t.Fatalf("%v", err)
}
return bin
}
func initTest(t *testing.T) (*prog.Target, rand.Source, int, EnvFlags) {
t.Parallel()
iters := 100
if testing.Short() {
iters = 10
}
seed := int64(time.Now().UnixNano())
rs := rand.NewSource(seed)
t.Logf("seed=%v", seed)
target, err := prog.GetTarget(runtime.GOOS, runtime.GOARCH)
if err != nil {
t.Fatal(err)
}
cfg, _, err := DefaultConfig()
if err != nil {
t.Fatal(err)
}
flags := cfg.Flags & (FlagUseShmem | FlagUseForkServer)
return target, rs, iters, flags
}
func TestEmptyProg(t *testing.T) {
target, _, _, flags0 := initTest(t)
bin := buildExecutor(t, target)
defer os.Remove(bin)
cfg := &Config{
Executor: bin,
Flags: flags0,
Timeout: timeout,
}
env, err := MakeEnv(cfg, 0)
if err != nil {
t.Fatalf("failed to create env: %v", err)
}
defer env.Close()
p := new(prog.Prog)
opts := &ExecOpts{}
output, _, failed, hanged, err := env.Exec(opts, p)
if err != nil {
t.Fatalf("failed to run executor: %v", err)
}
if len(output) != 0 {
t.Fatalf("output on empty program")
}
if failed || hanged {
t.Fatalf("empty program failed")
}
}
func TestExecute(t *testing.T) {
target, rs, iters, configFlags := initTest(t)
bin := buildExecutor(t, target)
defer os.Remove(bin)
flags := []ExecFlags{0, FlagThreaded, FlagThreaded | FlagCollide}
for _, flag := range flags {
t.Logf("testing flags 0x%x\n", flag)
cfg := &Config{
Executor: bin,
Flags: configFlags,
Timeout: timeout,
}
env, err := MakeEnv(cfg, 0)
if err != nil {
t.Fatalf("failed to create env: %v", err)
}
defer env.Close()
for i := 0; i < iters/len(flags); i++ {
p := target.Generate(rs, 10, nil)
if i == 0 {
p = target.GenerateSimpleProg()
}
opts := &ExecOpts{
Flags: flag,
}
output, _, _, _, err := env.Exec(opts, p)
if err != nil {
t.Logf("program:\n%s\n", p.Serialize())
t.Fatalf("failed to run executor: %v\n%s", err, output)
}
}
}
}
|
