Linux 阻塞和非阻塞 IO 实验

一、阻塞和非阻塞简介这里的 “IO” 并不是 STM32 或者其他单片机的时候所说的 “GPIO”(也就是引脚)。 这里的 IO 指的是 Input/Output也就是输入 / 输出是应用程序对驱动设备的输入 / 输出操作。当应用程序对设备驱动进行操作的时候如果不能获取到设备资源那么阻塞式 IO 就会将应用程序对应的线程挂起直到设备资源可以获取为止。对于非阻塞 IO应用程序对应的线程不会挂起它要么一直轮询等待直到设备资源可以使用要么就直接放弃。阻塞式 IO 如图所示图中应用程序调用 read 函数从设备中读取数据当设备不可用或数据未准备好的时候就会进入到休眠态。等设备可用的时候就会从休眠态唤醒然后从设备中读取数据返回给应用程序。非阻塞 IO 如图所示从图可以看出应用程序使用非阻塞访问方式从设备读取数据当设备不可用或数据未准备好的时候会立即向内核返回一个错误码表示数据读取失败。应用程序会再次重新读取数据这样一直往复循环直到数据读取成功。应用程序可以使用如下所示示例代码来实现阻塞访问int fd; int data 0; fd open(/dev/xxx_dev, O_RDWR); /* 阻塞方式打开 */ ret read(fd, data, sizeof(data)); /* 读取数据 */从示例代码可以看出对于设备驱动文件的默认读取方式就是阻塞式的所以我们前面所有的例程测试 APP 都是采用阻塞 IO。如果应用程序要采用非阻塞的方式来访问驱动设备文件可以使用如下所示代码int fd; int data 0; fd open(/dev/xxx_dev, O_RDWR | O_NONBLOCK); /* 非阻塞方式打开 */ ret read(fd, data, sizeof(data)); /* 读取数据 */第 4 行使用 open 函数打开 /dev/xxx_dev 设备文件的时候添加了参数 “O_NONBLOCK”表示以非阻塞方式打开设备这样从设备中读取数据的时候就是非阻塞方式的了。二、等待队列1、等待队列头阻塞访问最大的好处就是当设备文件不可操作的时候进程可以进入休眠态这样可以将 CPU 资源让出来。但是当设备文件可以操作的时候就必须唤醒进程一般在中断函数里面完成唤醒工作。Linux 内核提供了等待队列 (wait queue) 来实现阻塞进程的唤醒工作如果我们要在驱动中使用等待队列必须创建并初始化一个等待队列头等待队列头使用结构体 wait_queue_head_t 表示wait_queue_head_t 结构体定义在文件 include/linux/wait.h 中结构体内容如下所示39 struct __wait_queue_head { 40 spinlock_t lock; 41 struct list_head task_list; 42 }; 43 typedef struct __wait_queue_head wait_queue_head_t;定义好等待队列头以后需要初始化使用 init_waitqueue_head 函数初始化等待队列头函数原型如下void init_waitqueue_head(wait_queue_head_t *q)参数 q 就是要初始化的等待队列头。也可以使用宏 DECLARE_WAIT_QUEUE_HEAD 来一次性完成等待队列头的定义的初始化。2、等待队列项等待队列头就是一个等待队列的头部每个访问设备的进程都是一个队列项当设备不可用的时候就要将这些进程对应的等待队列项添加到等待队列里面。结构体 wait_queue_t 表示等待队列项结构体内容如下struct __wait_queue { unsigned int flags; void *private; wait_queue_func_t func; struct list_head task_list; }; typedef struct __wait_queue wait_queue_t;使用宏DECLARE_WAITQUEUE定义并初始化一个等待队列项宏的内容如下DECLARE_WAITQUEUE(name, tsk)name就是等待队列项的名字tsk表示这个等待队列项属于哪个任务(进程)一般设置为current在Linux内核中current相当于一个全局变量表示当前进程。因此宏DECLARE_WAITQUEUE 就是给当前正在运行的进程创建并初始化了一个等待队列项。3、将队列项添加/移除等待队列头当设备不可访问的时候就需要将进程对应的等待队列项添加到前面创建的等待队列头中只有添加到等待队列头中以后进程才能进入休眠态。当设备可以访问以后再将进程对应的等待队列项从等待队列头中移除即可等待队列项添加 API 函数如下void add_wait_queue(wait_queue_head_t *q, wait_queue_t *wait)函数参数和返回值含义如下q等待队列项要加入的等待队列头。wait要加入的等待队列项。返回值无。等待队列项移除 API 函数如下void remove_wait_queue(wait_queue_head_t *q, wait_queue_t *wait)函数参数和返回值含义如下q要删除的等待队列项所处的等待队列头。wait要删除的等待队列项。返回值无。4、等待唤醒当设备可以使用的时候就要唤醒进入休眠态的进程唤醒可以使用如下两个函数void wake_up(wait_queue_head_t *q) void wake_up_interruptible(wait_queue_head_t *q)参数 q 就是要唤醒的等待队列头这两个函数会将这个等待队列头中的所有进程都唤醒。 wake_up 函数可以唤醒处于 TASK_INTERRUPTIBLE 和 TASK_UNINTERRUPTIBLE 状态的进程而 wake_up_interruptible 函数只能唤醒处于 TASK_INTERRUPTIBLE 状态的进程。5、等待事件除了主动唤醒以外也可以设置等待队列等待某个事件当这个事件满足以后就自动唤醒等待队列中的进程和等待事件有关的 API 函数如表所示函数描述wait_event(wq, condition)等待以 wq 为等待队列头的等待队列被唤醒前提是 condition 条件必须满足 (为真)否则一直阻塞。此函数会将进程设置为 TASK_UNINTERRUPTIBLE 状态wait_event_timeout(wq, condition, timeout)功能和 wait_event 类似但是此函数可以添加超时时间以 jiffies 为单位。此函数有返回值如果返回 0 的话表示超时时间到而且 condition 为假。为 1 的话表示 condition 为真也就是条件满足了。wait_event_interruptible(wq, condition)与 wait_event 函数类似但是此函数将进程设置为 TASK_INTERRUPTIBLE就是可以被信号打断。wait_event_interruptible_timeout(wq, condition, timeout)与 wait_event_timeout 函数类似此函数也将进程设置为 TASK_INTERRUPTIBLE可以被信号打断。三、轮询如果用户应用程序以非阻塞的方式访问设备设备驱动程序就要提供非阻塞的处理方式也就是轮询。poll、epoll 和 select 可以用于处理轮询应用程序通过 select、epoll 或 poll 函数来查询设备是否可以操作如果可以操作的话就从设备读取或者向设备写入数据。当应用程序调用 select、epoll 或 poll 函数的时候设备驱动程序中的 poll 函数就会执行因此需要在设备驱动程序中编写 poll 函数。我们先来看一下应用程序中使用的 select、poll 和 epoll 这三个函数。1、select函数select 函数原型如下int select(int nfds, fd_set *readfds, fd_set *writefds, fd_set *exceptfds, struct timeval *timeout)函数参数和返回值含义如下nfds所要监视的这三类文件描述集合中最大文件描述符加 1。readfds、writefds 和 exceptfds这三个指针指向描述符集合这三个参数指明了关心哪些描述符、需要满足哪些条件等等这三个参数都是 fd_set 类型的fd_set 类型变量的每一个位都代表了一个文件描述符。readfds 用于监视指定描述符集的读变化也就是监视这些文件是否可以读取只要这些集合里面有一个文件可以读取那么 select 就会返回一个大于 0 的值表示文件可以读取。如果没有文件可以读取那么就会根据 timeout 参数来判断是否超时。可以将 readfds 设置为 NULL表示不关心任何文件的读的变化。writefds 和 readfs 类似只是 writefs 用于监视这些文件是否可以进行写操作。exceptfds 用于监视这些文件的异常。比如我们现在要从一个设备文件中读取数据那么就可以定义一个 fd_set 变量这个变量要传递给参数 readfds。当我们定义好一个 fd_set 变量以后可以使用如下所示几个宏进行操作void FD_ZERO(fd_set *set) void FD_SET(int fd, fd_set *set) void FD_CLR(int fd, fd_set *set) int FD_ISSET(int fd, fd_set *set)FD_ZERO用于将 fd_set 变量的所有位都清零FD_SET用于将 fd_set 变量的某个位置 1也就是向 fd_set 添加一个文件描述符参数 fd 就是要加入的文件描述符。FD_CLR用于将 fd_set 变量的某个位清零也就是将一个文件描述符从 fd_set 中删除参数 fd 就是要删除的文件描述符。FD_ISSET用于测试一个文件是否属于某个集合参数 fd 就是要判断的文件描述符。timeout: 超时时间当我们调用 select 函数等待某些文件描述符可以设置超时时间超时时间使用结构体 timeval 表示结构体定义如下所示struct timeval { long tv_sec; /* 秒 */ long tv_usec; /* 微秒 */ };当 timeout 为 NULL 的时候就表示无限期的等待。返回值 0表示超时发生但是没有任何文件描述符可以进行操作 -1发生错误 其他值可以进行操作的文件描述符个数。使用 select 函数对某个设备驱动文件进行读非阻塞访问的操作示例如下所示void main(void) { int ret, fd; /* 要监视的文件描述符 */ fd_set readfds; /* 读操作文件描述符集 */ struct timeval timeout; /* 超时结构体 */ fd open(dev_xxx, O_RDWR | O_NONBLOCK); /* 非阻塞式访问 */ FD_ZERO(readfds); /* 清除 readfds */ FD_SET(fd, readfds); /* 将 fd 添加到 readfds 里面 */ /* 构造超时时间 */ timeout.tv_sec 0; timeout.tv_usec 500000; /* 500ms */ ret select(fd 1, readfds, NULL, NULL, timeout); switch (ret) { case 0: /* 超时 */ printf(timeout!\r\n); break; case -1: /* 错误 */ printf(error!\r\n); break; default: /* 可以读取数据 */ if(FD_ISSET(fd, readfds)) { /* 判断是否为 fd 文件描述符 */ /* 使用 read 函数读取数据 */ } break; } }2、poll 函数在单个线程中select 函数能够监视的文件描述符数量有最大的限制一般为 1024可以修改内核将监视的文件描述符数量改大但是这样会降低效率这个时候就可以使用 poll 函数poll 函数本质上和 select 没有太大的差别但是 poll 函数没有最大文件描述符限制Linux 应用程序中 poll 函数原型如下所示int poll(struct pollfd *fds, nfds_t nfds, int timeout)函数参数和返回值含义如下fds要监视的文件描述符集合以及要监视的事件为一个数组数组元素都是结构体 pollfd 类型的pollfd 结构体如下所示struct pollfd { int fd; /* 文件描述符 */ short events; /* 请求的事件 */ short revents; /* 返回的事件 */ };fd 是要监视的文件描述符如果 fd 无效的话那么 events 监视事件也就无效并且 revents 返回 0。events 是要监视的事件可监视的事件类型如下所示宏含义POLLIN有数据可以读取。POLLPRI有紧急的数据需要读取。POLLOUT可以写数据。POLLERR指定的文件描述符发生错误。POLLHUP指定的文件描述符挂起。POLLNVAL无效的请求。POLLRDNORM等同于 POLLINrevents 是返回参数也就是返回的事件由 Linux 内核设置具体的返回事件。nfdspoll 函数要监视的文件描述符数量。timeout返回时间单位为 ms。返回值返回 revents 域中不为 0 的 pollfd 结构体个数也就是发生事件或错误的文件描述符数量0超时-1发生错误并且设置 errno 为错误类型。使用 poll 函数对某个设备驱动文件进行读非阻塞访问的操作示例如下所示void main(void) { int ret; int fd; /* 要监视的文件描述符 */ struct pollfd fds; fd open(filename, O_RDWR | O_NONBLOCK); /* 非阻塞式访问 */ /* 构造结构体 */ fds.fd fd; fds.events POLLIN; /* 监视数据是否可以读取 */ ret poll(fds, 1, 500); /* 轮询文件是否可操作超时500ms */ if (ret 0) { /* 数据有效 */ /* 读取数据 */ } else if (ret 0) { /* 超时 */ } else if (ret 0) { /* 错误 */ } }3、epoll 函数传统的 select 和 poll 函数都会随着所监听的 fd 数量的增加出现效率低下的问题而且 poll 函数每次必须遍历所有的描述符来检查就绪的描述符这个过程很浪费时间。为此epoll 应运而生epoll 就是为处理大并发而准备的一般常常在网络编程中使用 epoll 函数。应用程序需要先使用 epoll_create 函数创建一个 epoll 句柄epoll_create 函数原型如下int epoll_create(int size)函数参数和返回值含义如下size从 Linux2.6.8 开始此参数已经没有意义了随便填写一个大于 0 的值就可以。返回值epoll 句柄如果为‑1 的话表示创建失败。epoll 句柄创建成功以后使用 epoll_ctl 函数向其中添加要监视的文件描述符以及监视的事件epoll_ctl 函数原型如下所示int epoll_ctl(int epfd, int op, int fd, struct epoll_event *event)函数参数和返回值含义如下epfd要操作的 epoll 句柄也就是使用 epoll_create 函数创建的 epoll 句柄。op表示要对 epfd (epoll 句柄) 进行的操作可以设置为宏含义EPOLL_CTL_ADD向 epfd 添加文件参数 fd 表示的描述符。EPOLL_CTL_MOD修改参数 fd 的 event 事件。EPOLL_CTL_DEL从 epfd 中删除 fd 描述符。fd要监视的文件描述符。event要监视的事件类型为 epoll_event 结构体类型指针epoll_event 结构体类型如下所示struct epoll_event { uint32_t events; /* epoll事件 */ epoll_data_t data; /* 用户数据 */ };结构体 epoll_event 的 events 成员变量表示要监视的事件可选的事件如下所示宏含义EPOLLIN有数据可以读取。EPOLLOUT可以写数据。EPOLLPRI有紧急的数据需要读取。EPOLLERR指定的文件描述符发生错误。EPOLLHUP指定的文件描述符挂起。EPOLLET设置 epoll 为边沿触发默认触发模式为水平触发。EPOLLONESHOT一次性的监视当监视完成以后还需要再次监视某个 fd那么就需要将 fd 重新添加到 epoll 里面。上面这些事件可以进行 “或” 操作也就是说可以设置监视多个事件。返回值0成功‑1失败并且设置 errno 的值为相应的错误码。一切都设置好以后应用程序就可以通过 epoll_wait 函数来等待事件的发生类似 select 函数。epoll_wait 函数原型如下int epoll_wait(int epfd, struct epoll_event *events, int maxevents, int timeout)函数参数和返回值含义如下epfd要等待的 epoll。events指向 epoll_event 结构体的数组当有事件发生的时候 Linux 内核会填写 events调用者可以根据 events 判断发生了哪些事件。maxeventsevents 数组大小必须大于 0。timeout超时时间单位为 ms。返回值0超时‑1错误其他值准备就绪的文件描述符数量。epoll 更多的是用在大规模的并发服务器上因为在这种场合下 select 和 poll 并不适合。当设计到的文件描述符 (fd) 比较少的时候就适合用 select 和 poll。4、Linux 驱动下的 poll 操作函数当应用程序调用 select 或 poll 函数来对驱动程序进行非阻塞访问的时候驱动程序 file_operations 操作集中的 poll 函数就会执行。所以驱动程序的编写者需要提供对应的 poll 函数poll 函数原型如下所示unsigned int (*poll) (struct file *filp, struct poll_table_struct *wait)函数参数和返回值含义如下filp要打开的设备文件 (文件描述符)。wait结构体 poll_table_struct 类型指针由应用程序传递进来的。一般将此参数传递给 poll_wait 函数。返回值向应用程序返回设备或者资源状态可以返回的资源状态如下宏含义POLLIN有数据可以读取。POLLPRI有紧急的数据需要读取。POLLOUT可以写数据。POLLERR指定的文件描述符发生错误。POLLHUP指定的文件描述符挂起。POLLNVAL无效的请求。POLLRDNORM等同于 POLLIN普通数据可读我们需要在驱动程序的 poll 函数中调用 poll_wait 函数poll_wait 函数不会引起阻塞只是将应用程序添加到 poll_table 中poll_wait 函数原型如下void poll_wait(struct file * filp, wait_queue_head_t * wait_address, poll_table *p)参数 wait_address 是要添加到 poll_table 中的等待队列头参数 p 就是 poll_table就是 file_operations 中 poll 函数的 wait 参数。觉得有帮助的话打赏一下呗。。