Go语言中实现UDP Ping与ICMP端口不可达错误解析

package main

import (
    "errors"
    "fmt"
    "net"
    "time"
)

// UDPSendAndReceive 尝试向目标地址发送UDP数据并读取响应
func UDPSendAndReceive(srcAddr, dstAddr *net.UDPAddr, message []byte) ([]byte, error) {
    conn, err := net.ListenUDP("udp4", srcAddr)
    if err != nil {
        return nil, fmt.Errorf("无法监听UDP端口: %w", err)
    }
    defer conn.Close()

    // 设置发送和读取超时，避免无限等待
    conn.SetWriteDeadline(time.Now().Add(100 * time.Millisecond))
    conn.SetReadDeadline(time.Now().Add(250 * time.Millisecond))

    // 发送UDP数据包
    _, err = conn.WriteTo(message, dstAddr)
    if err != nil {
        return nil, fmt.Errorf("发送UDP数据失败: %w", err)
    }

    // 尝试读取响应
    buffer := make([]byte, 1024)
    n, _, err := conn.ReadFromUDP(buffer)
    if err != nil {
        // 如果是超时错误，可能意味着端口不可达或无响应
        if netErr, ok := err.(net.Error); ok && netErr.Timeout() {
            return nil, errors.New("UDP读取超时，可能端口不可达或无响应")
        }
        return nil, fmt.Errorf("读取UDP响应失败: %w", err)
    }

    return buffer[:n], nil
}

func main() {
    // 示例：发送到本地一个可能关闭的端口
    src := &net.UDPAddr{IP: net.ParseIP("127.0.0.1"), Port: 0} // 0表示系统分配一个临时端口
    dst := &net.UDPAddr{IP: net.ParseIP("127.0.0.1"), Port: 8888} // 假设这是一个未监听的端口

    fmt.Printf("尝试向 %s 发送UDP Ping...\n", dst.String())
    response, err := UDPSendAndReceive(src, dst, []byte("PING"))

    if err != nil {
        fmt.Printf("发生错误: %v\n", err)
    } else {
        fmt.Printf("收到响应: %s\n", string(response))
    }

    // 示例：发送到本地一个可能监听的端口 (如果8080有UDP服务)
    // dst2 := &net.UDPAddr{IP: net.ParseIP("127.0.0.1"), Port: 8080}
    // fmt.Printf("尝试向 %s 发送UDP Ping...\n", dst2.String())
    // response2, err2 := UDPSendAndReceive(src, dst2, []byte("PING"))
    // if err2 != nil {
    //  fmt.Printf("发生错误: %v\n", err2)
    // } else {
    //  fmt.Printf("收到响应: %s\n", string(response2))
    // }
}

    0                   1                   2                   3
    0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
   |     Type      |     Code      |          Checksum             |
   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
   |                             unused                            |
   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
   |      Internet Header + 64 bits of Original Data Datagram      |
   +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+

package main

import (
    "fmt"
)

// ParseICMPPortUnreachable 尝试解析ICMP数据包以识别端口不可达错误
func ParseICMPPortUnreachable(icmpPacketBytes []byte) bool {
    if len(icmpPacketBytes) < 2 {
        fmt.Println("ICMP数据包长度不足，无法解析类型和代码。")
        return false
    }

    icmpType := icmpPacketBytes[0]
    icmpCode := icmpPacketBytes[1]

    if icmpType == 3 && icmpCode == 3 {
        fmt.Println("检测到ICMP端口不可达错误 (Type 3, Code 3)。")
        return true
    } else {
        fmt.Printf("检测到其他ICMP消息: Type=%d, Code=%d\n", icmpType, icmpCode)
        return false
    }
}

func main() {
    // 示例：模拟一个ICMP端口不可达数据包的前两个字节
    // 实际场景中，这些字节会从原始套接字读取
    portUnreachablePacket := []byte{3, 3, 0x12, 0x34 /* ... 更多ICMP数据 */ }
    otherICMPPacket := []byte{3, 0, 0x56, 0x78 /* ... 更多ICMP数据 */ } // 模拟网络不可达

    fmt.Println("--- 解析端口不可达示例 ---")
    ParseICMPPortUnreachable(portUnreachablePacket)

    fmt.Println("\n--- 解析其他ICMP示例 ---")
    ParseICMPPortUnreachable(otherICMPPacket)

    fmt.Println("\n--- 解析短数据包示例 ---")
    ParseICMPPortUnreachable([]byte{3})
}