DNS客户端的实现还是用了Go语言,毕竟这个最熟悉。我的实现只是简单的实现了发送DNS请求,解析响应内容这些功能,对于多线程并发等机制都没有考虑。
我最先参考了《计算机网络》,上面提了一句,DNS请求就是发送一个UDP数据包,然后我就天真了,直接把域名的字符串用UDP发送了。结果就是没有响应。
后来就老老实实的分析协议,协议的分析请参考上一篇文章《DNS协议分析》。
域名服务器选用了阿里提供的233.5.5.5,53端口。详情可以查看阿里DNS官网。
先说一下Golang发送UDP数据报的方法。所有语言这块儿好像都长差不多,我就不多介绍了。
service := "223.5.5.5:53"
udpAddr, err := net.ResolveUDPAddr("udp", service)
checkError(err)
conn, err := net.DialUDP("udp", nil, udpAddr)
checkError(err)
_, err = conn.Write(out.Pack())
checkError(err)
buf := []byte{}
buf = make([]byte, 512)
n, err := conn.Read(buf[0:])
checkError(err)
接着就是发送和接收要用到的消息体,它由一个TransationID、Flags、查询数Questions、Answer是响应数和查询Question组成。查询Question由域名、类别、分类组成。查询结果ANswer由查询域名、记录类型、分类、数据长度和Primaryname组成。这些结构可以定义为:
type DnsMsg struct {
Id uint16
Bits uint16
Qdcount, Ancount, Nscount, Arcount uint16
Questions []dnsQuestion
Answers []dnsAnswer
}
type dnsQuestion struct {
Name string `net:"domain-name"`
Qtype uint16
Qclass uint16
}
type dnsAnswer struct {
Name uint16
Qtype uint16
Qclass uint16
QLive uint32
QLen uint16
CName string `net:"domain-name"`
}
DNS协议规定是两个字节的,都用uint16进行处理,Question里面的Name和Answer里面定义的CNAME应该定义成[]byte
,但是为了方便查看,先定义成了字符串,最后编码的时候再处理就行了。
有了结构体,就可以为这些结构体赋值,然后将其编码成字节码进行发送。编码方法如下:
func (this *DnsMsg) Pack() []byte {
bs := make([]byte, 12)
binary.BigEndian.PutUint16(bs[0:2], this.Id)
binary.BigEndian.PutUint16(bs[2:4], this.Bits)
binary.BigEndian.PutUint16(bs[4:6], uint16(len(this.Questions)))
binary.BigEndian.PutUint16(bs[6:8], this.Ancount)
binary.BigEndian.PutUint16(bs[8:10], this.Nscount)
binary.BigEndian.PutUint16(bs[10:12], this.Arcount)
ds := strings.Split(this.Questions[0].Name, ".")
for _, d := range ds {
bs = append(bs, byte(len(d)))
bs = append(bs, []byte(d)...)
}
bs = append(bs, 0)
temp := make([]byte, 2)
binary.BigEndian.PutUint16(temp, this.Questions[0].Qtype)
bs = append(bs, temp...)
binary.BigEndian.PutUint16(temp, this.Questions[0].Qclass)
bs = append(bs, temp...)
return bs
}
DNS头是固定的12个字节,所以先bs := make([]byte, 12)
申请12个字节空间。接着,填充这些字节。就是将结构体里面的uint16
转成[]byte
。这要用到binary
包进行转换。最后就是填充Question,要将域名进行转换。转换剔除了域名里面的.
,改成其后面部分长度的信息,也就是把www.cyeam.com
换成3www5cyeam3com0
。转换完成之后,还要添加一位0来表示结束。最后,再加上请求的类型和分类即可。
这些都请求成功之后,打印结果,终于能看到点东西了:一堆乱码里面有一些字母,能看得出来是我的域名的CDN地址。
最后,就是解析响应内容。响应和请求格式是一样的,区别就是最后比请求多了一些响应内容。
DNS头的12个字节和请求里面是完全一样的,返过来解析就可以了。需要注意两点,响应的头两个字节和请求的头两个字节是完全一样的,如果不一样,忽略这个请求。还有,请求的时候ancount
是0,这个代表响应数量,在响应里面就会是找到的结果数。
查询请求的Question也会在响应里面,所以也需要进行解析。请求查询时Question有可能是多个,这里都处理了一下。从第13个字节开始处理:
i := 13
for ; j < int(res.Qdcount); j++ {
domain_count := int(buf[i-1])
question := dnsQuestion{}
for buf[i] != 0 {
if domain_count > 0 {
question.Name += string(buf[i:i+domain_count]) + "."
i += domain_count
domain_count = 0
} else {
domain_count = int(buf[i])
i++
}
}
i++
question.Name = strings.TrimRight(question.Name, ".")
question.Qtype = binary.BigEndian.Uint16(buf[i : i+2])
question.Qclass = binary.BigEndian.Uint16(buf[i+2 : i+4])
i += 4
res.Questions[j] = question
}
剩下的东西都是Answer。前5个部分都是固定长度,分别是2、2、2、4、2。
answer := dnsAnswer{}
answer.Name = binary.BigEndian.Uint16(buf[i : i+2])
i += 2
answer.Qtype = binary.BigEndian.Uint16(buf[i : i+2])
i += 2
answer.Qclass = binary.BigEndian.Uint16(buf[i : i+2])
i += 2
answer.QLive = binary.BigEndian.Uint32(buf[i : i+4])
i += 4
answer.QLen = binary.BigEndian.Uint16(buf[i : i+2])
i += 2
后面的CName是关键,解析也要按照类型进行。一种类型是CNAME
类型,也就是别名,我的域名www.cyeam.com
的别名就是vm68h.x.incapdns.net
,这是我的CDN。还有一种就是真实IP,返回别名也没多大用,IP才能解决问题,所以Answer里面的第二个就是返回这个别名的真实IP149.126.77.152
。所以一个按字符串处理,一个按数字处理。Answer里面的分类说明了解析类型,如果值是1,说明是A
记录,也就是IP,如果值是5,是CNAME
记录。解析Answer代码如下:
if answer.Qtype == dnsTypeCNAME {
domain_count := int(buf[i])
i++
for buf[i] != 0 {
if domain_count > 0 {
answer.CName += string(buf[i:i+domain_count]) + "."
i += domain_count
domain_count = 0
} else {
domain_count = int(buf[i])
i++
}
}
i++
answer.CName = strings.TrimRight(answer.CName, ".")
} else if answer.Qtype == dnsTypeA {
for m := 0; m < int(answer.QLen); m++ {
answer.CName += fmt.Sprintf("%d.", buf[i+m])
}
answer.CName = strings.TrimRight(answer.CName, ".")
}
Answer里面还会包含结果的长度信息,用这个来遍历得到结果内容。
完整请求如下:
service := "223.5.5.5:53"
udpAddr, err := net.ResolveUDPAddr("udp", service)
checkError(err)
conn, err := net.DialUDP("udp", nil, udpAddr)
checkError(err)
question := dnsQuestion{"www.cyeam.com", dnsTypeA, dnsClassINET}
out := DnsMsg{}
out.Id = 2015
out.Bits |= _RD
out.Questions = append(out.Questions, question)
fmt.Println(out.Pack())
_, err = conn.Write(out.Pack())
checkError(err)
buf := []byte{}
buf = make([]byte, 512)
n, err := conn.Read(buf[0:])
checkError(err)
fmt.Println(buf[0:n])
fmt.Println(out.Unpack(buf[0:n]))
os.Exit(0)
解析完成后的结构体打印结果如下:
&{2015 33152 1 2 0 0 [{www.cyeam.com 1 1}] [{49164 5 1 30 22 vm68h.x.incapdns.net} {49195 1 1 30 4 149.126.77.152}]}
参考文献
- 《计算机网络(第五版)》谢希仁
- What does a DNS request look like? - serverfault
原文链接:DNS协议Golang实现,转载请注明来源!
有疑问加站长微信联系(非本文作者)
![](https://static.golangjob.cn/static/img/footer.png?imageView2/2/w/280)