1 // SPDX-License-Identifier: GPL-2.0
2
3 #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
4
5 #include <linux/init.h>
6 #include <linux/sched.h>
7 #include <linux/kthread.h>
8 #include <linux/workqueue.h>
9 #include <linux/memblock.h>
10
11 #include <asm/proto.h>
12 #include <asm/setup.h>
13
14 /*
15 * Some BIOSes seem to corrupt the low 64k of memory during events
16 * like suspend/resume and unplugging an HDMI cable. Reserve all
17 * remaining free memory in that area and fill it with a distinct
18 * pattern.
19 */
20 #define MAX_SCAN_AREAS 8
21
22 static int __read_mostly memory_corruption_check = -1;
23
24 static unsigned __read_mostly corruption_check_size = 64*1024;
25 static unsigned __read_mostly corruption_check_period = 60; /* seconds */
26
27 static struct scan_area {
28 u64 addr;
29 u64 size;
30 } scan_areas[MAX_SCAN_AREAS];
31 static int num_scan_areas;
32
set_corruption_check(char * arg)33 static __init int set_corruption_check(char *arg)
34 {
35 ssize_t ret;
36 unsigned long val;
37
38 if (!arg) {
39 pr_err("memory_corruption_check config string not provided\n");
40 return -EINVAL;
41 }
42
43 ret = kstrtoul(arg, 10, &val);
44 if (ret)
45 return ret;
46
47 memory_corruption_check = val;
48
49 return 0;
50 }
51 early_param("memory_corruption_check", set_corruption_check);
52
set_corruption_check_period(char * arg)53 static __init int set_corruption_check_period(char *arg)
54 {
55 ssize_t ret;
56 unsigned long val;
57
58 if (!arg) {
59 pr_err("memory_corruption_check_period config string not provided\n");
60 return -EINVAL;
61 }
62
63 ret = kstrtoul(arg, 10, &val);
64 if (ret)
65 return ret;
66
67 corruption_check_period = val;
68 return 0;
69 }
70 early_param("memory_corruption_check_period", set_corruption_check_period);
71
set_corruption_check_size(char * arg)72 static __init int set_corruption_check_size(char *arg)
73 {
74 char *end;
75 unsigned size;
76
77 if (!arg) {
78 pr_err("memory_corruption_check_size config string not provided\n");
79 return -EINVAL;
80 }
81
82 size = memparse(arg, &end);
83
84 if (*end == '\0')
85 corruption_check_size = size;
86
87 return (size == corruption_check_size) ? 0 : -EINVAL;
88 }
89 early_param("memory_corruption_check_size", set_corruption_check_size);
90
91
setup_bios_corruption_check(void)92 void __init setup_bios_corruption_check(void)
93 {
94 phys_addr_t start, end;
95 u64 i;
96
97 if (memory_corruption_check == -1) {
98 memory_corruption_check =
99 #ifdef CONFIG_X86_BOOTPARAM_MEMORY_CORRUPTION_CHECK
100 1
101 #else
102 0
103 #endif
104 ;
105 }
106
107 if (corruption_check_size == 0)
108 memory_corruption_check = 0;
109
110 if (!memory_corruption_check)
111 return;
112
113 corruption_check_size = round_up(corruption_check_size, PAGE_SIZE);
114
115 for_each_free_mem_range(i, NUMA_NO_NODE, MEMBLOCK_NONE, &start, &end,
116 NULL) {
117 start = clamp_t(phys_addr_t, round_up(start, PAGE_SIZE),
118 PAGE_SIZE, corruption_check_size);
119 end = clamp_t(phys_addr_t, round_down(end, PAGE_SIZE),
120 PAGE_SIZE, corruption_check_size);
121 if (start >= end)
122 continue;
123
124 memblock_reserve(start, end - start);
125 scan_areas[num_scan_areas].addr = start;
126 scan_areas[num_scan_areas].size = end - start;
127
128 /* Assume we've already mapped this early memory */
129 memset(__va(start), 0, end - start);
130
131 if (++num_scan_areas >= MAX_SCAN_AREAS)
132 break;
133 }
134
135 if (num_scan_areas)
136 pr_info("Scanning %d areas for low memory corruption\n", num_scan_areas);
137 }
138
139
check_for_bios_corruption(void)140 static void check_for_bios_corruption(void)
141 {
142 int i;
143 int corruption = 0;
144
145 if (!memory_corruption_check)
146 return;
147
148 for (i = 0; i < num_scan_areas; i++) {
149 unsigned long *addr = __va(scan_areas[i].addr);
150 unsigned long size = scan_areas[i].size;
151
152 for (; size; addr++, size -= sizeof(unsigned long)) {
153 if (!*addr)
154 continue;
155 pr_err("Corrupted low memory at %p (%lx phys) = %08lx\n", addr, __pa(addr), *addr);
156 corruption = 1;
157 *addr = 0;
158 }
159 }
160
161 WARN_ONCE(corruption, KERN_ERR "Memory corruption detected in low memory\n");
162 }
163
164 static void check_corruption(struct work_struct *dummy);
165 static DECLARE_DELAYED_WORK(bios_check_work, check_corruption);
166
check_corruption(struct work_struct * dummy)167 static void check_corruption(struct work_struct *dummy)
168 {
169 check_for_bios_corruption();
170 schedule_delayed_work(&bios_check_work,
171 round_jiffies_relative(corruption_check_period*HZ));
172 }
173
start_periodic_check_for_corruption(void)174 static int start_periodic_check_for_corruption(void)
175 {
176 if (!num_scan_areas || !memory_corruption_check || corruption_check_period == 0)
177 return 0;
178
179 pr_info("Scanning for low memory corruption every %d seconds\n", corruption_check_period);
180
181 /* First time we run the checks right away */
182 schedule_delayed_work(&bios_check_work, 0);
183
184 return 0;
185 }
186 device_initcall(start_periodic_check_for_corruption);
187
188