Some coding style and consistency fixes (trivial).
[coreboot.git] / util / flashrom / spi.c
index 7fdad82c4910f35691ebf9697842645e7cdb8ab0..036c836d1ade34d92ed4a76e74bdc387c0a093c6 100644 (file)
 #include "flash.h"
 #include "spi.h"
 
-
 void spi_prettyprint_status_register(struct flashchip *flash);
 
-int spi_command(unsigned int writecnt, unsigned int readcnt, const unsigned char *writearr, unsigned char *readarr)
+int spi_command(unsigned int writecnt, unsigned int readcnt,
+               const unsigned char *writearr, unsigned char *readarr)
 {
-       if (it8716f_flashport)
-               return it8716f_spi_command(writecnt, readcnt, writearr, readarr);
-       else if ((ich7_detected) || (viaspi_detected))
-               return ich_spi_command(writecnt, readcnt, writearr, readarr);
-       else if (ich9_detected)
+       switch (flashbus) {
+       case BUS_TYPE_IT87XX_SPI:
+               return it8716f_spi_command(writecnt, readcnt, writearr,
+                                          readarr);
+       case BUS_TYPE_ICH7_SPI:
+       case BUS_TYPE_ICH9_SPI:
+       case BUS_TYPE_VIA_SPI:
                return ich_spi_command(writecnt, readcnt, writearr, readarr);
-       printf_debug("%s called, but no SPI chipset detected\n", __FUNCTION__);
+       case BUS_TYPE_SB600_SPI:
+               return sb600_spi_command(writecnt, readcnt, writearr, readarr);
+       case BUS_TYPE_WBSIO_SPI:
+               return wbsio_spi_command(writecnt, readcnt, writearr, readarr);
+       default:
+               printf_debug
+                   ("%s called, but no SPI chipset/strapping detected\n",
+                    __FUNCTION__);
+       }
        return 1;
 }
 
 static int spi_rdid(unsigned char *readarr, int bytes)
 {
-       const unsigned char cmd[JEDEC_RDID_OUTSIZE] = {JEDEC_RDID};
+       const unsigned char cmd[JEDEC_RDID_OUTSIZE] = { JEDEC_RDID };
+
+       if (spi_command(sizeof(cmd), bytes, cmd, readarr))
+               return 1;
+       printf_debug("RDID returned %02x %02x %02x.\n", readarr[0], readarr[1],
+                    readarr[2]);
+       return 0;
+}
+
+static int spi_rems(unsigned char *readarr)
+{
+       const unsigned char cmd[JEDEC_REMS_OUTSIZE] = { JEDEC_REMS, 0, 0, 0 };
 
-       if (spi_command(JEDEC_RDID_OUTSIZE, bytes, cmd, readarr))
+       if (spi_command(sizeof(cmd), JEDEC_REMS_INSIZE, cmd, readarr))
                return 1;
-       printf_debug("RDID returned %02x %02x %02x.\n", readarr[0], readarr[1], readarr[2]);
+       printf_debug("REMS returned %02x %02x.\n", readarr[0], readarr[1]);
        return 0;
 }
 
 static int spi_res(unsigned char *readarr)
 {
-       const unsigned char cmd[JEDEC_RES_OUTSIZE] = {JEDEC_RES, 0, 0, 0};
+       const unsigned char cmd[JEDEC_RES_OUTSIZE] = { JEDEC_RES, 0, 0, 0 };
 
-       if (spi_command(JEDEC_RES_OUTSIZE, JEDEC_RES_INSIZE, cmd, readarr))
+       if (spi_command(sizeof(cmd), JEDEC_RES_INSIZE, cmd, readarr))
                return 1;
        printf_debug("RES returned %02x.\n", readarr[0]);
        return 0;
 }
 
-void spi_write_enable()
+int spi_write_enable(void)
 {
-       const unsigned char cmd[JEDEC_WREN_OUTSIZE] = {JEDEC_WREN};
+       const unsigned char cmd[JEDEC_WREN_OUTSIZE] = { JEDEC_WREN };
 
        /* Send WREN (Write Enable) */
-       spi_command(JEDEC_WREN_OUTSIZE, JEDEC_WREN_INSIZE, cmd, NULL);
+       return spi_command(sizeof(cmd), 0, cmd, NULL);
 }
 
-void spi_write_disable()
+int spi_write_disable(void)
 {
-       const unsigned char cmd[JEDEC_WRDI_OUTSIZE] = {JEDEC_WRDI};
+       const unsigned char cmd[JEDEC_WRDI_OUTSIZE] = { JEDEC_WRDI };
 
        /* Send WRDI (Write Disable) */
-       spi_command(JEDEC_WRDI_OUTSIZE, JEDEC_WRDI_INSIZE, cmd, NULL);
+       return spi_command(sizeof(cmd), 0, cmd, NULL);
 }
 
 static int probe_spi_rdid_generic(struct flashchip *flash, int bytes)
@@ -107,10 +128,10 @@ static int probe_spi_rdid_generic(struct flashchip *flash, int bytes)
                model_id = (readarr[1] << 8) | readarr[2];
        }
 
-       printf_debug("%s: id1 0x%x, id2 0x%x\n", __FUNCTION__, manuf_id, model_id);
+       printf_debug("%s: id1 0x%02x, id2 0x%02x\n", __FUNCTION__, manuf_id,
+                    model_id);
 
-       if (manuf_id == flash->manufacture_id &&
-           model_id == flash->model_id) {
+       if (manuf_id == flash->manufacture_id && model_id == flash->model_id) {
                /* Print the status register to tell the
                 * user about possible write protection.
                 */
@@ -127,17 +148,58 @@ static int probe_spi_rdid_generic(struct flashchip *flash, int bytes)
        return 0;
 }
 
-int probe_spi_rdid(struct flashchip *flash) {
+int probe_spi_rdid(struct flashchip *flash)
+{
        return probe_spi_rdid_generic(flash, 3);
 }
 
 /* support 4 bytes flash ID */
-int probe_spi_rdid4(struct flashchip *flash) {
-
+int probe_spi_rdid4(struct flashchip *flash)
+{
        /* only some SPI chipsets support 4 bytes commands */
-       if (!((ich7_detected) || (ich9_detected) || (viaspi_detected)))
+       switch (flashbus) {
+       case BUS_TYPE_ICH7_SPI:
+       case BUS_TYPE_ICH9_SPI:
+       case BUS_TYPE_VIA_SPI:
+       case BUS_TYPE_SB600_SPI:
+       case BUS_TYPE_WBSIO_SPI:
+               return probe_spi_rdid_generic(flash, 4);
+       default:
+               printf_debug("4b ID not supported on this SPI controller\n");
+       }
+
+       return 0;
+}
+
+int probe_spi_rems(struct flashchip *flash)
+{
+       unsigned char readarr[JEDEC_REMS_INSIZE];
+       uint32_t manuf_id, model_id;
+
+       if (spi_rems(readarr))
                return 0;
-       return probe_spi_rdid_generic(flash, 4);
+
+       manuf_id = readarr[0];
+       model_id = readarr[1];
+
+       printf_debug("%s: id1 0x%x, id2 0x%x\n", __FUNCTION__, manuf_id,
+                    model_id);
+
+       if (manuf_id == flash->manufacture_id && model_id == flash->model_id) {
+               /* Print the status register to tell the
+                * user about possible write protection.
+                */
+               spi_prettyprint_status_register(flash);
+
+               return 1;
+       }
+
+       /* Test if this is a pure vendor match. */
+       if (manuf_id == flash->manufacture_id &&
+           GENERIC_DEVICE_ID == flash->model_id)
+               return 1;
+
+       return 0;
 }
 
 int probe_spi_res(struct flashchip *flash)
@@ -145,13 +207,11 @@ int probe_spi_res(struct flashchip *flash)
        unsigned char readarr[3];
        uint32_t model_id;
 
-       if (spi_rdid(readarr, 3))
-               /* We couldn't issue RDID, it's pointless to try RES. */
-               return 0;
-
-       /* Check if RDID returns 0xff 0xff 0xff, then we use RES. */
-       if ((readarr[0] != 0xff) || (readarr[1] != 0xff) ||
-           (readarr[2] != 0xff))
+       /* Check if RDID was successful and did not return 0xff 0xff 0xff.
+        * In that case, RES is pointless.
+        */
+       if (!spi_rdid(readarr, 3) && ((readarr[0] != 0xff) ||
+           (readarr[1] != 0xff) || (readarr[2] != 0xff)))
                return 0;
 
        if (spi_res(readarr))
@@ -169,32 +229,37 @@ int probe_spi_res(struct flashchip *flash)
        return 1;
 }
 
-uint8_t spi_read_status_register()
+uint8_t spi_read_status_register(void)
 {
-       const unsigned char cmd[JEDEC_RDSR_OUTSIZE] = {JEDEC_RDSR};
-       unsigned char readarr[1];
+       const unsigned char cmd[JEDEC_RDSR_OUTSIZE] = { JEDEC_RDSR };
+       unsigned char readarr[2]; /* JEDEC_RDSR_INSIZE=1 but wbsio needs 2 */
 
        /* Read Status Register */
-       spi_command(JEDEC_RDSR_OUTSIZE, JEDEC_RDSR_INSIZE, cmd, readarr);
+       if (flashbus == BUS_TYPE_SB600_SPI) {
+               /* SB600 uses a different way to read status register. */
+               return sb600_read_status_register();
+       } else {
+               spi_command(sizeof(cmd), sizeof(readarr), cmd, readarr);
+       }
+
        return readarr[0];
 }
 
-/* Prettyprint the status register. Common definitions.
- */
+/* Prettyprint the status register. Common definitions. */
 void spi_prettyprint_status_register_common(uint8_t status)
 {
        printf_debug("Chip status register: Bit 5 / Block Protect 3 (BP3) is "
-               "%sset\n", (status & (1 << 5)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 5)) ? "" : "not ");
        printf_debug("Chip status register: Bit 4 / Block Protect 2 (BP2) is "
-               "%sset\n", (status & (1 << 4)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 4)) ? "" : "not ");
        printf_debug("Chip status register: Bit 3 / Block Protect 1 (BP1) is "
-               "%sset\n", (status & (1 << 3)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 3)) ? "" : "not ");
        printf_debug("Chip status register: Bit 2 / Block Protect 0 (BP0) is "
-               "%sset\n", (status & (1 << 2)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 2)) ? "" : "not ");
        printf_debug("Chip status register: Write Enable Latch (WEL) is "
-               "%sset\n", (status & (1 << 1)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 1)) ? "" : "not ");
        printf_debug("Chip status register: Write In Progress (WIP/BUSY) is "
-               "%sset\n", (status & (1 << 0)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 0)) ? "" : "not ");
 }
 
 /* Prettyprint the status register. Works for
@@ -204,9 +269,9 @@ void spi_prettyprint_status_register_common(uint8_t status)
 void spi_prettyprint_status_register_st_m25p(uint8_t status)
 {
        printf_debug("Chip status register: Status Register Write Disable "
-               "(SRWD) is %sset\n", (status & (1 << 7)) ? "" : "not ");
+                    "(SRWD) is %sset\n", (status & (1 << 7)) ? "" : "not ");
        printf_debug("Chip status register: Bit 6 is "
-               "%sset\n", (status & (1 << 6)) ? "" : "not ");
+                    "%sset\n", (status & (1 << 6)) ? "" : "not ");
        spi_prettyprint_status_register_common(status);
 }
 
@@ -224,13 +289,31 @@ void spi_prettyprint_status_register_sst25vf016(uint8_t status)
                "100000H-1FFFFFH",
                "all", "all"
        };
+       printf_debug("Chip status register: Block Protect Write Disable "
+                    "(BPL) is %sset\n", (status & (1 << 7)) ? "" : "not ");
+       printf_debug("Chip status register: Auto Address Increment Programming "
+                    "(AAI) is %sset\n", (status & (1 << 6)) ? "" : "not ");
+       spi_prettyprint_status_register_common(status);
+       printf_debug("Resulting block protection : %s\n",
+                    bpt[(status & 0x1c) >> 2]);
+}
+
+void spi_prettyprint_status_register_sst25vf040b(uint8_t status)
+{
+       const char *bpt[] = {
+               "none",
+               "0x70000-0x7ffff",
+               "0x60000-0x7ffff",
+               "0x40000-0x7ffff",
+               "all blocks", "all blocks", "all blocks", "all blocks"
+       };
        printf_debug("Chip status register: Block Protect Write Disable "
                "(BPL) is %sset\n", (status & (1 << 7)) ? "" : "not ");
        printf_debug("Chip status register: Auto Address Increment Programming "
                "(AAI) is %sset\n", (status & (1 << 6)) ? "" : "not ");
        spi_prettyprint_status_register_common(status);
        printf_debug("Resulting block protection : %s\n",
-               bpt[(status & 0x1c) >> 2]);
+               bpt[(status & 0x3c) >> 2]);
 }
 
 void spi_prettyprint_status_register(struct flashchip *flash)
@@ -250,28 +333,108 @@ void spi_prettyprint_status_register(struct flashchip *flash)
                        spi_prettyprint_status_register_st_m25p(status);
                break;
        case SST_ID:
-               if (flash->model_id == SST_25VF016B)
+               switch (flash->model_id) {
+               case 0x2541:
                        spi_prettyprint_status_register_sst25vf016(status);
+                       break;
+               case 0x8d:
+               case 0x258d:
+                       spi_prettyprint_status_register_sst25vf040b(status);
+                       break;
+               }
                break;
        }
 }
+
+int spi_chip_erase_60(struct flashchip *flash)
+{
+       const unsigned char cmd[JEDEC_CE_60_OUTSIZE] = {JEDEC_CE_60};
+       int result;
        
+       result = spi_disable_blockprotect();
+       if (result) {
+               printf_debug("spi_disable_blockprotect failed\n");
+               return result;
+       }
+       result = spi_write_enable();
+       if (result) {
+               printf_debug("spi_write_enable failed\n");
+               return result;
+       }
+       /* Send CE (Chip Erase) */
+       result = spi_command(sizeof(cmd), 0, cmd, NULL);
+       if (result) {
+               printf_debug("spi_chip_erase_60 failed sending erase\n");
+               return result;
+       }
+       /* Wait until the Write-In-Progress bit is cleared.
+        * This usually takes 1-85 s, so wait in 1 s steps.
+        */
+       /* FIXME: We assume spi_read_status_register will never fail. */
+       while (spi_read_status_register() & JEDEC_RDSR_BIT_WIP)
+               sleep(1);
+       return 0;
+}
+
 int spi_chip_erase_c7(struct flashchip *flash)
 {
-       const unsigned char cmd[JEDEC_CE_C7_OUTSIZE] = {JEDEC_CE_C7};
-       
-       spi_disable_blockprotect();
-       spi_write_enable();
+       const unsigned char cmd[JEDEC_CE_C7_OUTSIZE] = { JEDEC_CE_C7 };
+       int result;
+
+       result = spi_disable_blockprotect();
+       if (result) {
+               printf_debug("spi_disable_blockprotect failed\n");
+               return result;
+       }
+       result = spi_write_enable();
+       if (result) {
+               printf_debug("spi_write_enable failed\n");
+               return result;
+       }
        /* Send CE (Chip Erase) */
-       spi_command(JEDEC_CE_C7_OUTSIZE, JEDEC_CE_C7_INSIZE, cmd, NULL);
+       result = spi_command(sizeof(cmd), 0, cmd, NULL);
+       if (result) {
+               printf_debug("spi_chip_erase_60 failed sending erase\n");
+               return result;
+       }
        /* Wait until the Write-In-Progress bit is cleared.
         * This usually takes 1-85 s, so wait in 1 s steps.
         */
+       /* FIXME: We assume spi_read_status_register will never fail. */
        while (spi_read_status_register() & JEDEC_RDSR_BIT_WIP)
                sleep(1);
        return 0;
 }
 
+int spi_chip_erase_60_c7(struct flashchip *flash)
+{
+       int result;
+       result = spi_chip_erase_60(flash);
+       if (result) {
+               printf_debug("spi_chip_erase_60 failed, trying c7\n");
+               result = spi_chip_erase_c7(flash);
+       }
+       return result;
+}
+
+int spi_block_erase_52(const struct flashchip *flash, unsigned long addr)
+{
+       unsigned char cmd[JEDEC_BE_52_OUTSIZE] = {JEDEC_BE_52};
+
+       cmd[1] = (addr & 0x00ff0000) >> 16;
+       cmd[2] = (addr & 0x0000ff00) >> 8;
+       cmd[3] = (addr & 0x000000ff);
+       spi_write_enable();
+       /* Send BE (Block Erase) */
+       spi_command(sizeof(cmd), 0, cmd, NULL);
+       /* Wait until the Write-In-Progress bit is cleared.
+        * This usually takes 100-4000 ms, so wait in 100 ms steps.
+        */
+       while (spi_read_status_register() & JEDEC_RDSR_BIT_WIP)
+               usleep(100 * 1000);
+       return 0;
+}
+
 /* Block size is usually
  * 64k for Macronix
  * 32k for SST
@@ -279,14 +442,14 @@ int spi_chip_erase_c7(struct flashchip *flash)
  */
 int spi_block_erase_d8(const struct flashchip *flash, unsigned long addr)
 {
-       unsigned char cmd[JEDEC_BE_D8_OUTSIZE] = {JEDEC_BE_D8};
+       unsigned char cmd[JEDEC_BE_D8_OUTSIZE] = { JEDEC_BE_D8 };
 
        cmd[1] = (addr & 0x00ff0000) >> 16;
        cmd[2] = (addr & 0x0000ff00) >> 8;
        cmd[3] = (addr & 0x000000ff);
        spi_write_enable();
        /* Send BE (Block Erase) */
-       spi_command(JEDEC_BE_D8_OUTSIZE, JEDEC_BE_D8_INSIZE, cmd, NULL);
+       spi_command(sizeof(cmd), 0, cmd, NULL);
        /* Wait until the Write-In-Progress bit is cleared.
         * This usually takes 100-4000 ms, so wait in 100 ms steps.
         */
@@ -295,17 +458,40 @@ int spi_block_erase_d8(const struct flashchip *flash, unsigned long addr)
        return 0;
 }
 
+int spi_chip_erase_d8(struct flashchip *flash)
+{
+       int i, rc = 0;
+       int total_size = flash->total_size * 1024;
+       int erase_size = 64 * 1024;
+
+       spi_disable_blockprotect();
+
+       printf("Erasing chip: \n");
+
+       for (i = 0; i < total_size / erase_size; i++) {
+               rc = spi_block_erase_d8(flash, i * erase_size);
+               if (rc) {
+                       printf("Error erasing block at 0x%x\n", i);
+                       break;
+               }
+       }
+
+       printf("\n");
+
+       return rc;
+}
+
 /* Sector size is usually 4k, though Macronix eliteflash has 64k */
 int spi_sector_erase(const struct flashchip *flash, unsigned long addr)
 {
-       unsigned char cmd[JEDEC_SE_OUTSIZE] = {JEDEC_SE};
+       unsigned char cmd[JEDEC_SE_OUTSIZE] = { JEDEC_SE };
        cmd[1] = (addr & 0x00ff0000) >> 16;
        cmd[2] = (addr & 0x0000ff00) >> 8;
        cmd[3] = (addr & 0x000000ff);
 
        spi_write_enable();
        /* Send SE (Sector Erase) */
-       spi_command(JEDEC_SE_OUTSIZE, JEDEC_SE_INSIZE, cmd, NULL);
+       spi_command(sizeof(cmd), 0, cmd, NULL);
        /* Wait until the Write-In-Progress bit is cleared.
         * This usually takes 15-800 ms, so wait in 10 ms steps.
         */
@@ -314,86 +500,145 @@ int spi_sector_erase(const struct flashchip *flash, unsigned long addr)
        return 0;
 }
 
-void spi_page_program(int block, uint8_t *buf, uint8_t *bios)
+int spi_write_status_enable(void)
 {
-       if (it8716f_flashport) {
-               it8716f_spi_page_program(block, buf, bios);
-               return;
-       }
-       printf_debug("%s called, but no SPI chipset detected\n", __FUNCTION__);
+       const unsigned char cmd[JEDEC_EWSR_OUTSIZE] = { JEDEC_EWSR };
+
+       /* Send EWSR (Enable Write Status Register). */
+       return spi_command(JEDEC_EWSR_OUTSIZE, JEDEC_EWSR_INSIZE, cmd, NULL);
 }
 
 /*
  * This is according the SST25VF016 datasheet, who knows it is more
  * generic that this...
  */
-void spi_write_status_register(int status)
+int spi_write_status_register(int status)
 {
-       const unsigned char cmd[JEDEC_WRSR_OUTSIZE] = {JEDEC_WRSR, (unsigned char)status};
+       const unsigned char cmd[JEDEC_WRSR_OUTSIZE] =
+           { JEDEC_WRSR, (unsigned char)status };
 
        /* Send WRSR (Write Status Register) */
-       spi_command(JEDEC_WRSR_OUTSIZE, JEDEC_WRSR_INSIZE, cmd, NULL);
+       return spi_command(sizeof(cmd), 0, cmd, NULL);
 }
 
 void spi_byte_program(int address, uint8_t byte)
 {
-       const unsigned char cmd[JEDEC_BYTE_PROGRAM_OUTSIZE] = {JEDEC_BYTE_PROGRAM,
-               (address>>16)&0xff,
-               (address>>8)&0xff,
-               (address>>0)&0xff,
+       const unsigned char cmd[JEDEC_BYTE_PROGRAM_OUTSIZE] = {
+               JEDEC_BYTE_PROGRAM,
+               (address >> 16) & 0xff,
+               (address >> 8) & 0xff,
+               (address >> 0) & 0xff,
                byte
        };
 
        /* Send Byte-Program */
-       spi_command(JEDEC_BYTE_PROGRAM_OUTSIZE, JEDEC_BYTE_PROGRAM_INSIZE, cmd, NULL);
+       spi_command(sizeof(cmd), 0, cmd, NULL);
 }
 
-void spi_disable_blockprotect(void)
+int spi_disable_blockprotect(void)
 {
        uint8_t status;
+       int result;
 
        status = spi_read_status_register();
        /* If there is block protection in effect, unprotect it first. */
        if ((status & 0x3c) != 0) {
                printf_debug("Some block protection in effect, disabling\n");
-               spi_write_enable();
-               spi_write_status_register(status & ~0x3c);
+               result = spi_write_status_enable();
+               if (result) {
+                       printf_debug("spi_write_status_enable failed\n");
+                       return result;
+               }
+               result = spi_write_status_register(status & ~0x3c);
+               if (result) {
+                       printf_debug("spi_write_status_register failed\n");
+                       return result;
+               }
        }
+       return 0;
 }
 
-void spi_nbyte_read(int address, uint8_t *bytes, int len)
+int spi_nbyte_read(int address, uint8_t *bytes, int len)
 {
-       const unsigned char cmd[JEDEC_READ_OUTSIZE] = {JEDEC_READ,
+       const unsigned char cmd[JEDEC_READ_OUTSIZE] = {
+               JEDEC_READ,
                (address >> 16) & 0xff,
                (address >> 8) & 0xff,
                (address >> 0) & 0xff,
        };
 
        /* Send Read */
-       spi_command(JEDEC_READ_OUTSIZE, len, cmd, bytes);
+       return spi_command(sizeof(cmd), len, cmd, bytes);
 }
 
 int spi_chip_read(struct flashchip *flash, uint8_t *buf)
 {
-       if (it8716f_flashport)
+       switch (flashbus) {
+       case BUS_TYPE_IT87XX_SPI:
                return it8716f_spi_chip_read(flash, buf);
-       else if ((ich7_detected) || (viaspi_detected))
+       case BUS_TYPE_SB600_SPI:
+               return sb600_spi_read(flash, buf);
+       case BUS_TYPE_ICH7_SPI:
+       case BUS_TYPE_ICH9_SPI:
+       case BUS_TYPE_VIA_SPI:
                return ich_spi_read(flash, buf);
-       else if (ich9_detected)
-               return ich_spi_read(flash, buf);
-       printf_debug("%s called, but no SPI chipset detected\n", __FUNCTION__);
+       case BUS_TYPE_WBSIO_SPI:
+               return wbsio_spi_read(flash, buf);
+       default:
+               printf_debug
+                   ("%s called, but no SPI chipset/strapping detected\n",
+                    __FUNCTION__);
+       }
+
        return 1;
 }
 
 int spi_chip_write(struct flashchip *flash, uint8_t *buf)
 {
-       if (it8716f_flashport)
+       switch (flashbus) {
+       case BUS_TYPE_IT87XX_SPI:
                return it8716f_spi_chip_write(flash, buf);
-       else if ((ich7_detected) || (viaspi_detected))
-               return ich_spi_write(flash, buf);
-       else if (ich9_detected)
+       case BUS_TYPE_SB600_SPI:
+               return sb600_spi_write(flash, buf);
+       case BUS_TYPE_ICH7_SPI:
+       case BUS_TYPE_ICH9_SPI:
+       case BUS_TYPE_VIA_SPI:
                return ich_spi_write(flash, buf);
-       printf_debug("%s called, but no SPI chipset detected\n", __FUNCTION__);
+       case BUS_TYPE_WBSIO_SPI:
+               return wbsio_spi_write(flash, buf);
+       default:
+               printf_debug
+                   ("%s called, but no SPI chipset/strapping detected\n",
+                    __FUNCTION__);
+       }
+
        return 1;
 }
 
+int spi_aai_write(struct flashchip *flash, uint8_t *buf)
+{
+       uint32_t pos = 2, size = flash->total_size * 1024;
+       unsigned char w[6] = {0xad, 0, 0, 0, buf[0], buf[1]};
+       switch (flashbus) {
+       case BUS_TYPE_WBSIO_SPI:
+               fprintf(stderr, "%s: impossible with Winbond SPI masters,"
+                               " degrading to byte program\n", __func__);
+               return spi_chip_write(flash, buf);
+       default:
+               break;
+       }
+       flash->erase(flash);
+       spi_write_enable();
+       spi_command(6, 0, w, NULL);
+       while (spi_read_status_register() & JEDEC_RDSR_BIT_WIP)
+               myusec_delay(5); /* SST25VF040B Tbp is max 10us */
+       while (pos < size) {
+               w[1] = buf[pos++];
+               w[2] = buf[pos++];
+               spi_command(3, 0, w, NULL);
+               while (spi_read_status_register() & JEDEC_RDSR_BIT_WIP)
+                       myusec_delay(5); /* SST25VF040B Tbp is max 10us */
+       }
+       spi_write_disable();
+       return 0;
+}