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;
}