xref: /rk3399_rockchip-uboot/drivers/mtd/spi/spi_flash.c (revision f1ba13f8e2acf648740a4a8a3594509a3e16b3aa)
1 /*
2  * SPI Flash Core
3  *
4  * Copyright (C) 2015 Jagan Teki <jteki@openedev.com>
5  * Copyright (C) 2013 Jagannadha Sutradharudu Teki, Xilinx Inc.
6  * Copyright (C) 2010 Reinhard Meyer, EMK Elektronik
7  * Copyright (C) 2008 Atmel Corporation
8  *
9  * SPDX-License-Identifier:	GPL-2.0+
10  */
11 
12 #include <common.h>
13 #include <errno.h>
14 #include <malloc.h>
15 #include <mapmem.h>
16 #include <spi.h>
17 #include <spi_flash.h>
18 #include <linux/log2.h>
19 #include <dma.h>
20 
21 #include "sf_internal.h"
22 
23 DECLARE_GLOBAL_DATA_PTR;
24 
25 static void spi_flash_addr(u32 addr, u8 *cmd)
26 {
27 	/* cmd[0] is actual command */
28 	cmd[1] = addr >> 16;
29 	cmd[2] = addr >> 8;
30 	cmd[3] = addr >> 0;
31 }
32 
33 static int read_sr(struct spi_flash *flash, u8 *rs)
34 {
35 	int ret;
36 	u8 cmd;
37 
38 	cmd = CMD_READ_STATUS;
39 	ret = spi_flash_read_common(flash, &cmd, 1, rs, 1);
40 	if (ret < 0) {
41 		debug("SF: fail to read status register\n");
42 		return ret;
43 	}
44 
45 	return 0;
46 }
47 
48 static int read_fsr(struct spi_flash *flash, u8 *fsr)
49 {
50 	int ret;
51 	const u8 cmd = CMD_FLAG_STATUS;
52 
53 	ret = spi_flash_read_common(flash, &cmd, 1, fsr, 1);
54 	if (ret < 0) {
55 		debug("SF: fail to read flag status register\n");
56 		return ret;
57 	}
58 
59 	return 0;
60 }
61 
62 static int write_sr(struct spi_flash *flash, u8 ws)
63 {
64 	u8 cmd;
65 	int ret;
66 
67 	cmd = CMD_WRITE_STATUS;
68 	ret = spi_flash_write_common(flash, &cmd, 1, &ws, 1);
69 	if (ret < 0) {
70 		debug("SF: fail to write status register\n");
71 		return ret;
72 	}
73 
74 	return 0;
75 }
76 
77 #if defined(CONFIG_SPI_FLASH_SPANSION) || defined(CONFIG_SPI_FLASH_WINBOND)
78 static int read_cr(struct spi_flash *flash, u8 *rc)
79 {
80 	int ret;
81 	u8 cmd;
82 
83 	cmd = CMD_READ_CONFIG;
84 	ret = spi_flash_read_common(flash, &cmd, 1, rc, 1);
85 	if (ret < 0) {
86 		debug("SF: fail to read config register\n");
87 		return ret;
88 	}
89 
90 	return 0;
91 }
92 
93 static int write_cr(struct spi_flash *flash, u8 wc)
94 {
95 	u8 data[2];
96 	u8 cmd;
97 	int ret;
98 
99 	ret = read_sr(flash, &data[0]);
100 	if (ret < 0)
101 		return ret;
102 
103 	cmd = CMD_WRITE_STATUS;
104 	data[1] = wc;
105 	ret = spi_flash_write_common(flash, &cmd, 1, &data, 2);
106 	if (ret) {
107 		debug("SF: fail to write config register\n");
108 		return ret;
109 	}
110 
111 	return 0;
112 }
113 #endif
114 
115 #ifdef CONFIG_SPI_FLASH_BAR
116 /*
117  * This "clean_bar" is necessary in a situation when one was accessing
118  * spi flash memory > 16 MiB by using Bank Address Register's BA24 bit.
119  *
120  * After it the BA24 bit shall be cleared to allow access to correct
121  * memory region after SW reset (by calling "reset" command).
122  *
123  * Otherwise, the BA24 bit may be left set and then after reset, the
124  * ROM would read/write/erase SPL from 16 MiB * bank_sel address.
125  */
126 static int clean_bar(struct spi_flash *flash)
127 {
128 	u8 cmd, bank_sel = 0;
129 
130 	if (flash->bank_curr == 0)
131 		return 0;
132 	cmd = flash->bank_write_cmd;
133 
134 	return spi_flash_write_common(flash, &cmd, 1, &bank_sel, 1);
135 }
136 
137 static int write_bar(struct spi_flash *flash, u32 offset)
138 {
139 	u8 cmd, bank_sel;
140 	int ret;
141 
142 	bank_sel = offset / (SPI_FLASH_16MB_BOUN << flash->shift);
143 	if (bank_sel == flash->bank_curr)
144 		goto bar_end;
145 
146 	cmd = flash->bank_write_cmd;
147 	ret = spi_flash_write_common(flash, &cmd, 1, &bank_sel, 1);
148 	if (ret < 0) {
149 		debug("SF: fail to write bank register\n");
150 		return ret;
151 	}
152 
153 bar_end:
154 	flash->bank_curr = bank_sel;
155 	return flash->bank_curr;
156 }
157 
158 static int read_bar(struct spi_flash *flash, const struct spi_flash_info *info)
159 {
160 	u8 curr_bank = 0;
161 	int ret;
162 
163 	if (flash->size <= SPI_FLASH_16MB_BOUN)
164 		goto bar_end;
165 
166 	switch (JEDEC_MFR(info)) {
167 	case SPI_FLASH_CFI_MFR_SPANSION:
168 		flash->bank_read_cmd = CMD_BANKADDR_BRRD;
169 		flash->bank_write_cmd = CMD_BANKADDR_BRWR;
170 		break;
171 	default:
172 		flash->bank_read_cmd = CMD_EXTNADDR_RDEAR;
173 		flash->bank_write_cmd = CMD_EXTNADDR_WREAR;
174 	}
175 
176 	ret = spi_flash_read_common(flash, &flash->bank_read_cmd, 1,
177 				    &curr_bank, 1);
178 	if (ret) {
179 		debug("SF: fail to read bank addr register\n");
180 		return ret;
181 	}
182 
183 bar_end:
184 	flash->bank_curr = curr_bank;
185 	return 0;
186 }
187 #endif
188 
189 #ifdef CONFIG_SF_DUAL_FLASH
190 static void spi_flash_dual(struct spi_flash *flash, u32 *addr)
191 {
192 	switch (flash->dual_flash) {
193 	case SF_DUAL_STACKED_FLASH:
194 		if (*addr >= (flash->size >> 1)) {
195 			*addr -= flash->size >> 1;
196 			flash->flags |= SNOR_F_USE_UPAGE;
197 		} else {
198 			flash->flags &= ~SNOR_F_USE_UPAGE;
199 		}
200 		break;
201 	case SF_DUAL_PARALLEL_FLASH:
202 		*addr >>= flash->shift;
203 		break;
204 	default:
205 		debug("SF: Unsupported dual_flash=%d\n", flash->dual_flash);
206 		break;
207 	}
208 }
209 #endif
210 
211 static int spi_flash_sr_ready(struct spi_flash *flash)
212 {
213 	u8 sr;
214 	int ret;
215 
216 	ret = read_sr(flash, &sr);
217 	if (ret < 0)
218 		return ret;
219 
220 	return !(sr & STATUS_WIP);
221 }
222 
223 static int spi_flash_fsr_ready(struct spi_flash *flash)
224 {
225 	u8 fsr;
226 	int ret;
227 
228 	ret = read_fsr(flash, &fsr);
229 	if (ret < 0)
230 		return ret;
231 
232 	return fsr & STATUS_PEC;
233 }
234 
235 static int spi_flash_ready(struct spi_flash *flash)
236 {
237 	int sr, fsr;
238 
239 	sr = spi_flash_sr_ready(flash);
240 	if (sr < 0)
241 		return sr;
242 
243 	fsr = 1;
244 	if (flash->flags & SNOR_F_USE_FSR) {
245 		fsr = spi_flash_fsr_ready(flash);
246 		if (fsr < 0)
247 			return fsr;
248 	}
249 
250 	return sr && fsr;
251 }
252 
253 static int spi_flash_wait_till_ready(struct spi_flash *flash,
254 				     unsigned long timeout)
255 {
256 	unsigned long timebase;
257 	int ret;
258 
259 	timebase = get_timer(0);
260 
261 	while (get_timer(timebase) < timeout) {
262 		ret = spi_flash_ready(flash);
263 		if (ret < 0)
264 			return ret;
265 		if (ret)
266 			return 0;
267 	}
268 
269 	printf("SF: Timeout!\n");
270 
271 	return -ETIMEDOUT;
272 }
273 
274 int spi_flash_write_common(struct spi_flash *flash, const u8 *cmd,
275 		size_t cmd_len, const void *buf, size_t buf_len)
276 {
277 	struct spi_slave *spi = flash->spi;
278 	unsigned long timeout = SPI_FLASH_PROG_TIMEOUT;
279 	int ret;
280 
281 	if (buf == NULL)
282 		timeout = SPI_FLASH_PAGE_ERASE_TIMEOUT;
283 
284 	ret = spi_claim_bus(spi);
285 	if (ret) {
286 		debug("SF: unable to claim SPI bus\n");
287 		return ret;
288 	}
289 
290 	ret = spi_flash_cmd_write_enable(flash);
291 	if (ret < 0) {
292 		debug("SF: enabling write failed\n");
293 		return ret;
294 	}
295 
296 	ret = spi_flash_cmd_write(spi, cmd, cmd_len, buf, buf_len);
297 	if (ret < 0) {
298 		debug("SF: write cmd failed\n");
299 		return ret;
300 	}
301 
302 	ret = spi_flash_wait_till_ready(flash, timeout);
303 	if (ret < 0) {
304 		debug("SF: write %s timed out\n",
305 		      timeout == SPI_FLASH_PROG_TIMEOUT ?
306 			"program" : "page erase");
307 		return ret;
308 	}
309 
310 	spi_release_bus(spi);
311 
312 	return ret;
313 }
314 
315 int spi_flash_cmd_erase_ops(struct spi_flash *flash, u32 offset, size_t len)
316 {
317 	u32 erase_size, erase_addr;
318 	u8 cmd[SPI_FLASH_CMD_LEN];
319 	int ret = -1;
320 
321 	erase_size = flash->erase_size;
322 	if (offset % erase_size || len % erase_size) {
323 		debug("SF: Erase offset/length not multiple of erase size\n");
324 		return -1;
325 	}
326 
327 	if (flash->flash_is_locked) {
328 		if (flash->flash_is_locked(flash, offset, len) > 0) {
329 			printf("offset 0x%x is protected and cannot be erased\n",
330 			       offset);
331 			return -EINVAL;
332 		}
333 	}
334 
335 	cmd[0] = flash->erase_cmd;
336 	while (len) {
337 		erase_addr = offset;
338 
339 #ifdef CONFIG_SF_DUAL_FLASH
340 		if (flash->dual_flash > SF_SINGLE_FLASH)
341 			spi_flash_dual(flash, &erase_addr);
342 #endif
343 #ifdef CONFIG_SPI_FLASH_BAR
344 		ret = write_bar(flash, erase_addr);
345 		if (ret < 0)
346 			return ret;
347 #endif
348 		spi_flash_addr(erase_addr, cmd);
349 
350 		debug("SF: erase %2x %2x %2x %2x (%x)\n", cmd[0], cmd[1],
351 		      cmd[2], cmd[3], erase_addr);
352 
353 		ret = spi_flash_write_common(flash, cmd, sizeof(cmd), NULL, 0);
354 		if (ret < 0) {
355 			debug("SF: erase failed\n");
356 			break;
357 		}
358 
359 		offset += erase_size;
360 		len -= erase_size;
361 	}
362 
363 #ifdef CONFIG_SPI_FLASH_BAR
364 	ret = clean_bar(flash);
365 #endif
366 
367 	return ret;
368 }
369 
370 int spi_flash_cmd_write_ops(struct spi_flash *flash, u32 offset,
371 		size_t len, const void *buf)
372 {
373 	struct spi_slave *spi = flash->spi;
374 	unsigned long byte_addr, page_size;
375 	u32 write_addr;
376 	size_t chunk_len, actual;
377 	u8 cmd[SPI_FLASH_CMD_LEN];
378 	int ret = -1;
379 
380 	page_size = flash->page_size;
381 
382 	if (flash->flash_is_locked) {
383 		if (flash->flash_is_locked(flash, offset, len) > 0) {
384 			printf("offset 0x%x is protected and cannot be written\n",
385 			       offset);
386 			return -EINVAL;
387 		}
388 	}
389 
390 	cmd[0] = flash->write_cmd;
391 	for (actual = 0; actual < len; actual += chunk_len) {
392 		write_addr = offset;
393 
394 #ifdef CONFIG_SF_DUAL_FLASH
395 		if (flash->dual_flash > SF_SINGLE_FLASH)
396 			spi_flash_dual(flash, &write_addr);
397 #endif
398 #ifdef CONFIG_SPI_FLASH_BAR
399 		ret = write_bar(flash, write_addr);
400 		if (ret < 0)
401 			return ret;
402 #endif
403 		byte_addr = offset % page_size;
404 		chunk_len = min(len - actual, (size_t)(page_size - byte_addr));
405 
406 		if (spi->max_write_size)
407 			chunk_len = min(chunk_len,
408 					spi->max_write_size - sizeof(cmd));
409 
410 		spi_flash_addr(write_addr, cmd);
411 
412 		debug("SF: 0x%p => cmd = { 0x%02x 0x%02x%02x%02x } chunk_len = %zu\n",
413 		      buf + actual, cmd[0], cmd[1], cmd[2], cmd[3], chunk_len);
414 
415 		ret = spi_flash_write_common(flash, cmd, sizeof(cmd),
416 					buf + actual, chunk_len);
417 		if (ret < 0) {
418 			debug("SF: write failed\n");
419 			break;
420 		}
421 
422 		offset += chunk_len;
423 	}
424 
425 #ifdef CONFIG_SPI_FLASH_BAR
426 	ret = clean_bar(flash);
427 #endif
428 
429 	return ret;
430 }
431 
432 int spi_flash_read_common(struct spi_flash *flash, const u8 *cmd,
433 		size_t cmd_len, void *data, size_t data_len)
434 {
435 	struct spi_slave *spi = flash->spi;
436 	int ret;
437 
438 	ret = spi_claim_bus(spi);
439 	if (ret) {
440 		debug("SF: unable to claim SPI bus\n");
441 		return ret;
442 	}
443 
444 	ret = spi_flash_cmd_read(spi, cmd, cmd_len, data, data_len);
445 	if (ret < 0) {
446 		debug("SF: read cmd failed\n");
447 		return ret;
448 	}
449 
450 	spi_release_bus(spi);
451 
452 	return ret;
453 }
454 
455 /*
456  * TODO: remove the weak after all the other spi_flash_copy_mmap
457  * implementations removed from drivers
458  */
459 void __weak spi_flash_copy_mmap(void *data, void *offset, size_t len)
460 {
461 #ifdef CONFIG_DMA
462 	if (!dma_memcpy(data, offset, len))
463 		return;
464 #endif
465 	memcpy(data, offset, len);
466 }
467 
468 int spi_flash_cmd_read_ops(struct spi_flash *flash, u32 offset,
469 		size_t len, void *data)
470 {
471 	struct spi_slave *spi = flash->spi;
472 	u8 *cmd, cmdsz;
473 	u32 remain_len, read_len, read_addr;
474 	int bank_sel = 0;
475 	int ret = -1;
476 
477 	/* Handle memory-mapped SPI */
478 	if (flash->memory_map) {
479 		ret = spi_claim_bus(spi);
480 		if (ret) {
481 			debug("SF: unable to claim SPI bus\n");
482 			return ret;
483 		}
484 		spi_xfer(spi, 0, NULL, NULL, SPI_XFER_MMAP);
485 		spi_flash_copy_mmap(data, flash->memory_map + offset, len);
486 		spi_xfer(spi, 0, NULL, NULL, SPI_XFER_MMAP_END);
487 		spi_release_bus(spi);
488 		return 0;
489 	}
490 
491 	cmdsz = SPI_FLASH_CMD_LEN + flash->dummy_byte;
492 	cmd = calloc(1, cmdsz);
493 	if (!cmd) {
494 		debug("SF: Failed to allocate cmd\n");
495 		return -ENOMEM;
496 	}
497 
498 	cmd[0] = flash->read_cmd;
499 	while (len) {
500 		read_addr = offset;
501 
502 #ifdef CONFIG_SF_DUAL_FLASH
503 		if (flash->dual_flash > SF_SINGLE_FLASH)
504 			spi_flash_dual(flash, &read_addr);
505 #endif
506 #ifdef CONFIG_SPI_FLASH_BAR
507 		ret = write_bar(flash, read_addr);
508 		if (ret < 0)
509 			return ret;
510 		bank_sel = flash->bank_curr;
511 #endif
512 		remain_len = ((SPI_FLASH_16MB_BOUN << flash->shift) *
513 				(bank_sel + 1)) - offset;
514 		if (len < remain_len)
515 			read_len = len;
516 		else
517 			read_len = remain_len;
518 
519 		if (spi->max_read_size)
520 			read_len = min(read_len, spi->max_read_size);
521 
522 		spi_flash_addr(read_addr, cmd);
523 
524 		ret = spi_flash_read_common(flash, cmd, cmdsz, data, read_len);
525 		if (ret < 0) {
526 			debug("SF: read failed\n");
527 			break;
528 		}
529 
530 		offset += read_len;
531 		len -= read_len;
532 		data += read_len;
533 	}
534 
535 #ifdef CONFIG_SPI_FLASH_BAR
536 	ret = clean_bar(flash);
537 #endif
538 
539 	free(cmd);
540 	return ret;
541 }
542 
543 #ifdef CONFIG_SPI_FLASH_SST
544 static int sst_byte_write(struct spi_flash *flash, u32 offset, const void *buf)
545 {
546 	struct spi_slave *spi = flash->spi;
547 	int ret;
548 	u8 cmd[4] = {
549 		CMD_SST_BP,
550 		offset >> 16,
551 		offset >> 8,
552 		offset,
553 	};
554 
555 	debug("BP[%02x]: 0x%p => cmd = { 0x%02x 0x%06x }\n",
556 	      spi_w8r8(spi, CMD_READ_STATUS), buf, cmd[0], offset);
557 
558 	ret = spi_flash_cmd_write_enable(flash);
559 	if (ret)
560 		return ret;
561 
562 	ret = spi_flash_cmd_write(spi, cmd, sizeof(cmd), buf, 1);
563 	if (ret)
564 		return ret;
565 
566 	return spi_flash_wait_till_ready(flash, SPI_FLASH_PROG_TIMEOUT);
567 }
568 
569 int sst_write_wp(struct spi_flash *flash, u32 offset, size_t len,
570 		const void *buf)
571 {
572 	struct spi_slave *spi = flash->spi;
573 	size_t actual, cmd_len;
574 	int ret;
575 	u8 cmd[4];
576 
577 	ret = spi_claim_bus(spi);
578 	if (ret) {
579 		debug("SF: Unable to claim SPI bus\n");
580 		return ret;
581 	}
582 
583 	/* If the data is not word aligned, write out leading single byte */
584 	actual = offset % 2;
585 	if (actual) {
586 		ret = sst_byte_write(flash, offset, buf);
587 		if (ret)
588 			goto done;
589 	}
590 	offset += actual;
591 
592 	ret = spi_flash_cmd_write_enable(flash);
593 	if (ret)
594 		goto done;
595 
596 	cmd_len = 4;
597 	cmd[0] = CMD_SST_AAI_WP;
598 	cmd[1] = offset >> 16;
599 	cmd[2] = offset >> 8;
600 	cmd[3] = offset;
601 
602 	for (; actual < len - 1; actual += 2) {
603 		debug("WP[%02x]: 0x%p => cmd = { 0x%02x 0x%06x }\n",
604 		      spi_w8r8(spi, CMD_READ_STATUS), buf + actual,
605 		      cmd[0], offset);
606 
607 		ret = spi_flash_cmd_write(spi, cmd, cmd_len,
608 					buf + actual, 2);
609 		if (ret) {
610 			debug("SF: sst word program failed\n");
611 			break;
612 		}
613 
614 		ret = spi_flash_wait_till_ready(flash, SPI_FLASH_PROG_TIMEOUT);
615 		if (ret)
616 			break;
617 
618 		cmd_len = 1;
619 		offset += 2;
620 	}
621 
622 	if (!ret)
623 		ret = spi_flash_cmd_write_disable(flash);
624 
625 	/* If there is a single trailing byte, write it out */
626 	if (!ret && actual != len)
627 		ret = sst_byte_write(flash, offset, buf + actual);
628 
629  done:
630 	debug("SF: sst: program %s %zu bytes @ 0x%zx\n",
631 	      ret ? "failure" : "success", len, offset - actual);
632 
633 	spi_release_bus(spi);
634 	return ret;
635 }
636 
637 int sst_write_bp(struct spi_flash *flash, u32 offset, size_t len,
638 		const void *buf)
639 {
640 	struct spi_slave *spi = flash->spi;
641 	size_t actual;
642 	int ret;
643 
644 	ret = spi_claim_bus(spi);
645 	if (ret) {
646 		debug("SF: Unable to claim SPI bus\n");
647 		return ret;
648 	}
649 
650 	for (actual = 0; actual < len; actual++) {
651 		ret = sst_byte_write(flash, offset, buf + actual);
652 		if (ret) {
653 			debug("SF: sst byte program failed\n");
654 			break;
655 		}
656 		offset++;
657 	}
658 
659 	if (!ret)
660 		ret = spi_flash_cmd_write_disable(flash);
661 
662 	debug("SF: sst: program %s %zu bytes @ 0x%zx\n",
663 	      ret ? "failure" : "success", len, offset - actual);
664 
665 	spi_release_bus(spi);
666 	return ret;
667 }
668 #endif
669 
670 #if defined(CONFIG_SPI_FLASH_STMICRO) || defined(CONFIG_SPI_FLASH_SST)
671 static void stm_get_locked_range(struct spi_flash *flash, u8 sr, loff_t *ofs,
672 				 u64 *len)
673 {
674 	u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
675 	int shift = ffs(mask) - 1;
676 	int pow;
677 
678 	if (!(sr & mask)) {
679 		/* No protection */
680 		*ofs = 0;
681 		*len = 0;
682 	} else {
683 		pow = ((sr & mask) ^ mask) >> shift;
684 		*len = flash->size >> pow;
685 		*ofs = flash->size - *len;
686 	}
687 }
688 
689 /*
690  * Return 1 if the entire region is locked, 0 otherwise
691  */
692 static int stm_is_locked_sr(struct spi_flash *flash, loff_t ofs, u64 len,
693 			    u8 sr)
694 {
695 	loff_t lock_offs;
696 	u64 lock_len;
697 
698 	stm_get_locked_range(flash, sr, &lock_offs, &lock_len);
699 
700 	return (ofs + len <= lock_offs + lock_len) && (ofs >= lock_offs);
701 }
702 
703 /*
704  * Check if a region of the flash is (completely) locked. See stm_lock() for
705  * more info.
706  *
707  * Returns 1 if entire region is locked, 0 if any portion is unlocked, and
708  * negative on errors.
709  */
710 int stm_is_locked(struct spi_flash *flash, u32 ofs, size_t len)
711 {
712 	int status;
713 	u8 sr;
714 
715 	status = read_sr(flash, &sr);
716 	if (status < 0)
717 		return status;
718 
719 	return stm_is_locked_sr(flash, ofs, len, sr);
720 }
721 
722 /*
723  * Lock a region of the flash. Compatible with ST Micro and similar flash.
724  * Supports only the block protection bits BP{0,1,2} in the status register
725  * (SR). Does not support these features found in newer SR bitfields:
726  *   - TB: top/bottom protect - only handle TB=0 (top protect)
727  *   - SEC: sector/block protect - only handle SEC=0 (block protect)
728  *   - CMP: complement protect - only support CMP=0 (range is not complemented)
729  *
730  * Sample table portion for 8MB flash (Winbond w25q64fw):
731  *
732  *   SEC  |  TB   |  BP2  |  BP1  |  BP0  |  Prot Length  | Protected Portion
733  *  --------------------------------------------------------------------------
734  *    X   |   X   |   0   |   0   |   0   |  NONE         | NONE
735  *    0   |   0   |   0   |   0   |   1   |  128 KB       | Upper 1/64
736  *    0   |   0   |   0   |   1   |   0   |  256 KB       | Upper 1/32
737  *    0   |   0   |   0   |   1   |   1   |  512 KB       | Upper 1/16
738  *    0   |   0   |   1   |   0   |   0   |  1 MB         | Upper 1/8
739  *    0   |   0   |   1   |   0   |   1   |  2 MB         | Upper 1/4
740  *    0   |   0   |   1   |   1   |   0   |  4 MB         | Upper 1/2
741  *    X   |   X   |   1   |   1   |   1   |  8 MB         | ALL
742  *
743  * Returns negative on errors, 0 on success.
744  */
745 int stm_lock(struct spi_flash *flash, u32 ofs, size_t len)
746 {
747 	u8 status_old, status_new;
748 	u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
749 	u8 shift = ffs(mask) - 1, pow, val;
750 	int ret;
751 
752 	ret = read_sr(flash, &status_old);
753 	if (ret < 0)
754 		return ret;
755 
756 	/* SPI NOR always locks to the end */
757 	if (ofs + len != flash->size) {
758 		/* Does combined region extend to end? */
759 		if (!stm_is_locked_sr(flash, ofs + len, flash->size - ofs - len,
760 				      status_old))
761 			return -EINVAL;
762 		len = flash->size - ofs;
763 	}
764 
765 	/*
766 	 * Need smallest pow such that:
767 	 *
768 	 *   1 / (2^pow) <= (len / size)
769 	 *
770 	 * so (assuming power-of-2 size) we do:
771 	 *
772 	 *   pow = ceil(log2(size / len)) = log2(size) - floor(log2(len))
773 	 */
774 	pow = ilog2(flash->size) - ilog2(len);
775 	val = mask - (pow << shift);
776 	if (val & ~mask)
777 		return -EINVAL;
778 
779 	/* Don't "lock" with no region! */
780 	if (!(val & mask))
781 		return -EINVAL;
782 
783 	status_new = (status_old & ~mask) | val;
784 
785 	/* Only modify protection if it will not unlock other areas */
786 	if ((status_new & mask) <= (status_old & mask))
787 		return -EINVAL;
788 
789 	write_sr(flash, status_new);
790 
791 	return 0;
792 }
793 
794 /*
795  * Unlock a region of the flash. See stm_lock() for more info
796  *
797  * Returns negative on errors, 0 on success.
798  */
799 int stm_unlock(struct spi_flash *flash, u32 ofs, size_t len)
800 {
801 	uint8_t status_old, status_new;
802 	u8 mask = SR_BP2 | SR_BP1 | SR_BP0;
803 	u8 shift = ffs(mask) - 1, pow, val;
804 	int ret;
805 
806 	ret = read_sr(flash, &status_old);
807 	if (ret < 0)
808 		return ret;
809 
810 	/* Cannot unlock; would unlock larger region than requested */
811 	if (stm_is_locked_sr(flash, ofs - flash->erase_size, flash->erase_size,
812 			     status_old))
813 		return -EINVAL;
814 	/*
815 	 * Need largest pow such that:
816 	 *
817 	 *   1 / (2^pow) >= (len / size)
818 	 *
819 	 * so (assuming power-of-2 size) we do:
820 	 *
821 	 *   pow = floor(log2(size / len)) = log2(size) - ceil(log2(len))
822 	 */
823 	pow = ilog2(flash->size) - order_base_2(flash->size - (ofs + len));
824 	if (ofs + len == flash->size) {
825 		val = 0; /* fully unlocked */
826 	} else {
827 		val = mask - (pow << shift);
828 		/* Some power-of-two sizes are not supported */
829 		if (val & ~mask)
830 			return -EINVAL;
831 	}
832 
833 	status_new = (status_old & ~mask) | val;
834 
835 	/* Only modify protection if it will not lock other areas */
836 	if ((status_new & mask) >= (status_old & mask))
837 		return -EINVAL;
838 
839 	write_sr(flash, status_new);
840 
841 	return 0;
842 }
843 #endif
844 
845 
846 #if defined(CONFIG_SPI_FLASH_MACRONIX) || defined(CONFIG_SPI_FLASH_GIGADEVICE)
847 static int macronix_quad_enable(struct spi_flash *flash)
848 {
849 	u8 qeb_status;
850 	int ret;
851 
852 	ret = read_sr(flash, &qeb_status);
853 	if (ret < 0)
854 		return ret;
855 
856 	if (qeb_status & STATUS_QEB_MXIC)
857 		return 0;
858 
859 	ret = write_sr(flash, qeb_status | STATUS_QEB_MXIC);
860 	if (ret < 0)
861 		return ret;
862 
863 	/* read SR and check it */
864 	ret = read_sr(flash, &qeb_status);
865 	if (!(ret >= 0 && (qeb_status & STATUS_QEB_MXIC))) {
866 		printf("SF: Macronix SR Quad bit not clear\n");
867 		return -EINVAL;
868 	}
869 
870 	return ret;
871 }
872 #endif
873 
874 #if defined(CONFIG_SPI_FLASH_SPANSION) || defined(CONFIG_SPI_FLASH_WINBOND)
875 static int spansion_quad_enable(struct spi_flash *flash)
876 {
877 	u8 qeb_status;
878 	int ret;
879 
880 	ret = read_cr(flash, &qeb_status);
881 	if (ret < 0)
882 		return ret;
883 
884 	if (qeb_status & STATUS_QEB_WINSPAN)
885 		return 0;
886 
887 	ret = write_cr(flash, qeb_status | STATUS_QEB_WINSPAN);
888 	if (ret < 0)
889 		return ret;
890 
891 	/* read CR and check it */
892 	ret = read_cr(flash, &qeb_status);
893 	if (!(ret >= 0 && (qeb_status & STATUS_QEB_WINSPAN))) {
894 		printf("SF: Spansion CR Quad bit not clear\n");
895 		return -EINVAL;
896 	}
897 
898 	return ret;
899 }
900 #endif
901 
902 static const struct spi_flash_info *spi_flash_read_id(struct spi_flash *flash)
903 {
904 	int				tmp;
905 	u8				id[SPI_FLASH_MAX_ID_LEN];
906 	const struct spi_flash_info	*info;
907 
908 	tmp = spi_flash_cmd(flash->spi, CMD_READ_ID, id, SPI_FLASH_MAX_ID_LEN);
909 	if (tmp < 0) {
910 		printf("SF: error %d reading JEDEC ID\n", tmp);
911 		return ERR_PTR(tmp);
912 	}
913 
914 	info = spi_flash_ids;
915 	for (; info->name != NULL; info++) {
916 		if (info->id_len) {
917 			if (!memcmp(info->id, id, info->id_len))
918 				return info;
919 		}
920 	}
921 
922 	printf("SF: unrecognized JEDEC id bytes: %02x, %02x, %02x\n",
923 	       id[0], id[1], id[2]);
924 	return ERR_PTR(-ENODEV);
925 }
926 
927 static int set_quad_mode(struct spi_flash *flash,
928 			 const struct spi_flash_info *info)
929 {
930 	switch (JEDEC_MFR(info)) {
931 #if defined(CONFIG_SPI_FLASH_MACRONIX) || defined(CONFIG_SPI_FLASH_GIGADEVICE)
932 	case SPI_FLASH_CFI_MFR_MACRONIX:
933 	case SPI_FLASH_CIF_MFR_GIGADEVICE:
934 		return macronix_quad_enable(flash);
935 #endif
936 #if defined(CONFIG_SPI_FLASH_SPANSION) || defined(CONFIG_SPI_FLASH_WINBOND)
937 	case SPI_FLASH_CFI_MFR_SPANSION:
938 	case SPI_FLASH_CFI_MFR_WINBOND:
939 		return spansion_quad_enable(flash);
940 #endif
941 #ifdef CONFIG_SPI_FLASH_STMICRO
942 	case SPI_FLASH_CFI_MFR_STMICRO:
943 		debug("SF: QEB is volatile for %02x flash\n", JEDEC_MFR(info));
944 		return 0;
945 #endif
946 	default:
947 		printf("SF: Need set QEB func for %02x flash\n",
948 		       JEDEC_MFR(info));
949 		return -1;
950 	}
951 }
952 
953 #if CONFIG_IS_ENABLED(OF_CONTROL)
954 int spi_flash_decode_fdt(struct spi_flash *flash)
955 {
956 #ifdef CONFIG_DM_SPI_FLASH
957 	fdt_addr_t addr;
958 	fdt_size_t size;
959 
960 	addr = dev_read_addr_size(flash->dev, "memory-map", &size);
961 	if (addr == FDT_ADDR_T_NONE) {
962 		debug("%s: Cannot decode address\n", __func__);
963 		return 0;
964 	}
965 
966 	if (flash->size > size) {
967 		debug("%s: Memory map must cover entire device\n", __func__);
968 		return -1;
969 	}
970 	flash->memory_map = map_sysmem(addr, size);
971 #endif
972 
973 	return 0;
974 }
975 #endif /* CONFIG_IS_ENABLED(OF_CONTROL) */
976 
977 int spi_flash_scan(struct spi_flash *flash)
978 {
979 	struct spi_slave *spi = flash->spi;
980 	const struct spi_flash_info *info = NULL;
981 	int ret;
982 
983 	info = spi_flash_read_id(flash);
984 	if (IS_ERR_OR_NULL(info))
985 		return -ENOENT;
986 
987 	/*
988 	 * Flash powers up read-only, so clear BP# bits.
989 	 *
990 	 * Note on some flash (like Macronix), QE (quad enable) bit is in the
991 	 * same status register as BP# bits, and we need preserve its original
992 	 * value during a reboot cycle as this is required by some platforms
993 	 * (like Intel ICH SPI controller working under descriptor mode).
994 	 */
995 	if (JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_ATMEL ||
996 	   (JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_SST) ||
997 	   (JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_MACRONIX)) {
998 		u8 sr = 0;
999 
1000 		if (JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_MACRONIX) {
1001 			read_sr(flash, &sr);
1002 			sr &= STATUS_QEB_MXIC;
1003 		}
1004 		write_sr(flash, sr);
1005 	}
1006 
1007 	flash->name = info->name;
1008 	flash->memory_map = spi->memory_map;
1009 
1010 	if (info->flags & SST_WR)
1011 		flash->flags |= SNOR_F_SST_WR;
1012 
1013 #ifndef CONFIG_DM_SPI_FLASH
1014 	flash->write = spi_flash_cmd_write_ops;
1015 #if defined(CONFIG_SPI_FLASH_SST)
1016 	if (flash->flags & SNOR_F_SST_WR) {
1017 		if (spi->mode & SPI_TX_BYTE)
1018 			flash->write = sst_write_bp;
1019 		else
1020 			flash->write = sst_write_wp;
1021 	}
1022 #endif
1023 	flash->erase = spi_flash_cmd_erase_ops;
1024 	flash->read = spi_flash_cmd_read_ops;
1025 #endif
1026 
1027 #if defined(CONFIG_SPI_FLASH_STMICRO) || defined(CONFIG_SPI_FLASH_SST)
1028 	/* NOR protection support for STmicro/Micron chips and similar */
1029 	if (JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_STMICRO ||
1030 	    JEDEC_MFR(info) == SPI_FLASH_CFI_MFR_SST) {
1031 		flash->flash_lock = stm_lock;
1032 		flash->flash_unlock = stm_unlock;
1033 		flash->flash_is_locked = stm_is_locked;
1034 	}
1035 #endif
1036 
1037 	/* Compute the flash size */
1038 	flash->shift = (flash->dual_flash & SF_DUAL_PARALLEL_FLASH) ? 1 : 0;
1039 	flash->page_size = info->page_size;
1040 	/*
1041 	 * The Spansion S25FL032P and S25FL064P have 256b pages, yet use the
1042 	 * 0x4d00 Extended JEDEC code. The rest of the Spansion flashes with
1043 	 * the 0x4d00 Extended JEDEC code have 512b pages. All of the others
1044 	 * have 256b pages.
1045 	 */
1046 	if (JEDEC_EXT(info) == 0x4d00) {
1047 		if ((JEDEC_ID(info) != 0x0215) &&
1048 		    (JEDEC_ID(info) != 0x0216))
1049 			flash->page_size = 512;
1050 	}
1051 	flash->page_size <<= flash->shift;
1052 	flash->sector_size = info->sector_size << flash->shift;
1053 	flash->size = flash->sector_size * info->n_sectors << flash->shift;
1054 #ifdef CONFIG_SF_DUAL_FLASH
1055 	if (flash->dual_flash & SF_DUAL_STACKED_FLASH)
1056 		flash->size <<= 1;
1057 #endif
1058 
1059 #ifdef CONFIG_SPI_FLASH_USE_4K_SECTORS
1060 	/* Compute erase sector and command */
1061 	if (info->flags & SECT_4K) {
1062 		flash->erase_cmd = CMD_ERASE_4K;
1063 		flash->erase_size = 4096 << flash->shift;
1064 	} else
1065 #endif
1066 	{
1067 		flash->erase_cmd = CMD_ERASE_64K;
1068 		flash->erase_size = flash->sector_size;
1069 	}
1070 
1071 	/* Now erase size becomes valid sector size */
1072 	flash->sector_size = flash->erase_size;
1073 
1074 	/* Look for read commands */
1075 	flash->read_cmd = CMD_READ_ARRAY_FAST;
1076 	if (spi->mode & SPI_RX_SLOW)
1077 		flash->read_cmd = CMD_READ_ARRAY_SLOW;
1078 	else if (spi->mode & SPI_RX_QUAD && info->flags & RD_QUAD)
1079 		flash->read_cmd = CMD_READ_QUAD_OUTPUT_FAST;
1080 	else if (spi->mode & SPI_RX_DUAL && info->flags & RD_DUAL)
1081 		flash->read_cmd = CMD_READ_DUAL_OUTPUT_FAST;
1082 
1083 	/* Look for write commands */
1084 	if (info->flags & WR_QPP && spi->mode & SPI_TX_QUAD)
1085 		flash->write_cmd = CMD_QUAD_PAGE_PROGRAM;
1086 	else
1087 		/* Go for default supported write cmd */
1088 		flash->write_cmd = CMD_PAGE_PROGRAM;
1089 
1090 	/* Set the quad enable bit - only for quad commands */
1091 	if ((flash->read_cmd == CMD_READ_QUAD_OUTPUT_FAST) ||
1092 	    (flash->read_cmd == CMD_READ_QUAD_IO_FAST) ||
1093 	    (flash->write_cmd == CMD_QUAD_PAGE_PROGRAM)) {
1094 		ret = set_quad_mode(flash, info);
1095 		if (ret) {
1096 			debug("SF: Fail to set QEB for %02x\n",
1097 			      JEDEC_MFR(info));
1098 			return -EINVAL;
1099 		}
1100 	}
1101 
1102 	/* Read dummy_byte: dummy byte is determined based on the
1103 	 * dummy cycles of a particular command.
1104 	 * Fast commands - dummy_byte = dummy_cycles/8
1105 	 * I/O commands- dummy_byte = (dummy_cycles * no.of lines)/8
1106 	 * For I/O commands except cmd[0] everything goes on no.of lines
1107 	 * based on particular command but incase of fast commands except
1108 	 * data all go on single line irrespective of command.
1109 	 */
1110 	switch (flash->read_cmd) {
1111 	case CMD_READ_QUAD_IO_FAST:
1112 		flash->dummy_byte = 2;
1113 		break;
1114 	case CMD_READ_ARRAY_SLOW:
1115 		flash->dummy_byte = 0;
1116 		break;
1117 	default:
1118 		flash->dummy_byte = 1;
1119 	}
1120 
1121 #ifdef CONFIG_SPI_FLASH_STMICRO
1122 	if (info->flags & E_FSR)
1123 		flash->flags |= SNOR_F_USE_FSR;
1124 #endif
1125 
1126 	/* Configure the BAR - discover bank cmds and read current bank */
1127 #ifdef CONFIG_SPI_FLASH_BAR
1128 	ret = read_bar(flash, info);
1129 	if (ret < 0)
1130 		return ret;
1131 #endif
1132 
1133 #if CONFIG_IS_ENABLED(OF_CONTROL) && !CONFIG_IS_ENABLED(OF_PLATDATA)
1134 	ret = spi_flash_decode_fdt(flash);
1135 	if (ret) {
1136 		debug("SF: FDT decode error\n");
1137 		return -EINVAL;
1138 	}
1139 #endif
1140 
1141 #ifndef CONFIG_SPL_BUILD
1142 	printf("SF: Detected %s with page size ", flash->name);
1143 	print_size(flash->page_size, ", erase size ");
1144 	print_size(flash->erase_size, ", total ");
1145 	print_size(flash->size, "");
1146 	if (flash->memory_map)
1147 		printf(", mapped at %p", flash->memory_map);
1148 	puts("\n");
1149 #endif
1150 
1151 #ifndef CONFIG_SPI_FLASH_BAR
1152 	if (((flash->dual_flash == SF_SINGLE_FLASH) &&
1153 	     (flash->size > SPI_FLASH_16MB_BOUN)) ||
1154 	     ((flash->dual_flash > SF_SINGLE_FLASH) &&
1155 	     (flash->size > SPI_FLASH_16MB_BOUN << 1))) {
1156 		puts("SF: Warning - Only lower 16MiB accessible,");
1157 		puts(" Full access #define CONFIG_SPI_FLASH_BAR\n");
1158 	}
1159 #endif
1160 
1161 	return 0;
1162 }
1163