Linux设备驱动工程师之路——简单字符设备驱动程序

一、重要知识点 

1. 主次设备号

dev_t

         dev_t是内核中用来表示设备编号的数据类型;

int MAJOR(dev_t dev)

int MINOR(dev_t dev)

         这两个宏抽取主次设备号。

dev­_t MKDEV(unsigned int major, unsignedint minor)

         这个宏由主/次设备号构造一个dev_t结构。

 

2. 分配和释放设备号

int register_chardev_region(dev_t first,unsigned int count, char *name)

         静态申请设备号。

Int alloc_chardev_region(dev_t *dev,unsigned int firstminor, unsigned int count, char *name)

         动态申请设备号,注意第一个参数是传地址,而静态则是传值。        

         3. 几种重要的数据结构

struct file

         file结构代表一个打开的文件,它由内核在open时创建,并传递给该文件上进行操作的所有函数,直到最后的close函数。

         file结构private_data是跨系统调用时保存状态信息非常有用的资源。

         file结构的f_ops 保存了文件的当前读写位置。

struct inode

         内核用inode代表一个磁盘上的文件,它和file结构不同,后者表示打开的文件描述符。对于单个文件,可能会有许多个表示打开文件的文件描述符file结构,但他们都指单个inode结构。inode的dev_t i_rdev成员包含了真正的设备编号,struct cdev *i_cdev包含了指向struct cdev结构的指针。

struct file_operations

         file_operations结构保存了字符设备驱动程序的方法。

 

         4. 字符设备的注册和注销

struct cdev *cdev_alloc(void);

void cdev_init(struct cdev *dev, structfile_operations *fops);

int cdev_add(struct cdev *dev, dev_t num,unsigned int count);

void cdev_del(struct cdev *dev);

         用来管理cdev结构的函数,内核中使用该结构表示字符设备。注意cdev_add函数的count参数为次设备的个数,要想拥有多个次设备,就必须将该参数设为次设备的个数。

 

         5. 并发处理

         信号量和自旋锁的区别,使用信号量时当调用进程试图获得一个锁定了的锁时会导致进程睡眠,而自旋锁则是一直循法的等待一直到该锁解锁了为止。

         1)信号量

DECLARE_MUTEX(name);

DECLARE_MUTEX_LOCKED(name);

         声明和初始化用在互斥模式中的信号量的两个宏

void init_MUTEX(struct semaphore *sem)

void init_MUTEX_LOCKER(struct semaphore*sem);

         这两个函数可以在运行时初始化信号量

void down(struct semaphore *sem);

int down_interruptible(struct semaphore*sem);

int down_trylock(struct semahpore *sem);

void up(struct semaphore *sem);

         锁定和解锁信号量。如果必要,down会将调用进程置于不可中断的休眠状态;相反,down_interruptible可被信号中断。down_trylock不会休眠,并且会在信号量不可用时立即返回。锁定信号量的代码最后必须使用up解锁该信号量。

         2)自旋锁

spionlock_t lock = SPIN_LOCK_UNLOCKED;

spin_lock_init(spinlock_t *lock);

初始化自旋锁的两种方式。

         voidspin_lock(spinlock_t *lock);

锁定自旋锁

         voidspin_unlock(spinlock_t *lock);

         解锁自旋锁

        

二、驱动代码

  1. #include <linux/module.h>  
  2. #include <linux/types.h>  
  3. #include <linux/fs.h>  
  4. #include <linux/errno.h>  
  5. #include <linux/mm.h>  
  6. #include <linux/sched.h>  
  7. #include <linux/init.h>  
  8. #include <linux/cdev.h>  
  9. #include <asm/io.h>  
  10. #include <asm/system.h>  
  11. #include <asm/uaccess.h>  
  12.    
  13.    
  14. #define MEMDEV_MAJOR 251  
  15. #define MEMDEV_NUM 2  
  16. #define MEMDEV_SIZE 1024  
  17.    
  18. struct mem_dev  
  19. {  
  20.          unsignedint size;  
  21.          char*data;  
  22.          structsemaphore sem;  
  23. };  
  24.    
  25.    
  26. static int mem_major = MEMDEV_MAJOR;  
  27.    
  28. struct cdev mem_cdev;  
  29. struct mem_dev *mem_devp;  
  30.    
  31.    
  32. static int mem_open(struct inode *inode,struct file *filp)  
  33. {  
  34.          structmem_dev *dev;  
  35.          unsignedint num;  
  36.           
  37.          printk("mem_open.\n");  
  38.           
  39.          numMINOR(inode->i_rdev);//获得次设备号  
  40.          if(num> (MEMDEV_NUM -1))          //检查次设备号有效性  
  41.                    return-ENODEV;  
  42.                     
  43.          dev= &mem_devp[num];  
  44.          filp->private_datadev; //将设备结构保存为私有数据  
  45.           
  46.          return0;  
  47. }  
  48.    
  49. static int mem_release(struct inode *inode,struct file *filp)  
  50. {  
  51.          printk("mem_release.\n");  
  52.          return0;  
  53. }  
  54.    
  55. static ssize_t mem_read(struct file *filp,char __user *buf, size_t size, loff_t *ppos)  
  56. {  
  57.          intret = 0;  
  58.          structmem_dev *dev;  
  59.          unsignedlong p;  
  60.          unsignedlong count;  
  61.           
  62.          printk("mem_read.\n");  
  63.           
  64.          devfilp->private_data;//获得设备结构  
  65.          countsize;  
  66.          p= *ppos;  
  67.           
  68.          //检查偏移量和数据大小的有效性  
  69.          if(p> MEMDEV_SIZE)  
  70.                    return0;  
  71.          if(count> (MEMDEV_SIZE-p))  
  72.                    countMEMDEV_SIZE - p;  
  73.                     
  74.          if(down_interruptible(&dev->sem))//锁定互斥信号量  
  75.                   return -ERESTARTSYS;  
  76.                     
  77.          //读取数据到用户空间  
  78.          if(copy_to_user(buf,dev->data+p, count)){  
  79.                    ret= -EFAULT;  
  80.                    printk("copyfrom user failed\n");  
  81.          }  
  82.          else{  
  83.                    *ppos+= count;  
  84.                    retcount;  
  85.                    printk("read%d bytes from dev\n", count);  
  86.          }  
  87.           
  88.          up(&dev->sem);//解锁互斥信号量  
  89.           
  90.          returnret;  
  91. }  
  92.    
  93. static ssize_t mem_write(struct file *filp,const char __user *buf, size_t size, loff_t *ppos)//注意:第二个参数和read方法不同  
  94. {  
  95.          intret = 0;  
  96.          structmem_dev *dev;  
  97.          unsignedlong p;  
  98.          unsignedlong count;  
  99.           
  100.          printk("mem_write.\n");  
  101.           
  102.          devfilp->private_data;  
  103.          countsize;  
  104.          p= *ppos;  
  105.           
  106.          if(p> MEMDEV_SIZE)  
  107.                    return0;  
  108.          if(count> (MEMDEV_SIZE-p))  
  109.                    countMEMDEV_SIZE - p;  
  110.                     
  111.          if(down_interruptible(&dev->sem))//锁定互斥信号量  
  112.                    return-ERESTARTSYS;  
  113.                     
  114.          if(copy_from_user(dev->data+p,buf, count)){  
  115.                    ret= -EFAULT;  
  116.                    printk("copyfrom user failed\n");  
  117.          }  
  118.          else{  
  119.                    *ppos+= count;  
  120.                    retcount;  
  121.                    printk("write%d bytes to dev\n", count);  
  122.          }  
  123.           
  124.          up(&dev->sem);//解锁互斥信号量  
  125.           
  126.          returnret;  
  127. }  
  128.    
  129. static loff_t mem_llseek(struct file *filp,loff_t offset, int whence)  
  130. {  
  131.          intnewpos;  
  132.           
  133.          printk("mem_llseek.\n");  
  134.           
  135.          switch(whence)  
  136.          {  
  137.                    case0:  
  138.                             newposoffset;  
  139.                             break;  
  140.                     
  141.                    case1:  
  142.                             newposfilp->f_pos + offset;  
  143.                             break;  
  144.                              
  145.                    case2:  
  146.                             newposMEMDEV_SIZE - 1 + offset;  
  147.                             break;  
  148.                              
  149.                    default:  
  150.                             return-EINVAL;  
  151.          }  
  152.           
  153.          if((newpos<0)|| (newpos>(MEMDEV_SIZE - 1)))  
  154.                    return-EINVAL;  
  155.                     
  156.          filp->f_posnewpos;  
  157.          returnnewpos;  
  158. }  
  159. static const struct file_operationsmem_fops = {  
  160.          .ownerTHIS_MODULE,  
  161.          .openmem_open,  
  162.          .writemem_write,  
  163.          .readmem_read,  
  164.          .releasemem_release,  
  165.          .llseekmem_llseek,  
  166. };  
  167.    
  168. static int __init memdev_init(void)  
  169. {  
  170.          intresult;  
  171.          interr;  
  172.          inti;  
  173.           
  174.          //申请设备号  
  175.          dev_tdevno = MKDEV(mem_major, 0);  
  176.           
  177.          if(mem_major)  
  178.                    resultregister_chrdev_region(devno, MEMDEV_NUM, "memdev");//注意静态申请的dev_t参数和动态dev_t参数的区别  
  179.          else{                                                                                                                          //静态直接传变量,动态传变量指针  
  180.                    resultalloc_chrdev_region(&devno, 0, MEMDEV_NUM, "memdev");  
  181.                    mem_majorMAJOR(devno);  
  182.          }  
  183.           
  184.          if(result< 0){  
  185.                    printk("can'tget major devno:%d\n", mem_major);  
  186.                    returnresult;  
  187.          }  
  188.           
  189.          //注册设备驱动  
  190.          cdev_init(&mem_cdev,&mem_fops);  
  191.          mem_cdev.ownerTHIS_MODULE;  
  192.           
  193.          errcdev_add(&mem_cdev, MKDEV(mem_major, 0), MEMDEV_NUM);//如果有N个设备就要添加N个设备号  
  194.          if(err)  
  195.                    printk("addcdev faild,err is %d\n", err);  
  196.                     
  197.          //分配设备内存  
  198.          mem_devpkmalloc(MEMDEV_NUM*(sizeof(struct mem_dev)), GFP_KERNEL);  
  199.          if(!mem_devp){  
  200.                     result = - ENOMEM;  
  201.                     goto fail_malloc;  
  202.          }  
  203.          memset(mem_devp,0, MEMDEV_NUM*(sizeof(struct mem_dev)));  
  204.           
  205.           
  206.          for(i=0;i<MEMDEV_NUM; i++){  
  207.                    mem_devp[i].sizeMEMDEV_SIZE;  
  208.                    mem_devp[i].datakmalloc(MEMDEV_SIZE, GFP_KERNEL);  
  209.                    memset(mem_devp[i].data,0, MEMDEV_SIZE);  
  210.                    init_MUTEX(&mem_devp[i].sem);//初始化互斥锁  
  211.          }  
  212.           
  213.          returnresult;  
  214.           
  215. fail_malloc:  
  216.          unregister_chrdev_region(MKDEV(mem_major,0), MEMDEV_NUM);  
  217.          returnresult;  
  218. }  
  219.    
  220. static void memdev_exit(void)  
  221. {  
  222.          cdev_del(&mem_cdev);  
  223.          unregister_chrdev_region(MKDEV(mem_major,0), MEMDEV_NUM);//注意释放的设备号个数一定要和申请的设备号个数保存一致  
  224.                                                                                                                          //否则会导致设备号资源流失  
  225.          printk("memdev_exit\n");  
  226. }  
  227.    
  228. module_init(memdev_init);  
  229. module_exit(memdev_exit);  
  230.    
  231. MODULE_AUTHOR("Y-Kee");  
  232. MODULE_LICENSE("GPL");  
  233.    

相关推荐