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

運行結果如下：