diff --git a/controllers/login.go b/controllers/login.go index db68a45..e3e3006 100644 --- a/controllers/login.go +++ b/controllers/login.go @@ -35,6 +35,8 @@ func Login(localAddr net.Addr, debugOutput bool) error { if localAddr != nil && global.Config.Settings.Basic.UseDhcpIP { ip = localAddr.(*net.TCPAddr).IP.String() + } else if global.Flags.ClientIP != "" { + ip = global.Flags.ClientIP } log.Debugln("认证客户端 ip: ", ip) diff --git a/global/flags.go b/global/flags.go index 9880692..3700336 100644 --- a/global/flags.go +++ b/global/flags.go @@ -11,10 +11,13 @@ var Flags struct { RunningDaemon bool //强制daemon Daemon bool + //指定 client ip + ClientIP string } func initFlags() { flag.StringVar(&Flags.Path, "config", "Config.yaml", "config path") + flag.StringVar(&Flags.ClientIP, "ip", "", "client ip for login") flag.BoolVar(&Flags.RunningDaemon, "running-daemon", false, "") flag.BoolVar(&Flags.Daemon, "daemon", false, "") flag.Parse()