从socket开始讲解网络模式
windows采用IOCP网络模型,而linux采用epoll网络模型(Linux得以实现高并发,并被作为服务器首选的重要原因),接下来讲下epoll模型对网络编程高并发的作用
简单的socket连接
socket连接交互的流程如图:
服务端中个api的作用:
- socket(IPv4/IPv6,TCP/UDP,0):创建socket套接字,获取
listenfd
- bind(listenfd,服务器地址,服务器地址长度):将套接字绑定服务器地址
- listen(listenfd,size): 该套接字最多连接size个连接
- accept(listenfd,客户端信息,len):客户端使用connect()后,与服务端进行三次握手,三次握手成功后,生成一个连接文件描述符
connfd
- recv(connfd,buff,len,0):从该连接的的buff中读取len字节数据 (对应图中的read())
- send(connfd,buff,n,0): 读完数据后向buff写数据,以回应客户端。(对应图中的write())
原始代码实现:
int main(int argc, char **argv)
{
int listenfd, connfd, n;
struct sockaddr_in servaddr;
char buff[MAXLNE];
if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { // 监听tcp连接
printf("create socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
memset(&servaddr, 0, sizeof(servaddr)); // 服务地址
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 将uint32值转换为网络字节顺序,0.0.0.0系统将调用默认ip地址,为啥不是127.0.0.1呢
servaddr.sin_port = htons(9999); // 将整型变量转换成网络字节顺序,通过端口socket绑定到某一进程
// 当用socket()函数创建套接字以后,套接字在名称空间(网络地址族)中存在,但没有任何地址给它赋值
// bind()把用addr指定的地址赋值给sockfd。addrlen指定了以addr所指向的地址结构体的字节长度。一般来说,该操作称为“给套接字命名”。
if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1) {
printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
// socket创建套接字=》bind(),给套接字地址=》listen监听连接=》accept
// 该监听fd最多连接10个连接
if (listen(listenfd, 10) == -1) {
printf("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("========waiting for client's request========\n");
while (1) {
n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
}
}
}
现在加入有两个客户端同时连接服务器,第二个客户端能连接成功,但发送不了数据,只有第一个客户端能发送数据。这是因为第二个客户端发送连接请求后,被服务器监听并成功连接,但是accept只取了第一个客户端的connfd,然后服务器就一直在while(1){}里跑了,只有第一个连接能发送数据。
解决方法,将accept()放入while(1)循环里
int main(int argc, char **argv)
{
int listenfd, connfd, n;
struct sockaddr_in servaddr;
char buff[MAXLNE];
if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { // 监听tcp连接
printf("create socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
memset(&servaddr, 0, sizeof(servaddr)); // 服务地址
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 将uint32值转换为网络字节顺序,0.0.0.0系统将调用默认ip地址,为啥不是127.0.0.1呢
servaddr.sin_port = htons(9999); // 将整型变量转换成网络字节顺序,通过端口socket绑定到某一进程
// 当用socket()函数创建套接字以后,套接字在名称空间(网络地址族)中存在,但没有任何地址给它赋值
// bind()把用addr指定的地址赋值给sockfd。addrlen指定了以addr所指向的地址结构体的字节长度。一般来说,该操作称为“给套接字命名”。
if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1) {
printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
// socket创建套接字=》bind(),给套接字地址=》listen监听连接=》accept
// 该监听fd最多连接10个连接
if (listen(listenfd, 10) == -1) {
printf("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("========waiting for client's request========\n");
while (1) {
// 把accept放到while循环里
// 新问题:每个连接都能发送数据,但只能发送一条数据
// 原因是while中又两个阻塞点:accept和recv,一个连接发送数据后,服务器将阻塞在accept
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
}
//close(connfd);
}
}
产生了新问题:服务端无法正常接收数据,原因是while中又两个阻塞点:accept和recv,一个连接发送数据后,服务器将阻塞在accept
如:客户端A连接服务器=》客户端B连接服务器=>客户端B发送5次数据“B”(此时服务将无法读取这5个B,因为客户端A连接服务器后,还阻塞在recv(),等待读取A的数据),如果此时 =》 客户端A发送数据"A" => 服务器将会读取A,然后通过accept()获取客户端B的connfd,再读取客户端B之前发送的五个“B”
为每个socket连接设置一个线程
多线程:来一个连接新建一个线程,把connfd传给入口函数,接收发送数据
// 为方便讲解,listen()以上的代码略,最后会有一个整体的代码
int main(){
...
// 客户端不多,可以用这种方法,客户端太多就不行
// 如一个线程分配8M的栈空间,1G的内存只能分配128个线程左右 ,性能突破不了C10K
while (1) {
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
pthread_t threadid;
pthread_create(&threadid, NULL, client_routine, (void*)&connfd);
}
}
void *client_routine(void *arg) { // 线程入口函数,参数为connfd
int connfd = *(int *)arg;
char buff[MAXLNE];
while (1) {
int n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
break;
}
}
return NULL;
}
这种方法简单,方便管理,不用担心一个连接阻塞其他连接了,但是有一个问题:线程需要独立的运行栈和其他的开销,一个线程大约是8M的栈空间,4G的内存最多只能支持512个连接,无法达到C10K级的并发。
select网络模型
多线程的方式并发量上不去,能不能用少数的线程处理所有fd呢?
select: 当一个fd接收到数据时,服务器能知道是哪个连接发的数据,并进行处理,其处理流程为(select用的很少,可直接看下一节的poll网络模型):
- 创建事件集合,事件分为三类:可读、可写、异常,集合为长度为1024的bit-set,当有事件发生时,将对应的bit位设置为1
- 进入while(1){}循环
- 调用select(),将所有fd拷贝到内核,select会轮询所有fd是否有事件触发,触发了就置为1,并分别把(可读、可写、异常)事件从内核返回到用户态,这里可只考虑可读事件
- 对fd进行进行处理:socket中的fd分为两类listenfd和connfd,需分别处理
- listenfd对应的bit-set[listenfd]为1,且为可读事件时,说明服务器监听到了新的连接,需要将该connfd加入到事件集合中
- listenfd是一个bit-map的做法,01固定,为保准输入输出,从3开始递增分配(3,4,5,6),如果4回收了,再从4分配,所以listenfd比所有connfd都小
- 遍历bit-set,判断事件类型做出处理,如bit-set[connfd]=1,表示该fd有可读事件,就recv()读取数据,并send()响应客户端
代码如下,注意下select()函数参数的意义:
int main(){
// 对fd的处理包括两部分,listenfd和读写fd
// 因为首先要将rset拷贝到内核,再全部拷贝出来,开销太大
fd_set rfds, rset;
FD_ZERO(&rfds); // 先把bit-set清空
FD_SET(listenfd, &rfds); // 将listenfd加入 rfds读事件集合
int max_fd = listenfd; // 当前管理的所有文件描述符的最大值,也就bit-set的长度
while (1) {
rset = rfds; // 为啥还要弄这两个变量:防止读集合rfds在select被修改了
// 第二、三、四个参数:要管理哪些文件描述符的读、写、异常的事件,放到相应的集合
// 第五个超时时间:如果隔这么久一直没有事件发生,就返回,为NULL就是没有事件一直阻塞(select自带阻塞)
// 调用select需要把fd从用户态拷贝到内核态,而且需要在内核遍历传递进来的所有fd
// 把rfds给内核,rset是返回给用户的发生事件的文件描述符
int nready = select(max_fd+1, &rset, &wset, NULL, NULL); // 返回事件的数量,这里其实只有读事件
if (FD_ISSET(listenfd, &rset)) { // listenfd是否在读事件集合中
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) { // 将connfd加入到读事件集合
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
FD_SET(connfd, &rfds); // 将connfd加入读事件集合
if (connfd > max_fd) max_fd = connfd;
if (--nready == 0) continue; // 全部加完了,去对事件进行操作
}
int i = 0;
for (i = listenfd+1;i <= max_fd;i ++) { // 遍历所有fd
if (FD_ISSET(i, &rset)) { //
n = recv(i, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(i, buff, n, 0);
} else if (n == 0) { //
FD_CLR(i, &rfds); // 从读事件集合删除
close(i);
}
if (--nready == 0) break;
}
}
}
}
一个select能管理1024个fd,那么多弄几个select(一个进程或线程一个select),就能到达C10K了,但很难达到C1000K,其有以下缺点:
- 调用select时,事件集合需要从内核态拷贝到内核态,返回时,又需要全部从内核态拷贝到用户态。
- 需要轮询所有fd
- 单个select支持的fd太少了,默认为1024
poll网络模型
和select模型很像,区别就是用pollfd结构(fd的数量可自定义)代替了select的bit_set结构,
pollfd结构为:
struct pollfd
{
int fd; /* File descriptor to poll. */
short int events; /* Types of events poller cares about. */
short int revents; /* Types of events that actually occurred. */
};
其处理流程为:
- 定义pollfd列表,其中第一个元素为listenfd
- 初始化每个列表元素,fd为-1,event为(POLLIN、POLLOUT、POLLPRI等),select将事件分为三类,poll统一管理
- 进入while(1){}循环
- 接下里的对fd的处理流程和select一样了
- 调用
poll()
, 把fd都拷贝到内核,轮询后拷贝回用户态 - 如果listenfd有可读事件发生,将connfd加入到poll
- 遍历所有fd,如果fd有可读事件发生,recv()读取数据并send()响应客户端,数据读取完成后关闭connfd
- 调用
代码实现:
int main(){
...
struct pollfd fds[POLL_SIZE] = {0}; // fd的数量可自定义
fds[0].fd = listenfd; // 先将listenfd加入poll
fds[0].events = POLLIN; // select将事件分为三类,poll将这三类事件统一管理
int max_fd = listenfd;
int i = 0;
for (i = 1;i < POLL_SIZE;i ++) {
fds[i].fd = -1; // 将poll中的fd置为-1
}
while (1) {
int nready = poll(fds, max_fd+1, -1); // 把fd拷贝到内核,再拷贝出来
if (fds[0].revents & POLLIN) { // 如果listenfd在poll中,且有可读事件发生(也就是来连接了),revents实际发生的事件,pollout为可写事件
struct sockaddr_in client;
socklen_t len = sizeof(client); // 取connfd
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("accept \n");
fds[connfd].fd = connfd; // 将connfd加入poll
fds[connfd].events = POLLIN;
if (connfd > max_fd) max_fd = connfd;
if (--nready == 0) continue;
}
//int i = 0;
for (i = listenfd+1;i <= max_fd;i ++) {
if (fds[i].revents & POLLIN) { // fd i 发生了且为POLLIN类型
n = recv(i, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(i, buff, n, 0);
} else if (n == 0) { // 无数据可读后,关闭该connfd
fds[i].fd = -1;
close(i);
}
if (--nready == 0) break;
}
}
}
}
poll解决的问题:没有最大文件描述符数量的限制
。
但fd在内核态与用户态的来回拷贝,以及需要轮询所有fd,使得其监听事件的开销过大,无法支持太大的并发量,且poll不像select可以跨平台,其只能在Linux平台使用
epoll网络模型
select和poll都是只需调用一个函数,epoll需要调用三个:epoll_create、epoll_ctl(ADD, DEL, MOD)、epoll_wait
epoll结构为:
struct epoll_event
{
uint32_t events; /* Epoll events */
epoll_data_t data; /* User data variable */
} __EPOLL_PACKED;
其处理流程为:
- epoll_create(): 创建epfd,创建红黑树以及就绪列表(链表)
- 将listendfd绑定可读事件,加入epoll (listenfd和connfd都只需要从用户态拷贝到内核态一次了)
- 进入while(1){}循环
- epoll_wait():将就绪列表从内核态拷贝到用户态(从内核态拷贝到用户态只要拷贝就绪的事件了)
- 遍历就绪列表(不需要遍历全部fd了)
- 如果是listenfd有可读事件,将connfd加入到epoll中
- 如果是connfd有可读事件,读取数据,并send(),读完了从epoll中移除,并关闭该fd
代码实现:
int main(){
int epfd = epoll_create(1); //int size(为了兼容,以前就绪队列是固定的,后面改成链表了) 创建epfd
struct epoll_event events[POLL_SIZE] = {0}; // 这里POLL_SIZE就是每次取事件的最大数量,小一点没关系(如50),因为即使百万并发,活跃的也就1w,多跑几次了就是了
struct epoll_event ev;
ev.events = EPOLLIN;
ev.data.fd = listenfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev); // 将listenfd加入epoll,拷贝到内核:只需要拷贝一次,不需要拷贝出来
while (1) {
int nready = epoll_wait(epfd, events, POLL_SIZE, 5); // 取事件, 拷贝到用户态:只拷贝就绪的事件了
if (nready == -1) {
continue;
}
int i = 0;
// 遍历就绪队列
for (i = 0;i < nready;i ++) {
int clientfd = events[i].data.fd;
if (clientfd == listenfd) { // 处理listenfd
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("accept\n");
ev.events = EPOLLIN;
ev.data.fd = connfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
} else if (events[i].events & EPOLLIN) { // 处理connfd
n = recv(clientfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(clientfd, buff, n, 0);
} else if (n == 0) { // 读完了就从epoll中移除connfd
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
close(clientfd);
}
}
}
}
close(listenfd);
return 0;
}
epoll解决了select面临的三大问题,可实现C1000K的并发量
完整代码
#include <errno.h>
#include <netinet/in.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/select.h>
#include <sys/poll.h>
#include <sys/epoll.h>
#include <pthread.h>
#define MAXLNE 4096
#define POLL_SIZE 1024
//8m * 4G = 128 , 512
//C10k
void *client_routine(void *arg) { //
int connfd = *(int *)arg;
char buff[MAXLNE];
while (1) {
int n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
break;
}
}
return NULL;
}
int main(int argc, char **argv)
{
int listenfd, connfd, n;
struct sockaddr_in servaddr;
char buff[MAXLNE];
if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) { // 监听tcp连接
printf("create socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
memset(&servaddr, 0, sizeof(servaddr)); // 服务地址
servaddr.sin_family = AF_INET;
servaddr.sin_addr.s_addr = htonl(INADDR_ANY); // 将uint32值转换为网络字节顺序,0.0.0.0系统将调用默认ip地址,为啥不是127.0.0.1呢
servaddr.sin_port = htons(9999); // 将整型变量转换成网络字节顺序,通过端口socket绑定到某一进程
// 当用socket()函数创建套接字以后,套接字在名称空间(网络地址族)中存在,但没有任何地址给它赋值
// bind()把用addr指定的地址赋值给sockfd。addrlen指定了以addr所指向的地址结构体的字节长度。一般来说,该操作称为“给套接字命名”。
if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) == -1) {
printf("bind socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
// socket创建套接字=》bind(),给套接字地址=》listen监听连接=》accept
// 该监听fd最多连接10个连接
if (listen(listenfd, 10) == -1) {
printf("listen socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
#if 0
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("========waiting for client's request========\n");
while (1) {
n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
}
//close(connfd);
}
// tip: 至此,都两个客户端同时连接accept时,第二个客户端只能连接成功,但发送不了数据
// 这是因为accept只取了一个连接,只有第一个连接能发送数据
#elif 0
printf("========waiting for client's request========\n");
while (1) {
// 把accept放到while循环里
// 新问题:每个连接都能发送数据,但只能发送一条数据
// 原因是while中又两个阻塞点:accept和recv,一个连接发送数据后,服务器将阻塞在accept
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
n = recv(connfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(connfd, buff, n, 0);
} else if (n == 0) {
close(connfd);
}
//close(connfd);
}
#elif 0
// 多线程:来一个连接新建一个线程,把connfd传给入口函数,接收发送数据
// 客户端不多,可以用这种方法,客户端太多就不行
// 如一个线程分配8M的栈空间,1G的内存只能分配128个线程左右 ,性能突破不了C10K
while (1) {
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
pthread_t threadid;
pthread_create(&threadid, NULL, client_routine, (void*)&connfd);
}
#elif 0
// 所以能不能用少数的线程处理所有fd呢
// 当一个fd接收到数据时,服务器能知道是哪个连接发的数据,并进行处理
// // fd_set就是个bit-set,第n个bit有事件到达,就将第n个bit位置1
// 由于是bit-set,能监听的fd是固定的,要改还得去内核改,默认是1024
// 对fd的处理包括两部分,listenfd和读写fd
// 一个select能管理1024个fd,那么多弄几个select(一个进程或线程一个select),就能到达C10K了,但很难达到C1000K
// 因为首先要将rset拷贝到内核,再全部拷贝出来,开销太大
fd_set rfds, rset, wfds, wset;
FD_ZERO(&rfds); // 先把bit-set清空
// 设置listenfd-set (也就是我们要监控哪些集合,这个集合copy到内核);还有个集合是哪些fd置1了(这个集合从内核copy出来)
FD_SET(listenfd, &rfds); // 将listenfd加入 rfds读事件集合
FD_ZERO(&wfds); // 写事件集合
int max_fd = listenfd; // 当前管理的所有文件描述符的最大值,也就bit-set的长度
while (1) {
rset = rfds; // 为啥还要弄这两个变量:防止读集合rfds在select被修改了
wset = wfds;
// 第二、三个参数:要管理哪些文件描述符的读(写)的事件,放到相应的集合
// 第四个是异常事件,第五个超时时间:如果隔这么久一直没有事件发生,就返回,为空就是没有事件一直阻塞(select自带阻塞)
// 调用select需要把fd从用户态拷贝到内核态,而且需要在内核遍历传递进来的所有fd
// 把rfds和wfds给内核,rset和wset是返回给用户的发生事件的文件描述符
int nready = select(max_fd+1, &rset, &wset, NULL, NULL); // 返回事件的数量,这里其实只有读事件
if (FD_ISSET(listenfd, &rset)) { // listenfd是否在读事件集合中
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) { // 将connfd加入到读事件集合
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
FD_SET(connfd, &rfds); // 将connfd加入读事件集合
if (connfd > max_fd) max_fd = connfd;
if (--nready == 0) continue; // 全部加完了,去对读写事件进行操作
}
int i = 0;
// listenfd是一个bit-map的做法,01固定,从3开始递增分配(3,4,5,6),如果4回收了,再从4分配
for (i = listenfd+1;i <= max_fd;i ++) { // 遍历所有fd,依次进行读写操作,,这里不应该是可读可写事件集合吗
if (FD_ISSET(i, &rset)) { //
n = recv(i, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
FD_SET(i, &wfds); // 数据读完要加入写事件集合??一次没读完怎么办?
//reactor
//send(i, buff, n, 0);
} else if (n == 0) { //
FD_CLR(i, &rfds); // 从读事件集合删除
//printf("disconnect\n");
close(i);
}
if (--nready == 0) break;
} else if (FD_ISSET(i, &wset)) {
send(i, buff, n, 0);
FD_SET(i, &rfds); // 发送完了这个fd为啥要加入读事件集合,为啥不从写事件集合删除??
}
}
}
#elif 0
struct pollfd fds[POLL_SIZE] = {0}; // fd的数量可自定义
fds[0].fd = listenfd; // 先将listenfd加入poll
fds[0].events = POLLIN; // select将事件分为三类,poll将这三类事件统一管理
int max_fd = listenfd;
int i = 0;
for (i = 1;i < POLL_SIZE;i ++) {
fds[i].fd = -1; // 将poll中的fd置为-1
}
while (1) {
int nready = poll(fds, max_fd+1, -1); // 把fd拷贝到内核,再拷贝出来
if (fds[0].revents & POLLIN) { // 如果listenfd在poll中,且有可读事件发生(也就是来连接了),revents实际发生的事件,pollout为可写事件
struct sockaddr_in client;
socklen_t len = sizeof(client); // 取connfd
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("accept \n");
fds[connfd].fd = connfd; // 将connfd加入poll
fds[connfd].events = POLLIN;
if (connfd > max_fd) max_fd = connfd;
if (--nready == 0) continue;
}
//int i = 0;
for (i = listenfd+1;i <= max_fd;i ++) {
if (fds[i].revents & POLLIN) { // fd i 发生了且为POLLIN类型
n = recv(i, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(i, buff, n, 0);
} else if (n == 0) { // 无数据可读后,关闭该connfd
fds[i].fd = -1;
close(i);
}
if (--nready == 0) break;
}
}
}
#else
int epfd = epoll_create(1); //int size(为了兼容,以前就绪队列是固定的,后面改成链表了) 创建epfd
struct epoll_event events[POLL_SIZE] = {0}; // 这里POLL_SIZE就是就绪队列的大小了,小一点没关系(如50),因为即使百万并发,活跃的也就1w,多跑几次了就是了
struct epoll_event ev;
ev.events = EPOLLIN;
ev.data.fd = listenfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, listenfd, &ev); // 将listenfd加入epoll,拷贝到内核:只需要拷贝一次,不需要拷贝出来
while (1) {
int nready = epoll_wait(epfd, events, POLL_SIZE, 5); // 取事件, 拷贝到用户态:只拷贝就绪的事件了
if (nready == -1) {
continue;
}
int i = 0;
// 遍历就绪队列
for (i = 0;i < nready;i ++) {
int clientfd = events[i].data.fd;
if (clientfd == listenfd) { // 处理listenfd
struct sockaddr_in client;
socklen_t len = sizeof(client);
if ((connfd = accept(listenfd, (struct sockaddr *)&client, &len)) == -1) {
printf("accept socket error: %s(errno: %d)\n", strerror(errno), errno);
return 0;
}
printf("accept\n");
ev.events = EPOLLIN;
ev.data.fd = connfd;
epoll_ctl(epfd, EPOLL_CTL_ADD, connfd, &ev);
} else if (events[i].events & EPOLLIN) { // 处理connfd
n = recv(clientfd, buff, MAXLNE, 0);
if (n > 0) {
buff[n] = '\0';
printf("recv msg from client: %s\n", buff);
send(clientfd, buff, n, 0);
} else if (n == 0) { // 读完了就从epoll中移除connfd
ev.events = EPOLLIN;
ev.data.fd = clientfd;
epoll_ctl(epfd, EPOLL_CTL_DEL, clientfd, &ev);
close(clientfd);
}
}
}
}
#endif
close(listenfd);
return 0;
}