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
141struct atmel_qspi_caps {
142	bool has_qspick;
143	bool has_ricr;
144};
145
146struct 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
160struct atmel_qspi_mode {
161	u8 cmd_buswidth;
162	u8 addr_buswidth;
163	u8 data_buswidth;
164	u32 config;
165};
166
167static 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
178static 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
226static 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
240static 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
252static 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
267static 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 -EOPNOTSUPP;
276}
277
278static 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
295static 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		atmel_qspi_write(QSPI_MR_SMM, aq, QSPI_MR);
380		aq->mr = QSPI_MR_SMM;
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
407static 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
460pm_runtime_put:
461	pm_runtime_mark_last_busy(&aq->pdev->dev);
462	pm_runtime_put_autosuspend(&aq->pdev->dev);
463	return err;
464}
465
466static const char *atmel_qspi_get_name(struct spi_mem *spimem)
467{
468	return dev_name(spimem->spi->dev.parent);
469}
470
471static 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
477static 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(scbr);
505	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
506
507	pm_runtime_mark_last_busy(ctrl->dev.parent);
508	pm_runtime_put_autosuspend(ctrl->dev.parent);
509
510	return 0;
511}
512
513static int atmel_qspi_set_cs_timing(struct spi_device *spi)
514{
515	struct spi_controller *ctrl = spi->controller;
516	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
517	unsigned long clk_rate;
518	u32 cs_setup;
519	int delay;
520	int ret;
521
522	delay = spi_delay_to_ns(&spi->cs_setup, NULL);
523	if (delay <= 0)
524		return delay;
525
526	clk_rate = clk_get_rate(aq->pclk);
527	if (!clk_rate)
528		return -EINVAL;
529
530	cs_setup = DIV_ROUND_UP((delay * DIV_ROUND_UP(clk_rate, 1000000)),
531				1000);
532
533	ret = pm_runtime_resume_and_get(ctrl->dev.parent);
534	if (ret < 0)
535		return ret;
536
537	aq->scr |= QSPI_SCR_DLYBS(cs_setup);
538	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
539
540	pm_runtime_mark_last_busy(ctrl->dev.parent);
541	pm_runtime_put_autosuspend(ctrl->dev.parent);
542
543	return 0;
544}
545
546static void atmel_qspi_init(struct atmel_qspi *aq)
547{
548	/* Reset the QSPI controller */
549	atmel_qspi_write(QSPI_CR_SWRST, aq, QSPI_CR);
550
551	/* Set the QSPI controller by default in Serial Memory Mode */
552	atmel_qspi_write(QSPI_MR_SMM, aq, QSPI_MR);
553	aq->mr = QSPI_MR_SMM;
554
555	/* Enable the QSPI controller */
556	atmel_qspi_write(QSPI_CR_QSPIEN, aq, QSPI_CR);
557}
558
559static irqreturn_t atmel_qspi_interrupt(int irq, void *dev_id)
560{
561	struct atmel_qspi *aq = dev_id;
562	u32 status, mask, pending;
563
564	status = atmel_qspi_read(aq, QSPI_SR);
565	mask = atmel_qspi_read(aq, QSPI_IMR);
566	pending = status & mask;
567
568	if (!pending)
569		return IRQ_NONE;
570
571	aq->pending |= pending;
572	if ((aq->pending & QSPI_SR_CMD_COMPLETED) == QSPI_SR_CMD_COMPLETED)
573		complete(&aq->cmd_completion);
574
575	return IRQ_HANDLED;
576}
577
578static int atmel_qspi_probe(struct platform_device *pdev)
579{
580	struct spi_controller *ctrl;
581	struct atmel_qspi *aq;
582	struct resource *res;
583	int irq, err = 0;
584
585	ctrl = devm_spi_alloc_host(&pdev->dev, sizeof(*aq));
586	if (!ctrl)
587		return -ENOMEM;
588
589	ctrl->mode_bits = SPI_RX_DUAL | SPI_RX_QUAD | SPI_TX_DUAL | SPI_TX_QUAD;
590	ctrl->setup = atmel_qspi_setup;
591	ctrl->set_cs_timing = atmel_qspi_set_cs_timing;
592	ctrl->bus_num = -1;
593	ctrl->mem_ops = &atmel_qspi_mem_ops;
594	ctrl->num_chipselect = 1;
595	ctrl->dev.of_node = pdev->dev.of_node;
596	platform_set_drvdata(pdev, ctrl);
597
598	aq = spi_controller_get_devdata(ctrl);
599
600	init_completion(&aq->cmd_completion);
601	aq->pdev = pdev;
602
603	/* Map the registers */
604	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "qspi_base");
605	aq->regs = devm_ioremap_resource(&pdev->dev, res);
606	if (IS_ERR(aq->regs)) {
607		dev_err(&pdev->dev, "missing registers\n");
608		return PTR_ERR(aq->regs);
609	}
610
611	/* Map the AHB memory */
612	res = platform_get_resource_byname(pdev, IORESOURCE_MEM, "qspi_mmap");
613	aq->mem = devm_ioremap_resource(&pdev->dev, res);
614	if (IS_ERR(aq->mem)) {
615		dev_err(&pdev->dev, "missing AHB memory\n");
616		return PTR_ERR(aq->mem);
617	}
618
619	aq->mmap_size = resource_size(res);
620
621	/* Get the peripheral clock */
622	aq->pclk = devm_clk_get(&pdev->dev, "pclk");
623	if (IS_ERR(aq->pclk))
624		aq->pclk = devm_clk_get(&pdev->dev, NULL);
625
626	if (IS_ERR(aq->pclk)) {
627		dev_err(&pdev->dev, "missing peripheral clock\n");
628		return PTR_ERR(aq->pclk);
629	}
630
631	/* Enable the peripheral clock */
632	err = clk_prepare_enable(aq->pclk);
633	if (err) {
634		dev_err(&pdev->dev, "failed to enable the peripheral clock\n");
635		return err;
636	}
637
638	aq->caps = of_device_get_match_data(&pdev->dev);
639	if (!aq->caps) {
640		dev_err(&pdev->dev, "Could not retrieve QSPI caps\n");
641		err = -EINVAL;
642		goto disable_pclk;
643	}
644
645	if (aq->caps->has_qspick) {
646		/* Get the QSPI system clock */
647		aq->qspick = devm_clk_get(&pdev->dev, "qspick");
648		if (IS_ERR(aq->qspick)) {
649			dev_err(&pdev->dev, "missing system clock\n");
650			err = PTR_ERR(aq->qspick);
651			goto disable_pclk;
652		}
653
654		/* Enable the QSPI system clock */
655		err = clk_prepare_enable(aq->qspick);
656		if (err) {
657			dev_err(&pdev->dev,
658				"failed to enable the QSPI system clock\n");
659			goto disable_pclk;
660		}
661	}
662
663	/* Request the IRQ */
664	irq = platform_get_irq(pdev, 0);
665	if (irq < 0) {
666		err = irq;
667		goto disable_qspick;
668	}
669	err = devm_request_irq(&pdev->dev, irq, atmel_qspi_interrupt,
670			       0, dev_name(&pdev->dev), aq);
671	if (err)
672		goto disable_qspick;
673
674	pm_runtime_set_autosuspend_delay(&pdev->dev, 500);
675	pm_runtime_use_autosuspend(&pdev->dev);
676	pm_runtime_set_active(&pdev->dev);
677	pm_runtime_enable(&pdev->dev);
678	pm_runtime_get_noresume(&pdev->dev);
679
680	atmel_qspi_init(aq);
681
682	err = spi_register_controller(ctrl);
683	if (err) {
684		pm_runtime_put_noidle(&pdev->dev);
685		pm_runtime_disable(&pdev->dev);
686		pm_runtime_set_suspended(&pdev->dev);
687		pm_runtime_dont_use_autosuspend(&pdev->dev);
688		goto disable_qspick;
689	}
690	pm_runtime_mark_last_busy(&pdev->dev);
691	pm_runtime_put_autosuspend(&pdev->dev);
692
693	return 0;
694
695disable_qspick:
696	clk_disable_unprepare(aq->qspick);
697disable_pclk:
698	clk_disable_unprepare(aq->pclk);
699
700	return err;
701}
702
703static void atmel_qspi_remove(struct platform_device *pdev)
704{
705	struct spi_controller *ctrl = platform_get_drvdata(pdev);
706	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
707	int ret;
708
709	spi_unregister_controller(ctrl);
710
711	ret = pm_runtime_get_sync(&pdev->dev);
712	if (ret >= 0) {
713		atmel_qspi_write(QSPI_CR_QSPIDIS, aq, QSPI_CR);
714		clk_disable(aq->qspick);
715		clk_disable(aq->pclk);
716	} else {
717		/*
718		 * atmel_qspi_runtime_{suspend,resume} just disable and enable
719		 * the two clks respectively. So after resume failed these are
720		 * off, and we skip hardware access and disabling these clks again.
721		 */
722		dev_warn(&pdev->dev, "Failed to resume device on remove\n");
723	}
724
725	clk_unprepare(aq->qspick);
726	clk_unprepare(aq->pclk);
727
728	pm_runtime_disable(&pdev->dev);
729	pm_runtime_put_noidle(&pdev->dev);
730}
731
732static int __maybe_unused atmel_qspi_suspend(struct device *dev)
733{
734	struct spi_controller *ctrl = dev_get_drvdata(dev);
735	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
736	int ret;
737
738	ret = pm_runtime_resume_and_get(dev);
739	if (ret < 0)
740		return ret;
741
742	atmel_qspi_write(QSPI_CR_QSPIDIS, aq, QSPI_CR);
743
744	pm_runtime_mark_last_busy(dev);
745	pm_runtime_force_suspend(dev);
746
747	clk_unprepare(aq->qspick);
748	clk_unprepare(aq->pclk);
749
750	return 0;
751}
752
753static int __maybe_unused atmel_qspi_resume(struct device *dev)
754{
755	struct spi_controller *ctrl = dev_get_drvdata(dev);
756	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
757	int ret;
758
759	clk_prepare(aq->pclk);
760	clk_prepare(aq->qspick);
761
762	ret = pm_runtime_force_resume(dev);
763	if (ret < 0)
764		return ret;
765
766	atmel_qspi_init(aq);
767
768	atmel_qspi_write(aq->scr, aq, QSPI_SCR);
769
770	pm_runtime_mark_last_busy(dev);
771	pm_runtime_put_autosuspend(dev);
772
773	return 0;
774}
775
776static int __maybe_unused atmel_qspi_runtime_suspend(struct device *dev)
777{
778	struct spi_controller *ctrl = dev_get_drvdata(dev);
779	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
780
781	clk_disable(aq->qspick);
782	clk_disable(aq->pclk);
783
784	return 0;
785}
786
787static int __maybe_unused atmel_qspi_runtime_resume(struct device *dev)
788{
789	struct spi_controller *ctrl = dev_get_drvdata(dev);
790	struct atmel_qspi *aq = spi_controller_get_devdata(ctrl);
791	int ret;
792
793	ret = clk_enable(aq->pclk);
794	if (ret)
795		return ret;
796
797	ret = clk_enable(aq->qspick);
798	if (ret)
799		clk_disable(aq->pclk);
800
801	return ret;
802}
803
804static const struct dev_pm_ops __maybe_unused atmel_qspi_pm_ops = {
805	SET_SYSTEM_SLEEP_PM_OPS(atmel_qspi_suspend, atmel_qspi_resume)
806	SET_RUNTIME_PM_OPS(atmel_qspi_runtime_suspend,
807			   atmel_qspi_runtime_resume, NULL)
808};
809
810static const struct atmel_qspi_caps atmel_sama5d2_qspi_caps = {};
811
812static const struct atmel_qspi_caps atmel_sam9x60_qspi_caps = {
813	.has_qspick = true,
814	.has_ricr = true,
815};
816
817static const struct of_device_id atmel_qspi_dt_ids[] = {
818	{
819		.compatible = "atmel,sama5d2-qspi",
820		.data = &atmel_sama5d2_qspi_caps,
821	},
822	{
823		.compatible = "microchip,sam9x60-qspi",
824		.data = &atmel_sam9x60_qspi_caps,
825	},
826	{ /* sentinel */ }
827};
828
829MODULE_DEVICE_TABLE(of, atmel_qspi_dt_ids);
830
831static struct platform_driver atmel_qspi_driver = {
832	.driver = {
833		.name	= "atmel_qspi",
834		.of_match_table	= atmel_qspi_dt_ids,
835		.pm	= pm_ptr(&atmel_qspi_pm_ops),
836	},
837	.probe		= atmel_qspi_probe,
838	.remove_new	= atmel_qspi_remove,
839};
840module_platform_driver(atmel_qspi_driver);
841
842MODULE_AUTHOR("Cyrille Pitchen <cyrille.pitchen@atmel.com>");
843MODULE_AUTHOR("Piotr Bugalski <bugalski.piotr@gmail.com");
844MODULE_DESCRIPTION("Atmel QSPI Controller driver");
845MODULE_LICENSE("GPL v2");
846