zoukankan      html  css  js  c++  java
  • 虚拟内存设备驱动memdev及实例代码

    驱动:

    #include <linux/module.h>
    #include <linux/types.h>
    #include <linux/fs.h>
    #include <linux/errno.h>
    #include <linux/mm.h>
    #include <linux/sched.h>
    #include <linux/init.h>
    #include <linux/cdev.h>
    #include <asm/io.h>
    #include <asm/system.h>
    #include <asm/uaccess.h>
    #include <linux/slab.h>
    #include <linux/device.h>
    
    #ifndef MEMDEV_MAJOR
    #define MEMDEV_MAJOR 0   /*预设的mem的主设备号*/
    #endif
    
    #ifndef MEMDEV_NR_DEVS
    #define MEMDEV_NR_DEVS 2    /*设备数*/
    #endif
    
    #ifndef MEMDEV_SIZE
    #define MEMDEV_SIZE 4096
    #endif
    
    #define DEVICE_NAME "memdev"
    
    /*udev自动添加设备节点*/
    struct class *mem_class;
    
    static int mem_major = MEMDEV_MAJOR;
    
    module_param(mem_major, int, S_IRUGO);
    
    struct mem_dev                                     
    {                                                        
      struct cdev cdev; /*cdev结构体*/                       
      unsigned char mem[MEMDEV_SIZE]; /*全局内存*/        
    };
    
    struct mem_dev *mem_devp; /*设备结构体指针*/
    
    /*文件打开函数*/
    int mem_open(struct inode *inode, struct file *filp)
    {
        struct mem_dev *dev;
    
        dev=container_of(inode->i_cdev,struct mem_dev,cdev);//通过结构体成员的指针找到对应结构体的指针
        
        /*将设备描述结构指针赋值给文件私有数据指针*/
        filp->private_data = dev;
        
        return 0; 
    }
    
    /*文件释放函数*/
    int mem_release(struct inode *inode, struct file *filp)
    {
      return 0;
    }
    
    /*读函数*/
    static ssize_t mem_read(struct file *filp, char __user *buf, size_t size, loff_t *ppos)
    {
      unsigned long p =  *ppos;
      unsigned int count = size;
      int ret = 0;
      struct mem_dev *dev = filp->private_data; /*获得设备结构体指针*/
    
      /*判断读位置是否有效*/
      if (p >= MEMDEV_SIZE)
        return 0;
      if (count > MEMDEV_SIZE - p)
        count = MEMDEV_SIZE - p;
    
      /*读数据到用户空间*/
      if (copy_to_user(buf, (void*)(dev->mem+ p), count))
      {
        ret =  - EFAULT;
      }
      else
      {
        *ppos += count;
        ret = count;
        
        printk(KERN_INFO "read %d bytes(s) from postion  %d\n", count, p);
      }
    
      return ret;
    }
    
    /*写函数*/
    static ssize_t mem_write(struct file *filp, const char __user *buf, size_t size, loff_t *ppos)
    {
      unsigned long p =  *ppos;
      unsigned int count = size;
      int ret = 0;
      struct mem_dev *dev = filp->private_data; /*获得设备结构体指针*/
      
      /*分析和获取有效的写长度*/
      if (p >= MEMDEV_SIZE)
        return 0;
      if (count > MEMDEV_SIZE - p)
        count = MEMDEV_SIZE - p;
        
      /*从用户空间写入数据*/
      if (copy_from_user(dev->mem + p, buf, count))
        ret =  - EFAULT;
      else
      {
        *ppos += count;
        ret = count;
        
        printk(KERN_INFO "written %d bytes(s) to postion %d\n", count, p);
      }
    
      return ret;
    }
    
    /* seek文件定位函数 */
    static loff_t mem_llseek(struct file *filp, loff_t offset, int whence)
    { 
        loff_t newpos;
    
        switch(whence) {
          case 0: /* SEEK_SET */
            newpos = offset;
            break;
    
          case 1: /* SEEK_CUR */
            newpos = filp->f_pos + offset;
            break;
    
          case 2: /* SEEK_END */
            newpos = MEMDEV_SIZE -1 + offset;
            break;
    
          default: /* can't happen */
            return -EINVAL;
        }
        if ((newpos<0) || (newpos>MEMDEV_SIZE))
            return -EINVAL;
            
        filp->f_pos = newpos;
        return newpos;
    
    }
    
    /*文件操作结构体*/
    static const struct file_operations mem_fops =
    {
      .owner = THIS_MODULE,
      .llseek = mem_llseek,
      .read = mem_read,
      .write = mem_write,
      .open = mem_open,
      .release = mem_release,
    };
    
    /*初始化并注册cdev*/
    static void mem_setup_cdev(struct mem_dev *dev, int index)
    {
      int err, devno = MKDEV(mem_major, index);
    
      cdev_init(&dev->cdev, &mem_fops);
      dev->cdev.owner = THIS_MODULE;
      dev->cdev.ops = &mem_fops;
      err = cdev_add(&dev->cdev, devno, 1);
      if (err)
        printk(KERN_NOTICE "Error %d adding LED%d", err, index);
    }
    
    
    /*设备驱动模块加载函数*/
    static int memdev_init(void)
    {
      int result;
    
    
      dev_t devno = MKDEV(mem_major, 0);
    
      /* 静态申请设备号*/
      if (mem_major)
        result = register_chrdev_region(devno, MEMDEV_NR_DEVS, DEVICE_NAME);
      else  /* 动态分配设备号 */
      {
        result = alloc_chrdev_region(&devno, 0, MEMDEV_NR_DEVS, DEVICE_NAME);
        mem_major = MAJOR(devno);
      }  
      
      if (result < 0)
        return result;
    
    
      
    
       
      /* 为设备描述结构分配内存*/
      mem_devp = kmalloc(MEMDEV_NR_DEVS * sizeof(struct mem_dev), GFP_KERNEL);
      if (!mem_devp)    /*申请失败*/
      {
        result =  - ENOMEM;
        goto fail_malloc;
      }
      memset(mem_devp, 0, 2*sizeof(struct mem_dev));
      
      /*为设备分配内存*/
      mem_setup_cdev(&mem_devp[0], 0);
      mem_setup_cdev(&mem_devp[1], 1);
    
      
      /*udev机制可以自动添加设备节点,只需要添加xxx_class这个类,以及device_create()*/
      mem_class = class_create(THIS_MODULE, "mem_class");/*在sys目录下创建xx_class这个类,/sys/class/~*/
      device_create(mem_class, NULL, mem_devp[0].cdev.dev,      NULL, "memdev0");/*自动创建设备/dev/$DEVICE_NAME*/
      device_create(mem_class, NULL, mem_devp[1].cdev.dev,   NULL, "memdev1");/*自动创建设备/dev/$DEVICE_NAME*/
    
      printk("memdev driver installed, with major %d\n", mem_major);
      printk("the device name is %s\n",DEVICE_NAME);
      return 0;
    
      fail_malloc: 
      unregister_chrdev_region(devno, 1);
      
      return result;
    }
    
    /*模块卸载函数*/
    static void memdev_exit(void)
    {
     
      
      device_destroy(mem_class, mem_devp[0].cdev.dev);
      device_destroy(mem_class, mem_devp[1].cdev.dev);
      class_destroy(mem_class);
      cdev_del(&(mem_devp[0].cdev));   /*注销设备*/
      cdev_del(&(mem_devp[1].cdev)); 
    
          
      kfree(mem_devp);     /*释放设备结构体内存*/
      unregister_chrdev_region(MKDEV(mem_major, 0), 2); /*释放设备号*/
      printk("memdev driver uninstalled\n");
    }
    
    MODULE_AUTHOR("mhb@SEU");
    MODULE_LICENSE("GPL");
    
    module_init(memdev_init);
    module_exit(memdev_exit);

    测试代码:

    #include <stdio.h>
    #include <stdlib.h>
    #include <unistd.h>
    #include <fcntl.h>
    
    #include <string.h>
    
    int main()
    {
        FILE *fp0 = NULL;
        char Buf[4096];
        int result;
        
        /*初始化Buf*/
        strcpy(Buf,"Mem is char dev!");
        printf("BUF: %s\n",Buf);
        
        /*打开设备文件*/
        fp0 = fopen("/dev/memdev0","r+");
        if (fp0 == NULL)
        {
            perror("Open Memdev0 Error!\n");
            return -1;
        }
        
        /*写入设备*/
        result = fwrite(Buf, sizeof(Buf), 1, fp0);
        if (result  == -1)
        {
            perror("fwrite Error!\n");
            return -1;
        }
        
        /*重新定位文件位置(思考没有该指令,会有何后果)*/
        fseek(fp0,0,SEEK_SET);
        
        /*清除Buf*/
        strcpy(Buf,"Buf is NULL!");
        printf("BUF: %s\n",Buf);
        
        sleep(1);
        /*读出设备*/
        result = fread(Buf, sizeof(Buf), 1, fp0);
        if (result  == -1)
        {
            perror("fread Error!\n");
            return -1;
        }
        
        /*检测结果*/
        printf("BUF: %s\n",Buf);
        
        return 0;    
    
    }
  • 相关阅读:
    如何快速生成数据文件(fsutil命令,使用CreateFile和SetEndOfFile API函数,fopen和fseek RTL函数)
    TestDisk 数据恢复 重建分区表恢复文件-恢复diskpart clean
    利用winIO3.0进行windows10 64bit端口读取
    一个字体,大小,颜色可定义的自绘静态框控件-XColorStatic 类(比较好看,一共19篇自绘文章)
    美国富人与穷人的四个决定性差别!(冒风险、交朋友、敬业、生活习惯好)
    WebRTC 音视频开发
    协程的理解
    线程、进程、协程和队列
    Confluent
    “Options模式”下各种类型的Options对象是如何绑定的?
  • 原文地址:https://www.cnblogs.com/hello2mhb/p/3284239.html
Copyright © 2011-2022 走看看