// SPDX-License-Identifier: GPL-2.0-only
/****************************************************************************
 * Driver for Solarflare network controllers and boards
 * Copyright 2005-2006 Fen Systems Ltd.
 * Copyright 2006-2013 Solarflare Communications Inc.
 */

#include <linux/module.h>
#include <linux/mtd/mtd.h>
#include <linux/slab.h>
#include <linux/rtnetlink.h>

#include "net_driver.h"
#include "efx.h"

#define to_ef4_mtd_partition(mtd)    \
 container_of(mtd, struct ef4_mtd_partition, mtd)

/* MTD interface */

static int ef4_mtd_erase(struct mtd_info *mtd, struct erase_info *erase)
{
 struct ef4_nic *efx = mtd->priv;

 return efx->type->mtd_erase(mtd, erase->addr, erase->len);
}

static void ef4_mtd_sync(struct mtd_info *mtd)
{
 struct ef4_mtd_partition *part = to_ef4_mtd_partition(mtd);
 struct ef4_nic *efx = mtd->priv;
 int rc;

 rc = efx->type->mtd_sync(mtd);
 if (rc)
  pr_err("%s: %s sync failed (%d)\n",
         part->name, part->dev_type_name, rc);
}

static void ef4_mtd_remove_partition(struct ef4_mtd_partition *part)
{
 int rc;

 for (;;) {
  rc = mtd_device_unregister(&part->mtd);
  if (rc != -EBUSY)
   break;
  ssleep(1);
 }
 WARN_ON(rc);
 list_del(&part->node);
}

int ef4_mtd_add(struct ef4_nic *efx, struct ef4_mtd_partition *parts,
  size_t n_parts, size_t sizeof_part)
{
 struct ef4_mtd_partition *part;
 size_t i;

 for (i = 0; i < n_parts; i++) {
  part = (struct ef4_mtd_partition *)((char *)parts +
          i * sizeof_part);

  part->mtd.writesize = 1;

  part->mtd.owner = THIS_MODULE;
  part->mtd.priv = efx;
  part->mtd.name = part->name;
  part->mtd._erase = ef4_mtd_erase;
  part->mtd._read = efx->type->mtd_read;
  part->mtd._write = efx->type->mtd_write;
  part->mtd._sync = ef4_mtd_sync;

  efx->type->mtd_rename(part);

  if (mtd_device_register(&part->mtd, NULL, 0))
   goto fail;

  /* Add to list in order - ef4_mtd_remove() depends on this */
  list_add_tail(&part->node, &efx->mtd_list);
 }

 return 0;

fail:
 while (i--) {
  part = (struct ef4_mtd_partition *)((char *)parts +
          i * sizeof_part);
  ef4_mtd_remove_partition(part);
 }
 /* Failure is unlikely here, but probably means we're out of memory */
 return -ENOMEM;
}

void ef4_mtd_remove(struct ef4_nic *efx)
{
 struct ef4_mtd_partition *parts, *part, *next;

 WARN_ON(ef4_dev_registered(efx));

 if (list_empty(&efx->mtd_list))
  return;

 parts = list_first_entry(&efx->mtd_list, struct ef4_mtd_partition,
     node);

 list_for_each_entry_safe(part, next, &efx->mtd_list, node)
  ef4_mtd_remove_partition(part);

 kfree(parts);
}

void ef4_mtd_rename(struct ef4_nic *efx)
{
 struct ef4_mtd_partition *part;

 ASSERT_RTNL();

 list_for_each_entry(part, &efx->mtd_list, node)
  efx->type->mtd_rename(part);
}