diff options
Diffstat (limited to 'pkg/csource')
| -rw-r--r-- | pkg/csource/csource.go | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/pkg/csource/csource.go b/pkg/csource/csource.go index 6ac5d2eba..5c4b2f848 100644 --- a/pkg/csource/csource.go +++ b/pkg/csource/csource.go @@ -79,12 +79,12 @@ func Write(p *prog.Prog, opts Options) ([]byte, error) { fmt.Fprint(w, "// autogenerated by syzkaller (http://github.com/google/syzkaller)\n\n") - handled := make(map[string]int) + handled := make(map[string]uint64) for _, c := range p.Calls { handled[c.Meta.CallName] = c.Meta.NR } for name, nr := range handled { - // Only generate defines for new (added after commit 8a1ab3155c2ac on 2012-10-04) native syscalls. + // Only generate defines for new syscalls (added after commit 8a1ab3155c2ac on 2012-10-04). // TODO: the syscall number 313 implies that we're dealing with linux/amd64. if nr >= 313 && !strings.HasPrefix(name, "syz_") { fmt.Fprintf(w, "#ifndef __NR_%v\n", name) @@ -375,8 +375,9 @@ loop: if !opts.EnableTun && (meta.CallName == "syz_emit_ethernet" || meta.CallName == "syz_extract_tcp_res") { emitCall = false } + native := !strings.HasPrefix(meta.CallName, "syz_") if emitCall { - if meta.Native { + if native { fmt.Fprintf(w, "\tr[%v] = syscall(__NR_%v", n, meta.CallName) } else { fmt.Fprintf(w, "\tr[%v] = %v(", n, meta.CallName) @@ -387,7 +388,7 @@ loop: typ := read() size := read() _ = size - if emitCall && (meta.Native || i > 0) { + if emitCall && (native || i > 0) { fmt.Fprintf(w, ", ") } switch typ { @@ -419,7 +420,7 @@ loop: return calls, n } -func preprocessCommonHeader(opts Options, handled map[string]int, useBitmasks, useChecksums bool) (string, error) { +func preprocessCommonHeader(opts Options, handled map[string]uint64, useBitmasks, useChecksums bool) (string, error) { var defines []string if useBitmasks { defines = append(defines, "SYZ_USE_BITMASKS") |