使用第三方 DNS 库解析 Golang 中的 DNS 数据包

go get github.com/miekg/dns

package main

import (
    "fmt"
    "github.com/miekg/dns"
    "log"
    "net"
)

func main() {
    // 模拟一个 DNS 数据包 (例如从网络接收到的)
    // 这里使用一个简单的 A 记录查询示例
    packetData := []byte{
        0xaa, 0xaa, 0x01, 0x00, 0x00, 0x01, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x03, 0x77,
        0x77, 0x77, 0x05, 0x62, 0x61, 0x69, 0x64, 0x75, 0x02, 0x63, 0x6e, 0x00, 0x00, 0x01,
        0x00, 0x01,
    }

    // 创建一个 dns.Msg 对象
    msg := new(dns.Msg)

    // 使用 Unpack 方法解析数据包
    err := msg.Unpack(packetData)
    if err != nil {
        log.Fatalf("Failed to unpack DNS message: %v", err)
    }

    // 打印解析后的 DNS 消息
    fmt.Printf("DNS Header: %+v\n", msg.MsgHdr)

    // 遍历并打印 DNS 问题 (查询) 部分
    for _, question := range msg.Question {
        fmt.Printf("Question: %+v\n", question)
        fmt.Printf("  Name: %s\n", question.Name)
        fmt.Printf("  Qtype: %s\n", dns.TypeToString[question.Qtype])
        fmt.Printf("  Qclass: %s\n", dns.ClassToString[question.Qclass])
    }

    // 通常，你会在这里处理 DNS 答案、权威记录和额外记录
    // 但在这个例子中，我们只解析了问题部分

    // 示例：发送 DNS 查询
    c := new(dns.Client)
    m := new(dns.Msg)
    m.SetQuestion("www.example.com.", dns.TypeA)
    m.RecursionDesired = true

    r, _, err := c.Exchange(m, net.JoinHostPort("8.8.8.8", "53")) // Google Public DNS
    if err != nil {
        log.Fatalf("Failed to exchange DNS message: %v", err)
    }

    if r.Rcode != dns.RcodeSuccess {
        log.Fatalf("DNS query failed with code: %v", dns.RcodeToString[r.Rcode])
    }

    // 打印返回的答案
    for _, ans := range r.Answer {
        fmt.Printf("Answer: %+v\n", ans)
        // 进一步处理答案，例如提取 IP 地址
        if a, ok := ans.(*dns.A); ok {
            fmt.Printf("  IP Address: %s\n", a.A.String())
        }
    }
}