diff --git a/Main.go b/Main.go index 4a8047f..9e5cf13 100644 --- a/Main.go +++ b/Main.go @@ -7,15 +7,28 @@ import ( ) func main() { - printHeader() - var input string var wantPrefixes bool + var wantHeader bool + + flag.StringVar(&input, "i", "", "IP address or domain") + flag.BoolVar(&wantPrefixes, "p", false, "print BGP prefixes") + flag.BoolVar(&wantHeader, "b", true, "enable/disable header") + + flag.Usage = func() { + fmt.Printf("Usage of iPilot: \n") + fmt.Printf(" -b bool enable/disable header (default true)\n") + fmt.Printf(" -h bool view help\n") + fmt.Printf(" -i string IP address or domain\n") + fmt.Printf(" -p bool print BGP prefixes (default false)\n") + } - flag.StringVar(&input, "i", "", "Specify IP address or domain name.") - flag.BoolVar(&wantPrefixes, "p", false, "Enable printing of advertised BGP prefixes.") flag.Parse() + if wantHeader { + printHeader() + } + if input == "" { fmt.Println("FATAL: No IP address or domain name was specified.") os.Exit(1)