atcspi200: avoid possible NULL dereference

[u-boot] / drivers / spi / atcspi200_spi.c

diff --git a/drivers/spi/atcspi200_spi.c b/drivers/spi/atcspi200_spi.c
index 3e29df03a4da991c48c103a32afd7962d253324b..5b2e9d6264bd0f6e75447eae629d477193405494 100644 (file)
--- a/drivers/spi/atcspi200_spi.c
+++ b/drivers/spi/atcspi200_spi.c
@@ -297,6 +297,8 @@ struct spi_slave *spi_setup_slave(unsigned int bus, unsigned int cs,
                return NULL;
 
        ns = spi_alloc_slave(struct nds_spi_slave, bus, cs);
+       if (!ns)
+               return NULL;
 
        switch (bus) {
        case SPI0_BUS:
@@ -316,8 +318,6 @@ struct spi_slave *spi_setup_slave(unsigned int bus, unsigned int cs,
        ns->to = SPI_TIMEOUT;
        ns->max_transfer_length = MAX_TRANSFER_LEN;
        ns->slave.max_write_size = MAX_TRANSFER_LEN;
-       if (!ns)
-               return NULL;
 
        return &ns->slave;
 }