// SPDX-License-Identifier: GPL-2.0
/*
 * SDK7786 FPGA NMI Support.
 *
 * Copyright (C) 2010  Paul Mundt
 */
#include <linux/init.h>
#include <linux/kernel.h>
#include <linux/string.h>
#include <mach/fpga.h>

enum {
	NMI_MODE_MANUAL,
	NMI_MODE_AUX,
	NMI_MODE_MASKED,
	NMI_MODE_ANY,
	NMI_MODE_UNKNOWN,
};

/*
 * Default to the manual NMI switch.
 */
static unsigned int __initdata nmi_mode = NMI_MODE_ANY;

static int __init nmi_mode_setup(char *str)
{
	if (!str)
		return 0;

	if (strcmp(str, "manual") == 0)
		nmi_mode = NMI_MODE_MANUAL;
	else if (strcmp(str, "aux") == 0)
		nmi_mode = NMI_MODE_AUX;
	else if (strcmp(str, "masked") == 0)
		nmi_mode = NMI_MODE_MASKED;
	else if (strcmp(str, "any") == 0)
		nmi_mode = NMI_MODE_ANY;
	else {
		nmi_mode = NMI_MODE_UNKNOWN;
		pr_warn("Unknown NMI mode %s\n", str);
	}

	printk("Set NMI mode to %d\n", nmi_mode);
	return 0;
}
early_param("nmi_mode", nmi_mode_setup);

void __init sdk7786_nmi_init(void)
{
	unsigned int source, mask, tmp;

	switch (nmi_mode) {
	case NMI_MODE_MANUAL:
		source = NMISR_MAN_NMI;
		mask = NMIMR_MAN_NMIM;
		break;
	case NMI_MODE_AUX:
		source = NMISR_AUX_NMI;
		mask = NMIMR_AUX_NMIM;
		break;
	case NMI_MODE_ANY:
		source = NMISR_MAN_NMI | NMISR_AUX_NMI;
		mask = NMIMR_MAN_NMIM | NMIMR_AUX_NMIM;
		break;
	case NMI_MODE_MASKED:
	case NMI_MODE_UNKNOWN:
	default:
		source = mask = 0;
		break;
	}

	/* Set the NMI source */
	tmp = fpga_read_reg(NMISR);
	tmp &= ~NMISR_MASK;
	tmp |= source;
	fpga_write_reg(tmp, NMISR);

	/* And the IRQ masking */
	fpga_write_reg(NMIMR_MASK ^ mask, NMIMR);
}