歡迎來到Linux教程網
Linux教程網
Linux教程網
Linux教程網
您现在的位置: Linux教程網 >> UnixLinux >  >> Linux編程 >> Linux編程

Linux poll函數深入理解

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);
}

Copyright © Linux教程網 All Rights Reserved