// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2018 Toradex AG
 *
 * Author: Marcel Ziswiler <marcel.ziswiler@toradex.com>
 */

#include <linux/mtd/rawnand.h>
#include "internals.h"

static void esmt_nand_decode_id(struct nand_chip *chip)
{
        struct nand_device *base = &chip->base;
        struct nand_ecc_props requirements = {};

        nand_decode_ext_id(chip);

        /* Extract ECC requirements from 5th id byte. */
        if (chip->id.len >= 5 && nand_is_slc(chip)) {
                requirements.step_size = 512;
                switch (chip->id.data[4] & 0x3) {
                case 0x0:
                        requirements.strength = 4;
                        break;
                case 0x1:
                        requirements.strength = 2;
                        break;
                case 0x2:
                        requirements.strength = 1;
                        break;
                default:
                        WARN(1, "Could not get ECC info");
                        requirements.step_size = 0;
                        break;
                }
        }

        nanddev_set_ecc_requirements(base, &requirements);
}

static int esmt_nand_init(struct nand_chip *chip)
{
        if (nand_is_slc(chip))
                /*
                 * It is known that some ESMT SLC NANDs have been shipped
                 * with the factory bad block markers in the first or last page
                 * of the block, instead of the first or second page. To be on
                 * the safe side, let's check all three locations.
                 */
                chip->options |= NAND_BBM_FIRSTPAGE | NAND_BBM_SECONDPAGE |
                                 NAND_BBM_LASTPAGE;

        return 0;
}

const struct nand_manufacturer_ops esmt_nand_manuf_ops = {
        .detect = esmt_nand_decode_id,
        .init = esmt_nand_init,
};