1 // SPDX-License-Identifier: GPL-2.0+
2 /*
3  * Copyright (c) 2014 Google, Inc
4  */
5 
6 #define LOG_CATEGORY UCLASS_SPI_FLASH
7 
8 #include <common.h>
9 #include <bootdev.h>
10 #include <dm.h>
11 #include <log.h>
12 #include <malloc.h>
13 #include <spi.h>
14 #include <spi_flash.h>
15 #include <asm/global_data.h>
16 #include <dm/device-internal.h>
17 #include <test/test.h>
18 #include "sf_internal.h"
19 
20 DECLARE_GLOBAL_DATA_PTR;
21 
spi_flash_read_dm(struct udevice * dev,u32 offset,size_t len,void * buf)22 int spi_flash_read_dm(struct udevice *dev, u32 offset, size_t len, void *buf)
23 {
24 	return log_ret(sf_get_ops(dev)->read(dev, offset, len, buf));
25 }
26 
spi_flash_write_dm(struct udevice * dev,u32 offset,size_t len,const void * buf)27 int spi_flash_write_dm(struct udevice *dev, u32 offset, size_t len,
28 		       const void *buf)
29 {
30 	return log_ret(sf_get_ops(dev)->write(dev, offset, len, buf));
31 }
32 
spi_flash_erase_dm(struct udevice * dev,u32 offset,size_t len)33 int spi_flash_erase_dm(struct udevice *dev, u32 offset, size_t len)
34 {
35 	return log_ret(sf_get_ops(dev)->erase(dev, offset, len));
36 }
37 
spl_flash_get_sw_write_prot(struct udevice * dev)38 int spl_flash_get_sw_write_prot(struct udevice *dev)
39 {
40 	struct dm_spi_flash_ops *ops = sf_get_ops(dev);
41 
42 	if (!ops->get_sw_write_prot)
43 		return -ENOSYS;
44 	return log_ret(ops->get_sw_write_prot(dev));
45 }
46 
47 /*
48  * TODO(sjg@chromium.org): This is an old-style function. We should remove
49  * it when all SPI flash drivers use dm
50  */
spi_flash_probe(unsigned int busnum,unsigned int cs,unsigned int max_hz,unsigned int spi_mode)51 struct spi_flash *spi_flash_probe(unsigned int busnum, unsigned int cs,
52 				  unsigned int max_hz, unsigned int spi_mode)
53 {
54 	struct spi_slave *slave;
55 	struct udevice *bus;
56 	char *str;
57 
58 #if defined(CONFIG_SPL_BUILD) && CONFIG_IS_ENABLED(USE_TINY_PRINTF)
59 	str = "spi_flash";
60 #else
61 	char name[30];
62 
63 	snprintf(name, sizeof(name), "spi_flash@%d:%d", busnum, cs);
64 	str = strdup(name);
65 #endif
66 
67 	if (_spi_get_bus_and_cs(busnum, cs, max_hz, spi_mode,
68 				"jedec_spi_nor", str, &bus, &slave))
69 		return NULL;
70 
71 	return dev_get_uclass_priv(slave->dev);
72 }
73 
spi_flash_probe_bus_cs(unsigned int busnum,unsigned int cs,struct udevice ** devp)74 int spi_flash_probe_bus_cs(unsigned int busnum, unsigned int cs,
75 			   struct udevice **devp)
76 {
77 	struct spi_slave *slave;
78 	struct udevice *bus;
79 	int ret;
80 
81 	ret = spi_get_bus_and_cs(busnum, cs, &bus, &slave);
82 	if (ret)
83 		return ret;
84 
85 	*devp = slave->dev;
86 	return 0;
87 }
88 
spi_flash_post_bind(struct udevice * dev)89 static int spi_flash_post_bind(struct udevice *dev)
90 {
91 	int ret;
92 
93 	if (CONFIG_IS_ENABLED(BOOTDEV_SPI_FLASH) && test_sf_bootdev_enabled()) {
94 		ret = bootdev_setup_for_dev(dev, "sf_bootdev");
95 		if (ret)
96 			return log_msg_ret("bd", ret);
97 	}
98 
99 #if defined(CONFIG_NEEDS_MANUAL_RELOC)
100 	struct dm_spi_flash_ops *ops = sf_get_ops(dev);
101 	static int reloc_done;
102 
103 	if (!reloc_done) {
104 		if (ops->read)
105 			ops->read += gd->reloc_off;
106 		if (ops->write)
107 			ops->write += gd->reloc_off;
108 		if (ops->erase)
109 			ops->erase += gd->reloc_off;
110 
111 		reloc_done++;
112 	}
113 #endif
114 
115 	return 0;
116 }
117 
118 UCLASS_DRIVER(spi_flash) = {
119 	.id		= UCLASS_SPI_FLASH,
120 	.name		= "spi_flash",
121 	.post_bind	= spi_flash_post_bind,
122 	.per_device_auto	= sizeof(struct spi_nor),
123 };
124