十二月29号驱动作业

把IO多路复用在内核上的实现

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>

int major;//用于存放驱动的主设备号
char kbuf[128]={0};
wait_queue_head_t wq;

struct class *cls;
struct device *dev;
int condition=0;
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;//定义变量接收函数返回值
     printk("%s:%s:%d\n",__FILE__,__func__,__LINE__);
     //先判断size的大小是否大于内核kbuf的大小，如果比kbuf大，把size的数据修改 为kbuf的大小
   
         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 size;
}

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);
    //从用户空间拷贝数据
     ret=copy_from_user(kbuf,ubuf,size);
    if(ret)
    {
        printk("copy_from_user filed\n");
        return -EIO;
    }
    condition=1;
    wake_up_interruptible(&wq);
     

    return size;
}

__poll_t mycdev_poll(struct file* file,struct poll_table_struct* wait)
{
    //1.定义返回值的变量
    __poll_t mask = 0;
    //2.调用poll_wait,猜测是和阻塞相关的函数
    poll_wait(file,&wq,wait);
    //3.如果数据准备好位置mask
    if(condition){
        mask |= EPOLLIN; //EPOLL数据可读  EPOLLOUT数据可写
    }
    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,
    .poll = mycdev_poll,
    .release=mycdev_close,
};
//入口函数
static int __init mycdev_init(void)
{
   //注册字符设备驱动
   major=register_chrdev(0,"mycdev",&fops);
   if(major<0)
   {
    printk("字符设备驱动注册失败\n");
    return major;
   }
   printk("字符设备驱动注册成功,major=%d\n",major);
   
 
     //向上提交目录信息
   cls=class_create(THIS_MODULE,"led");
   if(IS_ERR(cls))
   {
        printk("向上提交目录信息失败\n");
        return PTR_ERR(cls);
   }
   printk("向上提交目录信息成功\n");
   //向上提交节点信息
   dev=device_create(cls,NULL,MKDEV(major,0),NULL,"myled1");
   if(IS_ERR(dev))
   {
        printk("向上提交节点信息失败\n");
        return PTR_ERR(dev);
   }
   printk("向上提交节点信息成功\n");
    init_waitqueue_head(&wq);
    
    return 0;
}

//出口函数
static void __exit mycdev_exit(void)
{

    //销毁节点
    device_destroy(cls,MKDEV(major,0));
    //销毁目录
    class_destroy(cls);
                                                                                                                                                                                                                
    //注销字符设备驱动
    unregister_chrdev(major,"mycdev");
}

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

epoll.c

#include<stdio.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <unistd.h>
#include <stdlib.h>
#include<string.h>
#include <sys/wait.h>
#include <sys/epoll.h>

int main(int argc, char const *argv[])
{
    int epfd,fd1,fd2;
    int ret,i;
    char buf[128]={0};
    struct epoll_event event; //用于操作epoll
    struct epoll_event events[10]; //用于存放准备好的事件描述符

    fd1=open("/dev/mycdev1",O_RDWR);
    if(fd1<0)
    {
        printg("oppen err\n");
        exit(-1);
    }
    fd2=open("/dev/input/mouse0",O_RDWR);
    if(fd2<0)
    {
        ptintf("open err\n");
        exit(-1);
    }

    //创建一个epoll
    epfd=epoll_create(10);
    if(epfd<0)
    {
        printf("epoll_create err\n");
        exit(-1);
    }

    //将目标文件描述符添加到eppoll
    event.events=EPOLLIN;
    event.data.fd=fd1;
    if(epoll_ctl(epfd,EPOLL_CTL_ADD,fd1,&event)<0)
    {
        printf("epoll_ctl ern");
        exit(-1);
    }
    event.events=EPOLLIN;
    event.data.fd=fd2;
    if(epoll_ctl(epfd,EPOLL_CTL_ADD,fd2,&event)<0)
    {
        printf("epoll_ctl ern");
        exit(-1);
    }
    while (1)
    {
       ret=epoll_wait(epfd,events,10,-1);
       if(ret<0)
       {
        printf("epoll_wait err\n");
        rexit(-1);
       }
       for(i=0;i<ret;i++)
       {
        if(events[i].events&EPOLLIN)
        {
            read(events[i].data.fd,buf,sizeof(buf));
            printf("buf:%s\n",buf);
        }
       }
       close(fd1);
       close(fd2);
       close(epfd);
        return 0;
    }
    
}

IO多路复用内部实现总结

USER:
	select(fd2 + 1, &rfds, NULL, NULL, NULL)
*****************************************************************************************************
VFS:  vi -t sys_select
    1.校验最大文件描述符的值
    2.将用户空间文件描述符的集合拷贝到内核空间
	3.从文件描述符的集合中找出文件描述符,进而调用驱动的poll函数
    	mask = fd_set-->fd-->fd_array[fd]-->file-->f_op-->poll(file,wait);
		如果所有文件描述符对应的驱动的poll函数的返回值都为0，说明所有硬件中的
        数据都没有准备好，此时进程休眠。
    4.如果有文件描述符的数据准备好了，唤醒这个休眠的进程，再次遍历文件描述符
        mask = fd_set-->fd-->fd_array[fd]-->file-->f_op-->poll(file,wait);
		如果文件描述符对应的poll函数返回的值不是0，就表示这个文件描述符的数
        据准备好了，把它放到准备好的文件描述符的集合中
	5.将准备好的文件描述符拷贝到用户空间
*****************************************************************************************************
DRIVER:
    __poll_t mycdev_poll(struct file* file, struct poll_table_struct* wait)
    {
        // 1.定义返回值的变量
        __poll_t mask = 0;
        // 2.调用poll_wait,向上提交等待队列头，构造等待队列
        poll_wait(file, &wq, wait);
        // 3.如果数据准备好置位mask
        if (condition) {
            mask |= EPOLLIN; // EPOLLIN 数据可读   EPOLLOUT数据可写
        }
        // 4.返回mask
        return mask;
    }