1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
23
24#include <linux/kernel.h>
25#include <linux/mm.h>
26#include <linux/slab.h>
27#include <linux/vmalloc.h>
28#include <linux/hyperv.h>
29#include <linux/version.h>
30#include <linux/interrupt.h>
31#include <asm/hyperv.h>
32#include "hyperv_vmbus.h"
33
34
35struct hv_context hv_context = {
36 .synic_initialized = false,
37 .hypercall_page = NULL,
38};
39
40
41
42
43unsigned int host_info_eax;
44unsigned int host_info_ebx;
45unsigned int host_info_ecx;
46unsigned int host_info_edx;
47
48static int query_hypervisor_info(void)
49{
50 unsigned int eax;
51 unsigned int ebx;
52 unsigned int ecx;
53 unsigned int edx;
54 unsigned int max_leaf;
55 unsigned int op;
56
57
58
59
60
61 eax = 0;
62 ebx = 0;
63 ecx = 0;
64 edx = 0;
65 op = HVCPUID_VENDOR_MAXFUNCTION;
66 cpuid(op, &eax, &ebx, &ecx, &edx);
67
68 max_leaf = eax;
69
70 if (max_leaf >= HVCPUID_VERSION) {
71 eax = 0;
72 ebx = 0;
73 ecx = 0;
74 edx = 0;
75 op = HVCPUID_VERSION;
76 cpuid(op, &eax, &ebx, &ecx, &edx);
77 host_info_eax = eax;
78 host_info_ebx = ebx;
79 host_info_ecx = ecx;
80 host_info_edx = edx;
81 }
82 return max_leaf;
83}
84
85
86
87
88static u64 do_hypercall(u64 control, void *input, void *output)
89{
90#ifdef CONFIG_X86_64
91 u64 hv_status = 0;
92 u64 input_address = (input) ? virt_to_phys(input) : 0;
93 u64 output_address = (output) ? virt_to_phys(output) : 0;
94 void *hypercall_page = hv_context.hypercall_page;
95
96 __asm__ __volatile__("mov %0, %%r8" : : "r" (output_address) : "r8");
97 __asm__ __volatile__("call *%3" : "=a" (hv_status) :
98 "c" (control), "d" (input_address),
99 "m" (hypercall_page));
100
101 return hv_status;
102
103#else
104
105 u32 control_hi = control >> 32;
106 u32 control_lo = control & 0xFFFFFFFF;
107 u32 hv_status_hi = 1;
108 u32 hv_status_lo = 1;
109 u64 input_address = (input) ? virt_to_phys(input) : 0;
110 u32 input_address_hi = input_address >> 32;
111 u32 input_address_lo = input_address & 0xFFFFFFFF;
112 u64 output_address = (output) ? virt_to_phys(output) : 0;
113 u32 output_address_hi = output_address >> 32;
114 u32 output_address_lo = output_address & 0xFFFFFFFF;
115 void *hypercall_page = hv_context.hypercall_page;
116
117 __asm__ __volatile__ ("call *%8" : "=d"(hv_status_hi),
118 "=a"(hv_status_lo) : "d" (control_hi),
119 "a" (control_lo), "b" (input_address_hi),
120 "c" (input_address_lo), "D"(output_address_hi),
121 "S"(output_address_lo), "m" (hypercall_page));
122
123 return hv_status_lo | ((u64)hv_status_hi << 32);
124#endif
125}
126
127
128
129
130
131
132int hv_init(void)
133{
134 int max_leaf;
135 union hv_x64_msr_hypercall_contents hypercall_msr;
136 void *virtaddr = NULL;
137
138 memset(hv_context.synic_event_page, 0, sizeof(void *) * NR_CPUS);
139 memset(hv_context.synic_message_page, 0,
140 sizeof(void *) * NR_CPUS);
141 memset(hv_context.vp_index, 0,
142 sizeof(int) * NR_CPUS);
143 memset(hv_context.event_dpc, 0,
144 sizeof(void *) * NR_CPUS);
145
146 max_leaf = query_hypervisor_info();
147
148
149
150
151 hv_context.guestid = generate_guest_id(0, LINUX_VERSION_CODE, 0);
152 wrmsrl(HV_X64_MSR_GUEST_OS_ID, hv_context.guestid);
153
154
155 rdmsrl(HV_X64_MSR_HYPERCALL, hypercall_msr.as_uint64);
156
157 virtaddr = __vmalloc(PAGE_SIZE, GFP_KERNEL, PAGE_KERNEL_EXEC);
158
159 if (!virtaddr)
160 goto cleanup;
161
162 hypercall_msr.enable = 1;
163
164 hypercall_msr.guest_physical_address = vmalloc_to_pfn(virtaddr);
165 wrmsrl(HV_X64_MSR_HYPERCALL, hypercall_msr.as_uint64);
166
167
168 hypercall_msr.as_uint64 = 0;
169 rdmsrl(HV_X64_MSR_HYPERCALL, hypercall_msr.as_uint64);
170
171 if (!hypercall_msr.enable)
172 goto cleanup;
173
174 hv_context.hypercall_page = virtaddr;
175
176 return 0;
177
178cleanup:
179 if (virtaddr) {
180 if (hypercall_msr.enable) {
181 hypercall_msr.as_uint64 = 0;
182 wrmsrl(HV_X64_MSR_HYPERCALL, hypercall_msr.as_uint64);
183 }
184
185 vfree(virtaddr);
186 }
187
188 return -ENOTSUPP;
189}
190
191
192
193
194
195
196void hv_cleanup(void)
197{
198 union hv_x64_msr_hypercall_contents hypercall_msr;
199
200
201 wrmsrl(HV_X64_MSR_GUEST_OS_ID, 0);
202
203 if (hv_context.hypercall_page) {
204 hypercall_msr.as_uint64 = 0;
205 wrmsrl(HV_X64_MSR_HYPERCALL, hypercall_msr.as_uint64);
206 vfree(hv_context.hypercall_page);
207 hv_context.hypercall_page = NULL;
208 }
209}
210
211
212
213
214
215
216int hv_post_message(union hv_connection_id connection_id,
217 enum hv_message_type message_type,
218 void *payload, size_t payload_size)
219{
220 struct aligned_input {
221 u64 alignment8;
222 struct hv_input_post_message msg;
223 };
224
225 struct hv_input_post_message *aligned_msg;
226 u16 status;
227 unsigned long addr;
228
229 if (payload_size > HV_MESSAGE_PAYLOAD_BYTE_COUNT)
230 return -EMSGSIZE;
231
232 addr = (unsigned long)kmalloc(sizeof(struct aligned_input), GFP_ATOMIC);
233 if (!addr)
234 return -ENOMEM;
235
236 aligned_msg = (struct hv_input_post_message *)
237 (ALIGN(addr, HV_HYPERCALL_PARAM_ALIGN));
238
239 aligned_msg->connectionid = connection_id;
240 aligned_msg->message_type = message_type;
241 aligned_msg->payload_size = payload_size;
242 memcpy((void *)aligned_msg->payload, payload, payload_size);
243
244 status = do_hypercall(HVCALL_POST_MESSAGE, aligned_msg, NULL)
245 & 0xFFFF;
246
247 kfree((void *)addr);
248
249 return status;
250}
251
252
253
254
255
256
257
258
259u16 hv_signal_event(void *con_id)
260{
261 u16 status;
262
263 status = (do_hypercall(HVCALL_SIGNAL_EVENT, con_id, NULL) & 0xFFFF);
264
265 return status;
266}
267
268
269
270
271
272
273
274
275void hv_synic_init(void *arg)
276{
277 u64 version;
278 union hv_synic_simp simp;
279 union hv_synic_siefp siefp;
280 union hv_synic_sint shared_sint;
281 union hv_synic_scontrol sctrl;
282 u64 vp_index;
283
284 int cpu = smp_processor_id();
285
286 if (!hv_context.hypercall_page)
287 return;
288
289
290 rdmsrl(HV_X64_MSR_SVERSION, version);
291
292 hv_context.event_dpc[cpu] = kmalloc(sizeof(struct tasklet_struct),
293 GFP_ATOMIC);
294 if (hv_context.event_dpc[cpu] == NULL) {
295 pr_err("Unable to allocate event dpc\n");
296 goto cleanup;
297 }
298 tasklet_init(hv_context.event_dpc[cpu], vmbus_on_event, cpu);
299
300 hv_context.synic_message_page[cpu] =
301 (void *)get_zeroed_page(GFP_ATOMIC);
302
303 if (hv_context.synic_message_page[cpu] == NULL) {
304 pr_err("Unable to allocate SYNIC message page\n");
305 goto cleanup;
306 }
307
308 hv_context.synic_event_page[cpu] =
309 (void *)get_zeroed_page(GFP_ATOMIC);
310
311 if (hv_context.synic_event_page[cpu] == NULL) {
312 pr_err("Unable to allocate SYNIC event page\n");
313 goto cleanup;
314 }
315
316
317 rdmsrl(HV_X64_MSR_SIMP, simp.as_uint64);
318 simp.simp_enabled = 1;
319 simp.base_simp_gpa = virt_to_phys(hv_context.synic_message_page[cpu])
320 >> PAGE_SHIFT;
321
322 wrmsrl(HV_X64_MSR_SIMP, simp.as_uint64);
323
324
325 rdmsrl(HV_X64_MSR_SIEFP, siefp.as_uint64);
326 siefp.siefp_enabled = 1;
327 siefp.base_siefp_gpa = virt_to_phys(hv_context.synic_event_page[cpu])
328 >> PAGE_SHIFT;
329
330 wrmsrl(HV_X64_MSR_SIEFP, siefp.as_uint64);
331
332
333 rdmsrl(HV_X64_MSR_SINT0 + VMBUS_MESSAGE_SINT, shared_sint.as_uint64);
334
335 shared_sint.as_uint64 = 0;
336 shared_sint.vector = HYPERVISOR_CALLBACK_VECTOR;
337 shared_sint.masked = false;
338 shared_sint.auto_eoi = true;
339
340 wrmsrl(HV_X64_MSR_SINT0 + VMBUS_MESSAGE_SINT, shared_sint.as_uint64);
341
342
343 rdmsrl(HV_X64_MSR_SCONTROL, sctrl.as_uint64);
344 sctrl.enable = 1;
345
346 wrmsrl(HV_X64_MSR_SCONTROL, sctrl.as_uint64);
347
348 hv_context.synic_initialized = true;
349
350
351
352
353
354
355 rdmsrl(HV_X64_MSR_VP_INDEX, vp_index);
356 hv_context.vp_index[cpu] = (u32)vp_index;
357 return;
358
359cleanup:
360 if (hv_context.synic_event_page[cpu])
361 free_page((unsigned long)hv_context.synic_event_page[cpu]);
362
363 if (hv_context.synic_message_page[cpu])
364 free_page((unsigned long)hv_context.synic_message_page[cpu]);
365 return;
366}
367
368
369
370
371void hv_synic_cleanup(void *arg)
372{
373 union hv_synic_sint shared_sint;
374 union hv_synic_simp simp;
375 union hv_synic_siefp siefp;
376 int cpu = smp_processor_id();
377
378 if (!hv_context.synic_initialized)
379 return;
380
381 rdmsrl(HV_X64_MSR_SINT0 + VMBUS_MESSAGE_SINT, shared_sint.as_uint64);
382
383 shared_sint.masked = 1;
384
385
386
387 wrmsrl(HV_X64_MSR_SINT0 + VMBUS_MESSAGE_SINT, shared_sint.as_uint64);
388
389 rdmsrl(HV_X64_MSR_SIMP, simp.as_uint64);
390 simp.simp_enabled = 0;
391 simp.base_simp_gpa = 0;
392
393 wrmsrl(HV_X64_MSR_SIMP, simp.as_uint64);
394
395 rdmsrl(HV_X64_MSR_SIEFP, siefp.as_uint64);
396 siefp.siefp_enabled = 0;
397 siefp.base_siefp_gpa = 0;
398
399 wrmsrl(HV_X64_MSR_SIEFP, siefp.as_uint64);
400
401 free_page((unsigned long)hv_context.synic_message_page[cpu]);
402 free_page((unsigned long)hv_context.synic_event_page[cpu]);
403}
404