Merge branch 'next'
[platform/kernel/u-boot.git] / drivers / spi / zynqmp_gqspi.c
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * (C) Copyright 2018 Xilinx
4  *
5  * Xilinx ZynqMP Generic Quad-SPI(QSPI) controller driver(master mode only)
6  */
7
8 #include <common.h>
9 #include <cpu_func.h>
10 #include <log.h>
11 #include <asm/arch/sys_proto.h>
12 #include <asm/cache.h>
13 #include <asm/io.h>
14 #include <clk.h>
15 #include <dm.h>
16 #include <malloc.h>
17 #include <memalign.h>
18 #include <spi.h>
19 #include <spi-mem.h>
20 #include <ubi_uboot.h>
21 #include <wait_bit.h>
22 #include <dm/device_compat.h>
23 #include <linux/bitops.h>
24 #include <linux/err.h>
25 #include <linux/sizes.h>
26 #include <zynqmp_firmware.h>
27
28 #define GQSPI_GFIFO_STRT_MODE_MASK      BIT(29)
29 #define GQSPI_CONFIG_MODE_EN_MASK       (3 << 30)
30 #define GQSPI_CONFIG_DMA_MODE           (2 << 30)
31 #define GQSPI_CONFIG_CPHA_MASK          BIT(2)
32 #define GQSPI_CONFIG_CPOL_MASK          BIT(1)
33
34 /*
35  * QSPI Interrupt Registers bit Masks
36  *
37  * All the four interrupt registers (Status/Mask/Enable/Disable) have the same
38  * bit definitions.
39  */
40 #define GQSPI_IXR_TXNFULL_MASK          0x00000004 /* QSPI TX FIFO Overflow */
41 #define GQSPI_IXR_TXFULL_MASK           0x00000008 /* QSPI TX FIFO is full */
42 #define GQSPI_IXR_TXFIFOEMPTY_MASK      0x00000100 /* QSPI TX FIFO is Empty */
43 #define GQSPI_IXR_RXNEMTY_MASK          0x00000010 /* QSPI RX FIFO Not Empty */
44 #define GQSPI_IXR_GFEMTY_MASK           0x00000080 /* QSPI Generic FIFO Empty */
45 #define GQSPI_IXR_GFNFULL_MASK          0x00000200 /* QSPI GENFIFO not full */
46 #define GQSPI_IXR_ALL_MASK              (GQSPI_IXR_TXNFULL_MASK | \
47                                          GQSPI_IXR_RXNEMTY_MASK)
48
49 /*
50  * QSPI Enable Register bit Masks
51  *
52  * This register is used to enable or disable the QSPI controller
53  */
54 #define GQSPI_ENABLE_ENABLE_MASK        0x00000001 /* QSPI Enable Bit Mask */
55
56 #define GQSPI_GFIFO_LOW_BUS             BIT(14)
57 #define GQSPI_GFIFO_CS_LOWER            BIT(12)
58 #define GQSPI_GFIFO_UP_BUS              BIT(15)
59 #define GQSPI_GFIFO_CS_UPPER            BIT(13)
60 #define GQSPI_SPI_MODE_QSPI             (3 << 10)
61 #define GQSPI_SPI_MODE_SPI              BIT(10)
62 #define GQSPI_SPI_MODE_DUAL_SPI         (2 << 10)
63 #define GQSPI_IMD_DATA_CS_ASSERT        5
64 #define GQSPI_IMD_DATA_CS_DEASSERT      5
65 #define GQSPI_GFIFO_TX                  BIT(16)
66 #define GQSPI_GFIFO_RX                  BIT(17)
67 #define GQSPI_GFIFO_STRIPE_MASK         BIT(18)
68 #define GQSPI_GFIFO_IMD_MASK            0xFF
69 #define GQSPI_GFIFO_EXP_MASK            BIT(9)
70 #define GQSPI_GFIFO_DATA_XFR_MASK       BIT(8)
71 #define GQSPI_STRT_GEN_FIFO             BIT(28)
72 #define GQSPI_GEN_FIFO_STRT_MOD         BIT(29)
73 #define GQSPI_GFIFO_WP_HOLD             BIT(19)
74 #define GQSPI_BAUD_DIV_MASK             (7 << 3)
75 #define GQSPI_DFLT_BAUD_RATE_DIV        BIT(3)
76 #define GQSPI_GFIFO_ALL_INT_MASK        0xFBE
77 #define GQSPI_DMA_DST_I_STS_DONE        BIT(1)
78 #define GQSPI_DMA_DST_I_STS_MASK        0xFE
79 #define MODEBITS                        0x6
80
81 #define GQSPI_GFIFO_SELECT              BIT(0)
82 #define GQSPI_FIFO_THRESHOLD            1
83 #define GQSPI_GENFIFO_THRESHOLD         31
84
85 #define SPI_XFER_ON_BOTH                0
86 #define SPI_XFER_ON_LOWER               1
87 #define SPI_XFER_ON_UPPER               2
88
89 #define GQSPI_DMA_ALIGN                 0x4
90 #define GQSPI_MAX_BAUD_RATE_VAL         7
91 #define GQSPI_DFLT_BAUD_RATE_VAL        2
92
93 #define GQSPI_TIMEOUT                   100000000
94
95 #define GQSPI_BAUD_DIV_SHIFT            2
96 #define GQSPI_LPBK_DLY_ADJ_LPBK_SHIFT   5
97 #define GQSPI_LPBK_DLY_ADJ_DLY_1        0x2
98 #define GQSPI_LPBK_DLY_ADJ_DLY_1_SHIFT  3
99 #define GQSPI_LPBK_DLY_ADJ_DLY_0        0x3
100 #define GQSPI_USE_DATA_DLY              0x1
101 #define GQSPI_USE_DATA_DLY_SHIFT        31
102 #define GQSPI_DATA_DLY_ADJ_VALUE        0x2
103 #define GQSPI_DATA_DLY_ADJ_SHIFT        28
104 #define TAP_DLY_BYPASS_LQSPI_RX_VALUE   0x1
105 #define TAP_DLY_BYPASS_LQSPI_RX_SHIFT   2
106 #define GQSPI_DATA_DLY_ADJ_OFST         0x000001F8
107 #define IOU_TAPDLY_BYPASS_OFST !IS_ENABLED(CONFIG_ARCH_VERSAL) ? \
108                                 0xFF180390 : 0xF103003C
109 #define GQSPI_LPBK_DLY_ADJ_LPBK_MASK    0x00000020
110 #define GQSPI_FREQ_37_5MHZ              37500000
111 #define GQSPI_FREQ_40MHZ                40000000
112 #define GQSPI_FREQ_100MHZ               100000000
113 #define GQSPI_FREQ_150MHZ               150000000
114 #define IOU_TAPDLY_BYPASS_MASK          0x7
115
116 #define GQSPI_REG_OFFSET                0x100
117 #define GQSPI_DMA_REG_OFFSET            0x800
118
119 /* QSPI register offsets */
120 struct zynqmp_qspi_regs {
121         u32 confr;      /* 0x00 */
122         u32 isr;        /* 0x04 */
123         u32 ier;        /* 0x08 */
124         u32 idisr;      /* 0x0C */
125         u32 imaskr;     /* 0x10 */
126         u32 enbr;       /* 0x14 */
127         u32 dr;         /* 0x18 */
128         u32 txd0r;      /* 0x1C */
129         u32 drxr;       /* 0x20 */
130         u32 sicr;       /* 0x24 */
131         u32 txftr;      /* 0x28 */
132         u32 rxftr;      /* 0x2C */
133         u32 gpior;      /* 0x30 */
134         u32 reserved0;  /* 0x34 */
135         u32 lpbkdly;    /* 0x38 */
136         u32 reserved1;  /* 0x3C */
137         u32 genfifo;    /* 0x40 */
138         u32 gqspisel;   /* 0x44 */
139         u32 reserved2;  /* 0x48 */
140         u32 gqfifoctrl; /* 0x4C */
141         u32 gqfthr;     /* 0x50 */
142         u32 gqpollcfg;  /* 0x54 */
143         u32 gqpollto;   /* 0x58 */
144         u32 gqxfersts;  /* 0x5C */
145         u32 gqfifosnap; /* 0x60 */
146         u32 gqrxcpy;    /* 0x64 */
147         u32 reserved3[36];      /* 0x68 */
148         u32 gqspidlyadj;        /* 0xF8 */
149 };
150
151 struct zynqmp_qspi_dma_regs {
152         u32 dmadst;     /* 0x00 */
153         u32 dmasize;    /* 0x04 */
154         u32 dmasts;     /* 0x08 */
155         u32 dmactrl;    /* 0x0C */
156         u32 reserved0;  /* 0x10 */
157         u32 dmaisr;     /* 0x14 */
158         u32 dmaier;     /* 0x18 */
159         u32 dmaidr;     /* 0x1C */
160         u32 dmaimr;     /* 0x20 */
161         u32 dmactrl2;   /* 0x24 */
162         u32 dmadstmsb;  /* 0x28 */
163 };
164
165 struct zynqmp_qspi_plat {
166         struct zynqmp_qspi_regs *regs;
167         struct zynqmp_qspi_dma_regs *dma_regs;
168         u32 frequency;
169         u32 speed_hz;
170         unsigned int io_mode;
171 };
172
173 struct zynqmp_qspi_priv {
174         struct zynqmp_qspi_regs *regs;
175         struct zynqmp_qspi_dma_regs *dma_regs;
176         const void *tx_buf;
177         void *rx_buf;
178         unsigned int len;
179         unsigned int io_mode;
180         int bytes_to_transfer;
181         int bytes_to_receive;
182         const struct spi_mem_op *op;
183 };
184
185 static int zynqmp_qspi_of_to_plat(struct udevice *bus)
186 {
187         struct zynqmp_qspi_plat *plat = dev_get_plat(bus);
188
189         debug("%s\n", __func__);
190
191         plat->regs = (struct zynqmp_qspi_regs *)(dev_read_addr(bus) +
192                                                  GQSPI_REG_OFFSET);
193         plat->dma_regs = (struct zynqmp_qspi_dma_regs *)
194                           (dev_read_addr(bus) + GQSPI_DMA_REG_OFFSET);
195
196         plat->io_mode = dev_read_bool(bus, "has-io-mode");
197
198         return 0;
199 }
200
201 static void zynqmp_qspi_init_hw(struct zynqmp_qspi_priv *priv)
202 {
203         u32 config_reg;
204         struct zynqmp_qspi_regs *regs = priv->regs;
205
206         writel(GQSPI_GFIFO_SELECT, &regs->gqspisel);
207         writel(GQSPI_GFIFO_ALL_INT_MASK, &regs->idisr);
208         writel(GQSPI_FIFO_THRESHOLD, &regs->txftr);
209         writel(GQSPI_FIFO_THRESHOLD, &regs->rxftr);
210         writel(GQSPI_GENFIFO_THRESHOLD, &regs->gqfthr);
211         writel(GQSPI_GFIFO_ALL_INT_MASK, &regs->isr);
212         writel(~GQSPI_ENABLE_ENABLE_MASK, &regs->enbr);
213
214         config_reg = readl(&regs->confr);
215         config_reg &= ~(GQSPI_GFIFO_STRT_MODE_MASK |
216                         GQSPI_CONFIG_MODE_EN_MASK);
217         config_reg |= GQSPI_GFIFO_WP_HOLD | GQSPI_DFLT_BAUD_RATE_DIV;
218         config_reg |= GQSPI_GFIFO_STRT_MODE_MASK;
219         if (!priv->io_mode)
220                 config_reg |= GQSPI_CONFIG_DMA_MODE;
221
222         writel(config_reg, &regs->confr);
223
224         writel(GQSPI_ENABLE_ENABLE_MASK, &regs->enbr);
225 }
226
227 static u32 zynqmp_qspi_bus_select(struct zynqmp_qspi_priv *priv)
228 {
229         u32 gqspi_fifo_reg = 0;
230
231         gqspi_fifo_reg = GQSPI_GFIFO_LOW_BUS |
232                          GQSPI_GFIFO_CS_LOWER;
233
234         return gqspi_fifo_reg;
235 }
236
237 static u32 zynqmp_qspi_genfifo_mode(u8 buswidth)
238 {
239         switch (buswidth) {
240         case 1:
241                 return GQSPI_SPI_MODE_SPI;
242         case 2:
243                 return GQSPI_SPI_MODE_DUAL_SPI;
244         case 4:
245                 return GQSPI_SPI_MODE_QSPI;
246         default:
247                 debug("Unsupported bus width %u\n", buswidth);
248                 return GQSPI_SPI_MODE_SPI;
249         }
250 }
251
252 static void zynqmp_qspi_fill_gen_fifo(struct zynqmp_qspi_priv *priv,
253                                       u32 gqspi_fifo_reg)
254 {
255         struct zynqmp_qspi_regs *regs = priv->regs;
256         u32 config_reg, ier;
257         int ret = 0;
258
259         writel(gqspi_fifo_reg, &regs->genfifo);
260
261         config_reg = readl(&regs->confr);
262         /* Manual start if needed */
263         config_reg |= GQSPI_STRT_GEN_FIFO;
264         writel(config_reg, &regs->confr);
265
266         /* Enable interrupts */
267         ier = readl(&regs->ier);
268         ier |= GQSPI_IXR_GFEMTY_MASK;
269         writel(ier, &regs->ier);
270
271         /* Wait until the gen fifo is empty to write the new command */
272         ret = wait_for_bit_le32(&regs->isr, GQSPI_IXR_GFEMTY_MASK, 1,
273                                 GQSPI_TIMEOUT, 1);
274         if (ret)
275                 printf("%s Timeout\n", __func__);
276
277 }
278
279 static void zynqmp_qspi_chipselect(struct zynqmp_qspi_priv *priv, int is_on)
280 {
281         u32 gqspi_fifo_reg = 0;
282
283         if (is_on) {
284                 gqspi_fifo_reg = zynqmp_qspi_bus_select(priv);
285                 gqspi_fifo_reg |= GQSPI_SPI_MODE_SPI |
286                                   GQSPI_IMD_DATA_CS_ASSERT;
287         } else {
288                 gqspi_fifo_reg = GQSPI_GFIFO_LOW_BUS;
289                 gqspi_fifo_reg |= GQSPI_IMD_DATA_CS_DEASSERT;
290         }
291
292         debug("GFIFO_CMD_CS: 0x%x\n", gqspi_fifo_reg);
293
294         zynqmp_qspi_fill_gen_fifo(priv, gqspi_fifo_reg);
295 }
296
297 void zynqmp_qspi_set_tapdelay(struct udevice *bus, u32 baudrateval)
298 {
299         struct zynqmp_qspi_plat *plat = dev_get_plat(bus);
300         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
301         struct zynqmp_qspi_regs *regs = priv->regs;
302         u32 tapdlybypass = 0, lpbkdlyadj = 0, datadlyadj = 0, clk_rate;
303         u32 reqhz = 0;
304
305         clk_rate = plat->frequency;
306         reqhz = (clk_rate / (GQSPI_BAUD_DIV_SHIFT << baudrateval));
307
308         debug("%s, req_hz:%d, clk_rate:%d, baudrateval:%d\n",
309               __func__, reqhz, clk_rate, baudrateval);
310
311         if (!(IS_ENABLED(CONFIG_ARCH_VERSAL) ||
312               IS_ENABLED(CONFIG_ARCH_VERSAL_NET))) {
313                 if (reqhz <= GQSPI_FREQ_40MHZ) {
314                         tapdlybypass = TAP_DLY_BYPASS_LQSPI_RX_VALUE <<
315                                         TAP_DLY_BYPASS_LQSPI_RX_SHIFT;
316                 } else if (reqhz <= GQSPI_FREQ_100MHZ) {
317                         tapdlybypass = TAP_DLY_BYPASS_LQSPI_RX_VALUE <<
318                                         TAP_DLY_BYPASS_LQSPI_RX_SHIFT;
319                         lpbkdlyadj = GQSPI_LPBK_DLY_ADJ_LPBK_MASK;
320                         datadlyadj = (GQSPI_USE_DATA_DLY <<
321                                       GQSPI_USE_DATA_DLY_SHIFT) |
322                                        (GQSPI_DATA_DLY_ADJ_VALUE <<
323                                         GQSPI_DATA_DLY_ADJ_SHIFT);
324                 } else if (reqhz <= GQSPI_FREQ_150MHZ) {
325                         lpbkdlyadj = GQSPI_LPBK_DLY_ADJ_LPBK_MASK |
326                                       GQSPI_LPBK_DLY_ADJ_DLY_0;
327                 }
328                 zynqmp_mmio_write(IOU_TAPDLY_BYPASS_OFST,
329                                   IOU_TAPDLY_BYPASS_MASK, tapdlybypass);
330         } else {
331                 if (reqhz <= GQSPI_FREQ_37_5MHZ) {
332                         tapdlybypass = TAP_DLY_BYPASS_LQSPI_RX_VALUE <<
333                                         TAP_DLY_BYPASS_LQSPI_RX_SHIFT;
334                 } else if (reqhz <= GQSPI_FREQ_100MHZ) {
335                         tapdlybypass = TAP_DLY_BYPASS_LQSPI_RX_VALUE <<
336                                         TAP_DLY_BYPASS_LQSPI_RX_SHIFT;
337                         lpbkdlyadj = GQSPI_LPBK_DLY_ADJ_LPBK_MASK;
338                         datadlyadj = GQSPI_USE_DATA_DLY <<
339                                       GQSPI_USE_DATA_DLY_SHIFT;
340                 } else if (reqhz <= GQSPI_FREQ_150MHZ) {
341                         lpbkdlyadj = GQSPI_LPBK_DLY_ADJ_LPBK_MASK |
342                                       (GQSPI_LPBK_DLY_ADJ_DLY_1 <<
343                                        GQSPI_LPBK_DLY_ADJ_DLY_1_SHIFT);
344                 }
345                 writel(tapdlybypass, IOU_TAPDLY_BYPASS_OFST);
346         }
347         writel(lpbkdlyadj, &regs->lpbkdly);
348         writel(datadlyadj, &regs->gqspidlyadj);
349 }
350
351 static int zynqmp_qspi_set_speed(struct udevice *bus, uint speed)
352 {
353         struct zynqmp_qspi_plat *plat = dev_get_plat(bus);
354         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
355         struct zynqmp_qspi_regs *regs = priv->regs;
356         u32 confr;
357         u8 baud_rate_val = 0;
358
359         debug("%s\n", __func__);
360         if (speed > plat->frequency)
361                 speed = plat->frequency;
362
363         if (plat->speed_hz != speed) {
364                 /* Set the clock frequency */
365                 /* If speed == 0, default to lowest speed */
366                 while ((baud_rate_val < 8) &&
367                        ((plat->frequency /
368                        (2 << baud_rate_val)) > speed))
369                         baud_rate_val++;
370
371                 if (baud_rate_val > GQSPI_MAX_BAUD_RATE_VAL)
372                         baud_rate_val = GQSPI_DFLT_BAUD_RATE_VAL;
373
374                 plat->speed_hz = plat->frequency / (2 << baud_rate_val);
375
376                 confr = readl(&regs->confr);
377                 confr &= ~GQSPI_BAUD_DIV_MASK;
378                 confr |= (baud_rate_val << 3);
379                 writel(confr, &regs->confr);
380                 zynqmp_qspi_set_tapdelay(bus, baud_rate_val);
381
382                 debug("regs=%p, speed=%d\n", priv->regs, plat->speed_hz);
383         }
384
385         return 0;
386 }
387
388 static int zynqmp_qspi_probe(struct udevice *bus)
389 {
390         struct zynqmp_qspi_plat *plat = dev_get_plat(bus);
391         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
392         struct clk clk;
393         unsigned long clock;
394         int ret;
395
396         debug("%s: bus:%p, priv:%p\n", __func__, bus, priv);
397
398         priv->regs = plat->regs;
399         priv->dma_regs = plat->dma_regs;
400         priv->io_mode = plat->io_mode;
401
402         ret = clk_get_by_index(bus, 0, &clk);
403         if (ret < 0) {
404                 dev_err(bus, "failed to get clock\n");
405                 return ret;
406         }
407
408         clock = clk_get_rate(&clk);
409         if (IS_ERR_VALUE(clock)) {
410                 dev_err(bus, "failed to get rate\n");
411                 return clock;
412         }
413         debug("%s: CLK %ld\n", __func__, clock);
414
415         ret = clk_enable(&clk);
416         if (ret) {
417                 dev_err(bus, "failed to enable clock\n");
418                 return ret;
419         }
420         plat->frequency = clock;
421         plat->speed_hz = plat->frequency / 2;
422
423         /* init the zynq spi hw */
424         zynqmp_qspi_init_hw(priv);
425
426         return 0;
427 }
428
429 static int zynqmp_qspi_set_mode(struct udevice *bus, uint mode)
430 {
431         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
432         struct zynqmp_qspi_regs *regs = priv->regs;
433         u32 confr;
434
435         debug("%s\n", __func__);
436         /* Set the SPI Clock phase and polarities */
437         confr = readl(&regs->confr);
438         confr &= ~(GQSPI_CONFIG_CPHA_MASK | GQSPI_CONFIG_CPOL_MASK);
439
440         if (mode & SPI_CPHA)
441                 confr |= GQSPI_CONFIG_CPHA_MASK;
442         if (mode & SPI_CPOL)
443                 confr |= GQSPI_CONFIG_CPOL_MASK;
444
445         writel(confr, &regs->confr);
446
447         return 0;
448 }
449
450 static int zynqmp_qspi_fill_tx_fifo(struct zynqmp_qspi_priv *priv, u32 size)
451 {
452         u32 data;
453         int ret = 0;
454         struct zynqmp_qspi_regs *regs = priv->regs;
455         u32 *buf = (u32 *)priv->tx_buf;
456         u32 len = size;
457
458         debug("TxFIFO: 0x%x, size: 0x%x\n", readl(&regs->isr),
459               size);
460
461         while (size) {
462                 ret = wait_for_bit_le32(&regs->isr, GQSPI_IXR_TXNFULL_MASK, 1,
463                                         GQSPI_TIMEOUT, 1);
464                 if (ret) {
465                         printf("%s: Timeout\n", __func__);
466                         return ret;
467                 }
468
469                 if (size >= 4) {
470                         writel(*buf, &regs->txd0r);
471                         buf++;
472                         size -= 4;
473                 } else {
474                         switch (size) {
475                         case 1:
476                                 data = *((u8 *)buf);
477                                 buf += 1;
478                                 data |= GENMASK(31, 8);
479                                 break;
480                         case 2:
481                                 data = *((u16 *)buf);
482                                 buf += 2;
483                                 data |= GENMASK(31, 16);
484                                 break;
485                         case 3:
486                                 data = *buf;
487                                 buf += 3;
488                                 data |= GENMASK(31, 24);
489                                 break;
490                         }
491                         writel(data, &regs->txd0r);
492                         size = 0;
493                 }
494         }
495
496         ret = wait_for_bit_le32(&regs->isr, GQSPI_IXR_TXFIFOEMPTY_MASK, 1,
497                                 GQSPI_TIMEOUT, 1);
498         if (ret) {
499                 printf("%s: Timeout\n", __func__);
500                 return ret;
501         }
502
503         priv->tx_buf += len;
504         return 0;
505 }
506
507 static void zynqmp_qspi_genfifo_cmd(struct zynqmp_qspi_priv *priv)
508 {
509         const struct spi_mem_op *op = priv->op;
510         u32 gen_fifo_cmd;
511         u8 i, dummy_cycles, addr;
512
513         /* Send opcode */
514         gen_fifo_cmd = zynqmp_qspi_bus_select(priv);
515         gen_fifo_cmd |= zynqmp_qspi_genfifo_mode(op->cmd.buswidth);
516         gen_fifo_cmd |= GQSPI_GFIFO_TX;
517         gen_fifo_cmd |= op->cmd.opcode;
518         zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
519
520         /* Send address */
521         for (i = 0; i < op->addr.nbytes; i++) {
522                 addr = op->addr.val >> (8 * (op->addr.nbytes - i - 1));
523
524                 gen_fifo_cmd = zynqmp_qspi_bus_select(priv);
525                 gen_fifo_cmd |= zynqmp_qspi_genfifo_mode(op->addr.buswidth);
526                 gen_fifo_cmd |= GQSPI_GFIFO_TX;
527                 gen_fifo_cmd |= addr;
528
529                 debug("GFIFO_CMD_Cmd = 0x%x\n", gen_fifo_cmd);
530
531                 zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
532         }
533
534         /* Send dummy */
535         if (op->dummy.nbytes) {
536                 dummy_cycles = op->dummy.nbytes * 8 / op->dummy.buswidth;
537
538                 gen_fifo_cmd = zynqmp_qspi_bus_select(priv);
539                 gen_fifo_cmd |= zynqmp_qspi_genfifo_mode(op->dummy.buswidth);
540                 gen_fifo_cmd &= ~(GQSPI_GFIFO_TX | GQSPI_GFIFO_RX);
541                 gen_fifo_cmd |= GQSPI_GFIFO_DATA_XFR_MASK;
542                 gen_fifo_cmd |= dummy_cycles;
543                 zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
544         }
545 }
546
547 static u32 zynqmp_qspi_calc_exp(struct zynqmp_qspi_priv *priv,
548                                 u32 *gen_fifo_cmd)
549 {
550         u32 expval = 8;
551         u32 len;
552
553         while (1) {
554                 if (priv->len > 255) {
555                         if (priv->len & (1 << expval)) {
556                                 *gen_fifo_cmd &= ~GQSPI_GFIFO_IMD_MASK;
557                                 *gen_fifo_cmd |= GQSPI_GFIFO_EXP_MASK;
558                                 *gen_fifo_cmd |= expval;
559                                 priv->len -= (1 << expval);
560                                 return expval;
561                         }
562                         expval++;
563                 } else {
564                         *gen_fifo_cmd &= ~(GQSPI_GFIFO_IMD_MASK |
565                                           GQSPI_GFIFO_EXP_MASK);
566                         *gen_fifo_cmd |= (u8)priv->len;
567                         len = (u8)priv->len;
568                         priv->len  = 0;
569                         return len;
570                 }
571         }
572 }
573
574 static int zynqmp_qspi_genfifo_fill_tx(struct zynqmp_qspi_priv *priv)
575 {
576         u32 gen_fifo_cmd;
577         u32 len;
578         int ret = 0;
579
580         gen_fifo_cmd = zynqmp_qspi_bus_select(priv);
581         gen_fifo_cmd |= zynqmp_qspi_genfifo_mode(priv->op->data.buswidth);
582         gen_fifo_cmd |= GQSPI_GFIFO_TX | GQSPI_GFIFO_DATA_XFR_MASK;
583
584         while (priv->len) {
585                 len = zynqmp_qspi_calc_exp(priv, &gen_fifo_cmd);
586                 zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
587
588                 debug("GFIFO_CMD_TX:0x%x\n", gen_fifo_cmd);
589
590                 if (gen_fifo_cmd & GQSPI_GFIFO_EXP_MASK)
591                         ret = zynqmp_qspi_fill_tx_fifo(priv, 1 << len);
592                 else
593                         ret = zynqmp_qspi_fill_tx_fifo(priv, len);
594
595                 if (ret)
596                         return ret;
597         }
598         return ret;
599 }
600
601 static int zynqmp_qspi_start_io(struct zynqmp_qspi_priv *priv,
602                                 u32 gen_fifo_cmd, u32 *buf)
603 {
604         u32 len;
605         u32 actuallen = priv->len;
606         u32 config_reg, ier, isr;
607         u32 timeout = GQSPI_TIMEOUT;
608         struct zynqmp_qspi_regs *regs = priv->regs;
609         u32 last_bits;
610         u32 *traverse = buf;
611
612         while (priv->len) {
613                 len = zynqmp_qspi_calc_exp(priv, &gen_fifo_cmd);
614                 /* If exponent bit is set, reset immediate to be 2^len */
615                 if (gen_fifo_cmd & GQSPI_GFIFO_EXP_MASK)
616                         priv->bytes_to_receive = (1 << len);
617                 else
618                         priv->bytes_to_receive = len;
619                 zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
620                 debug("GFIFO_CMD_RX:0x%x\n", gen_fifo_cmd);
621                 /* Manual start */
622                 config_reg = readl(&regs->confr);
623                 config_reg |= GQSPI_STRT_GEN_FIFO;
624                 writel(config_reg, &regs->confr);
625                 /* Enable RX interrupts for IO mode */
626                 ier = readl(&regs->ier);
627                 ier |= GQSPI_IXR_ALL_MASK;
628                 writel(ier, &regs->ier);
629                 while (priv->bytes_to_receive && timeout) {
630                         isr = readl(&regs->isr);
631                         if (isr & GQSPI_IXR_RXNEMTY_MASK) {
632                                 if (priv->bytes_to_receive >= 4) {
633                                         *traverse = readl(&regs->drxr);
634                                         traverse++;
635                                         priv->bytes_to_receive -= 4;
636                                 } else {
637                                         last_bits = readl(&regs->drxr);
638                                         memcpy(traverse, &last_bits,
639                                                priv->bytes_to_receive);
640                                         priv->bytes_to_receive = 0;
641                                 }
642                                 timeout = GQSPI_TIMEOUT;
643                         } else {
644                                 udelay(1);
645                                 timeout--;
646                         }
647                 }
648
649                 debug("buf:0x%lx, rxbuf:0x%lx, *buf:0x%x len: 0x%x\n",
650                       (unsigned long)buf, (unsigned long)priv->rx_buf,
651                       *buf, actuallen);
652                 if (!timeout) {
653                         printf("IO timeout: %d\n", readl(&regs->isr));
654                         return -1;
655                 }
656         }
657
658         return 0;
659 }
660
661 static int zynqmp_qspi_start_dma(struct zynqmp_qspi_priv *priv,
662                                  u32 gen_fifo_cmd, u32 *buf)
663 {
664         u32 addr;
665         u32 size;
666         u32 actuallen = priv->len;
667         u32 totallen = priv->len;
668         int ret = 0;
669         struct zynqmp_qspi_dma_regs *dma_regs = priv->dma_regs;
670
671         while (totallen) {
672                 if (totallen >= SZ_512M)
673                         priv->len = SZ_256M;
674                 else
675                         priv->len = totallen;
676
677                 totallen -= priv->len; /* Save remaining bytes length to read */
678                 actuallen = priv->len; /* Actual number of bytes reading */
679
680                 writel((unsigned long)buf, &dma_regs->dmadst);
681                 writel(roundup(priv->len, GQSPI_DMA_ALIGN), &dma_regs->dmasize);
682                 writel(GQSPI_DMA_DST_I_STS_MASK, &dma_regs->dmaier);
683                 addr = (unsigned long)buf;
684                 size = roundup(priv->len, GQSPI_DMA_ALIGN);
685                 flush_dcache_range(addr, addr + size);
686
687                 while (priv->len) {
688                         zynqmp_qspi_calc_exp(priv, &gen_fifo_cmd);
689                         zynqmp_qspi_fill_gen_fifo(priv, gen_fifo_cmd);
690
691                         debug("GFIFO_CMD_RX:0x%x\n", gen_fifo_cmd);
692                 }
693
694                 ret = wait_for_bit_le32(&dma_regs->dmaisr,
695                                         GQSPI_DMA_DST_I_STS_DONE, 1,
696                                         GQSPI_TIMEOUT, 1);
697                 if (ret) {
698                         printf("DMA Timeout:0x%x\n", readl(&dma_regs->dmaisr));
699                         return -ETIMEDOUT;
700                 }
701
702                 writel(GQSPI_DMA_DST_I_STS_DONE, &dma_regs->dmaisr);
703
704                 debug("buf:0x%lx, rxbuf:0x%lx, *buf:0x%x len: 0x%x\n",
705                       (unsigned long)buf, (unsigned long)priv->rx_buf, *buf,
706                       actuallen);
707
708                 if (buf != priv->rx_buf)
709                         memcpy(priv->rx_buf, buf, actuallen);
710
711                 buf = (u32 *)((u8 *)buf + actuallen);
712                 priv->rx_buf = (u8 *)priv->rx_buf + actuallen;
713         }
714
715         return 0;
716 }
717
718 static int zynqmp_qspi_genfifo_fill_rx(struct zynqmp_qspi_priv *priv)
719 {
720         u32 gen_fifo_cmd;
721         u32 *buf;
722         u32 actuallen = priv->len;
723
724         gen_fifo_cmd = zynqmp_qspi_bus_select(priv);
725         gen_fifo_cmd |= zynqmp_qspi_genfifo_mode(priv->op->data.buswidth);
726         gen_fifo_cmd |= GQSPI_GFIFO_RX | GQSPI_GFIFO_DATA_XFR_MASK;
727
728         /*
729          * Check if receive buffer is aligned to 4 byte and length
730          * is multiples of four byte as we are using dma to receive.
731          */
732         if ((!((unsigned long)priv->rx_buf & (GQSPI_DMA_ALIGN - 1)) &&
733              !(actuallen % GQSPI_DMA_ALIGN)) || priv->io_mode) {
734                 buf = (u32 *)priv->rx_buf;
735                 if (priv->io_mode)
736                         return zynqmp_qspi_start_io(priv, gen_fifo_cmd, buf);
737                 else
738                         return zynqmp_qspi_start_dma(priv, gen_fifo_cmd, buf);
739         }
740
741         ALLOC_CACHE_ALIGN_BUFFER(u8, tmp, roundup(priv->len, GQSPI_DMA_ALIGN));
742         buf = (u32 *)tmp;
743         return zynqmp_qspi_start_dma(priv, gen_fifo_cmd, buf);
744 }
745
746 static int zynqmp_qspi_claim_bus(struct udevice *dev)
747 {
748         struct udevice *bus = dev->parent;
749         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
750         struct zynqmp_qspi_regs *regs = priv->regs;
751
752         writel(GQSPI_ENABLE_ENABLE_MASK, &regs->enbr);
753
754         return 0;
755 }
756
757 static int zynqmp_qspi_release_bus(struct udevice *dev)
758 {
759         struct udevice *bus = dev->parent;
760         struct zynqmp_qspi_priv *priv = dev_get_priv(bus);
761         struct zynqmp_qspi_regs *regs = priv->regs;
762
763         writel(~GQSPI_ENABLE_ENABLE_MASK, &regs->enbr);
764
765         return 0;
766 }
767
768 static int zynqmp_qspi_exec_op(struct spi_slave *slave,
769                                const struct spi_mem_op *op)
770 {
771         struct zynqmp_qspi_priv *priv = dev_get_priv(slave->dev->parent);
772         int ret = 0;
773
774         priv->op = op;
775         priv->tx_buf = op->data.buf.out;
776         priv->rx_buf = op->data.buf.in;
777         priv->len = op->data.nbytes;
778
779         zynqmp_qspi_chipselect(priv, 1);
780
781         /* Send opcode, addr, dummy */
782         zynqmp_qspi_genfifo_cmd(priv);
783
784         /* Request the transfer */
785         if (op->data.dir == SPI_MEM_DATA_IN)
786                 ret = zynqmp_qspi_genfifo_fill_rx(priv);
787         else if (op->data.dir == SPI_MEM_DATA_OUT)
788                 ret = zynqmp_qspi_genfifo_fill_tx(priv);
789
790         zynqmp_qspi_chipselect(priv, 0);
791
792         return ret;
793 }
794
795 static const struct spi_controller_mem_ops zynqmp_qspi_mem_ops = {
796         .exec_op = zynqmp_qspi_exec_op,
797 };
798
799 static const struct dm_spi_ops zynqmp_qspi_ops = {
800         .claim_bus      = zynqmp_qspi_claim_bus,
801         .release_bus    = zynqmp_qspi_release_bus,
802         .set_speed      = zynqmp_qspi_set_speed,
803         .set_mode       = zynqmp_qspi_set_mode,
804         .mem_ops        = &zynqmp_qspi_mem_ops,
805 };
806
807 static const struct udevice_id zynqmp_qspi_ids[] = {
808         { .compatible = "xlnx,zynqmp-qspi-1.0" },
809         { .compatible = "xlnx,versal-qspi-1.0" },
810         { }
811 };
812
813 U_BOOT_DRIVER(zynqmp_qspi) = {
814         .name   = "zynqmp_qspi",
815         .id     = UCLASS_SPI,
816         .of_match = zynqmp_qspi_ids,
817         .ops    = &zynqmp_qspi_ops,
818         .of_to_plat = zynqmp_qspi_of_to_plat,
819         .plat_auto      = sizeof(struct zynqmp_qspi_plat),
820         .priv_auto      = sizeof(struct zynqmp_qspi_priv),
821         .probe  = zynqmp_qspi_probe,
822 };