2.2.2

cmd/afrog/main.go

@@ -114,6 +114,7 @@ func readConfig() {
 		flagSet.BoolVarP(&options.NoFinger, "nofinger", "nf", false, "disable fingerprint"),
 		flagSet.IntVar(&options.Retries, "retries", 1, "number of times to retry a failed request"),
 		flagSet.IntVar(&options.Timeout, "timeout", 10, "time to wait in seconds before timeout"),
+		flagSet.IntVar(&options.MaxHostNum, "mhe", 3, "max errors for a host before skipping from scan"),
 	)
 
 	flagSet.CreateGroup("update", "Update",

pkg/config/config.go

@@ -40,7 +40,7 @@ type Ceye struct {
 }
 
 const afrogConfigFilename = "afrog-config.yaml"
-const Version = "2.2.1"
+const Version = "2.2.2"
 
 // Create and initialize afrog-config.yaml configuration info
 func New() (*Config, error) {

pkg/config/options.go

@@ -79,6 +79,9 @@ type Options struct {
 	// number of times to retry a failed request (default 1)
 	Retries int
 
+	//
+	MaxHostNum int
+
 	// time to wait in seconds before timeout (default 10)
 	Timeout int
 

pkg/core/checker.go

@@ -166,14 +166,11 @@ func (c *Checker) Check(ctx context.Context, target string, pocItem *poc.Poc) (e
 	return err
 }
 
-var MaxCheckNum = 1
-
 func (c *Checker) checkURL(target string) (string, error) {
 	var err error
 
 	// if target check num more than MaxCheckNum
-	if c.Options.Targets.Num(target) > MaxCheckNum {
-		fmt.Printf("%s is blacklisted\n", target)
+	if c.Options.Targets.Num(target) > c.Options.MaxHostNum {
 		return "", fmt.Errorf("%s is blacklisted", target)
 	}