// SPDX-License-Identifier: GPL-2.0-only
/*
 * BCM47XX NAND flash driver
 *
 * Copyright (C) 2012 Rafał Miłecki <zajec5@gmail.com>
 */

#include "bcm47xxnflash.h"

#include <linux/module.h>
#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/platform_device.h>
#include <linux/bcma/bcma.h>

MODULE_DESCRIPTION("NAND flash driver for BCMA bus");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Rafał Miłecki");

static const char *probes[] = { "bcm47xxpart", NULL };

static int bcm47xxnflash_probe(struct platform_device *pdev)
{
 struct bcma_nflash *nflash = dev_get_platdata(&pdev->dev);
 struct bcm47xxnflash *b47n;
 struct mtd_info *mtd;
 int err = 0;

 b47n = devm_kzalloc(&pdev->dev, sizeof(*b47n), GFP_KERNEL);
 if (!b47n)
  return -ENOMEM;

 nand_set_controller_data(&b47n->nand_chip, b47n);
 mtd = nand_to_mtd(&b47n->nand_chip);
 mtd->dev.parent = &pdev->dev;
 b47n->cc = container_of(nflash, struct bcma_drv_cc, nflash);

 if (b47n->cc->core->bus->chipinfo.id == BCMA_CHIP_ID_BCM4706) {
  err = bcm47xxnflash_ops_bcm4706_init(b47n);
 } else {
  pr_err("Device not supported\n");
  err = -ENOTSUPP;
 }
 if (err) {
  pr_err("Initialization failed: %d\n", err);
  return err;
 }

 platform_set_drvdata(pdev, b47n);

 err = mtd_device_parse_register(mtd, probes, NULL, NULL, 0);
 if (err) {
  pr_err("Failed to register MTD device: %d\n", err);
  return err;
 }

 return 0;
}

static void bcm47xxnflash_remove(struct platform_device *pdev)
{
 struct bcm47xxnflash *nflash = platform_get_drvdata(pdev);
 struct nand_chip *chip = &nflash->nand_chip;
 int ret;

 ret = mtd_device_unregister(nand_to_mtd(chip));
 WARN_ON(ret);
 nand_cleanup(chip);
}

static struct platform_driver bcm47xxnflash_driver = {
 .probe = bcm47xxnflash_probe,
 .remove = bcm47xxnflash_remove,
 .driver = {
  .name = "bcma_nflash",
 },
};

module_platform_driver(bcm47xxnflash_driver);