]>
Commit | Line | Data |
---|---|---|
c88c4e4c HJ |
1 | /* |
2 | * Copyright (c) 2010, Microsoft Corporation. | |
3 | * | |
4 | * This program is free software; you can redistribute it and/or modify it | |
5 | * under the terms and conditions of the GNU General Public License, | |
6 | * version 2, as published by the Free Software Foundation. | |
7 | * | |
8 | * This program is distributed in the hope it will be useful, but WITHOUT | |
9 | * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
10 | * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for | |
11 | * more details. | |
12 | * | |
13 | * You should have received a copy of the GNU General Public License along with | |
14 | * this program; if not, write to the Free Software Foundation, Inc., 59 Temple | |
15 | * Place - Suite 330, Boston, MA 02111-1307 USA. | |
16 | * | |
17 | * Authors: | |
18 | * Haiyang Zhang <haiyangz@microsoft.com> | |
19 | * Hank Janssen <hjanssen@microsoft.com> | |
20 | */ | |
af7a5b6e HJ |
21 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
22 | ||
c88c4e4c HJ |
23 | #include <linux/kernel.h> |
24 | #include <linux/init.h> | |
25 | #include <linux/module.h> | |
26 | #include <linux/slab.h> | |
27 | #include <linux/sysctl.h> | |
9e629075 | 28 | #include <linux/reboot.h> |
c88c4e4c | 29 | |
3f335ea2 | 30 | #include "hyperv.h" |
245ba56a | 31 | #include "hv_kvp.h" |
c88c4e4c | 32 | |
a29b643c S |
33 | |
34 | static void shutdown_onchannelcallback(void *context); | |
35 | static struct hv_util_service util_shutdown = { | |
36 | .util_cb = shutdown_onchannelcallback, | |
37 | }; | |
38 | ||
39 | static void timesync_onchannelcallback(void *context); | |
40 | static struct hv_util_service util_timesynch = { | |
41 | .util_cb = timesync_onchannelcallback, | |
42 | }; | |
43 | ||
44 | static void heartbeat_onchannelcallback(void *context); | |
45 | static struct hv_util_service util_heartbeat = { | |
46 | .util_cb = heartbeat_onchannelcallback, | |
47 | }; | |
48 | ||
49 | static struct hv_util_service util_kvp = { | |
50 | .util_cb = hv_kvp_onchannelcallback, | |
51 | .util_init = hv_kvp_init, | |
52 | .util_deinit = hv_kvp_deinit, | |
53 | }; | |
c88c4e4c | 54 | |
6610944a | 55 | static void shutdown_onchannelcallback(void *context) |
c88c4e4c HJ |
56 | { |
57 | struct vmbus_channel *channel = context; | |
45241e50 | 58 | u32 recvlen; |
c88c4e4c HJ |
59 | u64 requestid; |
60 | u8 execute_shutdown = false; | |
a29b643c | 61 | u8 *shut_txf_buf = util_shutdown.recv_buffer; |
c88c4e4c HJ |
62 | |
63 | struct shutdown_msg_data *shutdown_msg; | |
64 | ||
65 | struct icmsg_hdr *icmsghdrp; | |
66 | struct icmsg_negotiate *negop = NULL; | |
67 | ||
45241e50 HJ |
68 | vmbus_recvpacket(channel, shut_txf_buf, |
69 | PAGE_SIZE, &recvlen, &requestid); | |
c88c4e4c HJ |
70 | |
71 | if (recvlen > 0) { | |
45241e50 | 72 | icmsghdrp = (struct icmsg_hdr *)&shut_txf_buf[ |
c88c4e4c HJ |
73 | sizeof(struct vmbuspipe_hdr)]; |
74 | ||
75 | if (icmsghdrp->icmsgtype == ICMSGTYPE_NEGOTIATE) { | |
45241e50 | 76 | prep_negotiate_resp(icmsghdrp, negop, shut_txf_buf); |
c88c4e4c | 77 | } else { |
45241e50 HJ |
78 | shutdown_msg = |
79 | (struct shutdown_msg_data *)&shut_txf_buf[ | |
80 | sizeof(struct vmbuspipe_hdr) + | |
81 | sizeof(struct icmsg_hdr)]; | |
c88c4e4c HJ |
82 | |
83 | switch (shutdown_msg->flags) { | |
84 | case 0: | |
85 | case 1: | |
86 | icmsghdrp->status = HV_S_OK; | |
87 | execute_shutdown = true; | |
88 | ||
af7a5b6e | 89 | pr_info("Shutdown request received -" |
32235b07 | 90 | " graceful shutdown initiated\n"); |
c88c4e4c HJ |
91 | break; |
92 | default: | |
93 | icmsghdrp->status = HV_E_FAIL; | |
94 | execute_shutdown = false; | |
95 | ||
af7a5b6e HJ |
96 | pr_info("Shutdown request received -" |
97 | " Invalid request\n"); | |
c88c4e4c | 98 | break; |
95cd17c9 | 99 | } |
c88c4e4c HJ |
100 | } |
101 | ||
102 | icmsghdrp->icflags = ICMSGHDRFLAG_TRANSACTION | |
103 | | ICMSGHDRFLAG_RESPONSE; | |
104 | ||
45241e50 | 105 | vmbus_sendpacket(channel, shut_txf_buf, |
c88c4e4c | 106 | recvlen, requestid, |
415f2287 | 107 | VM_PKT_DATA_INBAND, 0); |
c88c4e4c HJ |
108 | } |
109 | ||
c88c4e4c | 110 | if (execute_shutdown == true) |
ae2a0b42 | 111 | orderly_poweroff(true); |
c88c4e4c HJ |
112 | } |
113 | ||
39c4e9c3 | 114 | /* |
e9ec3603 | 115 | * Set guest time to host UTC time. |
39c4e9c3 | 116 | */ |
e9ec3603 | 117 | static inline void do_adj_guesttime(u64 hosttime) |
39c4e9c3 HZ |
118 | { |
119 | s64 host_tns; | |
120 | struct timespec host_ts; | |
39c4e9c3 HZ |
121 | |
122 | host_tns = (hosttime - WLTIMEDELTA) * 100; | |
123 | host_ts = ns_to_timespec(host_tns); | |
124 | ||
e9ec3603 HZ |
125 | do_settimeofday(&host_ts); |
126 | } | |
127 | ||
95ff7cde S |
128 | /* |
129 | * Set the host time in a process context. | |
130 | */ | |
131 | ||
132 | struct adj_time_work { | |
133 | struct work_struct work; | |
134 | u64 host_time; | |
135 | }; | |
136 | ||
137 | static void hv_set_host_time(struct work_struct *work) | |
138 | { | |
139 | struct adj_time_work *wrk; | |
140 | ||
141 | wrk = container_of(work, struct adj_time_work, work); | |
142 | do_adj_guesttime(wrk->host_time); | |
143 | kfree(wrk); | |
144 | } | |
145 | ||
e9ec3603 HZ |
146 | /* |
147 | * Synchronize time with host after reboot, restore, etc. | |
148 | * | |
149 | * ICTIMESYNCFLAG_SYNC flag bit indicates reboot, restore events of the VM. | |
150 | * After reboot the flag ICTIMESYNCFLAG_SYNC is included in the first time | |
151 | * message after the timesync channel is opened. Since the hv_utils module is | |
152 | * loaded after hv_vmbus, the first message is usually missed. The other | |
153 | * thing is, systime is automatically set to emulated hardware clock which may | |
154 | * not be UTC time or in the same time zone. So, to override these effects, we | |
155 | * use the first 50 time samples for initial system time setting. | |
156 | */ | |
157 | static inline void adj_guesttime(u64 hosttime, u8 flags) | |
158 | { | |
95ff7cde | 159 | struct adj_time_work *wrk; |
e9ec3603 HZ |
160 | static s32 scnt = 50; |
161 | ||
95ff7cde S |
162 | wrk = kmalloc(sizeof(struct adj_time_work), GFP_ATOMIC); |
163 | if (wrk == NULL) | |
164 | return; | |
165 | ||
166 | wrk->host_time = hosttime; | |
39c4e9c3 | 167 | if ((flags & ICTIMESYNCFLAG_SYNC) != 0) { |
95ff7cde S |
168 | INIT_WORK(&wrk->work, hv_set_host_time); |
169 | schedule_work(&wrk->work); | |
39c4e9c3 HZ |
170 | return; |
171 | } | |
172 | ||
e9ec3603 | 173 | if ((flags & ICTIMESYNCFLAG_SAMPLE) != 0 && scnt > 0) { |
39c4e9c3 | 174 | scnt--; |
95ff7cde S |
175 | INIT_WORK(&wrk->work, hv_set_host_time); |
176 | schedule_work(&wrk->work); | |
177 | } else | |
178 | kfree(wrk); | |
39c4e9c3 HZ |
179 | } |
180 | ||
181 | /* | |
182 | * Time Sync Channel message handler. | |
183 | */ | |
184 | static void timesync_onchannelcallback(void *context) | |
185 | { | |
186 | struct vmbus_channel *channel = context; | |
45241e50 | 187 | u32 recvlen; |
39c4e9c3 HZ |
188 | u64 requestid; |
189 | struct icmsg_hdr *icmsghdrp; | |
190 | struct ictimesync_data *timedatap; | |
a29b643c | 191 | u8 *time_txf_buf = util_timesynch.recv_buffer; |
39c4e9c3 | 192 | |
45241e50 HJ |
193 | vmbus_recvpacket(channel, time_txf_buf, |
194 | PAGE_SIZE, &recvlen, &requestid); | |
39c4e9c3 HZ |
195 | |
196 | if (recvlen > 0) { | |
45241e50 | 197 | icmsghdrp = (struct icmsg_hdr *)&time_txf_buf[ |
39c4e9c3 HZ |
198 | sizeof(struct vmbuspipe_hdr)]; |
199 | ||
200 | if (icmsghdrp->icmsgtype == ICMSGTYPE_NEGOTIATE) { | |
45241e50 | 201 | prep_negotiate_resp(icmsghdrp, NULL, time_txf_buf); |
39c4e9c3 | 202 | } else { |
45241e50 | 203 | timedatap = (struct ictimesync_data *)&time_txf_buf[ |
39c4e9c3 HZ |
204 | sizeof(struct vmbuspipe_hdr) + |
205 | sizeof(struct icmsg_hdr)]; | |
206 | adj_guesttime(timedatap->parenttime, timedatap->flags); | |
207 | } | |
208 | ||
209 | icmsghdrp->icflags = ICMSGHDRFLAG_TRANSACTION | |
210 | | ICMSGHDRFLAG_RESPONSE; | |
211 | ||
45241e50 | 212 | vmbus_sendpacket(channel, time_txf_buf, |
39c4e9c3 | 213 | recvlen, requestid, |
415f2287 | 214 | VM_PKT_DATA_INBAND, 0); |
39c4e9c3 | 215 | } |
39c4e9c3 HZ |
216 | } |
217 | ||
9153f7b9 HJ |
218 | /* |
219 | * Heartbeat functionality. | |
220 | * Every two seconds, Hyper-V send us a heartbeat request message. | |
221 | * we respond to this message, and Hyper-V knows we are alive. | |
222 | */ | |
223 | static void heartbeat_onchannelcallback(void *context) | |
224 | { | |
225 | struct vmbus_channel *channel = context; | |
45241e50 | 226 | u32 recvlen; |
9153f7b9 HJ |
227 | u64 requestid; |
228 | struct icmsg_hdr *icmsghdrp; | |
229 | struct heartbeat_msg_data *heartbeat_msg; | |
a29b643c | 230 | u8 *hbeat_txf_buf = util_heartbeat.recv_buffer; |
9153f7b9 | 231 | |
45241e50 HJ |
232 | vmbus_recvpacket(channel, hbeat_txf_buf, |
233 | PAGE_SIZE, &recvlen, &requestid); | |
9153f7b9 HJ |
234 | |
235 | if (recvlen > 0) { | |
45241e50 | 236 | icmsghdrp = (struct icmsg_hdr *)&hbeat_txf_buf[ |
9153f7b9 HJ |
237 | sizeof(struct vmbuspipe_hdr)]; |
238 | ||
239 | if (icmsghdrp->icmsgtype == ICMSGTYPE_NEGOTIATE) { | |
45241e50 | 240 | prep_negotiate_resp(icmsghdrp, NULL, hbeat_txf_buf); |
9153f7b9 | 241 | } else { |
45241e50 HJ |
242 | heartbeat_msg = |
243 | (struct heartbeat_msg_data *)&hbeat_txf_buf[ | |
244 | sizeof(struct vmbuspipe_hdr) + | |
245 | sizeof(struct icmsg_hdr)]; | |
9153f7b9 | 246 | |
9153f7b9 HJ |
247 | heartbeat_msg->seq_num += 1; |
248 | } | |
249 | ||
250 | icmsghdrp->icflags = ICMSGHDRFLAG_TRANSACTION | |
251 | | ICMSGHDRFLAG_RESPONSE; | |
252 | ||
45241e50 | 253 | vmbus_sendpacket(channel, hbeat_txf_buf, |
9153f7b9 | 254 | recvlen, requestid, |
415f2287 | 255 | VM_PKT_DATA_INBAND, 0); |
9153f7b9 | 256 | } |
9153f7b9 | 257 | } |
39c4e9c3 | 258 | |
84946899 S |
259 | static int util_probe(struct hv_device *dev, |
260 | const struct hv_vmbus_device_id *dev_id) | |
283f2129 | 261 | { |
a29b643c S |
262 | struct hv_util_service *srv = |
263 | (struct hv_util_service *)dev_id->driver_data; | |
264 | int ret; | |
265 | ||
266 | srv->recv_buffer = kmalloc(PAGE_SIZE, GFP_KERNEL); | |
267 | if (!srv->recv_buffer) | |
268 | return -ENOMEM; | |
269 | if (srv->util_init) { | |
270 | ret = srv->util_init(srv); | |
271 | if (ret) { | |
272 | kfree(srv->recv_buffer); | |
273 | return -ENODEV; | |
274 | } | |
275 | } | |
276 | ||
277 | hv_set_drvdata(dev, srv); | |
283f2129 S |
278 | return 0; |
279 | } | |
280 | ||
281 | static int util_remove(struct hv_device *dev) | |
282 | { | |
a29b643c S |
283 | struct hv_util_service *srv = hv_get_drvdata(dev); |
284 | ||
285 | if (srv->util_deinit) | |
286 | srv->util_deinit(); | |
287 | kfree(srv->recv_buffer); | |
288 | ||
283f2129 S |
289 | return 0; |
290 | } | |
291 | ||
292 | static const struct hv_vmbus_device_id id_table[] = { | |
c45cf2d4 GKH |
293 | /* Shutdown guid */ |
294 | { VMBUS_DEVICE(0x31, 0x60, 0x0B, 0X0E, 0x13, 0x52, 0x34, 0x49, | |
a29b643c S |
295 | 0x81, 0x8B, 0x38, 0XD9, 0x0C, 0xED, 0x39, 0xDB) |
296 | .driver_data = (unsigned long)&util_shutdown }, | |
c45cf2d4 GKH |
297 | /* Time synch guid */ |
298 | { VMBUS_DEVICE(0x30, 0xe6, 0x27, 0x95, 0xae, 0xd0, 0x7b, 0x49, | |
a29b643c S |
299 | 0xad, 0xce, 0xe8, 0x0a, 0xb0, 0x17, 0x5c, 0xaf) |
300 | .driver_data = (unsigned long)&util_timesynch }, | |
c45cf2d4 GKH |
301 | /* Heartbeat guid */ |
302 | { VMBUS_DEVICE(0x39, 0x4f, 0x16, 0x57, 0x15, 0x91, 0x78, 0x4e, | |
a29b643c S |
303 | 0xab, 0x55, 0x38, 0x2f, 0x3b, 0xd5, 0x42, 0x2d) |
304 | .driver_data = (unsigned long)&util_heartbeat }, | |
c45cf2d4 GKH |
305 | /* KVP guid */ |
306 | { VMBUS_DEVICE(0xe7, 0xf4, 0xa0, 0xa9, 0x45, 0x5a, 0x96, 0x4d, | |
a29b643c S |
307 | 0xb8, 0x27, 0x8a, 0x84, 0x1e, 0x8c, 0x3, 0xe6) |
308 | .driver_data = (unsigned long)&util_kvp }, | |
c45cf2d4 | 309 | { }, |
283f2129 S |
310 | }; |
311 | ||
312 | MODULE_DEVICE_TABLE(vmbus, id_table); | |
313 | ||
314 | /* The one and only one */ | |
315 | static struct hv_driver util_drv = { | |
768fa219 | 316 | .name = "hv_util", |
283f2129 S |
317 | .id_table = id_table, |
318 | .probe = util_probe, | |
319 | .remove = util_remove, | |
320 | }; | |
321 | ||
c88c4e4c HJ |
322 | static int __init init_hyperv_utils(void) |
323 | { | |
d531babe | 324 | int ret; |
af7a5b6e | 325 | pr_info("Registering HyperV Utility Driver\n"); |
c88c4e4c | 326 | |
45241e50 | 327 | |
d531babe S |
328 | ret = vmbus_driver_register(&util_drv); |
329 | ||
330 | if (ret != 0) | |
a29b643c | 331 | return ret; |
d531babe | 332 | |
c88c4e4c HJ |
333 | hv_cb_utils[HV_SHUTDOWN_MSG].callback = &shutdown_onchannelcallback; |
334 | ||
39c4e9c3 HZ |
335 | hv_cb_utils[HV_TIMESYNC_MSG].callback = ×ync_onchannelcallback; |
336 | ||
9153f7b9 HJ |
337 | hv_cb_utils[HV_HEARTBEAT_MSG].callback = &heartbeat_onchannelcallback; |
338 | ||
b9d8e352 | 339 | hv_cb_utils[HV_KVP_MSG].callback = &hv_kvp_onchannelcallback; |
245ba56a | 340 | |
d531babe S |
341 | return 0; |
342 | ||
c88c4e4c HJ |
343 | } |
344 | ||
345 | static void exit_hyperv_utils(void) | |
346 | { | |
af7a5b6e | 347 | pr_info("De-Registered HyperV Utility Driver\n"); |
c88c4e4c | 348 | |
b9d8e352 S |
349 | if (hv_cb_utils[HV_SHUTDOWN_MSG].channel != NULL) |
350 | hv_cb_utils[HV_SHUTDOWN_MSG].channel->onchannel_callback = | |
351 | &chn_cb_negotiate; | |
352 | hv_cb_utils[HV_SHUTDOWN_MSG].callback = NULL; | |
353 | ||
354 | if (hv_cb_utils[HV_TIMESYNC_MSG].channel != NULL) | |
355 | hv_cb_utils[HV_TIMESYNC_MSG].channel->onchannel_callback = | |
356 | &chn_cb_negotiate; | |
357 | hv_cb_utils[HV_TIMESYNC_MSG].callback = NULL; | |
39c4e9c3 | 358 | |
b9d8e352 S |
359 | if (hv_cb_utils[HV_HEARTBEAT_MSG].channel != NULL) |
360 | hv_cb_utils[HV_HEARTBEAT_MSG].channel->onchannel_callback = | |
361 | &chn_cb_negotiate; | |
362 | hv_cb_utils[HV_HEARTBEAT_MSG].callback = NULL; | |
9153f7b9 | 363 | |
b9d8e352 S |
364 | if (hv_cb_utils[HV_KVP_MSG].channel != NULL) |
365 | hv_cb_utils[HV_KVP_MSG].channel->onchannel_callback = | |
366 | &chn_cb_negotiate; | |
367 | hv_cb_utils[HV_KVP_MSG].callback = NULL; | |
45241e50 | 368 | |
768fa219 | 369 | vmbus_driver_unregister(&util_drv); |
c88c4e4c HJ |
370 | } |
371 | ||
372 | module_init(init_hyperv_utils); | |
373 | module_exit(exit_hyperv_utils); | |
374 | ||
375 | MODULE_DESCRIPTION("Hyper-V Utilities"); | |
376 | MODULE_VERSION(HV_DRV_VERSION); | |
377 | MODULE_LICENSE("GPL"); |