From 8520e8483ed81d923a925f7fa4a1b8df835977a7 Mon Sep 17 00:00:00 2001
From: CZ <cfd0123456789@qq.com>
Date: Mon, 6 Jun 2022 10:52:09 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=20IPV6=20=E6=94=AF=E6=8C=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 README.md |  9 +++++++++
 main.go   | 32 +++++++++++++++++++++++---------
 2 files changed, 32 insertions(+), 9 deletions(-)

diff --git a/README.md b/README.md
index b971cd9..b4734e2 100644
--- a/README.md
+++ b/README.md
@@ -60,3 +60,12 @@ cdnlookup.exe -d www.taobao.com
 59.47.225.XXX
 ......
 ````
+IPV6 查询
+
+````
+cdnlookup.exe -d www.jd.com -6 -ip 240e:382:701:7700:600c:5c8:0000:0000
+240e:c3:2800::26
+240e:c3:2800::22
+240e:95d:c02:200::3a
+
+````
\ No newline at end of file
diff --git a/main.go b/main.go
index 7d454d4..0b0f594 100644
--- a/main.go
+++ b/main.go
@@ -11,22 +11,34 @@ import (
 
 var IpMap map[string]bool
 
-func dnsquery(domain string, ip string, DnsServer string, OnlyIp bool, repeat int) {
+func dnsquery(domain string, ip string, DnsServer string, OnlyIp bool, repeat int, v6 bool) {
 	if !strings.HasSuffix(domain, ".") {
 		domain += "."
 	}
 	c := new(dns.Client)
 	m := new(dns.Msg)
-	m.SetQuestion(domain, dns.TypeA)
+	if v6 {
+		m.SetQuestion(domain, dns.TypeAAAA)
+	} else {
+		m.SetQuestion(domain, dns.TypeA)
+	}
+
 	o := new(dns.OPT)
 	o.Hdr.Name = "."
 	o.Hdr.Rrtype = dns.TypeOPT
 	e := new(dns.EDNS0_SUBNET) //EDNS
 	e.Code = dns.EDNS0SUBNET
-	e.Family = 1         // 1 IPv4 2 IPv6
-	e.SourceNetmask = 24 //  地址掩码 一般为24 (谷歌不支持大于24)
+	if v6 {
+		e.Family = 2         // 1 IPv4 2 IPv6
+		e.SourceNetmask = 56 //  地址掩码 ipv4 一般为 /24  ipv6为 /56
+		e.Address = net.ParseIP(ip).To16()
+	} else {
+		e.Family = 1
+		e.SourceNetmask = 24
+		e.Address = net.ParseIP(ip).To4()
+	}
+
 	e.SourceScope = 0
-	e.Address = net.ParseIP(ip).To4()
 	o.Option = append(o.Option, e)
 	m.Extra = append(m.Extra, o)
 	for i := 0; i < repeat; i++ {
@@ -43,6 +55,8 @@ func dnsquery(domain string, ip string, DnsServer string, OnlyIp bool, repeat in
 				} else {
 					println(answer.(*dns.A).A.String())
 				}
+			} else if answer.Header().Rrtype == dns.TypeAAAA {
+				IpMap[answer.(*dns.AAAA).AAAA.String()] = true
 			}
 		}
 	}
@@ -56,17 +70,18 @@ func main() {
 	var ip = flag.String("ip", "", "client ip")
 	var OnlyIp = flag.Bool("i", false, "Only output ip addr")
 	var repeat = flag.Int("r", 1, "repeat query rounds")
+	var v6 = flag.Bool("6", false, "query AAAA (ipv6)")
 	flag.Parse()
 	IpMap = make(map[string]bool)
-	if *ip != "" {
+	if (*ip != "") || (*v6) {
 		*OnlyIp = true
-		dnsquery(*domain, *ip, *DnsServer, *OnlyIp, *repeat)
+		dnsquery(*domain, *ip, *DnsServer, *OnlyIp, *repeat, *v6)
 	} else {
 		for city, ip := range CityMap {
 			if !*OnlyIp {
 				fmt.Println(city)
 			}
-			dnsquery(*domain, ip, *DnsServer, *OnlyIp, *repeat)
+			dnsquery(*domain, ip, *DnsServer, *OnlyIp, *repeat, *v6)
 
 		}
 	}
@@ -76,6 +91,5 @@ func main() {
 			println(ip)
 		}
 	}
-	//log.Println(in)
 
 }