https://blog.csdn.net/men_wen/article/details/53456474
Linux網絡編程—I/O復用模型之poll
1.函數poll
poll系統調用和select類似,也是在指定時間內輪詢一定數量的文件描述符,以測試其中是否有就緒者。
#include <poll.h>int poll(struct pollfd *fds, nfds_t nfds, int timeout);
//返回值:若成功,返回就緒的描述符個數,若超時則為0,若出錯則為-1
- fds是指向一個pollfd結構第一個元素的指針。每一個元素都是pollfd結構類型,用于測試某個給定描述符fd的條件
struct pollfd {int fd; /* file descriptor */short events; /* requested events */short revents; /* returned events */
};
要測試的條件在events中指定,函數相應的revents成員返回該描述符的狀態,一個為調用值,一個為返回結果,這與select的傳入傳出參數不同。?
events和revents可以設置以下常量的一個或多個的按位或結果:
POLLIN //普通或帶外優先數據可讀,即POLLRDNORM | POLLRDBAND
POLLRDNORM //數據可讀
POLLRDBAND //優先級帶數據可讀
POLLPRI //高優先級可讀數據
POLLOUT //普通或帶外數據可寫
POLLWRNORM //數據可寫
POLLWRBAND //優先級帶數據可寫
//下面三個revents不包含
POLLERR //發生錯誤
POLLHUP //發生掛起
POLLNVAL //描述字不是一個打開的文件
- out指定poll函數返回前等待多長時間,它是一個指定應等待毫秒數的正值。
|timeout值|說明|?
|:-:|:-:|?
|-1|永遠等待|?
|0|立即返回,不阻塞進程|?
|>0|等待指定數目的毫秒數|
- nfds 是監控數組中有多少文件描述符需要被監控
如果我們不關心某個特定的描述符,那么可以把與它對應的pollfd結構的fd成員設置為一個負值。poll函數會忽略這樣的pollfd結構的events成員,返回時將其revents成員的值置為0。
select模型最多可以創建一個1024個文件描述符大小的集合,這是因為受到fd_set的固定數據類型的限制,而poll則沒有這樣的問題,因為分配一個pollfd結構的數組并把數組中的元素通知內核稱為調用者的責任。但是poll模型任然使用的是輪詢模型,效率比較低下。
2. poll模型實現
2.1 服務器端
#include "wrap.h"
#include <poll.h>
#include <limits.h>
#include <sys/stropts.h>#define MAXLINE 1024
#define INFTIM -1
#define OPEN_MAX 3000
int main(int argc, char *argv[])
{int i, maxi, listenfd, connfd, sockfd;int nready;ssize_t n;char buf[MAXLINE];socklen_t clilen;struct pollfd client[OPEN_MAX];struct sockaddr_in clientaddr;listenfd = start_ser(argv[1], argv[2]);//監聽文件描述符 該函數參考selectclient[0].fd = listenfd;//將監聽文件描述符加入pollfd數組中client[0].events = POLLIN;//設置為輸入可讀for(i = 1; i < OPEN_MAX; i++){//初始化結構體數組client[i].fd = -1;}maxi = 0;while(1){nready = poll(client, maxi+1, INFTIM); //阻塞等待注冊時間發生if(client[0].revents & POLLIN){//當監聽文件描述符響應有新連接請求clilen = sizeof(clientaddr);connfd = Accept(listenfd, (struct sockaddr *)&clientaddr, &clilen);//接受新客戶for(i = 1; i < OPEN_MAX; i++){if(client[i].fd < 0){client[i].fd = connfd; //把新客戶的文件描述符加入數組break;}}if(i == OPEN_MAX){ //連接新客戶上限Close(connfd);perr_exit("too many clients");}client[i].events = POLLIN; //新客戶端的處理事件if(i > maxi){maxi = i; //更新最大值}if(--nready == 0){ //沒有可讀的描述符跳過本次循環continue;}}for(i = 1; i <= maxi; i++){ //請求所有客戶端的數據if((sockfd = client[i].fd) < 0){continue;}if(client[i].revents & POLLIN){ //POLLIN事件發生memset(buf, '\0', MAXLINE);if((n = Read(sockfd, buf, MAXLINE-1)) < 0){if(errno == ECONNRESET){ //當客戶端發送resetClose(sockfd);client[i].fd = -1;}else{perr_exit("read err");}}else if(n == 0){Close(sockfd);client[i].fd = -1;}else{printf("client : %s\n", buf);}if(--nready == 0){ //沒有可讀的描述符跳過本次循環break;}}}}Close(connfd);Close(listenfd);return 0;
}
2.2客戶端
#include "wrap.h"int main(int argc, char *argv[])
{struct sockaddr_in serveraddr;int connfd;connfd = Socket(AF_INET, SOCK_STREAM, 0);bzero(&serveraddr, sizeof(serveraddr));serveraddr.sin_family = AF_INET;serveraddr.sin_port = htons(atoi(argv[2]));inet_pton(AF_INET, argv[1], &serveraddr.sin_addr);Connect(connfd, (struct sockaddr *)&serveraddr, sizeof(serveraddr));char buf[1024];while((fgets(buf, 1024, stdin)) != NULL){Write(connfd, buf, strlen(buf));}Close(connfd);return 0;
}