Lucas De Marchi | 25985ed | 2011-03-30 22:57:33 -0300 | [diff] [blame] | 1 | /* Inject a hwpoison memory failure on a arbitrary pfn */ |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 2 | #include <linux/module.h> |
| 3 | #include <linux/debugfs.h> |
| 4 | #include <linux/kernel.h> |
| 5 | #include <linux/mm.h> |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 6 | #include <linux/swap.h> |
| 7 | #include <linux/pagemap.h> |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 8 | #include <linux/hugetlb.h> |
Wu Fengguang | 7c116f2 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 9 | #include "internal.h" |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 10 | |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 11 | static struct dentry *hwpoison_dir; |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 12 | |
| 13 | static int hwpoison_inject(void *data, u64 val) |
| 14 | { |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 15 | unsigned long pfn = val; |
| 16 | struct page *p; |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 17 | struct page *hpage; |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 18 | int err; |
| 19 | |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 20 | if (!capable(CAP_SYS_ADMIN)) |
| 21 | return -EPERM; |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 22 | |
| 23 | if (!pfn_valid(pfn)) |
| 24 | return -ENXIO; |
| 25 | |
| 26 | p = pfn_to_page(pfn); |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 27 | hpage = compound_head(p); |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 28 | /* |
| 29 | * This implies unable to support free buddy pages. |
| 30 | */ |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 31 | if (!get_page_unless_zero(hpage)) |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 32 | return 0; |
| 33 | |
Wanpeng Li | fb31ba3 | 2013-09-30 13:45:24 -0700 | [diff] [blame] | 34 | if (!hwpoison_filter_enable) |
| 35 | goto inject; |
| 36 | |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 37 | if (!PageLRU(p) && !PageHuge(p)) |
Andi Kleen | facb601 | 2009-12-16 12:20:00 +0100 | [diff] [blame] | 38 | shake_page(p, 0); |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 39 | /* |
| 40 | * This implies unable to support non-LRU pages. |
| 41 | */ |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 42 | if (!PageLRU(p) && !PageHuge(p)) |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 43 | return 0; |
| 44 | |
| 45 | /* |
| 46 | * do a racy check with elevated page count, to make sure PG_hwpoison |
| 47 | * will only be set for the targeted owner (or on a free page). |
| 48 | * We temporarily take page lock for try_get_mem_cgroup_from_page(). |
Tony Luck | cd42f4a | 2011-12-15 10:48:12 -0800 | [diff] [blame] | 49 | * memory_failure() will redo the check reliably inside page lock. |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 50 | */ |
Naoya Horiguchi | 43131e1 | 2010-05-28 09:29:22 +0900 | [diff] [blame] | 51 | lock_page(hpage); |
| 52 | err = hwpoison_filter(hpage); |
| 53 | unlock_page(hpage); |
Wu Fengguang | 31d3d34 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 54 | if (err) |
| 55 | return 0; |
| 56 | |
Andi Kleen | 0d57eb8d | 2009-12-16 12:20:01 +0100 | [diff] [blame] | 57 | inject: |
Wanpeng Li | 4883e99 | 2014-01-21 15:50:56 -0800 | [diff] [blame^] | 58 | pr_info("Injecting memory failure at pfn %#lx\n", pfn); |
Tony Luck | cd42f4a | 2011-12-15 10:48:12 -0800 | [diff] [blame] | 59 | return memory_failure(pfn, 18, MF_COUNT_INCREASED); |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 60 | } |
| 61 | |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 62 | static int hwpoison_unpoison(void *data, u64 val) |
| 63 | { |
| 64 | if (!capable(CAP_SYS_ADMIN)) |
| 65 | return -EPERM; |
| 66 | |
| 67 | return unpoison_memory(val); |
| 68 | } |
| 69 | |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 70 | DEFINE_SIMPLE_ATTRIBUTE(hwpoison_fops, NULL, hwpoison_inject, "%lli\n"); |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 71 | DEFINE_SIMPLE_ATTRIBUTE(unpoison_fops, NULL, hwpoison_unpoison, "%lli\n"); |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 72 | |
| 73 | static void pfn_inject_exit(void) |
| 74 | { |
| 75 | if (hwpoison_dir) |
| 76 | debugfs_remove_recursive(hwpoison_dir); |
| 77 | } |
| 78 | |
| 79 | static int pfn_inject_init(void) |
| 80 | { |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 81 | struct dentry *dentry; |
| 82 | |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 83 | hwpoison_dir = debugfs_create_dir("hwpoison", NULL); |
| 84 | if (hwpoison_dir == NULL) |
| 85 | return -ENOMEM; |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 86 | |
| 87 | /* |
| 88 | * Note that the below poison/unpoison interfaces do not involve |
| 89 | * hardware status change, hence do not require hardware support. |
| 90 | * They are mainly for testing hwpoison in software level. |
| 91 | */ |
Wanpeng Li | 2d1e8b3 | 2013-09-11 14:23:00 -0700 | [diff] [blame] | 92 | dentry = debugfs_create_file("corrupt-pfn", 0200, hwpoison_dir, |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 93 | NULL, &hwpoison_fops); |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 94 | if (!dentry) |
| 95 | goto fail; |
| 96 | |
Wanpeng Li | 2d1e8b3 | 2013-09-11 14:23:00 -0700 | [diff] [blame] | 97 | dentry = debugfs_create_file("unpoison-pfn", 0200, hwpoison_dir, |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 98 | NULL, &unpoison_fops); |
| 99 | if (!dentry) |
| 100 | goto fail; |
| 101 | |
Haicheng Li | 1bfe5fe | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 102 | dentry = debugfs_create_u32("corrupt-filter-enable", 0600, |
| 103 | hwpoison_dir, &hwpoison_filter_enable); |
| 104 | if (!dentry) |
| 105 | goto fail; |
| 106 | |
Wu Fengguang | 7c116f2 | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 107 | dentry = debugfs_create_u32("corrupt-filter-dev-major", 0600, |
| 108 | hwpoison_dir, &hwpoison_filter_dev_major); |
| 109 | if (!dentry) |
| 110 | goto fail; |
| 111 | |
| 112 | dentry = debugfs_create_u32("corrupt-filter-dev-minor", 0600, |
| 113 | hwpoison_dir, &hwpoison_filter_dev_minor); |
| 114 | if (!dentry) |
| 115 | goto fail; |
| 116 | |
Wu Fengguang | 478c5ff | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 117 | dentry = debugfs_create_u64("corrupt-filter-flags-mask", 0600, |
| 118 | hwpoison_dir, &hwpoison_filter_flags_mask); |
| 119 | if (!dentry) |
| 120 | goto fail; |
| 121 | |
| 122 | dentry = debugfs_create_u64("corrupt-filter-flags-value", 0600, |
| 123 | hwpoison_dir, &hwpoison_filter_flags_value); |
| 124 | if (!dentry) |
| 125 | goto fail; |
| 126 | |
Andrew Morton | c255a45 | 2012-07-31 16:43:02 -0700 | [diff] [blame] | 127 | #ifdef CONFIG_MEMCG_SWAP |
Andi Kleen | 4fd466e | 2009-12-16 12:19:59 +0100 | [diff] [blame] | 128 | dentry = debugfs_create_u64("corrupt-filter-memcg", 0600, |
| 129 | hwpoison_dir, &hwpoison_filter_memcg); |
| 130 | if (!dentry) |
| 131 | goto fail; |
| 132 | #endif |
| 133 | |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 134 | return 0; |
Wu Fengguang | 847ce40 | 2009-12-16 12:19:58 +0100 | [diff] [blame] | 135 | fail: |
| 136 | pfn_inject_exit(); |
| 137 | return -ENOMEM; |
Andi Kleen | cae681f | 2009-09-16 11:50:17 +0200 | [diff] [blame] | 138 | } |
| 139 | |
| 140 | module_init(pfn_inject_init); |
| 141 | module_exit(pfn_inject_exit); |
| 142 | MODULE_LICENSE("GPL"); |