Go语言中处理DNS报文的实践：推荐miekg/dns库的使用

go get github.com/miekg/dns

package main

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

func main() {
    // 示例：一个简单的DNS响应报文的字节数据
    // 这是一个对 example.com A记录的查询响应，包含一个A记录
    // 实际应用中，这些字节会从网络连接中读取
    rawDNSResponse := []byte{
        0x00, 0x01, // ID
        0x81, 0x80, // Flags: Standard query response, No error
        0x00, 0x01, // Questions: 1
        0x00, 0x01, // Answer RRs: 1
        0x00, 0x00, // Authority RRs: 0
        0x00, 0x00, // Additional RRs: 0
        // Question Section
        0x07, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 0x03, 'c', 'o', 'm', 0x00, // example.com
        0x00, 0x01, // Type: A (1)
        0x00, 0x01, // Class: IN (1)
        // Answer Section
        0x07, 'e', 'x', 'a', 'm', 'p', 'l', 'e', 0x03, 'c', 'o', 'm', 0x00, // example.com
        0x00, 0x01, // Type: A (1)
        0x00, 0x01, // Class: IN (1)
        0x00, 0x00, 0x00, 0x3c, // TTL: 60 seconds
        0x00, 0x04, // Data length: 4 bytes
        0xc0, 0x00, 0x00, 0x01, // Address: 192.0.0.1 (example IP)
    }

    msg := new(dns.Msg)
    err := msg.Unpack(rawDNSResponse)
    if err != nil {
        log.Fatalf("Failed to unpack DNS message: %v", err)
    }

    fmt.Printf("DNS Message ID: %d\n", msg.Id)
    fmt.Printf("Response Code: %s\n", dns.RcodeToString[msg.Rcode])

    fmt.Println("\nQuestions:")
    for _, q := range msg.Question {
        fmt.Printf("  Name: %s, Type: %s, Class: %s\n", q.Name, dns.TypeToString[q.Qtype], dns.ClassToString[q.Qclass])
    }

    fmt.Println("\nAnswers:")
    for _, rr := range msg.Answer {
        fmt.Printf("  %s\n", rr.String()) // RR的String()方法提供了友好的输出
    }
}

package main

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

func main() {
    // 1. 构建DNS查询报文
    m := new(dns.Msg)
    m.SetQuestion(dns.Fqdn("www.example.com."), dns.TypeA) // 查询 www.example.com 的 A 记录
    m.RecursionDesired = true // 请求递归查询

    // 2. 将报文打包成字节流（可选，通常由Client自动完成）
    // packedMsg, err := m.Pack()
    // if err != nil {
    //  log.Fatalf("Failed to pack DNS message: %v", err)
    // }
    // fmt.Printf("Packed DNS message length: %d bytes\n", len(packedMsg))

    // 3. 创建DNS客户端并发送查询
    c := new(dns.Client)
    c.Net = "udp" // 使用UDP协议
    c.Timeout = 5 * time.Second // 设置超时

    // 假设的DNS服务器地址，可以是公共DNS（如Google的8.8.8.8）或本地DNS
    dnsServer := "8.8.8.8:53"

    // 发送查询并获取响应
    r, _, err := c.Exchange(m, dnsServer)
    if err != nil {
        log.Fatalf("DNS query failed: %v", err)
    }

    if r == nil {
        log.Fatal("No DNS response received.")
    }

    // 4. 处理DNS响应
    if r.Rcode != dns.RcodeSuccess {
        fmt.Printf("DNS query failed with Rcode: %s\n", dns.RcodeToString[r.Rcode])
        return
    }

    fmt.Printf("DNS Response ID: %d\n", r.Id)
    fmt.Printf("Response Code: %s\n", dns.RcodeToString[r.Rcode])

    fmt.Println("\nAnswers:")
    if len(r.Answer) == 0 {
        fmt.Println("  No answers found.")
    }
    for _, rr := range r.Answer {
        fmt.Printf("  %s\n", rr.String())
        if a, ok := rr.(*dns.A); ok {
            fmt.Printf("    IP Address: %s\n", a.A.String())
        }
    }
}

// 查询MX记录
m.SetQuestion(dns.Fqdn("example.com."), dns.TypeMX)

// 在响应中处理MX记录
for _, rr := range r.Answer {
    if mx, ok := rr.(*dns.MX); ok {
        fmt.Printf("  MX Record: Host=%s, Preference=%d\n", mx.Mx, mx.Preference)
    }
}