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 
34 /*
35  * TODO(sjg@chromium.org): This is an old-style function. We should remove
36  * it when all SPI flash drivers use dm
37  */
spi_flash_probe(unsigned int bus,unsigned int cs,unsigned int max_hz,unsigned int spi_mode)38 struct spi_flash *spi_flash_probe(unsigned int bus, unsigned int cs,
39 				  unsigned int max_hz, unsigned int spi_mode)
40 {
41 	struct udevice *dev;
42 
43 	if (spi_flash_probe_bus_cs(bus, cs, max_hz, spi_mode, &dev))
44 		return NULL;
45 
46 	return dev_get_uclass_priv(dev);
47 }
48 
spi_flash_free(struct spi_flash * flash)49 void spi_flash_free(struct spi_flash *flash)
50 {
51 	device_remove(flash->spi->dev, DM_REMOVE_NORMAL);
52 }
53 
spi_flash_probe_bus_cs(unsigned int busnum,unsigned int cs,unsigned int max_hz,unsigned int spi_mode,struct udevice ** devp)54 int spi_flash_probe_bus_cs(unsigned int busnum, unsigned int cs,
55 			   unsigned int max_hz, unsigned int spi_mode,
56 			   struct udevice **devp)
57 {
58 	struct spi_slave *slave;
59 	struct udevice *bus;
60 	char *str;
61 	int ret;
62 
63 #if defined(CONFIG_SPL_BUILD) && CONFIG_IS_ENABLED(USE_TINY_PRINTF)
64 	str = "spi_flash";
65 #else
66 	char name[30];
67 
68 	snprintf(name, sizeof(name), "spi_flash@%d:%d", busnum, cs);
69 	str = strdup(name);
70 #endif
71 	ret = spi_get_bus_and_cs(busnum, cs, max_hz, spi_mode,
72 				  "jedec_spi_nor", str, &bus, &slave);
73 	if (ret)
74 		return ret;
75 
76 	*devp = slave->dev;
77 	return 0;
78 }
79 
spi_flash_post_bind(struct udevice * dev)80 static int spi_flash_post_bind(struct udevice *dev)
81 {
82 #if defined(CONFIG_NEEDS_MANUAL_RELOC)
83 	struct dm_spi_flash_ops *ops = sf_get_ops(dev);
84 	static int reloc_done;
85 
86 	if (!reloc_done) {
87 		if (ops->read)
88 			ops->read += gd->reloc_off;
89 		if (ops->write)
90 			ops->write += gd->reloc_off;
91 		if (ops->erase)
92 			ops->erase += gd->reloc_off;
93 
94 		reloc_done++;
95 	}
96 #endif
97 	return 0;
98 }
99 
100 UCLASS_DRIVER(spi_flash) = {
101 	.id		= UCLASS_SPI_FLASH,
102 	.name		= "spi_flash",
103 	.post_bind	= spi_flash_post_bind,
104 	.per_device_auto	= sizeof(struct spi_nor),
105 };
106