字符驱动学习三（轮询 select调用）

chituhuan

   在应用中经常调用selec函数实现多路访问，select函数的实现依赖与驱动的 xxx_poll函数。在xxx_poll函数中，核心的函数是poll_wait函数。在网上查了很多资料，发现轮询的操作依赖于select和poll_wait函数的配合。网上对poll函数是这样解释的：

Poll函数原型：

Unsigned int(*poll)(struct file *filp, struct poll_table *wait);

第一个参数为file结构体指针，第二个参数为轮询表指针。这个函数应该进行以下两项工作

1、对可能引起设备文件状态变化的等待队列调用poll_wait()函数，将对应等待队列添加到          poll_table         

2、返回表示是否能对设备进行无阻塞、写访问的掩码

Poll_wait()函数不会引起阻塞。它所做的工作就是把当前进程添加到wait参数指定的等待列表（poll_table）中.

fifo_poll函数：

1. static unsigned int cfifo_poll(struct file *file, poll_table *wait)
   
2. {
   
3.     unsigned int mask = 0;
   
4.     struct cfifo_t *dev = file->private_data;
   
5. 
   
6.     down(&dev->sem);
   
7.     poll_wait(file, &dev->r_wait, wait);
   
8.     poll_wait(file, &dev->w_wait, wait);
   
9. 
   
10.     if(dev->current_len != 0)
    
11.         mask |= POLLIN | POLLRDNORM;
    
12. 
    
13.     if(dev->current_len != GLOBALMEM_SIZE)
    
14.         mask |= POLLOUT | POLLWRNORM;
    
15. 
    
16.     up(&dev->sem);
    
17.     return mask;
    
18. }

问题是：在我们的使用中select 函数经常是要阻塞的，但是在xxx_poll函数中只是通过poll_wait将读写的等待队列加入 poll_table中.并没有阻塞的地方。

poll_wait的实现，总结一下就是：将select的查询的文件描述符的进程等待队列加入到内核的等待队列的维护中，作用就是供select调用查询。

   在select的实现中我们就可以看到这些等待队列的作用了。在内核的fs/select.c中有poll 和select的

相关实现。select函数的核心是：do_select。在这里实现了阻塞，poll_wait(）函数也是为它服务的。在poll_wait中加入的等待队列也在这里使用。

1. int do_select(int n, fd_set_bits *fds, struct timespec *end_time)
   
2. {
   
3.     ktime_t expire, *to = NULL;
   
4.     struct poll_wqueues table;
   
5.     poll_table *wait;
   
6.     int retval, i, timed_out = 0;
   
7.     unsigned long slack = 0;
   
8. 
   
9.     rcu_read_lock();
   
10.     retval = max_select_fd(n, fds);
    
11.     rcu_read_unlock();
    
12. 
    
13.     if (retval < 0)
    
14.         return retval;
    
15.     n = retval;
    
16. 
    
17.     poll_initwait(&table);
    
18.     wait = &table.pt;
    
19.     if (end_time && !end_time->tv_sec && !end_time->tv_nsec) {
    
20.         wait = NULL;
    
21.         timed_out = 1;
    
22.     }
    
23. 
    
24.     if (end_time && !timed_out)
    
25.         slack = estimate_accuracy(end_time);
    
26. 
    
27.     retval = 0;
    
28.     for (;;) {                                                                  //
    
29.         unsigned long *rinp, *routp, *rexp, *inp, *outp, *exp;
    
30. 
    
31.         inp = fds->in; outp = fds->out; exp = fds->ex;
    
32.         rinp = fds->res_in; routp = fds->res_out; rexp = fds->res_ex;
    
33. 
    
34.         for (i = 0; i < n; ++rinp, ++routp, ++rexp) {
    
35.             unsigned long in, out, ex, all_bits, bit = 1, mask, j;
    
36.             unsigned long res_in = 0, res_out = 0, res_ex = 0;
    
37.             const struct file_operations *f_op = NULL;
    
38.             struct file *file = NULL;
    
39. 
    
40.             in = *inp++; out = *outp++; ex = *exp++;
    
41.             all_bits = in | out | ex;
    
42.             if (all_bits == 0) {
    
43.                 i += __NFDBITS;
    
44.                 continue;
    
45.             }
    
46. 
    
47.             for (j = 0; j < __NFDBITS; ++j, ++i, bit <<= 1) {
    
48.                 int fput_needed;
    
49.                 if (i >= n)
    
50.                     break;
    
51.                 if (!(bit & all_bits))
    
52.                     continue;
    
53.                 file = fget_light(i, &fput_needed);
    
54.                 if (file) {
    
55.                     f_op = file->f_op;
    
56.                     mask = DEFAULT_POLLMASK;
    
57.                     if (f_op && f_op->poll) {
    
58.                         wait_key_set(wait, in, out, bit);
    
59.                         mask = (*f_op->poll)(file, wait);
    
60.                     }
    
61.                     fput_light(file, fput_needed);
    
62.                     if ((mask & POLLIN_SET) && (in & bit)) {
    
63.                         res_in |= bit;
    
64.                         retval++;
    
65.                         wait = NULL;
    
66.                     }
    
67.                     if ((mask & POLLOUT_SET) && (out & bit)) {
    
68.                         res_out |= bit;
    
69.                         retval++;
    
70.                         wait = NULL;
    
71.                     }
    
72.                     if ((mask & POLLEX_SET) && (ex & bit)) {
    
73.                         res_ex |= bit;
    
74.                         retval++;
    
75.                         wait = NULL;
    
76.                     }
    
77.                 }
    
78.             }
    
79.             if (res_in)
    
80.                 *rinp = res_in;
    
81.             if (res_out)
    
82.                 *routp = res_out;
    
83.             if (res_ex)
    
84.                 *rexp = res_ex;
    
85.             cond_resched();
    
86.         }
    
87.         wait = NULL;
    
88.         if (retval || timed_out || signal_pending(current)) //如果条件（文件描述符有改变，超时，信号中断）成立 ，break跳出循环
    
89.             break;
    
90.         if (table.error) {
    
91.             retval = table.error;
    
92.             break;
    
93.         }
    
94. 
    
95.         /*
    
96.          * If this is the first loop and we have a timeout
    
97.          * given, then we convert to ktime_t and set the to
    
98.          * pointer to the expiry value.
    
99.          */
    
100.         if (end_time && !to) {
     
101.             expire = timespec_to_ktime(*end_time);
     
102.             to = &expire;
     
103.         }
     
104. //将进程休眠，设置为可被中断状态 可以被read和write 函数中的wake_up_interruptible(&dev->w_wait);wake_up_int//erruptible(&dev->r_wait);函数唤醒这也是poll_wait函数的作用。唤醒后，继续循环，会调用驱动中的xxx_poll函数//，在 if (retval || timed_out || signal_pending(current))完成select调用。
105. 
     
106.         if (!poll_schedule_timeout(&table, TASK_INTERRUPTIBLE, 
     
107.                      to, slack))
     
108.             timed_out = 1;
     
109.     }
     
110. 
     
111.     poll_freewait(&table);
     
112. 
     
113.     return retval;
     
114. }

测试代码：

2. #include <stdio.h>
3. #include <stdlib.h>
4. #include <fcntl.h>
5. #include <sys/ioctl.h>
6. #include <sys/time.h>
7. #include <sys/types.h>
8. #include <unistd.h>
9. #include <sys/select.h>
10. 
    
11. #define FIFO_CLEAR 0x01
12. #define BUFFER_LEN 20
13. 
    
14. int main(int argc, char *argv[])
15. {
16. int fd;
17. int i;
18. fd_set wfds;
19. fd_set rfds;
20. 
    
21. //fd = open("/dev/cfifo0", O_RDONLY | O_NONBLOCK);
22. fd = open("/dev/cfifo0", O_RDONLY | O_WRONLY);
23. if(fd > 0)
24. ioctl(fd, FIFO_CLEAR, 0);
25. else
26. return -1;
27. 
    
28.    for(i = 0; i < 3; i++)
29. {
30. FD_ZERO(&rfds);
31. FD_ZERO(&wfds);
32. FD_SET(fd, &rfds);
33. // FD_SET(fd, &wfds);
34. printf("[%s %s %d] select start\n \n", __FUNCTION__, __FILE__, __LINE__);
35. select(fd+1, &rfds, NULL, NULL, NULL);
36. printf("[%s %s %d] select end\n \n", __FUNCTION__, __FILE__, __LINE__);
37. 
    
38. if(FD_ISSET(fd, &rfds) )
39. {
40. printf("cfifo can read\n");
41. }
42. 
    
43. // if(FD_ISSET(fd, &wfds) )
44. // {
45. // printf("cfifo can write\n");
46. // }
47. }
48. 
    
49. return 0;
50. }

root@wang:/work/wanghuan/drives# ls

bdev.c  cdev.c  cfifo.c  cfifo.ko  cfifo_poll_test.c  Makefile  modules.order  test

root@wang:/work/wanghuan/drives# ./test &

[1] 4457

root@wang:/work/wanghuan/drives# [main cfifo_poll_test.c 51] select start 

//在这里select阻塞 就是在do_select if (!poll_schedule_timeout(&table, TASK_INTERRUPTIBLE, to, slack))

1.         

root@wang:/work/wanghuan/drives# ls > /dev/cfifo0 

//写入后 cfifo_write的wake_up_interruptible(&dev->r_wait); 唤醒select 进程

[main cfifo_poll_test.c 53] select end

 

cfifo can read

[main cfifo_poll_test.c 51] select start

 

[main cfifo_poll_test.c 53] select end

 

cfifo can read

[main cfifo_poll_test.c 51] select start

 

[main cfifo_poll_test.c 53] select end

 

cfifo can read

[1]+  Done                    ./test

http://cuaib.chinaunix.com/space.php?uid=25014876&do=blog&id=61749这里的一篇文章讲解了poll_wait的实现，

http://zhangwenxin82.blog.163.com/blog/static/1145959562011110111247600/ 这里讲解了select调用。