xref: /openbmc/linux/drivers/mtd/devices/spear_smi.c (revision d0e22329)
1 /*
2  * SMI (Serial Memory Controller) device driver for Serial NOR Flash on
3  * SPEAr platform
4  * The serial nor interface is largely based on m25p80.c, however the SPI
5  * interface has been replaced by SMI.
6  *
7  * Copyright © 2010 STMicroelectronics.
8  * Ashish Priyadarshi
9  * Shiraz Hashim <shiraz.linux.kernel@gmail.com>
10  *
11  * This file is licensed under the terms of the GNU General Public
12  * License version 2. This program is licensed "as is" without any
13  * warranty of any kind, whether express or implied.
14  */
15 
16 #include <linux/clk.h>
17 #include <linux/delay.h>
18 #include <linux/device.h>
19 #include <linux/err.h>
20 #include <linux/errno.h>
21 #include <linux/interrupt.h>
22 #include <linux/io.h>
23 #include <linux/ioport.h>
24 #include <linux/jiffies.h>
25 #include <linux/kernel.h>
26 #include <linux/module.h>
27 #include <linux/param.h>
28 #include <linux/platform_device.h>
29 #include <linux/pm.h>
30 #include <linux/mtd/mtd.h>
31 #include <linux/mtd/partitions.h>
32 #include <linux/mtd/spear_smi.h>
33 #include <linux/mutex.h>
34 #include <linux/sched.h>
35 #include <linux/slab.h>
36 #include <linux/wait.h>
37 #include <linux/of.h>
38 #include <linux/of_address.h>
39 
40 /* SMI clock rate */
41 #define SMI_MAX_CLOCK_FREQ	50000000 /* 50 MHz */
42 
43 /* MAX time out to safely come out of a erase or write busy conditions */
44 #define SMI_PROBE_TIMEOUT	(HZ / 10)
45 #define SMI_MAX_TIME_OUT	(3 * HZ)
46 
47 /* timeout for command completion */
48 #define SMI_CMD_TIMEOUT		(HZ / 10)
49 
50 /* registers of smi */
51 #define SMI_CR1		0x0	/* SMI control register 1 */
52 #define SMI_CR2		0x4	/* SMI control register 2 */
53 #define SMI_SR		0x8	/* SMI status register */
54 #define SMI_TR		0xC	/* SMI transmit register */
55 #define SMI_RR		0x10	/* SMI receive register */
56 
57 /* defines for control_reg 1 */
58 #define BANK_EN		(0xF << 0)	/* enables all banks */
59 #define DSEL_TIME	(0x6 << 4)	/* Deselect time 6 + 1 SMI_CK periods */
60 #define SW_MODE		(0x1 << 28)	/* enables SW Mode */
61 #define WB_MODE		(0x1 << 29)	/* Write Burst Mode */
62 #define FAST_MODE	(0x1 << 15)	/* Fast Mode */
63 #define HOLD1		(0x1 << 16)	/* Clock Hold period selection */
64 
65 /* defines for control_reg 2 */
66 #define SEND		(0x1 << 7)	/* Send data */
67 #define TFIE		(0x1 << 8)	/* Transmission Flag Interrupt Enable */
68 #define WCIE		(0x1 << 9)	/* Write Complete Interrupt Enable */
69 #define RD_STATUS_REG	(0x1 << 10)	/* reads status reg */
70 #define WE		(0x1 << 11)	/* Write Enable */
71 
72 #define TX_LEN_SHIFT	0
73 #define RX_LEN_SHIFT	4
74 #define BANK_SHIFT	12
75 
76 /* defines for status register */
77 #define SR_WIP		0x1	/* Write in progress */
78 #define SR_WEL		0x2	/* Write enable latch */
79 #define SR_BP0		0x4	/* Block protect 0 */
80 #define SR_BP1		0x8	/* Block protect 1 */
81 #define SR_BP2		0x10	/* Block protect 2 */
82 #define SR_SRWD		0x80	/* SR write protect */
83 #define TFF		0x100	/* Transfer Finished Flag */
84 #define WCF		0x200	/* Transfer Finished Flag */
85 #define ERF1		0x400	/* Forbidden Write Request */
86 #define ERF2		0x800	/* Forbidden Access */
87 
88 #define WM_SHIFT	12
89 
90 /* flash opcodes */
91 #define OPCODE_RDID	0x9f	/* Read JEDEC ID */
92 
93 /* Flash Device Ids maintenance section */
94 
95 /* data structure to maintain flash ids from different vendors */
96 struct flash_device {
97 	char *name;
98 	u8 erase_cmd;
99 	u32 device_id;
100 	u32 pagesize;
101 	unsigned long sectorsize;
102 	unsigned long size_in_bytes;
103 };
104 
105 #define FLASH_ID(n, es, id, psize, ssize, size)	\
106 {				\
107 	.name = n,		\
108 	.erase_cmd = es,	\
109 	.device_id = id,	\
110 	.pagesize = psize,	\
111 	.sectorsize = ssize,	\
112 	.size_in_bytes = size	\
113 }
114 
115 static struct flash_device flash_devices[] = {
116 	FLASH_ID("st m25p16"     , 0xd8, 0x00152020, 0x100, 0x10000, 0x200000),
117 	FLASH_ID("st m25p32"     , 0xd8, 0x00162020, 0x100, 0x10000, 0x400000),
118 	FLASH_ID("st m25p64"     , 0xd8, 0x00172020, 0x100, 0x10000, 0x800000),
119 	FLASH_ID("st m25p128"    , 0xd8, 0x00182020, 0x100, 0x40000, 0x1000000),
120 	FLASH_ID("st m25p05"     , 0xd8, 0x00102020, 0x80 , 0x8000 , 0x10000),
121 	FLASH_ID("st m25p10"     , 0xd8, 0x00112020, 0x80 , 0x8000 , 0x20000),
122 	FLASH_ID("st m25p20"     , 0xd8, 0x00122020, 0x100, 0x10000, 0x40000),
123 	FLASH_ID("st m25p40"     , 0xd8, 0x00132020, 0x100, 0x10000, 0x80000),
124 	FLASH_ID("st m25p80"     , 0xd8, 0x00142020, 0x100, 0x10000, 0x100000),
125 	FLASH_ID("st m45pe10"    , 0xd8, 0x00114020, 0x100, 0x10000, 0x20000),
126 	FLASH_ID("st m45pe20"    , 0xd8, 0x00124020, 0x100, 0x10000, 0x40000),
127 	FLASH_ID("st m45pe40"    , 0xd8, 0x00134020, 0x100, 0x10000, 0x80000),
128 	FLASH_ID("st m45pe80"    , 0xd8, 0x00144020, 0x100, 0x10000, 0x100000),
129 	FLASH_ID("sp s25fl004"   , 0xd8, 0x00120201, 0x100, 0x10000, 0x80000),
130 	FLASH_ID("sp s25fl008"   , 0xd8, 0x00130201, 0x100, 0x10000, 0x100000),
131 	FLASH_ID("sp s25fl016"   , 0xd8, 0x00140201, 0x100, 0x10000, 0x200000),
132 	FLASH_ID("sp s25fl032"   , 0xd8, 0x00150201, 0x100, 0x10000, 0x400000),
133 	FLASH_ID("sp s25fl064"   , 0xd8, 0x00160201, 0x100, 0x10000, 0x800000),
134 	FLASH_ID("atmel 25f512"  , 0x52, 0x0065001F, 0x80 , 0x8000 , 0x10000),
135 	FLASH_ID("atmel 25f1024" , 0x52, 0x0060001F, 0x100, 0x8000 , 0x20000),
136 	FLASH_ID("atmel 25f2048" , 0x52, 0x0063001F, 0x100, 0x10000, 0x40000),
137 	FLASH_ID("atmel 25f4096" , 0x52, 0x0064001F, 0x100, 0x10000, 0x80000),
138 	FLASH_ID("atmel 25fs040" , 0xd7, 0x0004661F, 0x100, 0x10000, 0x80000),
139 	FLASH_ID("mac 25l512"    , 0xd8, 0x001020C2, 0x010, 0x10000, 0x10000),
140 	FLASH_ID("mac 25l1005"   , 0xd8, 0x001120C2, 0x010, 0x10000, 0x20000),
141 	FLASH_ID("mac 25l2005"   , 0xd8, 0x001220C2, 0x010, 0x10000, 0x40000),
142 	FLASH_ID("mac 25l4005"   , 0xd8, 0x001320C2, 0x010, 0x10000, 0x80000),
143 	FLASH_ID("mac 25l4005a"  , 0xd8, 0x001320C2, 0x010, 0x10000, 0x80000),
144 	FLASH_ID("mac 25l8005"   , 0xd8, 0x001420C2, 0x010, 0x10000, 0x100000),
145 	FLASH_ID("mac 25l1605"   , 0xd8, 0x001520C2, 0x100, 0x10000, 0x200000),
146 	FLASH_ID("mac 25l1605a"  , 0xd8, 0x001520C2, 0x010, 0x10000, 0x200000),
147 	FLASH_ID("mac 25l3205"   , 0xd8, 0x001620C2, 0x100, 0x10000, 0x400000),
148 	FLASH_ID("mac 25l3205a"  , 0xd8, 0x001620C2, 0x100, 0x10000, 0x400000),
149 	FLASH_ID("mac 25l6405"   , 0xd8, 0x001720C2, 0x100, 0x10000, 0x800000),
150 };
151 
152 /* Define spear specific structures */
153 
154 struct spear_snor_flash;
155 
156 /**
157  * struct spear_smi - Structure for SMI Device
158  *
159  * @clk: functional clock
160  * @status: current status register of SMI.
161  * @clk_rate: functional clock rate of SMI (default: SMI_MAX_CLOCK_FREQ)
162  * @lock: lock to prevent parallel access of SMI.
163  * @io_base: base address for registers of SMI.
164  * @pdev: platform device
165  * @cmd_complete: queue to wait for command completion of NOR-flash.
166  * @num_flashes: number of flashes actually present on board.
167  * @flash: separate structure for each Serial NOR-flash attached to SMI.
168  */
169 struct spear_smi {
170 	struct clk *clk;
171 	u32 status;
172 	unsigned long clk_rate;
173 	struct mutex lock;
174 	void __iomem *io_base;
175 	struct platform_device *pdev;
176 	wait_queue_head_t cmd_complete;
177 	u32 num_flashes;
178 	struct spear_snor_flash *flash[MAX_NUM_FLASH_CHIP];
179 };
180 
181 /**
182  * struct spear_snor_flash - Structure for Serial NOR Flash
183  *
184  * @bank: Bank number(0, 1, 2, 3) for each NOR-flash.
185  * @dev_id: Device ID of NOR-flash.
186  * @lock: lock to manage flash read, write and erase operations
187  * @mtd: MTD info for each NOR-flash.
188  * @num_parts: Total number of partition in each bank of NOR-flash.
189  * @parts: Partition info for each bank of NOR-flash.
190  * @page_size: Page size of NOR-flash.
191  * @base_addr: Base address of NOR-flash.
192  * @erase_cmd: erase command may vary on different flash types
193  * @fast_mode: flash supports read in fast mode
194  */
195 struct spear_snor_flash {
196 	u32 bank;
197 	u32 dev_id;
198 	struct mutex lock;
199 	struct mtd_info mtd;
200 	u32 num_parts;
201 	struct mtd_partition *parts;
202 	u32 page_size;
203 	void __iomem *base_addr;
204 	u8 erase_cmd;
205 	u8 fast_mode;
206 };
207 
208 static inline struct spear_snor_flash *get_flash_data(struct mtd_info *mtd)
209 {
210 	return container_of(mtd, struct spear_snor_flash, mtd);
211 }
212 
213 /**
214  * spear_smi_read_sr - Read status register of flash through SMI
215  * @dev: structure of SMI information.
216  * @bank: bank to which flash is connected
217  *
218  * This routine will return the status register of the flash chip present at the
219  * given bank.
220  */
221 static int spear_smi_read_sr(struct spear_smi *dev, u32 bank)
222 {
223 	int ret;
224 	u32 ctrlreg1;
225 
226 	mutex_lock(&dev->lock);
227 	dev->status = 0; /* Will be set in interrupt handler */
228 
229 	ctrlreg1 = readl(dev->io_base + SMI_CR1);
230 	/* program smi in hw mode */
231 	writel(ctrlreg1 & ~(SW_MODE | WB_MODE), dev->io_base + SMI_CR1);
232 
233 	/* performing a rsr instruction in hw mode */
234 	writel((bank << BANK_SHIFT) | RD_STATUS_REG | TFIE,
235 			dev->io_base + SMI_CR2);
236 
237 	/* wait for tff */
238 	ret = wait_event_interruptible_timeout(dev->cmd_complete,
239 			dev->status & TFF, SMI_CMD_TIMEOUT);
240 
241 	/* copy dev->status (lower 16 bits) in order to release lock */
242 	if (ret > 0)
243 		ret = dev->status & 0xffff;
244 	else if (ret == 0)
245 		ret = -ETIMEDOUT;
246 
247 	/* restore the ctrl regs state */
248 	writel(ctrlreg1, dev->io_base + SMI_CR1);
249 	writel(0, dev->io_base + SMI_CR2);
250 	mutex_unlock(&dev->lock);
251 
252 	return ret;
253 }
254 
255 /**
256  * spear_smi_wait_till_ready - wait till flash is ready
257  * @dev: structure of SMI information.
258  * @bank: flash corresponding to this bank
259  * @timeout: timeout for busy wait condition
260  *
261  * This routine checks for WIP (write in progress) bit in Status register
262  * If successful the routine returns 0 else -EBUSY
263  */
264 static int spear_smi_wait_till_ready(struct spear_smi *dev, u32 bank,
265 		unsigned long timeout)
266 {
267 	unsigned long finish;
268 	int status;
269 
270 	finish = jiffies + timeout;
271 	do {
272 		status = spear_smi_read_sr(dev, bank);
273 		if (status < 0) {
274 			if (status == -ETIMEDOUT)
275 				continue; /* try till finish */
276 			return status;
277 		} else if (!(status & SR_WIP)) {
278 			return 0;
279 		}
280 
281 		cond_resched();
282 	} while (!time_after_eq(jiffies, finish));
283 
284 	dev_err(&dev->pdev->dev, "smi controller is busy, timeout\n");
285 	return -EBUSY;
286 }
287 
288 /**
289  * spear_smi_int_handler - SMI Interrupt Handler.
290  * @irq: irq number
291  * @dev_id: structure of SMI device, embedded in dev_id.
292  *
293  * The handler clears all interrupt conditions and records the status in
294  * dev->status which is used by the driver later.
295  */
296 static irqreturn_t spear_smi_int_handler(int irq, void *dev_id)
297 {
298 	u32 status = 0;
299 	struct spear_smi *dev = dev_id;
300 
301 	status = readl(dev->io_base + SMI_SR);
302 
303 	if (unlikely(!status))
304 		return IRQ_NONE;
305 
306 	/* clear all interrupt conditions */
307 	writel(0, dev->io_base + SMI_SR);
308 
309 	/* copy the status register in dev->status */
310 	dev->status |= status;
311 
312 	/* send the completion */
313 	wake_up_interruptible(&dev->cmd_complete);
314 
315 	return IRQ_HANDLED;
316 }
317 
318 /**
319  * spear_smi_hw_init - initializes the smi controller.
320  * @dev: structure of smi device
321  *
322  * this routine initializes the smi controller wit the default values
323  */
324 static void spear_smi_hw_init(struct spear_smi *dev)
325 {
326 	unsigned long rate = 0;
327 	u32 prescale = 0;
328 	u32 val;
329 
330 	rate = clk_get_rate(dev->clk);
331 
332 	/* functional clock of smi */
333 	prescale = DIV_ROUND_UP(rate, dev->clk_rate);
334 
335 	/*
336 	 * setting the standard values, fast mode, prescaler for
337 	 * SMI_MAX_CLOCK_FREQ (50MHz) operation and bank enable
338 	 */
339 	val = HOLD1 | BANK_EN | DSEL_TIME | (prescale << 8);
340 
341 	mutex_lock(&dev->lock);
342 	/* clear all interrupt conditions */
343 	writel(0, dev->io_base + SMI_SR);
344 
345 	writel(val, dev->io_base + SMI_CR1);
346 	mutex_unlock(&dev->lock);
347 }
348 
349 /**
350  * get_flash_index - match chip id from a flash list.
351  * @flash_id: a valid nor flash chip id obtained from board.
352  *
353  * try to validate the chip id by matching from a list, if not found then simply
354  * returns negative. In case of success returns index in to the flash devices
355  * array.
356  */
357 static int get_flash_index(u32 flash_id)
358 {
359 	int index;
360 
361 	/* Matches chip-id to entire list of 'serial-nor flash' ids */
362 	for (index = 0; index < ARRAY_SIZE(flash_devices); index++) {
363 		if (flash_devices[index].device_id == flash_id)
364 			return index;
365 	}
366 
367 	/* Memory chip is not listed and not supported */
368 	return -ENODEV;
369 }
370 
371 /**
372  * spear_smi_write_enable - Enable the flash to do write operation
373  * @dev: structure of SMI device
374  * @bank: enable write for flash connected to this bank
375  *
376  * Set write enable latch with Write Enable command.
377  * Returns 0 on success.
378  */
379 static int spear_smi_write_enable(struct spear_smi *dev, u32 bank)
380 {
381 	int ret;
382 	u32 ctrlreg1;
383 
384 	mutex_lock(&dev->lock);
385 	dev->status = 0; /* Will be set in interrupt handler */
386 
387 	ctrlreg1 = readl(dev->io_base + SMI_CR1);
388 	/* program smi in h/w mode */
389 	writel(ctrlreg1 & ~SW_MODE, dev->io_base + SMI_CR1);
390 
391 	/* give the flash, write enable command */
392 	writel((bank << BANK_SHIFT) | WE | TFIE, dev->io_base + SMI_CR2);
393 
394 	ret = wait_event_interruptible_timeout(dev->cmd_complete,
395 			dev->status & TFF, SMI_CMD_TIMEOUT);
396 
397 	/* restore the ctrl regs state */
398 	writel(ctrlreg1, dev->io_base + SMI_CR1);
399 	writel(0, dev->io_base + SMI_CR2);
400 
401 	if (ret == 0) {
402 		ret = -EIO;
403 		dev_err(&dev->pdev->dev,
404 			"smi controller failed on write enable\n");
405 	} else if (ret > 0) {
406 		/* check whether write mode status is set for required bank */
407 		if (dev->status & (1 << (bank + WM_SHIFT)))
408 			ret = 0;
409 		else {
410 			dev_err(&dev->pdev->dev, "couldn't enable write\n");
411 			ret = -EIO;
412 		}
413 	}
414 
415 	mutex_unlock(&dev->lock);
416 	return ret;
417 }
418 
419 static inline u32
420 get_sector_erase_cmd(struct spear_snor_flash *flash, u32 offset)
421 {
422 	u32 cmd;
423 	u8 *x = (u8 *)&cmd;
424 
425 	x[0] = flash->erase_cmd;
426 	x[1] = offset >> 16;
427 	x[2] = offset >> 8;
428 	x[3] = offset;
429 
430 	return cmd;
431 }
432 
433 /**
434  * spear_smi_erase_sector - erase one sector of flash
435  * @dev: structure of SMI information
436  * @command: erase command to be send
437  * @bank: bank to which this command needs to be send
438  * @bytes: size of command
439  *
440  * Erase one sector of flash memory at offset ``offset'' which is any
441  * address within the sector which should be erased.
442  * Returns 0 if successful, non-zero otherwise.
443  */
444 static int spear_smi_erase_sector(struct spear_smi *dev,
445 		u32 bank, u32 command, u32 bytes)
446 {
447 	u32 ctrlreg1 = 0;
448 	int ret;
449 
450 	ret = spear_smi_wait_till_ready(dev, bank, SMI_MAX_TIME_OUT);
451 	if (ret)
452 		return ret;
453 
454 	ret = spear_smi_write_enable(dev, bank);
455 	if (ret)
456 		return ret;
457 
458 	mutex_lock(&dev->lock);
459 
460 	ctrlreg1 = readl(dev->io_base + SMI_CR1);
461 	writel((ctrlreg1 | SW_MODE) & ~WB_MODE, dev->io_base + SMI_CR1);
462 
463 	/* send command in sw mode */
464 	writel(command, dev->io_base + SMI_TR);
465 
466 	writel((bank << BANK_SHIFT) | SEND | TFIE | (bytes << TX_LEN_SHIFT),
467 			dev->io_base + SMI_CR2);
468 
469 	ret = wait_event_interruptible_timeout(dev->cmd_complete,
470 			dev->status & TFF, SMI_CMD_TIMEOUT);
471 
472 	if (ret == 0) {
473 		ret = -EIO;
474 		dev_err(&dev->pdev->dev, "sector erase failed\n");
475 	} else if (ret > 0)
476 		ret = 0; /* success */
477 
478 	/* restore ctrl regs */
479 	writel(ctrlreg1, dev->io_base + SMI_CR1);
480 	writel(0, dev->io_base + SMI_CR2);
481 
482 	mutex_unlock(&dev->lock);
483 	return ret;
484 }
485 
486 /**
487  * spear_mtd_erase - perform flash erase operation as requested by user
488  * @mtd: Provides the memory characteristics
489  * @e_info: Provides the erase information
490  *
491  * Erase an address range on the flash chip. The address range may extend
492  * one or more erase sectors. Return an error is there is a problem erasing.
493  */
494 static int spear_mtd_erase(struct mtd_info *mtd, struct erase_info *e_info)
495 {
496 	struct spear_snor_flash *flash = get_flash_data(mtd);
497 	struct spear_smi *dev = mtd->priv;
498 	u32 addr, command, bank;
499 	int len, ret;
500 
501 	if (!flash || !dev)
502 		return -ENODEV;
503 
504 	bank = flash->bank;
505 	if (bank > dev->num_flashes - 1) {
506 		dev_err(&dev->pdev->dev, "Invalid Bank Num");
507 		return -EINVAL;
508 	}
509 
510 	addr = e_info->addr;
511 	len = e_info->len;
512 
513 	mutex_lock(&flash->lock);
514 
515 	/* now erase sectors in loop */
516 	while (len) {
517 		command = get_sector_erase_cmd(flash, addr);
518 		/* preparing the command for flash */
519 		ret = spear_smi_erase_sector(dev, bank, command, 4);
520 		if (ret) {
521 			mutex_unlock(&flash->lock);
522 			return ret;
523 		}
524 		addr += mtd->erasesize;
525 		len -= mtd->erasesize;
526 	}
527 
528 	mutex_unlock(&flash->lock);
529 
530 	return 0;
531 }
532 
533 /**
534  * spear_mtd_read - performs flash read operation as requested by the user
535  * @mtd: MTD information of the memory bank
536  * @from: Address from which to start read
537  * @len: Number of bytes to be read
538  * @retlen: Fills the Number of bytes actually read
539  * @buf: Fills this after reading
540  *
541  * Read an address range from the flash chip. The address range
542  * may be any size provided it is within the physical boundaries.
543  * Returns 0 on success, non zero otherwise
544  */
545 static int spear_mtd_read(struct mtd_info *mtd, loff_t from, size_t len,
546 		size_t *retlen, u8 *buf)
547 {
548 	struct spear_snor_flash *flash = get_flash_data(mtd);
549 	struct spear_smi *dev = mtd->priv;
550 	void __iomem *src;
551 	u32 ctrlreg1, val;
552 	int ret;
553 
554 	if (!flash || !dev)
555 		return -ENODEV;
556 
557 	if (flash->bank > dev->num_flashes - 1) {
558 		dev_err(&dev->pdev->dev, "Invalid Bank Num");
559 		return -EINVAL;
560 	}
561 
562 	/* select address as per bank number */
563 	src = flash->base_addr + from;
564 
565 	mutex_lock(&flash->lock);
566 
567 	/* wait till previous write/erase is done. */
568 	ret = spear_smi_wait_till_ready(dev, flash->bank, SMI_MAX_TIME_OUT);
569 	if (ret) {
570 		mutex_unlock(&flash->lock);
571 		return ret;
572 	}
573 
574 	mutex_lock(&dev->lock);
575 	/* put smi in hw mode not wbt mode */
576 	ctrlreg1 = val = readl(dev->io_base + SMI_CR1);
577 	val &= ~(SW_MODE | WB_MODE);
578 	if (flash->fast_mode)
579 		val |= FAST_MODE;
580 
581 	writel(val, dev->io_base + SMI_CR1);
582 
583 	memcpy_fromio(buf, src, len);
584 
585 	/* restore ctrl reg1 */
586 	writel(ctrlreg1, dev->io_base + SMI_CR1);
587 	mutex_unlock(&dev->lock);
588 
589 	*retlen = len;
590 	mutex_unlock(&flash->lock);
591 
592 	return 0;
593 }
594 
595 static inline int spear_smi_cpy_toio(struct spear_smi *dev, u32 bank,
596 		void __iomem *dest, const void *src, size_t len)
597 {
598 	int ret;
599 	u32 ctrlreg1;
600 
601 	/* wait until finished previous write command. */
602 	ret = spear_smi_wait_till_ready(dev, bank, SMI_MAX_TIME_OUT);
603 	if (ret)
604 		return ret;
605 
606 	/* put smi in write enable */
607 	ret = spear_smi_write_enable(dev, bank);
608 	if (ret)
609 		return ret;
610 
611 	/* put smi in hw, write burst mode */
612 	mutex_lock(&dev->lock);
613 
614 	ctrlreg1 = readl(dev->io_base + SMI_CR1);
615 	writel((ctrlreg1 | WB_MODE) & ~SW_MODE, dev->io_base + SMI_CR1);
616 
617 	memcpy_toio(dest, src, len);
618 
619 	writel(ctrlreg1, dev->io_base + SMI_CR1);
620 
621 	mutex_unlock(&dev->lock);
622 	return 0;
623 }
624 
625 /**
626  * spear_mtd_write - performs write operation as requested by the user.
627  * @mtd: MTD information of the memory bank.
628  * @to:	Address to write.
629  * @len: Number of bytes to be written.
630  * @retlen: Number of bytes actually wrote.
631  * @buf: Buffer from which the data to be taken.
632  *
633  * Write an address range to the flash chip. Data must be written in
634  * flash_page_size chunks. The address range may be any size provided
635  * it is within the physical boundaries.
636  * Returns 0 on success, non zero otherwise
637  */
638 static int spear_mtd_write(struct mtd_info *mtd, loff_t to, size_t len,
639 		size_t *retlen, const u8 *buf)
640 {
641 	struct spear_snor_flash *flash = get_flash_data(mtd);
642 	struct spear_smi *dev = mtd->priv;
643 	void __iomem *dest;
644 	u32 page_offset, page_size;
645 	int ret;
646 
647 	if (!flash || !dev)
648 		return -ENODEV;
649 
650 	if (flash->bank > dev->num_flashes - 1) {
651 		dev_err(&dev->pdev->dev, "Invalid Bank Num");
652 		return -EINVAL;
653 	}
654 
655 	/* select address as per bank number */
656 	dest = flash->base_addr + to;
657 	mutex_lock(&flash->lock);
658 
659 	page_offset = (u32)to % flash->page_size;
660 
661 	/* do if all the bytes fit onto one page */
662 	if (page_offset + len <= flash->page_size) {
663 		ret = spear_smi_cpy_toio(dev, flash->bank, dest, buf, len);
664 		if (!ret)
665 			*retlen += len;
666 	} else {
667 		u32 i;
668 
669 		/* the size of data remaining on the first page */
670 		page_size = flash->page_size - page_offset;
671 
672 		ret = spear_smi_cpy_toio(dev, flash->bank, dest, buf,
673 				page_size);
674 		if (ret)
675 			goto err_write;
676 		else
677 			*retlen += page_size;
678 
679 		/* write everything in pagesize chunks */
680 		for (i = page_size; i < len; i += page_size) {
681 			page_size = len - i;
682 			if (page_size > flash->page_size)
683 				page_size = flash->page_size;
684 
685 			ret = spear_smi_cpy_toio(dev, flash->bank, dest + i,
686 					buf + i, page_size);
687 			if (ret)
688 				break;
689 			else
690 				*retlen += page_size;
691 		}
692 	}
693 
694 err_write:
695 	mutex_unlock(&flash->lock);
696 
697 	return ret;
698 }
699 
700 /**
701  * spear_smi_probe_flash - Detects the NOR Flash chip.
702  * @dev: structure of SMI information.
703  * @bank: bank on which flash must be probed
704  *
705  * This routine will check whether there exists a flash chip on a given memory
706  * bank ID.
707  * Return index of the probed flash in flash devices structure
708  */
709 static int spear_smi_probe_flash(struct spear_smi *dev, u32 bank)
710 {
711 	int ret;
712 	u32 val = 0;
713 
714 	ret = spear_smi_wait_till_ready(dev, bank, SMI_PROBE_TIMEOUT);
715 	if (ret)
716 		return ret;
717 
718 	mutex_lock(&dev->lock);
719 
720 	dev->status = 0; /* Will be set in interrupt handler */
721 	/* put smi in sw mode */
722 	val = readl(dev->io_base + SMI_CR1);
723 	writel(val | SW_MODE, dev->io_base + SMI_CR1);
724 
725 	/* send readid command in sw mode */
726 	writel(OPCODE_RDID, dev->io_base + SMI_TR);
727 
728 	val = (bank << BANK_SHIFT) | SEND | (1 << TX_LEN_SHIFT) |
729 		(3 << RX_LEN_SHIFT) | TFIE;
730 	writel(val, dev->io_base + SMI_CR2);
731 
732 	/* wait for TFF */
733 	ret = wait_event_interruptible_timeout(dev->cmd_complete,
734 			dev->status & TFF, SMI_CMD_TIMEOUT);
735 	if (ret <= 0) {
736 		ret = -ENODEV;
737 		goto err_probe;
738 	}
739 
740 	/* get memory chip id */
741 	val = readl(dev->io_base + SMI_RR);
742 	val &= 0x00ffffff;
743 	ret = get_flash_index(val);
744 
745 err_probe:
746 	/* clear sw mode */
747 	val = readl(dev->io_base + SMI_CR1);
748 	writel(val & ~SW_MODE, dev->io_base + SMI_CR1);
749 
750 	mutex_unlock(&dev->lock);
751 	return ret;
752 }
753 
754 
755 #ifdef CONFIG_OF
756 static int spear_smi_probe_config_dt(struct platform_device *pdev,
757 				     struct device_node *np)
758 {
759 	struct spear_smi_plat_data *pdata = dev_get_platdata(&pdev->dev);
760 	struct device_node *pp = NULL;
761 	const __be32 *addr;
762 	u32 val;
763 	int len;
764 	int i = 0;
765 
766 	if (!np)
767 		return -ENODEV;
768 
769 	of_property_read_u32(np, "clock-rate", &val);
770 	pdata->clk_rate = val;
771 
772 	pdata->board_flash_info = devm_kzalloc(&pdev->dev,
773 					       sizeof(*pdata->board_flash_info),
774 					       GFP_KERNEL);
775 	if (!pdata->board_flash_info)
776 		return -ENOMEM;
777 
778 	/* Fill structs for each subnode (flash device) */
779 	while ((pp = of_get_next_child(np, pp))) {
780 		struct spear_smi_flash_info *flash_info;
781 
782 		flash_info = &pdata->board_flash_info[i];
783 		pdata->np[i] = pp;
784 
785 		/* Read base-addr and size from DT */
786 		addr = of_get_property(pp, "reg", &len);
787 		pdata->board_flash_info->mem_base = be32_to_cpup(&addr[0]);
788 		pdata->board_flash_info->size = be32_to_cpup(&addr[1]);
789 
790 		if (of_get_property(pp, "st,smi-fast-mode", NULL))
791 			pdata->board_flash_info->fast_mode = 1;
792 
793 		i++;
794 	}
795 
796 	pdata->num_flashes = i;
797 
798 	return 0;
799 }
800 #else
801 static int spear_smi_probe_config_dt(struct platform_device *pdev,
802 				     struct device_node *np)
803 {
804 	return -ENOSYS;
805 }
806 #endif
807 
808 static int spear_smi_setup_banks(struct platform_device *pdev,
809 				 u32 bank, struct device_node *np)
810 {
811 	struct spear_smi *dev = platform_get_drvdata(pdev);
812 	struct spear_smi_flash_info *flash_info;
813 	struct spear_smi_plat_data *pdata;
814 	struct spear_snor_flash *flash;
815 	struct mtd_partition *parts = NULL;
816 	int count = 0;
817 	int flash_index;
818 	int ret = 0;
819 
820 	pdata = dev_get_platdata(&pdev->dev);
821 	if (bank > pdata->num_flashes - 1)
822 		return -EINVAL;
823 
824 	flash_info = &pdata->board_flash_info[bank];
825 	if (!flash_info)
826 		return -ENODEV;
827 
828 	flash = devm_kzalloc(&pdev->dev, sizeof(*flash), GFP_ATOMIC);
829 	if (!flash)
830 		return -ENOMEM;
831 	flash->bank = bank;
832 	flash->fast_mode = flash_info->fast_mode ? 1 : 0;
833 	mutex_init(&flash->lock);
834 
835 	/* verify whether nor flash is really present on board */
836 	flash_index = spear_smi_probe_flash(dev, bank);
837 	if (flash_index < 0) {
838 		dev_info(&dev->pdev->dev, "smi-nor%d not found\n", bank);
839 		return flash_index;
840 	}
841 	/* map the memory for nor flash chip */
842 	flash->base_addr = devm_ioremap(&pdev->dev, flash_info->mem_base,
843 					flash_info->size);
844 	if (!flash->base_addr)
845 		return -EIO;
846 
847 	dev->flash[bank] = flash;
848 	flash->mtd.priv = dev;
849 
850 	if (flash_info->name)
851 		flash->mtd.name = flash_info->name;
852 	else
853 		flash->mtd.name = flash_devices[flash_index].name;
854 
855 	flash->mtd.dev.parent = &pdev->dev;
856 	mtd_set_of_node(&flash->mtd, np);
857 	flash->mtd.type = MTD_NORFLASH;
858 	flash->mtd.writesize = 1;
859 	flash->mtd.flags = MTD_CAP_NORFLASH;
860 	flash->mtd.size = flash_info->size;
861 	flash->mtd.erasesize = flash_devices[flash_index].sectorsize;
862 	flash->page_size = flash_devices[flash_index].pagesize;
863 	flash->mtd.writebufsize = flash->page_size;
864 	flash->erase_cmd = flash_devices[flash_index].erase_cmd;
865 	flash->mtd._erase = spear_mtd_erase;
866 	flash->mtd._read = spear_mtd_read;
867 	flash->mtd._write = spear_mtd_write;
868 	flash->dev_id = flash_devices[flash_index].device_id;
869 
870 	dev_info(&dev->pdev->dev, "mtd .name=%s .size=%llx(%lluM)\n",
871 			flash->mtd.name, flash->mtd.size,
872 			flash->mtd.size / (1024 * 1024));
873 
874 	dev_info(&dev->pdev->dev, ".erasesize = 0x%x(%uK)\n",
875 			flash->mtd.erasesize, flash->mtd.erasesize / 1024);
876 
877 #ifndef CONFIG_OF
878 	if (flash_info->partitions) {
879 		parts = flash_info->partitions;
880 		count = flash_info->nr_partitions;
881 	}
882 #endif
883 
884 	ret = mtd_device_register(&flash->mtd, parts, count);
885 	if (ret) {
886 		dev_err(&dev->pdev->dev, "Err MTD partition=%d\n", ret);
887 		return ret;
888 	}
889 
890 	return 0;
891 }
892 
893 /**
894  * spear_smi_probe - Entry routine
895  * @pdev: platform device structure
896  *
897  * This is the first routine which gets invoked during booting and does all
898  * initialization/allocation work. The routine looks for available memory banks,
899  * and do proper init for any found one.
900  * Returns 0 on success, non zero otherwise
901  */
902 static int spear_smi_probe(struct platform_device *pdev)
903 {
904 	struct device_node *np = pdev->dev.of_node;
905 	struct spear_smi_plat_data *pdata = NULL;
906 	struct spear_smi *dev;
907 	struct resource *smi_base;
908 	int irq, ret = 0;
909 	int i;
910 
911 	if (np) {
912 		pdata = devm_kzalloc(&pdev->dev, sizeof(*pdata), GFP_KERNEL);
913 		if (!pdata) {
914 			ret = -ENOMEM;
915 			goto err;
916 		}
917 		pdev->dev.platform_data = pdata;
918 		ret = spear_smi_probe_config_dt(pdev, np);
919 		if (ret) {
920 			ret = -ENODEV;
921 			dev_err(&pdev->dev, "no platform data\n");
922 			goto err;
923 		}
924 	} else {
925 		pdata = dev_get_platdata(&pdev->dev);
926 		if (!pdata) {
927 			ret = -ENODEV;
928 			dev_err(&pdev->dev, "no platform data\n");
929 			goto err;
930 		}
931 	}
932 
933 	irq = platform_get_irq(pdev, 0);
934 	if (irq < 0) {
935 		ret = -ENODEV;
936 		dev_err(&pdev->dev, "invalid smi irq\n");
937 		goto err;
938 	}
939 
940 	dev = devm_kzalloc(&pdev->dev, sizeof(*dev), GFP_ATOMIC);
941 	if (!dev) {
942 		ret = -ENOMEM;
943 		goto err;
944 	}
945 
946 	smi_base = platform_get_resource(pdev, IORESOURCE_MEM, 0);
947 
948 	dev->io_base = devm_ioremap_resource(&pdev->dev, smi_base);
949 	if (IS_ERR(dev->io_base)) {
950 		ret = PTR_ERR(dev->io_base);
951 		goto err;
952 	}
953 
954 	dev->pdev = pdev;
955 	dev->clk_rate = pdata->clk_rate;
956 
957 	if (dev->clk_rate > SMI_MAX_CLOCK_FREQ)
958 		dev->clk_rate = SMI_MAX_CLOCK_FREQ;
959 
960 	dev->num_flashes = pdata->num_flashes;
961 
962 	if (dev->num_flashes > MAX_NUM_FLASH_CHIP) {
963 		dev_err(&pdev->dev, "exceeding max number of flashes\n");
964 		dev->num_flashes = MAX_NUM_FLASH_CHIP;
965 	}
966 
967 	dev->clk = devm_clk_get(&pdev->dev, NULL);
968 	if (IS_ERR(dev->clk)) {
969 		ret = PTR_ERR(dev->clk);
970 		goto err;
971 	}
972 
973 	ret = clk_prepare_enable(dev->clk);
974 	if (ret)
975 		goto err;
976 
977 	ret = devm_request_irq(&pdev->dev, irq, spear_smi_int_handler, 0,
978 			       pdev->name, dev);
979 	if (ret) {
980 		dev_err(&dev->pdev->dev, "SMI IRQ allocation failed\n");
981 		goto err_irq;
982 	}
983 
984 	mutex_init(&dev->lock);
985 	init_waitqueue_head(&dev->cmd_complete);
986 	spear_smi_hw_init(dev);
987 	platform_set_drvdata(pdev, dev);
988 
989 	/* loop for each serial nor-flash which is connected to smi */
990 	for (i = 0; i < dev->num_flashes; i++) {
991 		ret = spear_smi_setup_banks(pdev, i, pdata->np[i]);
992 		if (ret) {
993 			dev_err(&dev->pdev->dev, "bank setup failed\n");
994 			goto err_irq;
995 		}
996 	}
997 
998 	return 0;
999 
1000 err_irq:
1001 	clk_disable_unprepare(dev->clk);
1002 err:
1003 	return ret;
1004 }
1005 
1006 /**
1007  * spear_smi_remove - Exit routine
1008  * @pdev: platform device structure
1009  *
1010  * free all allocations and delete the partitions.
1011  */
1012 static int spear_smi_remove(struct platform_device *pdev)
1013 {
1014 	struct spear_smi *dev;
1015 	struct spear_snor_flash *flash;
1016 	int ret, i;
1017 
1018 	dev = platform_get_drvdata(pdev);
1019 	if (!dev) {
1020 		dev_err(&pdev->dev, "dev is null\n");
1021 		return -ENODEV;
1022 	}
1023 
1024 	/* clean up for all nor flash */
1025 	for (i = 0; i < dev->num_flashes; i++) {
1026 		flash = dev->flash[i];
1027 		if (!flash)
1028 			continue;
1029 
1030 		/* clean up mtd stuff */
1031 		ret = mtd_device_unregister(&flash->mtd);
1032 		if (ret)
1033 			dev_err(&pdev->dev, "error removing mtd\n");
1034 	}
1035 
1036 	clk_disable_unprepare(dev->clk);
1037 
1038 	return 0;
1039 }
1040 
1041 #ifdef CONFIG_PM_SLEEP
1042 static int spear_smi_suspend(struct device *dev)
1043 {
1044 	struct spear_smi *sdev = dev_get_drvdata(dev);
1045 
1046 	if (sdev && sdev->clk)
1047 		clk_disable_unprepare(sdev->clk);
1048 
1049 	return 0;
1050 }
1051 
1052 static int spear_smi_resume(struct device *dev)
1053 {
1054 	struct spear_smi *sdev = dev_get_drvdata(dev);
1055 	int ret = -EPERM;
1056 
1057 	if (sdev && sdev->clk)
1058 		ret = clk_prepare_enable(sdev->clk);
1059 
1060 	if (!ret)
1061 		spear_smi_hw_init(sdev);
1062 	return ret;
1063 }
1064 #endif
1065 
1066 static SIMPLE_DEV_PM_OPS(spear_smi_pm_ops, spear_smi_suspend, spear_smi_resume);
1067 
1068 #ifdef CONFIG_OF
1069 static const struct of_device_id spear_smi_id_table[] = {
1070 	{ .compatible = "st,spear600-smi" },
1071 	{}
1072 };
1073 MODULE_DEVICE_TABLE(of, spear_smi_id_table);
1074 #endif
1075 
1076 static struct platform_driver spear_smi_driver = {
1077 	.driver = {
1078 		.name = "smi",
1079 		.bus = &platform_bus_type,
1080 		.of_match_table = of_match_ptr(spear_smi_id_table),
1081 		.pm = &spear_smi_pm_ops,
1082 	},
1083 	.probe = spear_smi_probe,
1084 	.remove = spear_smi_remove,
1085 };
1086 module_platform_driver(spear_smi_driver);
1087 
1088 MODULE_LICENSE("GPL");
1089 MODULE_AUTHOR("Ashish Priyadarshi, Shiraz Hashim <shiraz.linux.kernel@gmail.com>");
1090 MODULE_DESCRIPTION("MTD SMI driver for serial nor flash chips");
1091