2022.12.30work

内核完成IO多路复用

驱动文件定义__poll_t函数，完成等待队列的提交，返回唤醒条件变量

mycdev.c

#include <linux/init.h>
#include <linux/module.h>
#include<linux/cdev.h>
#include<linux/fs.h>
#include<linux/uaccess.h>
#include<linux/slab.h>
#include<linux/io.h>
#include<linux/device.h>
#include<linux/poll.h>

//定义字符设备驱动结构体对象指针
struct cdev *cdev;
#if 1
unsigned int major=0;//进行动态申请设备号
#else
unsigned int major=500;//进行静态指定设备号
#endif
unsigned int minor=0;//次设备号
int count=3;//设备数量
struct class *cls;//指向设备目录的指针
struct device *dev;//指向设备节点的指针
char kbuf[128]={0};
unsigned int condition=0;//用来表示数据是否准备好的标识
//定义等待队列头
wait_queue_head_t wq_head;


int mycdev_open(struct inode *inode, struct file *file)
{
    printk("%s:%s:%d\n",__FILE__,__func__,__LINE__);
    return 0;   
}

ssize_t mycdev_read(struct file *file, char __user *ubuf, size_t size, loff_t *off)
{
    int ret;//定义变量接收函数返回值
            //将进程2的数据拷贝给进程1，模拟硬件数据
            if(size>sizeof(kbuf))
                size=sizeof(kbuf);
            //将kbuf的数据拷贝给用户空间
            ret=copy_to_user(ubuf,kbuf,size);
            if(ret)
            {
                printk("copy_to_user filed\n");
                return -EIO;
            }
            condition=0;
     
    return 0;
}

ssize_t mycdev_write(struct file *file, const char __user *ubuf, size_t size, loff_t *off)
{
    int ret;//定义变量接收函数返回值
     printk("%s:%s:%d\n",__FILE__,__func__,__LINE__);
     //先判断size的大小是否大于内核kbuf的大小，如果比kbuf大，把size的数据修改 为kbuf的大小
     if(size>sizeof(kbuf))
        size=sizeof(kbuf);
    //从进程2拷贝数据，给进程1
     ret=copy_from_user(kbuf,ubuf,size);
    if(ret)
    {
        printk("copy_from_user filed\n");
        return -EIO;
    }
   
   //唤醒进程2
   condition=1;
   wake_up_interruptible(&wq_head);

    return 0;
}

 __poll_t mycdev_poll(struct file *file, struct poll_table_struct *wait)
 {
    __poll_t mask=0;
    //向上提交等待队列头
    poll_wait(file,&wq_head,wait);
    //返回数值
    if(condition)
    {
        mask |= POLLIN;
    }
    return mask;

 }
int mycdev_close(struct inode *inode, struct file *file)
{
   
     printk("%s:%s:%d\n",__FILE__,__func__,__LINE__);
    return 0;
}

//定义操作方法结构体
struct file_operations fops =
{
    .open=mycdev_open,
    .read=mycdev_read,
    .write=mycdev_write,
    .release=mycdev_close,
    .poll= mycdev_poll,
};


static int __init mycdev_init(void)
{
    int ret,i;
    dev_t devno;//设备号
    //给字符设备驱动对象申请空间
    cdev=cdev_alloc();
    if(cdev==NULL)
    {
        printk("给字符设备驱动对象申请空间失败\n");
        ret=-ENOMEM;
        goto ERR1;
    }
    printk("给字符设备驱动对象申请空间成功\n");
    //对象的初始化
    cdev_init(cdev,&fops);
    //申请设备号
    if(major==0)//动态申请
    {
        ret=alloc_chrdev_region(&devno,minor,count,"mycdev");
        if(ret)
        {
            printk("动态指定设备号失败\n");
            goto ERR2;
        }
        //获取主设备号和次设备号
        major=MAJOR(devno);//获取主设备号
        minor=MINOR(devno);//获取次设备号
    }
    else if(major>0)//静态指定
    {
        ret=register_chrdev_region(MKDEV(major,minor),count,"mycdev");
        if(ret)
        {
            printk("静态指定设备号失败\n");
            goto ERR2;
        }
    }
    printk("设备号申请成功\n");
    //将字符设备驱动注册进内核
    ret=cdev_add(cdev,MKDEV(major,minor),count);
    if(ret)
    {
        printk("字符设备驱动注册失败\n");
        goto ERR3;
    }
    //自动创建设备节点
    //向上提交提交目录
    cls=class_create(THIS_MODULE,"mycdev");
    if(IS_ERR(cls))
    {
        printk("向上提交目录失败\n");
        ret=PTR_ERR(cls);
        goto ERR4;
    }
    printk("向上提交目录成功\n");
    //向上提交设备节点
    for(i=0;i<count;i++)
    {
        dev=device_create(cls,NULL,MKDEV(major,i),NULL,"mycdev%d",i);
        if(IS_ERR(dev))
        {
            printk("向上提交节点信息失败\n");
            ret=PTR_ERR(dev);
            goto ERR5;
        }
    }
    printk("向上提交节点信息成功\n");

    //初始化等待队列头
    init_waitqueue_head(&wq_head);

    return 0;

ERR5:
    for(--i;i>=0;i--)
    {
        device_destroy(cls,MKDEV(major,i));
    }
    class_destroy(cls);
ERR4:
    cdev_del(cdev);
ERR3:
    unregister_chrdev_region(MKDEV(major,minor),count);
ERR2:
    kfree(cdev);
ERR1:
    return ret;
}


static void __exit mycdev_exit(void)
{
    //销毁设备节点
    int i;
    for(i=0;i<count;i++)
    {
        device_destroy(cls,MKDEV(major,i));
    }
    //销毁目录
    class_destroy(cls);
    
    //删除驱动对象
    cdev_del(cdev);
    //销毁设备号
    unregister_chrdev_region(MKDEV(major,minor),count);
    //释放动态申请的字符设备驱动的对象空间
     kfree(cdev);

}
module_init(mycdev_init);
module_exit(mycdev_exit);
MODULE_LICENSE("GPL");