Drivers: hv: vmbus: add special kexec handler
[cascardo/linux.git] / arch / x86 / kernel / cpu / mshyperv.c
1 /*
2  * HyperV  Detection code.
3  *
4  * Copyright (C) 2010, Novell, Inc.
5  * Author : K. Y. Srinivasan <ksrinivasan@novell.com>
6  *
7  * This program is free software; you can redistribute it and/or modify
8  * it under the terms of the GNU General Public License as published by
9  * the Free Software Foundation; version 2 of the License.
10  *
11  */
12
13 #include <linux/types.h>
14 #include <linux/time.h>
15 #include <linux/clocksource.h>
16 #include <linux/module.h>
17 #include <linux/hardirq.h>
18 #include <linux/efi.h>
19 #include <linux/interrupt.h>
20 #include <linux/irq.h>
21 #include <linux/kexec.h>
22 #include <asm/processor.h>
23 #include <asm/hypervisor.h>
24 #include <asm/hyperv.h>
25 #include <asm/mshyperv.h>
26 #include <asm/desc.h>
27 #include <asm/idle.h>
28 #include <asm/irq_regs.h>
29 #include <asm/i8259.h>
30 #include <asm/apic.h>
31 #include <asm/timer.h>
32 #include <asm/reboot.h>
33
34 struct ms_hyperv_info ms_hyperv;
35 EXPORT_SYMBOL_GPL(ms_hyperv);
36
37 static void (*hv_kexec_handler)(void);
38
39 #if IS_ENABLED(CONFIG_HYPERV)
40 static void (*vmbus_handler)(void);
41
42 void hyperv_vector_handler(struct pt_regs *regs)
43 {
44         struct pt_regs *old_regs = set_irq_regs(regs);
45
46         entering_irq();
47         inc_irq_stat(irq_hv_callback_count);
48         if (vmbus_handler)
49                 vmbus_handler();
50
51         exiting_irq();
52         set_irq_regs(old_regs);
53 }
54
55 void hv_setup_vmbus_irq(void (*handler)(void))
56 {
57         vmbus_handler = handler;
58         /*
59          * Setup the IDT for hypervisor callback. Prevent reallocation
60          * at module reload.
61          */
62         if (!test_bit(HYPERVISOR_CALLBACK_VECTOR, used_vectors))
63                 alloc_intr_gate(HYPERVISOR_CALLBACK_VECTOR,
64                                 hyperv_callback_vector);
65 }
66
67 void hv_remove_vmbus_irq(void)
68 {
69         /* We have no way to deallocate the interrupt gate */
70         vmbus_handler = NULL;
71 }
72 EXPORT_SYMBOL_GPL(hv_setup_vmbus_irq);
73 EXPORT_SYMBOL_GPL(hv_remove_vmbus_irq);
74
75 void hv_setup_kexec_handler(void (*handler)(void))
76 {
77         hv_kexec_handler = handler;
78 }
79 EXPORT_SYMBOL_GPL(hv_setup_kexec_handler);
80
81 void hv_remove_kexec_handler(void)
82 {
83         hv_kexec_handler = NULL;
84 }
85 EXPORT_SYMBOL_GPL(hv_remove_kexec_handler);
86 #endif
87
88 static void hv_machine_shutdown(void)
89 {
90         if (kexec_in_progress && hv_kexec_handler)
91                 hv_kexec_handler();
92         native_machine_shutdown();
93 }
94
95 static uint32_t  __init ms_hyperv_platform(void)
96 {
97         u32 eax;
98         u32 hyp_signature[3];
99
100         if (!boot_cpu_has(X86_FEATURE_HYPERVISOR))
101                 return 0;
102
103         cpuid(HYPERV_CPUID_VENDOR_AND_MAX_FUNCTIONS,
104               &eax, &hyp_signature[0], &hyp_signature[1], &hyp_signature[2]);
105
106         if (eax >= HYPERV_CPUID_MIN &&
107             eax <= HYPERV_CPUID_MAX &&
108             !memcmp("Microsoft Hv", hyp_signature, 12))
109                 return HYPERV_CPUID_VENDOR_AND_MAX_FUNCTIONS;
110
111         return 0;
112 }
113
114 static cycle_t read_hv_clock(struct clocksource *arg)
115 {
116         cycle_t current_tick;
117         /*
118          * Read the partition counter to get the current tick count. This count
119          * is set to 0 when the partition is created and is incremented in
120          * 100 nanosecond units.
121          */
122         rdmsrl(HV_X64_MSR_TIME_REF_COUNT, current_tick);
123         return current_tick;
124 }
125
126 static struct clocksource hyperv_cs = {
127         .name           = "hyperv_clocksource",
128         .rating         = 400, /* use this when running on Hyperv*/
129         .read           = read_hv_clock,
130         .mask           = CLOCKSOURCE_MASK(64),
131         .flags          = CLOCK_SOURCE_IS_CONTINUOUS,
132 };
133
134 static void __init ms_hyperv_init_platform(void)
135 {
136         /*
137          * Extract the features and hints
138          */
139         ms_hyperv.features = cpuid_eax(HYPERV_CPUID_FEATURES);
140         ms_hyperv.hints    = cpuid_eax(HYPERV_CPUID_ENLIGHTMENT_INFO);
141
142         printk(KERN_INFO "HyperV: features 0x%x, hints 0x%x\n",
143                ms_hyperv.features, ms_hyperv.hints);
144
145 #ifdef CONFIG_X86_LOCAL_APIC
146         if (ms_hyperv.features & HV_X64_MSR_APIC_FREQUENCY_AVAILABLE) {
147                 /*
148                  * Get the APIC frequency.
149                  */
150                 u64     hv_lapic_frequency;
151
152                 rdmsrl(HV_X64_MSR_APIC_FREQUENCY, hv_lapic_frequency);
153                 hv_lapic_frequency = div_u64(hv_lapic_frequency, HZ);
154                 lapic_timer_frequency = hv_lapic_frequency;
155                 printk(KERN_INFO "HyperV: LAPIC Timer Frequency: %#x\n",
156                                 lapic_timer_frequency);
157         }
158 #endif
159
160         if (ms_hyperv.features & HV_X64_MSR_TIME_REF_COUNT_AVAILABLE)
161                 clocksource_register_hz(&hyperv_cs, NSEC_PER_SEC/100);
162
163 #ifdef CONFIG_X86_IO_APIC
164         no_timer_check = 1;
165 #endif
166
167         machine_ops.shutdown = hv_machine_shutdown;
168 }
169
170 const __refconst struct hypervisor_x86 x86_hyper_ms_hyperv = {
171         .name                   = "Microsoft HyperV",
172         .detect                 = ms_hyperv_platform,
173         .init_platform          = ms_hyperv_init_platform,
174 };
175 EXPORT_SYMBOL(x86_hyper_ms_hyperv);