七、设备驱动中的阻塞与非阻塞 IO(二)
7.2 轮询
7.2.1 介绍
在用户程序中的 select() 和 poll() 函数最终会使设备驱动中的 poll() 函数被执行。
设备驱动程序中的轮询函数原型:
/** 用于询问设备是否可以非阻塞的读写。当询问条件未触发时,用户空间进行 select() 和 poll() 将引起进程阻塞 */
unsigned int (*poll) (struct file *filp, struct poll_table_struct *wait);
- wait:轮询表指针。
- 这个函数应该进行两项工作:
- 对可能引起设备文件状态变化的等待队列调用 poll_wait() 函数,将对应的等待队列头部添加到 poll_table 中
- 返回表示是否能对设备进行无阻塞读、写访问的掩码
7.2.2 globalfifo 支持轮询
globalfifo.c
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/init.h>
#include <linux/cdev.h>
#include <linux/slab.h>
#include <linux/uaccess.h>
#include <linux/mutex.h>
#include <linux/wait.h>
#include <linux/sched/signal.h> ///< 内核>5.0 使用
//#include <linux/sched.h>
#include <linux/poll.h> #define GLOBALFIFO_SIZE 0x1000
#define MEM_CLEAR 0X1
//#define GLOBALFIFO_MAGIC 'g'
//#define MEM_CLEAR _IO(GLOBALFIFO_MAGIC, 0)
#define GLOBALFIFO_MAJOR 230
#define DEVICE_NUMBER 10 static int globalfifo_major = GLOBALFIFO_MAJOR;
module_param(globalfifo_major, int, S_IRUGO); struct globalfifo_dev {
struct cdev cdev;
/**
* 目前 FIFO 中有效数据长度
* current_len = 0, 表示 FIFO 为空
* current_len = GLOBALFIFO_SIZE, 表示 FIFO 满
*/
unsigned int current_len;
unsigned char mem[GLOBALFIFO_SIZE];
struct mutex mutex;
wait_queue_head_t r_wait; ///< 读等待队列头
wait_queue_head_t w_wait; ///< 写等待队列头
}; struct globalfifo_dev *globalfifo_devp; /**
* 这里涉及到私有数据的定义,大多数遵循将文件私有数据 pirvate_data 指向设备结构体,
* 再用 read write llseek ioctl 等函数通过 private_data 访问设备结构体。
* 对于此驱动而言,私有数据的设置是在 open 函数中完成的
*/
static int globalfifo_open(struct inode *inode, struct file *filp)
{
/**
* NOTA:
* container_of 的作用是通过结构体成员的指针找到对应结构体的指针。
* 第一个参数是结构体成员的指针
* 第二个参数是整个结构体的类型
* 第三个参数为传入的第一个参数(即结构体成员)的类型
* container_of 返回值为整个结构体指针
*/
struct globalfifo_dev *dev = container_of(inode->i_cdev, struct globalfifo_dev, cdev);
filp->private_data = dev;
return ;
} static int globalfifo_release(struct inode *inode, struct file *filp)
{
return ;
} static long globalfifo_ioctl(struct file *filp, unsigned int cmd, unsigned long arg)
{
struct globalfifo_dev *dev = filp->private_data; switch(cmd){
case MEM_CLEAR:
mutex_lock(&dev->mutex);
memset(dev->mem, , GLOBALFIFO_SIZE);
printk(KERN_INFO "globalfifo is set to zero\n");
mutex_unlock(&dev->mutex);
break;
default:
return -EINVAL;
} return ;
} static unsigned int globalfifo_poll(struct file *filp, struct poll_table_struct *wait)
{
unsigned int mask = ;
struct globalfifo_dev *dev = filp->private_data; mutex_lock(&dev->mutex); poll_wait(filp, &dev->r_wait, wait);
poll_wait(filp, &dev->w_wait, wait); if(dev->current_len != ) {
mask |= POLLIN | POLLRDNORM;
} if(dev->current_len != GLOBALFIFO_SIZE){
mask |= POLLIN | POLLWRNORM;
} mutex_unlock(&dev->mutex);
return mask;
} static loff_t globalfifo_llseek(struct file *filp, loff_t offset, int orig)
{
loff_t ret = ;
switch(orig) {
case : /** 从文件开头位置 seek */
if(offset < ){
ret = -EINVAL;
break;
}
if((unsigned int)offset > GLOBALFIFO_SIZE){
ret = -EINVAL;
break;
}
filp->f_pos = (unsigned int)offset;
ret = filp->f_pos;
break;
case : /** 从文件当前位置开始 seek */
if((filp->f_pos + offset) > GLOBALFIFO_SIZE){
ret = -EINVAL;
break;
}
if((filp->f_pos + offset) < ){
ret = -EINVAL;
break;
}
filp->f_pos += offset;
ret = filp->f_pos;
break;
default:
ret = -EINVAL;
break;
} return ret;
} static ssize_t globalfifo_write(struct file *filp, const char __user *buf, size_t size, loff_t *ppos)
{
unsigned int count = size;
int ret = ;
struct globalfifo_dev *dev = filp->private_data; DECLARE_WAITQUEUE(wait, current); ///< 将当前进程加入到 wait 等待队列
mutex_lock(&dev->mutex);
add_wait_queue(&dev->w_wait, &wait); ///< 添加等待队列元到读队列头中 /** 判断设备是否可写 */
while(dev->current_len == GLOBALFIFO_SIZE){
/** 若是非阻塞访问, 设备忙时, 直接返回 -EAGAIN */
if(filp->f_flags & O_NONBLOCK) {
ret = -EAGAIN;
goto out;
} __set_current_state(TASK_INTERRUPTIBLE); ///<改变进程状态为睡眠
schedule();
if(signal_pending(current)){ ///< 因为信号而唤醒
ret = -ERESTARTSYS;
goto out2;
}
} if(count > GLOBALFIFO_SIZE - dev->current_len)
count = GLOBALFIFO_SIZE - dev->current_len; if(copy_from_user(dev->mem + dev->current_len, buf, count)){
ret = -EFAULT;
goto out;
} else {
dev->current_len += count;
printk(KERN_INFO "written %u bytes(s), current len:%d\n", count, dev->current_len); wake_up_interruptible(&dev->r_wait); ///< 唤醒读等待队列
ret = count;
}
out:
mutex_unlock(&dev->mutex);
out2:
remove_wait_queue(&dev->w_wait, &wait); ///< 移除等待队列
set_current_state(TASK_RUNNING);
return ret;
} /**
* *ppos 是要读的位置相对于文件开头的偏移,如果该偏移大于或等于 GLOBALFIFO_SIZE,意味着已经独到文件末尾
*/
static ssize_t globalfifo_read(struct file *filp, char __user *buf, size_t size, loff_t *ppos)
{
unsigned int count = size;
int ret = ;
struct globalfifo_dev *dev = filp->private_data; DECLARE_WAITQUEUE(wait, current); ///< 将当前进程加入到 wait 等待队列
mutex_lock(&dev->mutex);
add_wait_queue(&dev->r_wait, &wait); ///< 添加等待队列元到读队列头中 /** 等待 FIFO 非空,即判断设备是否可读 */
while(dev->current_len == ) {
/** 若是非阻塞访问, 设备忙时, 直接返回 -EAGAIN */
/** filp->f_flags 是用户空间 */
if(filp->f_flags & O_NONBLOCK) {
ret = -EAGAIN;
goto out;
} /**
* 阻塞访问,调度其他进程执行
* FIFO 为空的情况下,读进程阻塞,必须依赖写进程往 FIFO 里面写东西唤醒它;
* 但写的进程为了 FIFO,它必须拿到这个互斥体来访问 FIFO 这个临界资源;
* 如果读进程把自己调度出去之前不释放这个互斥体,那么读写进程之间就死锁了
*/
__set_current_state(TASK_INTERRUPTIBLE); ///<改变进程状态为睡眠
mutex_unlock(&dev->mutex);
schedule();
if(signal_pending(current)){ ///< 因为信号而唤醒
ret = -ERESTARTSYS;
goto out2;
} mutex_lock(&dev->mutex);
} /** 要读取的字节数大于设备文件中的有效数据长度 */
if(count > dev->current_len)
count = dev->current_len; /** 从用户空间拷贝数据 */
if(copy_to_user(buf, dev->mem, count)) {
ret = -EFAULT;
goto out;
} else {
/** FIFO 中数据前移 */
memcpy(dev->mem, dev->mem + count, dev->current_len - count);
dev->current_len -= count; ///< 有效数据长度减少
printk(KERN_INFO "read %u bytes(s), current_len: %d\n", count, dev->current_len); wake_up_interruptible(&dev->w_wait); ///< 唤醒写等待队列 ret = count;
}
out:
mutex_unlock(&dev->mutex);
out2:
remove_wait_queue(&dev->r_wait, &wait); ///< 移除等待队列
set_current_state(TASK_RUNNING);
return ret;
} static const struct file_operations globalfifo_fops = {
.owner = THIS_MODULE,
.poll = globalfifo_poll,
.llseek = globalfifo_llseek,
.read = globalfifo_read,
.write = globalfifo_write,
.unlocked_ioctl = globalfifo_ioctl,
.open = globalfifo_open,
.release = globalfifo_release,
}; /**
* @brief globalfifo_setup_cdev
*
* @param dev
* @param index 次设备号
*/
static void globalfifo_setup_cdev(struct globalfifo_dev *dev, int index)
{
int err;
int devno = MKDEV(globalfifo_major, index); /** 使用 cdev_init 即是静态初始化了 cdev */
cdev_init(&dev->cdev, &globalfifo_fops);
dev->cdev.owner = THIS_MODULE; /** 设备编号范围设置为1,表示我们只申请了一个设备 */
err = cdev_add(&dev->cdev, devno, );
if(err)
printk(KERN_NOTICE "Error %d adding globalfifo%d\n", err, index);
} static int __init globalfifo_init(void)
{
int ret;
int i;
dev_t devno = MKDEV(globalfifo_major, ); if(globalfifo_major)
ret = register_chrdev_region(devno, DEVICE_NUMBER, "globalfifo");
else {
ret = alloc_chrdev_region(&devno, , DEVICE_NUMBER, "globalfifo");
globalfifo_major = MAJOR(devno);
} if(ret < )
return ret; globalfifo_devp = kzalloc(sizeof(struct globalfifo_dev), GFP_KERNEL);
if(!globalfifo_devp){
ret = -ENOMEM;
goto fail_malloc;
} for(i = ; i < DEVICE_NUMBER; i++){
globalfifo_setup_cdev(globalfifo_devp + i, i);
} mutex_init(&globalfifo_devp->mutex); /** 初始化读写等待队列 */
init_waitqueue_head(&globalfifo_devp->r_wait);
init_waitqueue_head(&globalfifo_devp->w_wait); fail_malloc:
unregister_chrdev_region(devno, );
return ret;
} static void __exit globalfifo_exit(void)
{
int i;
for(i = ; i < DEVICE_NUMBER; i++) {
cdev_del(&(globalfifo_devp + i)->cdev);
}
kfree(globalfifo_devp);
unregister_chrdev_region(MKDEV(globalfifo_major, ), DEVICE_NUMBER);
} module_init(globalfifo_init);
module_exit(globalfifo_exit);
epoll_test.c
} epfd = epoll_create();
if(epfd < ) {
perror("epoll_create()");
return ;
} bzero(&ev_globalfifo, sizeof(struct epoll_event));
ev_globalfifo.events = EPOLLIN | EPOLLPRI;
err = epoll_ctl(epfd, EPOLL_CTL_ADD, fd, &ev_globalfifo);
if(err < ) {
perror("epoll_ctl()");
return ;
}
err = epoll_wait(epfd, &ev_globalfifo, , );
if(err < ) {
perror("epoll_wait()");
} else if(err == ) {
printf("No data input in FIFO within 15 seconds.\n");
} else {
printf("FIFO is not empty.\n");
} err = epoll_ctl(epfd, EPOLL_CTL_DEL, fd, &ev_globalfifo);
if(err < ) {
perror("epoll_ctl()");
return ;
} return; }
Makefile
.PHONY:all clean test TEST_SRC := poll_test.c
TEST_TARGET := poll_test ifneq ($(KERNELRELEASE),) obj-m := globalfifo.o else
LINUX_KERNEL := $(shell uname -r)
LINUX_KERNEL_PATH := /usr/src/linux-headers-$(LINUX_KERNEL)
CURRENT_PATH := $(shell pwd)
EXTRA_CFLAGS += -DDEBUG all:
make -C $(LINUX_KERNEL_PATH) M=$(CURRENT_PATH) modules
clean:
rm -fr *.ko *.o *.mod.o *.mod.c *.symvers *.order .*.ko *.cmd .tmp_versions $(TEST_TARGET) test:
gcc -o $(TEST_TARGET) $(TEST_SRC)
endif
七、设备驱动中的阻塞与非阻塞 IO(二)的更多相关文章
- 七、设备驱动中的阻塞与非阻塞 IO(一)
7.1 阻塞与非阻塞 IO 阻塞操作是指在执行设备操作的时候,若不能获取资源,则挂起进程直到满足可操作的条件后再进行操作.被挂起的进程进入睡眠状态,被从调度器的运行队列移走,直到等待的条件被满足. 非 ...
- Linux设备驱动中的阻塞和非阻塞I/O
[基本概念] 1.阻塞 阻塞操作是指在执行设备操作时,托不能获得资源,则挂起进程直到满足操作所需的条件后再进行操作.被挂起的进程进入休眠状态(不占用cpu资源),从调度器的运行队列转移到等待队列,直到 ...
- 蜕变成蝶~Linux设备驱动中的阻塞和非阻塞I/O
今天意外收到一个消息,真是惊呆我了,博客轩给我发了信息,说是俺的博客文章有特色可以出本书,,这简直让我受宠若惊,俺只是个大三的技术宅,写的博客也是自己所学的一些见解和在网上看到我一些博文以及帖子里综合 ...
- Linux设备驱动中的IO模型---阻塞和非阻塞IO【转】
在前面学习网络编程时,曾经学过I/O模型 Linux 系统应用编程——网络编程(I/O模型),下面学习一下I/O模型在设备驱动中的应用. 回顾一下在Unix/Linux下共有五种I/O模型,分别是: ...
- Linux设备驱动中的阻塞和非阻塞I/O <转载>
Green 博客园 首页 新随笔 联系 订阅 管理 Linux设备驱动中的阻塞和非阻塞I/O [基本概念] 1.阻塞 阻塞操作是指在执行设备操作时,托不能获得资源,则挂起进程直到满足操作所需的条件 ...
- 《linux设备驱动开发详解》笔记——8阻塞与非阻塞IO
8.1 阻塞与非阻塞IO 8.1.0 概述 阻塞:访问设备时,若不能获取资源,则进程挂起,进入睡眠状态:也就是进入等待队列 非阻塞:不能获取资源时,不睡眠,要么退出.要么一直查询:直接退出且无资源时, ...
- Linux设备驱动中的异步通知与异步I/O
异步通知概念: 异步通知的意识是,一旦设备就绪,则主动通知应用程序,这样应用程序根本就不需要查询设备状态,这一点非常类似于硬件上的“中断”概念,比较准确的称谓是“信号驱动的异步IO”,信号是在软件层次 ...
- linux驱动编写之阻塞与非阻塞
一.概念 应用程序使用API接口,如open.read等来最终操作驱动,有两种结果--成功和失败.成功,很好处理,直接返回想要的结果:但是,失败,是继续等待,还是返回失败类型呢? 如果继续等待,将进 ...
- IO模型浅析-阻塞、非阻塞、IO复用、信号驱动、异步IO、同步IO
最近看到OVS用户态的代码,在接收内核态信息的时候,使用了Epoll多路复用机制,对其十分不解,于是从网上找了一些资料,学习了一下<UNIX网络变成卷1:套接字联网API>这本书对应的章节 ...
随机推荐
- ControlTemplate in WPF —— Menu
<ResourceDictionary xmlns="http://schemas.microsoft.com/winfx/2006/xaml/presentation" x ...
- 微信小程序的生命周期和APP对象的使用
1.生命周期和APP对象的使用: //app.js App({ onLaunch: function () { //调用API从本地缓存中获取数据 var logs = wx.getStorageSy ...
- RTX系统整合记录
1.切换数据库RTX常见问题解答五. SQL数据库配置 2.同步系统组织机构 部门同步 用户同步https://blog.csdn.net/qq_21703215/article/details/80 ...
- Stream流实现斐波那契数列
1.前言 我们都知道斐波那契数列有很多种实现方法,在jdk1.8以前没有流操作,只能通过递归或者迭代等其他方式来实现斐波那契数列, 但是jdk1.8以后,有了流操作,我们就可以使用流来实现斐波那契数列 ...
- Redis 入门 3.1 热身
3.1 热身 1. 获得符合规则的键名列表 KEYS pattern pattern 支持 glob 风格通配符格式 语言 字符组 ? 匹配一个字符 * 匹配任意个(包括0个)字符 [] 匹配括号间的 ...
- 【神经网络与深度学习】【Python开发】Caffe配置 windows下怎么安装protobuf for python
首先从google上下载protobuf-2.5.0.zip和protoc-2.5.0-win32.zip,然后把protoc-2.5.0-win32.zip里的protoc.exe放到protobu ...
- 洛谷 P1306 斐波那契公约数 题解
题面 结论:gcd(F[n],F[m])=F[gcd(n,m)]; F[n]=a和F[n+1]=b F[n+2]=a+b,F[n+3]=a+2b,…F[m]=F[m?n?1]a+F[m?n]b F[n ...
- C++中的多重继承(一)
1,C++ 中是否允许一个类继承自多个父类? 1,可以: 2,这种情况就是多重继承: 3,多重继承的表象就是一个类有多个父类: 4,这是 C++ 非常特别的一个特性,在其他的程序设计语言中比如 C#. ...
- html之input
<input>在w3c中解释的是用于搜集用户的信息 它其实就是一个输入框,根据type的不同这个框可以用作不同的功能可以输入一个文本,可以定义一个按钮什么的. 属性type一些值的介绍 1 ...
- PythonDay15
第十五章装饰器_递归 今日内容 带参数装饰器 多个装饰器修饰一个函数 递归 带参数的装饰器 # 判断argv,当登录不同的网页,会有不同的装饰效果def auth(argv): def warpp ...