一、关于select、poll、epoll这三者使用、对比介绍这篇:
网络编程--IO多路复用之select,poll和epoll详解
二、poll的使用和上篇的select大同小异:
int main(void)
{
//当客户端结束时,服务器还维持一个僵尸进程
signal(SIGCHLD, SIG_IGN);//使用signal忽略SIGCHLD信号(第一种方法)
/*signal(SIGCHLD, handle_sigchld);//(第二种方法)*/
int listenfd;//建立套接字
if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
/* if((listenfd=socket(PF_INET,SOCK_STREAM, 0))<0)*/
ERR_EXIT("socket");
//建立服务器地址
struct sockaddr_in servaddr;
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(5188);
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
/*servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");*/
/*inet_aton("127.0.0.1",&servaddr.sin_addr);*/
int on = 1;
//为了使服务器处于TIME_WATTING状态,能再次绑定地址
if(setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))<0)
ERR_EXIT("setsockopt");
//绑定地址
if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("bind");
//监听套接字
if(listen(listenfd,SOMAXCONN)<0)//套接字变为被动套接字
ERR_EXIT("listen");
//存放客户端地址
struct sockaddr_in peeraddr;
socklen_t peerlen;
int conn;//接受accept的套接字
int i;
int maxi = 0;
struct pollfd client[2048];//假设一个进程可以打开最大的文件描述符数为2048
for(i=0; i<2048; i++)
{
client[i].fd = -1;//空闲可以容纳新的客户端
}
int nready;
client[0].fd = listenfd;
client[0].events = POLLIN;//读可读事件感兴趣
//目前,加入到数组中的文件描述符个数等于1
while(1)
{
nready = poll(client, maxi+1, -1);
//nready = select(maxfd+1, &rset, NULL,NULL, NULL);
if(nready==-1)
{
if(errno == EINTR)
continue;
ERR_EXIT("poll");
}
if(nready==0)
continue;
if(client[0].revents & POLLIN)//返回发生的事件与POLLIN相同
//if(FD_ISSET(listenfd, &rset))
{
peerlen= sizeof(peeraddr);
conn=accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen);
if(conn==-1)
ERR_EXIT("accept");
for(i=0; i<2048;i++)
{
if(client[i].fd<0)
{
client[i].fd=conn;//将新的套接字加入
if(i>maxi)
maxi=i;
break;
}
}
if(i==2048)
{
fprintf(stderr,"too many clients");
exit(EXIT_FAILURE);
}
printf("ip=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
//FD_SET(conn, &allset);
client[i].events = POLLIN;//添加事件
if(--nready<=0)
continue;
}
//监听套接口总为client[0]
//已连接套接口从1到maxi
for(i=1; i<=maxi; i++)
{
conn=client[i].fd;
if(conn == -1)
continue;
if(client[i].events & POLLIN)//事件为POLLIN则处理
//if(FD_ISSET(conn, &rset))
{
char recvbuf[1024] = {0};
int ret = readline(conn, recvbuf,1024);//先接受包头
if(ret==-1)//读取失败则退出
ERR_EXIT("readline");
else if(ret==0)//客户端关闭
{
printf("client_close\n");
//FD_CLR(conn, &allset);
client[i].fd = -1;
close(conn);
}
fputs(recvbuf,stdout);//打印标准输出
writen(conn,recvbuf, strlen(recvbuf));
if(--nready<=0)
break;
}
}
}
return 0;
}
三、我们主要关注epoll
//仅修改服务器端
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>//使用signal函数
#include <sys/wait.h>//使用wait函数
#include <sys/epoll.h>
#include <fcntl.h>
#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>
#include <vector>
#include <algorithm>
typedef std::vector<struct epoll_event> EventList;
#define ERR_EXIT(m) \
do \
{ \
perror(m); \
exit(EXIT_FAILURE);\
}while(0)
/*
activate_nonblock —— 设置I/O为非阻塞模式
*/
void activate_nonblock(int fd)
{
int ret;
int flags = fcntl(fd, F_GETFL);//获取文件标志
if(flags == -1)
{
ERR_EXIT("fcntl");
}
flags |= O_NONBLOCK;//设置非阻塞模式
ret = fcntl(fd, F_SETFL, flags);
if(ret == -1)
ERR_EXIT("fcntl");
}
//读确定大小的包
//ssize_t有符号数,size_t为无符号数
ssize_t readn(int fd, void *buf, size_t count)
{
size_t nleft=count;//剩余字节数
ssize_t nread;//已经接受的字节数
char *bufp = (char*)buf;//
while(nleft>0)
{
if((nread=read(fd, bufp, nleft))<0)
{
if(errno==EINTR)//信号中断
{
continue;
}
return -1;//否则出错
}
else if(nread==0)//表示对方关闭传送
{
return count-nleft;//返回已读字节数
}
bufp+=nread;//进行指针偏移
nleft -= nread;
}
return count;
}
ssize_t writen(int fd, const void *buf, size_t count)
{
size_t nleft=count;//剩余字节数
ssize_t nwritten;//已经写入的字节数
char *bufp = (char*)buf;//
while(nleft>0)
{
if((nwritten=write(fd, bufp, nleft))<0)
{
if(errno==EINTR)//信号中断
{
continue;
}
return -1;//否则出错
}
else if(nwritten==0)
{
continue;
}
bufp+=nwritten;//进行指针偏移
nleft -= nwritten;
}
return count;
}
ssize_t recv_peek(int sockfd, void *buf,size_t len)
{
while(1)
{
//recv函数只用于套接口
//recv函数读取后,不将数据在缓冲区清除
int ret= recv(sockfd, buf, len, MSG_PEEK);
if(ret == -1 && errno == EINTR)
continue;
return ret;
}
}
ssize_t readline(int sockfd, void *buf, size_t maxline)
{
int ret;//设置窥探返回值
int nread;//设置已窥探字符数
char *bufp = (char*)buf;//缓存buf
int nleft = maxline;//设置maxline为包最大长度,nleft为剩余需读取字符数
while(1)
{
ret=recv_peek(sockfd, bufp, nleft);
if(ret<0)
{
return ret;
}
else if(ret=0)
{
return ret;//对方终止了传送
}
nread=ret;
int i;//检测有没有‘\n’字符,有则读取
for(i=0; i<nread;i++)
{
if(bufp[i]=='\n')
{
ret= readn(sockfd,bufp, i+1);
if(ret != i+1)//已经窥探到有i+1字符,如果没有则错误
exit(EXIT_FAILURE);
return ret;
}
}
//如果没有读到‘\n’,则将消息读入,直到最大包
nleft -= nread;
ret= readn(sockfd, bufp, nread);
if(ret != nread)//已经窥探到有nread个字符,如果readn函数不能读取这么多,则错误
{
exit(EXIT_FAILURE);
}
bufp += nread;
}
return -1;
}
void do_service(int conn)//执行连接建立后,内容读取,和返回内容
{
char recvbuf[1024];
int n;
while(1)
{
memset(recvbuf, 0, sizeof(recvbuf));
int ret = readline(conn, recvbuf,1024);//先接受包头
if(ret==-1)//读取失败则退出
ERR_EXIT("readline");
else if(ret==0)//客户端关闭
{
printf("client_close\n");
break;
}
fputs(recvbuf,stdout);//打印标准输出
writen(conn,recvbuf, strlen(recvbuf));
}
}
void ehco_srv(int conn)//执行连接建立后,内容读取,和返回内容
{
char recvbuf[1024];
int n;
while(1)
{
memset(recvbuf, 0, sizeof(recvbuf));
int ret = readline(conn, recvbuf,1024);//先接受包头
if(ret==-1)//读取失败则退出
ERR_EXIT("readline");
else if(ret==0)//客户端关闭
{
printf("client_close\n");
break;
}
fputs(recvbuf,stdout);//打印标准输出
writen(conn,recvbuf, strlen(recvbuf));
}
}
void handle_sigchld(int sig)
{
//只等待一个子进程结束,就退出消除一个子进程产生的僵尸进程,无法处理并发的多个子进程,造成的僵尸进程
//wait(NULL);
//等待选项为不挂起,用来处理客户端多个并发子进程连接后关闭时造成的多个僵尸进程
while(waitpid(-1, NULL, WNOHANG)>0);
}
void handle_sigpipe(int sig)
{
printf("recv a sig=%d\n", sig);
}
int main(void)
{
int count = 0;
//处理A调用了close,A不能再接收和发送数据段,但对等方B可以向A发送数据包,
//如果B这样做了,A返回RST段,B再向A发送数据,A会产生SIGPIPE信号,该信号直接忽略就好
signal(SIGPIPE, handle_sigpipe);
//当客户端结束时,服务器还维持一个僵尸进程
/*signal(SIGCHLD, SIG_IGH);//使用signal忽略SIGCHLD信号(第一种方法)*/
signal(SIGCHLD, handle_sigchld);//(第二种方法)
int listenfd;//建立套接字
if((listenfd=socket(PF_INET,SOCK_STREAM,IPPROTO_TCP))<0)
/* if((listenfd=socket(PF_INET,SOCK_STREAM, 0))<0)*/
ERR_EXIT("socket");
//建立服务器地址
struct sockaddr_in servaddr;
memset(&servaddr,0,sizeof(servaddr));
servaddr.sin_family = AF_INET;
servaddr.sin_port = htons(5188);
servaddr.sin_addr.s_addr = htonl(INADDR_ANY);
/*servaddr.sin_addr.s_addr = inet_addr("127.0.0.1");*/
/*inet_aton("127.0.0.1",&servaddr.sin_addr);*/
int on = 1;
//为了使服务器处于TIME_WATTING状态,能再次绑定地址
if(setsockopt(listenfd,SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on))<0)
ERR_EXIT("setsockopt");
//绑定地址
if(bind(listenfd,(struct sockaddr*)&servaddr,sizeof(servaddr))<0)
ERR_EXIT("bind");
//监听套接字
if(listen(listenfd,SOMAXCONN)<0)//套接字变为被动套接字
ERR_EXIT("listen");
//epoll函数修改
std::vector<int> clients;//存放已连接套接字
int epollfd;//存放epoll实例
epollfd = epoll_create1(EPOLL_CLOEXEC);//创建一个EPOLL实例
//将感兴趣的事件加入到epoll队列
struct epoll_event event;
event.data.fd = listenfd;
event.events = EPOLLIN | EPOLLET;//边沿触发
epoll_ctl(epollfd, EPOLL_CTL_ADD, listenfd, &event);
EventList events(16);//存放并发事件
//存放客户端地址
struct sockaddr_in peeraddr;
socklen_t peerlen;
int conn;//接受accept的套接字
int i;
int nready;
while(1)
{//&*events.begin()动态数组首地址
nready = epoll_wait(epollfd, &*events.begin(), static_cast<int>(events.size()), -1);
//nready = poll(client,maxi+1, -1);
//nready = select(maxfd+1, &rset, NULL,NULL, NULL);
if(nready==-1)
{
if(errno == EINTR)
continue;
ERR_EXIT("epoll_wait");
}
if(nready==0)
continue;
//数组已满,等待的数已经满了,扩充数组
if((size_t)nready == events.size())//等待的事件的个数
events.resize(events.size()*2);
//开始处理已检测到的事件
for(i=0; i<nready; i++)
{
//处理监听套接口
if(events[i].data.fd == listenfd)
{
peerlen = sizeof(peeraddr);
conn = accept(listenfd, (struct sockaddr*)&peeraddr, &peerlen);
if(conn==-1)
{
ERR_EXIT("accept");
}
printf("ip=%s port=%d\n", inet_ntoa(peeraddr.sin_addr), ntohs(peeraddr.sin_port));
clients.push_back(conn);
activate_nonblock(conn);
//监听套接口变为已连接套接口,放入队列,进行管理,等待数据到来
event.data.fd = conn;
event.events = EPOLLIN | EPOLLET;
epoll_ctl(epollfd, EPOLL_CTL_ADD, conn, &event);
}
else if(events[i].events & EPOLLIN)//已连接套接字
{
conn = events[i].data.fd;
if(conn<0)
continue;
char recvbuf[1024] = {0};
int ret = readline(conn, recvbuf, 1024);
if(ret ==-1)
ERR_EXIT("readline");
if(ret ==0)
{
printf("client close\n");
close(conn);
event= events[i];
epoll_ctl(epollfd, EPOLL_CTL_DEL, conn, &event);
clients.erase(std::remove(clients.begin(), clients.end(), conn), clients.end());
}
fputs(recvbuf,stdout);//打印标准输出
writen(conn,recvbuf, strlen(recvbuf));
}
}
}
return 0;
}