poll()函数的使用

分类： LINUX

poll函数用于监测多个等待事件，若事件未发生，进程睡眠，放弃CPU控制权，若监测的任何一个事件发生，poll将唤醒睡眠的进程，并判断是什么等待事件发生，执行相应的操作。poll函数退出后，struct pollfd变量的所有值被清零，需要重新设置。

示例是使用poll函数来监测按键的输入

----------------------------------------------------------------------------------------------------------

 

驱动代码：

1. #include <linux/config.h>
2. #include <linux/module.h>
3. #include <linux/kernel.h>
4. #include <linux/fs.h>
5. #include <linux/init.h>
6. #include <linux/devfs_fs_kernel.h>
7. #include <linux/miscdevice.h>
8. #include <linux/delay.h>
9. #include <asm/irq.h>
10. #include <asm/arch/regs-gpio.h>
11. #include <asm/hardware.h>
12. #include <linux/cdev.h>
13. #include <linux/mm.h>
14. #include <linux/interrupt.h>
15. #include <linux/poll.h>
16. #include <asm/uaccess.h>
17. #include <asm/ioctl.h>
18. #include <asm/arch/regs-irq.h>
19. #include <asm/io.h>
21. #define key S3C2410_GPF0
22. #define key_irq IRQ_EINT0 //IRQ_EINT0是中断号
23. #define key_cfg S3C2410_GPF0_EINT0 //设置为外部中断功能
25. #define DEVICE_NAME "key" //注意加上双引号
26. #define DEVICE_MAJOR major
27. #define DEVICE_MINOR 0
29. static dev_t dev; //dev_t类型用于存放主设备号和次设备号
30. static int major;
32. struct cdev *p_cdev; //声明一个指向字符设备结构的指针
34. static int key_event=0; //唤醒中断的条件标记，1时满足唤醒条件，静态全局变量
35. static int key_value=1; //按键键值
36. static DECLARE_WAIT_QUEUE_HEAD(wq); //调用宏定义，静态创建一个名为wq的等待队列
38. static void key_interrupt(void) //中断处理函数，注册中断时已注册了中断程序的入口地址
39. {
40.  key_value=s3c2410_gpio_getpin(key);
41.  key_event=1; //唤醒标记置位，表示条件达到，可以唤醒进程继续执行
42.  wake_up_interruptible(&wq); //调用宏定义，唤醒标记置位后调用此函数，&wq是队列入口地址
43. }
45. static int key_read(struct file *filp,char __user *buff,size_t count,loff_t *offp)
46. {
47. // wait_event_interruptible(wq,key_event); //若key_event为0，从此处将进程放入wq等待队列休眠，等待中断；key_event==1时，此宏不执行操作
48. //调用poll的时候来等待，这里可以不用wait_event_interrupt()
49.  key_value=s3c2410_gpio_getpin(key);
50.  copy_to_user(buff,&key_value,sizeof(key_event)); //将&key_value地址的值从内核空间复制到用户空间
51.  key_event=0; //完成中断操作，将唤醒标记清零，继续休眠
52.  return 0;
53. }
55. static unsigned int key_poll(struct file *filp,poll_table *wait)
56. {
57.  unsigned int mask=0; //用来记录发生的事件，以unsigned int类型返回
58.  poll_wait(filp,&wq,wait); //将当前进程添加到wq等待队列中
59.  if(key_event==1)mask|=POLLIN|POLLRDNORM; //中断事件发生，这时有数据可读，在mask中标记是可读事件发生
60.  return mask; //返回事件记录，返回0则表示等待事件超时
61. }
63. //设置寄存器，申请中断号等在open函数中完成
64. static int key_open(struct inode *inode,struct file *filp)
65. {
66.  int ret;
67.  s3c2410_gpio_cfgpin(key,key_cfg); //设置引脚功能
68.  s3c2410_gpio_pullup(key,1); //第二个参数1为禁止内部上拉
69.  ret=request_irq(key_irq,(void *)key_interrupt,SA_INTERRUPT,DEVICE_NAME,NULL); //注册中断，中断不共享时最后一个参数为NULL
70.  if(ret) {
71.   printk("Could not register interrupt\n");
72.   return ret;
73.  }
74.  set_irq_type(key_irq,IRQT_BOTHEDGE); //设置中断方式为双边触发
75.  return 0;
76. }
78. static int key_close(struct inode *inode,struct file *filp)
79. {
80.  free_irq(key_irq,NULL); //中断无共享时第二个参数为NULL
81.  return 0;
82. }
84. static struct file_operations key_fops={
85.  .owner=THIS_MODULE,
86.  .open=key_open,
87.  .release=key_close,
88.  .read=key_read,
89.  .poll=key_poll,
90. };
92. int key_init(void)
93. {
94.  int ret;
95.  ret=alloc_chrdev_region(&dev,DEVICE_MINOR,1,DEVICE_NAME); //采用主设备号动态分配
96.  if(ret<0){
97.   printk("Register /dev/key failed!\n");
98.   return ret;
99.  }
100.  else printk("Register /dev/key successfully!\n");
101.  major=MAJOR(dev); //取得分配到的主设备号
102.  p_cdev=cdev_alloc(); //申请一个字符设备结构并返回指向它的指针
103.  cdev_init(p_cdev,&key_fops); //相当于p_cdev->ops=&key_fops
104.  p_cdev->owner=THIS_MODULE;
105.  ret=cdev_add(p_cdev,dev,1); //向系统添加这个字符设备
106.  if(ret<0){
107.   printk("Add cdev failed!\n");
108.   return ret;
109.  }
110.  devfs_mk_cdev(dev,S_IFCHR|S_IRUSR|S_IWUSR|S_IRGRP,DEVICE_NAME);
111.  return 0;
112. }
114. void key_exit(void)
115. {
116.  unregister_chrdev_region(dev,1);
117.  cdev_del(p_cdev); //删除字符设备
118.  devfs_remove(DEVICE_NAME);
119.  printk("Device unregister!\n");
120. }
122. MODULE_LICENSE("GPL");
123. MODULE_AUTHOR("HJW");
124. module_init(key_init);
125. module_exit(key_exit);
127. ---------------------------------------------------------------------------------------------------
129. 测试程序代码：
130. #include <stdio.h>
131. #include <stdlib.h>
132. #include <unistd.h>
133. #include <sys/ioctl.h>
134. #include <sys/types.h>
135. #include <sys/stat.h>
136. #include <fcntl.h> /*文件控制*/
137. #include <sys/select.h>
138. #include <sys/time.h> /*时间方面的函数*/
139. #include <errno.h> /*有关错误方面的宏*/
140. #include<sys/poll.h> //poll()
141. #include<fcntl.h>
142. #include<string.h> //memset()
144. int main(void)
145. {
146.  int fd,key_value,ret;
147.  struct pollfd event; //创建一个struct pollfd结构体变量，存放文件描述符、要等待发生的事件
148.  fd=open("/dev/key",O_RDWR);
149.  if(fd<0){
150.   perror("open /dev/key error!\n");
151.   exit(1);
152.  }
153.  printf("open /dev/key sucessfully!\n");
154.  while(1){ //poll结束后struct pollfd结构体变量的内容被全部清零，需要再次设置
155.   memset(&event,0,sizeof(event)); //memst函数对对象的内容设置为同一值
156.   event.fd=fd; //存放打开的文件描述符
157.   event.events=POLLIN; //存放要等待发生的事件
158.   ret=poll((struct pollfd *)&event,1,5000); //监测event，一个对象，等待5000毫秒后超时,-1为无限等待
160. //判断poll的返回值，负数是出错，0是设定的时间超时，整数表示等待的时间发生
161.   if(ret<0){
162.    printf("poll error!\n");
163.    exit(1);
164.   }
165.   if(ret==0){
166.    printf("Time out!\n");
167.    continue;
168.   }
169.   if(event.revents&POLLERR){ //revents是由内核记录的实际发生的事件，events是进程等待的事件
170.    printf("Device error!\n");
171.    exit(1);
172.   }
173.   if(event.revents&POLLIN){
174.    read(fd,&key_value,sizeof(key_value));
175.    printf("Key value is '%d'\n",key_value);
176.   }
177.  }
178.  close(fd);
179.  return 0;
180. }