From 2c2c4ecdbcab252fcd816c33d9151e70d6670e08 Mon Sep 17 00:00:00 2001 From: fatedier Date: Thu, 30 Mar 2023 22:02:00 +0800 Subject: [PATCH] feat(nathole): add flag for STUN server in nathole discovery command (#3383) --- cmd/frpc/sub/nathole.go | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/cmd/frpc/sub/nathole.go b/cmd/frpc/sub/nathole.go index 9e8a21d..377401c 100644 --- a/cmd/frpc/sub/nathole.go +++ b/cmd/frpc/sub/nathole.go @@ -26,11 +26,15 @@ import ( "github.com/fatedier/frp/pkg/nathole" ) +var natHoleSTUNServer string + func init() { RegisterCommonFlags(natholeCmd) rootCmd.AddCommand(natholeCmd) natholeCmd.AddCommand(natholeDiscoveryCmd) + + natholeCmd.PersistentFlags().StringVarP(&natHoleSTUNServer, "nat_hole_stun_server", "", "", "STUN server address for nathole") } var natholeCmd = &cobra.Command{ @@ -47,6 +51,9 @@ var natholeDiscoveryCmd = &cobra.Command{ fmt.Println(err) os.Exit(1) } + if natHoleSTUNServer != "" { + cfg.NatHoleSTUNServer = natHoleSTUNServer + } if err := validateForNatHoleDiscovery(cfg); err != nil { fmt.Println(err)