《网络编程》高级 UDP 套接字编程

概述

         UDP 是一个无连接、不可靠的数据报协议,任何可靠传输都需由应用程序提供,例如:超时重传、序列号应答机制,但是它在某些场合使用效率高,方便。它支持广播和多播。有关《基本 UDP 套接字编程》参照该文,这里只是在那个基础上,记录一些在 UDP 编程中容易出现的问题。

辅助数据

        辅助数据(也称为控制信息)可通过调用 recvmsg 和 sendmsg 函数使用,这里两个函数的定义可参考文章《高级 I/O》,使用 msghdr 结构体中的 msg_control 和 msg_controllen 成员发送和接收辅助数据。

以下是辅助数据的各种用途:


辅助数据由一个或多个辅助数据对象构成,每个对象以一个结构体 cmsghdr 开头。结构体定义如下:

   /* 结构 cmsghdr*/
   struct cmsghdr {
       socklen_t cmsg_len;    /* data byte count, including header */
       int       cmsg_level;  /* originating protocol */
       int       cmsg_type;   /* protocol-specific type */
       /* followed by unsigned char cmsg_data[]; */
   };

        而 msg_control 指向第一个辅助数据对象,辅助数据的总长度则有 msg_controllen 指定。每个对象开头都是一个描述该对象的 cmsghdr 结构。在 cmsg_type 成员和实际数据之间可以有填充字节,从数据结尾处到下一个辅助数据对象之前也可以有填充字节。如下图所示:



以下是处理辅助数据的宏定义:

 #include <sys/socket.h>

    /* 宏定义 */
   struct cmsghdr *CMSG_FIRSTHDR(struct msghdr *msgh);
   struct cmsghdr *CMSG_NXTHDR(struct msghdr *msgh, struct cmsghdr *cmsg);
   size_t CMSG_ALIGN(size_t length);
   size_t CMSG_SPACE(size_t length);
   size_t CMSG_LEN(size_t length);
   unsigned char *CMSG_DATA(struct cmsghdr *cmsg);

   /* 结构 cmsghdr*/
   struct cmsghdr {
       socklen_t cmsg_len;    /* data byte count, including header */
       int       cmsg_level;  /* originating protocol */
       int       cmsg_type;   /* protocol-specific type */
       /* followed by unsigned char cmsg_data[]; */
   };
/* 这些宏定义用来创建和访问不是套接字负载部分的control控制信息(也称为辅助数据);
 * 这些控制信息可能包含所接收数据报的接口和各种很少使用的描述头等信息;
 */
 CMSG_FIRSTHDR(); /* 返回指向辅助数据缓冲区的第一个cmsghdr结构的指针,若无辅助数据则返回NULL; */

 CMSG_NXTHDR();/* 返回指向下一个cmsghdr结构的指针,若不再有辅助对象则返回NULL;*/

 CMSG_ALIGN();/* given a length, returns it including the required  align‐
       ment.  This is a constant expression. */

 CMSG_SPACE();/* 返回给定数据量的一个辅助数据对象的大小 */

 CMSG_DATA();/* 返回指向与cmsghdr结构关联的数据的第一个字节的指针; */

 CMSG_LEN();/* 返回给定数据量的存放到cmsg_len中的值;*/


        CMSG_LEN 和 CMSG_SPACE 的区别在于,前者不计辅助数据对象中数据部分之后可能的填充字节,因而返回的是用于存放在 cmsg_len 成员中的值,后者是计上尾处可能的填充字节,因而返回的是为辅助数据对象动态分配空间的大小。上面这些宏定义的使用方式如下:

    struct msghdr msg;
    struct cmsghdr *cmptr;

    for(cmptr = CMSG_FIRSTHDR(&msg); cmptr != NULL; cmptr = CMSG_NXTHDR(&msg, cmptr))
    {
        if(cmptr->cmsg_level == ... &&
                cmptr->cmsg_type == ...)
        {
            u_char *ptr;

            ptr = CMSG_DATA(cmptr);
            /* process data pointer to by ptr */
        }
    }

接收标志、目的 IP 地址和接口索引

        下面利用辅助数据实现一个功能类似于 recvfrom 函数的编程,该函数返回三个值:返回 msg_flags 接收标志、接收数据报的目的 IP 地址、接收数据报的接口索引。其中头文件 un.h 只是定义了返回结构 目的 IP 地址和 接口索引。

#ifndef UN_H
#define UN_H
#include <netinet/in.h>

/* 自定义结构体,成员包含目的地址、接口索引 */
struct unp_in_pktinfo{
    struct in_addr  ipi_addr;   /* destination IPv4 address */
    int             ipi_ifindex;/* received interface index */
};
#endif


/* 函数功能:类似于recvfrom函数;
 * 返回值:
 * 1、返回msg_flags值;
 * 2、返回所接收数据报的目的地址(由IP_RECVDSTADDR套接字选项获取);
 * 3、返回所接收数据报接口的索引(由IP_RECVIF套接字选项获取);
 */
#include <sys/socket.h>
#include <unistd.h>
#include <netinet/in.h>
#include <string.h>
#include "un.h"


ssize_t recvfrom_flag(int fd, void *ptr, size_t nbytes, int *flags,
        struct sockaddr * sa, socklen_t *salenptr, struct unp_in_pktinfo *pktp)
{
    struct msghdr msg;/* 需要调用recvmsg函数,所以必须定义该结构 */
    struct iovec iov[1];/* 非连续缓冲区,在这里只定义一个缓冲区 */
    ssize_t n;

#ifdef  HAVE_MSGHDR_MSG_CONTROL
    /* 若支持msg_control成员则初始化以下值辅助数据 */
    struct cmsghdr *cmptr;
    union{
        struct cmsghdr cm;
        char control[CMSG_SPACE(sizeof(struct in_addr)) +
            CMSG_SPACE(sizeof(struct unp_in_pktinfo))];
    } control_un;

    msg.msg_control = control_un.control;
    msg.msg_controllen = sizeof(control_un.control);
    msg.msg_flags = 0;
#else
    /* 若不支持msg_control控制信息,则直接初始化为0 */
    bzero(&msg, sizeof(msg));
#endif
    /* 赋值初始化msghdr结构 */
    msg.msg_name = sa;
    msg.msg_namelen = *salenptr;
    iov[0].iov_base = ptr;
    iov[0].iov_len = nbytes;
    msg.msg_iov = iov;
    msg.msg_iovlen = 1;

    if( (n = recvmsg(fd, &msg, *flags)) < 0)
        return(n);/* 出错返回 */
    /* 若recvmsg调用成功返回,则执行以下程序 */
    *salenptr = msg.msg_namelen;/* 值-结果参数必须返回 */
    if(pktp)
        /* 初始化unp_in_pktinfo结构,置地址为0.0.0.0,置接口索引为0 */
        bzero(pktp, sizeof(struct unp_in_pktinfo));/* 0.0.0.0, i/f = 0 */
#ifndef HAVE_MSGHDR_MSG_CONTROL
    /* 若不支持msg_control控制信息,则把待返回标志置为0,并返回 */
    *flags = 0;/* 值-结果参数返回 */
    return(n);
    /* 以下程序都是处理支持msg_control控制信息的部分 */
#else
    /* 返回标志信息 */
    *flags = msg.msg_flags;/* 值-结果参数返回 */
    if(msg.msg_controllen < sizeof(struct cmsghdr) ||
            (msg.msg_flags & MSG_CTRUNC) || pktp == NULL)
        return(n);
    /* 处理辅助数据 */
    for(cmptr = CMSG_FIRSTHDR(&msg); cmptr != NULL; cmptr = CMSG_NXTHDR(&msg, cmptr))
    {
#ifdef IP_RECVDSTADDR
        /* 处理IP_RECVDSTADDR,返回接收数据报的目的地址 */
        /* 其中IPPROTO_IP表示IPv4域 */
        if(cmptr->cmsg_level == IPPROTO_IP &&
                cmptr->cmsg_type == IP_RECVDSTADDR)
        {
            memcpy(&pktp->ipi_addr, CMSG_DATA(cmptr), sizeof(struct in_addr));
            continue;
        }
#endif
#ifdef IP_RECVIF
        /* 处理IP_RECVIF,返回接收数据报的接口索引 */
        if(cmptr->cmsg_level == IPPROTO_IP &&
                cmptr->cmsg_type == IP_RECVIF)
        {
            struct sockaddr_dl *sdl;/* 数据链路地址结构中包含有接口索引成员 */
            sdl = (struct sockaddr_dl *)CMSG_DATA(cmptr);
            pktp->ipi_ifindex = sdl->sdl_index;
            continue;
        }
#endif
        err_quit("unknown ancillary data");
    }
    return(n);
#endif
}
下面我们利用该函数来实现前面所记录《 基本 UDP 套接字编程》的程序,其中服务器的处理程序变为以下实现:

#include <sys/socket.h>
#include <string.h>
#include <stdio.h>
#include <netinet/in.h>
#include "un.h"
#include <net/if.h>
#include <arpa/inet.h>

/* 限定数据报的大小为 20 字节 */
#undef	MAXLINE
#define	MAXLINE	20		/* to see datagram truncation */
extern ssize_t recvfrom_flag(int , void *, size_t, int *, struct sockaddr*, socklen_t *, struct unp_in_pktinfo *);
extern char * Sock_ntop(const struct sockaddr *, socklen_t );
void
dg_echo(int sockfd, struct sockaddr *pcliaddr, socklen_t clilen)
{
	int						flags;
	const int				on = 1;
	socklen_t				len;
	ssize_t					n;
	char					mesg[MAXLINE], str[INET6_ADDRSTRLEN],
							ifname[IFNAMSIZ];
	struct in_addr			in_zero;
	struct unp_in_pktinfo	pktinfo;
/* 若支持IP_RECVDSATDDR和IP_RECVIF套接字选项,则设置它们 */
#ifdef	IP_RECVDSTADDR
	if (setsockopt(sockfd, IPPROTO_IP, IP_RECVDSTADDR, &on, sizeof(on)) < 0)
		err_ret("setsockopt of IP_RECVDSTADDR");
#endif
#ifdef	IP_RECVIF
	if (setsockopt(sockfd, IPPROTO_IP, IP_RECVIF, &on, sizeof(on)) < 0)
		err_ret("setsockopt of IP_RECVIF");
#endif
	bzero(&in_zero, sizeof(struct in_addr));	/* all 0 IPv4 address */

	for ( ; ; ) {
		len = clilen;
		flags = 0;
        /* 读取来自套接字的数据报 */
		n = recvfrom_flag(sockfd, mesg, MAXLINE, &flags,
						   pcliaddr, &len, &pktinfo);
        /* 把所读取的字节数显示,最大字节数不能超过20字节,若超过,则发生截断情况;
         * 调用sock_ntop把源IP地址和端口号转换为表达格式并输出 */
		printf("%d-byte datagram from %s", n, Sock_ntop(pcliaddr, len));
        /* 若返回的IP地址不为0,则调用inet_ntop转化目的IP地址格式并输出 */
		if (memcmp(&pktinfo.ipi_addr, &in_zero, sizeof(in_zero)) != 0)
			printf(", to %s", inet_ntop(AF_INET, &pktinfo.ipi_addr,
										str, sizeof(str)));
        printf(", index = %d", pktinfo.ipi_ifindex);
        /* 若返回的接口索引不为0,则调用if_indextoname获取接口名字并显示 */
		if (pktinfo.ipi_ifindex > 0)
			printf(", recv i/f = %s",
				   if_indextoname(pktinfo.ipi_ifindex, ifname));
        printf(", flags = %d",flags);
        /* 以下是测试4个标志 */
#ifdef	MSG_TRUNC
		if (flags & MSG_TRUNC)	printf(" (datagram truncated)");
#endif
#ifdef	MSG_CTRUNC
		if (flags & MSG_CTRUNC)	printf(" (control info truncated)");
#endif
#ifdef	MSG_BCAST
		if (flags & MSG_BCAST)	printf(" (broadcast)");
#endif
#ifdef	MSG_MCAST
		if (flags & MSG_MCAST)	printf(" (multicast)");
#endif
		printf("\n");
        /* 回射文本字符串给客户端 */
		sendto(sockfd, mesg, n, 0, pcliaddr, len);
	}
}

使用 UDP 协议

UDP 协议的特性:支持广播和多播;不需要建立连接和拆除;

TCP 协议的特性:确认应答,超时重传,重复分组检测,排序乱序的分组;窗口式流量控制;慢启动和拥塞避免;

所以在以下情况必须使用 UDP 协议:广播或多播;简单的请求-应答应用程序;注意:对于海量数据传输应避免使用 UDP 协议。

由于 UDP 是不可靠传输协议,所以应用程序必须提供:超时重传、序列号确认应答机制;

并发 UDP 服务器

对于 TCP 并发服务器只需 fork 创建一个新子进程即可,然而对于 UDP 必须应对两种不同类型的服务器:

  1. UDP 简单服务器:即读入一个客户请求并发送一个确认应答后,与这个客户就不再相关了。这种情况下,读入客户请求的服务器可以 fork 创建一个新的子进程并让子进程去处理请求。
  2. UDP 复杂服务器:即 UDP 服务器与客户交换多个数据报。问题是客户知道服务器端口号只有服务器的一个众所周知端口。一个客户发送其请求的第一个数据报到达这个端口,但是服务器如何区分这是来自客户的同一个请求的后续数据报还是来自其他客户的请求数据报,该问题解决办法是让服务器为每个客户创建一个新的套接字,在其上bind 绑定一个临时端口,然后使用该套接字发送对客户的所有应答。


参考资料:

《Unix 网络编程》

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: UDP套接字编程是一种基于UDP协议网络编程技术。UDP是一种无连接的协议,它不保证数据传输的可靠性和顺序性,但是具有传输速度快、实时性好等优点。在Python中,可以使用socket模块来实现UDP套接字编程,通过创建UDP套接字、绑定端口、发送和接收数据等操作,实现网络通信。UDP套接字编程常用于实时性要求较高的应用场景,如音视频传输、游戏等。 ### 回答2: UDP是一种无连接的协议,在网络编程中它被广泛应用于实时通讯和游戏等领域。Python语言提供了相应的库支持,使得开发者可以利用UDP套接字来实现高性能的数据传输。 UDP套接字编程主要包括创建UDP套接字、绑定端口、发送数据、接收数据和关闭套接字等步骤。 首先,我们需要在Python中创建UDP套接字,可以使用socket库下的socket()函数。该函数可以接收两个参数,第一个参数指定了IP地址族,通常采用AF_INET表示IPv4地址族,第二个参数指定了套接字类型,可以使用SOCK_DGRAM表示UDP套接字。 ``` import socket udp_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) ``` 接着,我们需要将套接字绑定到指定的端口上。可以使用bind()函数来完成该操作。bind()函数接收一个元组类型的参数,第一个参数指定了要绑定的IP地址,通常设置为本机IP地址。第二个参数指定了端口号。 ``` udp_socket.bind(('127.0.0.1', 8888)) ``` 数据的发送和接收都需要指定对方的IP地址和端口号。使用sendto()函数发送数据时,需要指定一个元组参数,第一个参数为要发送的数据,第二个参数为接收方的IP地址和端口号。 ``` udp_socket.sendto(b'Hello, World!', ('127.0.0.1', 9999)) ``` 使用recvfrom()函数接收数据时,需要指定一个整数参数,表示要接收的数据的最大长度。函数的返回值为一个元组,包含接收到的数据和对方的IP地址和端口号。 ``` data, addr = udp_socket.recvfrom(1024) print(data.decode(), addr) ``` 最后,我们需要在程序结束时关闭UDP套接字,使用close()函数即可。 ``` udp_socket.close() ``` 总的来说,Python网络编程UDP套接字编程是一种高效的数据传输方式,可以用于实时通讯和游戏等领域。开发者可以按照上述流程来利用socket库实现相应的UDP套接字编程。 ### 回答3: UDP是一种无连接的传输协议,它可以实现快速的数据传输,常用于实时性要求高的场合,例如视频直播、语音通话等。Python作为一种高级编程语言,它拥有良好的网络编程库,可以方便地进行UDP套接字编程。 Python的socket模块是实现套接字编程的核心模块。UDP套接字编程可以通过socket模块实现。下面是一个简单UDP套接字编程的例子: ```python import socket server_address = ('localhost', 10000) data = 'Hello, UDP!' sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.sendto(data.encode(), server_address) received_data, server = sock.recvfrom(4096) print(received_data.decode()) sock.close() ``` 在这个例子中,首先定义了一个服务端地址`server_address`,然后定义了要发送的数据`data`。接下来通过socket模块创建了一个UDP套接字对象`sock`,并调用`sendto()`方法将数据发送给服务端。`sendto()`方法的参数包括要发送的数据和服务端地址。当服务端接收到数据后,它会给客户端返回一个响应消息。客户端可以通过`recvfrom()`方法获取服务端返回的数据。最后,调用`close()`方法关闭套接字。 需要注意的是,在UDP套接字编程中,数据的发送和接收是非阻塞的。这意味着一旦调用`sendto()`方法或`recvfrom()`方法,程序就会立即返回,而不会等待服务端的响应。因此,UDP套接字编程需要自行处理超时、数据丢失等异常情况。 总之,Python的UDP套接字编程可以非常方便地实现快速数据传输,适用于需要处理实时性要求高的场合。开发者可以通过socket模块的UDP套接字编程接口轻松地构建自己的应用程序。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值