Pārlūkot izejas kodu

feat(nathole): add flag for STUN server in nathole discovery command (#3383)

fatedier 1 gadu atpakaļ
vecāks
revīzija
2c2c4ecdbc
1 mainītis faili ar 7 papildinājumiem un 0 dzēšanām
  1. 7 0
      cmd/frpc/sub/nathole.go

+ 7 - 0
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)