// SPDX-License-Identifier: GPL-2.0+
/*
 * Cirrus EP93xx SoC reset driver
 *
 * Copyright (C) 2021 Nikita Shubin <nikita.shubin@maquefel.me>
 */

#include <linux/bits.h>
#include <linux/container_of.h>
#include <linux/delay.h>
#include <linux/errno.h>
#include <linux/mfd/syscon.h>
#include <linux/module.h>
#include <linux/mod_devicetable.h>
#include <linux/notifier.h>
#include <linux/reboot.h>
#include <linux/slab.h>

#include <linux/soc/cirrus/ep93xx.h>

#define EP93XX_SYSCON_DEVCFG		0x80
#define EP93XX_SYSCON_DEVCFG_SWRST	BIT(31)

struct ep93xx_restart {
	struct ep93xx_regmap_adev *aux_dev;
	struct notifier_block restart_handler;
};

static int ep93xx_restart_handle(struct notifier_block *this,
				 unsigned long mode, void *cmd)
{
	struct ep93xx_restart *priv =
		container_of(this, struct ep93xx_restart, restart_handler);
	struct ep93xx_regmap_adev *aux = priv->aux_dev;

	/* Issue the reboot */
	aux->update_bits(aux->map, aux->lock, EP93XX_SYSCON_DEVCFG,
			 EP93XX_SYSCON_DEVCFG_SWRST, EP93XX_SYSCON_DEVCFG_SWRST);
	aux->update_bits(aux->map, aux->lock, EP93XX_SYSCON_DEVCFG,
			 EP93XX_SYSCON_DEVCFG_SWRST, 0);

	return NOTIFY_DONE;
}

static int ep93xx_reboot_probe(struct auxiliary_device *adev,
			       const struct auxiliary_device_id *id)
{
	struct ep93xx_regmap_adev *rdev = to_ep93xx_regmap_adev(adev);
	struct device *dev = &adev->dev;
	struct ep93xx_restart *priv;
	int err;

	if (!rdev->update_bits)
		return -ENODEV;

	priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
	if (!priv)
		return -ENOMEM;

	priv->aux_dev = rdev;

	priv->restart_handler.notifier_call = ep93xx_restart_handle;
	priv->restart_handler.priority = 128;

	err = register_restart_handler(&priv->restart_handler);
	if (err)
		return dev_err_probe(dev, err, "can't register restart notifier\n");

	return 0;
}

static const struct auxiliary_device_id ep93xx_reboot_ids[] = {
	{
		.name = "soc_ep93xx.reset-ep93xx",
	},
	{ /* sentinel */ }
};
MODULE_DEVICE_TABLE(auxiliary, ep93xx_reboot_ids);

static struct auxiliary_driver ep93xx_reboot_driver = {
	.probe		= ep93xx_reboot_probe,
	.id_table	= ep93xx_reboot_ids,
};
module_auxiliary_driver(ep93xx_reboot_driver);