Linux 设备驱动中的异步通知与异步 I/O(2)

http://blog.csdn.net/xinyuwuxian/article/details/9359685

成于坚持，败于止步

Linux 2.6 异步 I/O 

AIO 概念与 GNU C 库函数 

Linux 系统中最常用的输入/输出（I/O）模型是同步 I/O。在这个模型中，当请求发出之后，应用程序就会阻塞，直到请求满足为止。这是很好的一种解决方案，因为调用应用程序在等待 I/O 请求完成时不需要使用任何中央处理单元（CPU）。但是在某
些情况下，I/O 请求可能需要与其他进程产生交叠。可移植操作系统接口（POSIX）异步 I/O（AIO）应用程序接口（API）就提供了这种功能。 

Linux 异步 I/O 是 2.6 版本内核的一个标准特性，但是我们在 2.4 版本内核的补丁中也可以找到它。AIO 基本思想是允许进程发起很多 I/O 操作，而不用阻塞或等待任何操作完成。稍后或在接收到 I/O 操作完成的通知时，进程就可以检索 I/O 操作的结
果。 

select()函数所提供的功能（异步阻塞 I/O）与 AIO 类似，它对通知事件进行阻塞，而不是对 I/O 调用进行阻塞。 

在异步非阻塞 I/O 中，我们可以同时发起多个传输操作。这需要每个传输操作都有惟一的上下文，这样才能在它们完成时区分到底是哪个传输操作完成了。在 AIO 中，通过 aiocb（AIO I/O Control Block）结构体进行区分。这个结构体包含了有关传输的所有信息，包括为数据准备的用户缓冲区。在产生 I/O（称为完成）通知时，aiocb 结构就被用来惟一标识所完成的 I/O 操作。 

AIO 系列 API 被 GNU C 库函数所包含，它被 POSIX.1b 所要求，主要包括如下函数。 

1．aio_read 

aio_read()函数请求对一个有效的文件描述符进行异步读操作。这个文件描述符可以表示一个文件、套接字甚至管道

aio_read 函数的原型如下： 

int aio_read( struct aiocb *aiocbp ); 

aio_read()函数在请求进行排队之后会立即返回。如果执行成功，返回值就为 0；如果出现错误，返回值就为?1，并设置 errno 的值。 

2．aio_write 

aio_write()函数用来请求一个异步写操作，其函数原型如下： 

int aio_write( struct aiocb *aiocbp ); 

aio_write()函数会立即返回，说明请求已经进行排队（成功时返回值为 0，失败时返回值为?1，并相应地设置 errno。 

3．aio_error 

aio_error 函数被用来确定请求的状态，其原型如下： 

int aio_error( struct aiocb *aiocbp ); 

这个函数可以返回以下内容。 

EINPROGRESS：说明请求尚未完成。 

ECANCELLED：说明请求被应用程序取消了。 

-1：说明发生了错误，具体错误原因由 errno 记录。 

4．aio_return 

异步 I/O 和标准块 I/O 之间的另外一个区别是不能立即访问这个函数的返回状态，因为并没有阻塞在 read()调用上。在标准的 read()调用中，返回状态是在该函数返回时提供的。但是在异步 I/O 中，我们要使用 aio_return()函数。这个函数的原型如下：

ssize_t aio_return( struct aiocb *aiocbp ); 

只有在 aio_error()调用确定请求已经完成（可能成功，也可能发生了错误）之后，才会调用这个函数。aio_return()的返回值就等价于同步情况中 read 或 write 系统调用的返回值（所传输的字节数，如果发生错误，返回值就为?1）。 

下面代码清单给出了用户空间应用程序进行异步读操作的一个例程，它首先打开文件，然后准备 aiocb 结构体，之后调用 aio_read（&my_aiocb）进行提出异步读请求，当 aio_error(&my_aiocb) = = EINPROGRESS 即操作还在进行中时，一直等待，结束后通过 aio_return（&my_aiocb）获得返回值。

1  #include <aio.h>
2  ...
3  int fd, ret;
4  struct aiocb my_aiocb;
5
6  fd = open("file.txt", O_RDONLY);
7  if (fd < 0)
8    perror("open");
9
10 /* 清零 aiocb 结构体 */
11 bzero((char*) &my_aiocb, sizeof(struct aiocb));
12
13 /* 为 aiocb 请求分配数据缓冲区 */
14 my_aiocb.aio_buf = malloc(BUFSIZE + 1);
15 if (!my_aiocb.aio_buf)
16   perror("malloc");
17
18 /* 初始化 aiocb 的成员 */
19 my_aiocb.aio_fildes = fd;
20 my_aiocb.aio_nbytes = BUFSIZE;
21 my_aiocb.aio_offset = 0;
22
23 ret = aio_read(&my_aiocb);
24 if (ret < 0)
25   perror("aio_read");
26
27 while (aio_error(&my_aiocb) == EINPROGRESS)
28   ;
29
30 if ((ret = aio_return(&my_iocb)) > 0)
31 {
32   /* 获得异步读的返回值 */
33 }
34 else
35 {
36   /* 读失败，分析 errorno */
37 }

用户可以使用 aio_suspend()函数来挂起（或阻塞）调用进程，直到异步请求完成为止，此时会产生一个信号，或者发生其他超时操作。调用者提供了一个 aiocb 引用列表，其中任何一个完成都会导致 aio_suspend()返回。aio_suspend 的函数原型如下： 

int aio_suspend( const struct aiocb *const cblist[], int n, const struct timespec *timeout ); 

下面代码清单给出了用户空间异步读操作时使用 aio_suspend()函数的例子。

1 struct aioct *cblist[MAX_LIST]
2 /* 清零 aioct 结构体链表 */
3 bzero((char *)cblist, sizeof(cblist));
4 /* 将一个或更多的 aiocb 放入 aioct 结构体链表 */
5 cblist[0] = &my_aiocb;
6 ret = aio_read( &my_aiocb );
7 ret = aio_suspend( cblist, MAX_LIST, NULL );

aio_cancel()函数允许用户取消对某个文件描述符执行的一个或所有 I/O 请求。其原型如下： 

int aio_cancel( int fd, struct aiocb *aiocbp ); 

如果要取消一个请求，用户需提供文件描述符和 aiocb 引用。如果这个请求被成功取消了，那么这个函数就会返回 AIO_CANCELED。如果请求完成了，这个函数就会返回AIO_NOTCANCELED。 如果要取消对某个给定文件描述符的所有请求，用户需要提供这个文件的描述符以及一个对 aiocbp 的 NULL 引用。如果所有的请求都取消了，这个函数就会返回AIO_CANCELED ；如果至少有一个请求没有被取消，那么这个函数就会返回AIO_NOT_CANCELED；如果没有一个请求可以被取消，那么这个函数就会返回AIO_ALLDONE。然后，可以使用 aio_error()来验证每个 AIO 请求，如果某请求已经被取消了，那么 aio_error()就会返回?1，并且 errno 会被设置为 ECANCELED。 lio_listio()函数可用于同时发起多个传输。这个函数非常重要，它使得用户可以在一个系统调用（一次内核上下文切换）中启动大量的 I/O 操作。lio_listio API 函数的原型如下： 

int lio_listio( int mode, struct aiocb *list[], int nent, struct sigevent *sig ); 

mode 参数可以是 LIO_WAIT 或 LIO_NOWAIT。LIO_WAIT 会阻塞这个调用，直到所有的 I/O 都完成为止。在操作进行排队之后，LIO_NOWAIT 就会返回。list 是一个 aiocb 引用的列表，最大元素的个数是由 nent 定义的。如果 list 的元素为 NULL，lio_listio()会将其忽略。 

代码清单给出了用户空间异步 I/O 操作时使用 lio_listio()函数的例子。

1  struct aiocb aiocb1, aiocb2;
2  struct aiocb *list[MAX_LIST];
3  ...
4  /* 准备第一个 aiocb */
5  aiocb1.aio_fildes = fd;
6  aiocb1.aio_buf = malloc( BUFSIZE+1 );
7  aiocb1.aio_nbytes = BUFSIZE;
8  aiocb1.aio_offset = next_offset;
9  aiocb1.aio_lio_opcode = LIO_READ; /*异步读操作*/
10 ...  /*准备多个 aiocb */
11 bzero( (char *)list, sizeof(list) );
12
13 /*将 aiocb 填入链表*/
14 list[0] = &aiocb1;
15 list[1] = &aiocb2;
16 ...
17 ret = lio_listio( LIO_WAIT, list, MAX_LIST, NULL );/*发起大量 I/O 操作*/

上述代码第 9 行中，因为是进行异步读操作，所以操作码为 LIO_READ，对于写操作来说，应该使用 LIO_WRITE 作为操作码，而 LIO_NOP 意味着空操作。 

使用信号作为 AIO 的通知 

上面讲述的信号作为异步通知的机制在 AIO 中仍然是适用的，为使用信号，使用 AIO 的应用程序同样需要定义信号处理程序，在指定的信号被产生时会触发调用这个处理程序。作为信号上下文的一部分，特定的 aiocb 请求被提供给信号处理函数用来区分 AIO 请求。 

下面代码清单给出了使用信号作为 AIO 异步 I/O 通知机制的例子。

1  /*设置异步 I/O 请求*/
2  void setup_io(...)
3  {
4    int fd;
5    struct sigaction sig_act;
6    struct aiocb my_aiocb;
7    ...
8    /* 设置信号处理函数 */
9    sigemptyset(&sig_act.sa_mask);
10   sig_act.sa_flags = SA_SIGINFO;
11   sig_act.sa_sigaction = aio_completion_handler;
12
13   /* 设置 AIO 请求 */
14   bzero((char*) &my_aiocb, sizeof(struct aiocb));
15   my_aiocb.aio_fildes = fd;
16   my_aiocb.aio_buf = malloc(BUF_SIZE + 1);
17   my_aiocb.aio_nbytes = BUF_SIZE;
18   my_aiocb.aio_offset = next_offset;
19
20   /* 连接 AIO 请求和信号处理函数 */
21   my_aiocb.aio_sigevent.sigev_notify = SIGEV_SIGNAL;
22   my_aiocb.aio_sigevent.sigev_signo = SIGIO;
23   my_aiocb.aio_sigevent.sigev_value.sival_ptr = &my_aiocb;
24
25   /* 将信号与信号处理函数绑定 */
26   ret = sigaction(SIGIO, &sig_act, NULL);
27   ...
28   ret = aio_read(&my_aiocb); /*发出异步读请求*/
29 }
30
31 /*信号处理函数*/
32  void  aio_completion_handler(int  signo,  siginfo_t  *info,  void *context)
33 {
34   struct aiocb *req;
35
36   /* 确定是我们需要的信号*/
37   if (info->si_signo == SIGIO)
38   {
39     req = (struct aiocb*)info->si_value.sival_ptr; /*获得 aiocb*/
40
41     /* 请求的操作完成了吗? */
42     if (aio_error(req) == 0)
43     {
44       /* 请求的操作完成，获取返回值 */
45       ret = aio_return(req);
46     }
47   }
48   return ;
49 }

特别要注意上述代码的第 39 行通过(struct aiocb*)info->si_value.sival_ptr 获得了信号对应的 aiocb。 

使用回调函数作为 AIO 的通知 

除了信号之外，应用程序还可提供一个回调（Callback）函数给内核，以便 AIO 的请求完成后内核调用这个函数。 

一般来说，下层对上层（如内核对应用）的调用都称为“回调”，而上层对下层（如进行 Linux 系统调用）的调用称为“调用”，如图 9.3 所示。

 

代码清单给出了使用回调函数作为 AIO 异步 I/O 请求完成的通知机制的例子。

1  /*设置异步 I/O 请求*/
2  void setup_io(...)
3  {
4    int fd;
5    struct aiocb my_aiocb;
6    ...
7    /* 设置 AIO 请求 */
8    bzero((char*) &my_aiocb, sizeof(struct aiocb));
9    my_aiocb.aio_fildes = fd;
10   my_aiocb.aio_buf = malloc(BUF_SIZE + 1);
11   my_aiocb.aio_nbytes = BUF_SIZE;
12   my_aiocb.aio_offset = next_offset;
13
14   /* 连接 AIO 请求和线程回调函数 */
15   my_aiocb.aio_sigevent.sigev_notify = SIGEV_THREAD;
16   my_aiocb.aio_sigevent.notify_function = aio_completion_handler;
17     /*设置回调函数*/
18   my_aiocb.aio_sigevent.notify_attributes = NULL;
19   my_aiocb.aio_sigevent.sigev_value.sival_ptr = &my_aiocb;
20   ... ret = aio_read(&my_aiocb); //发起 AIO 请求
21 }
22
23 /* 异步 I/O 完成回调函数 */
24 void aio_completion_handler(sigval_t sigval)
25 {
26   struct aiocb *req;
27   req = (struct aiocb*)sigval.sival_ptr;
28
29   /* AIO 请求完成? */
30   if (aio_error(req) == 0)
31   {
32     /* 请求完成，获得返回值 */
33     ret = aio_return(req);
34   }
35
36   return ;
37 }

上述程序在创建 aiocb 请求之后，使用 SIGEV_THREAD 请求了一个线程回调函数来作为通知方法。在回调函数中，通过(struct aiocb*)sigval.sival_ptr 可以获得对应的aiocb 指针，使用 AIO 函数可验证请求是否已经完成。 

proc 文件系统包含了两个虚拟文件，它们可以用来对异步 I/O 的性能进行优化。 

/proc/sys/fs/aio-nr 文件提供了系统范围异步 I/O 请求的数目。 

/proc/sys/fs/aio-max-nr 文件是所允许的并发请求的最大个数，最大个数通常是 64KB，这对于大部分应用程序来说都已经足够了。 

AIO 与设备驱动 

在内核中，每个 I/O 请求都对应于一个 kiocb 结构体，其 ki_filp 成员指向对应的file 指针，通过 is_sync_kiocb()可以判断某 kiocb 是否为同步 I/O 请求，如果返回非真，表示为异步 I/O 请求。 

块设备和网络设备本身是异步的，只有字符设备必须明确表明应支持 AIO。AIO对于大多数字符设备而言都不是必须的，只有极少数设备需要。比如，对于磁带机，由于 I/O 操作很慢，这时候使用异步 I/O 将改善性能。 字符设备驱动程序中file_operations 包含 3 个与 AIO 相关的成员函数，如下所示： 

ssize_t (*aio_read) (struct kiocb *iocb, char *buffer, size_t count, loff_t offset); 

ssize_t (*aio_write) (struct kiocb *iocb, const char *buffer, size_t count, loff_t offset); 

int (*aio_fsync) (struct kiocb *iocb, int datasync); 

aio_read()和 aio_write()与 file_operations 中的 read()和 write()中的 offset 参数不同，它直接传递值，而后者传递的是指针，这是因为 AIO 从来不需要改变文件的位置。 

aio_read()和 aio_write()函数本身不一定完成了读和写操作，它只是发起、初始化读和写操作，下面代码清单给出了驱动程序中aio_read()和aio_write()函数的实现例子。

1  /*异步读*/
2  static ssize_t xxx_aio_read(struct kiocb *iocb, char *buf, size_t count, loff_t
3    pos)
4  {
5    return xxx_defer_op(0, iocb, buf, count, pos);
6  }
7
8  /*异步写*/
9  static ssize_t xxx_aio_write(struct kiocb *iocb, const char *buf, size_t count,
10   loff_t pos)
11 {
12   return xxx_defer_op(1, iocb, (char*)buf, count, pos);
13 }
14
15 /*初始化异步 I/O*/
16 static int xxx_defer_op(int write, struct kiocb *iocb, char *buf, size_t count,
17   loff_t pos)
18 {
19   struct async_work *async_wk;
20   int result;
21   /* 当可以访问 buffer 时进行复制*/
22   if (write)
23     result = xxx_write(iocb->ki_filp, buf, count, &pos);
24   else
25     result = xxx_read(iocb->ki_filp, buf, count, &pos);
26   /* 如果是同步 IOCB，立即返回状态 */
27   if (is_sync_kiocb(iocb))
28     return result;
29
30   /* 否则，推后几μ s 执行 */
31   async_wk = kmalloc(sizeof(*async_wk), GFP_KERNEL);
32   if (async_wk == NULL)
33     return result;
34   /*调度延迟的工作*/
35   async_wk->iocb = iocb;
36   async_wk->result = result;
37   INIT_WORK(&async_wk->work, xxx_do_deferred_op, async_wk);
38   schedule_delayed_work(&async_wk->work, HZ / 100);
39   return  - EIOCBQUEUED; /*控制权返回用户空间*/
40 }
41
42 /*延迟后执行*/
43 static void xxx_do_deferred_op(void *p)
44 {
45   struct async_work *async_wk = (struct async_work*)p;
46   aio_complete(async_wk->iocb, async_wk->result, 0);
47   kfree(async_wk);
48 }

上述代码中最核心的是使用 aync_work（异步工作）结构体将操作延后执行，aync_work 结构体定义如代码清单所示，通过 schedule_delayed_work()函数可以调度其执行。第 46 行对 aio_complete()的调用用于通知内核驱动程序已经完成了操作。

1 struct async_work
2 {
3     struct kiocb *iocb;  //kiocb 结构体指针
4     int result;  //执行结果
5     struct work_struct work; //工作结构体
6 };

就到这里了，O(∩_∩)O~