Signed-off-by: Boris BREZILLON <[email protected]>
---
 drivers/mtd/nand/sunxi_nand.c |   69 +++++++++++++++++++++++++++++++++++++++--
 1 file changed, 66 insertions(+), 3 deletions(-)

diff --git a/drivers/mtd/nand/sunxi_nand.c b/drivers/mtd/nand/sunxi_nand.c
index f24ad8b..4693d9f 100644
--- a/drivers/mtd/nand/sunxi_nand.c
+++ b/drivers/mtd/nand/sunxi_nand.c
@@ -168,6 +168,17 @@ struct sunxi_nand_hw_ecc {
        struct nand_ecclayout layout;
 };
 
+struct sunxi_nand_part {
+       struct nand_part part;
+       struct nand_ecc_ctrl ecc;
+};
+
+static inline struct sunxi_nand_part *
+to_sunxi_nand_part(struct nand_part *part)
+{
+       return container_of(part, struct sunxi_nand_part, part);
+}
+
 struct sunxi_nand_chip {
        struct list_head node;
        struct nand_chip nand;
@@ -780,12 +791,58 @@ static int sunxi_nand_chip_ecc_init(struct device *dev,
        return 0;
 }
 
+static void sunxi_nand_part_release(struct nand_part *part)
+{
+       kfree(to_sunxi_nand_part(part));
+}
+
+struct nand_part *sunxi_ofnandpart_parse(void *priv, struct mtd_info *master,
+                                        struct device_node *pp)
+{
+       int mode = of_get_nand_ecc_mode(pp);
+       struct sunxi_nand_part *part;
+       struct nand_chip *chip = master->priv;
+       u32 strength;
+       u32 blk_size;
+
+       if (mode < 0)
+               return nandpart_alloc();
+
+       part = kzalloc(sizeof(*part), GFP_KERNEL);
+       part->part.release = sunxi_nand_part_release;
+
+       part->ecc.size = chip->ecc_step_ds;
+       part->ecc.strength = chip->ecc_strength_ds;
+       if (!of_get_nand_ecc_level(pp, &strength, &blk_size)) {
+               part->ecc.size = blk_size;
+               part->ecc.strength = strength;
+       }
+
+       part->ecc.mode = mode;
+       switch (mode) {
+       case NAND_ECC_SOFT_BCH:
+               part->ecc.bytes = ((part->ecc.strength *
+                                   fls(8 * part->ecc.size)) + 7) / 8;
+               break;
+       case NAND_ECC_HW:
+               break;
+       case NAND_ECC_NONE:
+               break;
+       default:
+               break;
+       }
+
+       part->part.ecc = &part->ecc;
+
+       return &part->part;
+}
+
 static int sunxi_nand_chip_init(struct device *dev, struct sunxi_nfc *nfc,
                                struct device_node *np)
 {
        const struct nand_sdr_timings *timings;
        struct sunxi_nand_chip *chip;
-       struct mtd_part_parser_data ppdata;
+       struct ofnandpart_data ppdata;
        struct mtd_info *mtd;
        struct nand_chip *nand;
        int nsels;
@@ -889,9 +946,15 @@ static int sunxi_nand_chip_init(struct device *dev, struct 
sunxi_nfc *nfc,
                mtd->name = chip->default_name;
        }
 
-       ppdata.of_node = np;
-       ret = mtd_device_parse_register(mtd, NULL, &ppdata, NULL, 0);
+       ppdata.node = np;
+       ppdata.parse = sunxi_ofnandpart_parse;
+       ret = ofnandpart_parse(mtd, &ppdata);
        if (!ret)
+               ret = mtd_device_register(mtd, NULL, 0);
+       else if (ret > 0)
+               ret = 0;
+
+       if (ret)
                return ret;
 
        list_add_tail(&chip->node, &nfc->chips);
-- 
1.7.9.5

--
To unsubscribe from this list: send the line "unsubscribe linux-kernel" in
the body of a message to [email protected]
More majordomo info at  http://vger.kernel.org/majordomo-info.html
Please read the FAQ at  http://www.tux.org/lkml/

Reply via email to