1
2
3
4
5
6
7
8
9
10
11
12
13
14#include <asm-generic/sections.h>
15#include <asm/cpu_device_id.h>
16#include <asm/imr.h>
17#include <linux/init.h>
18#include <linux/mm.h>
19#include <linux/types.h>
20
21#define SELFTEST KBUILD_MODNAME ": "
22
23
24
25
26
27
28
29static __printf(2, 3)
30void __init imr_self_test_result(int res, const char *fmt, ...)
31{
32 va_list vlist;
33
34
35 if (res)
36 pr_info(SELFTEST "pass ");
37 else
38 pr_info(SELFTEST "fail ");
39
40
41 va_start(vlist, fmt);
42 vprintk(fmt, vlist);
43 va_end(vlist);
44
45
46 WARN(res == 0, "test failed");
47}
48#undef SELFTEST
49
50
51
52
53
54
55
56
57static void __init imr_self_test(void)
58{
59 phys_addr_t base = virt_to_phys(&_text);
60 size_t size = virt_to_phys(&__end_rodata) - base;
61 const char *fmt_over = "overlapped IMR @ (0x%08lx - 0x%08lx)\n";
62 int ret;
63
64
65 ret = imr_add_range(0, 0, 0, 0);
66 imr_self_test_result(ret < 0, "zero sized IMR\n");
67
68
69 ret = imr_add_range(base, size, IMR_CPU, IMR_CPU);
70 imr_self_test_result(ret < 0, fmt_over, __va(base), __va(base + size));
71
72
73 base += size - IMR_ALIGN;
74 ret = imr_add_range(base, size, IMR_CPU, IMR_CPU);
75 imr_self_test_result(ret < 0, fmt_over, __va(base), __va(base + size));
76
77
78 base -= size + IMR_ALIGN * 2;
79 ret = imr_add_range(base, size, IMR_CPU, IMR_CPU);
80 imr_self_test_result(ret < 0, fmt_over, __va(base), __va(base + size));
81
82
83 ret = imr_add_range(0, IMR_ALIGN, IMR_READ_ACCESS_ALL,
84 IMR_WRITE_ACCESS_ALL);
85 imr_self_test_result(ret < 0, "1KiB IMR @ 0x00000000 - access-all\n");
86
87
88 ret = imr_add_range(0, IMR_ALIGN, IMR_CPU, IMR_CPU);
89 imr_self_test_result(ret >= 0, "1KiB IMR @ 0x00000000 - cpu-access\n");
90 if (ret >= 0) {
91 ret = imr_remove_range(0, IMR_ALIGN);
92 imr_self_test_result(ret == 0, "teardown - cpu-access\n");
93 }
94
95
96 size = IMR_ALIGN * 2;
97 ret = imr_add_range(0, size, IMR_READ_ACCESS_ALL, IMR_WRITE_ACCESS_ALL);
98 imr_self_test_result(ret >= 0, "2KiB IMR @ 0x00000000\n");
99 if (ret >= 0) {
100 ret = imr_remove_range(0, size);
101 imr_self_test_result(ret == 0, "teardown 2KiB\n");
102 }
103}
104
105static const struct x86_cpu_id imr_ids[] __initconst = {
106 { X86_VENDOR_INTEL, 5, 9 },
107 {}
108};
109
110
111
112
113
114
115static int __init imr_self_test_init(void)
116{
117 if (x86_match_cpu(imr_ids))
118 imr_self_test();
119 return 0;
120}
121
122
123
124
125
126
127device_initcall(imr_self_test_init);
128