xref: /linux/arch/x86/lib/msr-smp.c (revision 44f57d78)
1 // SPDX-License-Identifier: GPL-2.0
2 #include <linux/export.h>
3 #include <linux/preempt.h>
4 #include <linux/smp.h>
5 #include <linux/completion.h>
6 #include <asm/msr.h>
7 
8 static void __rdmsr_on_cpu(void *info)
9 {
10 	struct msr_info *rv = info;
11 	struct msr *reg;
12 	int this_cpu = raw_smp_processor_id();
13 
14 	if (rv->msrs)
15 		reg = per_cpu_ptr(rv->msrs, this_cpu);
16 	else
17 		reg = &rv->reg;
18 
19 	rdmsr(rv->msr_no, reg->l, reg->h);
20 }
21 
22 static void __wrmsr_on_cpu(void *info)
23 {
24 	struct msr_info *rv = info;
25 	struct msr *reg;
26 	int this_cpu = raw_smp_processor_id();
27 
28 	if (rv->msrs)
29 		reg = per_cpu_ptr(rv->msrs, this_cpu);
30 	else
31 		reg = &rv->reg;
32 
33 	wrmsr(rv->msr_no, reg->l, reg->h);
34 }
35 
36 int rdmsr_on_cpu(unsigned int cpu, u32 msr_no, u32 *l, u32 *h)
37 {
38 	int err;
39 	struct msr_info rv;
40 
41 	memset(&rv, 0, sizeof(rv));
42 
43 	rv.msr_no = msr_no;
44 	err = smp_call_function_single(cpu, __rdmsr_on_cpu, &rv, 1);
45 	*l = rv.reg.l;
46 	*h = rv.reg.h;
47 
48 	return err;
49 }
50 EXPORT_SYMBOL(rdmsr_on_cpu);
51 
52 int rdmsrl_on_cpu(unsigned int cpu, u32 msr_no, u64 *q)
53 {
54 	int err;
55 	struct msr_info rv;
56 
57 	memset(&rv, 0, sizeof(rv));
58 
59 	rv.msr_no = msr_no;
60 	err = smp_call_function_single(cpu, __rdmsr_on_cpu, &rv, 1);
61 	*q = rv.reg.q;
62 
63 	return err;
64 }
65 EXPORT_SYMBOL(rdmsrl_on_cpu);
66 
67 int wrmsr_on_cpu(unsigned int cpu, u32 msr_no, u32 l, u32 h)
68 {
69 	int err;
70 	struct msr_info rv;
71 
72 	memset(&rv, 0, sizeof(rv));
73 
74 	rv.msr_no = msr_no;
75 	rv.reg.l = l;
76 	rv.reg.h = h;
77 	err = smp_call_function_single(cpu, __wrmsr_on_cpu, &rv, 1);
78 
79 	return err;
80 }
81 EXPORT_SYMBOL(wrmsr_on_cpu);
82 
83 int wrmsrl_on_cpu(unsigned int cpu, u32 msr_no, u64 q)
84 {
85 	int err;
86 	struct msr_info rv;
87 
88 	memset(&rv, 0, sizeof(rv));
89 
90 	rv.msr_no = msr_no;
91 	rv.reg.q = q;
92 
93 	err = smp_call_function_single(cpu, __wrmsr_on_cpu, &rv, 1);
94 
95 	return err;
96 }
97 EXPORT_SYMBOL(wrmsrl_on_cpu);
98 
99 static void __rwmsr_on_cpus(const struct cpumask *mask, u32 msr_no,
100 			    struct msr *msrs,
101 			    void (*msr_func) (void *info))
102 {
103 	struct msr_info rv;
104 	int this_cpu;
105 
106 	memset(&rv, 0, sizeof(rv));
107 
108 	rv.msrs	  = msrs;
109 	rv.msr_no = msr_no;
110 
111 	this_cpu = get_cpu();
112 
113 	if (cpumask_test_cpu(this_cpu, mask))
114 		msr_func(&rv);
115 
116 	smp_call_function_many(mask, msr_func, &rv, 1);
117 	put_cpu();
118 }
119 
120 /* rdmsr on a bunch of CPUs
121  *
122  * @mask:       which CPUs
123  * @msr_no:     which MSR
124  * @msrs:       array of MSR values
125  *
126  */
127 void rdmsr_on_cpus(const struct cpumask *mask, u32 msr_no, struct msr *msrs)
128 {
129 	__rwmsr_on_cpus(mask, msr_no, msrs, __rdmsr_on_cpu);
130 }
131 EXPORT_SYMBOL(rdmsr_on_cpus);
132 
133 /*
134  * wrmsr on a bunch of CPUs
135  *
136  * @mask:       which CPUs
137  * @msr_no:     which MSR
138  * @msrs:       array of MSR values
139  *
140  */
141 void wrmsr_on_cpus(const struct cpumask *mask, u32 msr_no, struct msr *msrs)
142 {
143 	__rwmsr_on_cpus(mask, msr_no, msrs, __wrmsr_on_cpu);
144 }
145 EXPORT_SYMBOL(wrmsr_on_cpus);
146 
147 struct msr_info_completion {
148 	struct msr_info		msr;
149 	struct completion	done;
150 };
151 
152 /* These "safe" variants are slower and should be used when the target MSR
153    may not actually exist. */
154 static void __rdmsr_safe_on_cpu(void *info)
155 {
156 	struct msr_info_completion *rv = info;
157 
158 	rv->msr.err = rdmsr_safe(rv->msr.msr_no, &rv->msr.reg.l, &rv->msr.reg.h);
159 	complete(&rv->done);
160 }
161 
162 static void __wrmsr_safe_on_cpu(void *info)
163 {
164 	struct msr_info *rv = info;
165 
166 	rv->err = wrmsr_safe(rv->msr_no, rv->reg.l, rv->reg.h);
167 }
168 
169 int rdmsr_safe_on_cpu(unsigned int cpu, u32 msr_no, u32 *l, u32 *h)
170 {
171 	struct msr_info_completion rv;
172 	call_single_data_t csd = {
173 		.func	= __rdmsr_safe_on_cpu,
174 		.info	= &rv,
175 	};
176 	int err;
177 
178 	memset(&rv, 0, sizeof(rv));
179 	init_completion(&rv.done);
180 	rv.msr.msr_no = msr_no;
181 
182 	err = smp_call_function_single_async(cpu, &csd);
183 	if (!err) {
184 		wait_for_completion(&rv.done);
185 		err = rv.msr.err;
186 	}
187 	*l = rv.msr.reg.l;
188 	*h = rv.msr.reg.h;
189 
190 	return err;
191 }
192 EXPORT_SYMBOL(rdmsr_safe_on_cpu);
193 
194 int wrmsr_safe_on_cpu(unsigned int cpu, u32 msr_no, u32 l, u32 h)
195 {
196 	int err;
197 	struct msr_info rv;
198 
199 	memset(&rv, 0, sizeof(rv));
200 
201 	rv.msr_no = msr_no;
202 	rv.reg.l = l;
203 	rv.reg.h = h;
204 	err = smp_call_function_single(cpu, __wrmsr_safe_on_cpu, &rv, 1);
205 
206 	return err ? err : rv.err;
207 }
208 EXPORT_SYMBOL(wrmsr_safe_on_cpu);
209 
210 int wrmsrl_safe_on_cpu(unsigned int cpu, u32 msr_no, u64 q)
211 {
212 	int err;
213 	struct msr_info rv;
214 
215 	memset(&rv, 0, sizeof(rv));
216 
217 	rv.msr_no = msr_no;
218 	rv.reg.q = q;
219 
220 	err = smp_call_function_single(cpu, __wrmsr_safe_on_cpu, &rv, 1);
221 
222 	return err ? err : rv.err;
223 }
224 EXPORT_SYMBOL(wrmsrl_safe_on_cpu);
225 
226 int rdmsrl_safe_on_cpu(unsigned int cpu, u32 msr_no, u64 *q)
227 {
228 	u32 low, high;
229 	int err;
230 
231 	err = rdmsr_safe_on_cpu(cpu, msr_no, &low, &high);
232 	*q = (u64)high << 32 | low;
233 
234 	return err;
235 }
236 EXPORT_SYMBOL(rdmsrl_safe_on_cpu);
237 
238 /*
239  * These variants are significantly slower, but allows control over
240  * the entire 32-bit GPR set.
241  */
242 static void __rdmsr_safe_regs_on_cpu(void *info)
243 {
244 	struct msr_regs_info *rv = info;
245 
246 	rv->err = rdmsr_safe_regs(rv->regs);
247 }
248 
249 static void __wrmsr_safe_regs_on_cpu(void *info)
250 {
251 	struct msr_regs_info *rv = info;
252 
253 	rv->err = wrmsr_safe_regs(rv->regs);
254 }
255 
256 int rdmsr_safe_regs_on_cpu(unsigned int cpu, u32 *regs)
257 {
258 	int err;
259 	struct msr_regs_info rv;
260 
261 	rv.regs   = regs;
262 	rv.err    = -EIO;
263 	err = smp_call_function_single(cpu, __rdmsr_safe_regs_on_cpu, &rv, 1);
264 
265 	return err ? err : rv.err;
266 }
267 EXPORT_SYMBOL(rdmsr_safe_regs_on_cpu);
268 
269 int wrmsr_safe_regs_on_cpu(unsigned int cpu, u32 *regs)
270 {
271 	int err;
272 	struct msr_regs_info rv;
273 
274 	rv.regs = regs;
275 	rv.err  = -EIO;
276 	err = smp_call_function_single(cpu, __wrmsr_safe_regs_on_cpu, &rv, 1);
277 
278 	return err ? err : rv.err;
279 }
280 EXPORT_SYMBOL(wrmsr_safe_regs_on_cpu);
281