// SPDX-License-Identifier: GPL-2.0-only
/*
 * Ampere Computing SoC's SMpro Misc Driver
 *
 * Copyright (c) 2022, Ampere Computing LLC
 */
#include <linux/mod_devicetable.h>
#include <linux/module.h>
#include <linux/platform_device.h>
#include <linux/regmap.h>

/* Boot Stage/Progress Registers */
#define BOOTSTAGE	0xB0
#define BOOTSTAGE_LO	0xB1
#define CUR_BOOTSTAGE	0xB2
#define BOOTSTAGE_HI	0xB3

/* SOC State Registers */
#define SOC_POWER_LIMIT		0xE5

struct smpro_misc {
	struct regmap *regmap;
};

static ssize_t boot_progress_show(struct device *dev, struct device_attribute *da, char *buf)
{
	struct smpro_misc *misc = dev_get_drvdata(dev);
	u16 boot_progress[3] = { 0 };
	u32 bootstage;
	u8 boot_stage;
	u8 cur_stage;
	u32 reg_lo;
	u32 reg;
	int ret;

	/* Read current boot stage */
	ret = regmap_read(misc->regmap, CUR_BOOTSTAGE, &reg);
	if (ret)
		return ret;

	cur_stage = reg & 0xff;

	ret = regmap_read(misc->regmap, BOOTSTAGE, &bootstage);
	if (ret)
		return ret;

	boot_stage = (bootstage >> 8) & 0xff;

	if (boot_stage > cur_stage)
		return -EINVAL;

	ret = regmap_read(misc->regmap,	BOOTSTAGE_LO, &reg_lo);
	if (!ret)
		ret = regmap_read(misc->regmap, BOOTSTAGE_HI, &reg);
	if (ret)
		return ret;

	/* Firmware to report new boot stage next time */
	if (boot_stage < cur_stage) {
		ret = regmap_write(misc->regmap, BOOTSTAGE, ((bootstage & 0xff00) | 0x1));
		if (ret)
			return ret;
	}

	boot_progress[0] = bootstage;
	boot_progress[1] = swab16(reg);
	boot_progress[2] = swab16(reg_lo);

	return sysfs_emit(buf, "%*phN\n", (int)sizeof(boot_progress), boot_progress);
}

static DEVICE_ATTR_RO(boot_progress);

static ssize_t soc_power_limit_show(struct device *dev, struct device_attribute *da, char *buf)
{
	struct smpro_misc *misc = dev_get_drvdata(dev);
	unsigned int value;
	int ret;

	ret = regmap_read(misc->regmap, SOC_POWER_LIMIT, &value);
	if (ret)
		return ret;

	return sysfs_emit(buf, "%d\n", value);
}

static ssize_t soc_power_limit_store(struct device *dev, struct device_attribute *da,
				     const char *buf, size_t count)
{
	struct smpro_misc *misc = dev_get_drvdata(dev);
	unsigned long val;
	s32 ret;

	ret = kstrtoul(buf, 0, &val);
	if (ret)
		return ret;

	ret = regmap_write(misc->regmap, SOC_POWER_LIMIT, (unsigned int)val);
	if (ret)
		return -EPROTO;

	return count;
}

static DEVICE_ATTR_RW(soc_power_limit);

static struct attribute *smpro_misc_attrs[] = {
	&dev_attr_boot_progress.attr,
	&dev_attr_soc_power_limit.attr,
	NULL
};

ATTRIBUTE_GROUPS(smpro_misc);

static int smpro_misc_probe(struct platform_device *pdev)
{
	struct smpro_misc *misc;

	misc = devm_kzalloc(&pdev->dev, sizeof(struct smpro_misc), GFP_KERNEL);
	if (!misc)
		return -ENOMEM;

	platform_set_drvdata(pdev, misc);

	misc->regmap = dev_get_regmap(pdev->dev.parent, NULL);
	if (!misc->regmap)
		return -ENODEV;

	return 0;
}

static struct platform_driver smpro_misc_driver = {
	.probe		= smpro_misc_probe,
	.driver = {
		.name	= "smpro-misc",
		.dev_groups = smpro_misc_groups,
	},
};

module_platform_driver(smpro_misc_driver);

MODULE_AUTHOR("Tung Nguyen <tungnguyen@os.amperecomputing.com>");
MODULE_AUTHOR("Quan Nguyen <quan@os.amperecomputing.com>");
MODULE_DESCRIPTION("Ampere Altra SMpro Misc driver");
MODULE_LICENSE("GPL");