write_enable() was being duplicated to both m25p80.c and fsl-quadspi.c.
But this should be handled within the spi-nor abstraction layer.
At the same time, let's add write_disable() after erasing, so we don't
leave the flash in a write-enabled state afterward.
Signed-off-by: Brian Norris <computersforpeace@gmail.com>
Acked-by: Huang Shijie <shijie.huang@intel.com>
 static int m25p80_erase(struct spi_nor *nor, loff_t offset)
 {
        struct m25p *flash = nor->priv;
-       int ret;
 
        dev_dbg(nor->dev, "%dKiB at 0x%08x\n",
                flash->mtd.erasesize / 1024, (u32)offset);
 
-       /* Send write enable, then erase commands. */
-       ret = nor->write_reg(nor, SPINOR_OP_WREN, NULL, 0, 0);
-       if (ret)
-               return ret;
-
        /* Set up command buffer. */
        flash->command[0] = nor->erase_opcode;
        m25p_addr2cmd(nor, offset, flash->command);
 
        dev_dbg(nor->dev, "%dKiB at 0x%08x:0x%08x\n",
                nor->mtd->erasesize / 1024, q->chip_base_addr, (u32)offs);
 
-       /* Send write enable, then erase commands. */
-       ret = nor->write_reg(nor, SPINOR_OP_WREN, NULL, 0, 0);
-       if (ret)
-               return ret;
-
        ret = fsl_qspi_runcmd(q, nor->erase_opcode, offs, 0);
        if (ret)
                return ret;
 
 {
        dev_dbg(nor->dev, " %lldKiB\n", (long long)(nor->mtd->size >> 10));
 
-       /* Send write enable, then erase commands. */
-       write_enable(nor);
-
        return nor->write_reg(nor, SPINOR_OP_CHIP_ERASE, NULL, 0, 0);
 }
 
 
        /* whole-chip erase? */
        if (len == mtd->size) {
+               write_enable(nor);
+
                if (erase_chip(nor)) {
                        ret = -EIO;
                        goto erase_err;
        /* "sector"-at-a-time erase */
        } else {
                while (len) {
+                       write_enable(nor);
+
                        if (nor->erase(nor, addr)) {
                                ret = -EIO;
                                goto erase_err;
                }
        }
 
+       write_disable(nor);
+
        spi_nor_unlock_and_unprep(nor, SPI_NOR_OPS_ERASE);
 
        instr->state = MTD_ERASE_DONE;