xref: /openbmc/linux/drivers/spi/atmel-quadspi.c (revision 24f7989e)
1 // SPDX-License-Identifier: GPL-2.0
2 /*
3  * Driver for Atmel QSPI Controller
4  *
5  * Copyright (C) 2015 Atmel Corporation
6  * Copyright (C) 2018 Cryptera A/S
7  *
8  * Author: Cyrille Pitchen <cyrille.pitchen@atmel.com>
9  * Author: Piotr Bugalski <bugalski.piotr@gmail.com>
10  *
11  * This driver is based on drivers/mtd/spi-nor/fsl-quadspi.c from Freescale.
12  */
13 
14 #include <linux/clk.h>
15 #include <linux/delay.h>
16 #include <linux/err.h>
17 #include <linux/interrupt.h>
18 #include <linux/io.h>
19 #include <linux/kernel.h>
20 #include <linux/module.h>
21 #include <linux/of.h>
22 #include <linux/of_platform.h>
23 #include <linux/platform_device.h>
24 #include <linux/pm_runtime.h>
25 #include <linux/spi/spi-mem.h>
26 
27 /* QSPI register offsets */
28 #define QSPI_CR      0x0000  /* Control Register */
29 #define QSPI_MR      0x0004  /* Mode Register */
30 #define QSPI_RD      0x0008  /* Receive Data Register */
31 #define QSPI_TD      0x000c  /* Transmit Data Register */
32 #define QSPI_SR      0x0010  /* Status Register */
33 #define QSPI_IER     0x0014  /* Interrupt Enable Register */
34 #define QSPI_IDR     0x0018  /* Interrupt Disable Register */
35 #define QSPI_IMR     0x001c  /* Interrupt Mask Register */
36 #define QSPI_SCR     0x0020  /* Serial Clock Register */
37 
38 #define QSPI_IAR     0x0030  /* Instruction Address Register */
39 #define QSPI_ICR     0x0034  /* Instruction Code Register */
40 #define QSPI_WICR    0x0034  /* Write Instruction Code Register */
41 #define QSPI_IFR     0x0038  /* Instruction Frame Register */
42 #define QSPI_RICR    0x003C  /* Read Instruction Code Register */
43 
44 #define QSPI_SMR     0x0040  /* Scrambling Mode Register */
45 #define QSPI_SKR     0x0044  /* Scrambling Key Register */
46 
47 #define QSPI_WPMR    0x00E4  /* Write Protection Mode Register */
48 #define QSPI_WPSR    0x00E8  /* Write Protection Status Register */
49 
50 #define QSPI_VERSION 0x00FC  /* Version Register */
51 
52 
53 /* Bitfields in QSPI_CR (Control Register) */
54 #define QSPI_CR_QSPIEN                  BIT(0)
55 #define QSPI_CR_QSPIDIS                 BIT(1)
56 #define QSPI_CR_SWRST                   BIT(7)
57 #define QSPI_CR_LASTXFER                BIT(24)
58 
59 /* Bitfields in QSPI_MR (Mode Register) */
60 #define QSPI_MR_SMM                     BIT(0)
61 #define QSPI_MR_LLB                     BIT(1)
62 #define QSPI_MR_WDRBT                   BIT(2)
63 #define QSPI_MR_SMRM                    BIT(3)
64 #define QSPI_MR_CSMODE_MASK             GENMASK(5, 4)
65 #define QSPI_MR_CSMODE_NOT_RELOADED     (0 << 4)
66 #define QSPI_MR_CSMODE_LASTXFER         (1 << 4)
67 #define QSPI_MR_CSMODE_SYSTEMATICALLY   (2 << 4)
68 #define QSPI_MR_NBBITS_MASK             GENMASK(11, 8)
69 #define QSPI_MR_NBBITS(n)               ((((n) - 8) << 8) & QSPI_MR_NBBITS_MASK)
70 #define QSPI_MR_DLYBCT_MASK             GENMASK(23, 16)
71 #define QSPI_MR_DLYBCT(n)               (((n) << 16) & QSPI_MR_DLYBCT_MASK)
72 #define QSPI_MR_DLYCS_MASK              GENMASK(31, 24)
73 #define QSPI_MR_DLYCS(n)                (((n) << 24) & QSPI_MR_DLYCS_MASK)
74 
75 /* Bitfields in QSPI_SR/QSPI_IER/QSPI_IDR/QSPI_IMR  */
76 #define QSPI_SR_RDRF                    BIT(0)
77 #define QSPI_SR_TDRE                    BIT(1)
78 #define QSPI_SR_TXEMPTY                 BIT(2)
79 #define QSPI_SR_OVRES                   BIT(3)
80 #define QSPI_SR_CSR                     BIT(8)
81 #define QSPI_SR_CSS                     BIT(9)
82 #define QSPI_SR_INSTRE                  BIT(10)
83 #define QSPI_SR_QSPIENS                 BIT(24)
84 
85 #define QSPI_SR_CMD_COMPLETED	(QSPI_SR_INSTRE | QSPI_SR_CSR)
86 
87 /* Bitfields in QSPI_SCR (Serial Clock Register) */
88 #define QSPI_SCR_CPOL                   BIT(0)
89 #define QSPI_SCR_CPHA                   BIT(1)
90 #define QSPI_SCR_SCBR_MASK              GENMASK(15, 8)
91 #define QSPI_SCR_SCBR(n)                (((n) << 8) & QSPI_SCR_SCBR_MASK)
92 #define QSPI_SCR_DLYBS_MASK             GENMASK(23, 16)
93 #define QSPI_SCR_DLYBS(n)               (((n) << 16) & QSPI_SCR_DLYBS_MASK)
94 
95 /* Bitfields in QSPI_ICR (Read/Write Instruction Code Register) */
96 #define QSPI_ICR_INST_MASK              GENMASK(7, 0)
97 #define QSPI_ICR_INST(inst)             (((inst) << 0) & QSPI_ICR_INST_MASK)
98 #define QSPI_ICR_OPT_MASK               GENMASK(23, 16)
99 #define QSPI_ICR_OPT(opt)               (((opt) << 16) & QSPI_ICR_OPT_MASK)
100 
101 /* Bitfields in QSPI_IFR (Instruction Frame Register) */
102 #define QSPI_IFR_WIDTH_MASK             GENMASK(2, 0)
103 #define QSPI_IFR_WIDTH_SINGLE_BIT_SPI   (0 << 0)
104 #define QSPI_IFR_WIDTH_DUAL_OUTPUT      (1 << 0)
105 #define QSPI_IFR_WIDTH_QUAD_OUTPUT      (2 << 0)
106 #define QSPI_IFR_WIDTH_DUAL_IO          (3 << 0)
107 #define QSPI_IFR_WIDTH_QUAD_IO          (4 << 0)
108 #define QSPI_IFR_WIDTH_DUAL_CMD         (5 << 0)
109 #define QSPI_IFR_WIDTH_QUAD_CMD         (6 << 0)
110 #define QSPI_IFR_INSTEN                 BIT(4)
111 #define QSPI_IFR_ADDREN                 BIT(5)
112 #define QSPI_IFR_OPTEN                  BIT(6)
113 #define QSPI_IFR_DATAEN                 BIT(7)
114 #define QSPI_IFR_OPTL_MASK              GENMASK(9, 8)
115 #define QSPI_IFR_OPTL_1BIT              (0 << 8)
116 #define QSPI_IFR_OPTL_2BIT              (1 << 8)
117 #define QSPI_IFR_OPTL_4BIT              (2 << 8)
118 #define QSPI_IFR_OPTL_8BIT              (3 << 8)
119 #define QSPI_IFR_ADDRL                  BIT(10)
120 #define QSPI_IFR_TFRTYP_MEM		BIT(12)
121 #define QSPI_IFR_SAMA5D2_WRITE_TRSFR	BIT(13)
122 #define QSPI_IFR_CRM                    BIT(14)
123 #define QSPI_IFR_NBDUM_MASK             GENMASK(20, 16)
124 #define QSPI_IFR_NBDUM(n)               (((n) << 16) & QSPI_IFR_NBDUM_MASK)
125 #define QSPI_IFR_APBTFRTYP_READ		BIT(24)	/* Defined in SAM9X60 */
126 
127 /* Bitfields in QSPI_SMR (Scrambling Mode Register) */
128 #define QSPI_SMR_SCREN                  BIT(0)
129 #define QSPI_SMR_RVDIS                  BIT(1)
130 
131 /* Bitfields in QSPI_WPMR (Write Protection Mode Register) */
132 #define QSPI_WPMR_WPEN                  BIT(0)
133 #define QSPI_WPMR_WPKEY_MASK            GENMASK(31, 8)
134 #define QSPI_WPMR_WPKEY(wpkey)          (((wpkey) << 8) & QSPI_WPMR_WPKEY_MASK)
135 
136 /* Bitfields in QSPI_WPSR (Write Protection Status Register) */
137 #define QSPI_WPSR_WPVS                  BIT(0)
138 #define QSPI_WPSR_WPVSRC_MASK           GENMASK(15, 8)
139 #define QSPI_WPSR_WPVSRC(src)           (((src) << 8) & QSPI_WPSR_WPVSRC)
140 
141 struct atmel_qspi_caps {
142 	bool has_qspick;
143 	bool has_ricr;
144 };
145 
146 struct atmel_qspi {
147 	void __iomem		*regs;
148 	void __iomem		*mem;
149 	struct clk		*pclk;
150 	struct clk		*qspick;
151 	struct platform_device	*pdev;
152 	const struct atmel_qspi_caps *caps;
153 	resource_size_t		mmap_size;
154 	u32			pending;
155 	u32			mr;
156 	u32			scr;
157 	struct completion	cmd_completion;
158 };
159 
160 struct atmel_qspi_mode {
161 	u8 cmd_buswidth;
162 	u8 addr_buswidth;
163 	u8 data_buswidth;
164 	u32 config;
165 };
166 
167 static const struct atmel_qspi_mode atmel_qspi_modes[] = {
168 	{ 1, 1, 1, QSPI_IFR_WIDTH_SINGLE_BIT_SPI },
169 	{ 1, 1, 2, QSPI_IFR_WIDTH_DUAL_OUTPUT },
170 	{ 1, 1, 4, QSPI_IFR_WIDTH_QUAD_OUTPUT },
171 	{ 1, 2, 2, QSPI_IFR_WIDTH_DUAL_IO },
172 	{ 1, 4, 4, QSPI_IFR_WIDTH_QUAD_IO },
173 	{ 2, 2, 2, QSPI_IFR_WIDTH_DUAL_CMD },
174 	{ 4, 4, 4, QSPI_IFR_WIDTH_QUAD_CMD },
175 };
176 
177 #ifdef VERBOSE_DEBUG
178 static const char *atmel_qspi_reg_name(u32 offset, char *tmp, size_t sz)
179 {
180 	switch (offset) {
181 	case QSPI_CR:
182 		return "CR";
183 	case QSPI_MR:
184 		return "MR";
185 	case QSPI_RD:
186 		return "MR";
187 	case QSPI_TD:
188 		return "TD";
189 	case QSPI_SR:
190 		return "SR";
191 	case QSPI_IER:
192 		return "IER";
193 	case QSPI_IDR:
194 		return "IDR";
195 	case QSPI_IMR:
196 		return "IMR";
197 	case QSPI_SCR:
198 		return "SCR";
199 	case QSPI_IAR:
200 		return "IAR";
201 	case QSPI_ICR:
202 		return "ICR/WICR";
203 	case QSPI_IFR:
204 		return "IFR";
205 	case QSPI_RICR:
206 		return "RICR";
207 	case QSPI_SMR:
208 		return "SMR";
209 	case QSPI_SKR:
210 		return "SKR";
211 	case QSPI_WPMR:
212 		return "WPMR";
213 	case QSPI_WPSR:
214 		return "WPSR";
215 	case QSPI_VERSION:
216 		return "VERSION";
217 	default:
218 		snprintf(tmp, sz, "0x%02x", offset);
219 		break;
220 	}
221 
222 	return tmp;
223 }
224 #endif /* VERBOSE_DEBUG */
225 
226 static u32 atmel_qspi_read(struct atmel_qspi *aq, u32 offset)
227 {
228 	u32 value = readl_relaxed(aq->regs + offset);
229 
230 #ifdef VERBOSE_DEBUG
231 	char tmp[8];
232 
233 	dev_vdbg(&aq->pdev->dev, "read 0x%08x from %s\n", value,
234 		 atmel_qspi_reg_name(offset, tmp, sizeof(tmp)));
235 #endif /* VERBOSE_DEBUG */
236 
237 	return value;
238 }
239 
240 static void atmel_qspi_write(u32 value, struct atmel_qspi *aq, u32 offset)
241 {
242 #ifdef VERBOSE_DEBUG
243 	char tmp[8];
244 
245 	dev_vdbg(&aq->pdev->dev, "write 0x%08x into %s\n", value,
246 		 atmel_qspi_reg_name(offset, tmp, sizeof(tmp)));
247 #endif /* VERBOSE_DEBUG */
248 
249 	writel_relaxed(value, aq->regs + offset);
250 }
251 
252 static inline bool atmel_qspi_is_compatible(const struct spi_mem_op *op,
253 					    const struct atmel_qspi_mode *mode)
254 {
255 	if (op->cmd.buswidth != mode->cmd_buswidth)
256 		return false;
257 
258 	if (op->addr.nbytes && op->addr.buswidth != mode->addr_buswidth)
259 		return false;
260 
261 	if (op->data.nbytes && op->data.buswidth != mode->data_buswidth)
262 		return false;
263 
264 	return true;
265 }
266 
267 static int atmel_qspi_find_mode(const struct spi_mem_op *op)
268 {
269 	u32 i;
270 
271 	for (i = 0; i < ARRAY_SIZE(atmel_qspi_modes); i++)
272 		if (atmel_qspi_is_compatible(op, &atmel_qspi_modes[i]))
273 			return i;
274 
275 	return -ENOTSUPP;
276 }
277 
278 static bool atmel_qspi_supports_op(struct spi_mem *mem,
279 				   const struct spi_mem_op *op)
280 {
281 	if (!spi_mem_default_supports_op(mem, op))
282 		return false;
283 
284 	if (atmel_qspi_find_mode(op) < 0)
285 		return false;
286 
287 	/* special case not supported by hardware */
288 	if (op->addr.nbytes == 2 && op->cmd.buswidth != op->addr.buswidth &&
289 	    op->dummy.nbytes == 0)
290 		return false;
291 
292 	return true;
293 }
294 
295 static int atmel_qspi_set_cfg(struct atmel_qspi *aq,
296 			      const struct spi_mem_op *op, u32 *offset)
297 {
298 	u32 iar, icr, ifr;
299 	u32 dummy_cycles = 0;
300 	int mode;
301 
302 	iar = 0;
303 	icr = QSPI_ICR_INST(op->cmd.opcode);
304 	ifr = QSPI_IFR_INSTEN;
305 
306 	mode = atmel_qspi_find_mode(op);
307 	if (mode < 0)
308 		return mode;
309 	ifr |= atmel_qspi_modes[mode].config;
310 
311 	if (op->dummy.nbytes)
312 		dummy_cycles = op->dummy.nbytes * 8 / op->dummy.buswidth;
313 
314 	/*
315 	 * The controller allows 24 and 32-bit addressing while NAND-flash
316 	 * requires 16-bit long. Handling 8-bit long addresses is done using
317 	 * the option field. For the 16-bit addresses, the workaround depends
318 	 * of the number of requested dummy bits. If there are 8 or more dummy
319 	 * cycles, the address is shifted and sent with the first dummy byte.
320 	 * Otherwise opcode is disabled and the first byte of the address
321 	 * contains the command opcode (works only if the opcode and address
322 	 * use the same buswidth). The limitation is when the 16-bit address is
323 	 * used without enough dummy cycles and the opcode is using a different
324 	 * buswidth than the address.
325 	 */
326 	if (op->addr.buswidth) {
327 		switch (op->addr.nbytes) {
328 		case 0:
329 			break;
330 		case 1:
331 			ifr |= QSPI_IFR_OPTEN | QSPI_IFR_OPTL_8BIT;
332 			icr |= QSPI_ICR_OPT(op->addr.val & 0xff);
333 			break;
334 		case 2:
335 			if (dummy_cycles < 8 / op->addr.buswidth) {
336 				ifr &= ~QSPI_IFR_INSTEN;
337 				ifr |= QSPI_IFR_ADDREN;
338 				iar = (op->cmd.opcode << 16) |
339 					(op->addr.val & 0xffff);
340 			} else {
341 				ifr |= QSPI_IFR_ADDREN;
342 				iar = (op->addr.val << 8) & 0xffffff;
343 				dummy_cycles -= 8 / op->addr.buswidth;
344 			}
345 			break;
346 		case 3:
347 			ifr |= QSPI_IFR_ADDREN;
348 			iar = op->addr.val & 0xffffff;
349 			break;
350 		case 4:
351 			ifr |= QSPI_IFR_ADDREN | QSPI_IFR_ADDRL;
352 			iar = op->addr.val & 0x7ffffff;
353 			break;
354 		default:
355 			return -ENOTSUPP;
356 		}
357 	}
358 
359 	/* offset of the data access in the QSPI memory space */
360 	*offset = iar;
361 
362 	/* Set number of dummy cycles */
363 	if (dummy_cycles)
364 		ifr |= QSPI_IFR_NBDUM(dummy_cycles);
365 
366 	/* Set data enable and data transfer type. */
367 	if (op->data.nbytes) {
368 		ifr |= QSPI_IFR_DATAEN;
369 
370 		if (op->addr.nbytes)
371 			ifr |= QSPI_IFR_TFRTYP_MEM;
372 	}
373 
374 	/*
375 	 * If the QSPI controller is set in regular SPI mode, set it in
376 	 * Serial Memory Mode (SMM).
377 	 */
378 	if (!(aq->mr & QSPI_MR_SMM)) {
379 		aq->mr |= QSPI_MR_SMM;
380 		atmel_qspi_write(aq->mr, aq, QSPI_MR);
381 	}
382 
383 	/* Clear pending interrupts */
384 	(void)atmel_qspi_read(aq, QSPI_SR);
385 
386 	/* Set QSPI Instruction Frame registers. */
387 	if (op->addr.nbytes && !op->data.nbytes)
388 		atmel_qspi_write(iar, aq, QSPI_IAR);
389 
390 	if (aq->caps->has_ricr) {
391 		if (op->data.dir == SPI_MEM_DATA_IN)
392 			atmel_qspi_write(icr, aq, QSPI_RICR);
393 		else
394 			atmel_qspi_write(icr, aq, QSPI_WICR);
395 	} else {
396 		if (op->data.nbytes && op->data.dir == SPI_MEM_DATA_OUT)
397 			ifr |= QSPI_IFR_SAMA5D2_WRITE_TRSFR;
398 
399 		atmel_qspi_write(icr, aq, QSPI_ICR);
400 	}
401 
402 	atmel_qspi_write(ifr, aq, QSPI_IFR);
403 
404 	return 0;
405 }
406 
407 static int atmel_qspi_exec_op(struct spi_mem *mem, const struct spi_mem_op *op)
408 {
409 	struct atmel_qspi *aq = spi_controller_get_devdata(mem->spi->controller);
410 	u32 sr, offset;
411 	int err;
412 
413 	/*
414 	 * Check if the address exceeds the MMIO window size. An improvement
415 	 * would be to add support for regular SPI mode and fall back to it
416 	 * when the flash memories overrun the controller's memory space.
417 	 */
418 	if (op->addr.val + op->data.nbytes > aq->mmap_size)
419 		return -ENOTSUPP;
420 
421 	err = pm_runtime_resume_and_get(&aq->pdev->dev);
422 	if (err < 0)
423 		return err;
424 
425 	err = atmel_qspi_set_cfg(aq, op, &offset);
426 	if (err)
427 		goto pm_runtime_put;
428 
429 	/* Skip to the final steps if there is no data */
430 	if (op->data.nbytes) {
431 		/* Dummy read of QSPI_IFR to synchronize APB and AHB accesses */
432 		(void)atmel_qspi_read(aq, QSPI_IFR);
433 
434 		/* Send/Receive data */
435 		if (op->data.dir == SPI_MEM_DATA_IN)
436 			memcpy_fromio(op->data.buf.in, aq->mem + offset,
437 				      op->data.nbytes);
438 		else
439 			memcpy_toio(aq->mem + offset, op->data.buf.out,
440 				    op->data.nbytes);
441 
442 		/* Release the chip-select */
443 		atmel_qspi_write(QSPI_CR_LASTXFER, aq, QSPI_CR);
444 	}
445 
446 	/* Poll INSTRuction End status */
447 	sr = atmel_qspi_read(aq, QSPI_SR);
448 	if ((sr & QSPI_SR_CMD_COMPLETED) == QSPI_SR_CMD_COMPLETED)
449 		goto pm_runtime_put;
450 
451 	/* Wait for INSTRuction End interrupt */
452 	reinit_completion(&aq->cmd_completion);
453 	aq->pending = sr & QSPI_SR_CMD_COMPLETED;
454 	atmel_qspi_write(QSPI_SR_CMD_COMPLETED, aq, QSPI_IER);
455 	if (!wait_for_completion_timeout(&aq->cmd_completion,
456 					 msecs_to_jiffies(1000)))
457 		err = -ETIMEDOUT;
458 	atmel_qspi_write(QSPI_SR_CMD_COMPLETED, aq, QSPI_IDR);
459 
460 pm_runtime_put:
461 	pm_runtime_mark_last_busy(&aq->pdev->dev);
462 	pm_runtime_put_autosuspend(&aq->pdev->dev);
463 	return err;
464 }
465 
466 static const char *atmel_qspi_get_name(struct spi_mem *spimem)
467 {
468 	return dev_name(spimem->spi->dev.parent);
469 }
470 
471 static const struct spi_controller_mem_ops atmel_qspi_mem_ops = {
472 	.supports_op = atmel_qspi_supports_op,
473 	.exec_op = atmel_qspi_exec_op,
474 	.get_name = atmel_qspi_get_name
475 };
476 
477 static int atmel_qspi_setup(struct spi_device *spi)
478 {
479 	struct spi_controller *ctrl = spi->controller;
480 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
481 	unsigned long src_rate;
482 	u32 scbr;
483 	int ret;
484 
485 	if (ctrl->busy)
486 		return -EBUSY;
487 
488 	if (!spi->max_speed_hz)
489 		return -EINVAL;
490 
491 	src_rate = clk_get_rate(aq->pclk);
492 	if (!src_rate)
493 		return -EINVAL;
494 
495 	/* Compute the QSPI baudrate */
496 	scbr = DIV_ROUND_UP(src_rate, spi->max_speed_hz);
497 	if (scbr > 0)
498 		scbr--;
499 
500 	ret = pm_runtime_resume_and_get(ctrl->dev.parent);
501 	if (ret < 0)
502 		return ret;
503 
504 	aq->scr &= ~QSPI_SCR_SCBR_MASK;
505 	aq->scr |= QSPI_SCR_SCBR(scbr);
506 	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
507 
508 	pm_runtime_mark_last_busy(ctrl->dev.parent);
509 	pm_runtime_put_autosuspend(ctrl->dev.parent);
510 
511 	return 0;
512 }
513 
514 static int atmel_qspi_set_cs_timing(struct spi_device *spi)
515 {
516 	struct spi_controller *ctrl = spi->controller;
517 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
518 	unsigned long clk_rate;
519 	u32 cs_setup;
520 	int delay;
521 	int ret;
522 
523 	delay = spi_delay_to_ns(&spi->cs_setup, NULL);
524 	if (delay <= 0)
525 		return delay;
526 
527 	clk_rate = clk_get_rate(aq->pclk);
528 	if (!clk_rate)
529 		return -EINVAL;
530 
531 	cs_setup = DIV_ROUND_UP((delay * DIV_ROUND_UP(clk_rate, 1000000)),
532 				1000);
533 
534 	ret = pm_runtime_resume_and_get(ctrl->dev.parent);
535 	if (ret < 0)
536 		return ret;
537 
538 	aq->scr &= ~QSPI_SCR_DLYBS_MASK;
539 	aq->scr |= QSPI_SCR_DLYBS(cs_setup);
540 	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
541 
542 	pm_runtime_mark_last_busy(ctrl->dev.parent);
543 	pm_runtime_put_autosuspend(ctrl->dev.parent);
544 
545 	return 0;
546 }
547 
548 static void atmel_qspi_init(struct atmel_qspi *aq)
549 {
550 	/* Reset the QSPI controller */
551 	atmel_qspi_write(QSPI_CR_SWRST, aq, QSPI_CR);
552 
553 	/* Set the QSPI controller by default in Serial Memory Mode */
554 	aq->mr |= QSPI_MR_SMM;
555 	atmel_qspi_write(aq->mr, aq, QSPI_MR);
556 
557 	/* Enable the QSPI controller */
558 	atmel_qspi_write(QSPI_CR_QSPIEN, aq, QSPI_CR);
559 }
560 
561 static irqreturn_t atmel_qspi_interrupt(int irq, void *dev_id)
562 {
563 	struct atmel_qspi *aq = dev_id;
564 	u32 status, mask, pending;
565 
566 	status = atmel_qspi_read(aq, QSPI_SR);
567 	mask = atmel_qspi_read(aq, QSPI_IMR);
568 	pending = status & mask;
569 
570 	if (!pending)
571 		return IRQ_NONE;
572 
573 	aq->pending |= pending;
574 	if ((aq->pending & QSPI_SR_CMD_COMPLETED) == QSPI_SR_CMD_COMPLETED)
575 		complete(&aq->cmd_completion);
576 
577 	return IRQ_HANDLED;
578 }
579 
580 static int atmel_qspi_probe(struct platform_device *pdev)
581 {
582 	struct spi_controller *ctrl;
583 	struct atmel_qspi *aq;
584 	struct resource *res;
585 	int irq, err = 0;
586 
587 	ctrl = devm_spi_alloc_host(&pdev->dev, sizeof(*aq));
588 	if (!ctrl)
589 		return -ENOMEM;
590 
591 	ctrl->mode_bits = SPI_RX_DUAL | SPI_RX_QUAD | SPI_TX_DUAL | SPI_TX_QUAD;
592 	ctrl->setup = atmel_qspi_setup;
593 	ctrl->set_cs_timing = atmel_qspi_set_cs_timing;
594 	ctrl->bus_num = -1;
595 	ctrl->mem_ops = &atmel_qspi_mem_ops;
596 	ctrl->num_chipselect = 1;
597 	ctrl->dev.of_node = pdev->dev.of_node;
598 	platform_set_drvdata(pdev, ctrl);
599 
600 	aq = spi_controller_get_devdata(ctrl);
601 
602 	init_completion(&aq->cmd_completion);
603 	aq->pdev = pdev;
604 
605 	/* Map the registers */
606 	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "qspi_base");
607 	aq->regs = devm_ioremap_resource(&pdev->dev, res);
608 	if (IS_ERR(aq->regs)) {
609 		dev_err(&pdev->dev, "missing registers\n");
610 		return PTR_ERR(aq->regs);
611 	}
612 
613 	/* Map the AHB memory */
614 	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "qspi_mmap");
615 	aq->mem = devm_ioremap_resource(&pdev->dev, res);
616 	if (IS_ERR(aq->mem)) {
617 		dev_err(&pdev->dev, "missing AHB memory\n");
618 		return PTR_ERR(aq->mem);
619 	}
620 
621 	aq->mmap_size = resource_size(res);
622 
623 	/* Get the peripheral clock */
624 	aq->pclk = devm_clk_get(&pdev->dev, "pclk");
625 	if (IS_ERR(aq->pclk))
626 		aq->pclk = devm_clk_get(&pdev->dev, NULL);
627 
628 	if (IS_ERR(aq->pclk)) {
629 		dev_err(&pdev->dev, "missing peripheral clock\n");
630 		return PTR_ERR(aq->pclk);
631 	}
632 
633 	/* Enable the peripheral clock */
634 	err = clk_prepare_enable(aq->pclk);
635 	if (err) {
636 		dev_err(&pdev->dev, "failed to enable the peripheral clock\n");
637 		return err;
638 	}
639 
640 	aq->caps = of_device_get_match_data(&pdev->dev);
641 	if (!aq->caps) {
642 		dev_err(&pdev->dev, "Could not retrieve QSPI caps\n");
643 		err = -EINVAL;
644 		goto disable_pclk;
645 	}
646 
647 	if (aq->caps->has_qspick) {
648 		/* Get the QSPI system clock */
649 		aq->qspick = devm_clk_get(&pdev->dev, "qspick");
650 		if (IS_ERR(aq->qspick)) {
651 			dev_err(&pdev->dev, "missing system clock\n");
652 			err = PTR_ERR(aq->qspick);
653 			goto disable_pclk;
654 		}
655 
656 		/* Enable the QSPI system clock */
657 		err = clk_prepare_enable(aq->qspick);
658 		if (err) {
659 			dev_err(&pdev->dev,
660 				"failed to enable the QSPI system clock\n");
661 			goto disable_pclk;
662 		}
663 	}
664 
665 	/* Request the IRQ */
666 	irq = platform_get_irq(pdev, 0);
667 	if (irq < 0) {
668 		err = irq;
669 		goto disable_qspick;
670 	}
671 	err = devm_request_irq(&pdev->dev, irq, atmel_qspi_interrupt,
672 			       0, dev_name(&pdev->dev), aq);
673 	if (err)
674 		goto disable_qspick;
675 
676 	pm_runtime_set_autosuspend_delay(&pdev->dev, 500);
677 	pm_runtime_use_autosuspend(&pdev->dev);
678 	pm_runtime_set_active(&pdev->dev);
679 	pm_runtime_enable(&pdev->dev);
680 	pm_runtime_get_noresume(&pdev->dev);
681 
682 	atmel_qspi_init(aq);
683 
684 	err = spi_register_controller(ctrl);
685 	if (err) {
686 		pm_runtime_put_noidle(&pdev->dev);
687 		pm_runtime_disable(&pdev->dev);
688 		pm_runtime_set_suspended(&pdev->dev);
689 		pm_runtime_dont_use_autosuspend(&pdev->dev);
690 		goto disable_qspick;
691 	}
692 	pm_runtime_mark_last_busy(&pdev->dev);
693 	pm_runtime_put_autosuspend(&pdev->dev);
694 
695 	return 0;
696 
697 disable_qspick:
698 	clk_disable_unprepare(aq->qspick);
699 disable_pclk:
700 	clk_disable_unprepare(aq->pclk);
701 
702 	return err;
703 }
704 
705 static void atmel_qspi_remove(struct platform_device *pdev)
706 {
707 	struct spi_controller *ctrl = platform_get_drvdata(pdev);
708 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
709 	int ret;
710 
711 	spi_unregister_controller(ctrl);
712 
713 	ret = pm_runtime_get_sync(&pdev->dev);
714 	if (ret >= 0) {
715 		atmel_qspi_write(QSPI_CR_QSPIDIS, aq, QSPI_CR);
716 		clk_disable(aq->qspick);
717 		clk_disable(aq->pclk);
718 	} else {
719 		/*
720 		 * atmel_qspi_runtime_{suspend,resume} just disable and enable
721 		 * the two clks respectively. So after resume failed these are
722 		 * off, and we skip hardware access and disabling these clks again.
723 		 */
724 		dev_warn(&pdev->dev, "Failed to resume device on remove\n");
725 	}
726 
727 	clk_unprepare(aq->qspick);
728 	clk_unprepare(aq->pclk);
729 
730 	pm_runtime_disable(&pdev->dev);
731 	pm_runtime_dont_use_autosuspend(&pdev->dev);
732 	pm_runtime_put_noidle(&pdev->dev);
733 }
734 
735 static int __maybe_unused atmel_qspi_suspend(struct device *dev)
736 {
737 	struct spi_controller *ctrl = dev_get_drvdata(dev);
738 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
739 	int ret;
740 
741 	ret = pm_runtime_resume_and_get(dev);
742 	if (ret < 0)
743 		return ret;
744 
745 	atmel_qspi_write(QSPI_CR_QSPIDIS, aq, QSPI_CR);
746 
747 	pm_runtime_mark_last_busy(dev);
748 	pm_runtime_force_suspend(dev);
749 
750 	clk_unprepare(aq->qspick);
751 	clk_unprepare(aq->pclk);
752 
753 	return 0;
754 }
755 
756 static int __maybe_unused atmel_qspi_resume(struct device *dev)
757 {
758 	struct spi_controller *ctrl = dev_get_drvdata(dev);
759 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
760 	int ret;
761 
762 	ret = clk_prepare(aq->pclk);
763 	if (ret)
764 		return ret;
765 
766 	ret = clk_prepare(aq->qspick);
767 	if (ret) {
768 		clk_unprepare(aq->pclk);
769 		return ret;
770 	}
771 
772 	ret = pm_runtime_force_resume(dev);
773 	if (ret < 0)
774 		return ret;
775 
776 	atmel_qspi_init(aq);
777 
778 	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
779 
780 	pm_runtime_mark_last_busy(dev);
781 	pm_runtime_put_autosuspend(dev);
782 
783 	return 0;
784 }
785 
786 static int __maybe_unused atmel_qspi_runtime_suspend(struct device *dev)
787 {
788 	struct spi_controller *ctrl = dev_get_drvdata(dev);
789 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
790 
791 	clk_disable(aq->qspick);
792 	clk_disable(aq->pclk);
793 
794 	return 0;
795 }
796 
797 static int __maybe_unused atmel_qspi_runtime_resume(struct device *dev)
798 {
799 	struct spi_controller *ctrl = dev_get_drvdata(dev);
800 	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
801 	int ret;
802 
803 	ret = clk_enable(aq->pclk);
804 	if (ret)
805 		return ret;
806 
807 	ret = clk_enable(aq->qspick);
808 	if (ret)
809 		clk_disable(aq->pclk);
810 
811 	return ret;
812 }
813 
814 static const struct dev_pm_ops __maybe_unused atmel_qspi_pm_ops = {
815 	SET_SYSTEM_SLEEP_PM_OPS(atmel_qspi_suspend, atmel_qspi_resume)
816 	SET_RUNTIME_PM_OPS(atmel_qspi_runtime_suspend,
817 			   atmel_qspi_runtime_resume, NULL)
818 };
819 
820 static const struct atmel_qspi_caps atmel_sama5d2_qspi_caps = {};
821 
822 static const struct atmel_qspi_caps atmel_sam9x60_qspi_caps = {
823 	.has_qspick = true,
824 	.has_ricr = true,
825 };
826 
827 static const struct of_device_id atmel_qspi_dt_ids[] = {
828 	{
829 		.compatible = "atmel,sama5d2-qspi",
830 		.data = &atmel_sama5d2_qspi_caps,
831 	},
832 	{
833 		.compatible = "microchip,sam9x60-qspi",
834 		.data = &atmel_sam9x60_qspi_caps,
835 	},
836 	{ /* sentinel */ }
837 };
838 
839 MODULE_DEVICE_TABLE(of, atmel_qspi_dt_ids);
840 
841 static struct platform_driver atmel_qspi_driver = {
842 	.driver = {
843 		.name	= "atmel_qspi",
844 		.of_match_table	= atmel_qspi_dt_ids,
845 		.pm	= pm_ptr(&atmel_qspi_pm_ops),
846 	},
847 	.probe		= atmel_qspi_probe,
848 	.remove_new	= atmel_qspi_remove,
849 };
850 module_platform_driver(atmel_qspi_driver);
851 
852 MODULE_AUTHOR("Cyrille Pitchen <cyrille.pitchen@atmel.com>");
853 MODULE_AUTHOR("Piotr Bugalski <bugalski.piotr@gmail.com");
854 MODULE_DESCRIPTION("Atmel QSPI Controller driver");
855 MODULE_LICENSE("GPL v2");
856