kernel module编程（三）：获取（分配或注册）设备号

#ifndef _WEI_SCULL_H
 #define _WEI_SCULL_H

 #define SCULL_MAJOR         0  
 #define SCULL_MINOR_MIN 0 
 #define SCULL_DEV_NUM     4

 #endif

#include <linux/init.h>
 #include <linux/module.h>
 #include <linux/fs.h>
 #include "scull.h"

 MODULE_LICENSE("Dual BSD/GPL");

 dev_t   dev;
 int        is_get_dev = -1;

 static int      scull_major = SCULL_MAJOR;

 static int __init scull_init(void)
 {
         printk("Scull module init enter/n");
         if(scull_major){ /* 根据major和minor号码获得dev */ 
                dev = MKDEV(scull_major,SCULL_MINOR_MIN ); 
                 is_get_dev = register_chrdev_region (dev, SCULL_DEV_NUM,"scull");//注册 
         }else{  /* 系统动态分配 */ 
                is_get_dev = alloc_chrdev_region (&dev,SCULL_MINOR_MIN, SCULL_DEV_NUM,"scull");
                 scull_major = MAJOR(dev ); //获取主号码，MINOR（）可获取minor号码 
         }
         if(is_get_dev < 0){
                 printk(KERN_WARNING "scull: can't get device major number %d/n",scull_major);
         }
         return is_get_dev;
 } 
/*在模块结束的时候，应该释放注册的资源*/ 
 static void __exit scull_exit(void)
 {
         if(is_get_dev < 0){
                 return ;
         }else{
                 unregister_chrdev_region(dev,SCULL_DEV_NUM ); 
                 printk("Scull module exit/n");
         }
 }

 module_init(scull_init);
 module_exit(scull_exit);

[wei@wei ~]$ cat scull_load
 #!/bin/sh
 module="scull"
 device="scull"
 mode="644"#加载节点，如果加载不成功（可能已经load过），则退出 
/sbin/insmod ./$module.ko $* || exit 1 
#如果/dev中含有以前创建的节点，删除它们，当然我们也可以判断major号是否和分配的一直，来决定保留或者删除 
rm -f /dev/${device}[0-3]
 
#这是获取major号的过程，awk的使用，可以参考文档：http://www.ibm.com/developerworks/cn/linux/shell/awk/awk-1/    将/proc/devices作为输入，按行输入，如果第二个参数匹配$module，这打印第一个参数，也即赋值给major 
major=$(awk "/$2==/"$module/" {print /$1}" /proc/devices)
 echo major=${major} 
 
#mknod四个设备节点 
mknod /dev/${device}0 c $major 0
 mknod /dev/${device}1 c $major 1
 mknod /dev/${device}2 c $major 2
 mknod /dev/${device}3 c $major 3#一般而言，kernel模块的操作权限都是root，但是有时候我们希望某些用户也可以有读取的权限需要修改设备节点的权限。下面允许组wei的操作，如果组wei不存在，则为组moblin。 
group="wei"
 grep -q '^wei:' /etc/group|| group="moblin"
 chgrp $group /dev/${device}[0-3]
 chmod $mode /dev/${device}[0-3]

[wei@wei ~]$ cat scull_unload 
 #!/bin/sh
 module="scull"
 device="scull"#如果没有加载该模块结束，否则卸载模块，并删除相关的节点。 
/sbin/lsmod | grep $module || exit 1 /sbin/rmmod $module 
 rm -f /dev/${device}[0-3]