Linux 下网络 IO 的多路复用
2019-10-20
关键字:select 与 poll
在 Linux 系统下,IO 总共可以分为以下四种:
1、阻塞 IO;
2、非阻塞 IO;
3、IO多路复用;
允许同时对多个 IO 进行控制。
4、信号驱动 IO;
一种异步通信模型。前面三种 IO 都是同步型的,唯这一种是异步型的。
阻塞 IO
所谓阻塞 IO 就是在调用相关函数时,程序的运行指针会暂停往下执行,直至这个 IO 操作有结果返回为止。简单来说就是我发起一个 IO 操作请求,你有数据就返回给我,没数据我就等你到有数据为止。
阻塞型 IO 是最普遍使用的 IO 模式,大部分的程序都是使用这一模式的。套接字在缺省情况下使用的就是阻塞 IO 模式。
常见的阻塞模式函数有:read, recv, recvfrom, write, send, accept, connect。 这里需要强调,sendto 函数是非阻塞型的。
非阻塞 IO
非阻塞型 IO 比较干脆。当它发起一个 IO 请求时,若 IO 有数据就返回结果,若无数据则程序指针就继续往下执行了,不会死等的。
我们可以通过两个函数来切换阻塞型与非阻塞型 IO。
1、fcntl()
假设我们需要将阻塞型 IO 设置为非阻塞型 IO。
int flag;
flag = fcntl(sockfd, F_GETFL, 0);
flag |= O_NONBLOCK;
fcntl(sockfd, F_SETFL, flag);
这样一来一去,就将 IO 形式给更改过来了。
2、ioctl()
int b_on = 1;
ioctl(sock_fd, FIONBIO, &b_on);
IO 多路复用
IO 多路利用其实就是 C 编程中的 select/poll 模型和 epoll 模型。
Select 函数:
select 函数的作用就是将原本需要单独分别监听阻塞资源统一交由 select 来监听。每当被 select 所监听的资源有数据波动时,select 会采用轮询的方式去找出哪个阻塞资源有数据过来了。并将这些有数据波动的资源保存起来,然后中断 select 函数的阻塞态,执行后面的代码。select() 函数的原型如下:
int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout);
这个函数是一个阻塞函数。当它返回值 0 时表示等待超时。返回值 -1 时表示等待失败。返回值大于 0 时表示成功监测到信号。
参数 nfds 是指 fd 的个数。由于在 Linux 中,文件描述符总是顺序递增增长的,因此这个参数也可以填入最大文件描述符号的值再加一。即 maxfd + 1。
参数 readfds 是读集合。
参数 writefds 是写集合。
参数 exceptfds 是异常集合。
参数 timeout 是超时时长。
fd_set 是系统定义的用于存放文件描述符的信息的结构体,可以通过以下几个函数来设定:
1、void FD_ZERO(fd_set *fdset);
对指定集合清零。
2、void FD_SET(int fd, fd_set *fdset);
将 fd 加入到 fdset 集合中去。
3、void FD_CLR(int fd, fd_set *fdset);
从集合中清除指定的 fd。
4、int FD_ISSET(int fd, fd_set *fdset);
判断指定的 fd 是否可以读写。
对于 select 函数中,writefds 与 exceptfds 通常都填 NULL。同时,对于 select() 之前与之后的 fd_set 集合,它所包含的内容是不一样的。在 select() 之前,我们会往 fd_set 集合中填入所有我们想监听的资源 fd,但在 select() 之后,对应的 fd_set 集合中的数据就已经发生了变化。通常可以理解为只有有数据波动的资源集合才会出现在 select() 以后的代码中。
以下是一个运用 select() 来复用 TCP 编程的伪代码:
poll 函数:
int poll(struct pollfd *fds, nfds_t nfds, int timeout);
函数执行成功时返回大于0的值,失败时返回 EOF,超时返回值0。
参数 pollfd 是一个系统结体体,它的原型如下:
struct pollfd {
int fd; //要监听的文件描述符号。
short events; // 请求的事件。
short revents; // 返回的事件。
};
参数 nfds 与 timeout 就不再赘述了。
以下是一个示例程序,程序来自于 https://www.jianshu.com/p/6a6845464770
epoll 函数:
略。
信号驱动 IO
略。