Linux C——poll函数实现多路复用

最新推荐文章于 2023-07-31 16:36:19 发布

Eleanor_gao

最新推荐文章于 2023-07-31 16:36:19 发布

阅读量2k

点赞数

分类专栏： Linux系统基础

Linux系统基础专栏收录该内容

41 篇文章 1 订阅

订阅专栏

多路复用

多路复用一般用于I/O操作可能会被阻塞的情况，对可能会有阻塞的I/O的管道、网路进行编程。

采用管道函数创建有名管道，使用select函数替代使用poll函数实现多路复用：创建两个有名管道，获取3个文件描述符（2个管道1个标准输入），然后初始化读文件描述符，select监视文件描述符的文件读写，管道1输出到屏幕上，管道2输出到屏幕上，标准输入‘Q’来进行判读是否退出。

select参数函数介绍：

（1）第一个参数maxfdp1指定待测试的描述字个数，它的值是待测试的最大描述字加1（因此把该参数命名为maxfdp1），描述字0、1、2...maxfdp1-1均将被测试。

因为文件描述符是从0开始的。

（2）中间的三个参数readset、writeset和exceptset指定我们要让内核测试读、写和异常条件的描述字。如果对某一个的条件不感兴趣，就可以把它设为空指针。struct fd_set可以理解为一个集合，这个集合中存放的是文件描述符，可通过以下四个宏进行设置：

void FD_ZERO(fd_set *fdset); //清空集合

void FD_SET(int fd, fd_set *fdset); //将一个给定的文件描述符加入集合之中

void FD_CLR(int fd, fd_set *fdset); //将一个给定的文件描述符从集合中删除

int FD_ISSET(int fd, fd_set *fdset); // 检查集合中指定的文件描述符是否可以读写

（3）timeout告知内核等待所指定描述字中的任何一个就绪可花多少时间。其timeval结构用于指定这段时间的秒数和微秒数。

struct timeval{

long tv_sec; //seconds

long tv_usec; //microseconds

};

这个参数有三种可能：

（1）永远等待下去：仅在有一个描述字准备好I/O时才返回。为此，把该参数设置为空指针NULL。

（2）等待一段固定时间：在有一个描述字准备好I/O时返回，但是不超过由该参数所指向的timeval结构中指定的秒数和微秒数。

（3）根本不等待：检查描述字后立即返回，这称为轮询。为此，该参数必须指向一个timeval结构，而且其中的定时器值必须为0。

[html]view plaincopy 
   
 <span style="font-family:FangSong_GB2312;font-size:18px;"><strong>#include <fcntl.h>  
 #include <stdio.h>  
 #include <stdlib.h>  
 #include <unistd.h>  
 #include <string.h>  
 #include <time.h>  
 #include <errno.h>  
    
 #define FIFO1 "in1"  
 #define FIFO2 "in2"  
 #define MAX_BUFFER_SIZE 1024  //缓冲区大小  
 #define IN_FILES 3               //多路复用输入文件数目  
 #define TIME_DELAY 60           //超时秒数  
    
 #define MAX(a,b) ((a > b) ? (a) : (b))  
    
 int main(void)  
 {  
     int fds[IN_FILES];      //管道描述符  
     int i;  
     int res;  
     int real_read;  
     int maxfd;  
    
     char buf[MAX_BUFFER_SIZE];  
    
     struct timeval tv;  
    
     fd_set inset;  
     fd_set tmp_inset;           //文件描述符集  
    
     fds[0] = 0;          //终端的文件描述符  
    
     if(access(FIFO1,F_OK) == -1)         //创建两个有名管道  
     {  
         if((mkfifo(FIFO1,0666) < 0) && (errno != EEXIST))  
         {  
             printf("Cannot creat fifo1 file!\n");  
    
             exit(1);  
         }  
     }  
    
     if(access(FIFO2,F_OK) == -1)  
     {  
         if((mkfifo(FIFO2,0666) < 0) && (errno != EEXIST))  
         {  
             printf("Cannot creat fifo2 file\n");  
    
             exit(1);  
         }  
     }  
    
     if((fds[1] = open(FIFO1,O_RDONLY | O_NONBLOCK)) < 0)   //以只读非阻塞的方式打开两个管道文件  
     {  
         printf("open in1 error!\n");  
    
         return 1;  
     }  
    
     if((fds[2] = open(FIFO2,O_RDONLY | O_NONBLOCK)) < 0)  
     {  
         printf("open in2 error!\n");  
    
         return 1;  
     }  
    
     maxfd = MAX(MAX(fds[0],fds[1]),fds[2]);  //取出两个文件描述符中的较大者  
    
     //初始化读集inset,并在读文件描述符集中加入相应的描述集  
     FD_ZERO(&inset);     //将insert清零，使集合中不含任何fd  
     for(i = 0; i < IN_FILES; i++)  
     {  //将fds[i]加入inset集合  
         FD_SET(fds[i],&inset);  
     }  
    
     FD_SET(0,&inset);  
    
     tv.tv_sec = TIME_DELAY;   //设置超时60s  
     tv.tv_usec = 0;  
 //循环测试该文件描述符是否准备就绪，并调用selelct()函数对相关文件描述符做相应的操作  
 while(FD_ISSET(fds[0],&inset) || FD_ISSET(fds[1],&inset) || FD_ISSET(fds[2],&inset))  
     {    //文件描述符集的备份，以免每次都进行初始化  
         tmp_inset = inset;  
         res = select(maxfd+1,&tmp_inset,NULL,NULL,&tv);  
    
         switch(res)  
         {  
             case -1:  
                 {  
                     printf("Select error!\n");  
    
                     return 1;  
                 }  
                 break;  
    
             case 0:  
                 {  
                     printf("Time out!\n");  
    
                     return 1;  
                 }  
                 break;  
             default:  
                 {  
                     for(i = 0; i < IN_FILES; i++)  
                     {  
                         if(FD_ISSET(fds[i],&tmp_inset))  
                         {  
                             memset(buf,0,MAX_BUFFER_SIZE);  
    
                             real_read = read(fds[i],buf,MAX_BUFFER_SIZE);  
    
                             if(real_read < 0)  
                             {  
                                 if(errno != EAGAIN)  
                                 {  
                                     return 1;  
                                 }  
                             }  
                             else if(!real_read)  //已到达文件尾  
                             {  
                                 close(fds[i]);  
    
                                 FD_CLR(fds[i],&inset);  
                             }  
                             else  
                             {  
                                 if(i == 0)  
                                 {   //主程序终端控制  
                                     if((buf[0] == 'q') || (buf[0] == 'Q'))  
                                     {  
                                         return 1;  
                                     }  
                                 }  
                                 else  
                                 {   //显示管道输入字符串  
                                     buf[real_read] = '\0';  
    
                                     printf("%s",buf);  
                                 }  
                             }  
                         }  
                     }  
                 }  
              break;  
         }  
     }  
    
     return 0;  
 }</strong></span>  