【网络编程】多路I/O转接服务器之epoll

最近学习网络编程相关，学习资源在b站搜索【linux网络编程】。本文章记录相关学习心得。

相关API

1. 创建一个epoll句柄，参数size用来告诉内核监听的文件描述符的个数，跟内存大小有关。此步骤创建一个建议大小为size的红黑树（二叉平衡树），返回值为根结点的fd

 #include <sys/epoll.h>
 int epoll_create(int size)     size：监听数目

2. 控制某个epoll监控的文件描述符上的事件：注册、修改、删除。此步骤向根结点增加，删除修改节点等

#include <sys/epoll.h>
    int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
        epfd：  为epoll_creat的句柄
        op：    表示动作，用3个宏来表示：
           EPOLL_CTL_ADD (注册新的fd到epfd)，
           EPOLL_CTL_MOD (修改已经注册的fd的监听事件)，
           EPOLL_CTL_DEL (从epfd删除一个fd)；
        event： 告诉内核需要监听的事件
        struct epoll_event {
           __uint32_t events; /* Epoll events */
           epoll_data_t data; /* User data variable */
        };
        typedef union epoll_data {
           void *ptr;
           int fd;
           uint32_t u32;
           uint64_t u64;
        } epoll_data_t;
        EPOLLIN ：  表示对应的文件描述符可以读（包括对端SOCKET正常关闭）
        EPOLLOUT：  表示对应的文件描述符可以写
        EPOLLPRI：  表示对应的文件描述符有紧急的数据可读（这里应该表示有带外数据到来）
        EPOLLERR：  表示对应的文件描述符发生错误
        EPOLLHUP：  表示对应的文件描述符被挂断；
        EPOLLET：   将EPOLL设为边缘触发(Edge Triggered)模式，这是相对于水平触发(Level Triggered)而言的
        EPOLLONESHOT：只监听一次事件，当监听完这次事件之后，如果还需要继续监听这个socket的话，需要再次把这个socket加入到EPOLL队列里

3. 等待所监控文件描述符上有事件的产生，类似于select()调用。

#include <sys/epoll.h>
	    int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout)
	        events：        用来存内核得到事件的集合，
	        maxevents： 告之内核这个events有多大，这个maxevents的值不能大于创建epoll_create()时的size，
	        timeout：   是超时时间
	           -1：阻塞
	           0： 立即返回，非阻塞
	           > 0 ：指定毫秒
	        返回值： 成功返回有多少文件描述符就绪，时间到时返回0，出错返回-1

代码

#include <sys/time.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <ctype.h>
#include <arpa/inet.h> // sockaddr_in
#include <stdio.h>
#include <stdlib.h>
#include <strings.h>
#include <sys/epoll.h>
#include <errno.h>

#define PORT 6666
#define OPEN_MAX 1024

int main() {
	int listen_fd = socket(AF_INET, SOCK_STREAM, 0);
	
	struct sockaddr_in ser_addr;
	bzero(&ser_addr, sizeof(ser_addr));
	ser_addr.sin_family = AF_INET;
	ser_addr.sin_port = htons(PORT);
	ser_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	//端口复用
	int opt = 1;
	setsockopt(listen_fd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
	
	bind(listen_fd, (struct sockaddr *)&ser_addr, sizeof(ser_addr));
	listen(listen_fd, 128);
	
	struct epoll_event temp_event, epfd_event[OPEN_MAX];
	temp_event.events = EPOLLIN;
	temp_event.data.fd = listen_fd;
	
	int epfd = epoll_create(OPEN_MAX); // root
	epoll_ctl(epfd, EPOLL_CTL_ADD, listen_fd, &temp_event);
	
	int nready;
	int connect_fd;
	struct sockaddr_in cli_addr;
	bzero(&cli_addr, sizeof(cli_addr));
	socklen_t cli_len = sizeof(cli_addr);
	int socket_fd;
	int n;
	char buf[BUFSIZ];
	
	while(1) {
		nready = epoll_wait(epfd, epfd_event, OPEN_MAX, -1);

		for (int i = 0; i < nready; i++) {
			if(epfd_event[i].events == EPOLLIN) {
				if (epfd_event[i].data.fd == listen_fd) { //监听新的客户端
					connect_fd = accept(listen_fd, 
								(struct sockaddr*)&cli_addr, 
								&cli_len);
					if (connect_fd < 0) {
						printf("accept error!\n");
						exit(1);
					}
					char ip[BUFSIZ];
					printf("client ip: %s, port : %d\n", inet_ntop(AF_INET, &cli_addr.sin_addr.s_addr, ip, sizeof(ip)), ntohs(cli_addr.sin_port));
					temp_event.events = EPOLLIN;
					temp_event.data.fd = connect_fd;
					epoll_ctl(epfd, EPOLL_CTL_ADD, //向树上新增节点
							  connect_fd, &temp_event);
					
				} else { //数据传送
					socket_fd = epfd_event[i].data.fd;
					n = read(socket_fd, buf, sizeof(buf));
					if (n < 0) {
						printf("read error!\n");
					} else if (n == 0) {
						printf("cli[%d] is close!\n", i);
						epoll_ctl(epfd, EPOLL_CTL_DEL, 
								  socket_fd, NULL);
						close(socket_fd);
					} else {
						for (int j = 0; j < n; j++) {
							buf[j] = toupper(buf[j]);
						}
						write(socket_fd, buf, n);
						write(STDOUT_FILENO, buf, n);
						
					}
					
				}
			}
		}
	}
	close(listen_fd);
	return 0;
}