Commit | Line | Data |
---|---|---|
3e7ee490 | 1 | /* |
3e7ee490 HJ |
2 | * Copyright (c) 2009, 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> | |
3e7ee490 | 20 | */ |
0a46618d HJ |
21 | #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt |
22 | ||
a0086dc5 | 23 | #include <linux/kernel.h> |
0c3b7b2f S |
24 | #include <linux/sched.h> |
25 | #include <linux/wait.h> | |
a0086dc5 | 26 | #include <linux/mm.h> |
5a0e3ad6 | 27 | #include <linux/slab.h> |
53af545b | 28 | #include <linux/list.h> |
c88c4e4c | 29 | #include <linux/module.h> |
8b5d6d3b | 30 | #include <linux/completion.h> |
41571916 | 31 | #include <linux/delay.h> |
46a97191 | 32 | #include <linux/hyperv.h> |
3f335ea2 | 33 | |
0f2a6619 | 34 | #include "hyperv_vmbus.h" |
3e7ee490 | 35 | |
7047f17d S |
36 | static void init_vp_index(struct vmbus_channel *channel, u16 dev_type); |
37 | ||
38 | static const struct vmbus_device vmbus_devs[] = { | |
39 | /* IDE */ | |
40 | { .dev_type = HV_IDE, | |
41 | HV_IDE_GUID, | |
42 | .perf_device = true, | |
43 | }, | |
44 | ||
45 | /* SCSI */ | |
46 | { .dev_type = HV_SCSI, | |
47 | HV_SCSI_GUID, | |
48 | .perf_device = true, | |
49 | }, | |
50 | ||
51 | /* Fibre Channel */ | |
52 | { .dev_type = HV_FC, | |
53 | HV_SYNTHFC_GUID, | |
54 | .perf_device = true, | |
55 | }, | |
56 | ||
57 | /* Synthetic NIC */ | |
58 | { .dev_type = HV_NIC, | |
59 | HV_NIC_GUID, | |
60 | .perf_device = true, | |
61 | }, | |
62 | ||
63 | /* Network Direct */ | |
64 | { .dev_type = HV_ND, | |
65 | HV_ND_GUID, | |
66 | .perf_device = true, | |
67 | }, | |
68 | ||
69 | /* PCIE */ | |
70 | { .dev_type = HV_PCIE, | |
71 | HV_PCIE_GUID, | |
72 | .perf_device = true, | |
73 | }, | |
74 | ||
75 | /* Synthetic Frame Buffer */ | |
76 | { .dev_type = HV_FB, | |
77 | HV_SYNTHVID_GUID, | |
78 | .perf_device = false, | |
79 | }, | |
80 | ||
81 | /* Synthetic Keyboard */ | |
82 | { .dev_type = HV_KBD, | |
83 | HV_KBD_GUID, | |
84 | .perf_device = false, | |
85 | }, | |
86 | ||
87 | /* Synthetic MOUSE */ | |
88 | { .dev_type = HV_MOUSE, | |
89 | HV_MOUSE_GUID, | |
90 | .perf_device = false, | |
91 | }, | |
92 | ||
93 | /* KVP */ | |
94 | { .dev_type = HV_KVP, | |
95 | HV_KVP_GUID, | |
96 | .perf_device = false, | |
97 | }, | |
98 | ||
99 | /* Time Synch */ | |
100 | { .dev_type = HV_TS, | |
101 | HV_TS_GUID, | |
102 | .perf_device = false, | |
103 | }, | |
104 | ||
105 | /* Heartbeat */ | |
106 | { .dev_type = HV_HB, | |
107 | HV_HEART_BEAT_GUID, | |
108 | .perf_device = false, | |
109 | }, | |
110 | ||
111 | /* Shutdown */ | |
112 | { .dev_type = HV_SHUTDOWN, | |
113 | HV_SHUTDOWN_GUID, | |
114 | .perf_device = false, | |
115 | }, | |
116 | ||
117 | /* File copy */ | |
118 | { .dev_type = HV_FCOPY, | |
119 | HV_FCOPY_GUID, | |
120 | .perf_device = false, | |
121 | }, | |
122 | ||
123 | /* Backup */ | |
124 | { .dev_type = HV_BACKUP, | |
125 | HV_VSS_GUID, | |
126 | .perf_device = false, | |
127 | }, | |
128 | ||
129 | /* Dynamic Memory */ | |
130 | { .dev_type = HV_DM, | |
131 | HV_DM_GUID, | |
132 | .perf_device = false, | |
133 | }, | |
134 | ||
135 | /* Unknown GUID */ | |
136 | { .dev_type = HV_UNKOWN, | |
137 | .perf_device = false, | |
138 | }, | |
139 | }; | |
140 | ||
141 | static u16 hv_get_dev_type(const uuid_le *guid) | |
142 | { | |
143 | u16 i; | |
144 | ||
145 | for (i = HV_IDE; i < HV_UNKOWN; i++) { | |
146 | if (!uuid_le_cmp(*guid, vmbus_devs[i].guid)) | |
147 | return i; | |
148 | } | |
149 | pr_info("Unknown GUID: %pUl\n", guid); | |
150 | return i; | |
151 | } | |
f38e7dd7 | 152 | |
c88c4e4c | 153 | /** |
da0e9631 | 154 | * vmbus_prep_negotiate_resp() - Create default response for Hyper-V Negotiate message |
c88c4e4c HJ |
155 | * @icmsghdrp: Pointer to msg header structure |
156 | * @icmsg_negotiate: Pointer to negotiate message structure | |
157 | * @buf: Raw buffer channel data | |
158 | * | |
159 | * @icmsghdrp is of type &struct icmsg_hdr. | |
160 | * @negop is of type &struct icmsg_negotiate. | |
c836d0ab S |
161 | * Set up and fill in default negotiate response message. |
162 | * | |
6741335b S |
163 | * The fw_version specifies the framework version that |
164 | * we can support and srv_version specifies the service | |
165 | * version we can support. | |
c88c4e4c HJ |
166 | * |
167 | * Mainly used by Hyper-V drivers. | |
168 | */ | |
6741335b | 169 | bool vmbus_prep_negotiate_resp(struct icmsg_hdr *icmsghdrp, |
c836d0ab | 170 | struct icmsg_negotiate *negop, u8 *buf, |
6741335b | 171 | int fw_version, int srv_version) |
c88c4e4c | 172 | { |
6741335b S |
173 | int icframe_major, icframe_minor; |
174 | int icmsg_major, icmsg_minor; | |
175 | int fw_major, fw_minor; | |
176 | int srv_major, srv_minor; | |
c836d0ab | 177 | int i; |
6741335b | 178 | bool found_match = false; |
c836d0ab | 179 | |
a3605300 | 180 | icmsghdrp->icmsgsize = 0x10; |
6741335b S |
181 | fw_major = (fw_version >> 16); |
182 | fw_minor = (fw_version & 0xFFFF); | |
183 | ||
184 | srv_major = (srv_version >> 16); | |
185 | srv_minor = (srv_version & 0xFFFF); | |
c88c4e4c | 186 | |
a3605300 S |
187 | negop = (struct icmsg_negotiate *)&buf[ |
188 | sizeof(struct vmbuspipe_hdr) + | |
189 | sizeof(struct icmsg_hdr)]; | |
c88c4e4c | 190 | |
6741335b S |
191 | icframe_major = negop->icframe_vercnt; |
192 | icframe_minor = 0; | |
193 | ||
194 | icmsg_major = negop->icmsg_vercnt; | |
195 | icmsg_minor = 0; | |
c836d0ab S |
196 | |
197 | /* | |
198 | * Select the framework version number we will | |
199 | * support. | |
200 | */ | |
201 | ||
202 | for (i = 0; i < negop->icframe_vercnt; i++) { | |
6741335b S |
203 | if ((negop->icversion_data[i].major == fw_major) && |
204 | (negop->icversion_data[i].minor == fw_minor)) { | |
205 | icframe_major = negop->icversion_data[i].major; | |
206 | icframe_minor = negop->icversion_data[i].minor; | |
207 | found_match = true; | |
208 | } | |
c836d0ab S |
209 | } |
210 | ||
6741335b S |
211 | if (!found_match) |
212 | goto fw_error; | |
213 | ||
214 | found_match = false; | |
215 | ||
c836d0ab S |
216 | for (i = negop->icframe_vercnt; |
217 | (i < negop->icframe_vercnt + negop->icmsg_vercnt); i++) { | |
6741335b S |
218 | if ((negop->icversion_data[i].major == srv_major) && |
219 | (negop->icversion_data[i].minor == srv_minor)) { | |
220 | icmsg_major = negop->icversion_data[i].major; | |
221 | icmsg_minor = negop->icversion_data[i].minor; | |
222 | found_match = true; | |
223 | } | |
c88c4e4c | 224 | } |
a3605300 | 225 | |
c836d0ab | 226 | /* |
6741335b | 227 | * Respond with the framework and service |
c836d0ab S |
228 | * version numbers we can support. |
229 | */ | |
6741335b S |
230 | |
231 | fw_error: | |
232 | if (!found_match) { | |
233 | negop->icframe_vercnt = 0; | |
234 | negop->icmsg_vercnt = 0; | |
235 | } else { | |
236 | negop->icframe_vercnt = 1; | |
237 | negop->icmsg_vercnt = 1; | |
238 | } | |
239 | ||
240 | negop->icversion_data[0].major = icframe_major; | |
241 | negop->icversion_data[0].minor = icframe_minor; | |
242 | negop->icversion_data[1].major = icmsg_major; | |
243 | negop->icversion_data[1].minor = icmsg_minor; | |
244 | return found_match; | |
c88c4e4c | 245 | } |
a3605300 | 246 | |
da0e9631 | 247 | EXPORT_SYMBOL_GPL(vmbus_prep_negotiate_resp); |
c88c4e4c | 248 | |
3e189519 | 249 | /* |
e98cb276 | 250 | * alloc_channel - Allocate and initialize a vmbus channel object |
bd60c33e | 251 | */ |
50fe56d2 | 252 | static struct vmbus_channel *alloc_channel(void) |
3e7ee490 | 253 | { |
bc63b6f6 | 254 | static atomic_t chan_num = ATOMIC_INIT(0); |
aded7165 | 255 | struct vmbus_channel *channel; |
3e7ee490 | 256 | |
aded7165 | 257 | channel = kzalloc(sizeof(*channel), GFP_ATOMIC); |
3e7ee490 | 258 | if (!channel) |
3e7ee490 | 259 | return NULL; |
3e7ee490 | 260 | |
bc63b6f6 | 261 | channel->id = atomic_inc_return(&chan_num); |
fe760e4d | 262 | channel->acquire_ring_lock = true; |
54411c42 | 263 | spin_lock_init(&channel->inbound_lock); |
67fae053 | 264 | spin_lock_init(&channel->lock); |
e68d2971 S |
265 | |
266 | INIT_LIST_HEAD(&channel->sc_list); | |
3a28fa35 | 267 | INIT_LIST_HEAD(&channel->percpu_list); |
3e7ee490 | 268 | |
3e7ee490 HJ |
269 | return channel; |
270 | } | |
271 | ||
3e189519 | 272 | /* |
e98cb276 | 273 | * free_channel - Release the resources used by the vmbus channel object |
bd60c33e | 274 | */ |
9f3e28e3 | 275 | static void free_channel(struct vmbus_channel *channel) |
3e7ee490 | 276 | { |
aadc3780 | 277 | kfree(channel); |
3e7ee490 HJ |
278 | } |
279 | ||
3a28fa35 S |
280 | static void percpu_channel_enq(void *arg) |
281 | { | |
282 | struct vmbus_channel *channel = arg; | |
283 | int cpu = smp_processor_id(); | |
284 | ||
285 | list_add_tail(&channel->percpu_list, &hv_context.percpu_list[cpu]); | |
286 | } | |
8b5d6d3b | 287 | |
3a28fa35 S |
288 | static void percpu_channel_deq(void *arg) |
289 | { | |
290 | struct vmbus_channel *channel = arg; | |
291 | ||
292 | list_del(&channel->percpu_list); | |
293 | } | |
8b5d6d3b | 294 | |
ed6cfcc5 | 295 | |
f52078cf | 296 | static void vmbus_release_relid(u32 relid) |
4b2f9abe | 297 | { |
ed6cfcc5 | 298 | struct vmbus_channel_relid_released msg; |
4b2f9abe | 299 | |
c8705979 | 300 | memset(&msg, 0, sizeof(struct vmbus_channel_relid_released)); |
ed6cfcc5 | 301 | msg.child_relid = relid; |
c8705979 S |
302 | msg.header.msgtype = CHANNELMSG_RELID_RELEASED; |
303 | vmbus_post_msg(&msg, sizeof(struct vmbus_channel_relid_released)); | |
f52078cf | 304 | } |
c8705979 | 305 | |
f52078cf DC |
306 | void hv_process_channel_removal(struct vmbus_channel *channel, u32 relid) |
307 | { | |
308 | unsigned long flags; | |
309 | struct vmbus_channel *primary_channel; | |
310 | ||
311 | vmbus_release_relid(relid); | |
ed6cfcc5 | 312 | |
34c6801e | 313 | BUG_ON(!channel->rescind); |
85d9aa70 | 314 | BUG_ON(!mutex_is_locked(&vmbus_connection.channel_mutex)); |
34c6801e | 315 | |
2115b561 S |
316 | if (channel->target_cpu != get_cpu()) { |
317 | put_cpu(); | |
3a28fa35 S |
318 | smp_call_function_single(channel->target_cpu, |
319 | percpu_channel_deq, channel, true); | |
2115b561 | 320 | } else { |
3a28fa35 | 321 | percpu_channel_deq(channel); |
2115b561 S |
322 | put_cpu(); |
323 | } | |
3a28fa35 | 324 | |
e68d2971 | 325 | if (channel->primary_channel == NULL) { |
e68d2971 | 326 | list_del(&channel->listentry); |
ca1c4b74 DC |
327 | |
328 | primary_channel = channel; | |
e68d2971 S |
329 | } else { |
330 | primary_channel = channel->primary_channel; | |
67fae053 | 331 | spin_lock_irqsave(&primary_channel->lock, flags); |
565ce642 | 332 | list_del(&channel->sc_list); |
357e836a | 333 | primary_channel->num_sc--; |
67fae053 | 334 | spin_unlock_irqrestore(&primary_channel->lock, flags); |
e68d2971 | 335 | } |
ca1c4b74 DC |
336 | |
337 | /* | |
338 | * We need to free the bit for init_vp_index() to work in the case | |
339 | * of sub-channel, when we reload drivers like hv_netvsc. | |
340 | */ | |
341 | cpumask_clear_cpu(channel->target_cpu, | |
342 | &primary_channel->alloced_cpus_in_node); | |
343 | ||
c8705979 | 344 | free_channel(channel); |
4b2f9abe | 345 | } |
8b5d6d3b | 346 | |
93e5bd06 S |
347 | void vmbus_free_channels(void) |
348 | { | |
813c5b79 DC |
349 | struct vmbus_channel *channel, *tmp; |
350 | ||
351 | list_for_each_entry_safe(channel, tmp, &vmbus_connection.chn_list, | |
352 | listentry) { | |
34c6801e | 353 | /* hv_process_channel_removal() needs this */ |
813c5b79 | 354 | channel->rescind = true; |
93e5bd06 | 355 | |
93e5bd06 | 356 | vmbus_device_unregister(channel->device_obj); |
93e5bd06 S |
357 | } |
358 | } | |
359 | ||
3e189519 | 360 | /* |
e98cb276 | 361 | * vmbus_process_offer - Process the offer by creating a channel/device |
c88c4e4c | 362 | * associated with this offer |
bd60c33e | 363 | */ |
2dd37cb8 | 364 | static void vmbus_process_offer(struct vmbus_channel *newchannel) |
3e7ee490 | 365 | { |
aded7165 | 366 | struct vmbus_channel *channel; |
188963ec | 367 | bool fnew = true; |
0f5e44ca | 368 | unsigned long flags; |
7047f17d | 369 | u16 dev_type; |
85d9aa70 | 370 | int ret; |
3e7ee490 | 371 | |
454f18a9 | 372 | /* Make sure this is a new offer */ |
d6f591e3 | 373 | mutex_lock(&vmbus_connection.channel_mutex); |
3e7ee490 | 374 | |
da9fcb72 | 375 | list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) { |
358d2ee2 S |
376 | if (!uuid_le_cmp(channel->offermsg.offer.if_type, |
377 | newchannel->offermsg.offer.if_type) && | |
378 | !uuid_le_cmp(channel->offermsg.offer.if_instance, | |
379 | newchannel->offermsg.offer.if_instance)) { | |
188963ec | 380 | fnew = false; |
3e7ee490 HJ |
381 | break; |
382 | } | |
383 | } | |
384 | ||
8dfd3326 | 385 | if (fnew) |
c50f7fb2 | 386 | list_add_tail(&newchannel->listentry, |
da9fcb72 | 387 | &vmbus_connection.chn_list); |
bd60c33e | 388 | |
d6f591e3 | 389 | mutex_unlock(&vmbus_connection.channel_mutex); |
3e7ee490 | 390 | |
188963ec | 391 | if (!fnew) { |
e68d2971 S |
392 | /* |
393 | * Check to see if this is a sub-channel. | |
394 | */ | |
395 | if (newchannel->offermsg.offer.sub_channel_index != 0) { | |
396 | /* | |
397 | * Process the sub-channel. | |
398 | */ | |
399 | newchannel->primary_channel = channel; | |
67fae053 | 400 | spin_lock_irqsave(&channel->lock, flags); |
e68d2971 | 401 | list_add_tail(&newchannel->sc_list, &channel->sc_list); |
a13e8bbe | 402 | channel->num_sc++; |
357e836a | 403 | spin_unlock_irqrestore(&channel->lock, flags); |
8dfd3326 VK |
404 | } else |
405 | goto err_free_chan; | |
406 | } | |
e68d2971 | 407 | |
7047f17d S |
408 | dev_type = hv_get_dev_type(&newchannel->offermsg.offer.if_type); |
409 | ||
410 | init_vp_index(newchannel, dev_type); | |
f38e7dd7 | 411 | |
8dfd3326 VK |
412 | if (newchannel->target_cpu != get_cpu()) { |
413 | put_cpu(); | |
414 | smp_call_function_single(newchannel->target_cpu, | |
415 | percpu_channel_enq, | |
416 | newchannel, true); | |
417 | } else { | |
418 | percpu_channel_enq(newchannel); | |
419 | put_cpu(); | |
3e7ee490 HJ |
420 | } |
421 | ||
42dceebe S |
422 | /* |
423 | * This state is used to indicate a successful open | |
424 | * so that when we do close the channel normally, we | |
425 | * can cleanup properly | |
426 | */ | |
427 | newchannel->state = CHANNEL_OPEN_STATE; | |
428 | ||
8dfd3326 VK |
429 | if (!fnew) { |
430 | if (channel->sc_creation_callback != NULL) | |
431 | channel->sc_creation_callback(newchannel); | |
432 | return; | |
433 | } | |
434 | ||
bd60c33e GKH |
435 | /* |
436 | * Start the process of binding this offer to the driver | |
437 | * We need to set the DeviceObject field before calling | |
646f1ea3 | 438 | * vmbus_child_dev_add() |
bd60c33e | 439 | */ |
f2c73011 | 440 | newchannel->device_obj = vmbus_device_create( |
767dff68 HZ |
441 | &newchannel->offermsg.offer.if_type, |
442 | &newchannel->offermsg.offer.if_instance, | |
188963ec | 443 | newchannel); |
9c3a6f7e | 444 | if (!newchannel->device_obj) |
5b1e5b53 | 445 | goto err_deq_chan; |
3e7ee490 | 446 | |
7047f17d | 447 | newchannel->device_obj->device_id = dev_type; |
454f18a9 BP |
448 | /* |
449 | * Add the new device to the bus. This will kick off device-driver | |
450 | * binding which eventually invokes the device driver's AddDevice() | |
451 | * method. | |
452 | */ | |
85d9aa70 DC |
453 | mutex_lock(&vmbus_connection.channel_mutex); |
454 | ret = vmbus_device_register(newchannel->device_obj); | |
455 | mutex_unlock(&vmbus_connection.channel_mutex); | |
456 | ||
457 | if (ret != 0) { | |
d43e2fe7 DC |
458 | pr_err("unable to add child device object (relid %d)\n", |
459 | newchannel->offermsg.child_relid); | |
460 | kfree(newchannel->device_obj); | |
461 | goto err_deq_chan; | |
462 | } | |
9c3a6f7e | 463 | return; |
2dd37cb8 | 464 | |
5b1e5b53 | 465 | err_deq_chan: |
f52078cf DC |
466 | vmbus_release_relid(newchannel->offermsg.child_relid); |
467 | ||
d6f591e3 | 468 | mutex_lock(&vmbus_connection.channel_mutex); |
5b1e5b53 | 469 | list_del(&newchannel->listentry); |
d6f591e3 | 470 | mutex_unlock(&vmbus_connection.channel_mutex); |
5b1e5b53 S |
471 | |
472 | if (newchannel->target_cpu != get_cpu()) { | |
473 | put_cpu(); | |
474 | smp_call_function_single(newchannel->target_cpu, | |
475 | percpu_channel_deq, newchannel, true); | |
476 | } else { | |
477 | percpu_channel_deq(newchannel); | |
478 | put_cpu(); | |
479 | } | |
480 | ||
9c3a6f7e VK |
481 | err_free_chan: |
482 | free_channel(newchannel); | |
3e7ee490 HJ |
483 | } |
484 | ||
a119845f S |
485 | /* |
486 | * We use this state to statically distribute the channel interrupt load. | |
487 | */ | |
1f656ff3 | 488 | static int next_numa_node_id; |
a119845f S |
489 | |
490 | /* | |
491 | * Starting with Win8, we can statically distribute the incoming | |
1f656ff3 S |
492 | * channel interrupt load by binding a channel to VCPU. |
493 | * We do this in a hierarchical fashion: | |
494 | * First distribute the primary channels across available NUMA nodes | |
495 | * and then distribute the subchannels amongst the CPUs in the NUMA | |
496 | * node assigned to the primary channel. | |
497 | * | |
498 | * For pre-win8 hosts or non-performance critical channels we assign the | |
499 | * first CPU in the first NUMA node. | |
a119845f | 500 | */ |
7047f17d | 501 | static void init_vp_index(struct vmbus_channel *channel, u16 dev_type) |
a119845f S |
502 | { |
503 | u32 cur_cpu; | |
7047f17d | 504 | bool perf_chn = vmbus_devs[dev_type].perf_device; |
1f656ff3 S |
505 | struct vmbus_channel *primary = channel->primary_channel; |
506 | int next_node; | |
507 | struct cpumask available_mask; | |
9f01ec53 | 508 | struct cpumask *alloced_mask; |
a119845f | 509 | |
a119845f S |
510 | if ((vmbus_proto_version == VERSION_WS2008) || |
511 | (vmbus_proto_version == VERSION_WIN7) || (!perf_chn)) { | |
512 | /* | |
513 | * Prior to win8, all channel interrupts are | |
514 | * delivered on cpu 0. | |
515 | * Also if the channel is not a performance critical | |
516 | * channel, bind it to cpu 0. | |
517 | */ | |
1f656ff3 | 518 | channel->numa_node = 0; |
d3ba720d | 519 | channel->target_cpu = 0; |
9c6e64ad | 520 | channel->target_vp = hv_context.vp_index[0]; |
d3ba720d | 521 | return; |
a119845f | 522 | } |
ce59fec8 VK |
523 | |
524 | /* | |
1f656ff3 S |
525 | * We distribute primary channels evenly across all the available |
526 | * NUMA nodes and within the assigned NUMA node we will assign the | |
527 | * first available CPU to the primary channel. | |
528 | * The sub-channels will be assigned to the CPUs available in the | |
529 | * NUMA node evenly. | |
ce59fec8 | 530 | */ |
1f656ff3 S |
531 | if (!primary) { |
532 | while (true) { | |
533 | next_node = next_numa_node_id++; | |
534 | if (next_node == nr_node_ids) | |
535 | next_node = next_numa_node_id = 0; | |
536 | if (cpumask_empty(cpumask_of_node(next_node))) | |
537 | continue; | |
538 | break; | |
539 | } | |
540 | channel->numa_node = next_node; | |
541 | primary = channel; | |
542 | } | |
9f01ec53 | 543 | alloced_mask = &hv_context.hv_numa_map[primary->numa_node]; |
1f656ff3 | 544 | |
9f01ec53 | 545 | if (cpumask_weight(alloced_mask) == |
1f656ff3 | 546 | cpumask_weight(cpumask_of_node(primary->numa_node))) { |
ce59fec8 | 547 | /* |
1f656ff3 S |
548 | * We have cycled through all the CPUs in the node; |
549 | * reset the alloced map. | |
ce59fec8 | 550 | */ |
9f01ec53 | 551 | cpumask_clear(alloced_mask); |
ce59fec8 VK |
552 | } |
553 | ||
9f01ec53 | 554 | cpumask_xor(&available_mask, alloced_mask, |
1f656ff3 S |
555 | cpumask_of_node(primary->numa_node)); |
556 | ||
3b71107d | 557 | cur_cpu = -1; |
79fd8e70 VK |
558 | |
559 | /* | |
560 | * Normally Hyper-V host doesn't create more subchannels than there | |
561 | * are VCPUs on the node but it is possible when not all present VCPUs | |
562 | * on the node are initialized by guest. Clear the alloced_cpus_in_node | |
563 | * to start over. | |
564 | */ | |
565 | if (cpumask_equal(&primary->alloced_cpus_in_node, | |
566 | cpumask_of_node(primary->numa_node))) | |
567 | cpumask_clear(&primary->alloced_cpus_in_node); | |
568 | ||
3b71107d DC |
569 | while (true) { |
570 | cur_cpu = cpumask_next(cur_cpu, &available_mask); | |
571 | if (cur_cpu >= nr_cpu_ids) { | |
572 | cur_cpu = -1; | |
573 | cpumask_copy(&available_mask, | |
574 | cpumask_of_node(primary->numa_node)); | |
575 | continue; | |
576 | } | |
577 | ||
ca1c4b74 DC |
578 | /* |
579 | * NOTE: in the case of sub-channel, we clear the sub-channel | |
580 | * related bit(s) in primary->alloced_cpus_in_node in | |
581 | * hv_process_channel_removal(), so when we reload drivers | |
582 | * like hv_netvsc in SMP guest, here we're able to re-allocate | |
583 | * bit from primary->alloced_cpus_in_node. | |
584 | */ | |
3b71107d DC |
585 | if (!cpumask_test_cpu(cur_cpu, |
586 | &primary->alloced_cpus_in_node)) { | |
587 | cpumask_set_cpu(cur_cpu, | |
588 | &primary->alloced_cpus_in_node); | |
589 | cpumask_set_cpu(cur_cpu, alloced_mask); | |
590 | break; | |
591 | } | |
592 | } | |
1f656ff3 | 593 | |
d3ba720d S |
594 | channel->target_cpu = cur_cpu; |
595 | channel->target_vp = hv_context.vp_index[cur_cpu]; | |
a119845f S |
596 | } |
597 | ||
41571916 VK |
598 | static void vmbus_wait_for_unload(void) |
599 | { | |
cd95aad5 VK |
600 | int cpu; |
601 | void *page_addr; | |
602 | struct hv_message *msg; | |
41571916 | 603 | struct vmbus_channel_message_header *hdr; |
cd95aad5 | 604 | u32 message_type; |
41571916 | 605 | |
cd95aad5 VK |
606 | /* |
607 | * CHANNELMSG_UNLOAD_RESPONSE is always delivered to the CPU which was | |
608 | * used for initial contact or to CPU0 depending on host version. When | |
609 | * we're crashing on a different CPU let's hope that IRQ handler on | |
610 | * the cpu which receives CHANNELMSG_UNLOAD_RESPONSE is still | |
611 | * functional and vmbus_unload_response() will complete | |
612 | * vmbus_connection.unload_event. If not, the last thing we can do is | |
613 | * read message pages for all CPUs directly. | |
614 | */ | |
41571916 | 615 | while (1) { |
cd95aad5 VK |
616 | if (completion_done(&vmbus_connection.unload_event)) |
617 | break; | |
41571916 | 618 | |
cd95aad5 VK |
619 | for_each_online_cpu(cpu) { |
620 | page_addr = hv_context.synic_message_page[cpu]; | |
621 | msg = (struct hv_message *)page_addr + | |
622 | VMBUS_MESSAGE_SINT; | |
41571916 | 623 | |
cd95aad5 VK |
624 | message_type = READ_ONCE(msg->header.message_type); |
625 | if (message_type == HVMSG_NONE) | |
626 | continue; | |
41571916 | 627 | |
cd95aad5 VK |
628 | hdr = (struct vmbus_channel_message_header *) |
629 | msg->u.payload; | |
630 | ||
631 | if (hdr->msgtype == CHANNELMSG_UNLOAD_RESPONSE) | |
632 | complete(&vmbus_connection.unload_event); | |
633 | ||
634 | vmbus_signal_eom(msg, message_type); | |
635 | } | |
636 | ||
637 | mdelay(10); | |
638 | } | |
639 | ||
640 | /* | |
641 | * We're crashing and already got the UNLOAD_RESPONSE, cleanup all | |
642 | * maybe-pending messages on all CPUs to be able to receive new | |
643 | * messages after we reconnect. | |
644 | */ | |
645 | for_each_online_cpu(cpu) { | |
646 | page_addr = hv_context.synic_message_page[cpu]; | |
647 | msg = (struct hv_message *)page_addr + VMBUS_MESSAGE_SINT; | |
648 | msg->header.message_type = HVMSG_NONE; | |
41571916 VK |
649 | } |
650 | } | |
651 | ||
2db84eff S |
652 | /* |
653 | * vmbus_unload_response - Handler for the unload response. | |
654 | */ | |
655 | static void vmbus_unload_response(struct vmbus_channel_message_header *hdr) | |
656 | { | |
657 | /* | |
658 | * This is a global event; just wakeup the waiting thread. | |
659 | * Once we successfully unload, we can cleanup the monitor state. | |
660 | */ | |
661 | complete(&vmbus_connection.unload_event); | |
662 | } | |
663 | ||
75ff3a8a | 664 | void vmbus_initiate_unload(bool crash) |
2db84eff S |
665 | { |
666 | struct vmbus_channel_message_header hdr; | |
667 | ||
4a54243f VK |
668 | /* Pre-Win2012R2 hosts don't support reconnect */ |
669 | if (vmbus_proto_version < VERSION_WIN8_1) | |
670 | return; | |
671 | ||
2db84eff S |
672 | init_completion(&vmbus_connection.unload_event); |
673 | memset(&hdr, 0, sizeof(struct vmbus_channel_message_header)); | |
674 | hdr.msgtype = CHANNELMSG_UNLOAD; | |
675 | vmbus_post_msg(&hdr, sizeof(struct vmbus_channel_message_header)); | |
676 | ||
41571916 VK |
677 | /* |
678 | * vmbus_initiate_unload() is also called on crash and the crash can be | |
679 | * happening in an interrupt context, where scheduling is impossible. | |
680 | */ | |
75ff3a8a | 681 | if (!crash) |
41571916 VK |
682 | wait_for_completion(&vmbus_connection.unload_event); |
683 | else | |
684 | vmbus_wait_for_unload(); | |
2db84eff S |
685 | } |
686 | ||
3e189519 | 687 | /* |
e98cb276 | 688 | * vmbus_onoffer - Handler for channel offers from vmbus in parent partition. |
bd60c33e | 689 | * |
bd60c33e | 690 | */ |
e98cb276 | 691 | static void vmbus_onoffer(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 692 | { |
bd60c33e | 693 | struct vmbus_channel_offer_channel *offer; |
188963ec | 694 | struct vmbus_channel *newchannel; |
3e7ee490 | 695 | |
bd60c33e | 696 | offer = (struct vmbus_channel_offer_channel *)hdr; |
3e7ee490 | 697 | |
454f18a9 | 698 | /* Allocate the channel object and save this offer. */ |
e98cb276 | 699 | newchannel = alloc_channel(); |
188963ec | 700 | if (!newchannel) { |
0a46618d | 701 | pr_err("Unable to allocate channel object\n"); |
3e7ee490 HJ |
702 | return; |
703 | } | |
704 | ||
132368bd S |
705 | /* |
706 | * By default we setup state to enable batched | |
707 | * reading. A specific service can choose to | |
708 | * disable this prior to opening the channel. | |
709 | */ | |
710 | newchannel->batched_reading = true; | |
711 | ||
b3bf60c7 S |
712 | /* |
713 | * Setup state for signalling the host. | |
714 | */ | |
715 | newchannel->sig_event = (struct hv_input_signal_event *) | |
716 | (ALIGN((unsigned long) | |
717 | &newchannel->sig_buf, | |
718 | HV_HYPERCALL_PARAM_ALIGN)); | |
719 | ||
720 | newchannel->sig_event->connectionid.asu32 = 0; | |
721 | newchannel->sig_event->connectionid.u.id = VMBUS_EVENT_CONNECTION_ID; | |
722 | newchannel->sig_event->flag_number = 0; | |
723 | newchannel->sig_event->rsvdz = 0; | |
724 | ||
725 | if (vmbus_proto_version != VERSION_WS2008) { | |
726 | newchannel->is_dedicated_interrupt = | |
727 | (offer->is_dedicated_interrupt != 0); | |
728 | newchannel->sig_event->connectionid.u.id = | |
729 | offer->connection_id; | |
730 | } | |
731 | ||
c50f7fb2 | 732 | memcpy(&newchannel->offermsg, offer, |
bd60c33e | 733 | sizeof(struct vmbus_channel_offer_channel)); |
c50f7fb2 HZ |
734 | newchannel->monitor_grp = (u8)offer->monitorid / 32; |
735 | newchannel->monitor_bit = (u8)offer->monitorid % 32; | |
3e7ee490 | 736 | |
2dd37cb8 | 737 | vmbus_process_offer(newchannel); |
3e7ee490 HJ |
738 | } |
739 | ||
3e189519 | 740 | /* |
e98cb276 | 741 | * vmbus_onoffer_rescind - Rescind offer handler. |
bd60c33e GKH |
742 | * |
743 | * We queue a work item to process this offer synchronously | |
744 | */ | |
e98cb276 | 745 | static void vmbus_onoffer_rescind(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 746 | { |
bd60c33e | 747 | struct vmbus_channel_rescind_offer *rescind; |
aded7165 | 748 | struct vmbus_channel *channel; |
d43e2fe7 DC |
749 | unsigned long flags; |
750 | struct device *dev; | |
3e7ee490 | 751 | |
bd60c33e | 752 | rescind = (struct vmbus_channel_rescind_offer *)hdr; |
85d9aa70 DC |
753 | |
754 | mutex_lock(&vmbus_connection.channel_mutex); | |
d43e2fe7 | 755 | channel = relid2channel(rescind->child_relid); |
98e08702 | 756 | |
2dd37cb8 | 757 | if (channel == NULL) { |
f52078cf DC |
758 | /* |
759 | * This is very impossible, because in | |
760 | * vmbus_process_offer(), we have already invoked | |
761 | * vmbus_release_relid() on error. | |
762 | */ | |
85d9aa70 | 763 | goto out; |
2dd37cb8 | 764 | } |
3e7ee490 | 765 | |
d43e2fe7 DC |
766 | spin_lock_irqsave(&channel->lock, flags); |
767 | channel->rescind = true; | |
768 | spin_unlock_irqrestore(&channel->lock, flags); | |
769 | ||
770 | if (channel->device_obj) { | |
499e8401 DC |
771 | if (channel->chn_rescind_callback) { |
772 | channel->chn_rescind_callback(channel); | |
85d9aa70 | 773 | goto out; |
499e8401 | 774 | } |
d43e2fe7 DC |
775 | /* |
776 | * We will have to unregister this device from the | |
777 | * driver core. | |
778 | */ | |
779 | dev = get_device(&channel->device_obj->device); | |
780 | if (dev) { | |
781 | vmbus_device_unregister(channel->device_obj); | |
782 | put_device(dev); | |
783 | } | |
784 | } else { | |
785 | hv_process_channel_removal(channel, | |
786 | channel->offermsg.child_relid); | |
2dd37cb8 | 787 | } |
85d9aa70 DC |
788 | |
789 | out: | |
790 | mutex_unlock(&vmbus_connection.channel_mutex); | |
791 | } | |
792 | ||
793 | void vmbus_hvsock_device_unregister(struct vmbus_channel *channel) | |
794 | { | |
795 | mutex_lock(&vmbus_connection.channel_mutex); | |
796 | ||
797 | BUG_ON(!is_hvsock_channel(channel)); | |
798 | ||
799 | channel->rescind = true; | |
800 | vmbus_device_unregister(channel->device_obj); | |
801 | ||
802 | mutex_unlock(&vmbus_connection.channel_mutex); | |
3e7ee490 | 803 | } |
85d9aa70 DC |
804 | EXPORT_SYMBOL_GPL(vmbus_hvsock_device_unregister); |
805 | ||
3e7ee490 | 806 | |
3e189519 | 807 | /* |
e98cb276 HZ |
808 | * vmbus_onoffers_delivered - |
809 | * This is invoked when all offers have been delivered. | |
bd60c33e GKH |
810 | * |
811 | * Nothing to do here. | |
812 | */ | |
e98cb276 | 813 | static void vmbus_onoffers_delivered( |
bd60c33e | 814 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 815 | { |
3e7ee490 HJ |
816 | } |
817 | ||
3e189519 | 818 | /* |
e98cb276 | 819 | * vmbus_onopen_result - Open result handler. |
bd60c33e GKH |
820 | * |
821 | * This is invoked when we received a response to our channel open request. | |
822 | * Find the matching request, copy the response and signal the requesting | |
823 | * thread. | |
824 | */ | |
e98cb276 | 825 | static void vmbus_onopen_result(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 826 | { |
bd60c33e | 827 | struct vmbus_channel_open_result *result; |
188963ec HZ |
828 | struct vmbus_channel_msginfo *msginfo; |
829 | struct vmbus_channel_message_header *requestheader; | |
830 | struct vmbus_channel_open_channel *openmsg; | |
dd0813b6 | 831 | unsigned long flags; |
3e7ee490 | 832 | |
bd60c33e | 833 | result = (struct vmbus_channel_open_result *)hdr; |
3e7ee490 | 834 | |
bd60c33e GKH |
835 | /* |
836 | * Find the open msg, copy the result and signal/unblock the wait event | |
837 | */ | |
15b2f647 | 838 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 839 | |
ebb61e5f HJ |
840 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
841 | msglistentry) { | |
188963ec | 842 | requestheader = |
c50f7fb2 | 843 | (struct vmbus_channel_message_header *)msginfo->msg; |
188963ec | 844 | |
c50f7fb2 | 845 | if (requestheader->msgtype == CHANNELMSG_OPENCHANNEL) { |
188963ec | 846 | openmsg = |
c50f7fb2 HZ |
847 | (struct vmbus_channel_open_channel *)msginfo->msg; |
848 | if (openmsg->child_relid == result->child_relid && | |
849 | openmsg->openid == result->openid) { | |
850 | memcpy(&msginfo->response.open_result, | |
bd60c33e | 851 | result, |
9568a193 S |
852 | sizeof( |
853 | struct vmbus_channel_open_result)); | |
854 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
855 | break; |
856 | } | |
857 | } | |
858 | } | |
15b2f647 | 859 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
860 | } |
861 | ||
3e189519 | 862 | /* |
e98cb276 | 863 | * vmbus_ongpadl_created - GPADL created handler. |
bd60c33e GKH |
864 | * |
865 | * This is invoked when we received a response to our gpadl create request. | |
866 | * Find the matching request, copy the response and signal the requesting | |
867 | * thread. | |
868 | */ | |
e98cb276 | 869 | static void vmbus_ongpadl_created(struct vmbus_channel_message_header *hdr) |
3e7ee490 | 870 | { |
188963ec | 871 | struct vmbus_channel_gpadl_created *gpadlcreated; |
188963ec HZ |
872 | struct vmbus_channel_msginfo *msginfo; |
873 | struct vmbus_channel_message_header *requestheader; | |
874 | struct vmbus_channel_gpadl_header *gpadlheader; | |
dd0813b6 | 875 | unsigned long flags; |
3e7ee490 | 876 | |
188963ec | 877 | gpadlcreated = (struct vmbus_channel_gpadl_created *)hdr; |
3e7ee490 | 878 | |
bd60c33e GKH |
879 | /* |
880 | * Find the establish msg, copy the result and signal/unblock the wait | |
881 | * event | |
882 | */ | |
15b2f647 | 883 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 884 | |
ebb61e5f HJ |
885 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
886 | msglistentry) { | |
188963ec | 887 | requestheader = |
c50f7fb2 | 888 | (struct vmbus_channel_message_header *)msginfo->msg; |
188963ec | 889 | |
c50f7fb2 | 890 | if (requestheader->msgtype == CHANNELMSG_GPADL_HEADER) { |
188963ec HZ |
891 | gpadlheader = |
892 | (struct vmbus_channel_gpadl_header *)requestheader; | |
893 | ||
c50f7fb2 HZ |
894 | if ((gpadlcreated->child_relid == |
895 | gpadlheader->child_relid) && | |
896 | (gpadlcreated->gpadl == gpadlheader->gpadl)) { | |
897 | memcpy(&msginfo->response.gpadl_created, | |
188963ec | 898 | gpadlcreated, |
9568a193 S |
899 | sizeof( |
900 | struct vmbus_channel_gpadl_created)); | |
901 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
902 | break; |
903 | } | |
904 | } | |
905 | } | |
15b2f647 | 906 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
907 | } |
908 | ||
3e189519 | 909 | /* |
e98cb276 | 910 | * vmbus_ongpadl_torndown - GPADL torndown handler. |
bd60c33e GKH |
911 | * |
912 | * This is invoked when we received a response to our gpadl teardown request. | |
913 | * Find the matching request, copy the response and signal the requesting | |
914 | * thread. | |
915 | */ | |
e98cb276 | 916 | static void vmbus_ongpadl_torndown( |
bd60c33e | 917 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 918 | { |
188963ec | 919 | struct vmbus_channel_gpadl_torndown *gpadl_torndown; |
188963ec HZ |
920 | struct vmbus_channel_msginfo *msginfo; |
921 | struct vmbus_channel_message_header *requestheader; | |
922 | struct vmbus_channel_gpadl_teardown *gpadl_teardown; | |
dd0813b6 | 923 | unsigned long flags; |
3e7ee490 | 924 | |
188963ec | 925 | gpadl_torndown = (struct vmbus_channel_gpadl_torndown *)hdr; |
bd60c33e GKH |
926 | |
927 | /* | |
928 | * Find the open msg, copy the result and signal/unblock the wait event | |
929 | */ | |
15b2f647 | 930 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 931 | |
ebb61e5f HJ |
932 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
933 | msglistentry) { | |
188963ec | 934 | requestheader = |
c50f7fb2 | 935 | (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 936 | |
c50f7fb2 | 937 | if (requestheader->msgtype == CHANNELMSG_GPADL_TEARDOWN) { |
188963ec HZ |
938 | gpadl_teardown = |
939 | (struct vmbus_channel_gpadl_teardown *)requestheader; | |
3e7ee490 | 940 | |
c50f7fb2 HZ |
941 | if (gpadl_torndown->gpadl == gpadl_teardown->gpadl) { |
942 | memcpy(&msginfo->response.gpadl_torndown, | |
188963ec | 943 | gpadl_torndown, |
9568a193 S |
944 | sizeof( |
945 | struct vmbus_channel_gpadl_torndown)); | |
946 | complete(&msginfo->waitevent); | |
3e7ee490 HJ |
947 | break; |
948 | } | |
949 | } | |
950 | } | |
15b2f647 | 951 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
952 | } |
953 | ||
3e189519 | 954 | /* |
e98cb276 | 955 | * vmbus_onversion_response - Version response handler |
bd60c33e GKH |
956 | * |
957 | * This is invoked when we received a response to our initiate contact request. | |
958 | * Find the matching request, copy the response and signal the requesting | |
959 | * thread. | |
960 | */ | |
e98cb276 | 961 | static void vmbus_onversion_response( |
bd60c33e | 962 | struct vmbus_channel_message_header *hdr) |
3e7ee490 | 963 | { |
188963ec HZ |
964 | struct vmbus_channel_msginfo *msginfo; |
965 | struct vmbus_channel_message_header *requestheader; | |
188963ec | 966 | struct vmbus_channel_version_response *version_response; |
dd0813b6 | 967 | unsigned long flags; |
3e7ee490 | 968 | |
188963ec | 969 | version_response = (struct vmbus_channel_version_response *)hdr; |
15b2f647 | 970 | spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 | 971 | |
ebb61e5f HJ |
972 | list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list, |
973 | msglistentry) { | |
188963ec | 974 | requestheader = |
c50f7fb2 | 975 | (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 976 | |
c50f7fb2 HZ |
977 | if (requestheader->msgtype == |
978 | CHANNELMSG_INITIATE_CONTACT) { | |
c50f7fb2 | 979 | memcpy(&msginfo->response.version_response, |
188963ec | 980 | version_response, |
bd60c33e | 981 | sizeof(struct vmbus_channel_version_response)); |
9568a193 | 982 | complete(&msginfo->waitevent); |
3e7ee490 HJ |
983 | } |
984 | } | |
15b2f647 | 985 | spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags); |
3e7ee490 HJ |
986 | } |
987 | ||
c8212f04 | 988 | /* Channel message dispatch table */ |
652594c7 | 989 | struct vmbus_channel_message_table_entry |
b7c6b02f | 990 | channel_message_table[CHANNELMSG_COUNT] = { |
652594c7 DC |
991 | {CHANNELMSG_INVALID, 0, NULL}, |
992 | {CHANNELMSG_OFFERCHANNEL, 0, vmbus_onoffer}, | |
993 | {CHANNELMSG_RESCIND_CHANNELOFFER, 0, vmbus_onoffer_rescind}, | |
994 | {CHANNELMSG_REQUESTOFFERS, 0, NULL}, | |
995 | {CHANNELMSG_ALLOFFERS_DELIVERED, 1, vmbus_onoffers_delivered}, | |
996 | {CHANNELMSG_OPENCHANNEL, 0, NULL}, | |
997 | {CHANNELMSG_OPENCHANNEL_RESULT, 1, vmbus_onopen_result}, | |
998 | {CHANNELMSG_CLOSECHANNEL, 0, NULL}, | |
999 | {CHANNELMSG_GPADL_HEADER, 0, NULL}, | |
1000 | {CHANNELMSG_GPADL_BODY, 0, NULL}, | |
1001 | {CHANNELMSG_GPADL_CREATED, 1, vmbus_ongpadl_created}, | |
1002 | {CHANNELMSG_GPADL_TEARDOWN, 0, NULL}, | |
1003 | {CHANNELMSG_GPADL_TORNDOWN, 1, vmbus_ongpadl_torndown}, | |
1004 | {CHANNELMSG_RELID_RELEASED, 0, NULL}, | |
1005 | {CHANNELMSG_INITIATE_CONTACT, 0, NULL}, | |
1006 | {CHANNELMSG_VERSION_RESPONSE, 1, vmbus_onversion_response}, | |
1007 | {CHANNELMSG_UNLOAD, 0, NULL}, | |
2db84eff | 1008 | {CHANNELMSG_UNLOAD_RESPONSE, 1, vmbus_unload_response}, |
5c23a1a5 DC |
1009 | {CHANNELMSG_18, 0, NULL}, |
1010 | {CHANNELMSG_19, 0, NULL}, | |
1011 | {CHANNELMSG_20, 0, NULL}, | |
1012 | {CHANNELMSG_TL_CONNECT_REQUEST, 0, NULL}, | |
c8212f04 GKH |
1013 | }; |
1014 | ||
3e189519 | 1015 | /* |
e98cb276 | 1016 | * vmbus_onmessage - Handler for channel protocol messages. |
bd60c33e GKH |
1017 | * |
1018 | * This is invoked in the vmbus worker thread context. | |
1019 | */ | |
e98cb276 | 1020 | void vmbus_onmessage(void *context) |
3e7ee490 | 1021 | { |
188963ec | 1022 | struct hv_message *msg = context; |
82250213 | 1023 | struct vmbus_channel_message_header *hdr; |
3e7ee490 HJ |
1024 | int size; |
1025 | ||
f6feebe0 HZ |
1026 | hdr = (struct vmbus_channel_message_header *)msg->u.payload; |
1027 | size = msg->header.payload_size; | |
3e7ee490 | 1028 | |
c50f7fb2 | 1029 | if (hdr->msgtype >= CHANNELMSG_COUNT) { |
0a46618d | 1030 | pr_err("Received invalid channel message type %d size %d\n", |
c50f7fb2 | 1031 | hdr->msgtype, size); |
04f50c4d | 1032 | print_hex_dump_bytes("", DUMP_PREFIX_NONE, |
f6feebe0 | 1033 | (unsigned char *)msg->u.payload, size); |
3e7ee490 HJ |
1034 | return; |
1035 | } | |
1036 | ||
b7c6b02f S |
1037 | if (channel_message_table[hdr->msgtype].message_handler) |
1038 | channel_message_table[hdr->msgtype].message_handler(hdr); | |
3e7ee490 | 1039 | else |
0a46618d | 1040 | pr_err("Unhandled channel message type %d\n", hdr->msgtype); |
3e7ee490 HJ |
1041 | } |
1042 | ||
3e189519 | 1043 | /* |
e98cb276 | 1044 | * vmbus_request_offers - Send a request to get all our pending offers. |
bd60c33e | 1045 | */ |
e98cb276 | 1046 | int vmbus_request_offers(void) |
3e7ee490 | 1047 | { |
82250213 | 1048 | struct vmbus_channel_message_header *msg; |
188963ec | 1049 | struct vmbus_channel_msginfo *msginfo; |
51e5181d | 1050 | int ret; |
3e7ee490 | 1051 | |
188963ec | 1052 | msginfo = kmalloc(sizeof(*msginfo) + |
bd60c33e GKH |
1053 | sizeof(struct vmbus_channel_message_header), |
1054 | GFP_KERNEL); | |
188963ec | 1055 | if (!msginfo) |
75910f23 | 1056 | return -ENOMEM; |
3e7ee490 | 1057 | |
c50f7fb2 | 1058 | msg = (struct vmbus_channel_message_header *)msginfo->msg; |
3e7ee490 | 1059 | |
c50f7fb2 | 1060 | msg->msgtype = CHANNELMSG_REQUESTOFFERS; |
3e7ee490 | 1061 | |
3e7ee490 | 1062 | |
c6977677 | 1063 | ret = vmbus_post_msg(msg, |
bd60c33e GKH |
1064 | sizeof(struct vmbus_channel_message_header)); |
1065 | if (ret != 0) { | |
0a46618d | 1066 | pr_err("Unable to request offers - %d\n", ret); |
3e7ee490 | 1067 | |
0c3b7b2f S |
1068 | goto cleanup; |
1069 | } | |
3e7ee490 | 1070 | |
0c3b7b2f | 1071 | cleanup: |
dd9b15dc | 1072 | kfree(msginfo); |
3e7ee490 | 1073 | |
3e7ee490 HJ |
1074 | return ret; |
1075 | } | |
1076 | ||
e68d2971 S |
1077 | /* |
1078 | * Retrieve the (sub) channel on which to send an outgoing request. | |
a13e8bbe S |
1079 | * When a primary channel has multiple sub-channels, we try to |
1080 | * distribute the load equally amongst all available channels. | |
e68d2971 S |
1081 | */ |
1082 | struct vmbus_channel *vmbus_get_outgoing_channel(struct vmbus_channel *primary) | |
1083 | { | |
1084 | struct list_head *cur, *tmp; | |
87712bf8 | 1085 | int cur_cpu; |
e68d2971 S |
1086 | struct vmbus_channel *cur_channel; |
1087 | struct vmbus_channel *outgoing_channel = primary; | |
a13e8bbe S |
1088 | int next_channel; |
1089 | int i = 1; | |
e68d2971 S |
1090 | |
1091 | if (list_empty(&primary->sc_list)) | |
1092 | return outgoing_channel; | |
1093 | ||
a13e8bbe S |
1094 | next_channel = primary->next_oc++; |
1095 | ||
1096 | if (next_channel > (primary->num_sc)) { | |
1097 | primary->next_oc = 0; | |
1098 | return outgoing_channel; | |
1099 | } | |
1100 | ||
87712bf8 S |
1101 | cur_cpu = hv_context.vp_index[get_cpu()]; |
1102 | put_cpu(); | |
e68d2971 S |
1103 | list_for_each_safe(cur, tmp, &primary->sc_list) { |
1104 | cur_channel = list_entry(cur, struct vmbus_channel, sc_list); | |
1105 | if (cur_channel->state != CHANNEL_OPENED_STATE) | |
1106 | continue; | |
1107 | ||
1108 | if (cur_channel->target_vp == cur_cpu) | |
1109 | return cur_channel; | |
1110 | ||
a13e8bbe S |
1111 | if (i == next_channel) |
1112 | return cur_channel; | |
e68d2971 | 1113 | |
a13e8bbe | 1114 | i++; |
e68d2971 S |
1115 | } |
1116 | ||
1117 | return outgoing_channel; | |
1118 | } | |
1119 | EXPORT_SYMBOL_GPL(vmbus_get_outgoing_channel); | |
1120 | ||
1121 | static void invoke_sc_cb(struct vmbus_channel *primary_channel) | |
1122 | { | |
1123 | struct list_head *cur, *tmp; | |
1124 | struct vmbus_channel *cur_channel; | |
1125 | ||
1126 | if (primary_channel->sc_creation_callback == NULL) | |
1127 | return; | |
1128 | ||
1129 | list_for_each_safe(cur, tmp, &primary_channel->sc_list) { | |
1130 | cur_channel = list_entry(cur, struct vmbus_channel, sc_list); | |
1131 | ||
1132 | primary_channel->sc_creation_callback(cur_channel); | |
1133 | } | |
1134 | } | |
1135 | ||
1136 | void vmbus_set_sc_create_callback(struct vmbus_channel *primary_channel, | |
1137 | void (*sc_cr_cb)(struct vmbus_channel *new_sc)) | |
1138 | { | |
1139 | primary_channel->sc_creation_callback = sc_cr_cb; | |
1140 | } | |
1141 | EXPORT_SYMBOL_GPL(vmbus_set_sc_create_callback); | |
1142 | ||
1143 | bool vmbus_are_subchannels_present(struct vmbus_channel *primary) | |
1144 | { | |
1145 | bool ret; | |
1146 | ||
1147 | ret = !list_empty(&primary->sc_list); | |
1148 | ||
1149 | if (ret) { | |
1150 | /* | |
1151 | * Invoke the callback on sub-channel creation. | |
1152 | * This will present a uniform interface to the | |
1153 | * clients. | |
1154 | */ | |
1155 | invoke_sc_cb(primary); | |
1156 | } | |
1157 | ||
1158 | return ret; | |
1159 | } | |
1160 | EXPORT_SYMBOL_GPL(vmbus_are_subchannels_present); | |
499e8401 DC |
1161 | |
1162 | void vmbus_set_chn_rescind_callback(struct vmbus_channel *channel, | |
1163 | void (*chn_rescind_cb)(struct vmbus_channel *)) | |
1164 | { | |
1165 | channel->chn_rescind_callback = chn_rescind_cb; | |
1166 | } | |
1167 | EXPORT_SYMBOL_GPL(vmbus_set_chn_rescind_callback); |