在这之前我们需要先看看自己的虚拟机能够连接最大客户端数(可以说是设置最大打开文件描述符限制),并且进行修改。按一下步骤进行输入(不同版本的虚拟机查出来的数目可能会有些差异,但是这些指令都是可以用的)
树是Linux中的一种数据结构,在windows中是不存在的,它类似数组和链表的结合体(具备数组好访问的特点和链表好修改的特点)。在所有树的类型中,最强的就是红黑树。(这里就不想用太多时间写其他类型了,知道最常用的就行了……)
在epoll模型中我们主要要使用到的函数就三个不是很多。下面的三个函数的顺序也是我们后面写代码时需要的流程。
头文件
#include
1、创建一个epoll句柄
int epoll_create(int size)
2、控制某个epoll监控的文件描述符上的事件:注册、修改、删除。
int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)
3、等待所监控文件描述符上有事件的产生,类似于select()调用。
int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout)
struct epoll_event
{
__uint32_t events; /* Epoll 事件*/
epoll_data_t data; /* 用户数据变量 */
};
在epoll_event 结构体的第一个参数
在epoll_event 结构体的第二个参数(注意:这是个联合体)
//下面的联合体中,一般只用到fd
typedef union epoll_data
{
void *ptr;
int fd; //文件描述符
__uint32_t u32;
__uint64_t u64;
} epoll_data_t;
epoll对文件描述符的操作有两种模式: LT (level trigger)和ET ( edge trigger)。LT模式是默认模式,
两种区别如下:
LT模式:当epoll_wait检 测到描述符事件发生并 将此事件通知应用程序,应用程序可以不立即处理该事件。.下次调用epoll_wait时, 会再次响应应用程序并通知此事件。
ET模式:当epoll wait检 测到描述符事件发生并
将此事件通知应用程序,应用程序必须立即处理该事件。如果不处理,下次调用epoll_wait时,不会再次响应应用程序并通知此事件。
ET模式在很大程度上减少了epoll事件被重复触发
的次数,因此效率要比LT模式高。
#include
#include
#include
#include
#include
#include
#include
#define MAXLINE 80
#define SERV_PORT 8000 //端口号
#define OPEN_MAX 1024 //监听个数
//出错时使用的代码
void perr_exit(char *str)
{
perror(str);
exit(1);
}
int main(void)
{
//maxi是client的游标,listenfd用于接收服务器的套接字,connfd接收客户端套接字,sockfd用于临时套接字
int i, j, maxi, listenfd, connfd, sockfd;
int nready, efd, res;
ssize_t n;
//用于信息发送接收的缓冲区
char buf[MAXLINE], str[INET_ADDRSTRLEN];
socklen_t clilen;
int client[OPEN_MAX];
struct sockaddr_in cliaddr, servaddr;
struct epoll_event tep, ep[OPEN_MAX];
/*socket基础部分*/
listenfd = socket(AF_INET, SOCK_STREAM, 0);
bzero(&servaddr, sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
servaddr.sin_port = htons(SERV_PORT);
bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr));
listen(listenfd, 20);
//把client[]数组初始化为-1
for (i = 0; i < OPEN_MAX; i++)
{
client[i] = -1;
}
//
maxi = -1;
//epoll句柄创建、初始化,监听1024个
efd = epoll_create(OPEN_MAX);
if (efd == -1) //简历不成功报错退出
perr_exit("epoll_create");
//把新建的listenfd挂到红黑树
tep.events = EPOLLIN; //表示对应的文件描述符可以读(包括对端SOCKET正常关闭)
tep.data.fd = listenfd;
res = epoll_ctl(efd, EPOLL_CTL_ADD, listenfd, &tep);
if (res == -1)
perr_exit("epoll_ctl");
//
for ( ; ; )
{
//epoll_wait把符合条件的果实放到ep数组中
nready = epoll_wait(efd, ep, OPEN_MAX, -1);//-1表示不会超时
if (nready == -1)
perr_exit("epoll_wait");
for (i = 0; i < nready; i++)
{
if (!(ep[i].events & EPOLLIN)) //判断tep.events是不是EPOLLIN;如果不是就跳过
continue;
//如果listenfd被读取,就代表有新的客户端要接入
if (ep[i].data.fd == listenfd)
{
clilen = sizeof(cliaddr);
connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen);
printf("received from %s at PORT %d\n",inet_ntop(AF_INET, &cliaddr.sin_addr, str, sizeof(str)), ntohs(cliaddr.sin_port));
for (j = 0; j < OPEN_MAX; j++)
{
if (client[j] < 0)
{
client[j] = connfd; /* save descriptor */
break;
}
}
if (j == OPEN_MAX)
perr_exit("too many clients");
if (j > maxi)
maxi = j; /* 把client[]下标往前面移动 */
//把新来的客户端挂到树上
tep.events = EPOLLIN;
tep.data.fd = connfd;
res = epoll_ctl(efd, EPOLL_CTL_ADD, connfd, &tep);
if (res == -1)
perr_exit("epoll_ctl");
}
else
{
//客户端被读,sockfd 接收临时客户端
sockfd = ep[i].data.fd;
n = read(sockfd, buf, MAXLINE);
//没有读到数据我就把这果实摘掉
if (n == 0)
{
for (j = 0; j <= maxi; j++)
{
if (client[j] == sockfd)
{
client[j] = -1;
break;
}
}
//摘掉读不到数据的客户端
res = epoll_ctl(efd, EPOLL_CTL_DEL, sockfd, NULL);
if (res == -1)
perr_exit("epoll_ctl");
//关闭该客户端
close(sockfd);
printf("client[%d] closed connection\n", j);
}
else
{
//小写转大写返回
for (j = 0; j < n; j++)
buf[j] = toupper(buf[j]);
write(sockfd, buf, n);
}
}
}//for (i = 0; i < nready; i++)结束
}//for(;;)结束
close(listenfd);
close(efd);
return 0;
}