mirror of
https://github.com/reactos/syzkaller.git
synced 2024-11-23 11:29:46 +00:00
108 lines
2.3 KiB
Go
108 lines
2.3 KiB
Go
// Copyright 2017 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 repro
|
|
|
|
import (
|
|
"math/rand"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/google/syzkaller/pkg/csource"
|
|
"github.com/google/syzkaller/prog"
|
|
)
|
|
|
|
func initTest(t *testing.T) (*rand.Rand, int) {
|
|
iters := 1000
|
|
if testing.Short() {
|
|
iters = 100
|
|
}
|
|
seed := time.Now().UnixNano()
|
|
if os.Getenv("CI") != "" {
|
|
seed = 0 // required for deterministic coverage reports
|
|
}
|
|
rs := rand.NewSource(seed)
|
|
t.Logf("seed=%v", seed)
|
|
return rand.New(rs), iters
|
|
}
|
|
|
|
func TestBisect(t *testing.T) {
|
|
ctx := &context{
|
|
stats: new(Stats),
|
|
}
|
|
|
|
rd, iters := initTest(t)
|
|
for n := 0; n < iters; n++ {
|
|
var progs []*prog.LogEntry
|
|
numTotal := rd.Intn(300)
|
|
numGuilty := 0
|
|
for i := 0; i < numTotal; i++ {
|
|
var prog prog.LogEntry
|
|
if rd.Intn(30) == 0 {
|
|
prog.Proc = 42
|
|
numGuilty++
|
|
}
|
|
progs = append(progs, &prog)
|
|
}
|
|
if numGuilty == 0 {
|
|
var prog prog.LogEntry
|
|
prog.Proc = 42
|
|
progs = append(progs, &prog)
|
|
numGuilty++
|
|
}
|
|
progs, _ = ctx.bisectProgs(progs, func(p []*prog.LogEntry) (bool, error) {
|
|
guilty := 0
|
|
for _, prog := range p {
|
|
if prog.Proc == 42 {
|
|
guilty++
|
|
}
|
|
}
|
|
return guilty == numGuilty, nil
|
|
})
|
|
if numGuilty > 8 && len(progs) == 0 {
|
|
// Bisection has been aborted.
|
|
continue
|
|
}
|
|
if len(progs) != numGuilty {
|
|
t.Fatalf("bisect test failed: wrong number of guilty progs: got: %v, want: %v", len(progs), numGuilty)
|
|
}
|
|
for _, prog := range progs {
|
|
if prog.Proc != 42 {
|
|
t.Fatalf("bisect test failed: wrong program is guilty: progs: %v", progs)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestSimplifies(t *testing.T) {
|
|
opts := csource.Options{
|
|
Threaded: true,
|
|
Collide: true,
|
|
Repeat: true,
|
|
Procs: 10,
|
|
Sandbox: "namespace",
|
|
NetInjection: true,
|
|
NetDevices: true,
|
|
NetReset: true,
|
|
Cgroups: true,
|
|
UseTmpDir: true,
|
|
HandleSegv: true,
|
|
Repro: true,
|
|
}
|
|
var check func(opts csource.Options, i int)
|
|
check = func(opts csource.Options, i int) {
|
|
if err := opts.Check("linux"); err != nil {
|
|
t.Fatalf("opts are invalid: %v", err)
|
|
}
|
|
if i == len(cSimplifies) {
|
|
return
|
|
}
|
|
check(opts, i+1)
|
|
if cSimplifies[i](&opts) {
|
|
check(opts, i+1)
|
|
}
|
|
}
|
|
check(opts, 0)
|
|
}
|