Linux中使用select函數

select模型在win32API編程中也很常見，而且和linux中的select函數遵循同樣的berkeley標准，所以函數原型相同。select函數原型：

Ubuntu:

int select( int maxfdp, fd_set * readfds, fd_set * writefds, fd_set * errorfds, struct timeval * timeout);

Win32:

int select( intnfds, fd_set*readfds, fd_set*writefds, fd_set*exceptfds, const struct timeval*timeout);

可以看到是一樣的。不同點是，win32API中的select函數忽略第一個參數。而linux要求第一個參數設置成(參數2~4某個數組中文件描述符的最大值+1)。

下面是ubuntu10.0中使用select函數的一個例子，流程為：

1. 創建子進程
2. 子進程通過管道定時發送數據給父進程
3. 父進程通過select同時監聽管道數據和用戶輸入
4. 父進程等待子進程退出

Ubuntu10.0.4：

源代碼文件select.cpp：

#include <stdio.h>
#include <unistd.h>
#include <sys/wait.h>// waitpid
#include <sys/types.h>// waitpid
#include <string.h>// strlen
/*
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]);
fd_set readfdset;// file descriptor set
for( i = 0; i < TIMES; ++ i){
FD_ZERO( & readfdset);
FD_SET( pipefds[0], & readfdset);// add read file descriptor
FD_SET( 0, & readfdset);// add standard input
select( pipefds[0]+1, & readfdset, NULL, NULL, NULL);
if( FD_ISSET( pipefds[0], & readfdset)){
buf[ read( pipefds[0], buf, 256)] = '\0';
printf( "Receive:%s\n", buf);
}
if( FD_ISSET( 0, & readfdset)){
buf[ read( 0, buf, 256)] = '\0';
printf( "Print:%s\n", buf);
}
}
int status;
wait( & status);
}
}
return 0;
}

還需要一個Makefile：

select: select.cpp
g++ $< -o $@

然後在命令行中鍵入：

make select
./select

運行截圖：

注意點

父進程必須等待子進程結束。如果父進程先於子進程結束，子進程成為死進程。

每次調用select後參數2都會被改變，下次調用select前必須重新設置。