1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3 * SPI flash probing
4 *
5 * Copyright (C) 2008 Atmel Corporation
6 * Copyright (C) 2010 Reinhard Meyer, EMK Elektronik
7 * Copyright (C) 2013 Jagannadha Sutradharudu Teki, Xilinx Inc.
8 */
9
10 #include <common.h>
11 #include <dm.h>
12 #include <errno.h>
13 #include <log.h>
14 #include <malloc.h>
15 #include <spi.h>
16 #include <spi_flash.h>
17
18 #include "sf_internal.h"
19
20 /**
21 * spi_flash_probe_slave() - Probe for a SPI flash device on a bus
22 *
23 * @flashp: Pointer to place to put flash info, which may be NULL if the
24 * space should be allocated
25 */
spi_flash_probe_slave(struct spi_flash * flash)26 static int spi_flash_probe_slave(struct spi_flash *flash)
27 {
28 struct spi_slave *spi = flash->spi;
29 int ret;
30
31 /* Setup spi_slave */
32 if (!spi) {
33 printf("SF: Failed to set up slave\n");
34 return -ENODEV;
35 }
36
37 /* Claim spi bus */
38 ret = spi_claim_bus(spi);
39 if (ret) {
40 debug("SF: Failed to claim SPI bus: %d\n", ret);
41 return ret;
42 }
43
44 ret = spi_nor_scan(flash);
45 if (ret)
46 goto err_read_id;
47
48 if (CONFIG_IS_ENABLED(SPI_FLASH_MTD))
49 ret = spi_flash_mtd_register(flash);
50
51 err_read_id:
52 spi_release_bus(spi);
53 return ret;
54 }
55
56 #if !CONFIG_IS_ENABLED(DM_SPI_FLASH)
spi_flash_probe(unsigned int busnum,unsigned int cs,unsigned int max_hz,unsigned int spi_mode)57 struct spi_flash *spi_flash_probe(unsigned int busnum, unsigned int cs,
58 unsigned int max_hz, unsigned int spi_mode)
59 {
60 struct spi_slave *bus;
61 struct spi_flash *flash;
62
63 bus = spi_setup_slave(busnum, cs, max_hz, spi_mode);
64 if (!bus)
65 return NULL;
66
67 /* Allocate space if needed (not used by sf-uclass */
68 flash = calloc(1, sizeof(*flash));
69 if (!flash) {
70 debug("SF: Failed to allocate spi_flash\n");
71 return NULL;
72 }
73
74 flash->spi = bus;
75 if (spi_flash_probe_slave(flash)) {
76 spi_free_slave(bus);
77 free(flash);
78 return NULL;
79 }
80
81 return flash;
82 }
83
spi_flash_free(struct spi_flash * flash)84 void spi_flash_free(struct spi_flash *flash)
85 {
86 if (CONFIG_IS_ENABLED(SPI_FLASH_MTD))
87 spi_flash_mtd_unregister();
88
89 spi_free_slave(flash->spi);
90 free(flash);
91 }
92
93 #else /* defined CONFIG_DM_SPI_FLASH */
94
spi_flash_std_read(struct udevice * dev,u32 offset,size_t len,void * buf)95 static int spi_flash_std_read(struct udevice *dev, u32 offset, size_t len,
96 void *buf)
97 {
98 struct spi_flash *flash = dev_get_uclass_priv(dev);
99 struct mtd_info *mtd = &flash->mtd;
100 size_t retlen;
101
102 return log_ret(mtd->_read(mtd, offset, len, &retlen, buf));
103 }
104
spi_flash_std_write(struct udevice * dev,u32 offset,size_t len,const void * buf)105 static int spi_flash_std_write(struct udevice *dev, u32 offset, size_t len,
106 const void *buf)
107 {
108 struct spi_flash *flash = dev_get_uclass_priv(dev);
109 struct mtd_info *mtd = &flash->mtd;
110 size_t retlen;
111
112 return mtd->_write(mtd, offset, len, &retlen, buf);
113 }
114
spi_flash_std_erase(struct udevice * dev,u32 offset,size_t len)115 static int spi_flash_std_erase(struct udevice *dev, u32 offset, size_t len)
116 {
117 struct spi_flash *flash = dev_get_uclass_priv(dev);
118 struct mtd_info *mtd = &flash->mtd;
119 struct erase_info instr;
120
121 if (offset % mtd->erasesize || len % mtd->erasesize) {
122 debug("SF: Erase offset/length not multiple of erase size\n");
123 return -EINVAL;
124 }
125
126 memset(&instr, 0, sizeof(instr));
127 instr.addr = offset;
128 instr.len = len;
129
130 return mtd->_erase(mtd, &instr);
131 }
132
spi_flash_std_probe(struct udevice * dev)133 int spi_flash_std_probe(struct udevice *dev)
134 {
135 struct spi_slave *slave = dev_get_parent_priv(dev);
136 struct spi_flash *flash;
137
138 flash = dev_get_uclass_priv(dev);
139 flash->dev = dev;
140 flash->spi = slave;
141 return spi_flash_probe_slave(flash);
142 }
143
spi_flash_std_remove(struct udevice * dev)144 static int spi_flash_std_remove(struct udevice *dev)
145 {
146 if (CONFIG_IS_ENABLED(SPI_FLASH_MTD))
147 spi_flash_mtd_unregister();
148
149 return 0;
150 }
151
152 static const struct dm_spi_flash_ops spi_flash_std_ops = {
153 .read = spi_flash_std_read,
154 .write = spi_flash_std_write,
155 .erase = spi_flash_std_erase,
156 };
157
158 static const struct udevice_id spi_flash_std_ids[] = {
159 { .compatible = "jedec,spi-nor" },
160 { }
161 };
162
163 U_BOOT_DRIVER(jedec_spi_nor) = {
164 .name = "jedec_spi_nor",
165 .id = UCLASS_SPI_FLASH,
166 .of_match = spi_flash_std_ids,
167 .probe = spi_flash_std_probe,
168 .remove = spi_flash_std_remove,
169 .priv_auto = sizeof(struct spi_nor),
170 .ops = &spi_flash_std_ops,
171 };
172
173 DM_DRIVER_ALIAS(jedec_spi_nor, spansion_m25p16)
174
175 #endif /* CONFIG_DM_SPI_FLASH */
176