Linux C epoll网络模型详解

 epoll模型同样委托内核做IO检测。过程与select基本相同，但是实现的方式不同。

#include <netdb.h>
#include <stdio.h>
#include <sys/socket.h>
#include <unistd.h>
#include <string.h>
#include <stdlib.h>
#include <memory.h>
#include <signal.h>
#include <time.h>
#include <errno.h>
#include <sys/epoll.h>



void out_addr(struct sockaddr_in *clientaddr)
{
	int port = ntohs(clientaddr->sin_port);
	char ip[16];
	memset(ip,0,sizeof(ip));
	inet_ntop(AF_INET, &clientaddr->sin_addr.s_addr, ip, sizeof(ip));
	printf("client: %s(%d) connected\n", ip, port);
}

int main(int argc, char *argv[])
{
	if(argc<2){
		printf("usage: %s #port\n", argv[0]);
		exit(1);
	}

	// AF_INET: IPV4
	// SOCK_STREAM: tcp
	int lfd = socket(AF_INET, SOCK_STREAM, 0);
	if(lfd < 0){
		perror("socket error");
		exit(1);
	}
	struct sockaddr_in  serveraddr;

	memset(&serveraddr, 0, sizeof(serveraddr));

	serveraddr.sin_family = AF_INET;
	serveraddr.sin_port = htons(atoi(argv[1]));
	serveraddr.sin_addr.s_addr = INADDR_ANY;
	if(bind(lfd,(struct sockaddr*)&serveraddr,
				sizeof(serveraddr))<0)	{
		perror("bind error");
		exit(1);
	}

	if(listen(lfd,36) <0){
		perror("listen error");
		exit(1);
	}

	printf("Start accept ......\n");

	struct sockaddr_in client_addr;
	socklen_t cli_len = sizeof(client_addr);

	// 创建一个树的根节点
	int epfd = epoll_create(2000);
	// 创建一个树的节点
	struct epoll_event ev;
	ev.events = EPOLLIN;                       // 监听可读事件
	ev.data.fd = lfd;                          // 服务器的文件描述符
	epoll_ctl(epfd,EPOLL_CTL_ADD,lfd,&ev);     // 挂在epoll树上
	struct epoll_event all[2000];              // epoll事件数组
	while(1){
        // 委托内核进行事件监控，最后一个参数是阻塞等待参数
		int ret = epoll_wait(epfd,all,sizeof(all)/sizeof(all[0]),-1);
        // 返回值是发送变化的fd的数量
		for(int i=0;i<ret;i++){
			int fd = all[i].data.fd;       // 取出文件描述符

			if(fd==lfd){                   // 如果fd==服务器fd，那么说明有连接请求需要接受
				int cfd = accept(lfd,
					(struct sockaddr*)&client_addr,
				       &cli_len);

				if(cfd == -1){
					perror("accept error");
					exit(1);
				}
              
                // 将客户端的cfd挂到epoll树上
				struct epoll_event temp;
				temp.events = EPOLLIN;
				temp.data.fd = cfd;
				epoll_ctl(epfd,EPOLL_CTL_ADD,cfd,&temp);


				out_addr(&client_addr);

			}

			else{
				// 如果没有读事件，继续下一个循环。
				if(!all[i].events & EPOLLIN)
					continue;

				char buf[1024] = {0};
				int len = recv(fd,buf,sizeof(buf),0);
				if(len ==-1){
					perror("recv error");
					exit(1);
				}
				else if(len==0){
					printf("client disconnected ...\n");
					// 如果客户端断开连接，需要将客户端fd从epoll树种删除
					int ret = epoll_ctl(epfd,EPOLL_CTL_DEL,fd,NULL);
					if(ret == -1){
						perror("epoll_ctl - del error");
						exit(1);
					}
					close(fd);
				}

				else{
					// 发送消息
					printf(" recv buf:%s\n",buf);
					write(fd,buf,len);
				}
			}
		}
	
	}

	return 0;

}