[PATCH v2 2/3] mtd: rawnand: meson: refactor use of 'meson_nfc_cmd_access()'

[Arseniy Krasnov]

Move call 'meson_nfc_cmd_seed()' and check for 'NAND_NEED_SCRAMBLING'
to 'meson_nfc_cmd_access()', thus removing code duplication.

Signed-off-by: Arseniy Krasnov <avkras...@salutedevices.com>
---
 drivers/mtd/nand/raw/meson_nand.c | 30 ++++++++++++------------------
 1 file changed, 12 insertions(+), 18 deletions(-)

diff --git a/drivers/mtd/nand/raw/meson_nand.c 
b/drivers/mtd/nand/raw/meson_nand.c
index 5d411c4594..19f005202b 100644
--- a/drivers/mtd/nand/raw/meson_nand.c
+++ b/drivers/mtd/nand/raw/meson_nand.c
@@ -229,14 +229,19 @@ static void meson_nfc_cmd_seed(const struct meson_nfc 
*nfc, u32 seed)
               nfc->reg_base + NFC_REG_CMD);
 }
 
-static void meson_nfc_cmd_access(struct nand_chip *nand, bool raw, bool dir,
-                                int scrambler)
+static void meson_nfc_cmd_access(struct nand_chip *nand, bool raw, bool dir, 
int page)
 {
        struct mtd_info *mtd = nand_to_mtd(nand);
        const struct meson_nfc *nfc = 
nand_get_controller_data(mtd_to_nand(mtd));
        const struct meson_nfc_nand_chip *meson_chip = to_meson_nand(nand);
        u32 bch = meson_chip->bch_mode, cmd;
        int len = mtd->writesize, pagesize, pages;
+       int scrambler;
+
+       if (nand->options & NAND_NEED_SCRAMBLING)
+               scrambler = NFC_CMD_SCRAMBLER_ENABLE;
+       else
+               scrambler = NFC_CMD_SCRAMBLER_DISABLE;
 
        pagesize = nand->ecc.size;
 
@@ -252,6 +257,9 @@ static void meson_nfc_cmd_access(struct nand_chip *nand, 
bool raw, bool dir,
        cmd = CMDRWGEN(DMA_DIR(dir), scrambler, bch,
                       NFC_CMD_SHORTMODE_DISABLE, pagesize, pages);
 
+       if (scrambler == NFC_CMD_SCRAMBLER_ENABLE)
+               meson_nfc_cmd_seed(nfc, page);
+
        writel(cmd, nfc->reg_base + NFC_REG_CMD);
 }
 
@@ -566,14 +574,7 @@ static int meson_nfc_write_page_sub(struct nand_chip *nand,
                return ret;
        }
 
-       if (nand->options & NAND_NEED_SCRAMBLING) {
-               meson_nfc_cmd_seed(nfc, page);
-               meson_nfc_cmd_access(nand, raw, DIRWRITE,
-                                    NFC_CMD_SCRAMBLER_ENABLE);
-       } else {
-               meson_nfc_cmd_access(nand, raw, DIRWRITE,
-                                    NFC_CMD_SCRAMBLER_DISABLE);
-       }
+       meson_nfc_cmd_access(nand, raw, DIRWRITE, page);
 
        cmd = nfc->param.chip_select | NFC_CMD_CLE | NAND_CMD_PAGEPROG;
        writel(cmd, nfc->reg_base + NFC_REG_CMD);
@@ -644,14 +645,7 @@ static int meson_nfc_read_page_sub(struct nand_chip *nand,
        if (ret)
                return ret;
 
-       if (nand->options & NAND_NEED_SCRAMBLING) {
-               meson_nfc_cmd_seed(nfc, page);
-               meson_nfc_cmd_access(nand, raw, DIRREAD,
-                                    NFC_CMD_SCRAMBLER_ENABLE);
-       } else {
-               meson_nfc_cmd_access(nand, raw, DIRREAD,
-                                    NFC_CMD_SCRAMBLER_DISABLE);
-       }
+       meson_nfc_cmd_access(nand, raw, DIRREAD, page);
 
        meson_nfc_wait_dma_finish(nfc);
        meson_nfc_check_ecc_pages_valid(nfc, nand, raw);
-- 
2.30.1