【Linux内核驱动】字符设备驱动框架模板

#include <linux/init.h>
#include <linux/module.h>
#include <linux/fs.h>
#include <linux/cdev.h>
#include <linux/kdev_t.h>
#include <linux/slab.h>
#include <linux/device.h>

#define DEVICE_NAME "chrdev_test"
#define DEVICE_MAJOR 0
#define MINOR_BASE 0
#define MINOR_NUM 1

MODULE_LICENSE("Dual BSD/GPL");
MODULE_AUTHOR("colorfulshark@hotmail.com");

static int major = DEVICE_MAJOR;
static int minor_base = MINOR_BASE;
static int minor_num = MINOR_NUM;
static dev_t devno;
static struct cdev *pcdev;
static struct class *test_class;
static struct device *test_device;

static struct file_operations fops = {
.owner = THIS_MODULE,
};

#if 0
module_param(major, int, S_IRWXU);
module_param(minor_base, int, S_IRWXU);
module_param(minor_num, int, S_IRWXU);
#endif

static void request_devno(void) //申请设备号
{
int ret;
if(major) {
printk(KERN_ALERT "alloc device number static\n");
devno = MKDEV(major, minor_base);
ret = register_chrdev_region(devno, minor_num, DEVICE_NAME);
} else {
printk(KERN_ALERT "alloc device number dynamic\n");
ret = alloc_chrdev_region(&devno, minor_base, minor_num, DEVICE_NAME);
printk(KERN_ALERT "alloc major : %d\n", MAJOR(devno));
}
if(ret < 0) {
printk(KERN_ALERT "register chrdev region failed\n");
}
}

static void free_devno(void) //释放设备号
{
unregister_chrdev_region(devno, minor_num);
}

static int add_cdev(void) //添加字符设备
{
int error;
pcdev = cdev_alloc();
if(pcdev == NULL) {
goto err_no_mem;
}
cdev_init(pcdev, &fops);
pcdev->owner = THIS_MODULE;
error = cdev_add(pcdev, devno, minor_num);
if(error)
goto err_cdev_add;
printk(KERN_ALERT "success to add cdev\n");
return 0;
err_no_mem:
printk(KERN_ALERT "no enough memory\n");
free_devno();
return -ENOMEM;
err_cdev_add:
printk(KERN_ALERT "fail to add cdev\n");
free_devno();
return error;
}

static void del_cdev(void) //删除字符设备
{
cdev_del(pcdev);
}

static void add_dev_node(void) //创建设备节点
{
test_class = class_create(THIS_MODULE, "test_class");
test_device = device_create(test_class, NULL, devno, NULL, "chrdev_test");
}

static void del_dev_node(void) //删除设备节点
{
device_destroy(test_class, devno);
class_destroy(test_class);
}

static int chrdev_init(void)
{
printk(KERN_ALERT "chrdev init\n");
request_devno();
add_cdev();
add_dev_node();
return 0;
}

static void chrdev_exit(void)
{
del_dev_node();
del_cdev();
free_devno();
printk(KERN_ALERT "chrdev exit\n");
}

module_init(chrdev_init);
module_exit(chrdev_exit);