Linux中使用poll函数

poll函数用法可以man一下。这里提供一个可以运行的示例。

程序流程：

1. 父进程启动并创建子进程
2. 子进程通过管道发送数据给父进程
3. 父进程同时监听管道数据和shell输入，阻塞500毫秒发现没有数据就打印一个"Testing...."
4. 父进程等待子进程结束
5. 子进程结束，父进程结束

Ubuntu10.04:

poll.cpp源代码：

#include <stdio.h>
#include <unistd.h>
#include <sys/wait.h>// waitpid
#include <sys/types.h>// waitpid
#include <string.h>// strlen
#include <poll.h>// poll
/*
comment:
pipe is used between two processes on the same computer.
*/
#define TIMES 50
int main(){
 int pipefds[2];
 if( -1 == pipe( pipefds)){
  printf( "Error when create pipes\n");
 }else{
  int i;
  pid_t pid = fork();
  if( 0 == pid){// child
   printf( "child running\n");
   close( pipefds[0]);
   for( i = 0; i < TIMES; ++ i){
    write( pipefds[1], "iamagoodguy", strlen( "iamagoodguy"));
    sleep( 1);
   }
  }else{
   printf( "parent running\n");
   char buf[256];
   close( pipefds[1]);
   struct pollfd pf[2];// key structure
   pf[0].fd = 0;// console input
   pf[0].events = POLLIN;// wait for bytes input
   pf[1].fd = pipefds[0];// pipe input
   pf[1].events = POLLIN;// wait for bytes input
   for( i = 0; i < TIMES; ++ i){
    poll( pf, 2, 500);// wait for only 500 ms
    printf( "Testing...\n");
    if( pf[1].revents & POLLIN){
     buf[ read( pipefds[0], buf, 256)] = '\0';
     printf( "Receive:%s\n", buf);
    }
    if( pf[0].revents & POLLIN){
     buf[ read( 0, buf, 256)] = '\0';
     printf( "Print:%s\n", buf);
    }
   }
   int status;
   wait( & status);
  }
 }
 return 0;
}

Makefile（只有poll部分是这个程序使用的）：

COMPILE = g++ $< -o $@
mapfile: mapfile.cpp
 $(COMPILE)
pipe: pipe.cpp
 $(COMPILE)
select: select.cpp
 $(COMPILE)
poll: poll.cpp
 $(COMPILE)

然后进行编译：

make poll
./poll