Golang DNS 中转代码示例 - 实现域名修改和IP地址替换

以下是一个简单的 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中转功能。