首页 > 代码库 > Linux下I/O复用 Select与Poll
Linux下I/O复用 Select与Poll
Select
#include <sys/time.h>
#include <sys/types.h>
#include <sys/unistd.h>
int select (int n, fd_set readfds, fd_set writefds, fd_set exceptfds, struct timeval timeout);
FD_CLR(int fd, fd_set *set);
FD_ISSET(int fd, fd_set *set);
FD_SET(int fd, fd_set *set);
FD_ZERO(fd_set *set);
调用select 将受到阻塞,必须等到指定的文件描述符就绪可进行I/O,或者等到一个指定的事件限额过去。
此调用所监视的文件描述符分为三种事件类型,每个等待不同的事件。readfds中的文件描述符是准备读取的;writefds 分组中的文件描述符是准备写入数据的; exceptfds 分组的中的文件描述符是用于查看是否有异常发生或紧急数据可用。这三个分组可以是NULL,在此情况下 select() 将无法监视相应的事件。
此调用返回后,每个分组只会包含就绪可进行I/O的文件描述符。
第一个参数 n 等于任何分组中最高编号的文件描述符的值加1。
timeout 参数是一个指向 timeval 结构的指针,该结构的定义如下:
struct timeval {
long tv_sec;
seconds long tv_usec;
microseconds
};
如果此参数不是 NULL,则 select() 调用将在 tv_sec 秒与 tv_usec 微秒之后返回,即使尚无任何文件描述符就绪可进行I/O。如果 timeout 的这两个值都被设为零,则此调用会立即返回,报告此调用进行的时候是否有任何事件等待处理,但是不会等待其后的任何事件。
fd_set writefds;
FD_ZERO(&writefds);
FD_SET(fd,&writefds); //将fd加入分组
FD_CLR(fd,&writedfds); //从分组移除fd
FD_ISSET(fd,&writefds); //测试特定文件描述符是否已就绪
1 #include <stdio.h> 2 #include <sys/time.h> 3 #include <sys/types.h> 4 #include <unistd.h> 5 6 #define TIMEOUT 5 7 #define BUF_LEN 1024 8 9 int main() 10 { 11 struct timeval tv; 12 fd_set readfds; 13 14 FD_ZERO(&readfds); 15 FD_SET(STDIN_FILENO, &readfds); 16 17 tv.tv_sec = TIMEOUT; 18 tv.tv_usec = 0; 19 20 ret = select (STDIN_FILENO+1, &readfds, NULL,NULL,&tv); 21 22 if (FD_ISSET (STDIN_FILENO, &readfds)) { 23 char buf[BUF_LEN+1]; 24 int len = read(STDIN_FILENO, buf, BUF_LEN); 25 26 buf[len] = 0; 27 printf("read:%s\n",buf); 28 } 29 }
poll
#include <sys/poll.h>
int poll (struct pollfd *fds, unsigned int nfds, int timeout);
struct pollfd {
int fd; //文件描述符
short events; //所要查看的事件
short revents; //返回所目击事件
};
每个 pllfd 结构可用于制定一个要查看的文件描述符。每个pollfd结构的 events 字段是该文件描述符所要查看事件的位掩码,用户可以设定此字段。而 revents 字段则是该文件描述符所目击事件的位掩码,内核会在返回时设定此字段。
有效事件包括:
POLLIN 有数据可供读取
OLLRDNORM 有一般数据可供读取
POLLRDBAND 有优先数据可供读取
POLLPRI 有紧急数据可供读取
POLLOUT 写入操作将不受阻塞
POLLWRNORM 写入一般数据将不受组阻塞
POLLWRBAND 写入优先数据将不受阻塞
POLLMSG 有SIGPOLL消息可用
POLLER 所指定的文件描述符发生错误
POLLHUP 所指定的文件描述符发生挂起事件
POLLNVAL 所指定的文件描述符无效
例子:
1 #include <stdio.h> 2 #include <unistd.h> 3 #include <sys/poll.h> 4 5 #define TIMEOUT 5 6 7 int main() 8 { 9 struct pollfd fds[2]; 10 int ret; 11 12 fds[0].fd = STDIN_FILENO; 13 fds[0].events = POLLIN; 14 15 fds[1].fd = STDOUT_FILENO; 16 fds[1].events = POLLOUT; 17 18 int a; 19 //scanf("%d",&a); 20 ret = poll(fds,2,TIMEOUT*1000); 21 22 if (!ret) { 23 perror("poll"); 24 return 1; 25 } 26 27 if (!ret) { 28 printf("%d seconds elapsed. \n", TIMEOUT); 29 return 0; 30 } 31 32 if (fds[0].revents & POLLIN) 33 printf("stdin is readable\n"); 34 35 if (fds[1].revents & POLLOUT) 36 printf("stdout is writable\n"); 37 scanf("%d",&a); 38 return 0; 39 }