1// SPDX-License-Identifier: GPL-2.0+
2/*
3 * (C) Copyright 2008 Semihalf
4 *
5 * Written by: Piotr Ziecik <kosmo@semihalf.com>
6 */
7
8#include <common.h>
9#include <dma.h>
10#include <flash.h>
11#include <malloc.h>
12
13#include <linux/errno.h>
14#include <linux/mtd/mtd.h>
15#include <linux/mtd/concat.h>
16#include <mtd/cfi_flash.h>
17
18static struct mtd_info cfi_mtd_info[CFI_MAX_FLASH_BANKS];
19static char cfi_mtd_names[CFI_MAX_FLASH_BANKS][16];
20#ifdef CONFIG_MTD_CONCAT
21static char c_mtd_name[16];
22#endif
23
24static int cfi_mtd_erase(struct mtd_info *mtd, struct erase_info *instr)
25{
26	flash_info_t *fi = mtd->priv;
27	size_t a_start = fi->start[0] + instr->addr;
28	size_t a_end = a_start + instr->len;
29	int s_first = -1;
30	int s_last = -1;
31	int error, sect;
32
33	for (sect = 0; sect < fi->sector_count; sect++) {
34		if (a_start == fi->start[sect])
35			s_first = sect;
36
37		if (sect < fi->sector_count - 1) {
38			if (a_end == fi->start[sect + 1]) {
39				s_last = sect;
40				break;
41			}
42		} else {
43			s_last = sect;
44			break;
45		}
46	}
47
48	if (s_first >= 0 && s_first <= s_last) {
49		instr->state = MTD_ERASING;
50
51		flash_set_verbose(0);
52		error = flash_erase(fi, s_first, s_last);
53		flash_set_verbose(1);
54
55		if (error) {
56			instr->state = MTD_ERASE_FAILED;
57			return -EIO;
58		}
59
60		instr->state = MTD_ERASE_DONE;
61		return 0;
62	}
63
64	return -EINVAL;
65}
66
67static int cfi_mtd_read(struct mtd_info *mtd, loff_t from, size_t len,
68	size_t *retlen, u_char *buf)
69{
70	flash_info_t *fi = mtd->priv;
71	u_char *f = (u_char*)(fi->start[0]) + from;
72
73	if (dma_memcpy(buf, f, len) < 0)
74		memcpy(buf, f, len);
75	*retlen = len;
76
77	return 0;
78}
79
80static int cfi_mtd_write(struct mtd_info *mtd, loff_t to, size_t len,
81	size_t *retlen, const u_char *buf)
82{
83	flash_info_t *fi = mtd->priv;
84	u_long t = fi->start[0] + to;
85	int error;
86
87	flash_set_verbose(0);
88	error = write_buff(fi, (u_char*)buf, t, len);
89	flash_set_verbose(1);
90
91	if (!error) {
92		*retlen = len;
93		return 0;
94	}
95
96	return -EIO;
97}
98
99static void cfi_mtd_sync(struct mtd_info *mtd)
100{
101	/*
102	 * This function should wait until all pending operations
103	 * finish. However this driver is fully synchronous, so
104	 * this function returns immediately
105	 */
106}
107
108static int cfi_mtd_lock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
109{
110	flash_info_t *fi = mtd->priv;
111
112	flash_set_verbose(0);
113	flash_protect(FLAG_PROTECT_SET, fi->start[0] + ofs,
114					fi->start[0] + ofs + len - 1, fi);
115	flash_set_verbose(1);
116
117	return 0;
118}
119
120static int cfi_mtd_unlock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
121{
122	flash_info_t *fi = mtd->priv;
123
124	flash_set_verbose(0);
125	flash_protect(FLAG_PROTECT_CLEAR, fi->start[0] + ofs,
126					fi->start[0] + ofs + len - 1, fi);
127	flash_set_verbose(1);
128
129	return 0;
130}
131
132static int cfi_mtd_set_erasesize(struct mtd_info *mtd, flash_info_t *fi)
133{
134	int sect_size = 0;
135	int sect_size_old = 0;
136	int sect;
137	int regions = 0;
138	int numblocks = 0;
139	ulong offset;
140	ulong base_addr;
141
142	/*
143	 * First detect the number of eraseregions so that we can allocate
144	 * the array of eraseregions correctly
145	 */
146	for (sect = 0; sect < fi->sector_count; sect++) {
147		if (sect_size_old != flash_sector_size(fi, sect))
148			regions++;
149		sect_size_old = flash_sector_size(fi, sect);
150	}
151
152	switch (regions) {
153	case 0:
154		return 1;
155	case 1:	/* flash has uniform erase size */
156		mtd->numeraseregions = 0;
157		mtd->erasesize = sect_size_old;
158		return 0;
159	}
160
161	mtd->numeraseregions = regions;
162	mtd->eraseregions = malloc(sizeof(struct mtd_erase_region_info) * regions);
163
164	/*
165	 * Now detect the largest sector and fill the eraseregions
166	 */
167	regions = 0;
168	base_addr = offset = fi->start[0];
169	sect_size_old = flash_sector_size(fi, 0);
170	for (sect = 0; sect < fi->sector_count; sect++) {
171		if (sect_size_old != flash_sector_size(fi, sect)) {
172			mtd->eraseregions[regions].offset = offset - base_addr;
173			mtd->eraseregions[regions].erasesize = sect_size_old;
174			mtd->eraseregions[regions].numblocks = numblocks;
175			/* Now start counting the next eraseregions */
176			numblocks = 0;
177			regions++;
178			offset = fi->start[sect];
179		}
180		numblocks++;
181
182		/*
183		 * Select the largest sector size as erasesize (e.g. for UBI)
184		 */
185		if (flash_sector_size(fi, sect) > sect_size)
186			sect_size = flash_sector_size(fi, sect);
187
188		sect_size_old = flash_sector_size(fi, sect);
189	}
190
191	/*
192	 * Set the last region
193	 */
194	mtd->eraseregions[regions].offset = offset - base_addr;
195	mtd->eraseregions[regions].erasesize = sect_size_old;
196	mtd->eraseregions[regions].numblocks = numblocks;
197
198	mtd->erasesize = sect_size;
199
200	return 0;
201}
202
203int cfi_mtd_init(void)
204{
205	struct mtd_info *mtd;
206	flash_info_t *fi;
207	int error, i;
208#ifdef CONFIG_MTD_CONCAT
209	int devices_found = 0;
210	struct mtd_info *mtd_list[CFI_FLASH_BANKS];
211#endif
212
213	for (i = 0; i < CFI_FLASH_BANKS; i++) {
214		fi = &flash_info[i];
215		mtd = &cfi_mtd_info[i];
216
217		memset(mtd, 0, sizeof(struct mtd_info));
218
219		error = cfi_mtd_set_erasesize(mtd, fi);
220		if (error)
221			continue;
222
223		sprintf(cfi_mtd_names[i], "nor%d", i);
224#ifdef CONFIG_CFI_FLASH
225		mtd->dev		= fi->dev;
226#endif
227		mtd->name		= cfi_mtd_names[i];
228		mtd->type		= MTD_NORFLASH;
229		mtd->flags		= MTD_CAP_NORFLASH;
230		mtd->size		= fi->size;
231		mtd->writesize		= 1;
232		mtd->writebufsize	= mtd->writesize;
233
234		mtd->_erase		= cfi_mtd_erase;
235		mtd->_read		= cfi_mtd_read;
236		mtd->_write		= cfi_mtd_write;
237		mtd->_sync		= cfi_mtd_sync;
238		mtd->_lock		= cfi_mtd_lock;
239		mtd->_unlock		= cfi_mtd_unlock;
240		mtd->priv		= fi;
241
242		if (add_mtd_device(mtd))
243			return -ENOMEM;
244
245#ifdef CONFIG_MTD_CONCAT
246		mtd_list[devices_found++] = mtd;
247#endif
248	}
249
250#ifdef CONFIG_MTD_CONCAT
251	if (devices_found > 1) {
252		/*
253		 * We detected multiple devices. Concatenate them together.
254		 */
255		sprintf(c_mtd_name, "nor%d", devices_found);
256		mtd = mtd_concat_create(mtd_list, devices_found, c_mtd_name);
257
258		if (mtd == NULL)
259			return -ENXIO;
260
261		if (add_mtd_device(mtd))
262			return -ENOMEM;
263	}
264#endif /* CONFIG_MTD_CONCAT */
265
266	return 0;
267}
268