多路IO转接服务器实现方法二：poll()函数

相较于多路IO转接服务器实现方法一：select()函数，使用poll()函数的优点有

1. 文件描述符的上限可以突破1024
2. select()函数监听集合与返回的满足监听条件的集合为一个集合，而poll函数将监听集合与符合监听条件的集合实现了分离
3. 搜索满足条件的文件描述符的范围缩小了

但，poll函数不能实现跨平台，只能在Linux平台上使用，而select()函数可以跨平台

poll()函数

函数原型

int poll(struct pollfd *fds, nfds_t nfds, int timeout);

参数

• fds：类型为struct pollfd的数组
  如果不想再监控某个文件描述符时，把pollfd中，fd设置为-1

  struct pollfd{
  	int fd;//文件描述符
  	short events;//监控的事件
  	short revents;//监控事件中满足条件返回的事件
  }
  

• nfds：监控数组中有多少文件描述符需要被监控
• timeout：毫秒级等待
  • -1：阻塞等
  • 0：立即返回，不阻塞进程
  • >0：等待指定毫秒数，如当前系统时间精度不够毫秒，向上取值

返回值

• 成功：返回所监听的所有监听集合中满足条件的总数。
• 失败：返回-1，并设置errno。

例子

#include <stdio.h>
#include <sys/socket.h>
#include <poll.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <ctype.h>

#define PORT 8888
#define IP "127.0.0.1"
#define MAXNUM 6
int main()
{
    //获得通信套接字
    int sfd = socket(AF_INET,SOCK_STREAM,0);

    //绑定
    struct sockaddr_in serv_addr;
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(PORT);
    inet_pton(AF_INET,IP,&serv_addr.sin_addr.s_addr);
    socklen_t serv_len = sizeof (serv_addr);
    int ret = bind(sfd,(struct sockaddr*)&serv_addr,serv_len);
    if(ret != 0){
        printf("bind error：%s\n",strerror(ret));
    }

    //监听
    listen(sfd,128);

    //初始化struct pollfd结构体数组，将sfd放入数组的0号位置
    struct pollfd client[MAXNUM];
    client[0].fd = sfd;
    client[0].events = POLLIN;

    for(int i = 1; i < MAXNUM; i++){
        client[i].fd = -1;
    }
    int maxi = 0;
    char buf[BUFSIZ];
    memset(buf,0,sizeof (buf));
    while(1){
        ret = poll(client,maxi+1,-1);

        //此时监听的sfd如果有读事件发生，说明有新的连接要建立，调用accept()函数
        if(client[0].revents & POLLIN){
            struct sockaddr_in clie_addr;
            socklen_t clie_len = sizeof (clie_addr);
            //返回客户端 cfd
            int cfd = accept(client[0].fd,(struct sockaddr*)&clie_addr,&clie_len);
            printf("%sconnected and port=%d\n",inet_ntop(AF_INET,&clie_addr.sin_addr.s_addr,buf,sizeof (buf)),
                   ntohs(clie_addr.sin_port));

            //将客户端文件描述符加入监听数组中即client[]中
            //找到数组中的空位置才能加入
            int i;
            for(i = 1; i < MAXNUM;i++){
                if(client[i].fd < 0){
                    client[i].fd = cfd;
                    break;
                }
            }
            //这里设置了最大的监听数为MAXNUM，如果超过了这个值就不再监听，退出
            if(i == MAXNUM){
                printf("too many connected...\n");
                exit(1);
            }

            //将cfd加入监听数组client[]中后，设置其监听属性，即是要监听他的读事件还是写事件异常事件
            client[i].events = POLLIN;//此处设置为监听他的读事件

            //将数组下标位置后移，其实就是现在一共实际要监听的文件描述符个数
            if(i > maxi){
                maxi = i;
            }

            //此处if语句条件满足要continue的原因是，假如ret中只有一个sfd时，说明客户端没有向服务端发送数据，所以跳出继续监听sfd的下一个连接
            if(--ret == 0){
                continue;
            }
        }
        //代码走到这里说明监听数组client[]中，有客户端在发数据了，接下来就是读数据并写回去
        int sockfd;
        int k;
        for(k = 1; k <= MAXNUM; k++){
            if((sockfd = client[k].fd) < 0){
                continue;
            }
            //k从1开始，就是跳过了第一个sfd，其余的客户端发送数据的文件描述符
            char rwbuf[BUFSIZ];
            memset(buf,0,sizeof (buf));
            if(client[k].revents & POLLIN){
                //如果满足 条件，开始读数据
                int len = read(sockfd,rwbuf,sizeof (rwbuf));
                if(len == 0){
                    //说明客户端关闭了
                    printf("client %d disconnected...\n",k);
                    close(sockfd);
                    client[k].fd = -1;
                }else if(len > 0){
                    //读到了数，将小写转换为大写
                    for(int i = 0; i< len; i++){
                        rwbuf[i] = toupper(rwbuf[i]);
                    }
                    write(sockfd,rwbuf,len);
                }
                if(--ret == 0){
                    break;
                }
            }
        }
    }
    close(sfd);
    return 0;
}