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