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