Linux Poll函数初级实例

/**********************
poll回射客户服务器模型
**********************/


#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <sys/wait.h>
#include <poll.h>
#include <unistd.h>


#include <stdlib.h>
#include <stdio.h>
#include <errno.h>
#include <string.h>


#include <vector>
#include <iostream>


#define ERR_EXIT(m) \
do \
{  \
perror(m); \
exit(EXIT_FAILURE);   \
} while(0) \


typedef std::vector<struct pollfd> PollFdList;


int main()
{
signal(SIGPIPE, SIG_IGN);


/*******************************************************************
PIPE信号是如何产生：
如果客户端关闭套接字close
而服务器端调用write，服务器会接受一个RST segment(TCP传输层）
这时，表明服务器端不应该再调用write
但是，如果服务器再次调用了write，这时会产生SIGPIPE信号，
如果不忽略这个信号，会退出整个进程
对于高可用服务器来说 7*24小时服务，要忽略这个信号

应该避免TIME_WAIT状态的出现
它会在内核当中，一段时间内保留一些内核资源
应尽可能在服务器端避免

如果服务器端主动断开连接（先于client调用close），服务端就会进入TIME_WAIT
协议设计上，应该让客户端主动断开连接，这样就把TIME_WAIT状态分散到大量的客户端。
如果客户端不活跃了，一些客户端不断开连接，这样就会占用服务器端的资源。
服务器端也要有机制来T掉不活跃的连接 close
**********************************************************************/
signal(SIGCHLD, SIG_IGN);

int listenfd;
// SOCK_NONBLOCK | SOCK_CLOEXEC 作用：采用非阻塞模式 nonblocking socket + I/O 复用  效率比fcntl高
if ((listenfd = socket(PF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, IPPROTO_TCP)) < 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);

const int on = 1;
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");

/****************************
poll函数的使用
关注的是POLLIN事件 （POLLIN ==> 有事件可读）
*****************************/
struct pollfd pfd;
pfd.fd = listenfd;
pfd.events = POLLIN;

//把关注的文件描述符添加到vector中
PollFdList pollfds;
pollfds.push_back(pfd);

int nready;
struct sockaddr_in peeraddr;
socklen_t peerlen;
int connfd;

while (1)
{
//传入的是vector的首地址  *是第一个元素， &*是第一个元素的地址
//C++ 11中 可以写 pollfds.data()
//-1是超时时间，表示永远等待，不设定超时，直到发生时间返回
nready = poll(&*pollfds.begin(), pollfds.size(), -1);
if (-1 == nready)
{
if (errno == EINTR)
continue;
ERR_EXIT("poll");
}
if (0 == nready) //没有事件发生
continue;

//监听有POLLIN到来
//accept4 可以有 SOCK_NONBLOCK | SOCK_CLOEXEC 功能 accept没有
//SOCK_CLOEXEC 在新打开的文件描述符里设置 close - on - exec(FD_CLOEXEC) 标记
if (pollfds[0].revents & POLLIN)
{
peerlen = sizeof(peeraddr);
connfd = ::accept4(listenfd, (struct sockaddr*)&peeraddr, &peerlen, SOCK_NONBLOCK | SOCK_CLOEXEC);
if (-1 == connfd)
ERR_EXIT("accept4");

//把得到的已连接socket加入到监听
pfd.fd = connfd;
pfd.events = POLLIN;
pfd.revents = 0; //这时返回事件一定要等于0，因为当前只是先加入监听 没有任何返回
pollfds.push_back(pfd);
--nready; //处理了一个事件就--

//connect success
std::cout << "ip = " << inet_ntoa(peeraddr.sin_addr) << " port = " << ntohs(peeraddr.sin_port) << std::endl;

if (0 == nready) //事件都处理完毕
continue;//继续监听
}

//遍历哪些已连接套接字产生了可读事件
//pollfds.begin() + 1
//因为第一个fd 肯定是listenfd
//所以从begin()+1的位置，开始遍历connfd
//nready > 0 ：事件个数是否都处理完毕
for (PollFdList::iterator it = pollfds.begin() + 1; it != pollfds.end() && nready > 0; ++it)
{
//产生可读事件
if (it->revents & POLLIN)
{
--nready;
connfd = it->fd; //把已连接套接字取出来读取
char buf[1024] = { 0 };
int ret = read(connfd, buf, 1024);
if (ret == -1)
ERR_EXIT("read");

if (0 == ret) //对方关闭套接字
{
std::cout << "client close" << std::endl;
it = pollfds.erase(it);
--it; //--it是因为 vector删除一个元素 后面元素往前靠 it指向现在这个位置，下一次遍历++it时，会把现在当前位置漏掉

close(connfd);
continue;
}
//读到的内容大于0 则回射回去
std::cout << buf;
write(connfd, buf, strlen(buf));
}
}
}
return 0;
}