《Linux设备驱动开发详解》源码——globalmem

最新推荐文章于 2021-05-13 00:18:52 发布

口天土立口

最新推荐文章于 2021-05-13 00:18:52 发布

阅读量972

点赞数

分类专栏： Linux设备驱动开发详解驱动程序

驱动程序同时被 2 个专栏收录

26 篇文章 1 订阅

订阅专栏

Linux设备驱动开发详解

7 篇文章 0 订阅

订阅专栏

/home/lihacker/develop/svn/ldd6410-read-only/training/kernel/drivers/globalmem

< ch6 / globalmem.c >

/*
 * A globalmem driver as an example of char device drivers
 *
 * The initial developer of the original code is Barry Song
 * <author@linuxdriver.cn>. All Rights Reserved.
 */

#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>

#define GLOBALMEM_SIZE	0x1000	/*È«ŸÖÄÚŽæ×îŽó4K×ÖœÚ*/
#define MEM_CLEAR 0x1  /*Çå0È«ŸÖÄÚŽæ*/
#define GLOBALMEM_MAJOR 250    /*Ô€ÉèµÄglobalmemµÄÖ÷Éè±žºÅ*/

static int globalmem_major = GLOBALMEM_MAJOR;
/*globalmemÉè±žœá¹¹Ìå*/
struct globalmem_dev {
	struct cdev cdev; /*cdevœá¹¹Ìå*/
	unsigned char mem[GLOBALMEM_SIZE]; /*È«ŸÖÄÚŽæ*/
};

struct globalmem_dev *globalmem_devp; /*Éè±žœá¹¹ÌåÖžÕë*/
/*ÎÄŒþŽò¿ªº¯Êý*/
int globalmem_open(struct inode *inode, struct file *filp)
{
	/*œ«Éè±žœá¹¹ÌåÖžÕëž³ÖµžøÎÄŒþËœÓÐÊýŸÝÖžÕë*/
	filp->private_data = globalmem_devp;
	return 0;
}
/*ÎÄŒþÊÍ·Åº¯Êý*/
int globalmem_release(struct inode *inode, struct file *filp)
{
	return 0;
}

/* ioctlÉè±ž¿ØÖÆº¯Êý */
static int globalmem_ioctl(struct inode *inodep, struct file *filp, unsigned
	int cmd, unsigned long arg)
{
	struct globalmem_dev *dev = filp->private_data;/*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	switch (cmd) {
	case MEM_CLEAR:
		memset(dev->mem, 0, GLOBALMEM_SIZE);
		printk(KERN_INFO "globalmem is set to zero\n");
		break;

	default:
		return  - EINVAL;
	}

	return 0;
}

/*¶Áº¯Êý*/
static ssize_t globalmem_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 globalmem_dev *dev = filp->private_data; /*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	/*·ÖÎöºÍ»ñÈ¡ÓÐÐ§µÄÐŽ³€¶È*/
	if (p >= GLOBALMEM_SIZE)
		return 0;
	if (count > GLOBALMEM_SIZE - p)
		count = GLOBALMEM_SIZE - p;

	/*ÄÚºË¿ÕŒä->ÓÃ»§¿ÕŒä*/
	if (copy_to_user(buf, (void *)(dev->mem + p), count)) {
		ret =  - EFAULT;
	} else {
		*ppos += count;
		ret = count;

		printk(KERN_INFO "read %u bytes(s) from %lu\n", count, p);
	}

	return ret;
}

/*ÐŽº¯Êý*/
static ssize_t globalmem_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 globalmem_dev *dev = filp->private_data; /*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	/*·ÖÎöºÍ»ñÈ¡ÓÐÐ§µÄÐŽ³€¶È*/
	if (p >= GLOBALMEM_SIZE)
		return 0;
	if (count > GLOBALMEM_SIZE - p)
		count = GLOBALMEM_SIZE - p;

	/*ÓÃ»§¿ÕŒä->ÄÚºË¿ÕŒä*/
	if (copy_from_user(dev->mem + p, buf, count))
		ret =  - EFAULT;
	else {
		*ppos += count;
		ret = count;

		printk(KERN_INFO "written %u bytes(s) from %lu\n", count, p);
	}

	return ret;
}

/* seekÎÄŒþ¶šÎ»º¯Êý */
static loff_t globalmem_llseek(struct file *filp, loff_t offset, int orig)
{
	loff_t ret = 0;
	switch (orig) {
	case 0:   /*Ïà¶ÔÎÄŒþ¿ªÊŒÎ»ÖÃÆ«ÒÆ*/
		if (offset < 0)	{
			ret =  - EINVAL;
			break;
		}
		if ((unsigned int)offset > GLOBALMEM_SIZE) {
			ret =  - EINVAL;
			break;
		}
		filp->f_pos = (unsigned int)offset;
		ret = filp->f_pos;
		break;
	case 1:   /*Ïà¶ÔÎÄŒþµ±Ç°Î»ÖÃÆ«ÒÆ*/
		if ((filp->f_pos + offset) > GLOBALMEM_SIZE) {
			ret =  - EINVAL;
			break;
		}
		if ((filp->f_pos + offset) < 0) {
			ret =  - EINVAL;
			break;
		}
		filp->f_pos += offset;
		ret = filp->f_pos;
		break;
	default:
		ret =  - EINVAL;
		break;
	}
	return ret;
}

/*ÎÄŒþ²Ù×÷œá¹¹Ìå*/
static const struct file_operations globalmem_fops = {
	.owner = THIS_MODULE,
	.llseek = globalmem_llseek,
	.read = globalmem_read,
	.write = globalmem_write,
	.ioctl = globalmem_ioctl,
	.open = globalmem_open,
	.release = globalmem_release,
};

/*³õÊŒ»¯²¢×¢²ácdev*/
static void globalmem_setup_cdev(struct globalmem_dev *dev, int index)
{
	int err, devno = MKDEV(globalmem_major, index);

	cdev_init(&dev->cdev, &globalmem_fops);
	dev->cdev.owner = THIS_MODULE;
	err = cdev_add(&dev->cdev, devno, 1);
	if (err)
		printk(KERN_NOTICE "Error %d adding LED%d", err, index);
}

/*Éè±žÇý¶¯Ä£¿éŒÓÔØº¯Êý*/
int globalmem_init(void)
{
	int result;
	dev_t devno = MKDEV(globalmem_major, 0);

	/* ÉêÇëÉè±žºÅ*/
	if (globalmem_major)
		result = register_chrdev_region(devno, 1, "globalmem");
	else { /* ¶¯Ì¬ÉêÇëÉè±žºÅ */
		result = alloc_chrdev_region(&devno, 0, 1, "globalmem");
		globalmem_major = MAJOR(devno);
	}
	if (result < 0)
		return result;

	/* ¶¯Ì¬ÉêÇëÉè±žœá¹¹ÌåµÄÄÚŽæ*/
	globalmem_devp = kmalloc(sizeof(struct globalmem_dev), GFP_KERNEL);
	if (!globalmem_devp) {    /*ÉêÇëÊ§°Ü*/
		result =  - ENOMEM;
		goto fail_malloc;
	}

	memset(globalmem_devp, 0, sizeof(struct globalmem_dev));

	globalmem_setup_cdev(globalmem_devp, 0);
	return 0;

fail_malloc:
	unregister_chrdev_region(devno, 1);
	return result;
}

/*Ä£¿éÐ¶ÔØº¯Êý*/
void globalmem_exit(void)
{
	cdev_del(&globalmem_devp->cdev);   /*×¢Ïúcdev*/
	kfree(globalmem_devp);     /*ÊÍ·ÅÉè±žœá¹¹ÌåÄÚŽæ*/
	unregister_chrdev_region(MKDEV(globalmem_major, 0), 1); /*ÊÍ·ÅÉè±žºÅ*/
}

MODULE_AUTHOR("Barry Song <21cnbao@gmail.com>");
MODULE_LICENSE("Dual BSD/GPL");

module_param(globalmem_major, int, S_IRUGO);

module_init(globalmem_init);
module_exit(globalmem_exit);

< ch6 / Makefile >

KVERS = $(shell uname -r)

# Kernel modules
obj-m += globalmem.o

# Specify flags for the module compilation.
#EXTRA_CFLAGS=-g -O0

build: kernel_modules

kernel_modules:
	make -C /lib/modules/$(KVERS)/build M=$(CURDIR) modules

clean:
	make -C /lib/modules/$(KVERS)/build M=$(CURDIR) clean

< ch7 / globalmen.c >

/*
 * A globalmem driver as an example of char device drivers
 *
 * The initial developer of the original code is Barry Song
 * <author@linuxdriver.cn>. All Rights Reserved.
 */

#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>

#define GLOBALMEM_SIZE	0x1000	/*È«ŸÖÄÚŽæ×îŽó4K×ÖœÚ*/
#define MEM_CLEAR 0x1  /*Çå0È«ŸÖÄÚŽæ*/
#define GLOBALMEM_MAJOR 250    /*Ô€ÉèµÄglobalmemµÄÖ÷Éè±žºÅ*/

static int globalmem_major = GLOBALMEM_MAJOR;
/*globalmemÉè±žœá¹¹Ìå*/
struct globalmem_dev {
	struct cdev cdev; /*cdevœá¹¹Ìå*/
	unsigned char mem[GLOBALMEM_SIZE]; /*È«ŸÖÄÚŽæ*/
	struct semaphore sem; /*²¢·¢¿ØÖÆÓÃµÄÐÅºÅÁ¿*/
};

struct globalmem_dev *globalmem_devp; /*Éè±žœá¹¹ÌåÖžÕë*/
/*ÎÄŒþŽò¿ªº¯Êý*/
int globalmem_open(struct inode *inode, struct file *filp)
{
	/*œ«Éè±žœá¹¹ÌåÖžÕëž³ÖµžøÎÄŒþËœÓÐÊýŸÝÖžÕë*/
	filp->private_data = globalmem_devp;
	return 0;
}
/*ÎÄŒþÊÍ·Åº¯Êý*/
int globalmem_release(struct inode *inode, struct file *filp)
{
	return 0;
}

/* ioctlÉè±ž¿ØÖÆº¯Êý */
static int globalmem_ioctl(struct inode *inodep, struct file *filp, unsigned
	int cmd, unsigned long arg)
{
	struct globalmem_dev *dev = filp->private_data;/*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	switch (cmd) {
	case MEM_CLEAR:
		if (down_interruptible(&dev->sem)) /* »ñµÃÐÅºÅÁ¿*/
			return  - ERESTARTSYS;
		memset(dev->mem, 0, GLOBALMEM_SIZE);
		up(&dev->sem); /* ÊÍ·ÅÐÅºÅÁ¿*/

		printk(KERN_INFO "globalmem is set to zero\n");
		break;

	default:
		return  - EINVAL;
	}

	return 0;
}

/*¶Áº¯Êý*/
static ssize_t globalmem_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 globalmem_dev *dev = filp->private_data; /*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	/*·ÖÎöºÍ»ñÈ¡ÓÐÐ§µÄÐŽ³€¶È*/
	if (p >= GLOBALMEM_SIZE)
		return 0;
	if (count > GLOBALMEM_SIZE - p)
		count = GLOBALMEM_SIZE - p;

	if (down_interruptible(&dev->sem)) /* »ñµÃÐÅºÅÁ¿*/
		return  - ERESTARTSYS;

	/*ÄÚºË¿ÕŒä->ÓÃ»§¿ÕŒä*/
	if (copy_to_user(buf, (void *)(dev->mem + p), count)) {
		ret =  - EFAULT;
	} else {
		*ppos += count;
		ret = count;

		printk(KERN_INFO "read %u bytes(s) from %lu\n", count, p);
	}
	up(&dev->sem); /* ÊÍ·ÅÐÅºÅÁ¿*/

	return ret;
}

/*ÐŽº¯Êý*/
static ssize_t globalmem_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 globalmem_dev *dev = filp->private_data; /*»ñµÃÉè±žœá¹¹ÌåÖžÕë*/

	/*·ÖÎöºÍ»ñÈ¡ÓÐÐ§µÄÐŽ³€¶È*/
	if (p >= GLOBALMEM_SIZE)
		return 0;
	if (count > GLOBALMEM_SIZE - p)
		count = GLOBALMEM_SIZE - p;

	if (down_interruptible(&dev->sem)) /* »ñµÃÐÅºÅÁ¿*/
		return  - ERESTARTSYS;
	/*ÓÃ»§¿ÕŒä->ÄÚºË¿ÕŒä*/
	if (copy_from_user(dev->mem + p, buf, count))
		ret =  - EFAULT;
	else {
		*ppos += count;
		ret = count;

		printk(KERN_INFO "written %u bytes(s) from %lu\n", count, p);
	}
	up(&dev->sem); /* ÊÍ·ÅÐÅºÅÁ¿*/

	return ret;
}

/* seekÎÄŒþ¶šÎ»º¯Êý */
static loff_t globalmem_llseek(struct file *filp, loff_t offset, int orig)
{
	loff_t ret = 0;
	switch (orig) {
	case 0:   /*Ïà¶ÔÎÄŒþ¿ªÊŒÎ»ÖÃÆ«ÒÆ*/
		if (offset < 0)	{
			ret =  - EINVAL;
			break;
		}
		if ((unsigned int)offset > GLOBALMEM_SIZE) {
			ret =  - EINVAL;
			break;
		}
		filp->f_pos = (unsigned int)offset;
		ret = filp->f_pos;
		break;
	case 1:   /*Ïà¶ÔÎÄŒþµ±Ç°Î»ÖÃÆ«ÒÆ*/
		if ((filp->f_pos + offset) > GLOBALMEM_SIZE) {
			ret =  - EINVAL;
			break;
		}
		if ((filp->f_pos + offset) < 0) {
			ret =  - EINVAL;
			break;
		}
		filp->f_pos += offset;
		ret = filp->f_pos;
		break;
	default:
		ret =  - EINVAL;
		break;
	}
	return ret;
}

/*ÎÄŒþ²Ù×÷œá¹¹Ìå*/
static const struct file_operations globalmem_fops = {
	.owner = THIS_MODULE,
	.llseek = globalmem_llseek,
	.read = globalmem_read,
	.write = globalmem_write,
	.ioctl = globalmem_ioctl,
	.open = globalmem_open,
	.release = globalmem_release,
};

/*³õÊŒ»¯²¢×¢²ácdev*/
static void globalmem_setup_cdev(struct globalmem_dev *dev, int index)
{
	int err, devno = MKDEV(globalmem_major, index);

	cdev_init(&dev->cdev, &globalmem_fops);
	dev->cdev.owner = THIS_MODULE;
	err = cdev_add(&dev->cdev, devno, 1);
	if (err)
		printk(KERN_NOTICE "Error %d adding LED%d", err, index);
}

/*Éè±žÇý¶¯Ä£¿éŒÓÔØº¯Êý*/
int globalmem_init(void)
{
	int result;
	dev_t devno = MKDEV(globalmem_major, 0);

	/* ÉêÇëÉè±žºÅ*/
	if (globalmem_major)
		result = register_chrdev_region(devno, 1, "globalmem");
	else { /* ¶¯Ì¬ÉêÇëÉè±žºÅ */
		result = alloc_chrdev_region(&devno, 0, 1, "globalmem");
		globalmem_major = MAJOR(devno);
	}
	if (result < 0)
		return result;

	/* ¶¯Ì¬ÉêÇëÉè±žœá¹¹ÌåµÄÄÚŽæ*/
	globalmem_devp = kmalloc(sizeof(struct globalmem_dev), GFP_KERNEL);
	if (!globalmem_devp) {    /*ÉêÇëÊ§°Ü*/
		result =  - ENOMEM;
		goto fail_malloc;
	}

	memset(globalmem_devp, 0, sizeof(struct globalmem_dev));

	globalmem_setup_cdev(globalmem_devp, 0);
	init_MUTEX(&globalmem_devp->sem);   /*³õÊŒ»¯ÐÅºÅÁ¿*/
	return 0;

fail_malloc:
	unregister_chrdev_region(devno, 1);
	return result;
}

/*Ä£¿éÐ¶ÔØº¯Êý*/
void globalmem_exit(void)
{
	cdev_del(&globalmem_devp->cdev);   /*×¢Ïúcdev*/
	kfree(globalmem_devp);     /*ÊÍ·ÅÉè±žœá¹¹ÌåÄÚŽæ*/
	unregister_chrdev_region(MKDEV(globalmem_major, 0), 1); /*ÊÍ·ÅÉè±žºÅ*/
}

MODULE_AUTHOR("Barry Song <21cnbao@gmail.com>");
MODULE_LICENSE("Dual BSD/GPL");

module_param(globalmem_major, int, S_IRUGO);

module_init(globalmem_init);
module_exit(globalmem_exit);

< ch7 / Makefile >

KVERS = $(shell uname -r)

# Kernel modules
obj-m += globalmem.o

# Specify flags for the module compilation.
#EXTRA_CFLAGS=-g -O0

build: kernel_modules

kernel_modules:
	make -C /lib/modules/$(KVERS)/build M=$(CURDIR) modules

clean:
	make -C /lib/modules/$(KVERS)/build M=$(CURDIR) clean