🔨 Add -retries flag (close #10)

pwnesia · Jan 3, 2021 · 820501b · 820501b
1 parent 0489388
commit 820501b
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -4,7 +4,7 @@
 [![License](https://img.shields.io/badge/License-Apache%202.0-yellowgreen)](https://github.com/kitabisa/ssb/blob/master/LICENSE)
 [![contributions](https://img.shields.io/badge/contributions-welcome-magenta.svg?style=flat)](https://github.com/kitabisa/ssb/issues)
 [![made with Go](https://img.shields.io/badge/made%20with-Go-brightgreen)](http://golang.org)
-[![Version](https://img.shields.io/badge/version-0.0.2-blueviolet)](https://github.com/kitabisa/ssb/releases)
+[![Version](https://img.shields.io/badge/version-0.1.0-blueviolet)](https://github.com/kitabisa/ssb/releases)
 
 **S**_ecure_ **S**_hell_ **B**_ruteforcer_ — A faster & simpler way to bruteforce SSH server.
 
@@ -30,7 +30,7 @@ Need [go1.14+](https://golang.org/doc/install#download) compiler installed and c
 
 ```bash
 ▶ ssb [-p port] [-w wordlist.txt] [-t timeout]
-      [-c concurrent] [-o output] [user@]hostname
+      [-c concurrent] [-r retries] [-o output] [user@]hostname
 ```
 
 ### Options:
@@ -44,6 +44,8 @@ Need [go1.14+](https://golang.org/doc/install#download) compiler installed and c
      Connection timeout (default 30s).
   -c concurrent
      Concurrency/threads level (default 100).
+  -r retries
+     Specify the connection retries (default 1).
   -o output
      Save valid password to file.
   -v

diff --git a/internal/runner/constant.go b/internal/runner/constant.go
@@ -5,10 +5,11 @@ import "time"
 var (
   opt   *Options
   uhost []string
+  vld   bool
 )
 
 const (
-  version = "v0.0.2"
+  version = "v0.1.0"
   banner  = `
              _
      ` + version + ` | |
@@ -34,6 +35,8 @@ Options:
      Connection timeout (default 30s).
   -c concurrent
      Concurrency/threads level (default 100).
+  -r retries
+     Specify the connection retries (default 1).
   -o output
      Save valid password to file.
   -v

diff --git a/internal/runner/parser.go b/internal/runner/parser.go
@@ -15,6 +15,7 @@ import (
 type Options struct {
 	concurrent int
 	wordlist   string
+	retries    int
 	verbose    bool
 	timeout    time.Duration
 	output     string
@@ -32,6 +33,7 @@ func Parse() *Options {
 	opt.timeout = timeout
 
 	flag.IntVar(&opt.port, "p", 22, "")
+	flag.IntVar(&opt.retries, "r", 1, "")
 	flag.IntVar(&opt.concurrent, "c", 100, "")
 	flag.BoolVar(&opt.verbose, "v", false, "")
 	flag.StringVar(&opt.output, "o", "", "")

diff --git a/internal/runner/runner.go b/internal/runner/runner.go
@@ -25,7 +25,11 @@ func New(opt *Options) {
 		go func() {
 			defer swg.Done()
 			for pass := range job {
-				opt.run(pass)
+				for i := 0; i < opt.retries; i++ {
+					if opt.run(pass) {
+						break
+					}
+				}
 			}
 		}()
 	}
@@ -39,7 +43,7 @@ func New(opt *Options) {
 	gologger.Infof("Done!")
 }
 
-func (opt *Options) run(password string) {
+func (opt *Options) run(password string) bool {
 	cfg := ssb.New(opt.user, password, opt.timeout)
 
 	con, err := ssb.Connect(opt.host, opt.port, cfg)
@@ -55,7 +59,11 @@ func (opt *Options) run(password string) {
 		if opt.file != nil {
 			fmt.Fprintf(opt.file, "%s\n", password)
 		}
+
+		vld = true
 	}
+
+	return vld
 }
 
 func (opt *Options) showInfo() {