Commit | Line | Data |
---|---|---|
3b20eb23 | 1 | // SPDX-License-Identifier: GPL-2.0-only |
3e7ee490 | 2 | /* |
3e7ee490 HJ |
3 | * Copyright (c) 2009, Microsoft Corporation. |
4 | * | |
3e7ee490 HJ |
5 | * Authors: |
6 | * Haiyang Zhang <haiyangz@microsoft.com> | |
7 | * Hank Janssen <hjanssen@microsoft.com> | |
3e7ee490 | 8 | */ |
0a46618d HJ |
9 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
10 | ||
a0086dc5 | 11 | #include <linux/kernel.h> |
638fea33 | 12 | #include <linux/interrupt.h> |
0c3b7b2f S |
13 | #include <linux/sched.h> |
14 | #include <linux/wait.h> | |
a0086dc5 | 15 | #include <linux/mm.h> |
5a0e3ad6 | 16 | #include <linux/slab.h> |
53af545b | 17 | #include <linux/list.h> |
c88c4e4c | 18 | #include <linux/module.h> |
8b5d6d3b | 19 | #include <linux/completion.h> |
41571916 | 20 | #include <linux/delay.h> |
46a97191 | 21 | #include <linux/hyperv.h> |
e810e48c | 22 | #include <asm/mshyperv.h> |
3f335ea2 | 23 | |
0f2a6619 | 24 | #include "hyperv_vmbus.h" |
3e7ee490 | 25 | |
7047f17d S |
26 | static void init_vp_index(struct vmbus_channel *channel, u16 dev_type); |
27 | ||
28 | static const struct vmbus_device vmbus_devs[] = { | |
29 | /* IDE */ | |
30 | { .dev_type = HV_IDE, | |
31 | HV_IDE_GUID, | |
32 | .perf_device = true, | |
33 | }, | |
34 | ||
35 | /* SCSI */ | |
36 | { .dev_type = HV_SCSI, | |
37 | HV_SCSI_GUID, | |
38 | .perf_device = true, | |
39 | }, | |
40 | ||
41 | /* Fibre Channel */ | |
42 | { .dev_type = HV_FC, | |
43 | HV_SYNTHFC_GUID, | |
44 | .perf_device = true, | |
45 | }, | |
46 | ||
47 | /* Synthetic NIC */ | |
48 | { .dev_type = HV_NIC, | |
49 | HV_NIC_GUID, | |
50 | .perf_device = true, | |
51 | }, | |
52 | ||
53 | /* Network Direct */ | |
54 | { .dev_type = HV_ND, | |
55 | HV_ND_GUID, | |
56 | .perf_device = true, | |
57 | }, | |
58 | ||
59 | /* PCIE */ | |
60 | { .dev_type = HV_PCIE, | |
61 | HV_PCIE_GUID, | |
238064f1 | 62 | .perf_device = false, |
7047f17d S |
63 | }, |
64 | ||
65 | /* Synthetic Frame Buffer */ | |
66 | { .dev_type = HV_FB, | |
67 | HV_SYNTHVID_GUID, | |
68 | .perf_device = false, | |
69 | }, | |
70 | ||
71 | /* Synthetic Keyboard */ | |
72 | { .dev_type = HV_KBD, | |
73 | HV_KBD_GUID, | |
74 | .perf_device = false, | |
75 | }, | |
76 | ||
77 | /* Synthetic MOUSE */ | |
78 | { .dev_type = HV_MOUSE, | |
79 | HV_MOUSE_GUID, | |
80 | .perf_device = false, | |
81 | }, | |
82 | ||
83 | /* KVP */ | |
84 | { .dev_type = HV_KVP, | |
85 | HV_KVP_GUID, | |
86 | .perf_device = false, | |
87 | }, | |
88 | ||
89 | /* Time Synch */ | |
90 | { .dev_type = HV_TS, | |
91 | HV_TS_GUID, | |
92 | .perf_device = false, | |
93 | }, | |
94 | ||
95 | /* Heartbeat */ | |
96 | { .dev_type = HV_HB, | |
97 | HV_HEART_BEAT_GUID, | |
98 | .perf_device = false, | |
99 | }, | |
100 | ||
101 | /* Shutdown */ | |
102 | { .dev_type = HV_SHUTDOWN, | |
103 | HV_SHUTDOWN_GUID, | |
104 | .perf_device = false, | |
105 | }, | |
106 | ||
107 | /* File copy */ | |
108 | { .dev_type = HV_FCOPY, | |
109 | HV_FCOPY_GUID, | |
110 | .perf_device = false, | |
111 | }, | |
112 | ||
113 | /* Backup */ | |
114 | { .dev_type = HV_BACKUP, | |
115 | HV_VSS_GUID, | |
116 | .perf_device = false, | |
117 | }, | |
118 | ||
119 | /* Dynamic Memory */ | |
120 | { .dev_type = HV_DM, | |
121 | HV_DM_GUID, | |
122 | .perf_device = false, | |
123 | }, | |
124 | ||
125 | /* Unknown GUID */ | |
f45be72c | 126 | { .dev_type = HV_UNKNOWN, |
7047f17d S |
127 | .perf_device = false, |
128 | }, | |
129 | }; | |
130 | ||
0f98829a | 131 | static const struct { |
593db803 | 132 | guid_t guid; |
0f98829a DC |
133 | } vmbus_unsupported_devs[] = { |
134 | { HV_AVMA1_GUID }, | |
135 | { HV_AVMA2_GUID }, | |
136 | { HV_RDV_GUID }, | |
137 | }; | |
138 | ||
ccb61f8a S |
139 | /* |
140 | * The rescinded channel may be blocked waiting for a response from the host; | |
141 | * take care of that. | |
142 | */ | |
143 | static void vmbus_rescind_cleanup(struct vmbus_channel *channel) | |
144 | { | |
145 | struct vmbus_channel_msginfo *msginfo; | |
146 | unsigned long flags; | |
147 | ||
148 | ||
149 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); | |
192b2d78 | 150 | channel->rescind = true; |
ccb61f8a S |
151 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
152 | msglistentry) { | |
153 | ||
154 | if (msginfo->waiting_channel == channel) { | |
155 | complete(&msginfo->waitevent); | |
156 | break; | |
157 | } | |
158 | } | |
159 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); | |
160 | } | |
161 | ||
593db803 | 162 | static bool is_unsupported_vmbus_devs(const guid_t *guid) |
0f98829a DC |
163 | { |
164 | int i; | |
165 | ||
166 | for (i = 0; i < ARRAY_SIZE(vmbus_unsupported_devs); i++) | |
593db803 | 167 | if (guid_equal(guid, &vmbus_unsupported_devs[i].guid)) |
0f98829a DC |
168 | return true; |
169 | return false; | |
170 | } | |
171 | ||
172 | static u16 hv_get_dev_type(const struct vmbus_channel *channel) | |
7047f17d | 173 | { |
593db803 | 174 | const guid_t *guid = &channel->offermsg.offer.if_type; |
7047f17d S |
175 | u16 i; |
176 | ||
0f98829a | 177 | if (is_hvsock_channel(channel) || is_unsupported_vmbus_devs(guid)) |
f45be72c | 178 | return HV_UNKNOWN; |
0f98829a | 179 | |
f45be72c | 180 | for (i = HV_IDE; i < HV_UNKNOWN; i++) { |
593db803 | 181 | if (guid_equal(guid, &vmbus_devs[i].guid)) |
7047f17d S |
182 | return i; |
183 | } | |
184 | pr_info("Unknown GUID: %pUl\n", guid); | |
185 | return i; | |
186 | } | |
f38e7dd7 | 187 | |
c88c4e4c | 188 | /** |
fe857bb4 | 189 | * vmbus_prep_negotiate_resp() - Create default response for Negotiate message |
c88c4e4c | 190 | * @icmsghdrp: Pointer to msg header structure |
c88c4e4c | 191 | * @buf: Raw buffer channel data |
fe857bb4 DC |
192 | * @fw_version: The framework versions we can support. |
193 | * @fw_vercnt: The size of @fw_version. | |
194 | * @srv_version: The service versions we can support. | |
195 | * @srv_vercnt: The size of @srv_version. | |
196 | * @nego_fw_version: The selected framework version. | |
197 | * @nego_srv_version: The selected service version. | |
c88c4e4c | 198 | * |
fe857bb4 | 199 | * Note: Versions are given in decreasing order. |
c88c4e4c | 200 | * |
fe857bb4 | 201 | * Set up and fill in default negotiate response message. |
c88c4e4c HJ |
202 | * Mainly used by Hyper-V drivers. |
203 | */ | |
6741335b | 204 | bool vmbus_prep_negotiate_resp(struct icmsg_hdr *icmsghdrp, |
a1656454 AN |
205 | u8 *buf, const int *fw_version, int fw_vercnt, |
206 | const int *srv_version, int srv_vercnt, | |
207 | int *nego_fw_version, int *nego_srv_version) | |
c88c4e4c | 208 | { |
6741335b S |
209 | int icframe_major, icframe_minor; |
210 | int icmsg_major, icmsg_minor; | |
211 | int fw_major, fw_minor; | |
212 | int srv_major, srv_minor; | |
a1656454 | 213 | int i, j; |
6741335b | 214 | bool found_match = false; |
a1656454 | 215 | struct icmsg_negotiate *negop; |
c836d0ab | 216 | |
a3605300 | 217 | icmsghdrp->icmsgsize = 0x10; |
a3605300 S |
218 | negop = (struct icmsg_negotiate *)&buf[ |
219 | sizeof(struct vmbuspipe_hdr) + | |
220 | sizeof(struct icmsg_hdr)]; | |
c88c4e4c | 221 | |
6741335b S |
222 | icframe_major = negop->icframe_vercnt; |
223 | icframe_minor = 0; | |
224 | ||
225 | icmsg_major = negop->icmsg_vercnt; | |
226 | icmsg_minor = 0; | |
c836d0ab S |
227 | |
228 | /* | |
229 | * Select the framework version number we will | |
230 | * support. | |
231 | */ | |
232 | ||
a1656454 AN |
233 | for (i = 0; i < fw_vercnt; i++) { |
234 | fw_major = (fw_version[i] >> 16); | |
235 | fw_minor = (fw_version[i] & 0xFFFF); | |
236 | ||
237 | for (j = 0; j < negop->icframe_vercnt; j++) { | |
238 | if ((negop->icversion_data[j].major == fw_major) && | |
239 | (negop->icversion_data[j].minor == fw_minor)) { | |
240 | icframe_major = negop->icversion_data[j].major; | |
241 | icframe_minor = negop->icversion_data[j].minor; | |
242 | found_match = true; | |
243 | break; | |
244 | } | |
6741335b | 245 | } |
a1656454 AN |
246 | |
247 | if (found_match) | |
248 | break; | |
c836d0ab S |
249 | } |
250 | ||
6741335b S |
251 | if (!found_match) |
252 | goto fw_error; | |
253 | ||
254 | found_match = false; | |
255 | ||
a1656454 AN |
256 | for (i = 0; i < srv_vercnt; i++) { |
257 | srv_major = (srv_version[i] >> 16); | |
258 | srv_minor = (srv_version[i] & 0xFFFF); | |
259 | ||
260 | for (j = negop->icframe_vercnt; | |
261 | (j < negop->icframe_vercnt + negop->icmsg_vercnt); | |
262 | j++) { | |
263 | ||
264 | if ((negop->icversion_data[j].major == srv_major) && | |
265 | (negop->icversion_data[j].minor == srv_minor)) { | |
266 | ||
267 | icmsg_major = negop->icversion_data[j].major; | |
268 | icmsg_minor = negop->icversion_data[j].minor; | |
269 | found_match = true; | |
270 | break; | |
271 | } | |
6741335b | 272 | } |
a1656454 AN |
273 | |
274 | if (found_match) | |
275 | break; | |
c88c4e4c | 276 | } |
a3605300 | 277 | |
c836d0ab | 278 | /* |
6741335b | 279 | * Respond with the framework and service |
c836d0ab S |
280 | * version numbers we can support. |
281 | */ | |
6741335b S |
282 | |
283 | fw_error: | |
284 | if (!found_match) { | |
285 | negop->icframe_vercnt = 0; | |
286 | negop->icmsg_vercnt = 0; | |
287 | } else { | |
288 | negop->icframe_vercnt = 1; | |
289 | negop->icmsg_vercnt = 1; | |
290 | } | |
291 | ||
a1656454 AN |
292 | if (nego_fw_version) |
293 | *nego_fw_version = (icframe_major << 16) | icframe_minor; | |
294 | ||
295 | if (nego_srv_version) | |
296 | *nego_srv_version = (icmsg_major << 16) | icmsg_minor; | |
297 | ||
6741335b S |
298 | negop->icversion_data[0].major = icframe_major; |
299 | negop->icversion_data[0].minor = icframe_minor; | |
300 | negop->icversion_data[1].major = icmsg_major; | |
301 | negop->icversion_data[1].minor = icmsg_minor; | |
302 | return found_match; | |
c88c4e4c | 303 | } |
a3605300 | 304 | |
da0e9631 | 305 | EXPORT_SYMBOL_GPL(vmbus_prep_negotiate_resp); |
c88c4e4c | 306 | |
3e189519 | 307 | /* |
e98cb276 | 308 | * alloc_channel - Allocate and initialize a vmbus channel object |
bd60c33e | 309 | */ |
50fe56d2 | 310 | static struct vmbus_channel *alloc_channel(void) |
3e7ee490 | 311 | { |
aded7165 | 312 | struct vmbus_channel *channel; |
3e7ee490 | 313 | |
aded7165 | 314 | channel = kzalloc(sizeof(*channel), GFP_ATOMIC); |
3e7ee490 | 315 | if (!channel) |
3e7ee490 | 316 | return NULL; |
3e7ee490 | 317 | |
67fae053 | 318 | spin_lock_init(&channel->lock); |
7fa32e5e | 319 | init_completion(&channel->rescind_event); |
e68d2971 S |
320 | |
321 | INIT_LIST_HEAD(&channel->sc_list); | |
3a28fa35 | 322 | INIT_LIST_HEAD(&channel->percpu_list); |
3e7ee490 | 323 | |
631e63a9 SH |
324 | tasklet_init(&channel->callback_event, |
325 | vmbus_on_event, (unsigned long)channel); | |
326 | ||
14948e39 KB |
327 | hv_ringbuffer_pre_init(channel); |
328 | ||
3e7ee490 HJ |
329 | return channel; |
330 | } | |
331 | ||
3e189519 | 332 | /* |
e98cb276 | 333 | * free_channel - Release the resources used by the vmbus channel object |
bd60c33e | 334 | */ |
9f3e28e3 | 335 | static void free_channel(struct vmbus_channel *channel) |
3e7ee490 | 336 | { |
631e63a9 | 337 | tasklet_kill(&channel->callback_event); |
46fc1548 | 338 | vmbus_remove_channel_attr_group(channel); |
8200f208 | 339 | |
c2e5df61 | 340 | kobject_put(&channel->kobj); |
3e7ee490 HJ |
341 | } |
342 | ||
3a28fa35 S |
343 | static void percpu_channel_enq(void *arg) |
344 | { | |
345 | struct vmbus_channel *channel = arg; | |
37cdd991 SH |
346 | struct hv_per_cpu_context *hv_cpu |
347 | = this_cpu_ptr(hv_context.cpu_context); | |
3a28fa35 | 348 | |
8200f208 | 349 | list_add_tail_rcu(&channel->percpu_list, &hv_cpu->chan_list); |
3a28fa35 | 350 | } |
8b5d6d3b | 351 | |
3a28fa35 S |
352 | static void percpu_channel_deq(void *arg) |
353 | { | |
354 | struct vmbus_channel *channel = arg; | |
355 | ||
8200f208 | 356 | list_del_rcu(&channel->percpu_list); |
3a28fa35 | 357 | } |
8b5d6d3b | 358 | |
ed6cfcc5 | 359 | |
f52078cf | 360 | static void vmbus_release_relid(u32 relid) |
4b2f9abe | 361 | { |
ed6cfcc5 | 362 | struct vmbus_channel_relid_released msg; |
259cc665 | 363 | int ret; |
4b2f9abe | 364 | |
c8705979 | 365 | memset(&msg, 0, sizeof(struct vmbus_channel_relid_released)); |
ed6cfcc5 | 366 | msg.child_relid = relid; |
c8705979 | 367 | msg.header.msgtype = CHANNELMSG_RELID_RELEASED; |
259cc665 VK |
368 | ret = vmbus_post_msg(&msg, sizeof(struct vmbus_channel_relid_released), |
369 | true); | |
370 | ||
371 | trace_vmbus_release_relid(&msg, ret); | |
f52078cf | 372 | } |
c8705979 | 373 | |
800b9329 | 374 | void hv_process_channel_removal(struct vmbus_channel *channel) |
f52078cf | 375 | { |
800b9329 | 376 | struct vmbus_channel *primary_channel; |
f52078cf | 377 | unsigned long flags; |
f52078cf | 378 | |
85d9aa70 | 379 | BUG_ON(!mutex_is_locked(&vmbus_connection.channel_mutex)); |
192b2d78 | 380 | BUG_ON(!channel->rescind); |
800b9329 | 381 | |
2115b561 S |
382 | if (channel->target_cpu != get_cpu()) { |
383 | put_cpu(); | |
3a28fa35 S |
384 | smp_call_function_single(channel->target_cpu, |
385 | percpu_channel_deq, channel, true); | |
2115b561 | 386 | } else { |
3a28fa35 | 387 | percpu_channel_deq(channel); |
2115b561 S |
388 | put_cpu(); |
389 | } | |
3a28fa35 | 390 | |
e68d2971 | 391 | if (channel->primary_channel == NULL) { |
e68d2971 | 392 | list_del(&channel->listentry); |
ca1c4b74 DC |
393 | |
394 | primary_channel = channel; | |
e68d2971 S |
395 | } else { |
396 | primary_channel = channel->primary_channel; | |
67fae053 | 397 | spin_lock_irqsave(&primary_channel->lock, flags); |
565ce642 | 398 | list_del(&channel->sc_list); |
67fae053 | 399 | spin_unlock_irqrestore(&primary_channel->lock, flags); |
e68d2971 | 400 | } |
ca1c4b74 DC |
401 | |
402 | /* | |
403 | * We need to free the bit for init_vp_index() to work in the case | |
404 | * of sub-channel, when we reload drivers like hv_netvsc. | |
405 | */ | |
509879bd S |
406 | if (channel->affinity_policy == HV_LOCALIZED) |
407 | cpumask_clear_cpu(channel->target_cpu, | |
408 | &primary_channel->alloced_cpus_in_node); | |
ca1c4b74 | 409 | |
d8bd2d44 DC |
410 | /* |
411 | * Upon suspend, an in-use hv_sock channel is marked as "rescinded" and | |
412 | * the relid is invalidated; after hibernation, when the user-space app | |
413 | * destroys the channel, the relid is INVALID_RELID, and in this case | |
414 | * it's unnecessary and unsafe to release the old relid, since the same | |
415 | * relid can refer to a completely different channel now. | |
416 | */ | |
417 | if (channel->offermsg.child_relid != INVALID_RELID) | |
418 | vmbus_release_relid(channel->offermsg.child_relid); | |
638fea33 | 419 | |
c8705979 | 420 | free_channel(channel); |
4b2f9abe | 421 | } |
8b5d6d3b | 422 | |
93e5bd06 S |
423 | void vmbus_free_channels(void) |
424 | { | |
813c5b79 DC |
425 | struct vmbus_channel *channel, *tmp; |
426 | ||
427 | list_for_each_entry_safe(channel, tmp, &vmbus_connection.chn_list, | |
428 | listentry) { | |
34c6801e | 429 | /* hv_process_channel_removal() needs this */ |
813c5b79 | 430 | channel->rescind = true; |
93e5bd06 | 431 | |
93e5bd06 | 432 | vmbus_device_unregister(channel->device_obj); |
93e5bd06 S |
433 | } |
434 | } | |
435 | ||
37c2578c DC |
436 | /* Note: the function can run concurrently for primary/sub channels. */ |
437 | static void vmbus_add_channel_work(struct work_struct *work) | |
3e7ee490 | 438 | { |
37c2578c DC |
439 | struct vmbus_channel *newchannel = |
440 | container_of(work, struct vmbus_channel, add_channel_work); | |
441 | struct vmbus_channel *primary_channel = newchannel->primary_channel; | |
0f5e44ca | 442 | unsigned long flags; |
7047f17d | 443 | u16 dev_type; |
85d9aa70 | 444 | int ret; |
3e7ee490 | 445 | |
0f98829a | 446 | dev_type = hv_get_dev_type(newchannel); |
7047f17d S |
447 | |
448 | init_vp_index(newchannel, dev_type); | |
f38e7dd7 | 449 | |
8dfd3326 VK |
450 | if (newchannel->target_cpu != get_cpu()) { |
451 | put_cpu(); | |
452 | smp_call_function_single(newchannel->target_cpu, | |
453 | percpu_channel_enq, | |
454 | newchannel, true); | |
455 | } else { | |
456 | percpu_channel_enq(newchannel); | |
457 | put_cpu(); | |
3e7ee490 HJ |
458 | } |
459 | ||
42dceebe S |
460 | /* |
461 | * This state is used to indicate a successful open | |
462 | * so that when we do close the channel normally, we | |
37c2578c | 463 | * can cleanup properly. |
42dceebe S |
464 | */ |
465 | newchannel->state = CHANNEL_OPEN_STATE; | |
466 | ||
37c2578c DC |
467 | if (primary_channel != NULL) { |
468 | /* newchannel is a sub-channel. */ | |
469 | struct hv_device *dev = primary_channel->device_obj; | |
c2e5df61 | 470 | |
50229128 | 471 | if (vmbus_add_channel_kobj(dev, newchannel)) |
37c2578c DC |
472 | goto err_deq_chan; |
473 | ||
474 | if (primary_channel->sc_creation_callback != NULL) | |
475 | primary_channel->sc_creation_callback(newchannel); | |
c2e5df61 | 476 | |
192b2d78 | 477 | newchannel->probe_done = true; |
8dfd3326 VK |
478 | return; |
479 | } | |
480 | ||
bd60c33e | 481 | /* |
37c2578c | 482 | * Start the process of binding the primary channel to the driver |
bd60c33e | 483 | */ |
f2c73011 | 484 | newchannel->device_obj = vmbus_device_create( |
767dff68 HZ |
485 | &newchannel->offermsg.offer.if_type, |
486 | &newchannel->offermsg.offer.if_instance, | |
188963ec | 487 | newchannel); |
9c3a6f7e | 488 | if (!newchannel->device_obj) |
5b1e5b53 | 489 | goto err_deq_chan; |
3e7ee490 | 490 | |
7047f17d | 491 | newchannel->device_obj->device_id = dev_type; |
454f18a9 BP |
492 | /* |
493 | * Add the new device to the bus. This will kick off device-driver | |
494 | * binding which eventually invokes the device driver's AddDevice() | |
495 | * method. | |
496 | */ | |
85d9aa70 | 497 | ret = vmbus_device_register(newchannel->device_obj); |
85d9aa70 DC |
498 | |
499 | if (ret != 0) { | |
d43e2fe7 DC |
500 | pr_err("unable to add child device object (relid %d)\n", |
501 | newchannel->offermsg.child_relid); | |
502 | kfree(newchannel->device_obj); | |
503 | goto err_deq_chan; | |
504 | } | |
54a66265 | 505 | |
6f3d791f | 506 | newchannel->probe_done = true; |
9c3a6f7e | 507 | return; |
2dd37cb8 | 508 | |
5b1e5b53 | 509 | err_deq_chan: |
d6f591e3 | 510 | mutex_lock(&vmbus_connection.channel_mutex); |
37c2578c DC |
511 | |
512 | /* | |
513 | * We need to set the flag, otherwise | |
514 | * vmbus_onoffer_rescind() can be blocked. | |
515 | */ | |
516 | newchannel->probe_done = true; | |
517 | ||
518 | if (primary_channel == NULL) { | |
519 | list_del(&newchannel->listentry); | |
520 | } else { | |
521 | spin_lock_irqsave(&primary_channel->lock, flags); | |
522 | list_del(&newchannel->sc_list); | |
523 | spin_unlock_irqrestore(&primary_channel->lock, flags); | |
524 | } | |
525 | ||
d6f591e3 | 526 | mutex_unlock(&vmbus_connection.channel_mutex); |
5b1e5b53 S |
527 | |
528 | if (newchannel->target_cpu != get_cpu()) { | |
529 | put_cpu(); | |
530 | smp_call_function_single(newchannel->target_cpu, | |
37c2578c DC |
531 | percpu_channel_deq, |
532 | newchannel, true); | |
5b1e5b53 S |
533 | } else { |
534 | percpu_channel_deq(newchannel); | |
535 | put_cpu(); | |
536 | } | |
638fea33 DC |
537 | |
538 | vmbus_release_relid(newchannel->offermsg.child_relid); | |
5b1e5b53 | 539 | |
9c3a6f7e | 540 | free_channel(newchannel); |
3e7ee490 HJ |
541 | } |
542 | ||
37c2578c DC |
543 | /* |
544 | * vmbus_process_offer - Process the offer by creating a channel/device | |
545 | * associated with this offer | |
546 | */ | |
547 | static void vmbus_process_offer(struct vmbus_channel *newchannel) | |
548 | { | |
549 | struct vmbus_channel *channel; | |
550 | struct workqueue_struct *wq; | |
551 | unsigned long flags; | |
552 | bool fnew = true; | |
553 | ||
554 | mutex_lock(&vmbus_connection.channel_mutex); | |
555 | ||
b307b389 DC |
556 | /* Remember the channels that should be cleaned up upon suspend. */ |
557 | if (is_hvsock_channel(newchannel) || is_sub_channel(newchannel)) | |
558 | atomic_inc(&vmbus_connection.nr_chan_close_on_suspend); | |
559 | ||
37c2578c DC |
560 | /* |
561 | * Now that we have acquired the channel_mutex, | |
562 | * we can release the potentially racing rescind thread. | |
563 | */ | |
564 | atomic_dec(&vmbus_connection.offer_in_progress); | |
565 | ||
566 | list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) { | |
593db803 AS |
567 | if (guid_equal(&channel->offermsg.offer.if_type, |
568 | &newchannel->offermsg.offer.if_type) && | |
569 | guid_equal(&channel->offermsg.offer.if_instance, | |
570 | &newchannel->offermsg.offer.if_instance)) { | |
37c2578c DC |
571 | fnew = false; |
572 | break; | |
573 | } | |
574 | } | |
575 | ||
576 | if (fnew) | |
577 | list_add_tail(&newchannel->listentry, | |
578 | &vmbus_connection.chn_list); | |
579 | else { | |
580 | /* | |
581 | * Check to see if this is a valid sub-channel. | |
582 | */ | |
583 | if (newchannel->offermsg.offer.sub_channel_index == 0) { | |
584 | mutex_unlock(&vmbus_connection.channel_mutex); | |
585 | /* | |
586 | * Don't call free_channel(), because newchannel->kobj | |
587 | * is not initialized yet. | |
588 | */ | |
589 | kfree(newchannel); | |
590 | WARN_ON_ONCE(1); | |
591 | return; | |
592 | } | |
593 | /* | |
594 | * Process the sub-channel. | |
595 | */ | |
596 | newchannel->primary_channel = channel; | |
597 | spin_lock_irqsave(&channel->lock, flags); | |
598 | list_add_tail(&newchannel->sc_list, &channel->sc_list); | |
599 | spin_unlock_irqrestore(&channel->lock, flags); | |
600 | } | |
601 | ||
602 | mutex_unlock(&vmbus_connection.channel_mutex); | |
603 | ||
604 | /* | |
605 | * vmbus_process_offer() mustn't call channel->sc_creation_callback() | |
606 | * directly for sub-channels, because sc_creation_callback() -> | |
607 | * vmbus_open() may never get the host's response to the | |
608 | * OPEN_CHANNEL message (the host may rescind a channel at any time, | |
609 | * e.g. in the case of hot removing a NIC), and vmbus_onoffer_rescind() | |
610 | * may not wake up the vmbus_open() as it's blocked due to a non-zero | |
611 | * vmbus_connection.offer_in_progress, and finally we have a deadlock. | |
612 | * | |
613 | * The above is also true for primary channels, if the related device | |
614 | * drivers use sync probing mode by default. | |
615 | * | |
616 | * And, usually the handling of primary channels and sub-channels can | |
617 | * depend on each other, so we should offload them to different | |
618 | * workqueues to avoid possible deadlock, e.g. in sync-probing mode, | |
619 | * NIC1's netvsc_subchan_work() can race with NIC2's netvsc_probe() -> | |
620 | * rtnl_lock(), and causes deadlock: the former gets the rtnl_lock | |
621 | * and waits for all the sub-channels to appear, but the latter | |
622 | * can't get the rtnl_lock and this blocks the handling of | |
623 | * sub-channels. | |
624 | */ | |
625 | INIT_WORK(&newchannel->add_channel_work, vmbus_add_channel_work); | |
626 | wq = fnew ? vmbus_connection.handle_primary_chan_wq : | |
627 | vmbus_connection.handle_sub_chan_wq; | |
628 | queue_work(wq, &newchannel->add_channel_work); | |
629 | } | |
630 | ||
a119845f S |
631 | /* |
632 | * We use this state to statically distribute the channel interrupt load. | |
633 | */ | |
1f656ff3 | 634 | static int next_numa_node_id; |
37c2578c DC |
635 | /* |
636 | * init_vp_index() accesses global variables like next_numa_node_id, and | |
637 | * it can run concurrently for primary channels and sub-channels: see | |
638 | * vmbus_process_offer(), so we need the lock to protect the global | |
639 | * variables. | |
640 | */ | |
641 | static DEFINE_SPINLOCK(bind_channel_to_cpu_lock); | |
a119845f S |
642 | |
643 | /* | |
644 | * Starting with Win8, we can statically distribute the incoming | |
1f656ff3 | 645 | * channel interrupt load by binding a channel to VCPU. |
ce767047 HZ |
646 | * We distribute the interrupt loads to one or more NUMA nodes based on |
647 | * the channel's affinity_policy. | |
1f656ff3 S |
648 | * |
649 | * For pre-win8 hosts or non-performance critical channels we assign the | |
650 | * first CPU in the first NUMA node. | |
a119845f | 651 | */ |
7047f17d | 652 | static void init_vp_index(struct vmbus_channel *channel, u16 dev_type) |
a119845f S |
653 | { |
654 | u32 cur_cpu; | |
7047f17d | 655 | bool perf_chn = vmbus_devs[dev_type].perf_device; |
1f656ff3 S |
656 | struct vmbus_channel *primary = channel->primary_channel; |
657 | int next_node; | |
25355252 | 658 | cpumask_var_t available_mask; |
9f01ec53 | 659 | struct cpumask *alloced_mask; |
a119845f | 660 | |
a119845f | 661 | if ((vmbus_proto_version == VERSION_WS2008) || |
25355252 DC |
662 | (vmbus_proto_version == VERSION_WIN7) || (!perf_chn) || |
663 | !alloc_cpumask_var(&available_mask, GFP_KERNEL)) { | |
a119845f S |
664 | /* |
665 | * Prior to win8, all channel interrupts are | |
666 | * delivered on cpu 0. | |
667 | * Also if the channel is not a performance critical | |
668 | * channel, bind it to cpu 0. | |
25355252 | 669 | * In case alloc_cpumask_var() fails, bind it to cpu 0. |
a119845f | 670 | */ |
1f656ff3 | 671 | channel->numa_node = 0; |
d3ba720d | 672 | channel->target_cpu = 0; |
7415aea6 | 673 | channel->target_vp = hv_cpu_number_to_vp_number(0); |
d3ba720d | 674 | return; |
a119845f | 675 | } |
ce59fec8 | 676 | |
37c2578c DC |
677 | spin_lock(&bind_channel_to_cpu_lock); |
678 | ||
ce59fec8 | 679 | /* |
509879bd S |
680 | * Based on the channel affinity policy, we will assign the NUMA |
681 | * nodes. | |
ce59fec8 | 682 | */ |
509879bd S |
683 | |
684 | if ((channel->affinity_policy == HV_BALANCED) || (!primary)) { | |
1f656ff3 S |
685 | while (true) { |
686 | next_node = next_numa_node_id++; | |
509879bd | 687 | if (next_node == nr_node_ids) { |
1f656ff3 | 688 | next_node = next_numa_node_id = 0; |
509879bd S |
689 | continue; |
690 | } | |
1f656ff3 S |
691 | if (cpumask_empty(cpumask_of_node(next_node))) |
692 | continue; | |
693 | break; | |
694 | } | |
695 | channel->numa_node = next_node; | |
696 | primary = channel; | |
697 | } | |
9f01ec53 | 698 | alloced_mask = &hv_context.hv_numa_map[primary->numa_node]; |
1f656ff3 | 699 | |
9f01ec53 | 700 | if (cpumask_weight(alloced_mask) == |
1f656ff3 | 701 | cpumask_weight(cpumask_of_node(primary->numa_node))) { |
ce59fec8 | 702 | /* |
1f656ff3 S |
703 | * We have cycled through all the CPUs in the node; |
704 | * reset the alloced map. | |
ce59fec8 | 705 | */ |
9f01ec53 | 706 | cpumask_clear(alloced_mask); |
ce59fec8 VK |
707 | } |
708 | ||
25355252 | 709 | cpumask_xor(available_mask, alloced_mask, |
1f656ff3 S |
710 | cpumask_of_node(primary->numa_node)); |
711 | ||
3b71107d | 712 | cur_cpu = -1; |
79fd8e70 | 713 | |
509879bd S |
714 | if (primary->affinity_policy == HV_LOCALIZED) { |
715 | /* | |
716 | * Normally Hyper-V host doesn't create more subchannels | |
717 | * than there are VCPUs on the node but it is possible when not | |
718 | * all present VCPUs on the node are initialized by guest. | |
719 | * Clear the alloced_cpus_in_node to start over. | |
720 | */ | |
721 | if (cpumask_equal(&primary->alloced_cpus_in_node, | |
722 | cpumask_of_node(primary->numa_node))) | |
723 | cpumask_clear(&primary->alloced_cpus_in_node); | |
724 | } | |
79fd8e70 | 725 | |
3b71107d | 726 | while (true) { |
25355252 | 727 | cur_cpu = cpumask_next(cur_cpu, available_mask); |
3b71107d DC |
728 | if (cur_cpu >= nr_cpu_ids) { |
729 | cur_cpu = -1; | |
25355252 | 730 | cpumask_copy(available_mask, |
3b71107d DC |
731 | cpumask_of_node(primary->numa_node)); |
732 | continue; | |
733 | } | |
734 | ||
509879bd S |
735 | if (primary->affinity_policy == HV_LOCALIZED) { |
736 | /* | |
737 | * NOTE: in the case of sub-channel, we clear the | |
738 | * sub-channel related bit(s) in | |
739 | * primary->alloced_cpus_in_node in | |
740 | * hv_process_channel_removal(), so when we | |
741 | * reload drivers like hv_netvsc in SMP guest, here | |
742 | * we're able to re-allocate | |
743 | * bit from primary->alloced_cpus_in_node. | |
744 | */ | |
745 | if (!cpumask_test_cpu(cur_cpu, | |
746 | &primary->alloced_cpus_in_node)) { | |
747 | cpumask_set_cpu(cur_cpu, | |
748 | &primary->alloced_cpus_in_node); | |
749 | cpumask_set_cpu(cur_cpu, alloced_mask); | |
750 | break; | |
751 | } | |
752 | } else { | |
3b71107d DC |
753 | cpumask_set_cpu(cur_cpu, alloced_mask); |
754 | break; | |
755 | } | |
756 | } | |
1f656ff3 | 757 | |
d3ba720d | 758 | channel->target_cpu = cur_cpu; |
7415aea6 | 759 | channel->target_vp = hv_cpu_number_to_vp_number(cur_cpu); |
25355252 | 760 | |
37c2578c DC |
761 | spin_unlock(&bind_channel_to_cpu_lock); |
762 | ||
25355252 | 763 | free_cpumask_var(available_mask); |
a119845f S |
764 | } |
765 | ||
41571916 VK |
766 | static void vmbus_wait_for_unload(void) |
767 | { | |
cd95aad5 VK |
768 | int cpu; |
769 | void *page_addr; | |
770 | struct hv_message *msg; | |
41571916 | 771 | struct vmbus_channel_message_header *hdr; |
cd95aad5 | 772 | u32 message_type; |
41571916 | 773 | |
cd95aad5 VK |
774 | /* |
775 | * CHANNELMSG_UNLOAD_RESPONSE is always delivered to the CPU which was | |
776 | * used for initial contact or to CPU0 depending on host version. When | |
777 | * we're crashing on a different CPU let's hope that IRQ handler on | |
778 | * the cpu which receives CHANNELMSG_UNLOAD_RESPONSE is still | |
779 | * functional and vmbus_unload_response() will complete | |
780 | * vmbus_connection.unload_event. If not, the last thing we can do is | |
781 | * read message pages for all CPUs directly. | |
782 | */ | |
41571916 | 783 | while (1) { |
cd95aad5 VK |
784 | if (completion_done(&vmbus_connection.unload_event)) |
785 | break; | |
41571916 | 786 | |
cd95aad5 | 787 | for_each_online_cpu(cpu) { |
37cdd991 SH |
788 | struct hv_per_cpu_context *hv_cpu |
789 | = per_cpu_ptr(hv_context.cpu_context, cpu); | |
790 | ||
791 | page_addr = hv_cpu->synic_message_page; | |
792 | msg = (struct hv_message *)page_addr | |
793 | + VMBUS_MESSAGE_SINT; | |
41571916 | 794 | |
cd95aad5 VK |
795 | message_type = READ_ONCE(msg->header.message_type); |
796 | if (message_type == HVMSG_NONE) | |
797 | continue; | |
41571916 | 798 | |
cd95aad5 VK |
799 | hdr = (struct vmbus_channel_message_header *) |
800 | msg->u.payload; | |
801 | ||
802 | if (hdr->msgtype == CHANNELMSG_UNLOAD_RESPONSE) | |
803 | complete(&vmbus_connection.unload_event); | |
804 | ||
805 | vmbus_signal_eom(msg, message_type); | |
806 | } | |
807 | ||
808 | mdelay(10); | |
809 | } | |
810 | ||
811 | /* | |
812 | * We're crashing and already got the UNLOAD_RESPONSE, cleanup all | |
813 | * maybe-pending messages on all CPUs to be able to receive new | |
814 | * messages after we reconnect. | |
815 | */ | |
816 | for_each_online_cpu(cpu) { | |
37cdd991 SH |
817 | struct hv_per_cpu_context *hv_cpu |
818 | = per_cpu_ptr(hv_context.cpu_context, cpu); | |
819 | ||
820 | page_addr = hv_cpu->synic_message_page; | |
cd95aad5 VK |
821 | msg = (struct hv_message *)page_addr + VMBUS_MESSAGE_SINT; |
822 | msg->header.message_type = HVMSG_NONE; | |
41571916 VK |
823 | } |
824 | } | |
825 | ||
2db84eff S |
826 | /* |
827 | * vmbus_unload_response - Handler for the unload response. | |
828 | */ | |
829 | static void vmbus_unload_response(struct vmbus_channel_message_header *hdr) | |
830 | { | |
831 | /* | |
832 | * This is a global event; just wakeup the waiting thread. | |
833 | * Once we successfully unload, we can cleanup the monitor state. | |
834 | */ | |
835 | complete(&vmbus_connection.unload_event); | |
836 | } | |
837 | ||
75ff3a8a | 838 | void vmbus_initiate_unload(bool crash) |
2db84eff S |
839 | { |
840 | struct vmbus_channel_message_header hdr; | |
841 | ||
4a54243f VK |
842 | /* Pre-Win2012R2 hosts don't support reconnect */ |
843 | if (vmbus_proto_version < VERSION_WIN8_1) | |
844 | return; | |
845 | ||
2db84eff S |
846 | init_completion(&vmbus_connection.unload_event); |
847 | memset(&hdr, 0, sizeof(struct vmbus_channel_message_header)); | |
848 | hdr.msgtype = CHANNELMSG_UNLOAD; | |
c0bb0392 VK |
849 | vmbus_post_msg(&hdr, sizeof(struct vmbus_channel_message_header), |
850 | !crash); | |
2db84eff | 851 | |
41571916 VK |
852 | /* |
853 | * vmbus_initiate_unload() is also called on crash and the crash can be | |
854 | * happening in an interrupt context, where scheduling is impossible. | |
855 | */ | |
75ff3a8a | 856 | if (!crash) |
41571916 VK |
857 | wait_for_completion(&vmbus_connection.unload_event); |
858 | else | |
859 | vmbus_wait_for_unload(); | |
2db84eff S |
860 | } |
861 | ||
d8bd2d44 DC |
862 | static void check_ready_for_resume_event(void) |
863 | { | |
864 | /* | |
865 | * If all the old primary channels have been fixed up, then it's safe | |
866 | * to resume. | |
867 | */ | |
868 | if (atomic_dec_and_test(&vmbus_connection.nr_chan_fixup_on_resume)) | |
869 | complete(&vmbus_connection.ready_for_resume_event); | |
870 | } | |
871 | ||
872 | static void vmbus_setup_channel_state(struct vmbus_channel *channel, | |
873 | struct vmbus_channel_offer_channel *offer) | |
874 | { | |
875 | /* | |
876 | * Setup state for signalling the host. | |
877 | */ | |
878 | channel->sig_event = VMBUS_EVENT_CONNECTION_ID; | |
879 | ||
880 | if (vmbus_proto_version != VERSION_WS2008) { | |
881 | channel->is_dedicated_interrupt = | |
882 | (offer->is_dedicated_interrupt != 0); | |
883 | channel->sig_event = offer->connection_id; | |
884 | } | |
885 | ||
886 | memcpy(&channel->offermsg, offer, | |
887 | sizeof(struct vmbus_channel_offer_channel)); | |
888 | channel->monitor_grp = (u8)offer->monitorid / 32; | |
889 | channel->monitor_bit = (u8)offer->monitorid % 32; | |
890 | } | |
891 | ||
e3ede02a DC |
892 | /* |
893 | * find_primary_channel_by_offer - Get the channel object given the new offer. | |
894 | * This is only used in the resume path of hibernation. | |
895 | */ | |
896 | static struct vmbus_channel * | |
897 | find_primary_channel_by_offer(const struct vmbus_channel_offer_channel *offer) | |
898 | { | |
899 | struct vmbus_channel *channel = NULL, *iter; | |
900 | const guid_t *inst1, *inst2; | |
901 | ||
902 | /* Ignore sub-channel offers. */ | |
903 | if (offer->offer.sub_channel_index != 0) | |
904 | return NULL; | |
905 | ||
906 | mutex_lock(&vmbus_connection.channel_mutex); | |
907 | ||
908 | list_for_each_entry(iter, &vmbus_connection.chn_list, listentry) { | |
909 | inst1 = &iter->offermsg.offer.if_instance; | |
910 | inst2 = &offer->offer.if_instance; | |
911 | ||
912 | if (guid_equal(inst1, inst2)) { | |
913 | channel = iter; | |
914 | break; | |
915 | } | |
916 | } | |
917 | ||
918 | mutex_unlock(&vmbus_connection.channel_mutex); | |
919 | ||
920 | return channel; | |
921 | } | |
922 | ||
3e189519 | 923 | /* |
e98cb276 | 924 | * vmbus_onoffer - Handler for channel offers from vmbus in parent partition. |
bd60c33e | 925 | * |
bd60c33e | 926 | */ |
e98cb276 | 927 | static void vmbus_onoffer(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 928 | { |
bd60c33e | 929 | struct vmbus_channel_offer_channel *offer; |
e3ede02a DC |
930 | struct vmbus_channel *oldchannel, *newchannel; |
931 | size_t offer_sz; | |
3e7ee490 | 932 | |
bd60c33e | 933 | offer = (struct vmbus_channel_offer_channel *)hdr; |
3e7ee490 | 934 | |
03b81e66 VK |
935 | trace_vmbus_onoffer(offer); |
936 | ||
e3ede02a DC |
937 | oldchannel = find_primary_channel_by_offer(offer); |
938 | ||
939 | if (oldchannel != NULL) { | |
940 | atomic_dec(&vmbus_connection.offer_in_progress); | |
941 | ||
942 | /* | |
d8bd2d44 DC |
943 | * We're resuming from hibernation: all the sub-channel and |
944 | * hv_sock channels we had before the hibernation should have | |
945 | * been cleaned up, and now we must be seeing a re-offered | |
946 | * primary channel that we had before the hibernation. | |
e3ede02a | 947 | */ |
d8bd2d44 DC |
948 | |
949 | WARN_ON(oldchannel->offermsg.child_relid != INVALID_RELID); | |
950 | /* Fix up the relid. */ | |
951 | oldchannel->offermsg.child_relid = offer->child_relid; | |
952 | ||
e3ede02a | 953 | offer_sz = sizeof(*offer); |
d8bd2d44 DC |
954 | if (memcmp(offer, &oldchannel->offermsg, offer_sz) == 0) { |
955 | check_ready_for_resume_event(); | |
e3ede02a | 956 | return; |
d8bd2d44 | 957 | } |
e3ede02a | 958 | |
d8bd2d44 DC |
959 | /* |
960 | * This is not an error, since the host can also change the | |
961 | * other field(s) of the offer, e.g. on WS RS5 (Build 17763), | |
962 | * the offer->connection_id of the Mellanox VF vmbus device | |
963 | * can change when the host reoffers the device upon resume. | |
964 | */ | |
965 | pr_debug("vmbus offer changed: relid=%d\n", | |
e3ede02a DC |
966 | offer->child_relid); |
967 | ||
968 | print_hex_dump_debug("Old vmbus offer: ", DUMP_PREFIX_OFFSET, | |
969 | 16, 4, &oldchannel->offermsg, offer_sz, | |
970 | false); | |
971 | print_hex_dump_debug("New vmbus offer: ", DUMP_PREFIX_OFFSET, | |
972 | 16, 4, offer, offer_sz, false); | |
d8bd2d44 DC |
973 | |
974 | /* Fix up the old channel. */ | |
975 | vmbus_setup_channel_state(oldchannel, offer); | |
976 | ||
977 | check_ready_for_resume_event(); | |
978 | ||
e3ede02a DC |
979 | return; |
980 | } | |
981 | ||
454f18a9 | 982 | /* Allocate the channel object and save this offer. */ |
e98cb276 | 983 | newchannel = alloc_channel(); |
188963ec | 984 | if (!newchannel) { |
9a547602 | 985 | vmbus_release_relid(offer->child_relid); |
54a66265 | 986 | atomic_dec(&vmbus_connection.offer_in_progress); |
0a46618d | 987 | pr_err("Unable to allocate channel object\n"); |
3e7ee490 HJ |
988 | return; |
989 | } | |
990 | ||
d8bd2d44 | 991 | vmbus_setup_channel_state(newchannel, offer); |
3e7ee490 | 992 | |
2dd37cb8 | 993 | vmbus_process_offer(newchannel); |
3e7ee490 HJ |
994 | } |
995 | ||
b307b389 DC |
996 | static void check_ready_for_suspend_event(void) |
997 | { | |
998 | /* | |
999 | * If all the sub-channels or hv_sock channels have been cleaned up, | |
1000 | * then it's safe to suspend. | |
1001 | */ | |
1002 | if (atomic_dec_and_test(&vmbus_connection.nr_chan_close_on_suspend)) | |
1003 | complete(&vmbus_connection.ready_for_suspend_event); | |
1004 | } | |
1005 | ||
3e189519 | 1006 | /* |
e98cb276 | 1007 | * vmbus_onoffer_rescind - Rescind offer handler. |
bd60c33e GKH |
1008 | * |
1009 | * We queue a work item to process this offer synchronously | |
1010 | */ | |
e98cb276 | 1011 | static void vmbus_onoffer_rescind(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1012 | { |
bd60c33e | 1013 | struct vmbus_channel_rescind_offer *rescind; |
aded7165 | 1014 | struct vmbus_channel *channel; |
d43e2fe7 | 1015 | struct device *dev; |
b307b389 | 1016 | bool clean_up_chan_for_suspend; |
3e7ee490 | 1017 | |
bd60c33e | 1018 | rescind = (struct vmbus_channel_rescind_offer *)hdr; |
85d9aa70 | 1019 | |
5175167c VK |
1020 | trace_vmbus_onoffer_rescind(rescind); |
1021 | ||
54a66265 S |
1022 | /* |
1023 | * The offer msg and the corresponding rescind msg | |
1024 | * from the host are guranteed to be ordered - | |
1025 | * offer comes in first and then the rescind. | |
1026 | * Since we process these events in work elements, | |
1027 | * and with preemption, we may end up processing | |
1028 | * the events out of order. Given that we handle these | |
1029 | * work elements on the same CPU, this is possible only | |
1030 | * in the case of preemption. In any case wait here | |
1031 | * until the offer processing has moved beyond the | |
1032 | * point where the channel is discoverable. | |
1033 | */ | |
1034 | ||
1035 | while (atomic_read(&vmbus_connection.offer_in_progress) != 0) { | |
1036 | /* | |
1037 | * We wait here until any channel offer is currently | |
1038 | * being processed. | |
1039 | */ | |
1040 | msleep(1); | |
1041 | } | |
1042 | ||
85d9aa70 | 1043 | mutex_lock(&vmbus_connection.channel_mutex); |
d43e2fe7 | 1044 | channel = relid2channel(rescind->child_relid); |
54a66265 | 1045 | mutex_unlock(&vmbus_connection.channel_mutex); |
98e08702 | 1046 | |
2dd37cb8 | 1047 | if (channel == NULL) { |
f52078cf | 1048 | /* |
54a66265 S |
1049 | * We failed in processing the offer message; |
1050 | * we would have cleaned up the relid in that | |
1051 | * failure path. | |
f52078cf | 1052 | */ |
54a66265 | 1053 | return; |
2dd37cb8 | 1054 | } |
3e7ee490 | 1055 | |
b307b389 DC |
1056 | clean_up_chan_for_suspend = is_hvsock_channel(channel) || |
1057 | is_sub_channel(channel); | |
d3b26dd7 DC |
1058 | /* |
1059 | * Before setting channel->rescind in vmbus_rescind_cleanup(), we | |
1060 | * should make sure the channel callback is not running any more. | |
1061 | */ | |
1062 | vmbus_reset_channel_cb(channel); | |
1063 | ||
6f3d791f S |
1064 | /* |
1065 | * Now wait for offer handling to complete. | |
1066 | */ | |
7fa32e5e | 1067 | vmbus_rescind_cleanup(channel); |
6f3d791f S |
1068 | while (READ_ONCE(channel->probe_done) == false) { |
1069 | /* | |
1070 | * We wait here until any channel offer is currently | |
1071 | * being processed. | |
1072 | */ | |
1073 | msleep(1); | |
1074 | } | |
1075 | ||
1076 | /* | |
1077 | * At this point, the rescind handling can proceed safely. | |
1078 | */ | |
1079 | ||
d43e2fe7 | 1080 | if (channel->device_obj) { |
499e8401 DC |
1081 | if (channel->chn_rescind_callback) { |
1082 | channel->chn_rescind_callback(channel); | |
b307b389 DC |
1083 | |
1084 | if (clean_up_chan_for_suspend) | |
1085 | check_ready_for_suspend_event(); | |
1086 | ||
54a66265 | 1087 | return; |
499e8401 | 1088 | } |
d43e2fe7 DC |
1089 | /* |
1090 | * We will have to unregister this device from the | |
1091 | * driver core. | |
1092 | */ | |
1093 | dev = get_device(&channel->device_obj->device); | |
1094 | if (dev) { | |
1095 | vmbus_device_unregister(channel->device_obj); | |
1096 | put_device(dev); | |
1097 | } | |
2dd37cb8 | 1098 | } |
54a66265 S |
1099 | if (channel->primary_channel != NULL) { |
1100 | /* | |
1101 | * Sub-channel is being rescinded. Following is the channel | |
1102 | * close sequence when initiated from the driveri (refer to | |
1103 | * vmbus_close() for details): | |
1104 | * 1. Close all sub-channels first | |
1105 | * 2. Then close the primary channel. | |
1106 | */ | |
192b2d78 | 1107 | mutex_lock(&vmbus_connection.channel_mutex); |
54a66265 S |
1108 | if (channel->state == CHANNEL_OPEN_STATE) { |
1109 | /* | |
1110 | * The channel is currently not open; | |
1111 | * it is safe for us to cleanup the channel. | |
1112 | */ | |
800b9329 | 1113 | hv_process_channel_removal(channel); |
7fa32e5e S |
1114 | } else { |
1115 | complete(&channel->rescind_event); | |
54a66265 | 1116 | } |
192b2d78 | 1117 | mutex_unlock(&vmbus_connection.channel_mutex); |
54a66265 | 1118 | } |
b307b389 DC |
1119 | |
1120 | /* The "channel" may have been freed. Do not access it any longer. */ | |
1121 | ||
1122 | if (clean_up_chan_for_suspend) | |
1123 | check_ready_for_suspend_event(); | |
85d9aa70 DC |
1124 | } |
1125 | ||
1126 | void vmbus_hvsock_device_unregister(struct vmbus_channel *channel) | |
1127 | { | |
85d9aa70 DC |
1128 | BUG_ON(!is_hvsock_channel(channel)); |
1129 | ||
9d35593b DC |
1130 | /* We always get a rescind msg when a connection is closed. */ |
1131 | while (!READ_ONCE(channel->probe_done) || !READ_ONCE(channel->rescind)) | |
1132 | msleep(1); | |
1133 | ||
85d9aa70 | 1134 | vmbus_device_unregister(channel->device_obj); |
3e7ee490 | 1135 | } |
85d9aa70 DC |
1136 | EXPORT_SYMBOL_GPL(vmbus_hvsock_device_unregister); |
1137 | ||
3e7ee490 | 1138 | |
3e189519 | 1139 | /* |
e98cb276 HZ |
1140 | * vmbus_onoffers_delivered - |
1141 | * This is invoked when all offers have been delivered. | |
bd60c33e GKH |
1142 | * |
1143 | * Nothing to do here. | |
1144 | */ | |
e98cb276 | 1145 | static void vmbus_onoffers_delivered( |
bd60c33e | 1146 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1147 | { |
3e7ee490 HJ |
1148 | } |
1149 | ||
3e189519 | 1150 | /* |
e98cb276 | 1151 | * vmbus_onopen_result - Open result handler. |
bd60c33e GKH |
1152 | * |
1153 | * This is invoked when we received a response to our channel open request. | |
1154 | * Find the matching request, copy the response and signal the requesting | |
1155 | * thread. | |
1156 | */ | |
e98cb276 | 1157 | static void vmbus_onopen_result(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1158 | { |
bd60c33e | 1159 | struct vmbus_channel_open_result *result; |
188963ec HZ |
1160 | struct vmbus_channel_msginfo *msginfo; |
1161 | struct vmbus_channel_message_header *requestheader; | |
1162 | struct vmbus_channel_open_channel *openmsg; | |
dd0813b6 | 1163 | unsigned long flags; |
3e7ee490 | 1164 | |
bd60c33e | 1165 | result = (struct vmbus_channel_open_result *)hdr; |
3e7ee490 | 1166 | |
9f23ce1a VK |
1167 | trace_vmbus_onopen_result(result); |
1168 | ||
bd60c33e GKH |
1169 | /* |
1170 | * Find the open msg, copy the result and signal/unblock the wait event | |
1171 | */ | |
15b2f647 | 1172 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 1173 | |
ebb61e5f HJ |
1174 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
1175 | msglistentry) { | |
188963ec | 1176 | requestheader = |
c50f7fb2 | 1177 | (struct vmbus_channel_message_header *)msginfo->msg; |
188963ec | 1178 | |
c50f7fb2 | 1179 | if (requestheader->msgtype == CHANNELMSG_OPENCHANNEL) { |
188963ec | 1180 | openmsg = |
c50f7fb2 HZ |
1181 | (struct vmbus_channel_open_channel *)msginfo->msg; |
1182 | if (openmsg->child_relid == result->child_relid && | |
1183 | openmsg->openid == result->openid) { | |
1184 | memcpy(&msginfo->response.open_result, | |
bd60c33e | 1185 | result, |
9568a193 S |
1186 | sizeof( |
1187 | struct vmbus_channel_open_result)); | |
1188 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
1189 | break; |
1190 | } | |
1191 | } | |
1192 | } | |
15b2f647 | 1193 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
1194 | } |
1195 | ||
3e189519 | 1196 | /* |
e98cb276 | 1197 | * vmbus_ongpadl_created - GPADL created handler. |
bd60c33e GKH |
1198 | * |
1199 | * This is invoked when we received a response to our gpadl create request. | |
1200 | * Find the matching request, copy the response and signal the requesting | |
1201 | * thread. | |
1202 | */ | |
e98cb276 | 1203 | static void vmbus_ongpadl_created(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1204 | { |
188963ec | 1205 | struct vmbus_channel_gpadl_created *gpadlcreated; |
188963ec HZ |
1206 | struct vmbus_channel_msginfo *msginfo; |
1207 | struct vmbus_channel_message_header *requestheader; | |
1208 | struct vmbus_channel_gpadl_header *gpadlheader; | |
dd0813b6 | 1209 | unsigned long flags; |
3e7ee490 | 1210 | |
188963ec | 1211 | gpadlcreated = (struct vmbus_channel_gpadl_created *)hdr; |
3e7ee490 | 1212 | |
a13bf0ea VK |
1213 | trace_vmbus_ongpadl_created(gpadlcreated); |
1214 | ||
bd60c33e GKH |
1215 | /* |
1216 | * Find the establish msg, copy the result and signal/unblock the wait | |
1217 | * event | |
1218 | */ | |
15b2f647 | 1219 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 1220 | |
ebb61e5f HJ |
1221 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
1222 | msglistentry) { | |
188963ec | 1223 | requestheader = |
c50f7fb2 | 1224 | (struct vmbus_channel_message_header *)msginfo->msg; |
188963ec | 1225 | |
c50f7fb2 | 1226 | if (requestheader->msgtype == CHANNELMSG_GPADL_HEADER) { |
188963ec HZ |
1227 | gpadlheader = |
1228 | (struct vmbus_channel_gpadl_header *)requestheader; | |
1229 | ||
c50f7fb2 HZ |
1230 | if ((gpadlcreated->child_relid == |
1231 | gpadlheader->child_relid) && | |
1232 | (gpadlcreated->gpadl == gpadlheader->gpadl)) { | |
1233 | memcpy(&msginfo->response.gpadl_created, | |
188963ec | 1234 | gpadlcreated, |
9568a193 S |
1235 | sizeof( |
1236 | struct vmbus_channel_gpadl_created)); | |
1237 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
1238 | break; |
1239 | } | |
1240 | } | |
1241 | } | |
15b2f647 | 1242 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
1243 | } |
1244 | ||
3e189519 | 1245 | /* |
e98cb276 | 1246 | * vmbus_ongpadl_torndown - GPADL torndown handler. |
bd60c33e GKH |
1247 | * |
1248 | * This is invoked when we received a response to our gpadl teardown request. | |
1249 | * Find the matching request, copy the response and signal the requesting | |
1250 | * thread. | |
1251 | */ | |
e98cb276 | 1252 | static void vmbus_ongpadl_torndown( |
bd60c33e | 1253 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1254 | { |
188963ec | 1255 | struct vmbus_channel_gpadl_torndown *gpadl_torndown; |
188963ec HZ |
1256 | struct vmbus_channel_msginfo *msginfo; |
1257 | struct vmbus_channel_message_header *requestheader; | |
1258 | struct vmbus_channel_gpadl_teardown *gpadl_teardown; | |
dd0813b6 | 1259 | unsigned long flags; |
3e7ee490 | 1260 | |
188963ec | 1261 | gpadl_torndown = (struct vmbus_channel_gpadl_torndown *)hdr; |
bd60c33e | 1262 | |
602a8c52 VK |
1263 | trace_vmbus_ongpadl_torndown(gpadl_torndown); |
1264 | ||
bd60c33e GKH |
1265 | /* |
1266 | * Find the open msg, copy the result and signal/unblock the wait event | |
1267 | */ | |
15b2f647 | 1268 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 1269 | |
ebb61e5f HJ |
1270 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
1271 | msglistentry) { | |
188963ec | 1272 | requestheader = |
c50f7fb2 | 1273 | (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 1274 | |
c50f7fb2 | 1275 | if (requestheader->msgtype == CHANNELMSG_GPADL_TEARDOWN) { |
188963ec HZ |
1276 | gpadl_teardown = |
1277 | (struct vmbus_channel_gpadl_teardown *)requestheader; | |
3e7ee490 | 1278 | |
c50f7fb2 HZ |
1279 | if (gpadl_torndown->gpadl == gpadl_teardown->gpadl) { |
1280 | memcpy(&msginfo->response.gpadl_torndown, | |
188963ec | 1281 | gpadl_torndown, |
9568a193 S |
1282 | sizeof( |
1283 | struct vmbus_channel_gpadl_torndown)); | |
1284 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
1285 | break; |
1286 | } | |
1287 | } | |
1288 | } | |
15b2f647 | 1289 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
1290 | } |
1291 | ||
3e189519 | 1292 | /* |
e98cb276 | 1293 | * vmbus_onversion_response - Version response handler |
bd60c33e GKH |
1294 | * |
1295 | * This is invoked when we received a response to our initiate contact request. | |
1296 | * Find the matching request, copy the response and signal the requesting | |
1297 | * thread. | |
1298 | */ | |
e98cb276 | 1299 | static void vmbus_onversion_response( |
bd60c33e | 1300 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 1301 | { |
188963ec HZ |
1302 | struct vmbus_channel_msginfo *msginfo; |
1303 | struct vmbus_channel_message_header *requestheader; | |
188963ec | 1304 | struct vmbus_channel_version_response *version_response; |
dd0813b6 | 1305 | unsigned long flags; |
3e7ee490 | 1306 | |
188963ec | 1307 | version_response = (struct vmbus_channel_version_response *)hdr; |
c7924fb0 VK |
1308 | |
1309 | trace_vmbus_onversion_response(version_response); | |
1310 | ||
15b2f647 | 1311 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 1312 | |
ebb61e5f HJ |
1313 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
1314 | msglistentry) { | |
188963ec | 1315 | requestheader = |
c50f7fb2 | 1316 | (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 1317 | |
c50f7fb2 HZ |
1318 | if (requestheader->msgtype == |
1319 | CHANNELMSG_INITIATE_CONTACT) { | |
c50f7fb2 | 1320 | memcpy(&msginfo->response.version_response, |
188963ec | 1321 | version_response, |
bd60c33e | 1322 | sizeof(struct vmbus_channel_version_response)); |
9568a193 | 1323 | complete(&msginfo->waitevent); |
3e7ee490 HJ |
1324 | } |
1325 | } | |
15b2f647 | 1326 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
1327 | } |
1328 | ||
c8212f04 | 1329 | /* Channel message dispatch table */ |
e6242fa0 SH |
1330 | const struct vmbus_channel_message_table_entry |
1331 | channel_message_table[CHANNELMSG_COUNT] = { | |
1332 | { CHANNELMSG_INVALID, 0, NULL }, | |
1333 | { CHANNELMSG_OFFERCHANNEL, 0, vmbus_onoffer }, | |
1334 | { CHANNELMSG_RESCIND_CHANNELOFFER, 0, vmbus_onoffer_rescind }, | |
1335 | { CHANNELMSG_REQUESTOFFERS, 0, NULL }, | |
1336 | { CHANNELMSG_ALLOFFERS_DELIVERED, 1, vmbus_onoffers_delivered }, | |
1337 | { CHANNELMSG_OPENCHANNEL, 0, NULL }, | |
1338 | { CHANNELMSG_OPENCHANNEL_RESULT, 1, vmbus_onopen_result }, | |
1339 | { CHANNELMSG_CLOSECHANNEL, 0, NULL }, | |
1340 | { CHANNELMSG_GPADL_HEADER, 0, NULL }, | |
1341 | { CHANNELMSG_GPADL_BODY, 0, NULL }, | |
1342 | { CHANNELMSG_GPADL_CREATED, 1, vmbus_ongpadl_created }, | |
1343 | { CHANNELMSG_GPADL_TEARDOWN, 0, NULL }, | |
1344 | { CHANNELMSG_GPADL_TORNDOWN, 1, vmbus_ongpadl_torndown }, | |
1345 | { CHANNELMSG_RELID_RELEASED, 0, NULL }, | |
1346 | { CHANNELMSG_INITIATE_CONTACT, 0, NULL }, | |
1347 | { CHANNELMSG_VERSION_RESPONSE, 1, vmbus_onversion_response }, | |
1348 | { CHANNELMSG_UNLOAD, 0, NULL }, | |
1349 | { CHANNELMSG_UNLOAD_RESPONSE, 1, vmbus_unload_response }, | |
1350 | { CHANNELMSG_18, 0, NULL }, | |
1351 | { CHANNELMSG_19, 0, NULL }, | |
1352 | { CHANNELMSG_20, 0, NULL }, | |
1353 | { CHANNELMSG_TL_CONNECT_REQUEST, 0, NULL }, | |
c8212f04 GKH |
1354 | }; |
1355 | ||
3e189519 | 1356 | /* |
e98cb276 | 1357 | * vmbus_onmessage - Handler for channel protocol messages. |
bd60c33e GKH |
1358 | * |
1359 | * This is invoked in the vmbus worker thread context. | |
1360 | */ | |
e98cb276 | 1361 | void vmbus_onmessage(void *context) |
3e7ee490 | 1362 | { |
188963ec | 1363 | struct hv_message *msg = context; |
82250213 | 1364 | struct vmbus_channel_message_header *hdr; |
3e7ee490 HJ |
1365 | int size; |
1366 | ||
f6feebe0 HZ |
1367 | hdr = (struct vmbus_channel_message_header *)msg->u.payload; |
1368 | size = msg->header.payload_size; | |
3e7ee490 | 1369 | |
716fa52f VK |
1370 | trace_vmbus_on_message(hdr); |
1371 | ||
c50f7fb2 | 1372 | if (hdr->msgtype >= CHANNELMSG_COUNT) { |
0a46618d | 1373 | pr_err("Received invalid channel message type %d size %d\n", |
c50f7fb2 | 1374 | hdr->msgtype, size); |
04f50c4d | 1375 | print_hex_dump_bytes("", DUMP_PREFIX_NONE, |
f6feebe0 | 1376 | (unsigned char *)msg->u.payload, size); |
3e7ee490 HJ |
1377 | return; |
1378 | } | |
1379 | ||
b7c6b02f S |
1380 | if (channel_message_table[hdr->msgtype].message_handler) |
1381 | channel_message_table[hdr->msgtype].message_handler(hdr); | |
3e7ee490 | 1382 | else |
0a46618d | 1383 | pr_err("Unhandled channel message type %d\n", hdr->msgtype); |
3e7ee490 HJ |
1384 | } |
1385 | ||
3e189519 | 1386 | /* |
e98cb276 | 1387 | * vmbus_request_offers - Send a request to get all our pending offers. |
bd60c33e | 1388 | */ |
e98cb276 | 1389 | int vmbus_request_offers(void) |
3e7ee490 | 1390 | { |
82250213 | 1391 | struct vmbus_channel_message_header *msg; |
188963ec | 1392 | struct vmbus_channel_msginfo *msginfo; |
51e5181d | 1393 | int ret; |
3e7ee490 | 1394 | |
188963ec | 1395 | msginfo = kmalloc(sizeof(*msginfo) + |
bd60c33e GKH |
1396 | sizeof(struct vmbus_channel_message_header), |
1397 | GFP_KERNEL); | |
188963ec | 1398 | if (!msginfo) |
75910f23 | 1399 | return -ENOMEM; |
3e7ee490 | 1400 | |
c50f7fb2 | 1401 | msg = (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 1402 | |
c50f7fb2 | 1403 | msg->msgtype = CHANNELMSG_REQUESTOFFERS; |
3e7ee490 | 1404 | |
c0bb0392 VK |
1405 | ret = vmbus_post_msg(msg, sizeof(struct vmbus_channel_message_header), |
1406 | true); | |
d6706e63 VK |
1407 | |
1408 | trace_vmbus_request_offers(ret); | |
1409 | ||
bd60c33e | 1410 | if (ret != 0) { |
0a46618d | 1411 | pr_err("Unable to request offers - %d\n", ret); |
3e7ee490 | 1412 | |
0c3b7b2f S |
1413 | goto cleanup; |
1414 | } | |
3e7ee490 | 1415 | |
0c3b7b2f | 1416 | cleanup: |
dd9b15dc | 1417 | kfree(msginfo); |
3e7ee490 | 1418 | |
3e7ee490 HJ |
1419 | return ret; |
1420 | } | |
1421 | ||
e68d2971 S |
1422 | static void invoke_sc_cb(struct vmbus_channel *primary_channel) |
1423 | { | |
1424 | struct list_head *cur, *tmp; | |
1425 | struct vmbus_channel *cur_channel; | |
1426 | ||
1427 | if (primary_channel->sc_creation_callback == NULL) | |
1428 | return; | |
1429 | ||
1430 | list_for_each_safe(cur, tmp, &primary_channel->sc_list) { | |
1431 | cur_channel = list_entry(cur, struct vmbus_channel, sc_list); | |
1432 | ||
1433 | primary_channel->sc_creation_callback(cur_channel); | |
1434 | } | |
1435 | } | |
1436 | ||
1437 | void vmbus_set_sc_create_callback(struct vmbus_channel *primary_channel, | |
1438 | void (*sc_cr_cb)(struct vmbus_channel *new_sc)) | |
1439 | { | |
1440 | primary_channel->sc_creation_callback = sc_cr_cb; | |
1441 | } | |
1442 | EXPORT_SYMBOL_GPL(vmbus_set_sc_create_callback); | |
1443 | ||
1444 | bool vmbus_are_subchannels_present(struct vmbus_channel *primary) | |
1445 | { | |
1446 | bool ret; | |
1447 | ||
1448 | ret = !list_empty(&primary->sc_list); | |
1449 | ||
1450 | if (ret) { | |
1451 | /* | |
1452 | * Invoke the callback on sub-channel creation. | |
1453 | * This will present a uniform interface to the | |
1454 | * clients. | |
1455 | */ | |
1456 | invoke_sc_cb(primary); | |
1457 | } | |
1458 | ||
1459 | return ret; | |
1460 | } | |
1461 | EXPORT_SYMBOL_GPL(vmbus_are_subchannels_present); | |
499e8401 DC |
1462 | |
1463 | void vmbus_set_chn_rescind_callback(struct vmbus_channel *channel, | |
1464 | void (*chn_rescind_cb)(struct vmbus_channel *)) | |
1465 | { | |
1466 | channel->chn_rescind_callback = chn_rescind_cb; | |
1467 | } | |
1468 | EXPORT_SYMBOL_GPL(vmbus_set_chn_rescind_callback); |