/*
 * SPI flash probing
 *
 * Copyright (C) 2008 Atmel Corporation
 * Copyright (C) 2010 Reinhard Meyer, EMK Elektronik
 * Copyright (C) 2013 Jagannadha Sutradharudu Teki, Xilinx Inc.
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <dm.h>
#include <errno.h>
#include <malloc.h>
#include <spi.h>
#include <spi_flash.h>

#include "sf_internal.h"

/**
 * spi_flash_probe_slave() - Probe for a SPI flash device on a bus
 *
 * @flashp: Pointer to place to put flash info, which may be NULL if the
 * space should be allocated
 */
static int spi_flash_probe_slave(struct spi_flash *flash)
{
        struct spi_slave *spi = flash->spi;
        int ret;

        /* Setup spi_slave */
        if (!spi) {
                printf("SF: Failed to set up slave\n");
                return -ENODEV;
        }

        /* Claim spi bus */
        ret = spi_claim_bus(spi);
        if (ret) {
                debug("SF: Failed to claim SPI bus: %d\n", ret);
                return ret;
        }
#if 0
        if (spi->option == SF_DUAL_PARALLEL_FLASH)
                spi->flags |= SPI_XFER_LOWER;
        /* Read the ID codes */
        ret = spi_flash_cmd(spi, CMD_READ_ID, idcode, sizeof(idcode));
        if (ret) {
                printf("SF: Failed to get idcodes\n");
                goto err_read_id;
        }

#ifdef CONFIG_SPI_GENERIC
        if (spi->option == SF_DUAL_PARALLEL_FLASH) {
                spi->flags |= SPI_XFER_UPPER;
                ret = spi_flash_cmd(spi, CMD_READ_ID, idcode_up,
                                    sizeof(idcode_up));
                if (ret) {
                        printf("SF: Failed to get idcodes\n");
                        goto err_read_id;
                }
                for (i = 0; i < sizeof(idcode); i++) {
                        if (idcode[i] != idcode_up[i]) {
                                printf("SF: Failed to get same idcodes\n");
                                goto err_read_id;
                        }
                }
        }
#endif
#ifdef DEBUG
        printf("SF: Got idcodes\n");
        print_buffer(0, idcode, 1, sizeof(idcode), 0);
#endif

        if (spi_flash_validate_params(spi, idcode, flash)) {

#endif

        ret = spi_flash_scan(flash);
        if (ret)
                goto err_read_id;

#ifdef CONFIG_SPI_FLASH_MTD
        ret = spi_flash_mtd_register(flash);
#endif

err_read_id:
        spi_release_bus(spi);
        return ret;
}

#ifndef CONFIG_DM_SPI_FLASH
static struct spi_flash *spi_flash_probe_tail(struct spi_slave *bus)
{
        struct spi_flash *flash;

        /* Allocate space if needed (not used by sf-uclass */
        flash = calloc(1, sizeof(*flash));
        if (!flash) {
                debug("SF: Failed to allocate spi_flash\n");
                return NULL;
        }

        flash->spi = bus;
        if (spi_flash_probe_slave(flash)) {
                spi_free_slave(bus);
                free(flash);
                return NULL;
        }

        return flash;
}

struct spi_flash *spi_flash_probe(unsigned int busnum, unsigned int cs,
                unsigned int max_hz, unsigned int spi_mode)
{
        struct spi_slave *bus;

        bus = spi_setup_slave(busnum, cs, max_hz, spi_mode);
        if (!bus)
                return NULL;
        return spi_flash_probe_tail(bus);
}

#ifdef CONFIG_OF_SPI_FLASH
struct spi_flash *spi_flash_probe_fdt(const void *blob, int slave_node,
                                      int spi_node)
{
        struct spi_slave *bus;

        bus = spi_setup_slave_fdt(blob, slave_node, spi_node);
        if (!bus)
                return NULL;
        return spi_flash_probe_tail(bus);
}
#endif

void spi_flash_free(struct spi_flash *flash)
{
#ifdef CONFIG_SPI_FLASH_MTD
        spi_flash_mtd_unregister();
#endif
        spi_free_slave(flash->spi);
        free(flash);
}

#else /* defined CONFIG_DM_SPI_FLASH */

static int spi_flash_std_read(struct udevice *dev, u32 offset, size_t len,
                              void *buf)
{
        struct spi_flash *flash = dev_get_uclass_priv(dev);

        return spi_flash_cmd_read_ops(flash, offset, len, buf);
}

static int spi_flash_std_write(struct udevice *dev, u32 offset, size_t len,
                        const void *buf)
{
        struct spi_flash *flash = dev_get_uclass_priv(dev);

#if defined(CONFIG_SPI_FLASH_SST)
        if (flash->flags & SNOR_F_SST_WR) {
                if (flash->spi->mode & SPI_TX_BYTE)
                        return sst_write_bp(flash, offset, len, buf);
                else
                        return sst_write_wp(flash, offset, len, buf);
        }
#endif

        return spi_flash_cmd_write_ops(flash, offset, len, buf);
}

static int spi_flash_std_erase(struct udevice *dev, u32 offset, size_t len)
{
        struct spi_flash *flash = dev_get_uclass_priv(dev);

        return spi_flash_cmd_erase_ops(flash, offset, len);
}

static int spi_flash_std_probe(struct udevice *dev)
{
        struct spi_slave *slave = dev_get_parent_priv(dev);
        struct dm_spi_slave_platdata *plat = dev_get_parent_platdata(dev);
        struct spi_flash *flash;

        flash = dev_get_uclass_priv(dev);
        flash->dev = dev;
        flash->spi = slave;
        debug("%s: slave=%p, cs=%d\n", __func__, slave, plat->cs);
        return spi_flash_probe_slave(flash);
}

static const struct dm_spi_flash_ops spi_flash_std_ops = {
        .read = spi_flash_std_read,
        .write = spi_flash_std_write,
        .erase = spi_flash_std_erase,
};

static const struct udevice_id spi_flash_std_ids[] = {
        { .compatible = "spi-flash" },
        { }
};

U_BOOT_DRIVER(spi_flash_std) = {
        .name           = "spi_flash_std",
        .id             = UCLASS_SPI_FLASH,
        .of_match       = spi_flash_std_ids,
        .probe          = spi_flash_std_probe,
        .priv_auto_alloc_size = sizeof(struct spi_flash),
        .ops            = &spi_flash_std_ops,
};

#endif /* CONFIG_DM_SPI_FLASH */