epoll用到的所有函数都是在头文件sys/epoll.h中声明的,下面简要说明所用到的数据结构和函数: 所用到的数据结构
typedef union epoll_data {
void *ptr;
int fd;
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;
struct epoll_event {
__uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
};
结构体epoll_event 被用于注册所感兴趣的事件和回传所发生待处理的事件,其中epoll_data 联合体用来保存触发事件的某个文件描述符相关的数据,例如一个client连接到服务器,服务器通过调用accept函数可以得到于这个client对应的socket文件描述符,可以把这文件描述符赋给epoll_data的fd字段以便后面的读写操作在这个文件描述符上进行。epoll_event 结构体的events字段是表示感兴趣的事件和被触发的事件可能的取值为:
所用到的函数: 1、epoll_create函数 函数声明:int epoll_create(int size) 该函数生成一个epoll专用的文件描述符,其中的参数是指定生成描述符的最大范围(我觉得这个参数和select函数的第一个参数应该是类似的但是该怎么设置才好,我也不太清楚)。 2、epoll_ctl函数 函数声明:int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event) 该函数用于控制某个文件描述符上的事件,可以注册事件,修改事件,删除事件。 参数:epfd:由 epoll_create 生成的epoll专用的文件描述符; op:要进行的操作例如注册事件,可能的取值EPOLL_CTL_ADD 注册、EPOLL_CTL_MOD 修 改、EPOLL_CTL_DEL 删除 fd:关联的文件描述符; event:指向epoll_event的指针; 如果调用成功返回0,不成功返回-1 3、epoll_wait函数 函数声明:int epoll_wait(int epfd,struct epoll_event * events,int maxevents,int timeout) 该函数用于轮询I/O事件的发生; 参数: epfd:由epoll_create 生成的epoll专用的文件描述符; epoll_event:用于回传代处理事件的数组; maxevents:每次能处理的事件数; timeout:等待I/O事件发生的超时值; 返回发生事件数。
以代码来说明问题: 首先给出server的代码,需要说明的是每次accept的连接,加入可读集的时候采用的都是LT模式,而且接收缓冲区是5字节的,也就是每次只接收5字节的数据:
#include<iostream>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<netinet/in.h>
#include<arpa/inet.h>
#include<fcntl.h>
#include<unistd.h>
#include<stdio.h>
#include<errno.h>
using namespace std;
#define MAXLINE 5
#define OPEN_MAX 100
#define LISTENQ 20
#define SERV_PORT 5000
#define INFTIM 1000
void setnonblocking(int sock)
{
int opts;
opts = fcntl(sock,F_GETFL);
if(opts < 0)
{
perror(" fcntl(sock,GETFL)");
exit (1);
}
opts = opts|O_NONBLOCK;
if(fcntl(sock,F_SETFL,opts) < 0)
{
perror("fcntl(sock,SETFL,opts)");
exit (1);
}
}
int main()
{
int i, maxi, listenfd, connfd, sockfd, epfd, nfds;
ssize_t n;
char line[MAXLINE];
socklen_t clilen;
// 声明epoll_event结构体的变量,ev用于注册事件,数组用于回传要处理的事件
struct epoll_event ev,events[20];
// 生成用于处理accept的epoll专用的文件描述符
epfd = epoll_create(256);
struct sockaddr_in clientaddr;
struct sockaddr_in serveraddr;
listenfd = socket(AF_INET, SOCK_STREAM, 0 );
// 把socket设置为非阻塞方式
setnonblocking(listenfd);
// 设置与要处理的事件相关的文件描述符
ev.data.fd = listenfd;
// 设置要处理的事件类型
ev.events = EPOLLIN;
// 注册epoll事件
epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev);
bzero(&serveraddr, sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
char *local_addr = "127.0.0.1";
inet_aton(local_addr, &(serveraddr.sin_addr)); // htons(SERV_PORT);
serveraddr.sin_port = htons(SERV_PORT);
bind(listenfd,(sockaddr *)&serveraddr, sizeof(serveraddr));
listen(listenfd, LISTENQ);
maxi = 0;
for(;;)
{
// 等待epoll事件的发生
nfds = epoll_wait(epfd, events, 20 , 500);
// 处理所发生的所有事件
for(i = 0 ;i < nfds; ++ i)
{
if(events[i].data.fd == listenfd)
{
clilen=sizeof(struct sockaddr);
connfd = accept(listenfd, (struct sockaddr *)&clientaddr, &clilen);
if (connfd < 0 )
{
perror("connfd<0");
exit ( 1 );
}
// setnonblocking(connfd);
char *str = inet_ntoa(clientaddr.sin_addr);
cout << " accapt a connection from " << str << endl;
// 设置用于读操作的文件描述符
ev.data.fd = connfd;
// 设置用于注测的读操作事件
ev.events = EPOLLIN|EPOLLET;
// ev.events = EPOLLIN;
// 注册ev
epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
}
else if(events[i].events & EPOLLIN)
{
cout << " EPOLLIN " << endl;
if((sockfd = events[i].data.fd) < 0)
continue;
if((n = read(sockfd, line, MAXLINE)) < 0)
{
if(errno == ECONNRESET)
{
close(sockfd);
events[i].data.fd = - 1 ;
}
else
std::cout << " readline error " << std::endl;
}
else if (n == 0 )
{
close(sockfd);
events[i].data.fd = - 1 ;
}
line[n] = ' \0';
cout<<" read "<<line<<endl;
// 设置用于写操作的文件描述符
ev.data.fd = sockfd;
// 设置用于注测的写操作事件
ev.events = EPOLLOUT|EPOLLET;
// 修改sockfd上要处理的事件为EPOLLOUT
// epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd, & ev);
}
else if (events[i].events & EPOLLOUT)
{
sockfd = events[i].data.fd;
write(sockfd, line, n);
// 设置用于读操作的文件描述符
ev.data.fd = sockfd;
// 设置用于注测的读操作事件
ev.events = EPOLLIN|EPOLLET;
// 修改sockfd上要处理的事件为EPOLIN
epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd, & ev);
}
}
}
return 0 ;
}