golang DNS服務(wù)器的簡單實(shí)現(xiàn)操作
簡單的DNS服務(wù)器
提供一個(gè)簡單的可以查詢域名和反向查詢的DNS服務(wù)器。
dig命令主要用來從 DNS 域名服務(wù)器查詢主機(jī)地址信息。
查找www.baidu.com的ip (A記錄):
命令:dig @127.0.0.1 www.baidu.com
根據(jù)ip查找對應(yīng)域名 (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, }, } }
補(bǔ)充:Golang自定義DNS Nameserver
某些情況下我們希望程序通過自定義Nameserver去查詢域名,而不希望通過操作系統(tǒng)給定的Nameserver,本文介紹如何在Golang中實(shí)現(xiàn)自定義Nameserver。
DNS解析過程
Golang中一般通過net.Resolver的LookupHost(ctx context.Context, host string) (addrs []string, err error)去實(shí)現(xiàn)域名解析,
解析過程如下:
檢查本地hosts文件是否存在解析記錄,存在即返回解析地址
不存在即根據(jù)resolv.conf中讀取的nameserver發(fā)起遞歸查詢
nameserver不斷的向上級nameserver發(fā)起迭代查詢
nameserver最終返回查詢結(jié)果給請求者
用戶可以通過修改/etc/resolv.conf來添加特定的nameserver,但某些場景下我們不希望更改系統(tǒng)配置。比如在kubernetes中,作為sidecar服務(wù)需要通過service去訪問其他集群內(nèi)服務(wù),必須更改dnsPolicy為ClusterFirst,但這可能會(huì)影響其他容器的DNS查詢效率。
自定義Nameserver
在Golang中自定義Nameserver,需要我們自己實(shí)現(xiàn)一個(gè)Resolver,如果是httpClient需要自定義DialContext()
Resolver實(shí)現(xiàn)如下:
// 默認(rèn)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, // 用戶設(shè)置的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 { // 創(chuàng)建鏈接 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, }
總結(jié)
通過以上實(shí)現(xiàn)可解決自定義Nameserver,也可以在Dailer中添加緩存,實(shí)現(xiàn)DNS緩存。
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教。
相關(guān)文章
一文帶你使用Golang實(shí)現(xiàn)SSH客戶端
SSH?全稱為?Secure?Shell,是一種用于安全地遠(yuǎn)程登錄到網(wǎng)絡(luò)上的其他計(jì)算機(jī)的網(wǎng)絡(luò)協(xié)議,本文主要為大家詳細(xì)介紹了如何使用?Golang?實(shí)現(xiàn)?SSH?客戶端,需要的可以參考下2023-11-11Go?tablewriter庫提升命令行輸出專業(yè)度實(shí)例詳解
命令行工具大家都用過,如果是運(yùn)維人員可能會(huì)編寫命令行工具來完成各種任務(wù),命令行輸出的美觀和易讀性往往容易被忽視,很爛的輸出會(huì)讓人感覺不專業(yè),本文將介紹Go語言中牛逼的實(shí)戰(zhàn)工具tablewriter庫,使你在命令行輸出中展現(xiàn)出專業(yè)的一面2023-11-11go語言實(shí)現(xiàn)sftp包上傳文件和文件夾到遠(yuǎn)程服務(wù)器操作
這篇文章主要介紹了go語言實(shí)現(xiàn)sftp包上傳文件和文件夾到遠(yuǎn)程服務(wù)器操作,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-12-12Go語言實(shí)現(xiàn)遺傳算法的實(shí)例代碼
Go 是一個(gè)開源的編程語言,它能讓構(gòu)造簡單、可靠且高效的軟件變得容易。本文將重點(diǎn)介紹如何用Go語言實(shí)現(xiàn)遺傳算法。如果你還沒有參加過GoLang Tour,我還建議你快速看一下這門語言的介紹2017-11-11