[deliverable/linux.git] / drivers / mtd / devices / mtdram.c

/*
 * mtdram - a test mtd device
 * Author: Alexander Larsson <alex@cendio.se>
 *
 * Copyright (c) 1999 Alexander Larsson <alex@cendio.se>
 * Copyright (c) 2005 Joern Engel <joern@wh.fh-wedel.de>
 *
 * This code is GPL
 *
 */

#include <linux/module.h>
#include <linux/slab.h>
#include <linux/ioport.h>
#include <linux/vmalloc.h>
#include <linux/init.h>
#include <linux/mtd/mtd.h>
#include <linux/mtd/mtdram.h>

static unsigned long total_size = CONFIG_MTDRAM_TOTAL_SIZE;
static unsigned long erase_size = CONFIG_MTDRAM_ERASE_SIZE;
#define MTDRAM_TOTAL_SIZE (total_size * 1024)
#define MTDRAM_ERASE_SIZE (erase_size * 1024)

#ifdef MODULE
module_param(total_size, ulong, 0);
MODULE_PARM_DESC(total_size, "Total device size in KiB");
module_param(erase_size, ulong, 0);
MODULE_PARM_DESC(erase_size, "Device erase block size in KiB");
#endif

// We could store these in the mtd structure, but we only support 1 device..
static struct mtd_info *mtd_info;

static int check_offs_len(struct mtd_info *mtd, loff_t ofs, uint64_t len)
{
	int ret = 0;

	/* Start address must align on block boundary */
	if (mtd_mod_by_eb(ofs, mtd)) {
		pr_debug("%s: unaligned address\n", __func__);
		ret = -EINVAL;
	}

	/* Length must align on block boundary */
	if (mtd_mod_by_eb(len, mtd)) {
		pr_debug("%s: length not block aligned\n", __func__);
		ret = -EINVAL;
	}

	return ret;
}

static int ram_erase(struct mtd_info *mtd, struct erase_info *instr)
{
	if (check_offs_len(mtd, instr->addr, instr->len))
		return -EINVAL;
	memset((char *)mtd->priv + instr->addr, 0xff, instr->len);
	instr->state = MTD_ERASE_DONE;
	mtd_erase_callback(instr);
	return 0;
}

static int ram_point(struct mtd_info *mtd, loff_t from, size_t len,
		size_t *retlen, void **virt, resource_size_t *phys)
{
	*virt = mtd->priv + from;
	*retlen = len;
	return 0;
}

static int ram_unpoint(struct mtd_info *mtd, loff_t from, size_t len)
{
	return 0;
}

/*
 * Allow NOMMU mmap() to directly map the device (if not NULL)
 * - return the address to which the offset maps
 * - return -ENOSYS to indicate refusal to do the mapping
 */
static unsigned long ram_get_unmapped_area(struct mtd_info *mtd,
					   unsigned long len,
					   unsigned long offset,
					   unsigned long flags)
{
	return (unsigned long) mtd->priv + offset;
}

static int ram_read(struct mtd_info *mtd, loff_t from, size_t len,
		size_t *retlen, u_char *buf)
{
	memcpy(buf, mtd->priv + from, len);
	*retlen = len;
	return 0;
}

static int ram_write(struct mtd_info *mtd, loff_t to, size_t len,
		size_t *retlen, const u_char *buf)
{
	memcpy((char *)mtd->priv + to, buf, len);
	*retlen = len;
	return 0;
}

static void __exit cleanup_mtdram(void)
{
	if (mtd_info) {
		mtd_device_unregister(mtd_info);
		vfree(mtd_info->priv);
		kfree(mtd_info);
	}
}

int mtdram_init_device(struct mtd_info *mtd, void *mapped_address,
		unsigned long size, const char *name)
{
	memset(mtd, 0, sizeof(*mtd));

	/* Setup the MTD structure */
	mtd->name = name;
	mtd->type = MTD_RAM;
	mtd->flags = MTD_CAP_RAM;
	mtd->size = size;
	mtd->writesize = 1;
	mtd->writebufsize = 64; /* Mimic CFI NOR flashes */
	mtd->erasesize = MTDRAM_ERASE_SIZE;
	mtd->priv = mapped_address;

	mtd->owner = THIS_MODULE;
	mtd->_erase = ram_erase;
	mtd->_point = ram_point;
	mtd->_unpoint = ram_unpoint;
	mtd->_get_unmapped_area = ram_get_unmapped_area;
	mtd->_read = ram_read;
	mtd->_write = ram_write;

	if (mtd_device_register(mtd, NULL, 0))
		return -EIO;

	return 0;
}

static int __init init_mtdram(void)
{
	void *addr;
	int err;

	if (!total_size)
		return -EINVAL;

	/* Allocate some memory */
	mtd_info = kmalloc(sizeof(struct mtd_info), GFP_KERNEL);
	if (!mtd_info)
		return -ENOMEM;

	addr = vmalloc(MTDRAM_TOTAL_SIZE);
	if (!addr) {
		kfree(mtd_info);
		mtd_info = NULL;
		return -ENOMEM;
	}
	err = mtdram_init_device(mtd_info, addr, MTDRAM_TOTAL_SIZE, "mtdram test device");
	if (err) {
		vfree(addr);
		kfree(mtd_info);
		mtd_info = NULL;
		return err;
	}
	memset(mtd_info->priv, 0xff, MTDRAM_TOTAL_SIZE);
	return err;
}

module_init(init_mtdram);
module_exit(cleanup_mtdram);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Alexander Larsson <alexl@redhat.com>");
MODULE_DESCRIPTION("Simulated MTD driver for testing");
Commit	Line	Data
1da177e4 LT	1	/*
1da177e4 LT	2	* mtdram - a test mtd device
1da177e4 LT	3	* Author: Alexander Larsson <alex@cendio.se>
	4	*
	5	* Copyright (c) 1999 Alexander Larsson <alex@cendio.se>
c13cbf3b	6	* Copyright (c) 2005 Joern Engel <joern@wh.fh-wedel.de>
1da177e4 LT	7	*
	8	* This code is GPL
	9	*
	10	*/
	11
1da177e4 LT	12	#include <linux/module.h>
	13	#include <linux/slab.h>
	14	#include <linux/ioport.h>
	15	#include <linux/vmalloc.h>
	16	#include <linux/init.h>
1da177e4	17	#include <linux/mtd/mtd.h>
456d9fc9	18	#include <linux/mtd/mtdram.h>
1da177e4	19
1da177e4 LT	20	static unsigned long total_size = CONFIG_MTDRAM_TOTAL_SIZE;
1da177e4 LT	21	static unsigned long erase_size = CONFIG_MTDRAM_ERASE_SIZE;
1da177e4 LT	22	#define MTDRAM_TOTAL_SIZE (total_size * 1024)
1da177e4 LT	23	#define MTDRAM_ERASE_SIZE (erase_size * 1024)
1da177e4	24
c13cbf3b JE	25	#ifdef MODULE
	26	module_param(total_size, ulong, 0);
	27	MODULE_PARM_DESC(total_size, "Total device size in KiB");
	28	module_param(erase_size, ulong, 0);
	29	MODULE_PARM_DESC(erase_size, "Device erase block size in KiB");
	30	#endif
1da177e4 LT	31
	32	// We could store these in the mtd structure, but we only support 1 device..
	33	static struct mtd_info *mtd_info;
	34
ab84fce5 DY	35	static int check_offs_len(struct mtd_info *mtd, loff_t ofs, uint64_t len)
	36	{
	37	int ret = 0;
	38
	39	/* Start address must align on block boundary */
	40	if (mtd_mod_by_eb(ofs, mtd)) {
	41	pr_debug("%s: unaligned address\n", __func__);
	42	ret = -EINVAL;
	43	}
	44
	45	/* Length must align on block boundary */
	46	if (mtd_mod_by_eb(len, mtd)) {
	47	pr_debug("%s: length not block aligned\n", __func__);
	48	ret = -EINVAL;
	49	}
	50
	51	return ret;
	52	}
	53
c13cbf3b	54	static int ram_erase(struct mtd_info mtd, struct erase_info instr)
1da177e4	55	{
ab84fce5 DY	56	if (check_offs_len(mtd, instr->addr, instr->len))
ab84fce5 DY	57	return -EINVAL;
c13cbf3b	58	memset((char *)mtd->priv + instr->addr, 0xff, instr->len);
c13cbf3b JE	59	instr->state = MTD_ERASE_DONE;
c13cbf3b JE	60	mtd_erase_callback(instr);
c13cbf3b	61	return 0;
1da177e4 LT	62	}
1da177e4 LT	63
c13cbf3b	64	static int ram_point(struct mtd_info *mtd, loff_t from, size_t len,
a98889f3	65	size_t retlen, void virt, resource_size_t phys)
1da177e4	66	{
a98889f3	67	*virt = mtd->priv + from;
c13cbf3b JE	68	*retlen = len;
c13cbf3b JE	69	return 0;
1da177e4 LT	70	}
1da177e4 LT	71
5e4e6e3f	72	static int ram_unpoint(struct mtd_info *mtd, loff_t from, size_t len)
1da177e4	73	{
5e4e6e3f	74	return 0;
1da177e4 LT	75	}
1da177e4 LT	76
402d3265 DH	77	/*
	78	* Allow NOMMU mmap() to directly map the device (if not NULL)
	79	* - return the address to which the offset maps
	80	* - return -ENOSYS to indicate refusal to do the mapping
	81	*/
	82	static unsigned long ram_get_unmapped_area(struct mtd_info *mtd,
	83	unsigned long len,
	84	unsigned long offset,
	85	unsigned long flags)
	86	{
	87	return (unsigned long) mtd->priv + offset;
	88	}
	89
1da177e4	90	static int ram_read(struct mtd_info *mtd, loff_t from, size_t len,
c13cbf3b	91	size_t retlen, u_char buf)
1da177e4	92	{
c13cbf3b	93	memcpy(buf, mtd->priv + from, len);
c13cbf3b JE	94	*retlen = len;
c13cbf3b JE	95	return 0;
1da177e4 LT	96	}
	97
	98	static int ram_write(struct mtd_info *mtd, loff_t to, size_t len,
c13cbf3b	99	size_t retlen, const u_char buf)
1da177e4	100	{
c13cbf3b	101	memcpy((char *)mtd->priv + to, buf, len);
c13cbf3b JE	102	*retlen = len;
c13cbf3b JE	103	return 0;
1da177e4 LT	104	}
	105
	106	static void __exit cleanup_mtdram(void)
	107	{
c13cbf3b	108	if (mtd_info) {
ee0e87b1	109	mtd_device_unregister(mtd_info);
f9101210	110	vfree(mtd_info->priv);
c13cbf3b JE	111	kfree(mtd_info);
c13cbf3b JE	112	}
1da177e4 LT	113	}
1da177e4 LT	114
c13cbf3b	115	int mtdram_init_device(struct mtd_info mtd, void mapped_address,
0a8899b3	116	unsigned long size, const char *name)
1da177e4	117	{
c13cbf3b JE	118	memset(mtd, 0, sizeof(*mtd));
	119
	120	/* Setup the MTD structure */
	121	mtd->name = name;
21c8db9e	122	mtd->type = MTD_RAM;
c13cbf3b JE	123	mtd->flags = MTD_CAP_RAM;
c13cbf3b JE	124	mtd->size = size;
783ed81f	125	mtd->writesize = 1;
07be303d	126	mtd->writebufsize = 64; /* Mimic CFI NOR flashes */
c13cbf3b JE	127	mtd->erasesize = MTDRAM_ERASE_SIZE;
	128	mtd->priv = mapped_address;
	129
	130	mtd->owner = THIS_MODULE;
3c3c10bb AB	131	mtd->_erase = ram_erase;
	132	mtd->_point = ram_point;
	133	mtd->_unpoint = ram_unpoint;
	134	mtd->_get_unmapped_area = ram_get_unmapped_area;
	135	mtd->_read = ram_read;
	136	mtd->_write = ram_write;
c13cbf3b	137
ee0e87b1	138	if (mtd_device_register(mtd, NULL, 0))
c13cbf3b	139	return -EIO;
c13cbf3b JE	140
c13cbf3b JE	141	return 0;
1da177e4	142	}
1da177e4 LT	143
	144	static int __init init_mtdram(void)
	145	{
c13cbf3b JE	146	void *addr;
	147	int err;
	148
	149	if (!total_size)
	150	return -EINVAL;
	151
	152	/* Allocate some memory */
	153	mtd_info = kmalloc(sizeof(struct mtd_info), GFP_KERNEL);
	154	if (!mtd_info)
	155	return -ENOMEM;
	156
	157	addr = vmalloc(MTDRAM_TOTAL_SIZE);
	158	if (!addr) {
	159	kfree(mtd_info);
	160	mtd_info = NULL;
	161	return -ENOMEM;
	162	}
	163	err = mtdram_init_device(mtd_info, addr, MTDRAM_TOTAL_SIZE, "mtdram test device");
	164	if (err) {
	165	vfree(addr);
	166	kfree(mtd_info);
	167	mtd_info = NULL;
	168	return err;
	169	}
	170	memset(mtd_info->priv, 0xff, MTDRAM_TOTAL_SIZE);
	171	return err;
1da177e4	172	}
1da177e4 LT	173
	174	module_init(init_mtdram);
	175	module_exit(cleanup_mtdram);
	176
	177	MODULE_LICENSE("GPL");
	178	MODULE_AUTHOR("Alexander Larsson <alexl@redhat.com>");
	179	MODULE_DESCRIPTION("Simulated MTD driver for testing");