Skip to content

Commit 0c2b268

Browse files
committed
feat(dns_resolver): add custom DNS server support for reverse lookups
- Use config.DNSServers for reverse DNS lookups - Try each configured DNS server with fallback - Add proper port handling (default :53)
1 parent cc8f032 commit 0c2b268

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

modules/dns_resolver.go

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"context"
55
"ipmap/config"
66
"net"
7+
"strings"
78
"time"
89
)
910

@@ -18,6 +19,19 @@ func ReverseDNS(ip string) string {
1819
d := net.Dialer{
1920
Timeout: time.Second * 2,
2021
}
22+
// Use custom DNS servers if configured
23+
if len(config.DNSServers) > 0 {
24+
for _, dns := range config.DNSServers {
25+
dnsAddr := strings.TrimSpace(dns)
26+
if !strings.Contains(dnsAddr, ":") {
27+
dnsAddr = dnsAddr + ":53"
28+
}
29+
conn, err := d.DialContext(ctx, network, dnsAddr)
30+
if err == nil {
31+
return conn, nil
32+
}
33+
}
34+
}
2135
return d.DialContext(ctx, network, address)
2236
},
2337
}

0 commit comments

Comments
 (0)