acidvegas/massrdns
1
Fork 0
mirror of git://git.acid.vegas/massrdns.git synced 2024-11-25 00:56:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

IP addresses were being looked up more than once

Browse Source
This commit is contained in:
Dionysus 2023-09-21 15:38:49 -04:00
parent d95e119aac
commit 451d626c80
Signed by: acidvegas
GPG Key ID: EF4B922DB85DC9DE
2 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
massrdns
View File

Binary file not shown.

11
massrdns.go
View File

@ -17,6 +17,7 @@ import (
var dnsServers []string var dnsServers []string
var failureCounts = make(map[string]int) var failureCounts = make(map[string]int)
var showErrors bool var showErrors bool
var inProgressIPs sync.Map
func loadDNSServersFromFile(filePath string) ([]string, error) { func loadDNSServersFromFile(filePath string) ([]string, error) {
var servers []string var servers []string
@ -145,6 +146,12 @@ func worker(cidr *net.IPNet, resultsChan chan string) {
break break
} }
_, alreadyProcessing := inProgressIPs.LoadOrStore(ip.String(), true)
if alreadyProcessing {
continue
}
defer inProgressIPs.Delete(ip.String())
triedServers := make(map[string]bool) triedServers := make(map[string]bool)
retries := 10 retries := 10
success := false success := false
@ -216,14 +223,14 @@ func main() {
rand.Seed(time.Now().UnixNano()) rand.Seed(time.Now().UnixNano())
subnets, err := splitCIDR(cidr, concurrency*10) // Create more subnets than workers subnets, err := splitCIDR(cidr, concurrency*10)
if err != nil { if err != nil {
fmt.Printf("Error splitting CIDR: %s\n", err) fmt.Printf("Error splitting CIDR: %s\n", err)
os.Exit(1) os.Exit(1)
} }
if len(subnets) < concurrency { if len(subnets) < concurrency {
concurrency = len(subnets) // Limit concurrency to number of subnets concurrency = len(subnets)
} }
cidrChan := make(chan *net.IPNet, len(subnets)) cidrChan := make(chan *net.IPNet, len(subnets))