mirror of
https://github.com/reactos/syzkaller.git
synced 2024-11-23 11:29:46 +00:00
87 lines
2.8 KiB
Go
87 lines
2.8 KiB
Go
// 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"
|
|
"runtime"
|
|
|
|
"github.com/google/syzkaller/pkg/csource"
|
|
"github.com/google/syzkaller/prog"
|
|
_ "github.com/google/syzkaller/sys"
|
|
)
|
|
|
|
var (
|
|
flagOS = flag.String("os", runtime.GOOS, "target os")
|
|
flagArch = flag.String("arch", runtime.GOARCH, "target arch")
|
|
flagThreaded = flag.Bool("threaded", false, "create threaded program")
|
|
flagCollide = flag.Bool("collide", false, "create collide program")
|
|
flagRepeat = flag.Bool("repeat", false, "repeat program infinitely or not")
|
|
flagProcs = flag.Int("procs", 1, "number of parallel processes")
|
|
flagSandbox = flag.String("sandbox", "", "sandbox to use (none, setuid, namespace)")
|
|
flagProg = flag.String("prog", "", "file with program to convert (required)")
|
|
flagFaultCall = flag.Int("fault_call", -1, "inject fault into this call (0-based)")
|
|
flagFaultNth = flag.Int("fault_nth", 0, "inject fault on n-th operation (0-based)")
|
|
flagEnableTun = flag.Bool("tun", false, "set up TUN/TAP interface")
|
|
flagUseTmpDir = flag.Bool("tmpdir", false, "create a temporary dir and execute inside it")
|
|
flagHandleSegv = flag.Bool("segv", false, "catch and ignore SIGSEGV")
|
|
flagWaitRepeat = flag.Bool("waitrepeat", false, "wait for each repeat attempt")
|
|
flagCgroups = flag.Bool("cgroups", false, "enable cgroups support")
|
|
flagDebug = flag.Bool("debug", false, "generate debug printfs")
|
|
)
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
if *flagProg == "" {
|
|
flag.PrintDefaults()
|
|
os.Exit(1)
|
|
}
|
|
target, err := prog.GetTarget(*flagOS, *flagArch)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "%v", err)
|
|
os.Exit(1)
|
|
}
|
|
data, err := ioutil.ReadFile(*flagProg)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to read prog file: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
p, err := target.Deserialize(data)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to deserialize the program: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
opts := csource.Options{
|
|
Threaded: *flagThreaded,
|
|
Collide: *flagCollide,
|
|
Repeat: *flagRepeat,
|
|
Procs: *flagProcs,
|
|
Sandbox: *flagSandbox,
|
|
Fault: *flagFaultCall >= 0,
|
|
FaultCall: *flagFaultCall,
|
|
FaultNth: *flagFaultNth,
|
|
EnableTun: *flagEnableTun,
|
|
UseTmpDir: *flagUseTmpDir,
|
|
EnableCgroups: *flagCgroups,
|
|
HandleSegv: *flagHandleSegv,
|
|
WaitRepeat: *flagWaitRepeat,
|
|
Debug: *flagDebug,
|
|
Repro: false,
|
|
}
|
|
src, err := csource.Write(p, opts)
|
|
if err != nil {
|
|
fmt.Fprintf(os.Stderr, "failed to generate C source: %v\n", err)
|
|
os.Exit(1)
|
|
}
|
|
if formatted, err := csource.Format(src); err != nil {
|
|
fmt.Fprintf(os.Stderr, "%v\n", err)
|
|
} else {
|
|
src = formatted
|
|
}
|
|
os.Stdout.Write(src)
|
|
}
|