// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (c) 2018 MediaTek Inc.
 * Author: Owen Chen <Owen.Chen@mediatek.com>
 */
#include <linux/ktime.h>
#include <linux/mfd/syscon.h>
#include <linux/of_device.h>
#include <linux/regmap.h>
#include <linux/soc/mediatek/scpsys-ext.h>

#define MTK_POLL_DELAY_US   10
#define MTK_POLL_TIMEOUT    USEC_PER_SEC

static int set_bus_protection(struct regmap *map, u32 mask, u32 ack_mask,
		u32 reg_set, u32 reg_sta, u32 reg_en)
{
	u32 val;

	if (reg_set)
		regmap_write(map, reg_set, mask);
	else
		regmap_update_bits(map, reg_en, mask, mask);

	return regmap_read_poll_timeout(map, reg_sta,
			val, (val & ack_mask) == ack_mask,
			MTK_POLL_DELAY_US, MTK_POLL_TIMEOUT);
}

static int clear_bus_protection(struct regmap *map, u32 mask, u32 ack_mask,
		u32 reg_clr, u32 reg_sta, u32 reg_en)
{
	u32 val;

	if (reg_clr)
		regmap_write(map, reg_clr, mask);
	else
		regmap_update_bits(map, reg_en, mask, 0);

	return regmap_read_poll_timeout(map, reg_sta,
			val, !(val & ack_mask),
			MTK_POLL_DELAY_US, MTK_POLL_TIMEOUT);
}

int mtk_scpsys_ext_set_bus_protection(const struct bus_prot *bp_table,
	struct regmap *infracfg, struct regmap *smi_common)
{
	int i;

	for (i = 0; i < MAX_STEPS; i++) {
		struct regmap *map;
		int ret;

		if (bp_table[i].type == INVALID_TYPE)
			continue;
		else if (bp_table[i].type == IFR_TYPE)
			map = infracfg;
		else if (bp_table[i].type == SMI_TYPE)
			map = smi_common;

		ret = set_bus_protection(map,
				bp_table[i].mask, bp_table[i].mask,
				bp_table[i].set_ofs, bp_table[i].sta_ofs,
				bp_table[i].en_ofs);

		if (ret)
			return ret;
	}

	return 0;
}

int mtk_scpsys_ext_clear_bus_protection(const struct bus_prot *bp_table,
	struct regmap *infracfg, struct regmap *smi_common)
{
	int i;

	for (i = MAX_STEPS - 1; i >= 0; i--) {
		struct regmap *map;
		int ret;

		if (bp_table[i].type == INVALID_TYPE)
			continue;
		else if (bp_table[i].type == IFR_TYPE)
			map = infracfg;
		else if (bp_table[i].type == SMI_TYPE)
			map = smi_common;

		ret = clear_bus_protection(map,
				bp_table[i].mask, bp_table[i].clr_ack_mask,
				bp_table[i].clr_ofs, bp_table[i].sta_ofs,
				bp_table[i].en_ofs);

		if (ret)
			return ret;
	}

	return 0;
}