// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2016 MediaTek Inc.
 */

#include <linux/module.h>
#include <linux/of_fdt.h>
#include <linux/of.h>
#include <linux/io.h>
#include <linux/slab.h>
#include <linux/of_address.h>
#include <linux/arm-smccc.h>

#include <linux/soc/mediatek/mtk_sip_svc.h> /* for SMC ID table */
#include <mt-plat/mtk_platform_debug.h>
#include "dfd.h"

static struct dfd_drv *drv;

/* return -1 for error indication */
int dfd_setup(int version)
{
	int ret;
	int dfd_doe;
	struct arm_smccc_res res;

	if (drv && (drv->enabled == 1) && (drv->base_addr > 0)) {
		/* check support or not first */
		if (!drv->check_dfd_support)
			return -EINVAL;

		pr_info("dfd setup\n");

		ret = mtk_dbgtop_dfd_count_en(1);
		ret = mtk_dbgtop_dfd_therm1_dis(1);
		ret = mtk_dbgtop_dfd_therm2_dis(0);
		ret = mtk_dbgtop_dfd_timeout(drv->rg_dfd_timeout);
		if (drv->mem_reserve && drv->cachedump_en) {
			dfd_doe = DFD_CACHE_DUMP_ENABLE;
			if (drv->l2c_trigger)
				dfd_doe |= DFD_PARITY_ERR_TRIGGER;
			if (version == DFD_EXTENDED_DUMP)
				arm_smccc_smc(MTK_SIP_KERNEL_DFD,
					DFD_SMC_MAGIC_SETUP,
					(u64) drv->base_addr,
					drv->chain_length,
					dfd_doe, 0, 0, 0, &res);
			else
				arm_smccc_smc(MTK_SIP_KERNEL_DFD,
					DFD_SMC_MAGIC_SETUP,
					(u64) drv->base_addr,
					drv->chain_length,
					0, 0, 0, 0, &res);
		} else {
			arm_smccc_smc(MTK_SIP_KERNEL_DFD, DFD_SMC_MAGIC_SETUP,
				(u64) drv->base_addr, drv->chain_length,
				0, 0, 0, 0, &res);
		}

		if (set_sram_flag_dfd_valid() < 0)
			return -1;

		return ret;
	} else
		return -1;
}

static int __init fdt_get_chosen(unsigned long node, const char *uname,
		int depth, void *data)
{
	if (depth != 1 || (strcmp(uname, "chosen") != 0
				&& strcmp(uname, "chosen@0") != 0))
		return 0;

	*(unsigned long *)data = node;
	return 1;
}

static int __init dfd_init(void)
{
	struct device_node *dev_node, *infra_node, *rgu_node;
	unsigned long node;
	const void *prop;
	unsigned int val;
	void __iomem *toprgu_base;

	drv = kzalloc(sizeof(struct dfd_drv), GFP_KERNEL);
	if (!drv)
		return -ENOMEM;

	pr_info("In dfd init\n");

	/* get dfd settings */
	dev_node = of_find_compatible_node(NULL, NULL, "mediatek,dfd");
	if (dev_node) {
		if (of_property_read_u32(dev_node, "mediatek,dfd_latch_offset", &val)) {
			pr_info("%s: Latch offset not found.\n", __func__);
			return -EINVAL;
		}
		pr_info("%s: Latch offset is 0x%x\n", __func__, val);
		rgu_node = of_find_compatible_node(NULL, NULL, "mediatek,toprgu");
		toprgu_base = of_iomap(rgu_node, 0);

		if (!toprgu_base)
			pr_info("%s: RGU base not found.\n", __func__);
		else
			pr_info("%s: get topdbg base success\n", __func__);
		get_dfd_base(toprgu_base, val);

		if (of_property_read_u32(dev_node, "mediatek,enabled", &val))
			drv->enabled = 0;
		else
			drv->enabled = val;

		if (of_property_read_u32(dev_node,
					"mediatek,chain_length", &val))
			drv->chain_length = 0;
		else
			drv->chain_length = val;

		if (of_property_read_u32(dev_node,
					"mediatek,rg_dfd_timeout", &val))
			drv->rg_dfd_timeout = 0;
		else
			drv->rg_dfd_timeout = val;

		if (of_property_read_u32(dev_node,
					"mediatek,check_dfd_support", &val))
			drv->check_dfd_support = 0;
		else
			drv->check_dfd_support = val;

		if (of_property_read_u32(dev_node,
					"mediatek,dfd_infra_base", &val))
			drv->dfd_infra_base = 0;
		else
			drv->dfd_infra_base = val;

		if (of_property_read_u32(dev_node,
					"mediatek,dfd_ap_addr_offset", &val))
			drv->dfd_ap_addr_offset = 0;
		else
			drv->dfd_ap_addr_offset = val;
	} else
		return -ENODEV;

	/* for cachedump enable */
	dev_node = of_find_compatible_node(NULL, NULL,
		"mediatek,dfd_cache");
	if (dev_node) {
		if (of_property_read_u32(dev_node, "mediatek,enabled", &val))
			drv->cachedump_en = 0;
		else
			drv->cachedump_en = val;

		if (drv->cachedump_en) {
			if (!of_property_read_u32(dev_node,
				"mediatek,rg_dfd_timeout", &val))
				drv->rg_dfd_timeout = val;
			if (!of_property_read_u32(dev_node,
				"mediatek,l2c_trigger", &val))
				drv->l2c_trigger = val;
		}
	}

	if (drv->enabled == 0)
		return 0;

	of_scan_flat_dt(fdt_get_chosen, &node);
	if (node) {
		prop = of_get_flat_dt_prop(node, "dfd,base_addr_msb", NULL);
		drv->base_addr_msb = (prop) ? of_read_number(prop, 1) : 0;
	} else {
		drv->base_addr_msb = 0;
	}

	of_scan_flat_dt(fdt_get_chosen, &node);
	if (node) {
		prop = of_get_flat_dt_prop(node,
			"dfd,cache_dump_support", NULL);
		drv->mem_reserve = (prop) ? of_read_number(prop, 1) : 0;
	} else {
		drv->mem_reserve = 0;
	}

	infra_node = of_find_compatible_node(NULL, NULL,
			"mediatek,common-infracfg_ao");
	if (infra_node) {
		void __iomem *infra = of_iomap(infra_node, 0);

		if (infra && drv->base_addr_msb) {
			infra += drv->dfd_infra_base;
			writel(readl(infra)
				| (drv->base_addr_msb >>
					drv->dfd_ap_addr_offset),
				infra);
		}
	}

	/* get base address if enabled */
	of_scan_flat_dt(fdt_get_chosen, &node);
	if (node) {
		prop = of_get_flat_dt_prop(node, "dfd,base_addr", NULL);
		drv->base_addr = (prop) ? (u64) of_read_number(prop, 2) : 0;
	} else
		return -ENODEV;

	return 0;
}

core_initcall(dfd_init);