package main import ( "flag" "fmt" "net" ) func main() { var address string var port int var shouldDial bool flag.StringVar(&address, "address", "127.0.0.1", "Address to dial or listen") flag.IntVar(&port, "port", 10300, "Port to dial or listen") flag.BoolVar(&shouldDial, "dial", false, "Whether to dial target address or not") flag.Parse() if shouldDial { dial(fmt.Sprintf("%s:%d", address, port)) } else { listen(net.ParseIP(address), port) } }