UDP 套接字编程

UDP 套接字编程

UDP 套接字编程相较于TCP 套接字编程会简单一些, UDP 仅提供无连接的不可靠数据报协议，而TCP是面向连接的字节流协议.

UDP 函数介绍

这里, 服务器和客户端都不需要调用 TCP 的 connect 方法, 客户端可以直接发送数据, 而服务端仅在为套接字命名(绑定端口)之后, 就可以接收消息. 函数介绍如下:

1. socket 创建一个套接字并返回该套接字的文件描述符。

   int socket(int domain, int type, int protocol);

2. 套接字命名
   为套接字关联一个 IP 地址和端口号

   int bind(int socket, const struct sockaddr *address, size_t address_len);

3. 等待客户数据到达(阻塞)
   服务器不接收来自客户的连接，只需使用 recvfrom 系统调用，等待客户的数据到达。
   int recvfrom(int socket, void *buffer, size_t length, int flags, struct sockaddr *src_addr, socklen_t *src_len);

4. 客户端发送数据
   不需要建立连接, 客户只需调用 sendto 系统调用向服务器发送消息：

int sendto(int socket, const void *buffer, size_t length, int flags, const struct sockaddr *dest_addr, socklen_t dest_len);

5. 关闭套接字

   操作系统为每个套接字分配了一个文件描述符，使用 close 系统调用通知操作系统回收文件描述符，可以：

   close(fd);

UDP 程序示例

1. 服务端

#include	"unp.h"

int
main(int argc, char **argv)
{
	int					sockfd;
	struct sockaddr_in	servaddr, cliaddr;

	sockfd = Socket(AF_INET, SOCK_DGRAM, 0);

	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family      = AF_INET;
	servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
	servaddr.sin_port        = htons(SERV_PORT);

	Bind(sockfd, (SA *) &servaddr, sizeof(servaddr));

	dg_echo(sockfd, (SA *) &cliaddr, sizeof(cliaddr));
}

void
dg_echo(int sockfd, SA *pcliaddr, socklen_t clilen)
{
	int			n;
	socklen_t	len;
	char		mesg[MAXLINE];

	for ( ; ; ) {
		len = clilen;
		n = Recvfrom(sockfd, mesg, MAXLINE, 0, pcliaddr, &len);

		Sendto(sockfd, mesg, n, 0, pcliaddr, len);
	}
}

2. 客户端

#include	"unp.h"

int
main(int argc, char **argv)
{
	int					sockfd;
	struct sockaddr_in	servaddr;

	if (argc != 2)
		err_quit("usage: udpcli <IPaddress>");

	bzero(&servaddr, sizeof(servaddr));
	servaddr.sin_family = AF_INET;
	servaddr.sin_port = htons(SERV_PORT);
	Inet_pton(AF_INET, argv[1], &servaddr.sin_addr);

	sockfd = Socket(AF_INET, SOCK_DGRAM, 0);

	dg_cli(stdin, sockfd, (SA *) &servaddr, sizeof(servaddr));

	exit(0);
}

void
dg_cli(FILE *fp, int sockfd, const SA *pservaddr, socklen_t servlen)
{
	int	n;
	char	sendline[MAXLINE], recvline[MAXLINE + 1];

	while (Fgets(sendline, MAXLINE, fp) != NULL) {

		Sendto(sockfd, sendline, strlen(sendline), 0, pservaddr, servlen);

		n = Recvfrom(sockfd, recvline, MAXLINE, 0, NULL, NULL);

		recvline[n] = 0;
		Fputs(recvline, stdout);
	}
}