Merge pull request #5 from Nitr4x/develop

Fixing go routine

Author: Nitr4x

bin/orfinder

@@ -4,7 +4,6 @@ import (
 	"fmt"
 	"log"
 	"net"
-	"sync"
 	"time"
 
 	"github.com/fatih/color"
@@ -108,9 +107,9 @@ func inc(ip net.IP) {
 	}
 }
 
-// Function Scan(address string, wg *sync.WaitGroup)
+// Function Scan(address string)
 // Perform SYN scanning on a given IP range
-func Scan(address string, wg *sync.WaitGroup) {
+func Scan(address string) {
 	ip, ipnet, err := net.ParseCIDR(address)
 	if err != nil {
 		log.Fatal(err)
@@ -127,6 +126,4 @@ func Scan(address string, wg *sync.WaitGroup) {
 			}
 		}
 	}
-
-	wg.Done()
 }

engine/scanner/scanner.go

@@ -34,11 +34,23 @@ func main() {
 
 	country := parser.Parse()
 	IPs := loader.Load(country)
+	tasks := make(chan string)
 
-	for i := len(IPs) - 1; i >= 0; i-- {
+	for i := 0; i < 100; i++ {
 		wg.Add(1)
-		go scanner.Scan(IPs[i], &wg)
+		go func() {
+			for task := range tasks {
+				scanner.Scan(task)
+			}
+			wg.Done()
+		}()
 	}
 
+	for i := len(IPs) - 1; i >= 0; i-- {
+		tasks <- IPs[i]
+	}
+
+	close(tasks)
+
 	wg.Wait()
 }