// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (c) 2023, Qualcomm Innovation Center, Inc. All rights reserved.
 */

#include <linux/firmware/qcom/qcom_scm.h>
#include <linux/mod_devicetable.h>
#include <linux/nvmem-provider.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>

/**
 * struct sec_qfprom - structure holding secure qfprom attributes
 *
 * @base: starting physical address for secure qfprom corrected address space.
 * @dev: qfprom device structure.
 */
struct sec_qfprom {
	phys_addr_t base;
	struct device *dev;
};

static int sec_qfprom_reg_read(void *context, unsigned int reg, void *_val, size_t bytes)
{
	struct sec_qfprom *priv = context;
	unsigned int i;
	u8 *val = _val;
	u32 read_val;
	u8 *tmp;

	for (i = 0; i < bytes; i++, reg++) {
		if (i == 0 || reg % 4 == 0) {
			if (qcom_scm_io_readl(priv->base + (reg & ~3), &read_val)) {
				dev_err(priv->dev, "Couldn't access fuse register\n");
				return -EINVAL;
			}
			tmp = (u8 *)&read_val;
		}

		val[i] = tmp[reg & 3];
	}

	return 0;
}

static int sec_qfprom_probe(struct platform_device *pdev)
{
	struct nvmem_config econfig = {
		.name = "sec-qfprom",
		.add_legacy_fixed_of_cells = true,
		.stride = 1,
		.word_size = 1,
		.id = NVMEM_DEVID_AUTO,
		.reg_read = sec_qfprom_reg_read,
	};
	struct device *dev = &pdev->dev;
	struct nvmem_device *nvmem;
	struct sec_qfprom *priv;
	struct resource *res;

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

	res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
	if (!res)
		return -EINVAL;

	priv->base = res->start;

	econfig.size = resource_size(res);
	econfig.dev = dev;
	econfig.priv = priv;

	priv->dev = dev;

	nvmem = devm_nvmem_register(dev, &econfig);

	return PTR_ERR_OR_ZERO(nvmem);
}

static const struct of_device_id sec_qfprom_of_match[] = {
	{ .compatible = "qcom,sec-qfprom" },
	{/* sentinel */},
};
MODULE_DEVICE_TABLE(of, sec_qfprom_of_match);

static struct platform_driver qfprom_driver = {
	.probe = sec_qfprom_probe,
	.driver = {
		.name = "qcom_sec_qfprom",
		.of_match_table = sec_qfprom_of_match,
	},
};
module_platform_driver(qfprom_driver);
MODULE_DESCRIPTION("Qualcomm Secure QFPROM driver");
MODULE_LICENSE("GPL");