【韦东山驱动代码移植高通平台之七】通过poll机制读取按键值

通过poll机制读取按键值

poll允许进程在一个或者多个文件描述符上做无阻塞的读取和写入，这些调用也会阻塞进程，但不是阻塞在read或者write系统调用上

需要给file_operations添加poll函数

poll函数里通过poll_wait函数向poll_table添加一个等待队列，返回描述哪个操作可以立即执行的位掩码，比如POLLIN表明可以无阻塞地读取，POLLOUT可以无阻塞地写入

 

示例中是在中断函数里唤醒添加到poll table里的等待队列，然后poll函数会重新执行一下，然后返回立即可以执行的位掩码

 

参考代码

#include <linux/module.h>
#include <linux/init.h>
#include <linux/major.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/types.h>
#include <linux/fs.h>
#include <asm/uaccess.h>
#include <linux/io.h>
#include <linux/sched.h>
//#include <asm/irq.h>
#include <linux/gpio.h>
#include <linux/interrupt.h>
#include <linux/poll.h>

#define TAG "keyvol "
#define INT_GPIO 91

static int key_major;
static struct cdev key_cdev;
static struct class *key_class;
volatile unsigned long *tlmm_gpio_cfg;
volatile unsigned long *tlmm_in_out;
static int key_press = 0;
static int key_value = 0;
static int irq;

static wait_queue_head_t kwait;

static irqreturn_t key_irq_thread(int irq, void *data)
{
	int value;

	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	value = *tlmm_in_out;
	value &= 0x1;
	if (value ) {
		key_value = 0;
	} else {
		key_value = 1;
	}
	wake_up_interruptible(&kwait);
	key_press = 1;

	return IRQ_HANDLED;
}
static ssize_t key_read(struct file *file, char __user *buffer,
		size_t count, loff_t *ppos)
{
	if(count != 1 )
		return -EINVAL;

	if((file->f_flags & O_NONBLOCK) && !key_press)
		return -EAGAIN;

	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	if (!key_press) {
		wait_event_interruptible(kwait, key_press);
	}
	if(key_press) {
		if(!copy_to_user(buffer, &key_value, 1)) {
			printk(TAG"%s key is press\n", __func__);
			key_press = 0;
		} else {
			printk(TAG"%s copy to user error\n", __func__);
			return -EFAULT;
		}
	}

	return count;
}

ssize_t key_write(struct file *file, const char __user *buf, size_t count, loff_t *ppos)
{
	printk(TAG" func:%s line:%d\n", __func__, __LINE__);

	return count;
}

static int key_open(struct inode *inode, struct file *file)
{
	int ret;

	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	irq = gpio_to_irq(INT_GPIO);
	printk(TAG"%s irq is %d\n", __func__, irq);
	ret = request_threaded_irq(irq, NULL, key_irq_thread, IRQF_TRIGGER_FALLING | IRQF_TRIGGER_RISING | IRQF_ONESHOT, "vol_key", NULL);
	printk(TAG"%s ret is %d\n", __func__, ret);

	return ret;
}

static unsigned int key_poll(struct file *file, poll_table *wait)
{
	unsigned int mask;

	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	poll_wait(file, &kwait, wait);
	if (key_press) {
		mask |= POLLIN | POLLRDNORM;
	}

	return mask;
}

static int key_release(struct inode *inode, struct file *file)
{
	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	free_irq(irq, NULL);

	return 0;
}

static const struct file_operations key_ops = {
	.owner = THIS_MODULE,
	.read = key_read,
	.write = key_write,
	.open = key_open,
	.poll = key_poll,
	.release = key_release,
};

static int my_key_init(void)
{
	int retval;
	dev_t dev_id;

	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
	retval = alloc_chrdev_region(&dev_id, 0, 1, "key"); //0,1
	key_major = MAJOR(dev_id);
    printk(TAG"major is %d\n", key_major);
	if (retval < 0) {
		printk(TAG"can't get major number\n");
		goto error;
	}

	cdev_init(&key_cdev, &key_ops);
	retval = cdev_add(&key_cdev, dev_id, 1); //1
	if (retval < 0) {
		printk(TAG"cannot add cdev\n");
		goto cleanup_alloc_chrdev_region;
	}

	key_class = class_create(THIS_MODULE, "key");
	if (IS_ERR(key_class)) {
		printk(TAG "Error creating key class.\n");
		cdev_del(&key_cdev);
		retval = PTR_ERR(key_class);
		goto cleanup_alloc_chrdev_region;
	}

	device_create(key_class, NULL, MKDEV(key_major, 0), NULL, "keyvol");

	tlmm_gpio_cfg = (volatile unsigned long *)ioremap(0x105B000, 8);
	tlmm_in_out = tlmm_gpio_cfg + 1;
	*tlmm_gpio_cfg |= 0x3;

	init_waitqueue_head(&kwait);

	return 0;

cleanup_alloc_chrdev_region:
	unregister_chrdev_region(dev_id, 0);
error:
	return retval;
}

static void key_exit(void)
{
	dev_t dev_id = MKDEV(key_major, 0);

	iounmap(tlmm_gpio_cfg);
	device_destroy(key_class, MKDEV(key_major, 0));
	class_destroy(key_class);
	cdev_del(&key_cdev);  
	unregister_chrdev_region(dev_id, 0);
	printk(TAG" func:%s line:%d\n", __func__, __LINE__);
}

module_init(my_key_init);
module_exit(key_exit);
MODULE_LICENSE("GPL");

 

测试代码

#include <stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include <poll.h>


int main(int argc, char **argv)
{
    int fd;
    char buf;
    struct pollfd pfd;

    if(argc != 2) {
    	printf("Usage: %s [node]\n", argv[0]);
    	exit(-1);
    }
    fd = open(argv[1], O_RDWR | O_NONBLOCK);
    if (fd < 0) {
    	printf("open %s failed\n", argv[1]);
    	exit(-1);
    }

    pfd.fd=fd;
    pfd.events = POLLIN;
    while (1) {  	
        if(poll(&pfd, 1, -1) > 0) {
        	read(fd, &buf, 1);
        	//printf("buf is %d\n", buf);
        	if (buf == 1) {
        		printf("key is press, buf is %d\n", buf);
        	} else () {
        		printf("key is release, buf is %d\n", buf);
            }
        }
    //usleep(500000);
    }

    close(fd);

    return 0;
}

假如编译出的测试程序为testkey_poll，运行测试程序

./testkey_poll /dev/keyvol

该测试程序大部分时间都是在poll函数里休眠，唤醒后会立即无阻塞读取

 

问题

当有数据准备就绪时，可以让驱动程序通知应用程序去读取吗？