347 阿裏雲技術社區[雲棲]

linux驅動開發--字符設備：設備輪詢操作select/poll

select係統調用
用於多路監控，當沒有一個文件滿足要求時，select調用將引起進程阻塞
設備輪詢操作，對應select係統調用
unsigned int(*poll)(struct file *filp, struct poll_table_struct *wait);
unsigned int(*poll)(struct file *filp, poll_table *wait);
filp:文件指針
wait:輪詢表指針
poll設備操作的任務
調用poll_wait()函數將等待隊列添加到poll_table輪詢表

返回描述設備是否可讀或可寫的掩碼

poll_table數據類型定義
在<linux/poll.h>中

typedef struct poll_table_struct{
	poll_queue_proc qproc;
}poll_table;
typedef void(*poll_queue_proc)(struct file *, wait_queue_head_t *, struct poll_table_struct *);

poll_wait()函數
在<linux/poll.h>中

void poll_wait(struct file *filp, wait_queue_head_t *wait_address, poll_table *p);
filp:執行操作的設備文件指針
wait_address:設備驅動的等待隊列頭
p:內核傳遞的輪詢表指針

poll_wait()函數調用不會引起阻塞，它僅僅將當前進程添加到wait參數指定的等待隊列數據鏈中

poll操作返回值
通常返回下列定義“或”的結果
POLLIN 設備可無阻塞讀
POLLOUT 設別可無阻塞寫
POLLRDNORM 數據可讀
POLLWRNORM 數據可寫
設備可讀通常返回： POLLIN | POLLRDNORM
設備可寫通常返回： POLLOUT | POLLWRNORM

poll操作一般結構

xxx_poll()
{
unsigned int mask = 0;//定義返回值
...
//調用poll_wait()把進程添加到輪詢表
...
if(device is ready for read){
mask = POLLIN | POLLRDNORM;
}
...
return mask;
}

/**
*Copyright (c) 2013.TianYuan
*All rights reserved.
*
*文件名稱: cdd_poll.c
*文件標識: 按鍵1的中斷
*
*當前版本：1.0
*作者：wuyq 
*
*取代版本：xxx
*原作者：xxx
*完成日期：2013-12-05
*/
#include <linux/init.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/cdev.h>
#include <linux/device.h>
#include <linux/slab.h>
#include <asm/uaccess.h>

#include <asm/gpio.h>
#include <plat/gpio-cfg.h>
#include <linux/spinlock_types.h>
#include <linux/sched.h>
#include <linux/poll.h>



MODULE_LICENSE("GPL");
#define CDD_MAJOR	200//cat /proc/devices找一個尚未使用的
#define CDD_MINOR	0
#define CDD_COUNT	10
dev_t dev = 0;
u32 cdd_major = 0;
u32 cdd_minor = 0;

struct class *dev_class = NULL;
#define BUF_SIZE	100
struct cdd_cdev{
	struct cdev cdev;
	struct device *dev_device;
	u8 led;
	
	char kbuf[BUF_SIZE];
	
	u32 data_len;//記錄緩衝區中已經寫入數據的長度
	//定義等待隊列頭
	wait_quue_head_t wqh;
};

struct cdd_cdev *cdd_cdevp = NULL;

unsigned long led_gpio_table[2] = {
	S5PV210_GPC1(3),//數字
	S5PV210_GPC1(4),
};

int cdd_open(struct inode* inode, struct file *filp)
{
	struct cdd_cdev *pcdevp = NULL;
	printk("enter cdd_open!\n");

	pcdevp = container_of(inode->i_cdev, struct cdd_cdev, cdev);
	printk("led = %d\n", pcdevp->led);
	
	/*獲取信號量*/
	//down獲取信號量不成功，會導致進程睡眠（第3個進程的時候）
	//down(&pcdevp->sem_open);
	if(down_interruptible(&pcdevp->sem_open)<0){
		return -1;
	}
	filp->private_data = pcdevp;
	//申請gpio管腳
	gpio_request(led_gpio_table[0], "GPC1_3");
	gpio_request(led_gpio_table[1], "GPC1_4");
	
	return 0;
}

int cdd_read(struct file *filp, char __user *buf, size_t count, loff_t *offset)
{
	int ret = 0;
	u32 pos = *offset;
	u32 cnt = count;
	
	struct cdd_cdev *cdevp = filp->private_data;
#if 0
	//定義並初始化一個等待隊列
	DECLARE_WAITQUEUE(wq, current);
	//將等待隊列添加到wqh指向的鏈表
	add_wait_queue(&pcdevp->wqh, &wq);
	//判斷設備有沒有數據供用戶空間讀,假設led不為0，表示有數據供用戶空間讀取
	if(pcdevp->led == 0){
		printk("no data for reading! sleep...\n");
		//設置當前線程為睡眠狀態
		set_current_state(TASK_INTERRUPTIBLE);
		schedule();//內核調度cpu的算法
		printk("have data for reading!\n");
	}
	//從指定的鏈表中刪除等待隊列
	remove_wait_queue(&pcdevp->wqh, &wq);
#endif
	wait_event_interruptible(pcdevp->wqh, pcdevp->led != 0); 
	
	//printk("enter cdd_read!\n");
	if(cnt > (cdevp->data_len-pos) ){
		cnt = cdevp->data_len - pos;
	}
	
	ret = copy_to_user(buf, cdevp->kbuf+pos, cnt);
	//printk("kernel kbuf content:%s\n", cdevp->kbuf);
	*offset += cnt;
	
	pcdevp->led = 0;
	
	return ret;
}

int cdd_write(struct file *filp, const char __user *buf, size_t count, loff_t *offset)
{
	int ret = 0;
	struct cdd_cdev *cdevp = filp->private_data;
	u32 pos = *offset;
	u32 cnt = count;
	
	//printk("enter cdd_write!\n");
	if(cnt > (BUF_SIZE - pos) ){
		cnt = BUF_SIZE - pos;
	}
	ret = copy_from_user(cdevp->kbuf+pos, buf, cnt);
	*offset += cnt;
	if(*offset > cdevp->data_len){
		cdevp->data_len = *offset;
	}
	pcdevp->led = 1;
	//喚醒等待隊列頭中的一個等待隊列
	wake_up_interruptible(&pcdevp->wqh);
	
	
	return ret;
}

int cdd_ioctl(struct inode *inode, struct file *filp, unsigned int cmd, unsigned long data)
{
	//printk("enter cdd_ioctl!\n");
	switch(cmd){
		case 1://點亮燈
			//設置管腳為輸出功能
			//參數：1.要設置的管腳編號2.默認的輸出值 0低電平1高電平
			gpio_direction_output(led_gpio_table[data], 0);
			//禁止內部上拉
			s3c_gpio_setpull(led_gpio_table[data], SEC_GPIO_PULL_NONE);
			//設置輸出值
			gpio_set_value(led_gpio_table[data], 1);
			
			break;
		case 0://熄滅燈
			
			//設置管腳為輸出功能
			//參數：1.要設置的管腳編號2.默認的輸出值 0低電平1高電平
			gpio_direction_output(led_gpio_table[data], 0);
			//禁止內部上拉
			s3c_gpio_setpull(led_gpio_table[data], SEC_GPIO_PULL_NONE);
			//設置輸出值
			gpio_set_value(led_gpio_table[data], 0);
			
			break;
		default:
			return -EINVAL;
	}
	
	
	return 0;
}

int cdd_release(struct inode *inode, struct file *filp)
{
	struct cdd_cdev *pcdevp = filp->private_data;
	printk("enter cdd_release!\n");
	
	gpio_free(led_gpio_table[0]);
	gpio_free(led_gpio_table[1]);
	up(&pcdevp->sem_open);
	return 0;
}

loff_t cdd_llseek(struct file *filp, loff_t offset, int whence)
{
	struct cdd_cdev *pcdevp = filp->private_data;
	loff_t newpos = 0;
	switch(whence){
		case SEEK_SET:
			newpos = offset;
			break;
		case SEEK_CUR:
			newpos = filp->f_pos + offset;
			break;
		case SEEK_END:
			newpos = pcdevp->data_len + offset;
			break;
		default:
			return -EINVAL;//無效的參數
	}
	
	if( newpos<0 || newpos>= BUF_SIZE ){
		return -EINVAL;
	}
	filp->f_pos = newpos;
	return newpos;
}

unsigned int cdd_poll(struct file *filp, struct poll_table_struct *wait)
{
	unsigned int mask = 0;
	struct cdd_cdev pcdevp = filp->private_data;
	printk("enter cdd_poll!\n");
	poll_wait(filp, &pcdep->wqh, wait);
	//led不為0即可讀
	if(pcdevp->led){
		mask = POLLIN | POLLRDNORM;
	}
	return mask;
}

struct file_operations cdd_fops = {
	.owner = THIS_MODULE,
	.open = cdd_open,
	.read = cdd_read,
	.write = cdd_write,
	.ioctl = cdd_ioctl,
	.release = cdd_release,
	.llseek = cdd_llseek,
	.poll = cdd_poll,
};

irqreturn_t cdd_isr(int irq, void *dev_id)
{
	printk("occur up key press or release!\n");
	return IRQ_HANDLED;
}

int __init cdd_init(void)
{
	int ret = 0;
	int i = 0;
	
	if(cdd_major){
		dev = MKDEV(CDD_MAJOR, CDD_MINOR);//生成設備號
		//注冊設備號;1、要注冊的起始設備號2、連續注冊的設備號個數3、名字
		ret = register_chrdev_region(dev, CDD_COUNT, "cdd_demo");
	}else{
		// 動態分配設備號
		ret = alloc_chrdev_region(&dev, cdd_minor, CDD_COUNT, "cdd_demo02");
	}
	
	if(ret < 0){
		printk("register_chrdev_region failed!\n");
		goto failure_register_chrdev;
	}
	//獲取主設備號
	cdd_major = MAJOR(dev);
	printk("cdd_major = %d\n", cdd_major);
	
	cdd_cdevp = kzalloc(sizeof(struct cdd_cdev)*CDD_COUNT, GFP_KERNEL);
	if(IS_ERR(cdd_cdevp)){
		printk("kzalloc failed!\n");
		goto failure_kzalloc;
	}
	/*創建設備類*/
	dev_class = class_create(THIS_MODULE, "cdd_class");
	if(IS_ERR(dev_class)){
		printk("class_create failed!\n");
		goto failure_dev_class;
	}
	for(i=0; i<CDD_COUNT; i++){
		/*初始化cdev*/
		cdev_init(&(cdd_cdevp[i].cdev), &cdd_fops);
		/*添加cdev到內核*/
		cdev_add(&(cdd_cdevp[i].cdev), dev+i, 1);
		
		/* “/dev/xxx” */
		device_create(dev_class, NULL, dev+i, NULL, "cdd%d", i);
		
		cdd_cdevp[i].led = i;
		//初始化等待隊列頭
		init_waitqueue_head(&cdd_cdevp[i].wqh);
	}
	
	//注冊中斷
	request_irq(IRQ_EINT0, cdd_isr, IRQF_TRIGGER_FALLING  | IRQF_TRIGGER_RISING, "interrupt_demo", NULL);
	
	return 0;
failure_dev_class:
	kfree(cdd_cdevp);
failure_kzalloc:
	unregister_chrdev_region(dev, CDD_COUNT);
failure_register_chrdev:
	return ret;
}

void __exit cdd_exit(void)
{
/*逆序消除*/
	int i = 0;
	for(; i < CDD_COUNT; i++){
		device_destroy(dev_class, dev+i);
		cdev_del(&(cdd_cdevp[i].cdev));
		//cdev_del(&((cdd_cdevp+i)->cdev));
	}
	class_destroy(dev_class);
	kfree(cdd_cdevp);
	unregister_chrdev_region(dev, CDD_COUNT);
	
}	

module_init(cdd_init);
module_exit(cdd_exit);

/**
*Copyright (c) 2013.TianYuan
*All rights reserved.
*
*文件名稱: test_select.c
*文件標識: select
*
*當前版本：1.0
*作者：wuyq 
*
*取代版本：xxx
*原作者：xxx
*完成日期：2013-12-05
*/
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>

int fd0 = 0;
int fd1 = 0;
int fd2 = 0 ;
fd_set rfds;// 讀文件描述符集合
int ret = 0;
struct timeval tv;

int main()
{
	fd0 = open("/dev/cdd0", O_RDWR);
	fd1 = open("/dev/cdd1", O_RDWR);
	fd2 = open("/dev/cdd2", O_RDWR);
	while(1){
		//清空rfds
		FD_ZERO(&rfds);
		//將fdn加入到frds集合
		FD_SET(fd0, &rfds);
		FD_SET(fd1, &rfds);
		FD_SET(fd2, &rfds);
		
		//超時時間5s
		tv.tv_sec = 5;
		tv.tv_usec = 0;
		
		ret = select(fd2+1, &rfds, NULL, NULL, &tv);
		if(ret < 0){
			printf("select failed!\n");
			return -1
		}else if(ret = 0){
			printf("select timeout!\n");
		}else{
			if(FD_ISSET(fd0, &rfds)){
				printf("/dev/cdd0 is ready for reading!\n");
			}
			if(FD_ISSET(fd1, &rfds)){
				printf("/dev/cdd1 is ready for reading!\n");
			}
			if(FD_ISSET(fd2, &rfds)){
				printf("/dev/cdd2 is ready for reading!\n");
			}
		}
		sleep(1);
	}
	
	return 0;
}

最後更新：2017-04-03 12:54:47

linux驅動開發--字符設備：設備輪詢操作select/poll

上一篇： C#之析構函數與構造函數

下一篇：開發中的版本問題(2)—配置tomcat使用特定的jdk版本

相關內容

熱門內容

最新內容

linux驅動開發--字符設備：設備輪詢操作select/poll

上一篇： C#之析構函數與構造函數

下一篇： 開發中的版本問題(2)—配置tomcat使用特定的jdk版本

相關內容

熱門內容

最新內容

下一篇：開發中的版本問題(2)—配置tomcat使用特定的jdk版本