一、poll
poll的實現和select非常相似，只是描述fd集合的方式不同，poll使用pollfd結構而不是select的fd_set結構，其他的都差不多。
二、poll相關函數
#include <poll.h>
int poll(struct pollfd *fds, nfds_t nfds, int timeout);
//fds: pollfd結構體
events: 要監視的事件
revents: 已經發生的事件, 設置標誌 來反映相關條件的存在
常量 說明
POLLIN 普通或優先級帶數據可讀
POLLRDNORM 普通數據可讀
POLLRDBAND 優先級帶數據可讀
POLLPRI 高優先級數據可讀
POLLOUT 普通數據可寫
POLLWRNORM 普通數據可寫
POLLWRBAND 優先級帶數據可寫
// 只能作爲描述字的返回結果存儲在revents中
POLLERR 發生錯誤
POLLHUP 發生掛起
POLLNVAL 描述字不是一個打開的文件
struct pollfd
{ int fd; /* 文件描述符 */
short events; /* 請求的事件 */
short revents; /* 返回的事件 */
};//nfds: 要監視的描述符的數目
//timeout:指定poll在返回前沒有接收事件時應該等待的時間【單位：ms】。
INFTIM: 永不超時
0 ：立即返回
>0: 等待指定的時間
2、特點：
pollfd並沒有最大數量限制（但是數量過大後性能也是會下降）。
和select函數一樣，poll返回後，需要輪詢pollfd來獲取就緒的描述符。
select和poll都需要在返回後，通過遍歷文件描述符來獲取已經就緒的socket。事實上，同時連接的大量客戶端在一時刻可能只有很少的處於就緒狀態，因此隨着監視的描述符數量的增長，其效率也會線性下降。
三、poll 服務器
#include <stdio.h>
#include<stdlib.h>
#include<sys/types.h>
#include<sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include<unistd.h>
#include<string.h>
#include<errno.h>
#include<sys/select.h>
#include<sys/poll.h>
#define _BACKLOG_ 5
#define _MAX_NUM_ 64
void usage(const char *proc)
{
printf("Usage: %s [ip][port]\n",proc);
}
static int startup(const char *ip,const int port)
{
int sock=socket(AF_INET,SOCK_STREAM,0);
if(sock < 0)
{
perror("socket");
exit(1);
}
int opt=1;
setsockopt(sock,SOL_SOCKET,SO_REUSEADDR,&opt,sizeof(opt));
struct sockaddr_in local;
local.sin_family=AF_INET;
local.sin_port=htons(port);
local.sin_addr.s_addr=inet_addr(ip);
if(bind(sock,(struct sockaddr*)&local,sizeof(local)) < 0)
{
perror("bind");
exit(2);
}
if(listen(sock,_BACKLOG_) < 0)
{
perror("listen");
exit(3);
}
return sock;
}
static int poll_server(int listen_sock)
{
struct sockaddr_in client;
socklen_t len=sizeof(client);
struct pollfd fds[_MAX_NUM_];
fds[0].fd=listen_sock;
fds[0].events=POLLIN;
fds[0].revents=0;
size_t i=0;
for(;i<_MAX_NUM_;++i)
{
fds[i].fd=-1;
fds[i].events=0;
fds[i].revents=0;
}
int max_fd=1;
int timeout=5000;
while(1)
{
switch(poll(fds,max_fd,timeout))
{
case -1://error
perror("poll");
break;
case 0://timeout
printf("poll timeout");
break;
default://normal
{
for(i=0;i<_MAX_NUM_;++i)
{
//is listen events ready?
if(fds[i].fd == listen_sock && fds[i].revents == POLLIN)
{
int accept_sock=accept(listen_sock,(struct sockaddr*)&client,&len);
if(accept_sock < 0)
{
perror("accept");
continue;
}
printf("get a client...[ip: %s][port: %d]\n",\
inet_ntoa(client.sin_addr),ntohs(client.sin_port));
for(i=0;i<_MAX_NUM_;++i)
{
if(fds[i].fd == -1)
{
fds[i].fd=accept_sock;
fds[i].events=POLLIN;
max_fd++;
break;
}
}
if(i == _MAX_NUM_)
{
close(accept_sock);
}
}
else if(fds[i].fd > 0 && fds[i].revents == POLLIN)
{
char buf[1024];
ssize_t _size=read(fds[i].fd,buf,sizeof(buf)-1);
if(_size > 0)//read success
{
buf[_size]='\0';
printf("Client # %s",buf);
}
else if(_size == 0)//client close
{
printf("clint close...\n");
struct pollfd tmp=fds[i];
fds[i]=fds[max_fd-1];
close(fds[max_fd-1].fd);
fds[max_fd-1].fd=-1;
fds[max_fd-1].events=0;
fds[max_fd-1].revents=0;
--max_fd;
}
}
else
{}
}
}
break;
}
}
return 0;
}
int main(int argc,char *argv[])
{
if(argc != 3)
{
usage(argv[0]);
return 1;
}
char *ip=argv[1];
int port=atoi(argv[2]);
int listen_sock=startup(ip,port);
poll_server(listen_sock);
close(listen_sock);
return 0;
}總結：
poll在返回後，需要通過遍歷文件描述符來獲取已經就緒的socket從而進行下一步操作；而且使用完監聽套接字後，都需要進行關閉。