当前位置：首页 > news >正文

select 同时接收普通数据和带外数据

news 来源：原创 2024/5/8 6:59:06

这是一个服务器程序，使用select同时接收普通数据（监视可读）和带外数据（监视异常）。

#include<sys/socket.h>
#include<sys/types.h>
#include<unistd.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<assert.h>
#include<stdio.h>
#include<errno.h>
#include<string.h>
#include<fcntl.h>
#include<stdlib.h>

int main(int argc, char* argv[])
{
	if (argc <= 2)
	{
		return 1;
	}
	const char* ip = argv[1];
	int port = atoi(argv[2]);

	struct sockaddr_in address;
	bzero(&address, sizeof(address));
	address.sin_family = AF_INET;
	address.sin_port = htons(port);
	inet_pton(AF_INET, ip, &address.sin_addr);

	int listenfd = socket(PF_INET, SOCK_STREAM, 0);
	assert(listenfd >= 0);

	int ret = bind(listenfd, (struct sockaddr*)&address, sizeof(address));
	assert(ret != -1);

	ret = listen(listenfd, 5);
	assert(ret != -1);

	struct sockaddr_in client_address;
	socklen_t client_addrlength = sizeof(client_address);
	int connfd = accept(listenfd, (struct sockaddr*)&client_address, &client_addrlength);
	if (connfd < 0)
	{
		printf("errno is: %d\n", errno);
		close(listenfd);
	}

	char buf[1024];
	fd_set read_fds;	/* 一个用于测试 可读状态的 文件描述符集合 */
	fd_set exception_fds;	/* 一个用于测试 异常状态的 文件描述符集合 */
	FD_ZERO(&read_fds);
	FD_ZERO(&exception_fds);

	while (1)
	{
		memset(buf, '\0', sizeof(buf));
		/* 每次调用 select 前都要重新在 read_fds 和 exception_fds 中设置文件描述符 connfd，因为事件发生之后文件描述符集合将被内核修改 */
		FD_SET(connfd, &read_fds);
		FD_SET(connfd, &exception_fds);
		ret = select(connfd + 1, &read_fds, NULL, &exception_fds, NULL);
		if (ret < 0)
		{
			printf("selection failure\n");
			break;
		}

		/* 对于可读事件，采用普通的 recv 函数读取数据 */
		if (FD_ISSET(connfd, &read_fds))
		{
			ret = recv(connfd, buf, sizeof(buf) - 1, 0);
			if (ret <= 0)
			{
				break;
			}
			printf("get %d bytes of normal data: %s\n", ret, buf);
		}
		/* 对于异常事件，采用带 MSG_OOB 标志的 recv 函数读取带外数据 */
		else if (FD_ISSET(connfd, &exception_fds))
		{
			ret = recv(connfd, buf, sizeof(buf) - 1, MSG_OOB);
			if (ret <= 0)
			{
				break;
			}
			printf("get %d bytes of oob data:  %s\n", ret, buf);
		}
	}
	close(connfd);
	close(listenfd);
	return 0;
}