poll函數與select函數差不多
函數原型:
#include <poll.h> int poll(struct pollfd fd[], nfds_t nfds, int timeout);
struct pollfd的結構如下:
struct pollfd{ int fd; // 文件描述符 short event;// 請求的事件 short revent;// 返回的事件 }
每個pollfd結構體指定了一個被監視的文件描述符。第一個參數是一個數組,即poll函數可以監視多個文件描述符。每個結構體的events是監視該文件描述符的事件掩碼,由用戶來設置。revents是文件描述符的操作結果事件,內核在調用返回時設置。events中請求的任何事件都可能在revents中返回。合法的事件如下:
後三個只能作為描述字的返回結果存儲在revents中,而不能作為測試條件用於events中。
這些事件在events域中無意義,因為它們在合適的時候總是會從revents中返回。使用poll()和select()不一樣,你不需要顯式地請求異常情況報告。
POLLIN | POLLPRI等價於select()的讀事件,POLLOUT |POLLWRBAND等價於select()的寫事件。POLLIN等價於POLLRDNORM |POLLRDBAND,而POLLOUT則等價於POLLWRNORM。
例如,要同時監視一個文件描述符是否可讀和可寫,我們可以設置 events為POLLIN |POLLOUT。在poll返回時,我們可以檢查revents中的標志,對應於文件描述符請求的events結構體。如果POLLIN事件被設置,則文件描述符可以被讀取而不阻塞。如果POLLOUT被設置,則文件描述符可以寫入而不導致阻塞。這些標志並不是互斥的:它們可能被同時設置,表示這個文件描述符的讀取和寫入操作都會正常返回而不阻塞。
第二個參數nfds:要監視的描述符的數目。
timeout參數指定等待的毫秒數,無論I/O是否准備好,poll都會返回。timeout指定為負數值表示無限超時;timeout為0指示poll調用立即返回並列出准備好I/O的文件描述符,但並不等待其它的事件。這種情況下,poll()就像它的名字那樣,一旦選舉出來,立即返回。
成功時,poll()返回結構體中revents域不為0的文件描述符個數;如果在超時前沒有任何事件發生,poll()返回0;失敗時,poll()返回-1,並設置errno為下列值之一:
EBADF:一個或多個結構體中指定的文件描述符無效。
EFAULT:fds指針指向的地址超出進程的地址空間。
EINTR:請求的事件之前產生一個信號,調用可以重新發起。
EINVAL:nfds參數超出PLIMIT_NOFILE值。
ENOMEM:可用內存不足,無法完成請求。
demo:
代碼與上一篇文章中"利用select實現IO多路復用TCP服務端"中代碼差不多
#include <fcntl.h> #include <stdio.h> #include <unistd.h> #include <stdlib.h> #include <string.h> #include <time.h> #include <errno.h> #include <poll.h> #define MAX_BUFFER_SIZE 1024 #define IN_FILES 3 #define TIME_DELAY 60*5 #define MAX(a,b) ((a>b)?(a):(b)) int main(int argc ,char **argv) { struct pollfd fds[IN_FILES]; char buf[MAX_BUFFER_SIZE]; int i,res,real_read, maxfd; fds[0].fd = 0; if((fds[1].fd=open("data1",O_RDONLY|O_NONBLOCK)) < 0) { fprintf(stderr,"open data1 error:%s",strerror(errno)); return 1; } if((fds[2].fd=open("data2",O_RDONLY|O_NONBLOCK)) < 0) { fprintf(stderr,"open data2 error:%s",strerror(errno)); return 1; } for (i = 0; i < IN_FILES; i++) { fds[i].events = POLLIN; } for(i=0;i<IN_FILES;i++) { fds[i].events = POLLIN; } while(fds[0].events || fds[1].events || fds[2].events) { if (poll(fds, IN_FILES, TIME_DELAY) <= 0) { printf("Poll error\n"); return 1; } for (i = 0; i< IN_FILES; i++) { if (fds[i].revents) { memset(buf, 0, MAX_BUFFER_SIZE); real_read = read(fds[i].fd, buf, MAX_BUFFER_SIZE); if (real_read < 0) { if (errno != EAGAIN) { return 1; } } else if (!real_read) { close(fds[i].fd); fds[i].events = 0; } else { if (i == 0) { if ((buf[0] == 'q') || (buf[0] == 'Q')) { return 1; } } else { buf[real_read] = '\0'; printf("%s", buf); } } } } } exit(0); }