#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 <strings.h>
#include <string.h>
#include <stdio.h>
using namespace std;
#define MAXLINE 32
#define OPEN_MAX 100
#define LISTENQ 20
#define SERV_PORT 5555
#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()
{
/*
结构体epoll_event 被用于注册所感兴趣的事件和回传所发生待处理的事件，定义如下：
typedef union epoll_data {
void *ptr;
int fd;
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;//保存触发事件的某个文件描述符相关的数据
struct epoll_event {
__uint32_t events; // epoll event
epoll_data_t data; // User data variable
};
其中events表示感兴趣的事件和被触发的事件，可能的取值为：
EPOLLIN ：表示对应的文件描述符可以读；
EPOLLOUT：表示对应的文件描述符可以写；
EPOLLPRI： 表示对应的文件描述符有紧急的数可读；
EPOLLERR：表示对应的文件描述符发生错误；
EPOLLHUP：表示对应的文件描述符被挂断；
EPOLLET： ET的epoll工作模式；
*/
ssize_t n;
char line[MAXLINE];
socklen_t clilen;
struct epoll_event ev,events[20];
int i,maxi,listenfd,connfd,sockfd,epfd,nfds;
/*
1、epoll_create函数
函数声明：int epoll_create(int size)
功能：该函数生成一个epoll专用的文件描述符，其中的参数是指定生成描述符的最大范围；
*/
epfd = epoll_create(256);
struct sockaddr_in clientaddr;
struct sockaddr_in serveraddr;
listenfd = socket(AF_INET,SOCK_STREAM,0);
//设置非阻塞
setnonblocking(listenfd);
/*
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
*/
ev.data.fd = listenfd;
ev.events = EPOLLIN | EPOLLET;
epoll_ctl(epfd,EPOLL_CTL_ADD,listenfd,&ev);
//服务器地址绑定
bzero(&serveraddr,sizeof(serveraddr));
serveraddr.sin_family = AF_INET;
serveraddr.sin_addr.s_addr = INADDR_ANY;
serveraddr.sin_port = htons(1234);
bind(listenfd,(const sockaddr*)&serveraddr,sizeof(serveraddr));
listen(listenfd,LISTENQ);
while(true) {
/*
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事件发生的超时值；
成功：返回发生的事件数；失败：-1
*/
nfds = epoll_wait(epfd,events,20,500);
for(i = 0;i < nfds;i ++) {
if(events[i].data.fd == listenfd) {//如果新监测到一个SOCKET用户连接到了绑定的SOCKET端口，建立新的连接。
connfd = accept(listenfd,(sockaddr*)&clientaddr,&clilen);
if(connfd < 0){
perror("connfd<0");
return -1;
}
//设置非阻塞
setnonblocking(listenfd);
printf("ip client:%s\n",inet_ntoa(clientaddr.sin_addr));
ev.data.fd = connfd;
ev.events = EPOLLIN | EPOLLET;
epoll_ctl(epfd,EPOLL_CTL_ADD,connfd,&ev);
}else if (events[i].events & EPOLLIN) {//如果是已经连接的用户，并且收到数据，那么进行读入。
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
cout<<"readline error"<<endl;
}else if (0 == n){
close(sockfd);
events[i].data.fd = -1;
}
printf("Data ==:%s\n",line);
ev.data.fd = sockfd;//修改sockfd事件为EPOLLOUT
ev.events = EPOLLOUT | EPOLLET;
epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
} else if (events[i].events & EPOLLOUT) {
printf("send data to client\n");
sockfd = events[i].data.fd;
write(sockfd,"hello client",n);
ev.data.fd =sockfd;
ev.events = EPOLLIN | EPOLLET;
epoll_ctl(epfd,EPOLL_CTL_MOD,sockfd,&ev);
}
}
}
return 0;
}
