golang 写一个dns中转代码

以下是一个简单的DNS中转代码示例，使用Go语言编写：

package main

import (
    "fmt"
    "net"
)

func main() {
    // 监听地址和端口
    addr := ":53"
    udpAddr, err := net.ResolveUDPAddr("udp", addr)
    if err != nil {
        panic(err)
    }

    // 创建UDP连接
    conn, err := net.ListenUDP("udp", udpAddr)
    if err != nil {
        panic(err)
    }
    defer conn.Close()

    // 循环处理DNS请求
    for {
        // 读取DNS请求
        buf := make([]byte, 1024)
        n, clientAddr, err := conn.ReadFromUDP(buf)
        if err != nil {
            fmt.Println("error reading from UDP socket:", err)
            continue
        }

        // 解析DNS请求
        msg := new(dns.Msg)
        err = msg.Unpack(buf[:n])
        if err != nil {
            fmt.Println("error unpacking DNS message:", err)
            continue
        }

        // 更改DNS请求
        for _, q := range msg.Question {
            // 将请求的域名添加前缀
            q.Name = "prefix." + q.Name
        }

        // 发送DNS请求
        dnsAddr, err := net.ResolveUDPAddr("udp", "8.8.8.8:53")
        if err != nil {
            fmt.Println("error resolving DNS server address:", err)
            continue
        }
        _, err = conn.WriteToUDP(msg.Pack(), dnsAddr)
        if err != nil {
            fmt.Println("error writing to UDP socket:", err)
            continue
        }

        // 读取DNS响应
        buf = make([]byte, 1024)
        n, _, err = conn.ReadFromUDP(buf)
        if err != nil {
            fmt.Println("error reading from UDP socket:", err)
            continue
        }

        // 解析DNS响应
        msg = new(dns.Msg)
        err = msg.Unpack(buf[:n])
        if err != nil {
            fmt.Println("error unpacking DNS message:", err)
            continue
        }

        // 更改DNS响应
        for _, a := range msg.Answer {
            // 将响应的IP地址添加后缀
            switch t := a.(type) {
            case *dns.A:
                a.Header().Name = a.Header().Name[:len(a.Header().Name)-1] + ".suffix."
                a.(*dns.A).A = net.ParseIP("1.2.3.4")
            case *dns.AAAA:
                a.Header().Name = a.Header().Name[:len(a.Header().Name)-1] + ".suffix."
                a.(*dns.AAAA).AAAA = net.ParseIP("::1")
            default:
                continue
            }
        }

        // 发送DNS响应
        _, err = conn.WriteToUDP(msg.Pack(), clientAddr)
        if err != nil {
            fmt.Println("error writing to UDP socket:", err)
            continue
        }
    }
}

这个示例程序会监听本地的UDP 53端口，当收到DNS请求时，会将请求的域名添加前缀并发送到Google Public DNS服务器（8.8.8.8），等待响应后将响应的IP地址添加后缀并返回给客户端。您可以根据自己的需求更改代码来实现您所需的DNS中转功能