golang DNS服务器的简单实现操作
来源:脚本之家
时间:2022-12-31 11:33:01 154浏览 收藏
对于一个Golang开发者来说,牢固扎实的基础是十分重要的,golang学习网就来带大家一点点的掌握基础知识点。今天本篇文章带大家了解《golang DNS服务器的简单实现操作》,主要介绍了服务器、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)补充: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缓存。
以上为个人经验,希望能给大家一个参考,也希望大家多多支持golang学习网。如有错误或未考虑完全的地方,望不吝赐教。
以上就是本文的全部内容了,是否有顺利帮助你解决问题?若是能给你带来学习上的帮助,请大家多多支持golang学习网!更多关于Golang的相关知识,也可关注golang学习网公众号。
-
102 收藏
-
347 收藏
-
245 收藏
-
456 收藏
-
420 收藏
-
233 收藏
-
322 收藏
-
181 收藏
-
316 收藏
-
244 收藏
-
300 收藏
-
- 前端进阶之JavaScript设计模式
- 设计模式是开发人员在软件开发过程中面临一般问题时的解决方案,代表了最佳的实践。本课程的主打内容包括JS常见设计模式以及具体应用场景,打造一站式知识长龙服务,适合有JS基础的同学学习。
- 立即学习 542次学习
-
- GO语言核心编程课程
- 本课程采用真实案例,全面具体可落地,从理论到实践,一步一步将GO核心编程技术、编程思想、底层实现融会贯通,使学习者贴近时代脉搏,做IT互联网时代的弄潮儿。
- 立即学习 507次学习
-
- 简单聊聊mysql8与网络通信
- 如有问题加微信:Le-studyg;在课程中,我们将首先介绍MySQL8的新特性,包括性能优化、安全增强、新数据类型等,帮助学生快速熟悉MySQL8的最新功能。接着,我们将深入解析MySQL的网络通信机制,包括协议、连接管理、数据传输等,让
- 立即学习 497次学习
-
- JavaScript正则表达式基础与实战
- 在任何一门编程语言中,正则表达式,都是一项重要的知识,它提供了高效的字符串匹配与捕获机制,可以极大的简化程序设计。
- 立即学习 487次学习
-
- 从零制作响应式网站—Grid布局
- 本系列教程将展示从零制作一个假想的网络科技公司官网,分为导航,轮播,关于我们,成功案例,服务流程,团队介绍,数据部分,公司动态,底部信息等内容区块。网站整体采用CSSGrid布局,支持响应式,有流畅过渡和展现动画。
- 立即学习 484次学习