poll_server

poll与select不同在于描述符存储方式不同和参数类型不同。
1.结构体数组的管理：当每次有需要关心的描述符时，将其放入结构体中，每次有无效的描述符后，将其描述符置-1，下次poll函数会忽略它。当有新的描述符加入时，从头遍历结构体，将为-1的元素设为要关心的描述符事件状态。切记：当新的描述符加到结构体数组末尾时要更新关心描述符个数，即poll第二个参数。
2.每次调用poll后，结构体元素revents会存储就绪事件状态，当每次重新调用poll之前时，系统会自己设置其为0，重新监听关心事件（不需要用户重新置0）
3.poll中参数不是输入，输出型，因此timeout,struct pollfd *fds参数不需重置，nfds看情况（参照第一点），而select函数是输入输出类型，每次调用前需重置。

int poll（struct pollfd*fds， nfds_t nfds， int timeout）；

这个结构中fd表示文件描述符，events表示请求检测的事件，revents表示检测之后返回的事件，如果当某个文件描述符有状态变化时，revents的值就不为空。

struct pollfd结构如下

struct pollfd {
int fd;
short events;
short revents;
};

poll的基本事件都有：
POLLIN:当前有数据可读 （服务器角度）
POLLOUT:当前有数据可输出
POLLPRI:当前有紧急数据处理
若fds=-1,则poll()无效。

参数：

fds：存放需要被检测状态的Socket描述符；与select不同（select函数在调用之后，会清空检测socket描述符的数组），每当调用这个函数之后，系统不会清空这个数组，而是将有状态变化的描述符结构的revents变量状态变化，操作起来比较方便；
nfds：用于标记数组fds中的struct pollfd结构元素的总数量；

函数返回值：

0：表示数组fds中有socket描述符的状态发生变化，或可以读取、或可以写入、或出错。并且返回的值表示这些状态有变化的socket描述符的总数量；此时可以对fds数组进行遍历，以寻找那些revents不空的socket描述符，然后判断这个里面有哪些事件以读取数据。
=0：表示没有socket描述符有状态变化，并且调用超时。
<0：此时表示有错误发生，此时全局变量errno保存错误码。

#include <stdio.h>#include <sys/socket.h>#include <stdlib.h>#include <string.h>#include <sys/select.h>#include <sys/time.h>#include <sys/types.h>#include <unistd.h>#include <poll.h>#include <netinet/in.h>#include <arpa/inet.h>int startup(const char*ip, int port){    int sock = socket(AF_INET, SOCK_STREAM, 0);    if(sock < 0)    {        perror("socket");        exit(-1);    }    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(-1);    }    if(listen(sock, 10) < 0)    {        perror("listen");        exit(-1);    }    return sock;}static void usage(const char* proc){    printf("%s [local_ip] [local_port]\n", proc);}int main(int argc, char* argv[]){    if(argc != 3)    {        usage(argv[0]);        return -1;    }    int listen_sock = startup(argv[1], atoi(argv[2]));    int maxfd = 0;    struct pollfd evs[2];    evs[0].fd = listen_sock;//需要监测的文件描述符    evs[0].events = POLLIN;    evs[0].revents = 0;    int timeout = 8000;    int nums = sizeof(evs)/sizeof(evs[0]);    struct sockaddr_in client;    socklen_t len = sizeof(client);    int i=1;    for(; i<nums; i++)    {        evs[i].fd = -1;    }    while(1)    {        switch(poll(evs, maxfd+1, timeout))        {            case 0:                printf("timeout...\n");                break;            case -1:                perror("poll");                break;            default:                {                    for(i=0; i<nums; i++)                    {//listen_sock的读事件就绪，获取客户端fd                        if((evs[i].fd == listen_sock) && (evs[i].revents)&POLLIN)                        {                            //获取客户端                            int new_sock = accept(listen_sock, (struct sockaddr*)&client, &len);                            if(new_sock < 0)                            {//取数据失败                                perror("accept");                                continue;                            }                            printf("get new client:[%s:%d]\n", inet_ntoa(client.sin_addr), ntohs(client.sin_port));                            //将获取的客户端new_sock放入poll结构体数组中，以便poll()监测                            int j=1;                            for(; j<nums; j++)                            {                                if(evs[j].fd == -1)                                {//将new_sock添加进去                                    evs[j].fd = new_sock;                                    evs[j].events = POLLIN;                                    evs[j].revents = 0;                                    break;                                }                            }                            if(j == nums)                            {                                close(new_sock);                            }                            if(j > maxfd)//更新有效文件描述符个数                                maxfd = j;                        }                        else if((evs[i].fd>0) && (evs[i].revents)&POLLIN)                        {//其他fd就绪，如：read, write...                            char buf[1024];                            ssize_t s = read(evs[i].fd, buf, sizeof(buf)-1);                            if(s > 0)                            {//读成功                                buf[s] = 0;                                printf("client say# %s", buf);                            }                            //从标准输入中获取字符，写给客户端                            ssize_t _s = read(0, buf, sizeof(buf)-1);                            if(_s > 0)                            {                                write(evs[i].fd, buf, strlen(buf));                            }                            else if(s == 0)                            {                                printf("read done..., break\n");                                //关闭该文件描述符，获取其他有效的文件描述符                                close(evs[i].fd);                                int k = 1;                                for(; k < nums; k++)                                {                                    if(evs[k].fd != -1 && (k!=i))                                    {                                        int tmp = evs[i].fd;                                        evs[i].fd = evs[k].fd;                                        evs[k].fd = tmp;                                    }                                }                            }                            else                            {                                perror("read");                                continue；                            }                        }                    }            }            break;        }    }    return 0;}