手写一个UDP端口探测程序

原创于 2025-04-29 22:14:27 发布 · 797 阅读

14 ·

CC 4.0 BY-SA版权

文章标签：

#udp #网络协议 #网络

网络专栏收录该内容

12 篇文章

订阅专栏

今天让我们一起来看看UDP端口探测是怎么个事，本篇文章主要用到SOCK_DGRAM 套接字类型。

socket套接字相关的文章：

socket套接字你搞清楚了吗?

如何使用socket系统调用创建TCP三次握手呢？

SOCK_DGRAM 是 Linux 套接字(Socket)的一种类型，它提供无连接的、不可靠的数据报服务。这种套接字通常用于 UDP (User Datagram Protocol) 用户数据报协议。

UDP报文结构

特性

无连接：通信前不需要建立连接
数据报服务：数据以独立的数据包形式传输
不可靠传输：不保证数据到达顺序，可能丢失或重复
边界保留：数据保持发送时的消息边界
高效：相比 SOCK_STREAM 开销更小

示例：向目标服务发送UDP报文，测试目标服务是否回包（验证端口是否存活）

func main() {
    // 创建套接字
	fd, err := syscall.Socket(
		syscall.AF_INET,
		syscall.SOCK_DGRAM,
		syscall.IPPROTO_UDP,
	)
	if err != nil {
		log.Fatalf("create socket err: %v", err)
	}
	defer syscall.Close(fd)

	// 设置超时
	tv := syscall.Timeval{Sec: 3, Usec: 0}
	if err := syscall.SetsockoptTimeval(
		fd, syscall.SOL_SOCKET, syscall.SO_RCVTIMEO, &tv); err != nil {
		log.Fatalf("Set timeout error: %v\n", err)
		return
	}

	// 构造目标IP及端口
	dstIP := net.ParseIP("192.168.1.5").To4()
	dstPort := 1234
	dstAddr := &syscall.SockaddrInet4{Port: dstPort}
	copy(dstAddr.Addr[:], dstIP.To4())

	// 发送探测数据
	msg := []byte("hello udp!")
	if err := syscall.Sendto(fd, msg, 0, dstAddr); err != nil {
		log.Fatalf("send err: %v", err)
		return
	}

	// 尝试接收响应
	buf := make([]byte, 1024)
	_, _, err = syscall.Recvfrom(fd, buf, 0)
	if err != nil {
		if err == syscall.EAGAIN || err == syscall.EWOULDBLOCK {
			log.Fatalf("No response received\n")
		} else {
			log.Fatalf("Receive error: %v\n", err)
		}
		return
	}
	log.Fatalf("udp probe success!\n")
}