[net-next-2.6.git] / lib / cpu-notifier-error-inject.c

#include <linux/kernel.h>
#include <linux/cpu.h>
#include <linux/module.h>
#include <linux/notifier.h>

static int priority;
static int cpu_up_prepare_error;
static int cpu_down_prepare_error;

module_param(priority, int, 0);
MODULE_PARM_DESC(priority, "specify cpu notifier priority");

module_param(cpu_up_prepare_error, int, 0644);
MODULE_PARM_DESC(cpu_up_prepare_error,
		"specify error code to inject CPU_UP_PREPARE action");

module_param(cpu_down_prepare_error, int, 0644);
MODULE_PARM_DESC(cpu_down_prepare_error,
		"specify error code to inject CPU_DOWN_PREPARE action");

static int err_inject_cpu_callback(struct notifier_block *nfb,
				unsigned long action, void *hcpu)
{
	int err = 0;

	switch (action) {
	case CPU_UP_PREPARE:
	case CPU_UP_PREPARE_FROZEN:
		err = cpu_up_prepare_error;
		break;
	case CPU_DOWN_PREPARE:
	case CPU_DOWN_PREPARE_FROZEN:
		err = cpu_down_prepare_error;
		break;
	}
	if (err)
		printk(KERN_INFO "Injecting error (%d) at cpu notifier\n", err);

	return notifier_from_errno(err);
}

static struct notifier_block err_inject_cpu_notifier = {
	.notifier_call = err_inject_cpu_callback,
};

static int err_inject_init(void)
{
	err_inject_cpu_notifier.priority = priority;

	return register_hotcpu_notifier(&err_inject_cpu_notifier);
}

static void err_inject_exit(void)
{
	unregister_hotcpu_notifier(&err_inject_cpu_notifier);
}

module_init(err_inject_init);
module_exit(err_inject_exit);

MODULE_DESCRIPTION("CPU notifier error injection module");
MODULE_LICENSE("GPL");
MODULE_AUTHOR("Akinobu Mita <akinobu.mita@gmail.com>");
Commit	Line	Data
c9d221f8 AM	1	#include <linux/kernel.h>
	2	#include <linux/cpu.h>
	3	#include <linux/module.h>
	4	#include <linux/notifier.h>
	5
	6	static int priority;
	7	static int cpu_up_prepare_error;
	8	static int cpu_down_prepare_error;
	9
	10	module_param(priority, int, 0);
	11	MODULE_PARM_DESC(priority, "specify cpu notifier priority");
	12
	13	module_param(cpu_up_prepare_error, int, 0644);
	14	MODULE_PARM_DESC(cpu_up_prepare_error,
	15	"specify error code to inject CPU_UP_PREPARE action");
	16
	17	module_param(cpu_down_prepare_error, int, 0644);
	18	MODULE_PARM_DESC(cpu_down_prepare_error,
	19	"specify error code to inject CPU_DOWN_PREPARE action");
	20
	21	static int err_inject_cpu_callback(struct notifier_block *nfb,
	22	unsigned long action, void *hcpu)
	23	{
	24	int err = 0;
	25
	26	switch (action) {
	27	case CPU_UP_PREPARE:
	28	case CPU_UP_PREPARE_FROZEN:
	29	err = cpu_up_prepare_error;
	30	break;
	31	case CPU_DOWN_PREPARE:
	32	case CPU_DOWN_PREPARE_FROZEN:
	33	err = cpu_down_prepare_error;
	34	break;
	35	}
	36	if (err)
	37	printk(KERN_INFO "Injecting error (%d) at cpu notifier\n", err);
	38
	39	return notifier_from_errno(err);
	40	}
	41
	42	static struct notifier_block err_inject_cpu_notifier = {
	43	.notifier_call = err_inject_cpu_callback,
	44	};
	45
	46	static int err_inject_init(void)
	47	{
	48	err_inject_cpu_notifier.priority = priority;
	49
	50	return register_hotcpu_notifier(&err_inject_cpu_notifier);
	51	}
	52
	53	static void err_inject_exit(void)
	54	{
	55	unregister_hotcpu_notifier(&err_inject_cpu_notifier);
	56	}
	57
	58	module_init(err_inject_init);
	59	module_exit(err_inject_exit);
	60
	61	MODULE_DESCRIPTION("CPU notifier error injection module");
	62	MODULE_LICENSE("GPL");
	63	MODULE_AUTHOR("Akinobu Mita <akinobu.mita@gmail.com>");