【网络编程】poll函数

poll和select的区别不大，主要是poll没有连接数限定，因为它用的链表实现

1. #include <poll.h>
3. int poll(struct pollfd *fds, nfds_t nfds, int timeout);
4. struct pollfd {
5.     int   fd;         //要监控的文件描述符,如果fd为-1, 表示内核不再监控
6.     short events;     //输入参数, 表示告诉内核要监控的事件, 读事件, 写事件, 异常事件
7.     short revents;    //输出参数, 表示内核告诉应用程序有哪些文件描述符有事件发生
8. };
9. events/revents:
10.                    POLLIN:可读事件
11.                    POLLOUT: 可写事件
12.                    POLLERR: 异常事件
13. nfds: 告诉内核监控的范围, 具体是: 数组下标的最大值+1
14. timeout:
15.         =0: 不阻塞, 立刻返回
16.         -1: 表示一直阻塞, 直到有事件发生
17.         >0: 表示阻塞时长, 在时长范围内若有事件发生会立刻返回;
18.                 如果超过了时长也会立刻返回
19. 函数返回值:
20.         >0: 发生变化的文件描述符的个数
21.         =0: 没有文件描述符发生变化
22.         -1: 表示异常

复制代码

使用poll来监控多个文件描述符举行客户端通讯
代码：

1. #include "socketwrap.h"
2. #include <arpa/inet.h>
3. #include <poll.h>
4. #include <strings.h>
5. #include <string.h>
6. #include <ctype.h>
8. #define POLLLEN 1024
10. int main()
11. {
12.     int sfd = Socket(AF_INET, SOCK_STREAM, 0);
14.     // 设置端口复用
15.     int opt = 1;
16.     setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(int));
18.     struct sockaddr_in soaddr;
19.     bzero(&soaddr, sizeof(soaddr));
21.     soaddr.sin_family = AF_INET;
22.     soaddr.sin_port = htons(9999);
23.     soaddr.sin_addr.s_addr = htonl(INADDR_ANY);
25.     Bind(sfd, (struct sockaddr *)&soaddr, sizeof(soaddr));
27.     // 监听-listen
28.     Listen(sfd, 128);
30.     struct pollfd connfd[POLLLEN];
32.     int maxfd;  // 当前需要监听的最大的文件描述符
33.     int nready; // 返回的需要处理的个数
34.     int cfd;    // 通信描述符
35.     int i, maxi = 0;
37.     struct sockaddr_in clientsocket;
38.     socklen_t clilen;
39.     char buff[64]; // 通信数据
41.     // 初始化有效的文件描述符数组
42.     for (int i = 0; i < POLLLEN; i++)
43.     {
44.         bzero(&connfd[i], sizeof(connfd[i]));
45.         connfd[i].fd = -1;
46.     }
48.     maxfd = sfd;
49.     connfd[0].fd = sfd;
50.     connfd[0].events = POLLIN;
52.     while (1)
53.     {
54.         clilen = sizeof(clientsocket);
55.         bzero(&clientsocket, clilen);
57.         nready = poll(connfd, maxfd + 1, -1);
59.         if (nready < 0)
60.         {
61.             if (errno == EINTR) // 被信号中断
62.             {
63.                 continue;
64.             }
65.             perror("poll error");
66.             break;
67.         }
69.         if (nready > 0)
70.         {
71.             if (connfd[0].revents == POLLIN)
72.             {
73.                 // 有连接到来
74.                 cfd = Accept(sfd, (struct sockaddr *)&clientsocket, &clilen);
76.                 for (i = 0; i < POLLLEN; i++)
77.                 {
78.                     if (connfd[i].fd == -1)
79.                     {
80.                         connfd[i].events = POLLIN;
81.                         connfd[i].fd = cfd;
82.                         maxi = i;
83.                         if (maxfd < cfd)
84.                         {
85.                             maxfd = cfd;
86.                         }
87.                         // 打印客户端的IP和PORT
88.                         char sIP[16];
89.                         memset(sIP, 0x00, sizeof(sIP));
90.                         printf("client [%s:%d] connect\n", inet_ntop(AF_INET, &clientsocket.sin_addr.s_addr, sIP, sizeof(sIP)), htons(clientsocket.sin_port));
91.                         break;
92.                     }
93.                 }
94.                 if (i == POLLLEN)
95.                 {
96.                     close(cfd);
97.                     printf("connect too much, server busy\n");
98.                     continue;
99.                 }
101.                 if (--nready == 0)
102.                 {
103.                     continue;
104.                 }
105.             }
107.             for (i = 1; i <= maxi; i++)
108.             {
109.                 if (connfd[i].fd == -1)
110.                 {
111.                     continue;
112.                 }
113.                 if (connfd[i].revents == POLLIN)
114.                 {
115.                     // 有数据发送过来
116.                     int n;
117.                     int sockfd = connfd[i].fd;
118.                     memset(buff, 0x00, sizeof(buff));
119.                     n = Read(sockfd, buff, sizeof(buff));
120.                     if (n < 0)
121.                     {
122.                         perror("read over");
123.                         close(sockfd);
124.                         bzero(&connfd[i], sizeof(bzero));
125.                         connfd[i].fd = -1; // 将connfd[i]置为-1,表示该位置可用
126.                     }
127.                     else if (n == 0)
128.                     {
129.                         // printf("client is closed\n");
130.                         close(sockfd);
131.                         bzero(&connfd[i], sizeof(bzero));
132.                         connfd[i].fd = -1; // 将connfd[i]置为-1,表示该位置可用
133.                     }
134.                     else
135.                     {
136.                         printf("[%d]:[%s]\n", n, buff);
137.                         for (i = 0; i < n; i++)
138.                         {
139.                             buff[i] = toupper(buff[i]);
140.                         }
142.                         Write(sockfd, buff, n);
143.                     }
145.                     if (--nready <= 0)
146.                     {
147.                         break; // 注意这里是break,而不是continue, 应该是从最外层的while继续循环
148.                     }
149.                 }
150.             }
151.         }
152.     }
154.     close(sfd);
156.     return 0;
157. }

复制代码

保举一个零声教育学习教程，个人以为老师讲得不错，分享给大家：[Linux，Nginx，ZeroMQ，MySQL，Redis，
fastdfs，MongoDB，ZK，流媒体，CDN，P2P，K8S，Docker，
TCP/IP，协程，DPDK等技术内容，点击立即学习:链接

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。