I/O复用——epoll两种模式的测试代码

LT的服务端测试代码

#define _GUN_SOURCE
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <stdlib.h>

#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#include <sys/epoll.h>

#define MAXEVENTS 100


// 创建并初始化连接套接字
int InitSocket()
{
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd == -1)
	{
		return -1;
	}

	struct sockaddr_in ser;
	memset(&ser, 0, sizeof(ser));
	ser.sin_family = AF_INET;
	ser.sin_port = htons(6000);
	ser.sin_addr.s_addr = inet_addr("127.0.0.1");

	int res = bind(sockfd, (struct sockaddr*)&ser, sizeof(ser));
	if (res == -1)
	{
		return -1;
	}

	res = listen(sockfd, 5);
	if (res == -1)
	{
		return -1;
	}

	return sockfd;
}

// 获取新的客户端链接,并添加到epfd中
void GetNewClient(int fd, int epfd)
{
	struct sockaddr_in cli;
	socklen_t len = sizeof(cli);

	int c = accept(fd, (struct sockaddr*)&cli, &len);
	if (c < 0)
	{
		printf("accpet error\n");
		return;
	}

	printf("one client link success\n");

	// 将新的客户端文件描述符添加到内核事件表中
	struct epoll_event events;
	events.data.fd = c;
	events.events = EPOLLIN | EPOLLRDHUP;

	int res = epoll_ctl(epfd, EPOLL_CTL_ADD, c, &events);
	assert(res != -1);
}

// 处理客户端的数据
void DealClientData(int fd)
{
	char buff[128] = { 0 };
	int n = recv(fd, buff, 1, 0); // 将接收的字符设置为一个以便观察LT的结果
	if (n <= 0)
	{
		printf("%d recv error\n", fd);
		return;
	}

	printf("%d: %s\n", fd, buff);
	send(fd, "OK", 2, 0);
}

// 处理就绪事件
void DealFinshEvents(struct epoll_event *events, int n, int epfd, int sockfd)
{
	int i = 0;
	for (; i < n; ++i)
	{
		int fd = events[i].data.fd;
		if (fd == sockfd)
		{
			// 获取新的客户端链接,并添加到epfd中
			GetNewClient(fd, epfd);
		}
		else
		{
			if (events[i].events  & EPOLLRDHUP) // 客户端关闭了连接或者写操作
			{
				close(fd);
				// 删除内核事件表中的事件描述符
				epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL);
			}
			else
			{
				// 处理客户端的数据
				DealClientData(fd);
			}
		}
	}
}

int main()
{
	// 创建并初始化连接套接字
	int sockfd = InitSocket();
	assert(sockfd != -1);

	// 创建内核事件表
	int epfd = epoll_create(10);
	assert(epfd != -1);

	struct epoll_event event;
	event.events = EPOLLIN;
	event.data.fd = sockfd;

	// 将sockfd添加到内核事件表
	int res = epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &event);
	assert(res != -1);

	int count = 0; // 用于记录epoll_wait的返回次数
	while (1)
	{
		struct epoll_event events[MAXEVENTS];
		int n = epoll_wait(epfd, events, MAXEVENTS, -1);
		if (n <= 0)
		{
			printf("epoll_wait error\n");
			continue;
		}

		printf("epoll_wait return %d\n", count++);

		// 处理就绪事件
		DealFinshEvents(events, n, epfd, sockfd);
	}

	return 0;
}

ET的服务端测试代码

因为ET模式下需要一次性接收完客户端发送的所有数据,所以需要将客户端的监听套接字设置为非阻塞的,详情见GetNewClient()函数

#define _GUN_SOURCE
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <assert.h>
#include <stdlib.h>

#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>
#include <netinet/in.h>

#include <sys/epoll.h>
#include <fcntl.h>
#include <errno.h>

#define MAXEVENTS 100


// 创建并初始化连接套接字
int InitSocket()
{
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	if (sockfd == -1)
	{
		return -1;
	}

	struct sockaddr_in ser;
	memset(&ser, 0, sizeof(ser));
	ser.sin_family = AF_INET;
	ser.sin_port = htons(6000);
	ser.sin_addr.s_addr = inet_addr("127.0.0.1");

	int res = bind(sockfd, (struct sockaddr*)&ser, sizeof(ser));
	if (res == -1)
	{
		return -1;
	}

	res = listen(sockfd, 5);
	if (res == -1)
	{
		return -1;
	}

	return sockfd;
}

// 获取新的客户端链接,并添加到epfd中
void GetNewClient(int fd, int epfd)
{
	struct sockaddr_in cli;
	socklen_t len = sizeof(cli);

	int c = accept(fd, (struct sockaddr*)&cli, &len);
	if (c < 0)
	{
		printf("accpet error\n");
		return;
	}

	printf("one client link success\n");

	// 将新的客户端文件描述符添加到内核事件表中
	struct epoll_event events;
	events.data.fd = c;
	events.events = EPOLLIN | EPOLLRDHUP | EPOLLET; // 以ET方式来处理

	int res = epoll_ctl(epfd, EPOLL_CTL_ADD, c, &events);
	assert(res != -1);

	// 将c文件描述符设置为非阻塞方式
	int oldoption = fcntl(c, F_GETFL); // 获取到标志
	int newoption = oldoption | O_NONBLOCK; // 将c设置为非阻塞
	fcntl(c, F_SETFL, newoption);
}

// 处理客户端的数据
void DealClientData(int fd)
{
	while (1)
	{
		char buff[128] = { 0 };

		// 因为recv操作fd时是阻塞的,所以需要将fd设置为非阻塞的
		// 将fd设置为非阻塞后,recv不会阻塞,如果没有数据,recv返回-1,并将会设置全局的errno
		int n = recv(fd, buff, 1, 0);
		if (n == 0) // 接收数据出错
		{
			printf("%d recv error\n", fd);
			break;
		}
		else if (n == -1)
		{
			if (errno == EAGAIN || errno == EWOULDBLOCK) // 读取完数据
			{
				printf("read finsh\n");
				send(fd, "OK", 2, 0);
				break;
			}
			else // 读取失败
			{
				printf("%d recv error\n", fd);
				break;
			}
		}
		else // 打印接收的数据
		{
			printf("%d: %s\n", fd, buff);
		}
	}
}

// 处理就绪事件
void DealFinshEvents(struct epoll_event *events, int n, int epfd, int sockfd)
{
	int i = 0;
	for (; i < n; ++i)
	{
		int fd = events[i].data.fd;
		if (fd == sockfd)
		{
			// 获取新的客户端链接,并添加到epfd中
			GetNewClient(fd, epfd);
		}
		else
		{
			if (events[i].events  & EPOLLRDHUP) // 客户端关闭了连接或者写操作
			{
				close(fd);
				// 删除内核事件表中的事件描述符
				epoll_ctl(epfd, EPOLL_CTL_DEL, fd, NULL);
			}
			else
			{
				// 处理客户端的数据
				DealClientData(fd);
			}
		}
	}
}

int main()
{
	// 创建并初始化连接套接字
	int sockfd = InitSocket();
	assert(sockfd != -1);

	// 创建内核事件表
	int epfd = epoll_create(5);
	assert(epfd != -1);

	struct epoll_event event;
	event.events = EPOLLIN;
	event.data.fd = sockfd;

	// 将sockfd添加到内核事件表
	int res = epoll_ctl(epfd, EPOLL_CTL_ADD, sockfd, &event);
	assert(res != -1);

	int count = 0; // 用于记录epoll_wait的返回次数
	while (1)
	{
		struct epoll_event events[MAXEVENTS];
		int n = epoll_wait(epfd, events, MAXEVENTS, -1);
		if (n <= 0)
		{
			printf("epoll_wait error\n");
			continue;
		}

		printf("epoll_wait return %d\n", count++);

		// 处理就绪事件
		DealFinshEvents(events, n, epfd, sockfd);
	}

	return 0;
}

客户端的测试代码

客户端代码

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值