[net-next-2.6.git] / mm / hwpoison-inject.c

/* Inject a hwpoison memory failure on a arbitary pfn */
#include <linux/module.h>
#include <linux/debugfs.h>
#include <linux/kernel.h>
#include <linux/mm.h>

static struct dentry *hwpoison_dir;

static int hwpoison_inject(void *data, u64 val)
{
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;
	printk(KERN_INFO "Injecting memory failure at pfn %Lx\n", val);
	return __memory_failure(val, 18, 0);
}

static int hwpoison_unpoison(void *data, u64 val)
{
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

	return unpoison_memory(val);
}

DEFINE_SIMPLE_ATTRIBUTE(hwpoison_fops, NULL, hwpoison_inject, "%lli\n");
DEFINE_SIMPLE_ATTRIBUTE(unpoison_fops, NULL, hwpoison_unpoison, "%lli\n");

static void pfn_inject_exit(void)
{
	if (hwpoison_dir)
		debugfs_remove_recursive(hwpoison_dir);
}

static int pfn_inject_init(void)
{
	struct dentry *dentry;

	hwpoison_dir = debugfs_create_dir("hwpoison", NULL);
	if (hwpoison_dir == NULL)
		return -ENOMEM;

	/*
	 * Note that the below poison/unpoison interfaces do not involve
	 * hardware status change, hence do not require hardware support.
	 * They are mainly for testing hwpoison in software level.
	 */
	dentry = debugfs_create_file("corrupt-pfn", 0600, hwpoison_dir,
					  NULL, &hwpoison_fops);
	if (!dentry)
		goto fail;

	dentry = debugfs_create_file("unpoison-pfn", 0600, hwpoison_dir,
				     NULL, &unpoison_fops);
	if (!dentry)
		goto fail;

	return 0;
fail:
	pfn_inject_exit();
	return -ENOMEM;
}

module_init(pfn_inject_init);
module_exit(pfn_inject_exit);
MODULE_LICENSE("GPL");
Commit	Line	Data
cae681fc AK	1	/* Inject a hwpoison memory failure on a arbitary pfn */
	2	#include <linux/module.h>
	3	#include <linux/debugfs.h>
	4	#include <linux/kernel.h>
	5	#include <linux/mm.h>
	6
847ce401	7	static struct dentry *hwpoison_dir;
cae681fc AK	8
	9	static int hwpoison_inject(void *data, u64 val)
	10	{
	11	if (!capable(CAP_SYS_ADMIN))
	12	return -EPERM;
	13	printk(KERN_INFO "Injecting memory failure at pfn %Lx\n", val);
	14	return __memory_failure(val, 18, 0);
	15	}
	16
847ce401 WF	17	static int hwpoison_unpoison(void *data, u64 val)
	18	{
	19	if (!capable(CAP_SYS_ADMIN))
	20	return -EPERM;
	21
	22	return unpoison_memory(val);
	23	}
	24
cae681fc	25	DEFINE_SIMPLE_ATTRIBUTE(hwpoison_fops, NULL, hwpoison_inject, "%lli\n");
847ce401	26	DEFINE_SIMPLE_ATTRIBUTE(unpoison_fops, NULL, hwpoison_unpoison, "%lli\n");
cae681fc AK	27
	28	static void pfn_inject_exit(void)
	29	{
	30	if (hwpoison_dir)
	31	debugfs_remove_recursive(hwpoison_dir);
	32	}
	33
	34	static int pfn_inject_init(void)
	35	{
847ce401 WF	36	struct dentry *dentry;
847ce401 WF	37
cae681fc AK	38	hwpoison_dir = debugfs_create_dir("hwpoison", NULL);
	39	if (hwpoison_dir == NULL)
	40	return -ENOMEM;
847ce401 WF	41
	42	/*
	43	* Note that the below poison/unpoison interfaces do not involve
	44	* hardware status change, hence do not require hardware support.
	45	* They are mainly for testing hwpoison in software level.
	46	*/
	47	dentry = debugfs_create_file("corrupt-pfn", 0600, hwpoison_dir,
cae681fc	48	NULL, &hwpoison_fops);
847ce401 WF	49	if (!dentry)
	50	goto fail;
	51
	52	dentry = debugfs_create_file("unpoison-pfn", 0600, hwpoison_dir,
	53	NULL, &unpoison_fops);
	54	if (!dentry)
	55	goto fail;
	56
cae681fc	57	return 0;
847ce401 WF	58	fail:
	59	pfn_inject_exit();
	60	return -ENOMEM;
cae681fc AK	61	}
	62
	63	module_init(pfn_inject_init);
	64	module_exit(pfn_inject_exit);
	65	MODULE_LICENSE("GPL");