epoll的实现

关于epoll的介绍已经在这篇博客中介绍了 下面将展示这个具体实现

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
 
int main(int argc,char *argv[])
{
	int udpfd = 0;
	int ret = 0;
	struct pollfd fds[2];//监测文件描述结构体数组：2个  状态分离  所以需要两个数组
	//用于存放server和client端的地址
        struct sockaddr_in saddr;
	struct sockaddr_in caddr;
        //初始化
	bzero(&saddr,sizeof(saddr));
	saddr.sin_family = AF_INET;
	saddr.sin_port   = htons(8000);
	saddr.sin_addr.s_addr = htonl(INADDR_ANY);
	
	bzero(&caddr,sizeof(caddr));
	caddr.sin_family  = AF_INET;
	caddr.sin_port    = htons(8000);
	
	//创建套接字
	if( (udpfd = socket(AF_INET,SOCK_DGRAM, 0)) < 0)
	{
		perror("socket error");
		exit(-1);
	}
	
	//套接字端口绑字
	if(bind(udpfd, (struct sockaddr*)&saddr, sizeof(saddr)) != 0)
	{
		perror("bind error");
		close(udpfd);//绑定失败的话就直接关闭套接字		
		exit(-1);
	}
 
	printf("input: \"sayto 192.168.220.X\" to sendmsg to somebody\033[32m\n");	
	fds[0].fd = 0;		//标准输入描述符  数组一
	fds[1].fd = udpfd;	//udp描述符     数组二
	/*
        //这里会展示pollfd结构体
	struct pollfd{
	int fd;			//文件描述符
	short events;	//等待的事件
	short revents;	//实际发生的事件
         };
	*/
	fds[0].events = POLLIN; // 普通或优先级带数据可读  
    fds[1].events = POLLIN; // 普通或优先级带数据可读
	
	while(1)
	{	
		// 监视并等待多个文件（标准输入，udp套接字）描述符的属性变化（是否可读）  
        // 没有属性变化，这个函数会阻塞，直到有变化才往下执行，这里没有设置超时  
        /*
成功时，poll() 返回结构体中 revents 域不为 0 的文件描述符个数；如果在超时前没有任何事件发生，poll()返回 0,失败时，poll() 返回 -1，
*/
		write(1,"UdpQQ:",6);//向标准输出输出UdpQQ
        ret = poll(fds, 2, -1); //监听两个数组 由于timeout参数设置为-1,所以说会一直等待到有时间发生
		
		
        if(ret == -1){ // 出错  
            perror("poll()");  
        }
		else if(ret > 0){ // 准备就绪的文件描述符  
            char buf[100] = {0};  //缓冲区
            if( ( fds[0].revents & POLLIN ) ==  POLLIN ){ // 标准输入   通过位与操作 类似于select函数的ISSET
                
				fgets(buf, sizeof(buf), stdin);//从键盘输入
				buf[strlen(buf) - 1] = '\0';
				if(strncmp(buf, "sayto", 5) == 0)//出现sayto 后 就将其后面的ip地址赋值给addr
				{
					char ipbuf[16] = "";
					inet_pton(AF_INET, buf+6, &caddr.sin_addr);//给addr套接字地址再赋值.
					printf("\rsay to %s\n",inet_ntop(AF_INET,&caddr.sin_addr,ipbuf,sizeof(ipbuf)));
					continue;
				}
				else if(strcmp(buf, "exit")==0)//判断输入  exit 出现时候直接退出
				{
					close(udpfd);
					exit(0);
				}
				sendto(udpfd, buf, strlen(buf),0,(struct sockaddr*)&caddr, sizeof(caddr));  
                  
            }
			else if( ( fds[1].revents & POLLIN ) ==  POLLIN ){ //udp套接字  
                struct sockaddr_in addr;
				char ipbuf[INET_ADDRSTRLEN] = "";
				socklen_t addrlen = sizeof(addr);
				
				bzero(&addr,sizeof(addr));
				//显示输出
				recvfrom(udpfd, buf, 100, 0, (struct sockaddr*)&addr, &addrlen);
				printf("\r\033[31m[%s]:\033[32m%s\n",inet_ntop(AF_INET,&addr.sin_addr,ipbuf,sizeof(ipbuf)),buf);  
            }  
              
        }
		else if(0 == ret){ // 超时  
            printf("time out\n");  
        }  
	}
	
	return 0;
}
 

关于怎样测试 参照
添加链接描述
下面加入关于tcp并发服务器（poll实现）

#include <string.h>
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <poll.h>
#include <errno.h>
#define OPEN_MAX 100
 
int main(int argc, char *argv[])
{
	//1.创建tcp监听套接字
	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	
	//2.绑定sockfd
	struct sockaddr_in my_addr;
	bzero(&my_addr, sizeof(my_addr));
	my_addr.sin_family = AF_INET;
	my_addr.sin_port = htons(8000);
	my_addr.sin_addr.s_addr = htonl(INADDR_ANY);
	bind(sockfd, (struct sockaddr *)&my_addr, sizeof(my_addr));
	
	//3.监听listen
	listen(sockfd, 10);
	
	//4.poll相应参数准备
	struct pollfd client[OPEN_MAX];
	int i = 0, maxi = 0;
	for(;i<OPEN_MAX; i++)
		client[i].fd = -1;//初始化poll结构中的文件描述符fd  初始化的时候将其都设置为-1
	
	client[0].fd = sockfd;//需要监测的描述符
	client[0].events = POLLIN;//普通或优先级带数据可读
	
	//5.对已连接的客户端的数据处理
	while(1)
	{
		write(1,"UdpQQ:",6);
		 
		int ret = poll(client, maxi+1, -1);//对加入poll结构体数组所有元素进行监测  timeout 设置位-1  就是阻塞等待
		
		//5.1监测sockfd(监听套接字)是否存在连接
		if((client[0].revents & POLLIN) == POLLIN )//进行位操作 判断是否有事件
		{
			struct sockaddr_in cli_addr;
			int clilen = sizeof(cli_addr);
			int connfd = 0;
			//5.1.1 从tcp完成连接中提取客户端
			connfd = accept(sockfd, (struct sockaddr *)&cli_addr, &clilen);
			char buf[100] = {0};  //缓冲区
			
                
				fgets(buf, sizeof(buf), stdin);//从键盘输入
				buf[strlen(buf) - 1] = '\0';
				if(strncmp(buf, "sayto", 5) == 0)//出现sayto 后 就将其后面的ip地址赋值给addr
				{
					char ipbuf[16] = "";
					inet_pton(AF_INET, buf+6, &cli_addr.sin_addr);//给addr套接字地址再赋值.
					printf("\rsay to %s\n",inet_ntop(AF_INET,&cli_addr.sin_addr,ipbuf,sizeof(ipbuf)));
					continue;
				}
				else if(strcmp(buf, "exit")==0)//判断输入  exit 出现时候直接退出
				{
					close(connfd);
					exit(0);
				}
		
				sendto(connfd, buf, strlen(buf),0,(struct sockaddr*)&caddr, sizeof(caddr)); 
			//5.1.2 将提取到的connfd放入poll结构体数组中，以便于poll函数监测
			for(i=1; i<OPEN_MAX; i++)//因为client[0]已经放了sockfd 所以从位置1开始
			{
				if(client[i].fd < 0)//如果这个位置是初始化的-1  那么说明这个位置空缺
				{
					client[i].fd = connfd;
					client[i].events = POLLIN;
					break;
				}
			}
			
			//5.1.3 maxi更新
			if(i > maxi)//如果i超出了最大值 就将i赋值给最大值
				maxi = i;
				
			//5.1.4 如果没有就绪的描述符，就继续poll监测，否则继续向下看
			if(--ret <= 0)//如果poll没有监听到事件 那么返回继续监听
				continue;
		}
		
		//5.2继续响应就绪的描述符
		for(i=1; i<=maxi; i++)
		{
			if(client[i].fd < 0)
				continue;
			
			if(client[i].revents & (POLLIN | POLLERR))//通过位操作  判断是否有请求
			{
				int len = 0;
				char buf[128] = "";
				
				//5.2.1接受客户端数据
				if((len = recv(client[i].fd, buf, sizeof(buf), 0)) < 0)
				{
					if(errno == ECONNRESET)//tcp连接超时、RST  判断是否报错
					{
						close(client[i].fd);//关闭这个文件描述符
						client[i].fd = -1;//重新将这个位置置为-1
					}
					else
						perror("read error:");//输出读取错误
					
				}
				else if(len == 0)//客户端关闭连接
				{
					close(client[i].fd);
					client[i].fd = -1;
				}
				else//正常接收到服务器的数据
					printf("\r\033[31m[%s]:\033[32m%s\n",inet_ntop(AF_INET,&cli_addr.sin_addr,ipbuf,sizeof(ipbuf)),buf);  
				
				//5.2.2所有的就绪描述符处理完了，就退出当前的for循环，继续poll监测
				if(--ret <= 0)
					break;
				
			}
		}
	}
	return 0;
}