字符设备驱动之poll机制

poll 机制分析
所有的系统调用，基于都可以在它的名字前加上“sys_”前缀，这就是它在内核中对应
的函数。比如系统调用 open、read、write、poll，与之对应的内核函数为：sys_open、sys_read、
sys_write、sys_poll。

一、内核框架：

对于系统调用 poll 或 select，它们对应的内核函数都是 sys_poll。分析 sys_poll，即可理解 poll
机制。
sys_poll 函数位于 fs/select.c 文件中，代码如下：

它对超时参数稍作处理后，直接调用 do_sys_poll。
do_sys_poll 函数也位于位于 fs/select.c 文件中，我们忽略其他代码：
int do_sys_poll(struct pollfd __user *ufds, unsigned int nfds, s64 *timeout)
{
……
poll_initwait(&table);
……
fdcount = do_poll(nfds, head, &table, timeout);
……
}
poll_initwait 函数非常简单，它初始化一个 poll_wqueues 变量 table：
poll_initwait > init_poll_funcptr(&pwq->pt, __pollwait); > pt->qproc = qproc;
即 table->pt->qproc = __pollwait，__pollwait 将在驱动的 poll 函数里用到。
table 中的 qproc 函数指针指向“__pollwait”。

do_poll 函数位于 fs/select.c 文件中，代码如下：
static int do_poll(unsigned int nfds, struct poll_list *list,
struct poll_wqueues *wait, s64 *timeout)
{
01 ……
02 for (;;) { //有一个死循环.
03 ……
04 if (do_pollfd(pfd, pt)) {
05 count++;
06 pt = NULL;
07 }
08 ……
09 if (count || !*timeout || signal_pending(current))
10 break; //超时跳出。count不为0时，就跳出返回到应用程序中去了。
11 count = wait->error;
12 if (count)
13 break;
14
15 if (*timeout < 0) {
16 /* Wait indefinitely */
17 __timeout = MAX_SCHEDULE_TIMEOUT;
18 } else if (unlikely(*timeout >= (s64)MAX_SCHEDULE_TIMEOUT-1)) {
19 /*
20 * Wait for longer than MAX_SCHEDULE_TIMEOUT. Do it in
21 * a loop
22 */
23 __timeout = MAX_SCHEDULE_TIMEOUT - 1;
24 *timeout -= __timeout;
25 } else {
26 __timeout = *timeout;
27 *timeout = 0;
28 }
29 
30 __timeout = schedule_timeout(__timeout); //休眠。
31 if (*timeout >= 0)
32 *timeout += __timeout;
33 }
34 __set_current_state(TASK_RUNNING);
35 return count;
36 }

分析其中的代码，可以发现，它的作用如下：
从 02 行可以知道，这是个循环，它退出的条件为：
09 行的 3 个条件之一(count 非 0，超时、有信号等待处理)
count 非 0 表示 04 行的 do_pollfd 至少有一个成功。
11、12 行：发生错误
重点在 do_pollfd 函数，后面再分析
第 30 行，让本进程休眠一段时间，注意：应用程序执行 poll 调用后，如果①②的条件不满
足，进程就会进入休眠。那么，谁唤醒呢？除了休眠到指定时间被系统唤醒外，还可以被驱
动程序唤醒──记住这点，这就是为什么驱动的 poll 里要调用 poll_wait 的原因，后面分析。

do_pollfd 函数位于 fs/select.c 文件中，代码如下：
static inline unsigned int do_pollfd(struct pollfd *pollfd, poll_table *pwait)
{
……
if (file->f_op && file->f_op->poll)
mask = file->f_op->poll(file, pwait);
……
}

可见，do_pollfd 函数就是调用我们的驱动程序里注册的 poll 函数。这里有个“f_op”这就是
一个 file_operation

二、驱动程序：

驱动程序里与 poll 相关的地方有两处：一是构造 file_operation 结构时，要定义自己的 poll
函数。二是通过 poll_wait 来调用上面说到的__pollwait 函数，pollwait 的代码如下：

p->qproc 就是“__pollwait 函数”：
在“do_sys_poll”函数中定义一个“poll_wqueues”结构的变量“table”，且在“do_sys_poll”
中一个函数“poll_initwait(&table);”使用了这个变量。而"void poll_initwait(struct poll_wqueues
*pwq)"中调用 了一个函数“init_poll_funcptr(&pwq->pt, __pollwait);”则，最后要看如下操作：
“init_poll_funcptr(&table->pt, __pollwait);”，接着再看“init_poll_funcptr()”函数。
static inline void init_poll_funcptr(poll_table *pt, poll_queue_proc qproc)
{
pt->qproc = qproc;
}p->qproc 就是__pollwait 函数，从它的代码可知，它只是把当前进程挂入我们驱动程序里定
义的一个队列里而已。它的代码如下：
static void __pollwait(struct file *filp, wait_queue_head_t *wait_address,
poll_table *p)
{
struct poll_table_entry *entry = poll_get_entry§;
if (!entry)
return;
get_file(filp);
entry->filp = filp;
entry->wait_address = wait_address;
init_waitqueue_entry(&entry->wait, current);
add_wait_queue(wait_address, &entry->wait);
}

执行到驱动程序的 poll_wait 函数时，进程并没有休眠，我们的驱动程序里实现的 poll 函数
是不会引起休眠的。让进程进入休眠，是前面分析的 do_sys_poll 函数的 30 行“__timeout =
schedule_timeout(__timeout)”。
poll_wait 只是把本进程挂入某个队列，应用程序调用 poll > sys_poll > do_sys_poll >
poll_initwait，do_poll > do_pollfd > 我们自己写的 poll 函数后，再调用 schedule_timeout 进
入休眠。如果我们的驱动程序发现情况就绪，可以把这个队列上挂着的进程唤醒。可见，
poll_wait 的作用，只是为了让驱动程序能找到要唤醒的进程。即使不用 poll_wait，我们的程
序也有机会被唤醒：chedule_timeout(__timeout)，只是要休眠__time_out 这段时间。

总结

1. poll > sys_poll > do_sys_poll > poll_initwait，poll_initwait 函数注册一下回调函数__pollwait，
   它就是我们的驱动程序执行 poll_wait 时，真正被调用的函数。
2. 接下来执行 file->f_op->poll，即我们驱动程序里自己实现的 poll 函数
   它会调用 poll_wait 把自己挂入某个队列，这个队列也是我们的驱动自己定义的；
   它还判断一下设备是否就绪。
3. 如果设备未就绪，do_sys_poll 里会让进程休眠一定时间
4. 进程被唤醒的条件有 2：一是上面说的“一定时间”到了，二是被驱动程序唤醒。驱动程
   序发现条件就绪时，就把“某个队列”上挂着的进程唤醒，这个队列，就是前面通过 poll_wait
   把本进程挂过去的队列。
5. 如果驱动程序没有去唤醒进程，那么 chedule_timeout(__timeou)超时后，会重复 2、3 动
   作，直到应用程序的 poll 调用传入的时间到达。