在前面的文章中介绍了五种 I/O 模型《I/O 模型》,这里介绍 I/O 模型中 I/O 多路复用在 TCP 套接字编程中的使用。在 I/O 多路复用中主要是 select 和 poll 函数的使用。
select 函数
该函数允许进程指示内核等待多个事件中的任何一个发生,并只在一个或多个事件发生或超过指定时间后才被唤醒。进程调用 select 函数是告知内核,进程对哪些描述符(读、写或异常)感兴趣以及等待的时间。
/* IO多路复用 */
/*
* 函数功能:
* 返回值:准备就绪的描述符数,若超时则返回0,出错则返回-1;
* 函数原型:
*/
#include <sys/select.h>
int select(int maxfdpl, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, const struct timeval *tvptr);
/*
* 说明:
* 参数maxfdpl是“最大描述符加1”,即指定待测试的描述符个数;
* 参数readfds、writefds、exceptfds是指向描述符集的指针,即让内核测试读、写或异常条件的描述符;
* 时间参数有三种取值:
* tvptr == NULL;
* 永远等待;若捕获到信号则中断此无限期等待;当所指定的描述符中的一个已准备好或捕获到信号则返回;
* 若捕获到信号,则select返回-1,errno设置为EINTR;
*
* tvptr->tv_sec == 0 && tvptr->tv_usec == 0;
* 完全不等待;测试所有描述符并立即返回,这是得到多个描述符的状态而不阻塞select函数的轮回方法;
*
* tvptr->sec != 0 || tvptr->usec != 0;
* 等待指定的秒数和微妙数;当指定的描述符已准备好,或超过指定的时间立即返回;
* 若超过指定的时间还没有描述符准备好,则返回0;
*
* tvptr的结构如下:
*/
struct timeval
{
long tv_sec; /* seconds */
long tv_usec; /* and microseconds */
};
我们可以通过以下函数对 fd_set 数据结构进行处理。声明了一个描述符集后,必须使用 FD_ZERO 清空其所有位达到初始化,然后才可以设置各个位;从 select 返回时,使用 FD_ISSET 测试该集中的一个给定位是否仍旧设置;
#include <sys/select.h>
int FD_ISSET(int fd, fd_set *fdset); //测试描述符fd是否在描述符