reactos/syzkaller
1
0
Fork 0
mirror of https://github.com/reactos/syzkaller.git synced 2025-02-26 22:55:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

csourse: fix emitting syz_* syscalls in c reproducer

Browse Source
This commit is contained in:
Andrey Konovalov 2016-11-29 18:40:58 +01:00
parent 684beccb2a
commit b13dc4bf50
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
csource/csource.go
View File

@ -45,7 +45,7 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) {
}
fmt.Fprintf(w, "\n")
hdr, err := preprocessCommonHeader(opts)
hdr, err := preprocessCommonHeader(opts, handled)
if err != nil {
return nil, err
}
@ -246,7 +246,7 @@ loop:
return calls, n
}
func preprocessCommonHeader(opts Options) (string, error) {
func preprocessCommonHeader(opts Options, handled map[string]int) (string, error) {
cmd := exec.Command("cpp", "-nostdinc", "-undef", "-fdirectives-only", "-dDI", "-E", "-P", "-")
switch opts.Sandbox {
case "none":
@ -261,6 +261,9 @@ func preprocessCommonHeader(opts Options) (string, error) {
if opts.Repeat {
cmd.Args = append(cmd.Args, "-DSYZ_REPEAT")
}
for name, _ := range handled {
cmd.Args = append(cmd.Args, "-D__NR_" + name)
}
cmd.Stdin = strings.NewReader(commonHeader)
stderr := new(bytes.Buffer)
stdout := new(bytes.Buffer)
@ -271,6 +274,13 @@ func preprocessCommonHeader(opts Options) (string, error) {
}
out := strings.Replace(stdout.String(), "#define __STDC__ 1\n", "", -1)
out = strings.Replace(out, "#define __STDC_HOSTED__ 1\n", "", -1)
for _, arg := range cmd.Args {
if !strings.HasPrefix(arg, "-D") {
continue
}
define := strings.TrimPrefix(arg, "-D")
out = strings.Replace(out, "#define " + define + " 1\n", "", -1)
}
return out, nil
}