Merge pull request #90 from edoardottt/devel

Devel

Author: edoardottt

pkg/crawler/colly.go

@@ -212,7 +212,9 @@ func New(target string, txt string, html string, delayTime int, concurrency int,
 			// HERE SCAN FOR SECRETS
 			if secretsFlag && lengthOk {
 				secretsSlice := huntSecrets(secretsFile, r.Request.URL.String(), string(r.Body))
-				FinalSecrets = append(FinalSecrets, secretsSlice...)
+				for _, elem := range secretsSlice {
+					FinalSecrets = append(FinalSecrets, elem)
+				}
 			}
 			// HERE SCAN FOR ENDPOINTS
 			if endpointsFlag {
@@ -233,13 +235,17 @@ func New(target string, txt string, html string, delayTime int, concurrency int,
 			// HERE SCAN FOR ERRORS
 			if errorsFlag {
 				errorsSlice := huntErrors(r.Request.URL.String(), string(r.Body))
-				FinalErrors = append(FinalErrors, errorsSlice...)
+				for _, elem := range errorsSlice {
+					FinalErrors = append(FinalErrors, elem)
+				}
 			}
 
 			// HERE SCAN FOR INFOS
 			if infoFlag {
 				infosSlice := huntInfos(r.Request.URL.String(), string(r.Body))
-				FinalInfos = append(FinalInfos, infosSlice...)
+				for _, elem := range infosSlice {
+					FinalInfos = append(FinalInfos, elem)
+				}
 			}
 		}
 	})
@@ -281,15 +287,21 @@ func New(target string, txt string, html string, delayTime int, concurrency int,
 	// Setup graceful exit
 	chanC := make(chan os.Signal, 1)
 	lettersNum := 23
+	cCount := 0
 
 	signal.Notify(chanC, os.Interrupt)
 	rand.Seed(time.Now().UnixNano())
 
 	go func() {
 		for range chanC {
+			if cCount > 0 {
+				os.Exit(1)
+			}
+
 			if !plain {
 				fmt.Fprint(os.Stdout, "\r")
 				fmt.Println("CTRL+C pressed: Exiting")
+				cCount++
 			}
 
 			c.AllowedDomains = []string{sliceUtils.RandSeq(lettersNum)}