forked from google/syzkaller
/
repro.go
93 lines (86 loc) · 2.5 KB
/
repro.go
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
// 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 main
import (
"flag"
"fmt"
"io/ioutil"
"os"
"github.com/google/syzkaller/pkg/csource"
"github.com/google/syzkaller/pkg/log"
"github.com/google/syzkaller/pkg/mgrconfig"
"github.com/google/syzkaller/pkg/osutil"
"github.com/google/syzkaller/pkg/report"
"github.com/google/syzkaller/pkg/repro"
"github.com/google/syzkaller/prog"
"github.com/google/syzkaller/vm"
)
var (
flagConfig = flag.String("config", "", "configuration file")
flagCount = flag.Int("count", 0, "number of VMs to use (overrides config count param)")
flagDebug = flag.Bool("debug", false, "print debug output")
)
func main() {
os.Args = append(append([]string{}, os.Args[0], "-v=10"), os.Args[1:]...)
flag.Parse()
cfg, err := mgrconfig.LoadFile(*flagConfig)
if err != nil {
log.Fatalf("%v", err)
}
if len(flag.Args()) != 1 {
log.Fatalf("usage: syz-repro -config=config.file execution.log")
}
data, err := ioutil.ReadFile(flag.Args()[0])
if err != nil {
log.Fatalf("failed to open log file: %v", err)
}
if _, err := prog.GetTarget(cfg.TargetOS, cfg.TargetArch); err != nil {
log.Fatalf("%v", err)
}
vmPool, err := vm.Create(cfg, *flagDebug)
if err != nil {
log.Fatalf("%v", err)
}
vmCount := vmPool.Count()
if *flagCount > 0 && *flagCount < vmCount {
vmCount = *flagCount
}
if vmCount > 4 {
vmCount = 4
}
vmIndexes := make([]int, vmCount)
for i := range vmIndexes {
vmIndexes[i] = i
}
reporter, err := report.NewReporter(cfg)
if err != nil {
log.Fatalf("%v", err)
}
osutil.HandleInterrupts(vm.Shutdown)
res, stats, err := repro.Run(data, cfg, reporter, vmPool, vmIndexes)
if err != nil {
log.Logf(0, "reproduction failed: %v", err)
}
if stats != nil {
fmt.Printf("Extracting prog: %v\n", stats.ExtractProgTime)
fmt.Printf("Minimizing prog: %v\n", stats.MinimizeProgTime)
fmt.Printf("Simplifying prog options: %v\n", stats.SimplifyProgTime)
fmt.Printf("Extracting C: %v\n", stats.ExtractCTime)
fmt.Printf("Simplifying C: %v\n", stats.SimplifyCTime)
}
if res == nil {
return
}
fmt.Printf("opts: %+v crepro: %v\n\n", res.Opts, res.CRepro)
fmt.Printf("%s\n", res.Prog.Serialize())
if res.CRepro {
src, err := csource.Write(res.Prog, res.Opts)
if err != nil {
log.Fatalf("failed to generate C repro: %v", err)
}
if formatted, err := csource.Format(src); err == nil {
src = formatted
}
fmt.Printf("%s\n", src)
}
}