/*
 * Detect Hard Lockups using the NMI
 *
 * started by Don Zickus, Copyright (C) 2010 Red Hat, Inc.
 *
 * this code detects hard lockups: incidents in where on a CPU
 * the kernel does not respond to anything except NMI.
 *
 * Note: Most of this code is borrowed heavily from softlockup.c,
 * so thanks to Ingo for the initial implementation.
 * Some chunks also taken from arch/x86/kernel/apic/nmi.c, thanks
 * to those contributors as well.
 */

#include <linux/mm.h>
#include <linux/cpu.h>
#include <linux/nmi.h>
#include <linux/init.h>
#include <linux/delay.h>
#include <linux/freezer.h>
#include <linux/lockdep.h>
#include <linux/notifier.h>
#include <linux/module.h>
#include <linux/sysctl.h>

#include <asm/irq_regs.h>
#include <linux/perf_event.h>

static DEFINE_PER_CPU(struct perf_event *, nmi_watchdog_ev);
static DEFINE_PER_CPU(int, nmi_watchdog_touch);
static DEFINE_PER_CPU(long, alert_counter);

static int panic_on_timeout;

void touch_nmi_watchdog(void)
{
	__raw_get_cpu_var(nmi_watchdog_touch) = 1;
	touch_softlockup_watchdog();
}
EXPORT_SYMBOL(touch_nmi_watchdog);

void touch_all_nmi_watchdog(void)
{
	int cpu;

	for_each_online_cpu(cpu)
		per_cpu(nmi_watchdog_touch, cpu) = 1;
	touch_softlockup_watchdog();
}

static int __init setup_nmi_watchdog(char *str)
{
	if (!strncmp(str, "panic", 5)) {
		panic_on_timeout = 1;
		str = strchr(str, ',');
		if (!str)
			return 1;
		++str;
	}
	return 1;
}
__setup("nmi_watchdog=", setup_nmi_watchdog);

struct perf_event_attr wd_hw_attr = {
	.type		= PERF_TYPE_HARDWARE,
	.config		= PERF_COUNT_HW_CPU_CYCLES,
	.size		= sizeof(struct perf_event_attr),
	.pinned		= 1,
	.disabled	= 1,
};

struct perf_event_attr wd_sw_attr = {
	.type		= PERF_TYPE_SOFTWARE,
	.config		= PERF_COUNT_SW_CPU_CLOCK,
	.size		= sizeof(struct perf_event_attr),
	.pinned		= 1,
	.disabled	= 1,
};

void wd_overflow(struct perf_event *event, int nmi,
		 struct perf_sample_data *data,
		 struct pt_regs *regs)
{
	int cpu = smp_processor_id();
	int touched = 0;

	if (__get_cpu_var(nmi_watchdog_touch)) {
		per_cpu(nmi_watchdog_touch, cpu) = 0;
		touched = 1;
	}

	/* check to see if the cpu is doing anything */
	if (!touched && hw_nmi_is_cpu_stuck(regs)) {
		/*
		 * Ayiee, looks like this CPU is stuck ...
		 * wait a few IRQs (5 seconds) before doing the oops ...
		 */
		per_cpu(alert_counter, cpu) += 1;
		if (per_cpu(alert_counter, cpu) == 5) {
			if (panic_on_timeout)
				panic("NMI Watchdog detected LOCKUP on cpu %d", cpu);
			else
				WARN(1, "NMI Watchdog detected LOCKUP on cpu %d", cpu);
		}
	} else {
		per_cpu(alert_counter, cpu) = 0;
	}

	return;
}

static int enable_nmi_watchdog(int cpu)
{
	struct perf_event *event;
	struct perf_event_attr *wd_attr;

	event = per_cpu(nmi_watchdog_ev, cpu);
	if (event && event->state > PERF_EVENT_STATE_OFF)
		return 0;

	if (event == NULL) {
		/* Try to register using hardware perf events first */
		wd_attr = &wd_hw_attr;
		wd_attr->sample_period = hw_nmi_get_sample_period();
		event = perf_event_create_kernel_counter(wd_attr, cpu, -1, wd_overflow);
		if (IS_ERR(event)) {
			/* hardware doesn't exist or not supported, fallback to software events */
			printk(KERN_INFO "nmi_watchdog: hardware not available, trying software events\n");
			wd_attr = &wd_sw_attr;
			wd_attr->sample_period = NSEC_PER_SEC;
			event = perf_event_create_kernel_counter(wd_attr, cpu, -1, wd_overflow);
			if (IS_ERR(event)) {
				printk(KERN_ERR "nmi watchdog failed to create perf event on %i: %p\n", cpu, event);
				return -1;
			}
		}
		per_cpu(nmi_watchdog_ev, cpu) = event;
	}
	perf_event_enable(per_cpu(nmi_watchdog_ev, cpu));
	return 0;
}

static void disable_nmi_watchdog(int cpu)
{
	struct perf_event *event;

	event = per_cpu(nmi_watchdog_ev, cpu);
	if (event) {
		perf_event_disable(per_cpu(nmi_watchdog_ev, cpu));
		per_cpu(nmi_watchdog_ev, cpu) = NULL;
		perf_event_release_kernel(event);
	}
}

#ifdef CONFIG_SYSCTL
/*
 * proc handler for /proc/sys/kernel/nmi_watchdog
 */
int nmi_watchdog_enabled;

int proc_nmi_enabled(struct ctl_table *table, int write,
		     void __user *buffer, size_t *length, loff_t *ppos)
{
	int cpu;

	if (!write) {
		struct perf_event *event;
		for_each_online_cpu(cpu) {
			event = per_cpu(nmi_watchdog_ev, cpu);
			if (event && event->state > PERF_EVENT_STATE_OFF) {
				nmi_watchdog_enabled = 1;
				break;
			}
		}
		proc_dointvec(table, write, buffer, length, ppos);
		return 0;
	}

	touch_all_nmi_watchdog();
	proc_dointvec(table, write, buffer, length, ppos);
	if (nmi_watchdog_enabled) {
		for_each_online_cpu(cpu)
			if (enable_nmi_watchdog(cpu)) {
				printk(KERN_ERR "NMI watchdog failed configuration, "
					" can not be enabled\n");
			}
	} else {
		for_each_online_cpu(cpu)
			disable_nmi_watchdog(cpu);
	}
	return 0;
}

#endif /* CONFIG_SYSCTL */

/*
 * Create/destroy watchdog threads as CPUs come and go:
 */
static int __cpuinit
cpu_callback(struct notifier_block *nfb, unsigned long action, void *hcpu)
{
	int hotcpu = (unsigned long)hcpu;

	switch (action) {
	case CPU_UP_PREPARE:
	case CPU_UP_PREPARE_FROZEN:
		per_cpu(nmi_watchdog_touch, hotcpu) = 0;
		break;
	case CPU_ONLINE:
	case CPU_ONLINE_FROZEN:
		if (enable_nmi_watchdog(hotcpu))
			return NOTIFY_BAD;
		break;
#ifdef CONFIG_HOTPLUG_CPU
	case CPU_UP_CANCELED:
	case CPU_UP_CANCELED_FROZEN:
		disable_nmi_watchdog(hotcpu);
	case CPU_DEAD:
	case CPU_DEAD_FROZEN:
		break;
#endif /* CONFIG_HOTPLUG_CPU */
	}
	return NOTIFY_OK;
}

static struct notifier_block __cpuinitdata cpu_nfb = {
	.notifier_call = cpu_callback
};

static int __initdata nonmi_watchdog;

static int __init nonmi_watchdog_setup(char *str)
{
	nonmi_watchdog = 1;
	return 1;
}
__setup("nonmi_watchdog", nonmi_watchdog_setup);

static int __init spawn_nmi_watchdog_task(void)
{
	void *cpu = (void *)(long)smp_processor_id();
	int err;

	if (nonmi_watchdog)
		return 0;

	printk(KERN_INFO "NMI watchdog enabled, takes one hw-pmu counter.\n");

	err = cpu_callback(&cpu_nfb, CPU_UP_PREPARE, cpu);
	if (err == NOTIFY_BAD) {
		BUG();
		return 1;
	}
	cpu_callback(&cpu_nfb, CPU_ONLINE, cpu);
	register_cpu_notifier(&cpu_nfb);

	return 0;
}
early_initcall(spawn_nmi_watchdog_task);