Skip to content

Commit 01f249c

Browse files
author
Ethan Graham
committed
tools/kfuzztest-gen: add usage, simplify args parsing
Remove flag package, as we only parse a single argument every time. Instead, we can just look at os.Args[1] to get the path to VMLinux.
1 parent c9dbe9a commit 01f249c

File tree

1 file changed

+10
-5
lines changed

1 file changed

+10
-5
lines changed

tools/kfuzztest-gen/main.go

Lines changed: 10 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,20 +6,25 @@
66
package main
77

88
import (
9-
"flag"
109
"fmt"
10+
"io"
11+
"os"
1112

1213
"github.com/google/syzkaller/pkg/kfuzztest"
1314
"github.com/google/syzkaller/pkg/log"
1415
"github.com/google/syzkaller/pkg/tool"
1516
)
1617

17-
var vmlinuxPath = flag.String("vmlinux", "./vmlinux", "path to vmlinux")
18-
1918
func main() {
20-
flag.Parse()
19+
usage := func(w io.Writer) {
20+
fmt.Fprintln(w, "usage: ./kfuzztest-gen /path/to/vmlinux")
21+
}
22+
if len(os.Args) != 2 {
23+
usage(os.Stderr)
24+
os.Exit(1)
25+
}
2126

22-
extractor, err := kfuzztest.NewExtractor(*vmlinuxPath)
27+
extractor, err := kfuzztest.NewExtractor(os.Args[1])
2328
if err != nil {
2429
tool.Fail(err)
2530
}

0 commit comments

Comments
 (0)