golang DNS服务器的简单实现操作
简单的DNS服务器
提供一个简单的可以查询域名和反向查询的DNS服务器。
dig命令主要用来从 DNS 域名服务器查询主机地址信息。
查找www.baidu.com的ip (A记录):
命令:dig @127.0.0.1 www.baidu.com
根据ip查找对应域名 (PTR记录):
命令:dig @127.0.0.1 -x 220.181.38.150
源码 :
package main import ( "fmt" "net" "golang.org/x/net/dns/dnsmessage" ) func main() { conn, err := net.ListenUDP("udp", &net.UDPAddr{Port: 53}) if err != nil { panic(err) } defer conn.Close() fmt.Println("Listing ...") for { buf := make([]byte, 512) _, addr, _ := conn.ReadFromUDP(buf) var msg dnsmessage.Message if err := msg.Unpack(buf); err != nil { fmt.Println(err) continue } go ServerDNS(addr, conn, msg) } } // address books var ( addressBookOfA = map[string][4]byte{ "www.baidu.com.": [4]byte{220, 181, 38, 150}, } addressBookOfPTR = map[string]string{ "150.38.181.220.in-addr.arpa.": "www.baidu.com.", } ) // ServerDNS serve func ServerDNS(addr *net.UDPAddr, conn *net.UDPConn, msg dnsmessage.Message) { // query info if len(msg.Questions) < 1 { return } question := msg.Questions[0] var ( queryTypeStr = question.Type.String() queryNameStr = question.Name.String() queryType = question.Type queryName, _ = dnsmessage.NewName(queryNameStr) ) fmt.Printf("[%s] queryName: [%s]\n", queryTypeStr, queryNameStr) // find record var resource dnsmessage.Resource switch queryType { case dnsmessage.TypeA: if rst, ok := addressBookOfA[queryNameStr]; ok { resource = NewAResource(queryName, rst) } else { fmt.Printf("not fount A record queryName: [%s] \n", queryNameStr) Response(addr, conn, msg) return } case dnsmessage.TypePTR: if rst, ok := addressBookOfPTR[queryName.String()]; ok { resource = NewPTRResource(queryName, rst) } else { fmt.Printf("not fount PTR record queryName: [%s] \n", queryNameStr) Response(addr, conn, msg) return } default: fmt.Printf("not support dns queryType: [%s] \n", queryTypeStr) return } // send response msg.Response = true msg.Answers = append(msg.Answers, resource) Response(addr, conn, msg) } // Response return func Response(addr *net.UDPAddr, conn *net.UDPConn, msg dnsmessage.Message) { packed, err := msg.Pack() if err != nil { fmt.Println(err) return } if _, err := conn.WriteToUDP(packed, addr); err != nil { fmt.Println(err) } } // NewAResource A record func NewAResource(query dnsmessage.Name, a [4]byte) dnsmessage.Resource { return dnsmessage.Resource{ Header: dnsmessage.ResourceHeader{ Name: query, Class: dnsmessage.ClassINET, TTL: 600, }, Body: &dnsmessage.AResource{ A: a, }, } } // NewPTRResource PTR record func NewPTRResource(query dnsmessage.Name, ptr string) dnsmessage.Resource { name, _ := dnsmessage.NewName(ptr) return dnsmessage.Resource{ Header: dnsmessage.ResourceHeader{ Name: query, Class: dnsmessage.ClassINET, }, Body: &dnsmessage.PTRResource{ PTR: name, }, } }
补充:Golang自定义DNS Nameserver
某些情况下我们希望程序通过自定义Nameserver去查询域名,而不希望通过操作系统给定的Nameserver,本文介绍如何在Golang中实现自定义Nameserver。
DNS解析过程
Golang中一般通过net.Resolver的LookupHost(ctx context.Context, host string) (addrs []string, err error)去实现域名解析,
解析过程如下:
检查本地hosts文件是否存在解析记录,存在即返回解析地址
不存在即根据resolv.conf中读取的nameserver发起递归查询
nameserver不断的向上级nameserver发起迭代查询
nameserver最终返回查询结果给请求者
用户可以通过修改/etc/resolv.conf来添加特定的nameserver,但某些场景下我们不希望更改系统配置。比如在kubernetes中,作为sidecar服务需要通过service去访问其他集群内服务,必须更改dnsPolicy为ClusterFirst,但这可能会影响其他容器的DNS查询效率。
自定义Nameserver
在Golang中自定义Nameserver,需要我们自己实现一个Resolver,如果是httpClient需要自定义DialContext()
Resolver实现如下:
// 默认dialer dialer := &net.Dialer{ Timeout: 1 * time.Second, } // 定义resolver resolver := &net.Resolver{ Dial: func(ctx context.Context, network, address string) (net.Conn, error) { return dialer.DialContext(ctx, "tcp", nameserver) // 通过tcp请求nameserver解析域名 }, }
自定义Dialer如下:
type Dialer struct { dialer *net.Dialer resolver *net.Resolver nameserver string } // NewDialer create a Dialer with user's nameserver. func NewDialer(dialer *net.Dialer, nameserver string) (*Dialer, error) { conn, err := dialer.Dial("tcp", nameserver) if err != nil { return nil, err } defer conn.Close() return &Dialer{ dialer: dialer, resolver: &net.Resolver{ Dial: func(ctx context.Context, network, address string) (net.Conn, error) { return dialer.DialContext(ctx, "tcp", nameserver) }, }, nameserver: nameserver, // 用户设置的nameserver }, nil } // DialContext connects to the address on the named network using // the provided context. func (d *Dialer) DialContext(ctx context.Context, network, address string) (net.Conn, error) { host, port, err := net.SplitHostPort(address) if err != nil { return nil, err } ips, err := d.resolver.LookupHost(ctx, host) // 通过自定义nameserver查询域名 for _, ip := range ips { // 创建链接 conn, err := d.dialer.DialContext(ctx, network, ip+":"+port) if err == nil { return conn, nil } } return d.dialer.DialContext(ctx, network, address) }
httpClient中自定义DialContext()如下:
ndialer, _ := NewDialer(dialer, nameserver) client := &http.Client{ Transport: &http.Transport{ DialContext: ndialer.DialContext, TLSHandshakeTimeout: 10 * time.Second, }, Timeout: timeout, }
总结
通过以上实现可解决自定义Nameserver,也可以在Dailer中添加缓存,实现DNS缓存。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持猪先飞。如有错误或未考虑完全的地方,望不吝赐教。
相关文章
- 一般来说为了达到隐藏自身 IP,我们都会选择一些 CDN 服务,目前用的最多的免费 CDN 是 Cloudflare。但是正常情况下 Cloudflare 是不能自己选择IP,有时候分配给我们的 IP 可能表现不佳,也只能将就着用。但是我们都希望可以用上 CF 比较快的 IP。那么自选IP,就显得很必要了。...2022-09-23
- 简单的php获取linux服务器状态的代码,不多说-直接上函数:复制代码 代码如下:function get_used_status(){ $fp = popen('top -b -n 2 | grep -E "^(Cpu|Mem|Tasks)"',"r");//获取某一时刻系统cpu和内存使用情况 $rs =...2014-05-31
- 这篇文章主要介绍了golang 调用 php7详解及实例的相关资料,需要的朋友可以参考下...2017-01-15
- 这篇文章主要介绍了Springboot+TCP监听服务器搭建过程,本文通过图文并茂的形式给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下...2020-10-28
- 前一段时间使用NetStat命令查看服务器端口时,发现服务器udp端口开放了好多,最少在1000个以上,当时事情比较多,没有管它,今天终于有点时间,仔细检查了一下,排除了这个问题. ...2016-01-27
- “主机,用户名,密码”得到连接、“数据库,sql,连接”得到结果,最后是结果的处理显示。当然,数据库连接是扩展库为我们完成的,我们能做的仅仅是处理结果而已。...2013-09-29
解决HttpPost+json请求---服务器中文乱码及其他问题
这篇文章主要介绍了解决HttpPost+json请求---服务器中文乱码及其他问题,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-01-222022年最新Cloudflare免费自选IP教程(非Partner)
写在开头众所周知,CF在去年底大规模禁用Host API key,使得Partner自选法失效.但最近,Cloudflare为所有计划添加了100个SaaS域免费额度(以前$2一个).经过一番摸索,这个船新的...2022-09-23- 这篇文章主要介绍了用golang实现替换某个文件中的字符串操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-25
- 这篇文章主要介绍了解决Golang json序列化字符串时多了\的情况,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-24
- 这篇文章主要介绍了Hyper-V尝试连接到服务器出错无效类的解决方法,需要的朋友可以参考下...2016-09-28
- 这篇文章主要介绍了mac使用Shell(终端)SSH连接远程服务器的方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2020-07-11
Golang中的自定义类型之间的转换的实现(type conversion)
这篇文章主要介绍了Golang中的自定义类型之间的转换的实现(type conversion),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧...2021-02-21- 这篇文章主要介绍了golang在GRPC中设置client的超时时间,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
- 这篇文章主要为大家详细介绍了js实现上传图片到服务器,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2021-04-11
- 这篇文章主要介绍了golang与php实现计算两个经纬度之间距离的方法,结合实例形式对比分析了Go语言与php进行经纬度计算的相关数学运算技巧,需要的朋友可以参考下...2016-07-29
解决golang处理http response碰到的问题和需要注意的点
这篇文章主要介绍了解决golang处理http response碰到的问题和需要注意的点,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2020-12-16- 这篇文章主要介绍了golang http使用踩过的坑与填坑指南,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...2021-04-27
c# HttpWebRequest通过代理服务器抓取网页内容应用介绍
在C#项目开发过程中可能会有些特殊的需求比如:用HttpWebRequest通过代理服务器验证后抓取网页内容,要想实现此方法并不容易,本文整理了一下,有需求的朋友可以参考下...2020-06-25- 这篇文章主要为大家详细介绍了uploader秒传图片到服务器的完整代码,具有一定的参考价值,感兴趣的小伙伴们可以参考一下...2017-04-27