当前位置：首页 > 文章列表 > Golang > Go教程 > golang DNS服务器的简单实现操作

golang DNS服务器的简单实现操作

来源：脚本之家 2022-12-31 11:33:01 0浏览收藏

对于一个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) 

<p><strong>补充：Golang自定义DNS Nameserver</strong></p>
<p>某些情况下我们希望程序通过自定义Nameserver去查询域名，而不希望通过操作系统给定的Nameserver，本文介绍如何在Golang中实现自定义Nameserver。</p>
<h2>DNS解析过程</h2>
<p>Golang中一般通过net.Resolver的LookupHost(ctx context.Context, host string) (addrs []string, err error)去实现域名解析，</p>
<h3>解析过程如下：</h3>
<p>检查本地hosts文件是否存在解析记录，存在即返回解析地址</p>
<p>不存在即根据resolv.conf中读取的nameserver发起递归查询</p>
<p>nameserver不断的向上级nameserver发起迭代查询</p>
<p>nameserver最终返回查询结果给请求者</p>
<p>用户可以通过修改/etc/resolv.conf来添加特定的nameserver，但某些场景下我们不希望更改系统配置。比如在kubernetes中，作为sidecar服务需要通过service去访问其他集群内服务，必须更改dnsPolicy为ClusterFirst，但这可能会影响其他容器的DNS查询效率。</p>
<h2>自定义Nameserver</h2>
<p>在Golang中自定义Nameserver，需要我们自己实现一个Resolver，如果是httpClient需要自定义DialContext()</p>
<h3>Resolver实现如下：</h3>

<pre class="brush:plain;">
// 默认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,
}