max try times from command line

hugefiver · Dec 20, 2021 · 489ec28 · 489ec28
1 parent ee4d5e8
commit 489ec28
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/flags.go b/flags.go
@@ -35,6 +35,9 @@ type ArgsStruct struct {
 
 	// Anti honeypot scan
 	AntiScan bool
+
+	// Max try times
+	MaxTry uint
 }
 
 // GetArg : get args
@@ -69,6 +72,8 @@ func GetArg() (ArgsStruct, func()) {
 	f.BoolVar(&NoAntiScan, "A", false, "disable anti honeypot scan")
 	f.BoolVar(&AntiScan, "a", false, "enable anti honeypot scan (default)")
 
+	f.UintVar(&args.MaxTry, "try", 0, "max try times")
+
 	f.Parse(os.Args[1:])
 	//_args = &args
 

diff --git a/main.go b/main.go
@@ -122,10 +122,14 @@ func main() {
 		}
 	}
 
+	maxTry := 3
+	if args.MaxTry > 0 {
+		maxTry = int(args.MaxTry)
+	}
 	serverConfig := ssh.ServerConfig{
 		Config:             ssh.Config{},
 		NoClientAuth:       false,
-		MaxAuthTries:       3,
+		MaxAuthTries:       maxTry,
 		PasswordCallback:   rejectAll,
 		PublicKeyCallback:  nil,
 		AuthLogCallback:    nil,