Add a driver for Macronix NAND block protection function.

Signed-off-by: Mason Yang <masonccy...@mxic.com.tw>
---
 drivers/mtd/nand/raw/mxic_nand.c     |  3 +++
 drivers/mtd/nand/raw/nand_macronix.c | 47 ++++++++++++++++++++++++++++++++++++
 include/linux/mfd/mxic-mx25f0a.h     |  3 +++
 3 files changed, 53 insertions(+)

diff --git a/drivers/mtd/nand/raw/mxic_nand.c b/drivers/mtd/nand/raw/mxic_nand.c
index 03886b2..9307ca2 100644
--- a/drivers/mtd/nand/raw/mxic_nand.c
+++ b/drivers/mtd/nand/raw/mxic_nand.c
@@ -262,6 +262,9 @@ static int mx25f0a_nand_probe(struct platform_device *pdev)
        if (err)
                goto fail;
 
+       mtd->_lock = mxic_nand_lock;
+       mtd->_unlock = mxic_nand_unlock;
+
        err = mtd_device_register(mtd, NULL, 0);
        if (err)
                goto fail;
diff --git a/drivers/mtd/nand/raw/nand_macronix.c 
b/drivers/mtd/nand/raw/nand_macronix.c
index a19caa4..db63350 100644
--- a/drivers/mtd/nand/raw/nand_macronix.c
+++ b/drivers/mtd/nand/raw/nand_macronix.c
@@ -21,8 +21,12 @@
 #define MACRONIX_RANDOMIZER_BIT BIT(1)
 #define MACRONIX_READ_RETRY_MODE 5
 
+#define ONFI_FEATURE_ADDR_MXIC_PROTECTION 0xA0
 #define ONFI_FEATURE_ADDR_MXIC_RANDOMIZER 0xB0
 
+#define MXIC_BLOCK_PROTECTION_ALL_LOCK 0x38
+#define MXIC_BLOCK_PROTECTION_ALL_UNLOCK 0x0
+
 struct nand_onfi_vendor_macronix {
        u8 reserved[1];
        u8 reliability_func;
@@ -146,6 +150,13 @@ static void macronix_nand_onfi_init(struct nand_chip *chip)
                struct nand_onfi_vendor_macronix *mxic =
                                (void *)p->onfi->vendor;
 
+               if (p->supports_set_get_features) {
+                       set_bit(ONFI_FEATURE_ADDR_MXIC_PROTECTION,
+                               p->get_feature_list);
+                       set_bit(ONFI_FEATURE_ADDR_MXIC_PROTECTION,
+                               p->set_feature_list);
+               }
+
                if (mxic->reliability_func & MACRONIX_READ_RETRY_BIT) {
                        chip->read_retries = MACRONIX_READ_RETRY_MODE + 1;
                        chip->setup_read_retry =
@@ -241,3 +252,39 @@ static int macronix_nand_init(struct nand_chip *chip)
 const struct nand_manufacturer_ops macronix_nand_manuf_ops = {
        .init = macronix_nand_init,
 };
+
+int mxic_nand_lock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
+{
+       struct nand_chip *chip = mtd_to_nand(mtd);
+       u8 feature[ONFI_SUBFEATURE_PARAM_LEN] = {0};
+       int ret;
+
+       feature[0] = MXIC_BLOCK_PROTECTION_ALL_LOCK;
+       nand_select_target(chip, 0);
+       ret = nand_set_features(chip, ONFI_FEATURE_ADDR_MXIC_PROTECTION,
+                               feature);
+       nand_deselect_target(chip);
+       if (ret)
+               pr_err("Lock MXIC NAND all blocks failed, err:%d\n", ret);
+
+       return ret;
+}
+EXPORT_SYMBOL_GPL(mxic_nand_lock);
+
+int mxic_nand_unlock(struct mtd_info *mtd, loff_t ofs, uint64_t len)
+{
+       struct nand_chip *chip = mtd_to_nand(mtd);
+       u8 feature[ONFI_SUBFEATURE_PARAM_LEN] = {0};
+       int ret;
+
+       feature[0] = MXIC_BLOCK_PROTECTION_ALL_UNLOCK;
+       nand_select_target(chip, 0);
+       ret = nand_set_features(chip, ONFI_FEATURE_ADDR_MXIC_PROTECTION,
+                               feature);
+       nand_deselect_target(chip);
+       if (ret)
+               pr_err("Unlock MXIC NAND all blocks failed, err:%d\n", ret);
+
+       return ret;
+}
+EXPORT_SYMBOL_GPL(mxic_nand_unlock);
diff --git a/include/linux/mfd/mxic-mx25f0a.h b/include/linux/mfd/mxic-mx25f0a.h
index 5a8cfc7..cfce992 100644
--- a/include/linux/mfd/mxic-mx25f0a.h
+++ b/include/linux/mfd/mxic-mx25f0a.h
@@ -170,4 +170,7 @@ struct mx25f0a_mfd {
        void __iomem *dirmap;
 };
 
+int mxic_nand_lock(struct mtd_info *mtd, loff_t ofs, uint64_t len);
+int mxic_nand_unlock(struct mtd_info *mtd, loff_t ofs, uint64_t len);
+
 #endif // __MFD_MXIC_MX25F0A_H
-- 
1.9.1

Reply via email to