Linux驱动——注册字符类设备

一、分配内存空间

        前面介绍的杂项设备并没有分配内存空间这个过程， 是因为系统自带的代码已经给杂项设备分配好了。

        Linux 中注册字符类设备需要首先申请内存空间， 有一个专门分配小内存空间的函数 kmalloc， 这个函数在头文件“include/linux/slab.h” 中， 如下图所示， 使用命令“vim include/linux/slab.h” 打开头文件。

        如上图所示， 函数 static inline void *kzalloc(size_t size, gfp_t flags)有两个参数，参数 size_t size： 申请的内存大小(最大 128K)，代表优先权， 内存不够可以延迟分配。

        和申请内存的函数 kzalloc 对应的是 kfree 释放函数， 如下图所示。

 

        void kfree(const void *)函数只有一个参数， 就是内存的指针， 这个指针由申请内存的函数 kzalloc 返回。

二、注册字符类设备的函数

        注册字符类设备的初始化函数为 cdev_init， 这个函数在头文件“include/linux/cdev.h” 中， 使用命令“vim include/linux/cdev.h” 打开这个头文件如下图所示。

 

        如上图所示红框中的函数“void cdev_init(struct cdev *, const struct file_operations *)” 和结构体“cdev”。cdev_init 函数有两个参数，

        参数 struct cdev *： cdev 字符设备文件结构体

        参数 const struct file_operations *： file_operations 结构体。

        注册字符设备的函数为 cdev_add， 这个函数也是在头文件“include/linux/cdev.h” 中， 如下图所示。

        如上图所示， 和注册驱动的函数 cdev_add 对应的还有卸载驱动的函数 cdev_del。注册驱动的函数 int cdev_add(struct cdev *, dev_t, unsigned);有三个参数：

        参数 struct cdev *： cdev 字符设备文件结构

        参数 dev_t： 设备号 dev,前面已经介绍和使用过了参数 unsigned： 设备范围大小卸载驱动的函数 void cdev_del(struct cdev *)只有一个参数， cdev 字符设备结构体

三、注册字符类设备的例程

/*包含初始化宏定义的头文件,代码中的module_init和module_exit在此文件中*/
#include <linux/init.h>
/*包含初始化加载模块的头文件,代码中的MODULE_LICENSE在此头文件中*/
#include <linux/module.h>
/*定义module_param module_param_array的头文件*/
#include <linux/moduleparam.h>
/*定义module_param module_param_array中perm的头文件*/
#include <linux/stat.h>
/*三个字符设备函数*/
#include <linux/fs.h>
/*MKDEV转换设备号数据类型的宏定义*/
#include <linux/kdev_t.h>
/*定义字符设备的结构体*/
#include <linux/cdev.h>

/*分配内存空间函数头文件*/
#include <linux/slab.h>

#define DEVICE_NAME "ascdev"
#define DEVICE_MINOR_NUM 2
#define DEV_MAJOR 0
#define DEV_MINOR 0
#define REGDEV_SIZE 3000

MODULE_LICENSE("Dual BSD/GPL");
/*声明是开源的，没有内核版本限制*/
MODULE_AUTHOR("iTOPEET_dz");
/*声明作者*/

int numdev_major = DEV_MAJOR;
int numdev_minor = DEV_MINOR;

/*输入主设备号*/
module_param(numdev_major,int,S_IRUSR);
/*输入次设备号*/
module_param(numdev_minor,int,S_IRUSR);

struct reg_dev
{
	char *data;
	unsigned long size;
	
	struct cdev cdev;
};
struct reg_dev *my_devices;

struct file_operations my_fops = {
	.owner = THIS_MODULE,
};


/*设备注册到系统*/
static void reg_init_cdev(struct reg_dev *dev,int index){
	int err;
	int devno = MKDEV(numdev_major,numdev_minor+index);
	
	/*数据初始化*/
	cdev_init(&dev->cdev,&my_fops);
	dev->cdev.owner = THIS_MODULE;
	dev->cdev.ops = &my_fops;
	
	/*注册到系统*/
	err = cdev_add(&dev->cdev,devno,1);
	if(err){
		printk(KERN_EMERG "cdev_add %d is fail! %d\n",index,err);
	}
	else{
		printk(KERN_EMERG "cdev_add %d is success!\n",index);
	}
}

static int scdev_init(void)
{
	int ret = 0,i;
	dev_t num_dev;
	
	
	printk(KERN_EMERG "numdev_major is %d!\n",numdev_major);
	printk(KERN_EMERG "numdev_minor is %d!\n",numdev_minor);
	
	if(numdev_major){
		num_dev = MKDEV(numdev_major,numdev_minor);
		ret = register_chrdev_region(num_dev,DEVICE_MINOR_NUM,DEVICE_NAME);
	}
	else{
		/*动态注册设备号*/
		ret = alloc_chrdev_region(&num_dev,numdev_minor,DEVICE_MINOR_NUM,DEVICE_NAME);
		/*获得主设备号*/
		numdev_major = MAJOR(num_dev);
		printk(KERN_EMERG "adev_region req %d !\n",numdev_major);
	}
	if(ret<0){
		printk(KERN_EMERG "register_chrdev_region req %d is failed!\n",numdev_major);		
	}
	
	my_devices = kmalloc(DEVICE_MINOR_NUM * sizeof(struct reg_dev),GFP_KERNEL);
	if(!my_devices){
		ret = -ENOMEM;
		goto fail;
	}
	memset(my_devices,0,DEVICE_MINOR_NUM * sizeof(struct reg_dev));
	
	/*设备初始化*/
	for(i=0;i<DEVICE_MINOR_NUM;i++){
		my_devices[i].data = kmalloc(REGDEV_SIZE,GFP_KERNEL);
		memset(my_devices[i].data,0,REGDEV_SIZE);
		/*设备注册到系统*/
		reg_init_cdev(&my_devices[i],i);
	}
	
		
	printk(KERN_EMERG "scdev_init!\n");
	/*打印信息，KERN_EMERG表示紧急信息*/
	return 0;

fail:
	/*注销设备号*/
	unregister_chrdev_region(MKDEV(numdev_major,numdev_minor),DEVICE_MINOR_NUM);
	printk(KERN_EMERG "kmalloc is fail!\n");
	
	return ret;
}

static void scdev_exit(void)
{
	int i;
	printk(KERN_EMERG "scdev_exit!\n");
	
	/*除去字符设备*/
	for(i=0;i<DEVICE_MINOR_NUM;i++){
		cdev_del(&(my_devices[i].cdev));
	}
		
	unregister_chrdev_region(MKDEV(numdev_major,numdev_minor),DEVICE_MINOR_NUM);
}


module_init(scdev_init);
/*初始化函数*/
module_exit(scdev_exit);
/*卸载函数*/