drivers/mtd/spi/sf_probe.c

   1 /*
   2  * SPI flash probing
   3  *
   4  * Copyright (C) 2008 Atmel Corporation
   5  * Copyright (C) 2010 Reinhard Meyer, EMK Elektronik
   6  * Copyright (C) 2013 Jagannadha Sutradharudu Teki, Xilinx Inc.
   7  *
   8  * SPDX-License-Identifier:     GPL-2.0+
   9  */
  10
  11 #include <common.h>
  12 #include <dm.h>
  13 #include <errno.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  */
  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_flash_scan(flash);
  45         if (ret) {
  46                 ret = -EINVAL;
  47                 goto err_read_id;
  48         }
  49
  50 #ifdef CONFIG_SPI_FLASH_MTD
  51         ret = spi_flash_mtd_register(flash);
  52 #endif
  53
  54 err_read_id:
  55         spi_release_bus(spi);
  56         return ret;
  57 }
  58
  59 #ifndef CONFIG_DM_SPI_FLASH
  60 static struct spi_flash *spi_flash_probe_tail(struct spi_slave *bus)
  61 {
  62         struct spi_flash *flash;
  63
  64         /* Allocate space if needed (not used by sf-uclass */
  65         flash = calloc(1, sizeof(*flash));
  66         if (!flash) {
  67                 debug("SF: Failed to allocate spi_flash\n");
  68                 return NULL;
  69         }
  70
  71         flash->spi = bus;
  72         if (spi_flash_probe_slave(flash)) {
  73                 spi_free_slave(bus);
  74                 free(flash);
  75                 return NULL;
  76         }
  77
  78         return flash;
  79 }
  80
  81 struct spi_flash *spi_flash_probe(unsigned int busnum, unsigned int cs,
  82                 unsigned int max_hz, unsigned int spi_mode)
  83 {
  84         struct spi_slave *bus;
  85
  86         bus = spi_setup_slave(busnum, cs, max_hz, spi_mode);
  87         if (!bus)
  88                 return NULL;
  89         return spi_flash_probe_tail(bus);
  90 }
  91
  92 #ifdef CONFIG_OF_SPI_FLASH
  93 struct spi_flash *spi_flash_probe_fdt(const void *blob, int slave_node,
  94                                       int spi_node)
  95 {
  96         struct spi_slave *bus;
  97
  98         bus = spi_setup_slave_fdt(blob, slave_node, spi_node);
  99         if (!bus)
 100                 return NULL;
 101         return spi_flash_probe_tail(bus);
 102 }
 103 #endif
 104
 105 void spi_flash_free(struct spi_flash *flash)
 106 {
 107 #ifdef CONFIG_SPI_FLASH_MTD
 108         spi_flash_mtd_unregister();
 109 #endif
 110         spi_free_slave(flash->spi);
 111         free(flash);
 112 }
 113
 114 #else /* defined CONFIG_DM_SPI_FLASH */
 115
 116 static int spi_flash_std_read(struct udevice *dev, u32 offset, size_t len,
 117                               void *buf)
 118 {
 119         struct spi_flash *flash = dev_get_uclass_priv(dev);
 120
 121         return spi_flash_cmd_read_ops(flash, offset, len, buf);
 122 }
 123
 124 static int spi_flash_std_write(struct udevice *dev, u32 offset, size_t len,
 125                         const void *buf)
 126 {
 127         struct spi_flash *flash = dev_get_uclass_priv(dev);
 128
 129 #if defined(CONFIG_SPI_FLASH_SST)
 130         if (flash->flags & SNOR_F_SST_WR) {
 131                 if (flash->spi->mode & SPI_TX_BYTE)
 132                         return sst_write_bp(flash, offset, len, buf);
 133                 else
 134                         return sst_write_wp(flash, offset, len, buf);
 135         }
 136 #endif
 137
 138         return spi_flash_cmd_write_ops(flash, offset, len, buf);
 139 }
 140
 141 static int spi_flash_std_erase(struct udevice *dev, u32 offset, size_t len)
 142 {
 143         struct spi_flash *flash = dev_get_uclass_priv(dev);
 144
 145         return spi_flash_cmd_erase_ops(flash, offset, len);
 146 }
 147
 148 static int spi_flash_std_probe(struct udevice *dev)
 149 {
 150         struct spi_slave *slave = dev_get_parent_priv(dev);
 151         struct dm_spi_slave_platdata *plat = dev_get_parent_platdata(dev);
 152         struct spi_flash *flash;
 153
 154         flash = dev_get_uclass_priv(dev);
 155         flash->dev = dev;
 156         flash->spi = slave;
 157         debug("%s: slave=%p, cs=%d\n", __func__, slave, plat->cs);
 158         return spi_flash_probe_slave(flash);
 159 }
 160
 161 static const struct dm_spi_flash_ops spi_flash_std_ops = {
 162         .read = spi_flash_std_read,
 163         .write = spi_flash_std_write,
 164         .erase = spi_flash_std_erase,
 165 };
 166
 167 static const struct udevice_id spi_flash_std_ids[] = {
 168         { .compatible = "spi-flash" },
 169         { }
 170 };
 171
 172 U_BOOT_DRIVER(spi_flash_std) = {
 173         .name           = "spi_flash_std",
 174         .id             = UCLASS_SPI_FLASH,
 175         .of_match       = spi_flash_std_ids,
 176         .probe          = spi_flash_std_probe,
 177         .priv_auto_alloc_size = sizeof(struct spi_flash),
 178         .ops            = &spi_flash_std_ops,
 179 };
 180
 181 #endif /* CONFIG_DM_SPI_FLASH */