760
linux驅動開發--字符設備：內核等待隊列
在Linux驅動程序中，可以使用等待隊列（wait queue）來實現阻塞進程的喚醒。等待隊列可以用來同步對係統資源的訪問。
1.定義和初始化隊列頭
wait_queue_head_t wqh;
init_waitqueue_head(wait_queue_head_t *wqh);
2.定義和初始化等待隊列
DECLARE_WAITQUEUE(name, tsk);
3.添加、移除等待隊列
add_wait_queue(wait_queue_head_t *q, wait_queue_t *wait);
remove_wait_queue(wait_queue_head_t *q, wait_queue_t *wait);將等待隊列wait添加到等待隊列頭q執行的等待隊列鏈表中，或者從中刪除。
4.等待事件
wait_event(queue, condition);當condition為真時，立即返回；否則進程進入TASK_UNINTERRUPTIBLE類型的睡眠狀態，並掛在queue指定的等待隊列頭上。
add_wait_queue(queue, conditon);當condition為真時，立即返回；否則進程進入TASK_INTERRUPTIBLE類型的睡眠狀態，並掛在queue指定的等待隊列頭上。
5.喚醒隊列
wake_up(wait_queue_head_t *queue);
wake_up_interruptible(wait_queue_head_t *queue);喚醒由queue指向的等待隊列頭鏈表中所有等待隊列對應的進程。
6.在等待隊列中睡眠
sleep_on(wait_queue_head_t *q);讓進程進入不可中斷的睡眠，並將它放入等待隊列
interruptible_sleep_on(wait_queue_head_t *q);讓進程進入不可中斷的睡眠，並將它進入等待隊列。
內核等待隊列一般使用方法：
a.定義和初始化等待隊列，將進程狀態改變，並將等待隊列添加到等待隊列數據鏈中
b.改變進程狀態的方法：調用set_current_state(state_value)函數、調用set_task_state(task, state_value)函數、直接采用current->state = TASK_INTERRUPTIBLE,類似於賦值語句。
c.通過schedule()調用放棄cpu，調度其他進程執行
d.進程被其它地方喚醒，將等待隊列移出等待隊列頭指向的數據鏈
/**
*Copyright (c) 2013.TianYuan
*All rights reserved.
*
*文件名稱: char_device_driver13.c
*文件標識: 等待隊列
*
*當前版本：1.0
*作者：wuyq
*
*取代版本：xxx
*原作者：xxx
*完成日期：2013-11-29
*/
#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>
MODULE_LICENSE("GPL");
#define CDD_MAJOR 200//cat /proc/devices找一個尚未使用的
#define CDD_MINOR 0
#define CDD_COUNT 2
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_queue_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;
}
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,
};
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);
}
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.
*
*文件名稱: char_device_driver13_test0.c
*文件標識: 此程序運行r，此時發生阻塞，進入等待
*
*當前版本：1.0
*作者：wuyq
*
*取代版本：xxx
*原作者：xxx
*完成日期：2013-11-29
*/
#include <stdio.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
/*手工創建設備節點文件
mknod /dev/cdd c 248 0
*/
int fd = 0;
char rbuf[100];
char wbuf[100] = "nihao!\n";
int main()
{
char ch;
fd = open("/dev/cdd0", O_RDWR);
if(fd < 0){
printf("open failed!\n");
return -1;
}
printf("open successed fd = %d\n", fd);
while(1)
{
printf("starting to test /dev/cdd...\n");
ch = getchar();
getchar();//取走回車
if(ch == 'q'){
break;
}
switch(ch){
case 'r':
memset(rbuf, 0, 100);//清空
read(fd, rbuf, 3);
printf("user space from kernel: %s\n", rbuf);
break;
case 'w':
write(fd, wbuf, strlen(wbuf) );
break;
case 'o':
ioctl(fd, 0, 0);
break;
case 'O':
ioctl(fd, 1, 0);
break;
case 'p':
ioctl(fd, 0, 1);
break;
case 'P':
ioctl(fd, 1, 1);
break;
case 'l':
lseek(fd, 0, SEEK_SET);//移動的文件的開頭
break;
default:
break;
}
sleep(1);
}
close(fd);
return 0;
}
/**
*Copyright (c) 2013.TianYuan
*All rights reserved.
*
*文件名稱: char_device_driver13_test1.c
*文件標識: test0 和 test1兩個測試：此程序執行w，觀察阻塞的r進程被喚醒，完成操作。
*
*當前版本：1.0
*作者：wuyq
*
*取代版本：xxx
*原作者：xxx
*完成日期：2013-11-29
*/
#include <stdio.h>
#include <fcntl.h>
#include <stdlib.h>
#include <string.h>
/*手工創建設備節點文件
mknod /dev/cdd c 248 0
*/
int fd = 0;
char rbuf[100];
char wbuf[100] = "nihao!\n";
int main()
{
char ch;
fd = open("/dev/cdd1", O_RDWR);
if(fd < 0){
printf("open failed!\n");
return -1;
}
printf("open successed fd = %d\n", fd);
while(1)
{
printf("starting to test /dev/cdd...\n");
ch = getchar();
getchar();//取走回車
if(ch == 'q'){
break;
}
switch(ch){
case 'r':
memset(rbuf, 0, 100);//清空
read(fd, rbuf, 3);
printf("user space from kernel: %s\n", rbuf);
break;
case 'w':
write(fd, wbuf, strlen(wbuf) );
break;
case 'o':
ioctl(fd, 0, 0);
break;
case 'O':
ioctl(fd, 1, 0);
break;
case 'p':
ioctl(fd, 0, 1);
break;
case 'P':
ioctl(fd, 1, 1);
break;
case 'l':
lseek(fd, 0, SEEK_SET);//移動的文件的開頭
break;
default:
break;
}
sleep(1);
}
close(fd);
return 0;
}
最後更新：2017-04-03 12:54:47