欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

golang DNS服務(wù)器的簡單實(shí)現(xiàn)操作

 更新時(shí)間:2021年04月30日 15:00:30   作者:風(fēng)格色  
這篇文章主要介紹了golang DNS服務(wù)器的簡單實(shí)現(xiàn)操作,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧

簡單的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)文章

  • 在Go中格式化字符串的幾種常用方法

    在Go中格式化字符串的幾種常用方法

    Go對字符串格式化提供了良好的支持,這篇文章主要給大家介紹了關(guān)于在Go中格式化字符串的幾種常用方法,文中通過代碼介紹的非常詳細(xì),對大家學(xué)習(xí)或者使用Go具有一定的參考價(jià)值,需要的朋友可以參考下
    2023-10-10
  • Go使用sync.Pool提高性能的代碼示例

    Go使用sync.Pool提高性能的代碼示例

    在高性能應(yīng)用程序中,頻繁的內(nèi)存分配和回收是性能瓶頸的常見原因之一,Go 語言提供了 sync.Pool 類型,它可以用來存儲(chǔ)和重用臨時(shí)對象,本文將詳細(xì)介紹如何在 Go 中使用 sync.Pool,并通過實(shí)際代碼示例來展示其對性能的提升效果,需要的朋友可以參考下
    2024-04-04
  • 一文帶你使用Golang實(shí)現(xiàn)SSH客戶端

    一文帶你使用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-11
  • Go?tablewriter庫提升命令行輸出專業(yè)度實(shí)例詳解

    Go?tablewriter庫提升命令行輸出專業(yè)度實(shí)例詳解

    命令行工具大家都用過,如果是運(yùn)維人員可能會(huì)編寫命令行工具來完成各種任務(wù),命令行輸出的美觀和易讀性往往容易被忽視,很爛的輸出會(huì)讓人感覺不專業(yè),本文將介紹Go語言中牛逼的實(shí)戰(zhàn)工具tablewriter庫,使你在命令行輸出中展現(xiàn)出專業(yè)的一面
    2023-11-11
  • go語言實(shí)現(xiàn)sftp包上傳文件和文件夾到遠(yuǎn)程服務(wù)器操作

    go語言實(shí)現(xiàn)sftp包上傳文件和文件夾到遠(yuǎn)程服務(wù)器操作

    這篇文章主要介紹了go語言實(shí)現(xiàn)sftp包上傳文件和文件夾到遠(yuǎn)程服務(wù)器操作,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧
    2020-12-12
  • Go語言學(xué)習(xí)之鏈表的使用詳解

    Go語言學(xué)習(xí)之鏈表的使用詳解

    鏈表是一種物理存儲(chǔ)單元上非連續(xù)、非順序的存儲(chǔ)結(jié)構(gòu),數(shù)據(jù)元素的邏輯順序是通過鏈表中的指針鏈接次序?qū)崿F(xiàn)的。本文將詳細(xì)為大家介紹Go語言中鏈表的使用,感興趣的可以了解一下
    2022-04-04
  • go語言獲取系統(tǒng)盤符的方法

    go語言獲取系統(tǒng)盤符的方法

    這篇文章主要介紹了go語言獲取系統(tǒng)盤符的方法,涉及Go語言調(diào)用winapi獲取系統(tǒng)硬件信息的技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下
    2015-03-03
  • Go語言實(shí)現(xiàn)遺傳算法的實(shí)例代碼

    Go語言實(shí)現(xiàn)遺傳算法的實(shí)例代碼

    Go 是一個(gè)開源的編程語言,它能讓構(gòu)造簡單、可靠且高效的軟件變得容易。本文將重點(diǎn)介紹如何用Go語言實(shí)現(xiàn)遺傳算法。如果你還沒有參加過GoLang Tour,我還建議你快速看一下這門語言的介紹
    2017-11-11
  • Go語言中Slice常見陷阱與避免方法詳解

    Go語言中Slice常見陷阱與避免方法詳解

    這篇文章主要為大家詳細(xì)介紹的是 Go 語言中的 Slice 的常見陷阱以及如何避免這些錯(cuò)誤,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以學(xué)習(xí)一下
    2023-02-02
  • Go如何實(shí)現(xiàn)HTTP請求限流示例

    Go如何實(shí)現(xiàn)HTTP請求限流示例

    本篇文章主要介紹了Go如何實(shí)現(xiàn)HTTP請求限流示例,小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,也給大家做個(gè)參考。一起跟隨小編過來看看吧
    2018-04-04

最新評論