poll()函數:
這個函數是某些Unix系統提供的用于執行與select()函數同等功能的函數,下面是這個函數的聲明:
#include <poll.h>int poll(struct pollfd fds[], nfds_t nfds, int timeout);
參數說明:
fds:是一個struct pollfd結構類型的數組,用于存放需要檢測其狀態的Socket描述符;每當調用這個函數之后,系統不會清空這個數組,操作起來比較方便;特別是對于socket連接比較多的情況下,在一定程度上可以提高處理的效率;這一點與select()函數不同,調用select()函數之后,select()函數會清空它所檢測的socket描述符集合,導致每次調用select()之前都必須把socket描述符重新加入到待檢測的集合中;因此,select()函數適合于只檢測一個socket描述符的情況,而poll()函數適合于大量socket描述符的情況;
nfds:nfds_t類型的參數,用于標記數組fds中的結構體元素的總數量;
timeout:是poll函數調用阻塞的時間,單位:毫秒;
返回值:
>0:數組fds中準備好讀、寫或出錯狀態的那些socket描述符的總數量;
==0:數組fds中沒有任何socket描述符準備好讀、寫,或出錯;此時poll超時,超時時間是timeout毫秒;換句話說,如果所檢測的socket描述符上沒有任何事件發生的話,那么poll()函數會阻塞timeout所指定的毫秒時間長度之后返回,如果timeout==0,那么poll() 函數立即返回而不阻塞,如果timeout==INFTIM,那么poll() 函數會一直阻塞下去,直到所檢測的socket描述符上的感興趣的事件發生是才返回,如果感興趣的事件永遠不發生,那么poll()就會永遠阻塞下去;
-1: poll函數調用失敗,同時會自動設置全局變量errno;
pollfd的結構:
struct pollfd {int fd; /*文件描述符*/short events; /* 等待的需要測試事件 */short revents; /* 實際發生了的事件,也就是返回結果 */};
使用poll來實現TCP回射服務器
例子代碼源自UNIX網絡編程tcpcliserv/tcpcliservpoll01.c
#include <unistd.h>#include <sys/types.h> /* basic system data types */#include <sys/socket.h> /* basic socket definitions */#include <netinet/in.h> /* sockaddr_in{} and other Internet defns */#include <arpa/inet.h> /* inet(3) functions */#include <stdlib.h>#include <errno.h>#include <stdio.h>#include <string.h>#include <poll.h> /* poll function */#include <limits.h>#define MAXLINE 10240#ifndef OPEN_MAX#define OPEN_MAX 40960#endif#define NOTDEFintmain(int argc, char **argv){inti, maxi, listenfd, connfd, sockfd;intnready;ssize_tn;charbuf[MAXLINE];socklen_tclilen;struct pollfdclient[OPEN_MAX];struct sockaddr_incliaddr, servaddr;listenfd = socket(AF_INET, SOCK_STREAM, 0);//監聽fdbzero(&servaddr, sizeof(servaddr));servaddr.sin_family = AF_INET;servaddr.sin_addr.s_addr = htonl(INADDR_ANY);servaddr.sin_port = htons(6888);bind(listenfd, (struct sockaddr *) &servaddr, sizeof(servaddr));listen(listenfd, 1024);client[0].fd = listenfd;client[0].events = POLLRDNORM;for (i = 1; i < OPEN_MAX; i++)client[i].fd = -1;/* -1 indicates available entry */maxi = 0;/* max index into client[] array *//* end fig01 *//* include fig02 */for ( ; ; ) {nready = poll(client, maxi+1, -1);//maxi表示client數組大小if (client[0].revents & POLLRDNORM) {/**new client connection*每次有新連接都會執行這個if語句,然后將新添加的鏈接調用accept來接受鏈接*/clilen = sizeof(cliaddr);connfd = accept(listenfd, (struct sockaddr *) &cliaddr, &clilen);//accept函數返回了一個socketfd,#ifdefNOTDEFPRintf("new client: %s %d/n", (struct sockaddr *) &cliaddr, clilen);#endiffor (i = 1; i < OPEN_MAX; i++)//監視connfd是否可讀、可寫if (client[i].fd < 0) {client[i].fd = connfd;/* save descriptor */break;}if (i == OPEN_MAX)//err_quit("too many clients"); printf("too many clients/n");client[i].events = POLLRDNORM;//檢測connfd是否可讀if (i > maxi)maxi = i;/* max index in client[] array */if (--nready <= 0)/*如果除了listen的client[0]被激活,其他事件沒有沒有被激活則nready是1*自減1后,為0,表示此次處理poll結束。繼續下次監視。*/continue;/* no more readable descriptors */}for (i = 1; i <= maxi; i++) {/* 第0個元素是處理listen的,處理其余accept的所有可讀的connfd */if ( (sockfd = client[i].fd) < 0)//無效的fdcontinue;if (client[i].revents & (POLLRDNORM | POLLERR)) {//處理可讀的connfdif ( (n = read(sockfd, buf, MAXLINE)) < 0) {if (errno == ECONNRESET) {/*4connection reset by client */#ifdefNOTDEFprintf("client[%d] aborted connection/n", i);#endifclose(sockfd);client[i].fd = -1;} elseprintf("read error");//err_sys("read error");} else if (n == 0) {/*4connection closed by client */#ifdefNOTDEFprintf("client[%d] closed connection/n", i);#endifclose(sockfd);client[i].fd = -1;} else//writen(sockfd, buf, n); write(sockfd, buf, n);if (--nready <= 0)break;/* no more readable descriptors */}}}}/* end fig02 */
新聞熱點
疑難解答