xref: /openbmc/u-boot/drivers/spi/fsl_qspi.c (revision 8e51c0f2)
1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright 2013-2015 Freescale Semiconductor, Inc.
4  *
5  * Freescale Quad Serial Peripheral Interface (QSPI) driver
6  */
7 
8 #include <common.h>
9 #include <malloc.h>
10 #include <spi.h>
11 #include <asm/io.h>
12 #include <linux/sizes.h>
13 #include <dm.h>
14 #include <errno.h>
15 #include <watchdog.h>
16 #include <wait_bit.h>
17 #include "fsl_qspi.h"
18 
19 DECLARE_GLOBAL_DATA_PTR;
20 
21 #define RX_BUFFER_SIZE		0x80
22 #if defined(CONFIG_MX6SX) || defined(CONFIG_MX6UL) || \
23 	defined(CONFIG_MX6ULL) || defined(CONFIG_MX7D)
24 #define TX_BUFFER_SIZE		0x200
25 #else
26 #define TX_BUFFER_SIZE		0x40
27 #endif
28 
29 #define OFFSET_BITS_MASK	GENMASK(23, 0)
30 
31 #define FLASH_STATUS_WEL	0x02
32 
33 /* SEQID */
34 #define SEQID_WREN		1
35 #define SEQID_FAST_READ		2
36 #define SEQID_RDSR		3
37 #define SEQID_SE		4
38 #define SEQID_CHIP_ERASE	5
39 #define SEQID_PP		6
40 #define SEQID_RDID		7
41 #define SEQID_BE_4K		8
42 #ifdef CONFIG_SPI_FLASH_BAR
43 #define SEQID_BRRD		9
44 #define SEQID_BRWR		10
45 #define SEQID_RDEAR		11
46 #define SEQID_WREAR		12
47 #endif
48 #define SEQID_WRAR		13
49 #define SEQID_RDAR		14
50 
51 /* QSPI CMD */
52 #define QSPI_CMD_PP		0x02	/* Page program (up to 256 bytes) */
53 #define QSPI_CMD_RDSR		0x05	/* Read status register */
54 #define QSPI_CMD_WREN		0x06	/* Write enable */
55 #define QSPI_CMD_FAST_READ	0x0b	/* Read data bytes (high frequency) */
56 #define QSPI_CMD_BE_4K		0x20    /* 4K erase */
57 #define QSPI_CMD_CHIP_ERASE	0xc7	/* Erase whole flash chip */
58 #define QSPI_CMD_SE		0xd8	/* Sector erase (usually 64KiB) */
59 #define QSPI_CMD_RDID		0x9f	/* Read JEDEC ID */
60 
61 /* Used for Micron, winbond and Macronix flashes */
62 #define	QSPI_CMD_WREAR		0xc5	/* EAR register write */
63 #define	QSPI_CMD_RDEAR		0xc8	/* EAR reigster read */
64 
65 /* Used for Spansion flashes only. */
66 #define	QSPI_CMD_BRRD		0x16	/* Bank register read */
67 #define	QSPI_CMD_BRWR		0x17	/* Bank register write */
68 
69 /* Used for Spansion S25FS-S family flash only. */
70 #define QSPI_CMD_RDAR		0x65	/* Read any device register */
71 #define QSPI_CMD_WRAR		0x71	/* Write any device register */
72 
73 /* 4-byte address QSPI CMD - used on Spansion and some Macronix flashes */
74 #define QSPI_CMD_FAST_READ_4B	0x0c    /* Read data bytes (high frequency) */
75 #define QSPI_CMD_PP_4B		0x12    /* Page program (up to 256 bytes) */
76 #define QSPI_CMD_SE_4B		0xdc    /* Sector erase (usually 64KiB) */
77 
78 /* fsl_qspi_platdata flags */
79 #define QSPI_FLAG_REGMAP_ENDIAN_BIG	BIT(0)
80 
81 /* default SCK frequency, unit: HZ */
82 #define FSL_QSPI_DEFAULT_SCK_FREQ	50000000
83 
84 /* QSPI max chipselect signals number */
85 #define FSL_QSPI_MAX_CHIPSELECT_NUM     4
86 
87 #ifdef CONFIG_DM_SPI
88 /**
89  * struct fsl_qspi_platdata - platform data for Freescale QSPI
90  *
91  * @flags: Flags for QSPI QSPI_FLAG_...
92  * @speed_hz: Default SCK frequency
93  * @reg_base: Base address of QSPI registers
94  * @amba_base: Base address of QSPI memory mapping
95  * @amba_total_size: size of QSPI memory mapping
96  * @flash_num: Number of active slave devices
97  * @num_chipselect: Number of QSPI chipselect signals
98  */
99 struct fsl_qspi_platdata {
100 	u32 flags;
101 	u32 speed_hz;
102 	fdt_addr_t reg_base;
103 	fdt_addr_t amba_base;
104 	fdt_size_t amba_total_size;
105 	u32 flash_num;
106 	u32 num_chipselect;
107 };
108 #endif
109 
110 /**
111  * struct fsl_qspi_priv - private data for Freescale QSPI
112  *
113  * @flags: Flags for QSPI QSPI_FLAG_...
114  * @bus_clk: QSPI input clk frequency
115  * @speed_hz: Default SCK frequency
116  * @cur_seqid: current LUT table sequence id
117  * @sf_addr: flash access offset
118  * @amba_base: Base address of QSPI memory mapping of every CS
119  * @amba_total_size: size of QSPI memory mapping
120  * @cur_amba_base: Base address of QSPI memory mapping of current CS
121  * @flash_num: Number of active slave devices
122  * @num_chipselect: Number of QSPI chipselect signals
123  * @regs: Point to QSPI register structure for I/O access
124  */
125 struct fsl_qspi_priv {
126 	u32 flags;
127 	u32 bus_clk;
128 	u32 speed_hz;
129 	u32 cur_seqid;
130 	u32 sf_addr;
131 	u32 amba_base[FSL_QSPI_MAX_CHIPSELECT_NUM];
132 	u32 amba_total_size;
133 	u32 cur_amba_base;
134 	u32 flash_num;
135 	u32 num_chipselect;
136 	struct fsl_qspi_regs *regs;
137 };
138 
139 #ifndef CONFIG_DM_SPI
140 struct fsl_qspi {
141 	struct spi_slave slave;
142 	struct fsl_qspi_priv priv;
143 };
144 #endif
145 
146 static u32 qspi_read32(u32 flags, u32 *addr)
147 {
148 	return flags & QSPI_FLAG_REGMAP_ENDIAN_BIG ?
149 		in_be32(addr) : in_le32(addr);
150 }
151 
152 static void qspi_write32(u32 flags, u32 *addr, u32 val)
153 {
154 	flags & QSPI_FLAG_REGMAP_ENDIAN_BIG ?
155 		out_be32(addr, val) : out_le32(addr, val);
156 }
157 
158 static inline int is_controller_busy(const struct fsl_qspi_priv *priv)
159 {
160 	u32 val;
161 	const u32 mask = QSPI_SR_BUSY_MASK | QSPI_SR_AHB_ACC_MASK |
162 			 QSPI_SR_IP_ACC_MASK;
163 	unsigned int retry = 5;
164 
165 	do {
166 		val = qspi_read32(priv->flags, &priv->regs->sr);
167 
168 		if ((~val & mask) == mask)
169 			return 0;
170 
171 		udelay(1);
172 	} while (--retry);
173 
174 	return -ETIMEDOUT;
175 }
176 
177 /* QSPI support swapping the flash read/write data
178  * in hardware for LS102xA, but not for VF610 */
179 static inline u32 qspi_endian_xchg(u32 data)
180 {
181 #ifdef CONFIG_VF610
182 	return swab32(data);
183 #else
184 	return data;
185 #endif
186 }
187 
188 static void qspi_set_lut(struct fsl_qspi_priv *priv)
189 {
190 	struct fsl_qspi_regs *regs = priv->regs;
191 	u32 lut_base;
192 
193 	/* Unlock the LUT */
194 	qspi_write32(priv->flags, &regs->lutkey, LUT_KEY_VALUE);
195 	qspi_write32(priv->flags, &regs->lckcr, QSPI_LCKCR_UNLOCK);
196 
197 	/* Write Enable */
198 	lut_base = SEQID_WREN * 4;
199 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_WREN) |
200 		PAD0(LUT_PAD1) | INSTR0(LUT_CMD));
201 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], 0);
202 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
203 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
204 
205 	/* Fast Read */
206 	lut_base = SEQID_FAST_READ * 4;
207 #ifdef CONFIG_SPI_FLASH_BAR
208 	qspi_write32(priv->flags, &regs->lut[lut_base],
209 		     OPRND0(QSPI_CMD_FAST_READ) | PAD0(LUT_PAD1) |
210 		     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
211 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
212 #else
213 	if (FSL_QSPI_FLASH_SIZE  <= SZ_16M)
214 		qspi_write32(priv->flags, &regs->lut[lut_base],
215 			     OPRND0(QSPI_CMD_FAST_READ) | PAD0(LUT_PAD1) |
216 			     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
217 			     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
218 	else
219 		qspi_write32(priv->flags, &regs->lut[lut_base],
220 			     OPRND0(QSPI_CMD_FAST_READ_4B) |
221 			     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) |
222 			     OPRND1(ADDR32BIT) | PAD1(LUT_PAD1) |
223 			     INSTR1(LUT_ADDR));
224 #endif
225 	qspi_write32(priv->flags, &regs->lut[lut_base + 1],
226 		     OPRND0(8) | PAD0(LUT_PAD1) | INSTR0(LUT_DUMMY) |
227 		     OPRND1(RX_BUFFER_SIZE) | PAD1(LUT_PAD1) |
228 		     INSTR1(LUT_READ));
229 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
230 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
231 
232 	/* Read Status */
233 	lut_base = SEQID_RDSR * 4;
234 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_RDSR) |
235 		PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(1) |
236 		PAD1(LUT_PAD1) | INSTR1(LUT_READ));
237 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], 0);
238 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
239 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
240 
241 	/* Erase a sector */
242 	lut_base = SEQID_SE * 4;
243 #ifdef CONFIG_SPI_FLASH_BAR
244 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_SE) |
245 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
246 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
247 #else
248 	if (FSL_QSPI_FLASH_SIZE  <= SZ_16M)
249 		qspi_write32(priv->flags, &regs->lut[lut_base],
250 			     OPRND0(QSPI_CMD_SE) | PAD0(LUT_PAD1) |
251 			     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
252 			     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
253 	else
254 		qspi_write32(priv->flags, &regs->lut[lut_base],
255 			     OPRND0(QSPI_CMD_SE_4B) | PAD0(LUT_PAD1) |
256 			     INSTR0(LUT_CMD) | OPRND1(ADDR32BIT) |
257 			     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
258 #endif
259 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], 0);
260 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
261 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
262 
263 	/* Erase the whole chip */
264 	lut_base = SEQID_CHIP_ERASE * 4;
265 	qspi_write32(priv->flags, &regs->lut[lut_base],
266 		     OPRND0(QSPI_CMD_CHIP_ERASE) |
267 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD));
268 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], 0);
269 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
270 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
271 
272 	/* Page Program */
273 	lut_base = SEQID_PP * 4;
274 #ifdef CONFIG_SPI_FLASH_BAR
275 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_PP) |
276 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
277 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
278 #else
279 	if (FSL_QSPI_FLASH_SIZE  <= SZ_16M)
280 		qspi_write32(priv->flags, &regs->lut[lut_base],
281 			     OPRND0(QSPI_CMD_PP) | PAD0(LUT_PAD1) |
282 			     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
283 			     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
284 	else
285 		qspi_write32(priv->flags, &regs->lut[lut_base],
286 			     OPRND0(QSPI_CMD_PP_4B) | PAD0(LUT_PAD1) |
287 			     INSTR0(LUT_CMD) | OPRND1(ADDR32BIT) |
288 			     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
289 #endif
290 #if defined(CONFIG_MX6SX) || defined(CONFIG_MX6UL) || \
291 	defined(CONFIG_MX6ULL) || defined(CONFIG_MX7D)
292 	/*
293 	 * To MX6SX, OPRND0(TX_BUFFER_SIZE) can not work correctly.
294 	 * So, Use IDATSZ in IPCR to determine the size and here set 0.
295 	 */
296 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], OPRND0(0) |
297 		     PAD0(LUT_PAD1) | INSTR0(LUT_WRITE));
298 #else
299 	qspi_write32(priv->flags, &regs->lut[lut_base + 1],
300 		     OPRND0(TX_BUFFER_SIZE) |
301 		     PAD0(LUT_PAD1) | INSTR0(LUT_WRITE));
302 #endif
303 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
304 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
305 
306 	/* READ ID */
307 	lut_base = SEQID_RDID * 4;
308 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_RDID) |
309 		PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(8) |
310 		PAD1(LUT_PAD1) | INSTR1(LUT_READ));
311 	qspi_write32(priv->flags, &regs->lut[lut_base + 1], 0);
312 	qspi_write32(priv->flags, &regs->lut[lut_base + 2], 0);
313 	qspi_write32(priv->flags, &regs->lut[lut_base + 3], 0);
314 
315 	/* SUB SECTOR 4K ERASE */
316 	lut_base = SEQID_BE_4K * 4;
317 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_BE_4K) |
318 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
319 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
320 
321 #ifdef CONFIG_SPI_FLASH_BAR
322 	/*
323 	 * BRRD BRWR RDEAR WREAR are all supported, because it is hard to
324 	 * dynamically check whether to set BRRD BRWR or RDEAR WREAR during
325 	 * initialization.
326 	 */
327 	lut_base = SEQID_BRRD * 4;
328 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_BRRD) |
329 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(1) |
330 		     PAD1(LUT_PAD1) | INSTR1(LUT_READ));
331 
332 	lut_base = SEQID_BRWR * 4;
333 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_BRWR) |
334 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(1) |
335 		     PAD1(LUT_PAD1) | INSTR1(LUT_WRITE));
336 
337 	lut_base = SEQID_RDEAR * 4;
338 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_RDEAR) |
339 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(1) |
340 		     PAD1(LUT_PAD1) | INSTR1(LUT_READ));
341 
342 	lut_base = SEQID_WREAR * 4;
343 	qspi_write32(priv->flags, &regs->lut[lut_base], OPRND0(QSPI_CMD_WREAR) |
344 		     PAD0(LUT_PAD1) | INSTR0(LUT_CMD) | OPRND1(1) |
345 		     PAD1(LUT_PAD1) | INSTR1(LUT_WRITE));
346 #endif
347 
348 	/*
349 	 * Read any device register.
350 	 * Used for Spansion S25FS-S family flash only.
351 	 */
352 	lut_base = SEQID_RDAR * 4;
353 	qspi_write32(priv->flags, &regs->lut[lut_base],
354 		     OPRND0(QSPI_CMD_RDAR) | PAD0(LUT_PAD1) |
355 		     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
356 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
357 	qspi_write32(priv->flags, &regs->lut[lut_base + 1],
358 		     OPRND0(8) | PAD0(LUT_PAD1) | INSTR0(LUT_DUMMY) |
359 		     OPRND1(1) | PAD1(LUT_PAD1) |
360 		     INSTR1(LUT_READ));
361 
362 	/*
363 	 * Write any device register.
364 	 * Used for Spansion S25FS-S family flash only.
365 	 */
366 	lut_base = SEQID_WRAR * 4;
367 	qspi_write32(priv->flags, &regs->lut[lut_base],
368 		     OPRND0(QSPI_CMD_WRAR) | PAD0(LUT_PAD1) |
369 		     INSTR0(LUT_CMD) | OPRND1(ADDR24BIT) |
370 		     PAD1(LUT_PAD1) | INSTR1(LUT_ADDR));
371 	qspi_write32(priv->flags, &regs->lut[lut_base + 1],
372 		     OPRND0(1) | PAD0(LUT_PAD1) | INSTR0(LUT_WRITE));
373 
374 	/* Lock the LUT */
375 	qspi_write32(priv->flags, &regs->lutkey, LUT_KEY_VALUE);
376 	qspi_write32(priv->flags, &regs->lckcr, QSPI_LCKCR_LOCK);
377 }
378 
379 #if defined(CONFIG_SYS_FSL_QSPI_AHB)
380 /*
381  * If we have changed the content of the flash by writing or erasing,
382  * we need to invalidate the AHB buffer. If we do not do so, we may read out
383  * the wrong data. The spec tells us reset the AHB domain and Serial Flash
384  * domain at the same time.
385  */
386 static inline void qspi_ahb_invalid(struct fsl_qspi_priv *priv)
387 {
388 	struct fsl_qspi_regs *regs = priv->regs;
389 	u32 reg;
390 
391 	reg = qspi_read32(priv->flags, &regs->mcr);
392 	reg |= QSPI_MCR_SWRSTHD_MASK | QSPI_MCR_SWRSTSD_MASK;
393 	qspi_write32(priv->flags, &regs->mcr, reg);
394 
395 	/*
396 	 * The minimum delay : 1 AHB + 2 SFCK clocks.
397 	 * Delay 1 us is enough.
398 	 */
399 	udelay(1);
400 
401 	reg &= ~(QSPI_MCR_SWRSTHD_MASK | QSPI_MCR_SWRSTSD_MASK);
402 	qspi_write32(priv->flags, &regs->mcr, reg);
403 }
404 
405 /* Read out the data from the AHB buffer. */
406 static inline void qspi_ahb_read(struct fsl_qspi_priv *priv, u8 *rxbuf, int len)
407 {
408 	struct fsl_qspi_regs *regs = priv->regs;
409 	u32 mcr_reg;
410 	void *rx_addr;
411 
412 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
413 
414 	qspi_write32(priv->flags, &regs->mcr,
415 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
416 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
417 
418 	rx_addr = (void *)(uintptr_t)(priv->cur_amba_base + priv->sf_addr);
419 	/* Read out the data directly from the AHB buffer. */
420 	memcpy(rxbuf, rx_addr, len);
421 
422 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
423 }
424 
425 static void qspi_enable_ddr_mode(struct fsl_qspi_priv *priv)
426 {
427 	u32 reg, reg2;
428 	struct fsl_qspi_regs *regs = priv->regs;
429 
430 	reg = qspi_read32(priv->flags, &regs->mcr);
431 	/* Disable the module */
432 	qspi_write32(priv->flags, &regs->mcr, reg | QSPI_MCR_MDIS_MASK);
433 
434 	/* Set the Sampling Register for DDR */
435 	reg2 = qspi_read32(priv->flags, &regs->smpr);
436 	reg2 &= ~QSPI_SMPR_DDRSMP_MASK;
437 	reg2 |= (2 << QSPI_SMPR_DDRSMP_SHIFT);
438 	qspi_write32(priv->flags, &regs->smpr, reg2);
439 
440 	/* Enable the module again (enable the DDR too) */
441 	reg |= QSPI_MCR_DDR_EN_MASK;
442 	/* Enable bit 29 for imx6sx */
443 	reg |= BIT(29);
444 
445 	qspi_write32(priv->flags, &regs->mcr, reg);
446 }
447 
448 /*
449  * There are two different ways to read out the data from the flash:
450  *  the "IP Command Read" and the "AHB Command Read".
451  *
452  * The IC guy suggests we use the "AHB Command Read" which is faster
453  * then the "IP Command Read". (What's more is that there is a bug in
454  * the "IP Command Read" in the Vybrid.)
455  *
456  * After we set up the registers for the "AHB Command Read", we can use
457  * the memcpy to read the data directly. A "missed" access to the buffer
458  * causes the controller to clear the buffer, and use the sequence pointed
459  * by the QUADSPI_BFGENCR[SEQID] to initiate a read from the flash.
460  */
461 static void qspi_init_ahb_read(struct fsl_qspi_priv *priv)
462 {
463 	struct fsl_qspi_regs *regs = priv->regs;
464 
465 	/* AHB configuration for access buffer 0/1/2 .*/
466 	qspi_write32(priv->flags, &regs->buf0cr, QSPI_BUFXCR_INVALID_MSTRID);
467 	qspi_write32(priv->flags, &regs->buf1cr, QSPI_BUFXCR_INVALID_MSTRID);
468 	qspi_write32(priv->flags, &regs->buf2cr, QSPI_BUFXCR_INVALID_MSTRID);
469 	qspi_write32(priv->flags, &regs->buf3cr, QSPI_BUF3CR_ALLMST_MASK |
470 		     (0x80 << QSPI_BUF3CR_ADATSZ_SHIFT));
471 
472 	/* We only use the buffer3 */
473 	qspi_write32(priv->flags, &regs->buf0ind, 0);
474 	qspi_write32(priv->flags, &regs->buf1ind, 0);
475 	qspi_write32(priv->flags, &regs->buf2ind, 0);
476 
477 	/*
478 	 * Set the default lut sequence for AHB Read.
479 	 * Parallel mode is disabled.
480 	 */
481 	qspi_write32(priv->flags, &regs->bfgencr,
482 		     SEQID_FAST_READ << QSPI_BFGENCR_SEQID_SHIFT);
483 
484 	/*Enable DDR Mode*/
485 	qspi_enable_ddr_mode(priv);
486 }
487 #endif
488 
489 #ifdef CONFIG_SPI_FLASH_BAR
490 /* Bank register read/write, EAR register read/write */
491 static void qspi_op_rdbank(struct fsl_qspi_priv *priv, u8 *rxbuf, u32 len)
492 {
493 	struct fsl_qspi_regs *regs = priv->regs;
494 	u32 reg, mcr_reg, data, seqid;
495 
496 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
497 	qspi_write32(priv->flags, &regs->mcr,
498 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
499 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
500 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
501 
502 	qspi_write32(priv->flags, &regs->sfar, priv->cur_amba_base);
503 
504 	if (priv->cur_seqid == QSPI_CMD_BRRD)
505 		seqid = SEQID_BRRD;
506 	else
507 		seqid = SEQID_RDEAR;
508 
509 	qspi_write32(priv->flags, &regs->ipcr,
510 		     (seqid << QSPI_IPCR_SEQID_SHIFT) | len);
511 
512 	/* Wait previous command complete */
513 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
514 		;
515 
516 	while (1) {
517 		WATCHDOG_RESET();
518 
519 		reg = qspi_read32(priv->flags, &regs->rbsr);
520 		if (reg & QSPI_RBSR_RDBFL_MASK) {
521 			data = qspi_read32(priv->flags, &regs->rbdr[0]);
522 			data = qspi_endian_xchg(data);
523 			memcpy(rxbuf, &data, len);
524 			qspi_write32(priv->flags, &regs->mcr,
525 				     qspi_read32(priv->flags, &regs->mcr) |
526 				     QSPI_MCR_CLR_RXF_MASK);
527 			break;
528 		}
529 	}
530 
531 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
532 }
533 #endif
534 
535 static void qspi_op_rdid(struct fsl_qspi_priv *priv, u32 *rxbuf, u32 len)
536 {
537 	struct fsl_qspi_regs *regs = priv->regs;
538 	u32 mcr_reg, rbsr_reg, data, size;
539 	int i;
540 
541 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
542 	qspi_write32(priv->flags, &regs->mcr,
543 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
544 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
545 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
546 
547 	qspi_write32(priv->flags, &regs->sfar, priv->cur_amba_base);
548 
549 	qspi_write32(priv->flags, &regs->ipcr,
550 		     (SEQID_RDID << QSPI_IPCR_SEQID_SHIFT) | 0);
551 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
552 		;
553 
554 	i = 0;
555 	while ((RX_BUFFER_SIZE >= len) && (len > 0)) {
556 		WATCHDOG_RESET();
557 
558 		rbsr_reg = qspi_read32(priv->flags, &regs->rbsr);
559 		if (rbsr_reg & QSPI_RBSR_RDBFL_MASK) {
560 			data = qspi_read32(priv->flags, &regs->rbdr[i]);
561 			data = qspi_endian_xchg(data);
562 			size = (len < 4) ? len : 4;
563 			memcpy(rxbuf, &data, size);
564 			len -= size;
565 			rxbuf++;
566 			i++;
567 		}
568 	}
569 
570 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
571 }
572 
573 /* If not use AHB read, read data from ip interface */
574 static void qspi_op_read(struct fsl_qspi_priv *priv, u32 *rxbuf, u32 len)
575 {
576 	struct fsl_qspi_regs *regs = priv->regs;
577 	u32 mcr_reg, data;
578 	int i, size;
579 	u32 to_or_from;
580 	u32 seqid;
581 
582 	if (priv->cur_seqid == QSPI_CMD_RDAR)
583 		seqid = SEQID_RDAR;
584 	else
585 		seqid = SEQID_FAST_READ;
586 
587 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
588 	qspi_write32(priv->flags, &regs->mcr,
589 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
590 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
591 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
592 
593 	to_or_from = priv->sf_addr + priv->cur_amba_base;
594 
595 	while (len > 0) {
596 		WATCHDOG_RESET();
597 
598 		qspi_write32(priv->flags, &regs->sfar, to_or_from);
599 
600 		size = (len > RX_BUFFER_SIZE) ?
601 			RX_BUFFER_SIZE : len;
602 
603 		qspi_write32(priv->flags, &regs->ipcr,
604 			     (seqid << QSPI_IPCR_SEQID_SHIFT) |
605 			     size);
606 		while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
607 			;
608 
609 		to_or_from += size;
610 		len -= size;
611 
612 		i = 0;
613 		while ((RX_BUFFER_SIZE >= size) && (size > 0)) {
614 			data = qspi_read32(priv->flags, &regs->rbdr[i]);
615 			data = qspi_endian_xchg(data);
616 			if (size < 4)
617 				memcpy(rxbuf, &data, size);
618 			else
619 				memcpy(rxbuf, &data, 4);
620 			rxbuf++;
621 			size -= 4;
622 			i++;
623 		}
624 		qspi_write32(priv->flags, &regs->mcr,
625 			     qspi_read32(priv->flags, &regs->mcr) |
626 			     QSPI_MCR_CLR_RXF_MASK);
627 	}
628 
629 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
630 }
631 
632 static void qspi_op_write(struct fsl_qspi_priv *priv, u8 *txbuf, u32 len)
633 {
634 	struct fsl_qspi_regs *regs = priv->regs;
635 	u32 mcr_reg, data, reg, status_reg, seqid;
636 	int i, size, tx_size;
637 	u32 to_or_from = 0;
638 
639 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
640 	qspi_write32(priv->flags, &regs->mcr,
641 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
642 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
643 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
644 
645 	status_reg = 0;
646 	while ((status_reg & FLASH_STATUS_WEL) != FLASH_STATUS_WEL) {
647 		WATCHDOG_RESET();
648 
649 		qspi_write32(priv->flags, &regs->ipcr,
650 			     (SEQID_WREN << QSPI_IPCR_SEQID_SHIFT) | 0);
651 		while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
652 			;
653 
654 		qspi_write32(priv->flags, &regs->ipcr,
655 			     (SEQID_RDSR << QSPI_IPCR_SEQID_SHIFT) | 1);
656 		while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
657 			;
658 
659 		reg = qspi_read32(priv->flags, &regs->rbsr);
660 		if (reg & QSPI_RBSR_RDBFL_MASK) {
661 			status_reg = qspi_read32(priv->flags, &regs->rbdr[0]);
662 			status_reg = qspi_endian_xchg(status_reg);
663 		}
664 		qspi_write32(priv->flags, &regs->mcr,
665 			     qspi_read32(priv->flags, &regs->mcr) |
666 			     QSPI_MCR_CLR_RXF_MASK);
667 	}
668 
669 	/* Default is page programming */
670 	seqid = SEQID_PP;
671 	if (priv->cur_seqid == QSPI_CMD_WRAR)
672 		seqid = SEQID_WRAR;
673 #ifdef CONFIG_SPI_FLASH_BAR
674 	if (priv->cur_seqid == QSPI_CMD_BRWR)
675 		seqid = SEQID_BRWR;
676 	else if (priv->cur_seqid == QSPI_CMD_WREAR)
677 		seqid = SEQID_WREAR;
678 #endif
679 
680 	to_or_from = priv->sf_addr + priv->cur_amba_base;
681 
682 	qspi_write32(priv->flags, &regs->sfar, to_or_from);
683 
684 	tx_size = (len > TX_BUFFER_SIZE) ?
685 		TX_BUFFER_SIZE : len;
686 
687 	size = tx_size / 16;
688 	/*
689 	 * There must be atleast 128bit data
690 	 * available in TX FIFO for any pop operation
691 	 */
692 	if (tx_size % 16)
693 		size++;
694 	for (i = 0; i < size * 4; i++) {
695 		memcpy(&data, txbuf, 4);
696 		data = qspi_endian_xchg(data);
697 		qspi_write32(priv->flags, &regs->tbdr, data);
698 		txbuf += 4;
699 	}
700 
701 	qspi_write32(priv->flags, &regs->ipcr,
702 		     (seqid << QSPI_IPCR_SEQID_SHIFT) | tx_size);
703 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
704 		;
705 
706 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
707 }
708 
709 static void qspi_op_rdsr(struct fsl_qspi_priv *priv, void *rxbuf, u32 len)
710 {
711 	struct fsl_qspi_regs *regs = priv->regs;
712 	u32 mcr_reg, reg, data;
713 
714 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
715 	qspi_write32(priv->flags, &regs->mcr,
716 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
717 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
718 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
719 
720 	qspi_write32(priv->flags, &regs->sfar, priv->cur_amba_base);
721 
722 	qspi_write32(priv->flags, &regs->ipcr,
723 		     (SEQID_RDSR << QSPI_IPCR_SEQID_SHIFT) | 0);
724 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
725 		;
726 
727 	while (1) {
728 		WATCHDOG_RESET();
729 
730 		reg = qspi_read32(priv->flags, &regs->rbsr);
731 		if (reg & QSPI_RBSR_RDBFL_MASK) {
732 			data = qspi_read32(priv->flags, &regs->rbdr[0]);
733 			data = qspi_endian_xchg(data);
734 			memcpy(rxbuf, &data, len);
735 			qspi_write32(priv->flags, &regs->mcr,
736 				     qspi_read32(priv->flags, &regs->mcr) |
737 				     QSPI_MCR_CLR_RXF_MASK);
738 			break;
739 		}
740 	}
741 
742 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
743 }
744 
745 static void qspi_op_erase(struct fsl_qspi_priv *priv)
746 {
747 	struct fsl_qspi_regs *regs = priv->regs;
748 	u32 mcr_reg;
749 	u32 to_or_from = 0;
750 
751 	mcr_reg = qspi_read32(priv->flags, &regs->mcr);
752 	qspi_write32(priv->flags, &regs->mcr,
753 		     QSPI_MCR_CLR_RXF_MASK | QSPI_MCR_CLR_TXF_MASK |
754 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_END_CFD_LE);
755 	qspi_write32(priv->flags, &regs->rbct, QSPI_RBCT_RXBRD_USEIPS);
756 
757 	to_or_from = priv->sf_addr + priv->cur_amba_base;
758 	qspi_write32(priv->flags, &regs->sfar, to_or_from);
759 
760 	qspi_write32(priv->flags, &regs->ipcr,
761 		     (SEQID_WREN << QSPI_IPCR_SEQID_SHIFT) | 0);
762 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
763 		;
764 
765 	if (priv->cur_seqid == QSPI_CMD_SE) {
766 		qspi_write32(priv->flags, &regs->ipcr,
767 			     (SEQID_SE << QSPI_IPCR_SEQID_SHIFT) | 0);
768 	} else if (priv->cur_seqid == QSPI_CMD_BE_4K) {
769 		qspi_write32(priv->flags, &regs->ipcr,
770 			     (SEQID_BE_4K << QSPI_IPCR_SEQID_SHIFT) | 0);
771 	}
772 	while (qspi_read32(priv->flags, &regs->sr) & QSPI_SR_BUSY_MASK)
773 		;
774 
775 	qspi_write32(priv->flags, &regs->mcr, mcr_reg);
776 }
777 
778 int qspi_xfer(struct fsl_qspi_priv *priv, unsigned int bitlen,
779 		const void *dout, void *din, unsigned long flags)
780 {
781 	u32 bytes = DIV_ROUND_UP(bitlen, 8);
782 	static u32 wr_sfaddr;
783 	u32 txbuf;
784 
785 	WATCHDOG_RESET();
786 
787 	if (dout) {
788 		if (flags & SPI_XFER_BEGIN) {
789 			priv->cur_seqid = *(u8 *)dout;
790 			memcpy(&txbuf, dout, 4);
791 		}
792 
793 		if (flags == SPI_XFER_END) {
794 			priv->sf_addr = wr_sfaddr;
795 			qspi_op_write(priv, (u8 *)dout, bytes);
796 			return 0;
797 		}
798 
799 		if (priv->cur_seqid == QSPI_CMD_FAST_READ ||
800 		    priv->cur_seqid == QSPI_CMD_RDAR) {
801 			priv->sf_addr = swab32(txbuf) & OFFSET_BITS_MASK;
802 		} else if ((priv->cur_seqid == QSPI_CMD_SE) ||
803 			   (priv->cur_seqid == QSPI_CMD_BE_4K)) {
804 			priv->sf_addr = swab32(txbuf) & OFFSET_BITS_MASK;
805 			qspi_op_erase(priv);
806 		} else if (priv->cur_seqid == QSPI_CMD_PP ||
807 			   priv->cur_seqid == QSPI_CMD_WRAR) {
808 			wr_sfaddr = swab32(txbuf) & OFFSET_BITS_MASK;
809 		} else if ((priv->cur_seqid == QSPI_CMD_BRWR) ||
810 			 (priv->cur_seqid == QSPI_CMD_WREAR)) {
811 #ifdef CONFIG_SPI_FLASH_BAR
812 			wr_sfaddr = 0;
813 #endif
814 		}
815 	}
816 
817 	if (din) {
818 		if (priv->cur_seqid == QSPI_CMD_FAST_READ) {
819 #ifdef CONFIG_SYS_FSL_QSPI_AHB
820 			qspi_ahb_read(priv, din, bytes);
821 #else
822 			qspi_op_read(priv, din, bytes);
823 #endif
824 		} else if (priv->cur_seqid == QSPI_CMD_RDAR) {
825 			qspi_op_read(priv, din, bytes);
826 		} else if (priv->cur_seqid == QSPI_CMD_RDID)
827 			qspi_op_rdid(priv, din, bytes);
828 		else if (priv->cur_seqid == QSPI_CMD_RDSR)
829 			qspi_op_rdsr(priv, din, bytes);
830 #ifdef CONFIG_SPI_FLASH_BAR
831 		else if ((priv->cur_seqid == QSPI_CMD_BRRD) ||
832 			 (priv->cur_seqid == QSPI_CMD_RDEAR)) {
833 			priv->sf_addr = 0;
834 			qspi_op_rdbank(priv, din, bytes);
835 		}
836 #endif
837 	}
838 
839 #ifdef CONFIG_SYS_FSL_QSPI_AHB
840 	if ((priv->cur_seqid == QSPI_CMD_SE) ||
841 	    (priv->cur_seqid == QSPI_CMD_PP) ||
842 	    (priv->cur_seqid == QSPI_CMD_BE_4K) ||
843 	    (priv->cur_seqid == QSPI_CMD_WREAR) ||
844 	    (priv->cur_seqid == QSPI_CMD_BRWR))
845 		qspi_ahb_invalid(priv);
846 #endif
847 
848 	return 0;
849 }
850 
851 void qspi_module_disable(struct fsl_qspi_priv *priv, u8 disable)
852 {
853 	u32 mcr_val;
854 
855 	mcr_val = qspi_read32(priv->flags, &priv->regs->mcr);
856 	if (disable)
857 		mcr_val |= QSPI_MCR_MDIS_MASK;
858 	else
859 		mcr_val &= ~QSPI_MCR_MDIS_MASK;
860 	qspi_write32(priv->flags, &priv->regs->mcr, mcr_val);
861 }
862 
863 void qspi_cfg_smpr(struct fsl_qspi_priv *priv, u32 clear_bits, u32 set_bits)
864 {
865 	u32 smpr_val;
866 
867 	smpr_val = qspi_read32(priv->flags, &priv->regs->smpr);
868 	smpr_val &= ~clear_bits;
869 	smpr_val |= set_bits;
870 	qspi_write32(priv->flags, &priv->regs->smpr, smpr_val);
871 }
872 #ifndef CONFIG_DM_SPI
873 static unsigned long spi_bases[] = {
874 	QSPI0_BASE_ADDR,
875 #ifdef CONFIG_MX6SX
876 	QSPI1_BASE_ADDR,
877 #endif
878 };
879 
880 static unsigned long amba_bases[] = {
881 	QSPI0_AMBA_BASE,
882 #ifdef CONFIG_MX6SX
883 	QSPI1_AMBA_BASE,
884 #endif
885 };
886 
887 static inline struct fsl_qspi *to_qspi_spi(struct spi_slave *slave)
888 {
889 	return container_of(slave, struct fsl_qspi, slave);
890 }
891 
892 struct spi_slave *spi_setup_slave(unsigned int bus, unsigned int cs,
893 		unsigned int max_hz, unsigned int mode)
894 {
895 	u32 mcr_val;
896 	struct fsl_qspi *qspi;
897 	struct fsl_qspi_regs *regs;
898 	u32 total_size;
899 
900 	if (bus >= ARRAY_SIZE(spi_bases))
901 		return NULL;
902 
903 	if (cs >= FSL_QSPI_FLASH_NUM)
904 		return NULL;
905 
906 	qspi = spi_alloc_slave(struct fsl_qspi, bus, cs);
907 	if (!qspi)
908 		return NULL;
909 
910 #ifdef CONFIG_SYS_FSL_QSPI_BE
911 	qspi->priv.flags |= QSPI_FLAG_REGMAP_ENDIAN_BIG;
912 #endif
913 
914 	regs = (struct fsl_qspi_regs *)spi_bases[bus];
915 	qspi->priv.regs = regs;
916 	/*
917 	 * According cs, use different amba_base to choose the
918 	 * corresponding flash devices.
919 	 *
920 	 * If not, only one flash device is used even if passing
921 	 * different cs using `sf probe`
922 	 */
923 	qspi->priv.cur_amba_base = amba_bases[bus] + cs * FSL_QSPI_FLASH_SIZE;
924 
925 	qspi->slave.max_write_size = TX_BUFFER_SIZE;
926 
927 	mcr_val = qspi_read32(qspi->priv.flags, &regs->mcr);
928 
929 	/* Set endianness to LE for i.mx */
930 	if (IS_ENABLED(CONFIG_MX6) || IS_ENABLED(CONFIG_MX7))
931 		mcr_val = QSPI_MCR_END_CFD_LE;
932 
933 	qspi_write32(qspi->priv.flags, &regs->mcr,
934 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_MDIS_MASK |
935 		     (mcr_val & QSPI_MCR_END_CFD_MASK));
936 
937 	qspi_cfg_smpr(&qspi->priv,
938 		      ~(QSPI_SMPR_FSDLY_MASK | QSPI_SMPR_DDRSMP_MASK |
939 		      QSPI_SMPR_FSPHS_MASK | QSPI_SMPR_HSENA_MASK), 0);
940 
941 	total_size = FSL_QSPI_FLASH_SIZE * FSL_QSPI_FLASH_NUM;
942 	/*
943 	 * Any read access to non-implemented addresses will provide
944 	 * undefined results.
945 	 *
946 	 * In case single die flash devices, TOP_ADDR_MEMA2 and
947 	 * TOP_ADDR_MEMB2 should be initialized/programmed to
948 	 * TOP_ADDR_MEMA1 and TOP_ADDR_MEMB1 respectively - in effect,
949 	 * setting the size of these devices to 0.  This would ensure
950 	 * that the complete memory map is assigned to only one flash device.
951 	 */
952 	qspi_write32(qspi->priv.flags, &regs->sfa1ad,
953 		     FSL_QSPI_FLASH_SIZE | amba_bases[bus]);
954 	qspi_write32(qspi->priv.flags, &regs->sfa2ad,
955 		     FSL_QSPI_FLASH_SIZE | amba_bases[bus]);
956 	qspi_write32(qspi->priv.flags, &regs->sfb1ad,
957 		     total_size | amba_bases[bus]);
958 	qspi_write32(qspi->priv.flags, &regs->sfb2ad,
959 		     total_size | amba_bases[bus]);
960 
961 	qspi_set_lut(&qspi->priv);
962 
963 #ifdef CONFIG_SYS_FSL_QSPI_AHB
964 	qspi_init_ahb_read(&qspi->priv);
965 #endif
966 
967 	qspi_module_disable(&qspi->priv, 0);
968 
969 	return &qspi->slave;
970 }
971 
972 void spi_free_slave(struct spi_slave *slave)
973 {
974 	struct fsl_qspi *qspi = to_qspi_spi(slave);
975 
976 	free(qspi);
977 }
978 
979 int spi_claim_bus(struct spi_slave *slave)
980 {
981 	return 0;
982 }
983 
984 void spi_release_bus(struct spi_slave *slave)
985 {
986 	/* Nothing to do */
987 }
988 
989 int spi_xfer(struct spi_slave *slave, unsigned int bitlen,
990 		const void *dout, void *din, unsigned long flags)
991 {
992 	struct fsl_qspi *qspi = to_qspi_spi(slave);
993 
994 	return qspi_xfer(&qspi->priv, bitlen, dout, din, flags);
995 }
996 
997 void spi_init(void)
998 {
999 	/* Nothing to do */
1000 }
1001 #else
1002 static int fsl_qspi_child_pre_probe(struct udevice *dev)
1003 {
1004 	struct spi_slave *slave = dev_get_parent_priv(dev);
1005 
1006 	slave->max_write_size = TX_BUFFER_SIZE;
1007 
1008 	return 0;
1009 }
1010 
1011 static int fsl_qspi_probe(struct udevice *bus)
1012 {
1013 	u32 mcr_val;
1014 	u32 amba_size_per_chip;
1015 	struct fsl_qspi_platdata *plat = dev_get_platdata(bus);
1016 	struct fsl_qspi_priv *priv = dev_get_priv(bus);
1017 	struct dm_spi_bus *dm_spi_bus;
1018 	int i, ret;
1019 
1020 	dm_spi_bus = bus->uclass_priv;
1021 
1022 	dm_spi_bus->max_hz = plat->speed_hz;
1023 
1024 	priv->regs = (struct fsl_qspi_regs *)(uintptr_t)plat->reg_base;
1025 	priv->flags = plat->flags;
1026 
1027 	priv->speed_hz = plat->speed_hz;
1028 	/*
1029 	 * QSPI SFADR width is 32bits, the max dest addr is 4GB-1.
1030 	 * AMBA memory zone should be located on the 0~4GB space
1031 	 * even on a 64bits cpu.
1032 	 */
1033 	priv->amba_base[0] = (u32)plat->amba_base;
1034 	priv->amba_total_size = (u32)plat->amba_total_size;
1035 	priv->flash_num = plat->flash_num;
1036 	priv->num_chipselect = plat->num_chipselect;
1037 
1038 	/* make sure controller is not busy anywhere */
1039 	ret = is_controller_busy(priv);
1040 
1041 	if (ret) {
1042 		debug("ERROR : The controller is busy\n");
1043 		return ret;
1044 	}
1045 
1046 	mcr_val = qspi_read32(priv->flags, &priv->regs->mcr);
1047 
1048 	/* Set endianness to LE for i.mx */
1049 	if (IS_ENABLED(CONFIG_MX6) || IS_ENABLED(CONFIG_MX7))
1050 		mcr_val = QSPI_MCR_END_CFD_LE;
1051 
1052 	qspi_write32(priv->flags, &priv->regs->mcr,
1053 		     QSPI_MCR_RESERVED_MASK | QSPI_MCR_MDIS_MASK |
1054 		     (mcr_val & QSPI_MCR_END_CFD_MASK));
1055 
1056 	qspi_cfg_smpr(priv, ~(QSPI_SMPR_FSDLY_MASK | QSPI_SMPR_DDRSMP_MASK |
1057 		QSPI_SMPR_FSPHS_MASK | QSPI_SMPR_HSENA_MASK), 0);
1058 
1059 	/*
1060 	 * Assign AMBA memory zone for every chipselect
1061 	 * QuadSPI has two channels, every channel has two chipselects.
1062 	 * If the property 'num-cs' in dts is 2, the AMBA memory will be divided
1063 	 * into two parts and assign to every channel. This indicate that every
1064 	 * channel only has one valid chipselect.
1065 	 * If the property 'num-cs' in dts is 4, the AMBA memory will be divided
1066 	 * into four parts and assign to every chipselect.
1067 	 * Every channel will has two valid chipselects.
1068 	 */
1069 	amba_size_per_chip = priv->amba_total_size >>
1070 			     (priv->num_chipselect >> 1);
1071 	for (i = 1 ; i < priv->num_chipselect ; i++)
1072 		priv->amba_base[i] =
1073 			amba_size_per_chip + priv->amba_base[i - 1];
1074 
1075 	/*
1076 	 * Any read access to non-implemented addresses will provide
1077 	 * undefined results.
1078 	 *
1079 	 * In case single die flash devices, TOP_ADDR_MEMA2 and
1080 	 * TOP_ADDR_MEMB2 should be initialized/programmed to
1081 	 * TOP_ADDR_MEMA1 and TOP_ADDR_MEMB1 respectively - in effect,
1082 	 * setting the size of these devices to 0.  This would ensure
1083 	 * that the complete memory map is assigned to only one flash device.
1084 	 */
1085 	qspi_write32(priv->flags, &priv->regs->sfa1ad,
1086 		     priv->amba_base[0] + amba_size_per_chip);
1087 	switch (priv->num_chipselect) {
1088 	case 1:
1089 		break;
1090 	case 2:
1091 		qspi_write32(priv->flags, &priv->regs->sfa2ad,
1092 			     priv->amba_base[1]);
1093 		qspi_write32(priv->flags, &priv->regs->sfb1ad,
1094 			     priv->amba_base[1] + amba_size_per_chip);
1095 		qspi_write32(priv->flags, &priv->regs->sfb2ad,
1096 			     priv->amba_base[1] + amba_size_per_chip);
1097 		break;
1098 	case 4:
1099 		qspi_write32(priv->flags, &priv->regs->sfa2ad,
1100 			     priv->amba_base[2]);
1101 		qspi_write32(priv->flags, &priv->regs->sfb1ad,
1102 			     priv->amba_base[3]);
1103 		qspi_write32(priv->flags, &priv->regs->sfb2ad,
1104 			     priv->amba_base[3] + amba_size_per_chip);
1105 		break;
1106 	default:
1107 		debug("Error: Unsupported chipselect number %u!\n",
1108 		      priv->num_chipselect);
1109 		qspi_module_disable(priv, 1);
1110 		return -EINVAL;
1111 	}
1112 
1113 	qspi_set_lut(priv);
1114 
1115 #ifdef CONFIG_SYS_FSL_QSPI_AHB
1116 	qspi_init_ahb_read(priv);
1117 #endif
1118 
1119 	qspi_module_disable(priv, 0);
1120 
1121 	return 0;
1122 }
1123 
1124 static int fsl_qspi_ofdata_to_platdata(struct udevice *bus)
1125 {
1126 	struct fdt_resource res_regs, res_mem;
1127 	struct fsl_qspi_platdata *plat = bus->platdata;
1128 	const void *blob = gd->fdt_blob;
1129 	int node = dev_of_offset(bus);
1130 	int ret, flash_num = 0, subnode;
1131 
1132 	if (fdtdec_get_bool(blob, node, "big-endian"))
1133 		plat->flags |= QSPI_FLAG_REGMAP_ENDIAN_BIG;
1134 
1135 	ret = fdt_get_named_resource(blob, node, "reg", "reg-names",
1136 				     "QuadSPI", &res_regs);
1137 	if (ret) {
1138 		debug("Error: can't get regs base addresses(ret = %d)!\n", ret);
1139 		return -ENOMEM;
1140 	}
1141 	ret = fdt_get_named_resource(blob, node, "reg", "reg-names",
1142 				     "QuadSPI-memory", &res_mem);
1143 	if (ret) {
1144 		debug("Error: can't get AMBA base addresses(ret = %d)!\n", ret);
1145 		return -ENOMEM;
1146 	}
1147 
1148 	/* Count flash numbers */
1149 	fdt_for_each_subnode(subnode, blob, node)
1150 		++flash_num;
1151 
1152 	if (flash_num == 0) {
1153 		debug("Error: Missing flashes!\n");
1154 		return -ENODEV;
1155 	}
1156 
1157 	plat->speed_hz = fdtdec_get_int(blob, node, "spi-max-frequency",
1158 					FSL_QSPI_DEFAULT_SCK_FREQ);
1159 	plat->num_chipselect = fdtdec_get_int(blob, node, "num-cs",
1160 					      FSL_QSPI_MAX_CHIPSELECT_NUM);
1161 
1162 	plat->reg_base = res_regs.start;
1163 	plat->amba_base = res_mem.start;
1164 	plat->amba_total_size = res_mem.end - res_mem.start + 1;
1165 	plat->flash_num = flash_num;
1166 
1167 	debug("%s: regs=<0x%llx> <0x%llx, 0x%llx>, max-frequency=%d, endianess=%s\n",
1168 	      __func__,
1169 	      (u64)plat->reg_base,
1170 	      (u64)plat->amba_base,
1171 	      (u64)plat->amba_total_size,
1172 	      plat->speed_hz,
1173 	      plat->flags & QSPI_FLAG_REGMAP_ENDIAN_BIG ? "be" : "le"
1174 	      );
1175 
1176 	return 0;
1177 }
1178 
1179 static int fsl_qspi_xfer(struct udevice *dev, unsigned int bitlen,
1180 		const void *dout, void *din, unsigned long flags)
1181 {
1182 	struct fsl_qspi_priv *priv;
1183 	struct udevice *bus;
1184 
1185 	bus = dev->parent;
1186 	priv = dev_get_priv(bus);
1187 
1188 	return qspi_xfer(priv, bitlen, dout, din, flags);
1189 }
1190 
1191 static int fsl_qspi_claim_bus(struct udevice *dev)
1192 {
1193 	struct fsl_qspi_priv *priv;
1194 	struct udevice *bus;
1195 	struct dm_spi_slave_platdata *slave_plat = dev_get_parent_platdata(dev);
1196 	int ret;
1197 
1198 	bus = dev->parent;
1199 	priv = dev_get_priv(bus);
1200 
1201 	/* make sure controller is not busy anywhere */
1202 	ret = is_controller_busy(priv);
1203 
1204 	if (ret) {
1205 		debug("ERROR : The controller is busy\n");
1206 		return ret;
1207 	}
1208 
1209 	priv->cur_amba_base = priv->amba_base[slave_plat->cs];
1210 
1211 	qspi_module_disable(priv, 0);
1212 
1213 	return 0;
1214 }
1215 
1216 static int fsl_qspi_release_bus(struct udevice *dev)
1217 {
1218 	struct fsl_qspi_priv *priv;
1219 	struct udevice *bus;
1220 
1221 	bus = dev->parent;
1222 	priv = dev_get_priv(bus);
1223 
1224 	qspi_module_disable(priv, 1);
1225 
1226 	return 0;
1227 }
1228 
1229 static int fsl_qspi_set_speed(struct udevice *bus, uint speed)
1230 {
1231 	/* Nothing to do */
1232 	return 0;
1233 }
1234 
1235 static int fsl_qspi_set_mode(struct udevice *bus, uint mode)
1236 {
1237 	/* Nothing to do */
1238 	return 0;
1239 }
1240 
1241 static const struct dm_spi_ops fsl_qspi_ops = {
1242 	.claim_bus	= fsl_qspi_claim_bus,
1243 	.release_bus	= fsl_qspi_release_bus,
1244 	.xfer		= fsl_qspi_xfer,
1245 	.set_speed	= fsl_qspi_set_speed,
1246 	.set_mode	= fsl_qspi_set_mode,
1247 };
1248 
1249 static const struct udevice_id fsl_qspi_ids[] = {
1250 	{ .compatible = "fsl,vf610-qspi" },
1251 	{ .compatible = "fsl,imx6sx-qspi" },
1252 	{ .compatible = "fsl,imx6ul-qspi" },
1253 	{ .compatible = "fsl,imx7d-qspi" },
1254 	{ }
1255 };
1256 
1257 U_BOOT_DRIVER(fsl_qspi) = {
1258 	.name	= "fsl_qspi",
1259 	.id	= UCLASS_SPI,
1260 	.of_match = fsl_qspi_ids,
1261 	.ops	= &fsl_qspi_ops,
1262 	.ofdata_to_platdata = fsl_qspi_ofdata_to_platdata,
1263 	.platdata_auto_alloc_size = sizeof(struct fsl_qspi_platdata),
1264 	.priv_auto_alloc_size = sizeof(struct fsl_qspi_priv),
1265 	.probe	= fsl_qspi_probe,
1266 	.child_pre_probe = fsl_qspi_child_pre_probe,
1267 };
1268 #endif
1269