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