refactor(tools): replace global variables with local to prevent race conditions

- Convert IPBlocks, IPAddress, Websites from global to local variables
- Use lowercase naming convention (ipBlocks, ipAddress, websites)
- Prevents data accumulation across multiple runs
- Eliminates potential race condition risk

Author: lordixir

tools/find_asn.go

@@ -8,37 +8,36 @@ import (
 	"time"
 )
 
-var (
-	IPBlocks  []string
-	IPAddress []string
-	Websites  [][]string
-)
-
 func FindASN(asn string, domain string, domainTitle string, con bool, export bool, timeout int, interruptData *modules.InterruptData) {
+	// Use local variables instead of global to avoid race conditions
+	var ipBlocks []string
+	var ipAddress []string
+	var websites [][]string
+
 	re := regexp.MustCompile(`(?m)route:\s+([0-9\.\/]+)$`)
 	for _, match := range re.FindAllStringSubmatch(modules.FindIPBlocks(asn), -1) {
-		IPBlocks = append(IPBlocks, match[1])
+		ipBlocks = append(ipBlocks, match[1])
 	}
 
-	for _, block := range IPBlocks {
+	for _, block := range ipBlocks {
 		ips, err := modules.CalcIPAddress(block)
 		if err != nil {
 			return
 		}
 
-		IPAddress = append(IPAddress, ips...)
+		ipAddress = append(ipAddress, ips...)
 	}
 
 	// Update interrupt data with IP blocks
 	if interruptData != nil {
-		interruptData.IPBlocks = IPBlocks
+		interruptData.IPBlocks = ipBlocks
 	}
 
 	fmt.Println("ASN:         " + asn +
-		"\nIP Block:    " + strconv.Itoa(len(IPBlocks)) +
-		"\nIP Address:  " + strconv.Itoa(len(IPAddress)) +
+		"\nIP Block:    " + strconv.Itoa(len(ipBlocks)) +
+		"\nIP Address:  " + strconv.Itoa(len(ipAddress)) +
 		"\nStart Time:  " + time.Now().Local().String() +
-		"\nEnd Time:    " + time.Now().Add((time.Millisecond*time.Duration(timeout))*time.Duration(len(IPAddress))).Local().String())
+		"\nEnd Time:    " + time.Now().Add((time.Millisecond*time.Duration(timeout))*time.Duration(len(ipAddress))).Local().String())
 
-	modules.ResolveSite(IPAddress, Websites, domainTitle, IPBlocks, domain, con, export, timeout, interruptData)
+	modules.ResolveSite(ipAddress, websites, domainTitle, ipBlocks, domain, con, export, timeout, interruptData)
 }

tools/find_ip.go

@@ -7,20 +7,24 @@ import (
 	"time"
 )
 
-func FindIP(IPBlocks []string, domain string, domainTitle string, con bool, export bool, timeout int, interruptData *modules.InterruptData) {
-	for _, block := range IPBlocks {
+func FindIP(ipBlocks []string, domain string, domainTitle string, con bool, export bool, timeout int, interruptData *modules.InterruptData) {
+	// Use local variables instead of global to avoid race conditions
+	var ipAddress []string
+	var websites [][]string
+
+	for _, block := range ipBlocks {
 		ips, err := modules.CalcIPAddress(block)
 		if err != nil {
 			return
 		}
 
-		IPAddress = append(IPAddress, ips...)
+		ipAddress = append(ipAddress, ips...)
 	}
 
-	fmt.Println("IP Block:    " + strconv.Itoa(len(IPBlocks)) +
-		"\nIP Address:  " + strconv.Itoa(len(IPAddress)) +
+	fmt.Println("IP Block:    " + strconv.Itoa(len(ipBlocks)) +
+		"\nIP Address:  " + strconv.Itoa(len(ipAddress)) +
 		"\nStart Time:  " + time.Now().Local().String() +
-		"\nEnd Time:    " + time.Now().Add((time.Millisecond*time.Duration(timeout))*time.Duration(len(IPAddress))).Local().String())
+		"\nEnd Time:    " + time.Now().Add((time.Millisecond*time.Duration(timeout))*time.Duration(len(ipAddress))).Local().String())
 
-	modules.ResolveSite(IPAddress, Websites, domainTitle, IPBlocks, domain, con, export, timeout, interruptData)
+	modules.ResolveSite(ipAddress, websites, domainTitle, ipBlocks, domain, con, export, timeout, interruptData)
 }