waitqueue实现阻塞数据访问 - Linux等待队列

第五十二章Linux阻塞和非阻塞IO实验

阻塞和非阻塞IO是Linux驱动开发里面很常见的两种设备访问模式，在编写驱动的时候一定要考虑到阻塞和非阻塞。本章我们就来学习一下阻塞和非阻塞IO，以及如何在驱动程序中处理阻塞与非阻塞，如何在驱动程序使用等待队列和poll机制。

52.1 阻塞和非阻塞IO

52.1.1 阻塞和非阻塞简介

这里的“IO”并不是我们学习STM32或者其他单片机的时候所说的“GPIO”(也就是引脚)。这里的IO指的是Input/Output，也就是输入/输出，是应用程序对驱动设备的输入/输出操作。当应用程序对设备驱动进行操作的时候，如果不能获取到设备资源，那么阻塞式IO就会将应用程序对应的线程挂起，直到设备资源可以才做为止。对于非阻塞IO，应用程序对应的线程不会挂起，它要么一直轮询等待，直到设备资源可以使用，要么就直接放弃。阻塞式IO如图52.1.1.1所示：

图52.1.1.1 阻塞IO访问示意图。

图52.1.1.1中应用程序调用read函数从设备中读取数据，当设备不可用或数据未准备好的时候就会进入到休眠态。等设备可用的时候就会从休眠态唤醒，然后从设备中读取数据返回给应用程序。非阻塞IO如图52.1.2所示：

图52.1.1.2 非阻塞IO访问示意图

从图52.1.1.2可以看出，应用程序使用非阻塞访问方式从设备读取数据，当设备不可用或数据未准备好的时候会立即向内核返回一个错误码，表示数据读取失败。应用程序会再次重新读取数据，这样一直往复循环，直到数据读取成功。

应用程序可以使用如下所示示例代码来实现阻塞访问：

示例代码52.1.1.1 应用程序阻塞读取数据

1int fd;

2int data =0;

3

4 fd =open("/dev/xxx_dev", O_RDWR); /* 阻塞方式打开 */

5 ret =read(fd,&data,sizeof(data)); /* 读取数据 */

从示例代码52.1.1.1可以看出，对于设备驱动文件的默认读取方式就是阻塞式的，所以我们前面所有的例程测试APP都是采用阻塞IO。

如果应用程序要采用非阻塞的方式来访问驱动设备文件，可以使用如下所示代码：

示例代码52.1.1.2 应用程序非阻塞读取数据

1int fd;

2int data =0;

3

4 fd =open("/dev/xxx_dev", O_RDWR | O_NONBLOCK); /* 非阻塞方式打开 */

5 ret =read(fd,&data,sizeof(data)); /* 读取数据 */

第4行使用open函数打开“/dev/xxx_dev”设备文件的时候添加了参数“O_NONBLOCK”，表示以非阻塞方式打开设备，这样从设备中读取数据的时候就是非阻塞方式的了。

52.1.2 等待队列

1、等待队列头

阻塞访问最大的好处就是当设备文件不可操作的时候进程可以进入休眠态，这样可以将CPU资源让出来。但是，当设备文件可以操作的时候就必须唤醒进程，一般在中断函数里面完成唤醒工作。Linux内核提供了等待队列(wait queue)来实现阻塞进程的唤醒工作，如果我们要在驱动中使用等待队列，必须创建并初始化一个等待队列头，等待队列头使用结构体wait_queue_head_t表示，wait_queue_head_t结构体定义在文件include/linux/wait.h中，结构体内容如下所示：

示例代码52.1.2.1 wait_queue_head_t结构体

39struct __wait_queue_head {

40 spinlock_t lock;

41struct list_head task_list;

42};

43typedefstruct __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表示等待队列项，结构体内容如下：

示例代码52.1.2.2 wait_queue_t 结构体

struct __wait_queue {

unsignedint flags;

void *private;

wait_queue_func_t func;

struct list_head task_list;

};

typedefstruct __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函数如表52.1.2.1所示：

52.1.3 轮询

如果用户应用程序以非阻塞的方式访问设备，设备驱动程序就要提供非阻塞的处理方式，也就是轮询。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：要操作的文件描述符个数。

readfds、writefds和exceptfds：这三个指针指向描述符集合，这三个参数指明了关心哪些描述符、需要满足哪些条件等等，这三个参数都是fd_set类型的，fd_set类型变量的每一个位都代表了一个文件描述符。readfds用于监视指定描述符集的读变化，也就是监视这些文件是否可以读取，只要这些集合里面有一个文件可以读取那么seclect就会返回一个大于0的值表示文件可以读取。如果没有文件可以读取，那么就会根据timeout参数来判断是否超时。可以将readfs设置为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_set的某个位是否置1，也就是判断某个文件是否可以进行操作，参数fd就是要判断的文件描述符。

timeout:超时时间，当我们调用select函数等待某些文件描述符可以设置超时时间，超时时间使用结构体timeval表示，结构体定义如下所示：

struct timeval {

long tv_sec; /* 秒 */

long tv_usec; /* 微妙 */

};

当timeout为NULL的时候就表示无限期的等待。

返回值：0，表示的话就表示超时发生，但是没有任何文件描述符可以进行操作；-1，发生错误；其他值，可以进行操作的文件描述符个数。

使用select函数对某个设备驱动文件进行读非阻塞访问的操作示例如下所示：

示例代码52.1.3.1 select函数非阻塞读访问示例

1void main(void)

2{

3 int ret, fd;/* 要监视的文件描述符 */

4 fd_set readfds;/* 读操作文件描述符集 */

5 struct timeval timeout;/* 超时结构体 */

6

7 fd =open("dev_xxx", O_RDWR | O_NONBLOCK); /* 非阻塞式访问 */

8

9 FD_ZERO(&readfds);/* 清除readfds */

10 FD_SET(fd,&readfds);/* 将fd添加到readfds里面 */

11

12 /* 构造超时时间 */

13 timeout.tv_sec =0;

14 timeout.tv_usec =500000;/* 500ms */

15

16 ret =select(fd +1,&readfds,NULL,NULL,&timeout);

17 switch(ret){

18 case0:/* 超时 */

19 printf("timeout!\r\n");

20 break;

21 case-1:/* 错误 */

22 printf("error!\r\n");

23 break;

24 default:/* 可以读取数据 */

25 if(FD_ISSET(fd,&readfds)){/* 判断是否为fd文件描述符 */

26 /* 使用read函数读取数据 */

27 }

28 break;

29 }

30}

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 等同于POLLIN

revents是返回参数，也就是返回的事件，有Linux内核设置具体的返回事件。

nfds：poll函数要监视的文件描述符数量。

timeout：超时时间，单位为ms。

返回值：返回revents域中不为0的pollfd结构体个数，也就是发生事件或错误的文件描述符数量；0，超时；-1，发生错误，并且设置errno为错误类型。

使用poll函数对某个设备驱动文件进行读非阻塞访问的操作示例如下所示：

示例代码52.1.3.2 poll函数读非阻塞访问示例

1void main(void)

2{

3 int ret;

4 int fd;/* 要监视的文件描述符 */

5 struct pollfd fds;

6

7 fd =open(filename, O_RDWR | O_NONBLOCK);/* 非阻塞式访问 */

8

9 /* 构造结构体 */

10 fds.fd =fd;

11 fds.events =POLLIN;/* 监视数据是否可以读取 */

12

13 ret =poll(&fds,1,500); /* 轮询文件是否可操作，超时500ms */

14 if(ret){/* 数据有效 */

15 ......

16 /* 读取数据 */

17 ......

18 }elseif(ret ==0){/* 超时 */

19 ......

20 }elseif(ret <0){/* 错误 */

21 ......

22 }

23}

3、epoll函数

传统的selcet和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判断发生了哪些事件。

maxevents：events数组大小，必须大于0。

timeout：超时时间，单位为ms。

返回值：0，超时；-1，错误；其他值，准备就绪的文件描述符数量。

epoll更多的是用在大规模的并发服务器上，因为在这种场合下select和poll并不适合。当设计到的文件描述符(fd)比较少的时候就适合用selcet和poll，本章我们就使用sellect和poll这两个函数。

52.1.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参数。

52.2 阻塞IO实验

在上一章Linux中断实验中，我们直接在应用程序中通过read函数不断的读取按键状态，当按键有效的时候就打印出按键值。这种方法有个缺点，那就是imx6uirqApp这个测试应用程序拥有很高的CPU占用率，大家可以在开发板中加载上一章的驱动程序模块imx6uirq.ko，然后以后台运行模式打开imx6uirqApp这个测试软件，命令如下：

./imx6uirqApp /dev/imx6uirq &

测试驱动是否正常工作，如果驱动工作正常的话输入“top”命令查看imx6uirqApp这个应用程序的CPU使用率，结果如图52.2.1所示：

图52.2.1 CPU使用率

从图52.2.1可以看出，imx6uirqApp这个应用程序的CPU使用率竟然高达99.6%，这仅仅是一个读取按键值的应用程序，这么高的CPU使用率显然是有问题的！原因就在于我们是直接在while循环中通过read函数读取按键值，因此imx6uirqApp这个软件会一直运行，一直读取按键值，CPU使用率肯定就会很高。最好的方法就是在没有有效的按键事件发生的时候，imx6uirqApp这个应用程序应该处于休眠状态，当有按键事件发生以后imx6uirqApp这个应用程序才运行，打印出按键值，这样就会降低CPU使用率，本小节我们就使用阻塞IO来实现此功能。

52.2.1 硬件原理图分析

本章实验硬件原理图参考15.2小节即可。

52.2.2 实验程序编写

1、驱动程序编写

本实验对应的例程路径为：开发板光盘->2、Linux驱动例程->14_blockio。

本章实验我们在上一章的“13_irq”实验的基础上完成，主要是对其添加阻塞访问相关的代码。新建名为“14_blockio”的文件夹，然后在14_blockio文件夹里面创建vscode工程，工作区命名为“blockio”。将“13_irq”实验中的imx6uirq.c复制到14_blockio文件夹中，并重命名为blockio.c。接下来我们就修改blockio.c这个文件，在其中添加阻塞相关的代码，完成以后的blockio.c内容如下所示(因为是在上一章实验的imx6uirq.c文件的基础上修改的，因为了减少篇幅，下面的代码有省略)：

示例代码52.2.2.1 blockio.c文件代码(有省略)

1 #include 

2 #include 

......

18 #include 

19 #include 

20 #include 

21/***************************************************************

22 Copyright ALIENTEK Co., Ltd. 1998-2029. All rights reserved.

23文件名 : block.c

24作者 : 左忠凯

25版本 : V1.0

26描述 : 阻塞IO访问

27其他 : 无

28论坛 : www.openedv.com

29日志 : 初版V1.0 2019/7/26 左忠凯创建

30 ***************************************************************/

31 #define IMX6UIRQ_CNT 1 /* 设备号个数 */

32 #define IMX6UIRQ_NAME "blockio" /* 名字 */

33 #define KEY0VALUE 0X01 /* KEY0按键值 */

34 #define INVAKEY 0XFF /* 无效的按键值 */

35 #define KEY_NUM 1 /* 按键数量 */

36

37/* 中断IO描述结构体 */

38struct irq_keydesc {

39int gpio;/* gpio */

40int irqnum;/* 中断号 */

41unsignedchar value;/* 按键对应的键值 */

42char name[10];/* 名字 */

43 irqreturn_t (*handler)(int,void*);/* 中断服务函数 */

44};

45

46/* imx6uirq设备结构体 */

47struct imx6uirq_dev{

48 dev_t devid;/* 设备号 */

49struct cdev cdev;/* cdev */

50struct class *class;/* 类 */

51struct device *device;/* 设备 */

52int major;/* 主设备号 */

53int minor;/* 次设备号 */

54struct device_node *nd;/* 设备节点 */

55 atomic_t keyvalue;/* 有效的按键键值 */

56 atomic_t releasekey;/* 标记是否完成一次完成的按键 */

57struct timer_list timer; /* 定义一个定时器*/

58struct irq_keydesc irqkeydesc[KEY_NUM];/* 按键init述数组 */

59unsignedchar curkeynum;/* 当前init按键号 */

60

61 wait_queue_head_t r_wait; /* 读等待队列头 */

62};

63

64struct imx6uirq_dev imx6uirq;/* irq设备 */

65

66/* @description : 中断服务函数，开启定时器

67 * 定时器用于按键消抖。

68 * @param - irq : 中断号

69 * @param - dev_id : 设备结构。

70 * @return : 中断执行结果

71 */

72static irqreturn_t key0_handler(int irq,void*dev_id)

73{

74struct imx6uirq_dev *dev =(struct imx6uirq_dev*)dev_id;

75

76 dev->curkeynum =0;

77 dev->timer.data =(volatilelong)dev_id;

78 mod_timer(&dev->timer,jiffies + msecs_to_jiffies(10));

79returnIRQ_RETVAL(IRQ_HANDLED);

80}

81

82/* @description : 定时器服务函数，用于按键消抖，定时器到了以后

83 * 再次读取按键值，如果按键还是处于按下状态就表示按键有效。

84 * @param - arg : 设备结构变量

85 * @return : 无

86 */

87void timer_function(unsignedlong arg)

88{

89unsignedchar value;

90unsignedchar num;

91struct irq_keydesc *keydesc;

92struct imx6uirq_dev *dev =(struct imx6uirq_dev *)arg;

93

94 num =dev->curkeynum;

95 keydesc =&dev->irqkeydesc[num];

96

97 value =gpio_get_value(keydesc->gpio); /* 读取IO值 */

98if(value ==0){/* 按下按键 */

99 atomic_set(&dev->keyvalue,keydesc->value);

100}

101else{/* 按键松开 */

102 atomic_set(&dev->keyvalue,0x80|keydesc->value);

103 atomic_set(&dev->releasekey,1);

104}

105

106/* 唤醒进程 */

107if(atomic_read(&dev->releasekey)){/* 完成一次按键过程 */

108/* wake_up(&dev->r_wait); */

109 wake_up_interruptible(&dev->r_wait);

110}

111}

112

113/*

114 * @description : 按键IO初始化

115 * @param : 无

116 * @return : 无

117 */

118staticint keyio_init(void)

119{

120unsignedchar i =0;

121char name[10];

122int ret =0;

......

163/* 创建定时器 */

164 init_timer(&imx6uirq.timer);

165 imx6uirq.timer.function =timer_function;

166

167/* 初始化等待队列头 */

168 init_waitqueue_head(&imx6uirq.r_wait);

169return0;

170}

171

172/*

173 * @description : 打开设备

174 * @param – inode : 传递给驱动的inode

175 * @param - filp : 设备文件，file结构体有个叫做private_data的成员变量

176 * 一般在open的时候将private_data指向设备结构体。

177 * @return : 0 成功;其他失败

178 */

179staticint imx6uirq_open(struct inode *inode,struct file *filp)

180{

181 filp->private_data =&imx6uirq;/* 设置私有数据 */

182return0;

183}

184

185/*

186 * @description : 从设备读取数据

187 * @param - filp : 要打开的设备文件(文件描述符)

188 * @param - buf : 返回给用户空间的数据缓冲区

189 * @param - cnt : 要读取的数据长度

190 * @param - offt : 相对于文件首地址的偏移

191 * @return : 读取的字节数，如果为负值，表示读取失败

192 */

193static ssize_t imx6uirq_read(struct file *filp,char __user *buf,

size_t cnt,loff_t *offt)

194{

195int ret =0;

196unsignedchar keyvalue =0;

197unsignedchar releasekey =0;

198struct imx6uirq_dev *dev =(struct imx6uirq_dev *)

filp->private_data;

199

200 #if0

201/* 加入等待队列，等待被唤醒,也就是有按键按下 */

202 ret = wait_event_interruptible(dev->r_wait,

atomic_read(&dev->releasekey));

203if(ret){

204goto wait_error;

205}

206 #endif

207

208 DECLARE_WAITQUEUE(wait, current);/* 定义一个等待队列 */

209if(atomic_read(&dev->releasekey)==0){ /* 没有按键按下 */

210 add_wait_queue(&dev->r_wait,&wait); /*添加到等待队列头 */

211 __set_current_state(TASK_INTERRUPTIBLE);/* 设置任务状态 */

212 schedule(); /* 进行一次任务切换 */

213if(signal_pending(current)){ /* 判断是否为信号引起的唤醒 */

214 ret =-ERESTARTSYS;

215goto wait_error;

216}

217}

218 remove_wait_queue(&dev->r_wait,&wait);/* 唤醒以后将等待队列移除 */

219

220 keyvalue =atomic_read(&dev->keyvalue);

221 releasekey =atomic_read(&dev->releasekey);

......

234return0;

235

236 wait_error:

237 set_current_state(TASK_RUNNING); /* 设置任务为运行态 */

238 remove_wait_queue(&dev->r_wait,&wait); /* 将等待队列移除 */

239returnret;

240

241 data_error:

242return-EINVAL;

243}

244

245/* 设备操作函数 */

246staticstruct file_operations imx6uirq_fops ={

247.owner =THIS_MODULE,

248.open =imx6uirq_open,

249.read =imx6uirq_read,

250};

251

252/*

253 * @description : 驱动入口函数

254 * @param : 无

255 * @return : 无

256 */

257staticint __init imx6uirq_init(void)

258{

259/* 1、构建设备号 */

260if(imx6uirq.major){

261 imx6uirq.devid =MKDEV(imx6uirq.major,0);

262 register_chrdev_region(imx6uirq.devid,IMX6UIRQ_CNT,

IMX6UIRQ_NAME);

263}else{

264 alloc_chrdev_region(&imx6uirq.devid,0,IMX6UIRQ_CNT,

IMX6UIRQ_NAME);

265 imx6uirq.major =MAJOR(imx6uirq.devid);

266 imx6uirq.minor =MINOR(imx6uirq.devid);

267}

......

284

285/* 5、始化按键 */

286 atomic_set(&imx6uirq.keyvalue,INVAKEY);

287 atomic_set(&imx6uirq.releasekey,0);

288 keyio_init();

289return0;

290}

291

292/*

293 * @description : 驱动出口函数

294 * @param : 无

295 * @return : 无

296 */

297staticvoid __exit imx6uirq_exit(void)

298{

299unsigned i =0;

300/* 删除定时器 */

301 del_timer_sync(&imx6uirq.timer);/* 删除定时器 */

......

310 class_destroy(imx6uirq.class);

311}

312

313 module_init(imx6uirq_init);

314 module_exit(imx6uirq_exit);

315 MODULE_LICENSE("GPL");

第32行，修改设备文件名字为“blockio”，当驱动程序加载成功以后就会在根文件系统中出现一个名为“/dev/blockio”的文件。

第61行，在设备结构体中添加一个等待队列头r_wait，因为在Linux驱动中处理阻塞IO需要用到等待队列。

第107~110行，定时器中断处理函数执行，表示有按键按下，先在107行判断一下是否是一次有效的按键，如果是的话就通过wake_up或者wake_up_interruptible函数来唤醒等待队列r_wait。

第168行，调用init_waitqueue_head函数初始化等待队列头r_wait。

第200~206行，采用等待事件来处理read的阻塞访问，wait_event_interruptible函数等待releasekey有效，也就是有按键按下。如果按键没有按下的话进程就会进入休眠状态，因为采用了wait_event_interruptible函数，因此进入休眠态的进程可以被信号打断。

第208~218行，首先使用 DECLARE_WAITQUEUE宏定义一个等待队列，如果没有按键按下的话就使用add_wait_queue函数将当前任务的等待队列添加到等待队列头r_wait中。随后调用__set_current_state函数设置当前进程的状态为TASK_INTERRUPTIBLE，也就是可以被信号打断。接下来调用schedule函数进行一次任务切换，当前进程就会进入到休眠态。如果有按键按下，那么进入休眠态的进程就会唤醒，然后接着从休眠点开始运行。在这里也就是从第213行开始运行，首先通过signal_pending函数判断一下进程是不是由信号唤醒的，如果是由信号唤醒的话就直接返回-ERESTARTSYS这个错误码。如果不是由信号唤醒的(也就是被按键唤醒的)那么就在218行调用remove_wait_queue函数将进程从等待队列中删除。

使用等待队列实现阻塞访问重点注意两点：

①、将任务或者进程加入到等待队列头，

②、在合适的点唤醒等待队列，一般都是中断处理函数里面。

2、编写测试APP

本节实验的测试APP直接使用第51.3.3小节所编写的imx6uirqApp.c，将imx6uirqApp.c复制到本节实验文件夹下，并且重命名为blockioApp.c，不需要修改任何内容。

52.2.3 运行测试

1、编译驱动程序和测试APP

①、编译驱动程序

编写Makefile文件，本章实验的Makefile文件和第四十章实验基本一样，只是将obj-m变量的值改为blockio.o，Makefile内容如下所示：

示例代码52.2.3.1 Makefile文件

1 KERNELDIR:=/home/zuozhongkai/linux/IMX6ULL/linux/temp/linux-imx-rel_imx_4.1.15_2.1.0_ga_alientek

......

4 obj-m := blockio.o

......

11 clean:

12$(MAKE) -C $(KERNELDIR) M=$(CURRENT_PATH) clean

第4行，设置obj-m变量的值为blockio.o。

输入如下命令编译出驱动模块文件：

make-j32

编译成功以后就会生成一个名为“blockio.ko”的驱动模块文件。

②、编译测试APP

输入如下命令编译测试noblockioApp.c这个测试程序：

arm-linux-gnueabihf-gcc blockioApp.c -o blockioApp

编译成功以后就会生成blcokioApp这个应用程序。

2、运行测试

将上一小节编译出来blockio.ko和blockioApp这两个文件拷贝到rootfs/lib/modules/4.1.15目录中，重启开发板，进入到目录lib/modules/4.1.15中，输入如下命令加载blockio.ko驱动模块：

depmod //第一次加载驱动的时候需要运行此命令

modprobe blockio.ko //加载驱动

驱动加载成功以后使用如下命令打开blockioApp这个测试APP，并且以后台模式运行：

./blockioApp /dev/blockio &

按下开发板上的KEY0按键，结果如图52.2.3.1所示：

图52.2.3.1 测试APP运行测试

当按下KEY0按键以后blockioApp这个测试APP就会打印出按键值。输入“top”命令，查看blockioAPP这个应用APP的CPU使用率，如图52.2.3.2所示：

图52.2.3.2 应用程序CPU使用率

从图52.2.3.2可以看出，当我们在按键驱动程序里面加入阻塞访问以后，blockioApp这个应用程序的CPU使用率从图52.2.1中的99.6%降低到了0.0%。大家注意，这里的0.0%并不是说blockioApp这个应用程序不使用CPU了，只是因为使用率太小了，CPU使用率可能为0.00001%，但是图52.2.3.2只能显示出小数点后一位，因此就显示成了0.0%。

我们可以使用“kill”命令关闭后台运行的应用程序，比如我们关闭掉blockioApp这个后台运行的应用程序。首先输出“Ctrl+C”关闭top命令界面，进入到命令行模式。然后使用“ps”命令查看一下blockioApp这个应用程序的PID，如图52.2.3.3所示：

图52.2.3.3 当前系统所有进程的ID

从图图52.2.3.3可以看出，blockioApp这个应用程序的PID为149，使用“kill-9PID”即可“杀死”指定PID的进程，比如我们现在要“杀死”PID为149的blockioApp应用程序，可是使用如下命令：

kill-9 149

输入上述命令以后终端显示如图52.2.3.4所示：

图52.2.3.4 kill命令输出结果

从图52.2.3.4可以看出，“./blockioApp /dev/blockio”这个应用程序已经被“杀掉”了，在此输入“ps”命令查看当前系统运行的进程，会发现blockioApp已经不见了。这个就是使用kill命令“杀掉”指定进程的方法。

52.3 非阻塞IO实验

52.3.1 硬件原理图分析

本章实验硬件原理图参考15.2小节即可。

52.3.2 实验程序编写

1、驱动程序编写

本实验对应的例程路径为：开发板光盘->2、Linux驱动例程->15_noblockio。

本章实验我们在52.2小节中的“14_blockio”实验的基础上完成，上一小节实验我们已经在驱动中添加了阻塞IO的代码，本小节我们继续完善驱动，加入非阻塞IO驱动代码。新建名为“15_noblockio”的文件夹，然后在15_noblockio文件夹里面创建vscode工程，工作区命名为“noblockio”。将“14_blockio”实验中的blockio.c复制到15_noblockio文件夹中，并重命名为noblockio.c。接下来我们就修改noblockio.c这个文件，在其中添加非阻塞相关的代码，完成以后的noblockio.c内容如下所示(因为是在上一小节实验的blockio.c文件的基础上修改的，因为了减少篇幅，下面的代码有省略)：

示例代码52.3.3.1 noblockio.c文件(有省略)

1 #include 

2 #include 

......

18 #include 

19 #include 

20 #include 

21 #include 

22 #include 

23/***************************************************************

24 Copyright ALIENTEK Co., Ltd. 1998-2029. All rights reserved.

25文件名 : noblock.c

26 作者 : 左忠凯

27版本 : V1.0

28描述 : 非阻塞IO访问

29其他 : 无

30论坛 : www.openedv.com

31日志 : 初版V1.0 2019/7/26 左忠凯创建

32 ***************************************************************/

31 #define IMX6UIRQ_CNT 1 /* 设备号个数 */

32 #define IMX6UIRQ_NAME "noblockio" /* 名字 */

......

187/*

188 * @description : 从设备读取数据

189 * @param - filp : 要打开的设备文件(文件描述符)

190 * @param - buf : 返回给用户空间的数据缓冲区

191 * @param - cnt : 要读取的数据长度

192 * @param - offt : 相对于文件首地址的偏移

193 * @return : 读取的字节数，如果为负值，表示读取失败

194 */

195static ssize_t imx6uirq_read(struct file *filp,char __user *buf,

size_t cnt,loff_t *offt)

196{

197int ret =0;

198unsignedchar keyvalue =0;

199unsignedchar releasekey =0;

200struct imx6uirq_dev *dev =(struct imx6uirq_dev *)

filp->private_data;

201

202if(filp->f_flags & O_NONBLOCK){ /* 非阻塞访问 */

203if(atomic_read(&dev->releasekey)==0) /* 没有按键按下*/

204return-EAGAIN;

205}else{/* 阻塞访问 */

206/* 加入等待队列，等待被唤醒,也就是有按键按下 */

207 ret =wait_event_interruptible(dev->r_wait,

atomic_read(&dev->releasekey));

208if(ret){

209gotowait_error;

210}

211}

......

229 wait_error:

230returnret;

231 data_error:

232return-EINVAL;

233}

234

235/*

236 * @description : poll函数，用于处理非阻塞访问

237 * @param - filp : 要打开的设备文件(文件描述符)

238 * @param - wait : 等待列表(poll_table)

239 * @return : 设备或者资源状态，

240 */

241unsignedint imx6uirq_poll(struct file *filp,

struct poll_table_struct *wait)

242{

243unsignedint mask =0;

244struct imx6uirq_dev *dev =(struct imx6uirq_dev *)

filp->private_data;

245

246 poll_wait(filp,&dev->r_wait, wait);

247

248if(atomic_read(&dev->releasekey)){ /* 按键按下 */

249 mask = POLLIN | POLLRDNORM; /* 返回PLLIN */

250}

251return mask;

252}

253

254/* 设备操作函数 */

255staticstruct file_operations imx6uirq_fops ={

256.owner =THIS_MODULE,

257.open =imx6uirq_open,

258.read =imx6uirq_read,

259.poll = imx6uirq_poll,

260};

261

262/*

263 * @description : 驱动入口函数

264 * @param : 无

265 * @return : 无

266 */

267staticint __init imx6uirq_init(void)

268{

......

298 keyio_init();

299return0;

300}

301

302/*

303 * @description : 驱动出口函数

304 * @param : 无

305 * @return : 无

306 */

307staticvoid __exit imx6uirq_exit(void)

308{

309unsigned i =0;

310/* 删除定时器 */

311 del_timer_sync(&imx6uirq.timer);/* 删除定时器 */

......

320 class_destroy(imx6uirq.class);

321}

322

323 module_init(imx6uirq_init);

324 module_exit(imx6uirq_exit);

325 MODULE_LICENSE("GPL");

第32行，修改设备文件名字为“noblockio”，当驱动程序加载成功以后就会在根文件系统中出现一个名为“/dev/noblockio”的文件。

第202~204行，判断是否为非阻塞式读取访问，如果是的话就判断按键是否有效，也就是判断一下有没有按键按下，如果没有的话就返回-EAGAIN。

第241~252行，imx6uirq_poll函数就是file_operations驱动操作集中的poll函数，当应用程序调用select或者poll函数的时候imx6uirq_poll函数就会执行。第246行调用poll_wait函数将等待队列头添加到poll_table中，第248~250行判断按键是否有效，如果按键有效的话就向应用程序返回POLLIN这个事件，表示有数据可以读取。

第259行，设置file_operations的poll成员变量为imx6uirq_poll。

2、编写测试APP

新建名为noblockioApp.c测试APP文件，然后在其中输入如下所示内容：

示例代码52.3.3.2 noblockioApp.c文件代码

1 #include "stdio.h"

2 #include "unistd.h"

3 #include "sys/types.h"

4 #include "sys/stat.h"

5 #include "fcntl.h"

6 #include "stdlib.h"

7 #include "string.h"

8 #include "poll.h"

9 #include "sys/select.h"

10 #include "sys/time.h"

11 #include "linux/ioctl.h"

12/***************************************************************

13 Copyright ALIENTEK Co., Ltd. 1998-2029. All rights reserved.

14文件名 : noblockApp.c

15作者 : 左忠凯

16版本 : V1.0

17描述 : 非阻塞访问测试APP

18其他 : 无

19使用方法 ：./blockApp /dev/blockio 打开测试App

20论坛 : www.openedv.com

21日志 : 初版V1.0 2019/9/8 左忠凯创建

22 ***************************************************************/

23

24/*

25 * @description : main主程序

26 * @param - argc : argv数组元素个数

27 * @param - argv : 具体参数

28 * @return : 0 成功;其他失败

29 */

30int main(int argc,char*argv[])

31{

32int fd;

33int ret =0;

34char*filename;

35struct pollfd fds;

36 fd_set readfds;

37struct timeval timeout;

38unsignedchar data;

39

40if(argc !=2){

41 printf("Error Usage!\r\n");

42return-1;

43}

44

45 filename =argv[1];

46 fd =open(filename, O_RDWR | O_NONBLOCK);/* 非阻塞访问 */

47if(fd <0){

48 printf("Can't open file %s\r\n",filename);

49return-1;

50}

51

52 #if0

53/* 构造结构体 */

54 fds.fd =fd;

55 fds.events =POLLIN;

56

57while(1){

58 ret =poll(&fds,1,500);

59if(ret){/* 数据有效 */

60 ret =read(fd,&data,sizeof(data));

61if(ret <0){

62/* 读取错误 */

63}else{

64if(data)

65 printf("key value = %d \r\n",data);

66}

67}elseif(ret ==0){/* 超时 */

68/* 用户自定义超时处理 */

69}elseif(ret <0){/* 错误 */

70/* 用户自定义错误处理 */

71}

72}

73 #endif

74

75while(1){

76 FD_ZERO(&readfds);

77 FD_SET(fd,&readfds);

78/* 构造超时时间 */

79 timeout.tv_sec =0;

80 timeout.tv_usec =500000;/* 500ms */

81 ret =select(fd +1,&readfds,NULL,NULL,&timeout);

82switch(ret){

83case0:/* 超时 */

84/* 用户自定义超时处理 */

85break;

86case-1:/* 错误 */

87/* 用户自定义错误处理 */

88break;

89default:/* 可以读取数据 */

90if(FD_ISSET(fd,&readfds)){

91 ret =read(fd,&data,sizeof(data));

92if(ret <0){

93/* 读取错误 */

94}else{

95if(data)

96 printf("key value=%d\r\n",data);

97}

98}

99break;

100}

101}

102

103 close(fd);

104returnret;

105}

第52~73行，这段代码使用poll函数来实现非阻塞访问，在while循环中使用poll函数不断的轮询，检查驱动程序是否有数据可以读取，如果可以读取的话就调用read函数读取按键数据。

第75~101行，这段代码使用select函数来实现非阻塞访问。

52.3.3运行测试

1、编译驱动程序和测试APP

①、编译驱动程序

编写Makefile文件，本章实验的Makefile文件和第四十章实验基本一样，只是将obj-m变量的值改为noblockio.o，Makefile内容如下所示：

示例代码52.3.3.1 Makefile文件

1 KERNELDIR:=/home/zuozhongkai/linux/IMX6ULL/linux/temp/linux-imx-rel_imx_4.1.15_2.1.0_ga_alientek

......

4 obj-m := noblockio.o

......

11 clean:

12$(MAKE) -C $(KERNELDIR) M=$(CURRENT_PATH) clean

第4行，设置obj-m变量的值为noblockio.o。

输入如下命令编译出驱动模块文件：

make-j32

编译成功以后就会生成一个名为“noblockio.ko”的驱动模块文件。

②、编译测试APP

输入如下命令编译测试noblockioApp.c这个测试程序：

arm-linux-gnueabihf-gcc noblockioApp.c -o noblockioApp

编译成功以后就会生成noblcokioApp这个应用程序。

2、运行测试

将上一小节编译出来noblockio.ko和noblockioApp这两个文件拷贝到rootfs/lib/modules/4.1.15目录中，重启开发板，进入到目录lib/modules/4.1.15中，输入如下命令加载blockio.ko驱动模块：

depmod //第一次加载驱动的时候需要运行此命令

modprobe noblockio.ko //加载驱动

驱动加载成功以后使用如下命令打开noblockioApp这个测试APP，并且以后台模式运行：

./noblockioApp /dev/noblockio &

按下开发板上的KEY0按键，结果如图52.3.3.1所示：

图52.3.3.1 测试APP运行测试

当按下KEY0按键以后noblockioApp这个测试APP就会打印出按键值。输入“top”命令，查看noblockioAPP这个应用APP的CPU使用率，如图52.3.3.2所示：

图52.3.3.2 应用程序CPU使用率

从图52.3.3.2可以看出，采用非阻塞方式读处理以后，noblockioApp的CPU占用率也低至0.0%，和图52.2.3.2中的blockioApp一样，这里的0.0%并不是说noblockioApp这个应用程序不使用CPU了，只是因为使用率太小了，而图中只能显示出小数点后一位，因此就显示成了0.0%。

如果要“杀掉”处于后台运行模式的noblockioApp这个应用程序，可以参考52.2.3小节讲解的方法。