channel_mgmt.c 23 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882
  1. /*
  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>
  20. */
  21. #define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
  22. #include <linux/kernel.h>
  23. #include <linux/sched.h>
  24. #include <linux/wait.h>
  25. #include <linux/mm.h>
  26. #include <linux/slab.h>
  27. #include <linux/list.h>
  28. #include <linux/module.h>
  29. #include <linux/completion.h>
  30. #include <linux/hyperv.h>
  31. #include "hyperv_vmbus.h"
  32. struct vmbus_channel_message_table_entry {
  33. enum vmbus_channel_message_type message_type;
  34. void (*message_handler)(struct vmbus_channel_message_header *msg);
  35. };
  36. /**
  37. * vmbus_prep_negotiate_resp() - Create default response for Hyper-V Negotiate message
  38. * @icmsghdrp: Pointer to msg header structure
  39. * @icmsg_negotiate: Pointer to negotiate message structure
  40. * @buf: Raw buffer channel data
  41. *
  42. * @icmsghdrp is of type &struct icmsg_hdr.
  43. * @negop is of type &struct icmsg_negotiate.
  44. * Set up and fill in default negotiate response message.
  45. *
  46. * The fw_version specifies the framework version that
  47. * we can support and srv_version specifies the service
  48. * version we can support.
  49. *
  50. * Mainly used by Hyper-V drivers.
  51. */
  52. bool vmbus_prep_negotiate_resp(struct icmsg_hdr *icmsghdrp,
  53. struct icmsg_negotiate *negop, u8 *buf,
  54. int fw_version, int srv_version)
  55. {
  56. int icframe_major, icframe_minor;
  57. int icmsg_major, icmsg_minor;
  58. int fw_major, fw_minor;
  59. int srv_major, srv_minor;
  60. int i;
  61. bool found_match = false;
  62. icmsghdrp->icmsgsize = 0x10;
  63. fw_major = (fw_version >> 16);
  64. fw_minor = (fw_version & 0xFFFF);
  65. srv_major = (srv_version >> 16);
  66. srv_minor = (srv_version & 0xFFFF);
  67. negop = (struct icmsg_negotiate *)&buf[
  68. sizeof(struct vmbuspipe_hdr) +
  69. sizeof(struct icmsg_hdr)];
  70. icframe_major = negop->icframe_vercnt;
  71. icframe_minor = 0;
  72. icmsg_major = negop->icmsg_vercnt;
  73. icmsg_minor = 0;
  74. /*
  75. * Select the framework version number we will
  76. * support.
  77. */
  78. for (i = 0; i < negop->icframe_vercnt; i++) {
  79. if ((negop->icversion_data[i].major == fw_major) &&
  80. (negop->icversion_data[i].minor == fw_minor)) {
  81. icframe_major = negop->icversion_data[i].major;
  82. icframe_minor = negop->icversion_data[i].minor;
  83. found_match = true;
  84. }
  85. }
  86. if (!found_match)
  87. goto fw_error;
  88. found_match = false;
  89. for (i = negop->icframe_vercnt;
  90. (i < negop->icframe_vercnt + negop->icmsg_vercnt); i++) {
  91. if ((negop->icversion_data[i].major == srv_major) &&
  92. (negop->icversion_data[i].minor == srv_minor)) {
  93. icmsg_major = negop->icversion_data[i].major;
  94. icmsg_minor = negop->icversion_data[i].minor;
  95. found_match = true;
  96. }
  97. }
  98. /*
  99. * Respond with the framework and service
  100. * version numbers we can support.
  101. */
  102. fw_error:
  103. if (!found_match) {
  104. negop->icframe_vercnt = 0;
  105. negop->icmsg_vercnt = 0;
  106. } else {
  107. negop->icframe_vercnt = 1;
  108. negop->icmsg_vercnt = 1;
  109. }
  110. negop->icversion_data[0].major = icframe_major;
  111. negop->icversion_data[0].minor = icframe_minor;
  112. negop->icversion_data[1].major = icmsg_major;
  113. negop->icversion_data[1].minor = icmsg_minor;
  114. return found_match;
  115. }
  116. EXPORT_SYMBOL_GPL(vmbus_prep_negotiate_resp);
  117. /*
  118. * alloc_channel - Allocate and initialize a vmbus channel object
  119. */
  120. static struct vmbus_channel *alloc_channel(void)
  121. {
  122. struct vmbus_channel *channel;
  123. channel = kzalloc(sizeof(*channel), GFP_ATOMIC);
  124. if (!channel)
  125. return NULL;
  126. spin_lock_init(&channel->inbound_lock);
  127. spin_lock_init(&channel->sc_lock);
  128. INIT_LIST_HEAD(&channel->sc_list);
  129. INIT_LIST_HEAD(&channel->percpu_list);
  130. channel->controlwq = create_workqueue("hv_vmbus_ctl");
  131. if (!channel->controlwq) {
  132. kfree(channel);
  133. return NULL;
  134. }
  135. return channel;
  136. }
  137. /*
  138. * release_hannel - Release the vmbus channel object itself
  139. */
  140. static void release_channel(struct work_struct *work)
  141. {
  142. struct vmbus_channel *channel = container_of(work,
  143. struct vmbus_channel,
  144. work);
  145. destroy_workqueue(channel->controlwq);
  146. kfree(channel);
  147. }
  148. /*
  149. * free_channel - Release the resources used by the vmbus channel object
  150. */
  151. static void free_channel(struct vmbus_channel *channel)
  152. {
  153. /*
  154. * We have to release the channel's workqueue/thread in the vmbus's
  155. * workqueue/thread context
  156. * ie we can't destroy ourselves.
  157. */
  158. INIT_WORK(&channel->work, release_channel);
  159. queue_work(vmbus_connection.work_queue, &channel->work);
  160. }
  161. static void percpu_channel_enq(void *arg)
  162. {
  163. struct vmbus_channel *channel = arg;
  164. int cpu = smp_processor_id();
  165. list_add_tail(&channel->percpu_list, &hv_context.percpu_list[cpu]);
  166. }
  167. static void percpu_channel_deq(void *arg)
  168. {
  169. struct vmbus_channel *channel = arg;
  170. list_del(&channel->percpu_list);
  171. }
  172. /*
  173. * vmbus_process_rescind_offer -
  174. * Rescind the offer by initiating a device removal
  175. */
  176. static void vmbus_process_rescind_offer(struct work_struct *work)
  177. {
  178. struct vmbus_channel *channel = container_of(work,
  179. struct vmbus_channel,
  180. work);
  181. unsigned long flags;
  182. struct vmbus_channel *primary_channel;
  183. struct vmbus_channel_relid_released msg;
  184. if (channel->device_obj)
  185. vmbus_device_unregister(channel->device_obj);
  186. memset(&msg, 0, sizeof(struct vmbus_channel_relid_released));
  187. msg.child_relid = channel->offermsg.child_relid;
  188. msg.header.msgtype = CHANNELMSG_RELID_RELEASED;
  189. vmbus_post_msg(&msg, sizeof(struct vmbus_channel_relid_released));
  190. if (channel->target_cpu != get_cpu()) {
  191. put_cpu();
  192. smp_call_function_single(channel->target_cpu,
  193. percpu_channel_deq, channel, true);
  194. } else {
  195. percpu_channel_deq(channel);
  196. put_cpu();
  197. }
  198. if (channel->primary_channel == NULL) {
  199. spin_lock_irqsave(&vmbus_connection.channel_lock, flags);
  200. list_del(&channel->listentry);
  201. spin_unlock_irqrestore(&vmbus_connection.channel_lock, flags);
  202. } else {
  203. primary_channel = channel->primary_channel;
  204. spin_lock_irqsave(&primary_channel->sc_lock, flags);
  205. list_del(&channel->sc_list);
  206. spin_unlock_irqrestore(&primary_channel->sc_lock, flags);
  207. }
  208. free_channel(channel);
  209. }
  210. void vmbus_free_channels(void)
  211. {
  212. struct vmbus_channel *channel;
  213. list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) {
  214. vmbus_device_unregister(channel->device_obj);
  215. kfree(channel->device_obj);
  216. free_channel(channel);
  217. }
  218. }
  219. /*
  220. * vmbus_process_offer - Process the offer by creating a channel/device
  221. * associated with this offer
  222. */
  223. static void vmbus_process_offer(struct work_struct *work)
  224. {
  225. struct vmbus_channel *newchannel = container_of(work,
  226. struct vmbus_channel,
  227. work);
  228. struct vmbus_channel *channel;
  229. bool fnew = true;
  230. bool enq = false;
  231. int ret;
  232. unsigned long flags;
  233. /* The next possible work is rescind handling */
  234. INIT_WORK(&newchannel->work, vmbus_process_rescind_offer);
  235. /* Make sure this is a new offer */
  236. spin_lock_irqsave(&vmbus_connection.channel_lock, flags);
  237. list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) {
  238. if (!uuid_le_cmp(channel->offermsg.offer.if_type,
  239. newchannel->offermsg.offer.if_type) &&
  240. !uuid_le_cmp(channel->offermsg.offer.if_instance,
  241. newchannel->offermsg.offer.if_instance)) {
  242. fnew = false;
  243. break;
  244. }
  245. }
  246. if (fnew) {
  247. list_add_tail(&newchannel->listentry,
  248. &vmbus_connection.chn_list);
  249. enq = true;
  250. }
  251. spin_unlock_irqrestore(&vmbus_connection.channel_lock, flags);
  252. if (enq) {
  253. if (newchannel->target_cpu != get_cpu()) {
  254. put_cpu();
  255. smp_call_function_single(newchannel->target_cpu,
  256. percpu_channel_enq,
  257. newchannel, true);
  258. } else {
  259. percpu_channel_enq(newchannel);
  260. put_cpu();
  261. }
  262. }
  263. if (!fnew) {
  264. /*
  265. * Check to see if this is a sub-channel.
  266. */
  267. if (newchannel->offermsg.offer.sub_channel_index != 0) {
  268. /*
  269. * Process the sub-channel.
  270. */
  271. newchannel->primary_channel = channel;
  272. spin_lock_irqsave(&channel->sc_lock, flags);
  273. list_add_tail(&newchannel->sc_list, &channel->sc_list);
  274. spin_unlock_irqrestore(&channel->sc_lock, flags);
  275. if (newchannel->target_cpu != get_cpu()) {
  276. put_cpu();
  277. smp_call_function_single(newchannel->target_cpu,
  278. percpu_channel_enq,
  279. newchannel, true);
  280. } else {
  281. percpu_channel_enq(newchannel);
  282. put_cpu();
  283. }
  284. newchannel->state = CHANNEL_OPEN_STATE;
  285. if (channel->sc_creation_callback != NULL)
  286. channel->sc_creation_callback(newchannel);
  287. return;
  288. }
  289. free_channel(newchannel);
  290. return;
  291. }
  292. /*
  293. * This state is used to indicate a successful open
  294. * so that when we do close the channel normally, we
  295. * can cleanup properly
  296. */
  297. newchannel->state = CHANNEL_OPEN_STATE;
  298. /*
  299. * Start the process of binding this offer to the driver
  300. * We need to set the DeviceObject field before calling
  301. * vmbus_child_dev_add()
  302. */
  303. newchannel->device_obj = vmbus_device_create(
  304. &newchannel->offermsg.offer.if_type,
  305. &newchannel->offermsg.offer.if_instance,
  306. newchannel);
  307. /*
  308. * Add the new device to the bus. This will kick off device-driver
  309. * binding which eventually invokes the device driver's AddDevice()
  310. * method.
  311. */
  312. ret = vmbus_device_register(newchannel->device_obj);
  313. if (ret != 0) {
  314. pr_err("unable to add child device object (relid %d)\n",
  315. newchannel->offermsg.child_relid);
  316. spin_lock_irqsave(&vmbus_connection.channel_lock, flags);
  317. list_del(&newchannel->listentry);
  318. spin_unlock_irqrestore(&vmbus_connection.channel_lock, flags);
  319. kfree(newchannel->device_obj);
  320. free_channel(newchannel);
  321. }
  322. }
  323. enum {
  324. IDE = 0,
  325. SCSI,
  326. NIC,
  327. MAX_PERF_CHN,
  328. };
  329. /*
  330. * This is an array of device_ids (device types) that are performance critical.
  331. * We attempt to distribute the interrupt load for these devices across
  332. * all available CPUs.
  333. */
  334. static const struct hv_vmbus_device_id hp_devs[] = {
  335. /* IDE */
  336. { HV_IDE_GUID, },
  337. /* Storage - SCSI */
  338. { HV_SCSI_GUID, },
  339. /* Network */
  340. { HV_NIC_GUID, },
  341. };
  342. /*
  343. * We use this state to statically distribute the channel interrupt load.
  344. */
  345. static u32 next_vp;
  346. /*
  347. * Starting with Win8, we can statically distribute the incoming
  348. * channel interrupt load by binding a channel to VCPU. We
  349. * implement here a simple round robin scheme for distributing
  350. * the interrupt load.
  351. * We will bind channels that are not performance critical to cpu 0 and
  352. * performance critical channels (IDE, SCSI and Network) will be uniformly
  353. * distributed across all available CPUs.
  354. */
  355. static void init_vp_index(struct vmbus_channel *channel, const uuid_le *type_guid)
  356. {
  357. u32 cur_cpu;
  358. int i;
  359. bool perf_chn = false;
  360. u32 max_cpus = num_online_cpus();
  361. for (i = IDE; i < MAX_PERF_CHN; i++) {
  362. if (!memcmp(type_guid->b, hp_devs[i].guid,
  363. sizeof(uuid_le))) {
  364. perf_chn = true;
  365. break;
  366. }
  367. }
  368. if ((vmbus_proto_version == VERSION_WS2008) ||
  369. (vmbus_proto_version == VERSION_WIN7) || (!perf_chn)) {
  370. /*
  371. * Prior to win8, all channel interrupts are
  372. * delivered on cpu 0.
  373. * Also if the channel is not a performance critical
  374. * channel, bind it to cpu 0.
  375. */
  376. channel->target_cpu = 0;
  377. channel->target_vp = 0;
  378. return;
  379. }
  380. cur_cpu = (++next_vp % max_cpus);
  381. channel->target_cpu = cur_cpu;
  382. channel->target_vp = hv_context.vp_index[cur_cpu];
  383. }
  384. /*
  385. * vmbus_onoffer - Handler for channel offers from vmbus in parent partition.
  386. *
  387. */
  388. static void vmbus_onoffer(struct vmbus_channel_message_header *hdr)
  389. {
  390. struct vmbus_channel_offer_channel *offer;
  391. struct vmbus_channel *newchannel;
  392. offer = (struct vmbus_channel_offer_channel *)hdr;
  393. /* Allocate the channel object and save this offer. */
  394. newchannel = alloc_channel();
  395. if (!newchannel) {
  396. pr_err("Unable to allocate channel object\n");
  397. return;
  398. }
  399. /*
  400. * By default we setup state to enable batched
  401. * reading. A specific service can choose to
  402. * disable this prior to opening the channel.
  403. */
  404. newchannel->batched_reading = true;
  405. /*
  406. * Setup state for signalling the host.
  407. */
  408. newchannel->sig_event = (struct hv_input_signal_event *)
  409. (ALIGN((unsigned long)
  410. &newchannel->sig_buf,
  411. HV_HYPERCALL_PARAM_ALIGN));
  412. newchannel->sig_event->connectionid.asu32 = 0;
  413. newchannel->sig_event->connectionid.u.id = VMBUS_EVENT_CONNECTION_ID;
  414. newchannel->sig_event->flag_number = 0;
  415. newchannel->sig_event->rsvdz = 0;
  416. if (vmbus_proto_version != VERSION_WS2008) {
  417. newchannel->is_dedicated_interrupt =
  418. (offer->is_dedicated_interrupt != 0);
  419. newchannel->sig_event->connectionid.u.id =
  420. offer->connection_id;
  421. }
  422. init_vp_index(newchannel, &offer->offer.if_type);
  423. memcpy(&newchannel->offermsg, offer,
  424. sizeof(struct vmbus_channel_offer_channel));
  425. newchannel->monitor_grp = (u8)offer->monitorid / 32;
  426. newchannel->monitor_bit = (u8)offer->monitorid % 32;
  427. INIT_WORK(&newchannel->work, vmbus_process_offer);
  428. queue_work(newchannel->controlwq, &newchannel->work);
  429. }
  430. /*
  431. * vmbus_onoffer_rescind - Rescind offer handler.
  432. *
  433. * We queue a work item to process this offer synchronously
  434. */
  435. static void vmbus_onoffer_rescind(struct vmbus_channel_message_header *hdr)
  436. {
  437. struct vmbus_channel_rescind_offer *rescind;
  438. struct vmbus_channel *channel;
  439. rescind = (struct vmbus_channel_rescind_offer *)hdr;
  440. channel = relid2channel(rescind->child_relid);
  441. if (channel == NULL)
  442. /* Just return here, no channel found */
  443. return;
  444. /* work is initialized for vmbus_process_rescind_offer() from
  445. * vmbus_process_offer() where the channel got created */
  446. queue_work(channel->controlwq, &channel->work);
  447. }
  448. /*
  449. * vmbus_onoffers_delivered -
  450. * This is invoked when all offers have been delivered.
  451. *
  452. * Nothing to do here.
  453. */
  454. static void vmbus_onoffers_delivered(
  455. struct vmbus_channel_message_header *hdr)
  456. {
  457. }
  458. /*
  459. * vmbus_onopen_result - Open result handler.
  460. *
  461. * This is invoked when we received a response to our channel open request.
  462. * Find the matching request, copy the response and signal the requesting
  463. * thread.
  464. */
  465. static void vmbus_onopen_result(struct vmbus_channel_message_header *hdr)
  466. {
  467. struct vmbus_channel_open_result *result;
  468. struct vmbus_channel_msginfo *msginfo;
  469. struct vmbus_channel_message_header *requestheader;
  470. struct vmbus_channel_open_channel *openmsg;
  471. unsigned long flags;
  472. result = (struct vmbus_channel_open_result *)hdr;
  473. /*
  474. * Find the open msg, copy the result and signal/unblock the wait event
  475. */
  476. spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
  477. list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list,
  478. msglistentry) {
  479. requestheader =
  480. (struct vmbus_channel_message_header *)msginfo->msg;
  481. if (requestheader->msgtype == CHANNELMSG_OPENCHANNEL) {
  482. openmsg =
  483. (struct vmbus_channel_open_channel *)msginfo->msg;
  484. if (openmsg->child_relid == result->child_relid &&
  485. openmsg->openid == result->openid) {
  486. memcpy(&msginfo->response.open_result,
  487. result,
  488. sizeof(
  489. struct vmbus_channel_open_result));
  490. complete(&msginfo->waitevent);
  491. break;
  492. }
  493. }
  494. }
  495. spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
  496. }
  497. /*
  498. * vmbus_ongpadl_created - GPADL created handler.
  499. *
  500. * This is invoked when we received a response to our gpadl create request.
  501. * Find the matching request, copy the response and signal the requesting
  502. * thread.
  503. */
  504. static void vmbus_ongpadl_created(struct vmbus_channel_message_header *hdr)
  505. {
  506. struct vmbus_channel_gpadl_created *gpadlcreated;
  507. struct vmbus_channel_msginfo *msginfo;
  508. struct vmbus_channel_message_header *requestheader;
  509. struct vmbus_channel_gpadl_header *gpadlheader;
  510. unsigned long flags;
  511. gpadlcreated = (struct vmbus_channel_gpadl_created *)hdr;
  512. /*
  513. * Find the establish msg, copy the result and signal/unblock the wait
  514. * event
  515. */
  516. spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
  517. list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list,
  518. msglistentry) {
  519. requestheader =
  520. (struct vmbus_channel_message_header *)msginfo->msg;
  521. if (requestheader->msgtype == CHANNELMSG_GPADL_HEADER) {
  522. gpadlheader =
  523. (struct vmbus_channel_gpadl_header *)requestheader;
  524. if ((gpadlcreated->child_relid ==
  525. gpadlheader->child_relid) &&
  526. (gpadlcreated->gpadl == gpadlheader->gpadl)) {
  527. memcpy(&msginfo->response.gpadl_created,
  528. gpadlcreated,
  529. sizeof(
  530. struct vmbus_channel_gpadl_created));
  531. complete(&msginfo->waitevent);
  532. break;
  533. }
  534. }
  535. }
  536. spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
  537. }
  538. /*
  539. * vmbus_ongpadl_torndown - GPADL torndown handler.
  540. *
  541. * This is invoked when we received a response to our gpadl teardown request.
  542. * Find the matching request, copy the response and signal the requesting
  543. * thread.
  544. */
  545. static void vmbus_ongpadl_torndown(
  546. struct vmbus_channel_message_header *hdr)
  547. {
  548. struct vmbus_channel_gpadl_torndown *gpadl_torndown;
  549. struct vmbus_channel_msginfo *msginfo;
  550. struct vmbus_channel_message_header *requestheader;
  551. struct vmbus_channel_gpadl_teardown *gpadl_teardown;
  552. unsigned long flags;
  553. gpadl_torndown = (struct vmbus_channel_gpadl_torndown *)hdr;
  554. /*
  555. * Find the open msg, copy the result and signal/unblock the wait event
  556. */
  557. spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
  558. list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list,
  559. msglistentry) {
  560. requestheader =
  561. (struct vmbus_channel_message_header *)msginfo->msg;
  562. if (requestheader->msgtype == CHANNELMSG_GPADL_TEARDOWN) {
  563. gpadl_teardown =
  564. (struct vmbus_channel_gpadl_teardown *)requestheader;
  565. if (gpadl_torndown->gpadl == gpadl_teardown->gpadl) {
  566. memcpy(&msginfo->response.gpadl_torndown,
  567. gpadl_torndown,
  568. sizeof(
  569. struct vmbus_channel_gpadl_torndown));
  570. complete(&msginfo->waitevent);
  571. break;
  572. }
  573. }
  574. }
  575. spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
  576. }
  577. /*
  578. * vmbus_onversion_response - Version response handler
  579. *
  580. * This is invoked when we received a response to our initiate contact request.
  581. * Find the matching request, copy the response and signal the requesting
  582. * thread.
  583. */
  584. static void vmbus_onversion_response(
  585. struct vmbus_channel_message_header *hdr)
  586. {
  587. struct vmbus_channel_msginfo *msginfo;
  588. struct vmbus_channel_message_header *requestheader;
  589. struct vmbus_channel_version_response *version_response;
  590. unsigned long flags;
  591. version_response = (struct vmbus_channel_version_response *)hdr;
  592. spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
  593. list_for_each_entry(msginfo, &vmbus_connection.chn_msg_list,
  594. msglistentry) {
  595. requestheader =
  596. (struct vmbus_channel_message_header *)msginfo->msg;
  597. if (requestheader->msgtype ==
  598. CHANNELMSG_INITIATE_CONTACT) {
  599. memcpy(&msginfo->response.version_response,
  600. version_response,
  601. sizeof(struct vmbus_channel_version_response));
  602. complete(&msginfo->waitevent);
  603. }
  604. }
  605. spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
  606. }
  607. /* Channel message dispatch table */
  608. static struct vmbus_channel_message_table_entry
  609. channel_message_table[CHANNELMSG_COUNT] = {
  610. {CHANNELMSG_INVALID, NULL},
  611. {CHANNELMSG_OFFERCHANNEL, vmbus_onoffer},
  612. {CHANNELMSG_RESCIND_CHANNELOFFER, vmbus_onoffer_rescind},
  613. {CHANNELMSG_REQUESTOFFERS, NULL},
  614. {CHANNELMSG_ALLOFFERS_DELIVERED, vmbus_onoffers_delivered},
  615. {CHANNELMSG_OPENCHANNEL, NULL},
  616. {CHANNELMSG_OPENCHANNEL_RESULT, vmbus_onopen_result},
  617. {CHANNELMSG_CLOSECHANNEL, NULL},
  618. {CHANNELMSG_GPADL_HEADER, NULL},
  619. {CHANNELMSG_GPADL_BODY, NULL},
  620. {CHANNELMSG_GPADL_CREATED, vmbus_ongpadl_created},
  621. {CHANNELMSG_GPADL_TEARDOWN, NULL},
  622. {CHANNELMSG_GPADL_TORNDOWN, vmbus_ongpadl_torndown},
  623. {CHANNELMSG_RELID_RELEASED, NULL},
  624. {CHANNELMSG_INITIATE_CONTACT, NULL},
  625. {CHANNELMSG_VERSION_RESPONSE, vmbus_onversion_response},
  626. {CHANNELMSG_UNLOAD, NULL},
  627. };
  628. /*
  629. * vmbus_onmessage - Handler for channel protocol messages.
  630. *
  631. * This is invoked in the vmbus worker thread context.
  632. */
  633. void vmbus_onmessage(void *context)
  634. {
  635. struct hv_message *msg = context;
  636. struct vmbus_channel_message_header *hdr;
  637. int size;
  638. hdr = (struct vmbus_channel_message_header *)msg->u.payload;
  639. size = msg->header.payload_size;
  640. if (hdr->msgtype >= CHANNELMSG_COUNT) {
  641. pr_err("Received invalid channel message type %d size %d\n",
  642. hdr->msgtype, size);
  643. print_hex_dump_bytes("", DUMP_PREFIX_NONE,
  644. (unsigned char *)msg->u.payload, size);
  645. return;
  646. }
  647. if (channel_message_table[hdr->msgtype].message_handler)
  648. channel_message_table[hdr->msgtype].message_handler(hdr);
  649. else
  650. pr_err("Unhandled channel message type %d\n", hdr->msgtype);
  651. }
  652. /*
  653. * vmbus_request_offers - Send a request to get all our pending offers.
  654. */
  655. int vmbus_request_offers(void)
  656. {
  657. struct vmbus_channel_message_header *msg;
  658. struct vmbus_channel_msginfo *msginfo;
  659. int ret, t;
  660. msginfo = kmalloc(sizeof(*msginfo) +
  661. sizeof(struct vmbus_channel_message_header),
  662. GFP_KERNEL);
  663. if (!msginfo)
  664. return -ENOMEM;
  665. init_completion(&msginfo->waitevent);
  666. msg = (struct vmbus_channel_message_header *)msginfo->msg;
  667. msg->msgtype = CHANNELMSG_REQUESTOFFERS;
  668. ret = vmbus_post_msg(msg,
  669. sizeof(struct vmbus_channel_message_header));
  670. if (ret != 0) {
  671. pr_err("Unable to request offers - %d\n", ret);
  672. goto cleanup;
  673. }
  674. t = wait_for_completion_timeout(&msginfo->waitevent, 5*HZ);
  675. if (t == 0) {
  676. ret = -ETIMEDOUT;
  677. goto cleanup;
  678. }
  679. cleanup:
  680. kfree(msginfo);
  681. return ret;
  682. }
  683. /*
  684. * Retrieve the (sub) channel on which to send an outgoing request.
  685. * When a primary channel has multiple sub-channels, we choose a
  686. * channel whose VCPU binding is closest to the VCPU on which
  687. * this call is being made.
  688. */
  689. struct vmbus_channel *vmbus_get_outgoing_channel(struct vmbus_channel *primary)
  690. {
  691. struct list_head *cur, *tmp;
  692. int cur_cpu = hv_context.vp_index[smp_processor_id()];
  693. struct vmbus_channel *cur_channel;
  694. struct vmbus_channel *outgoing_channel = primary;
  695. int cpu_distance, new_cpu_distance;
  696. if (list_empty(&primary->sc_list))
  697. return outgoing_channel;
  698. list_for_each_safe(cur, tmp, &primary->sc_list) {
  699. cur_channel = list_entry(cur, struct vmbus_channel, sc_list);
  700. if (cur_channel->state != CHANNEL_OPENED_STATE)
  701. continue;
  702. if (cur_channel->target_vp == cur_cpu)
  703. return cur_channel;
  704. cpu_distance = ((outgoing_channel->target_vp > cur_cpu) ?
  705. (outgoing_channel->target_vp - cur_cpu) :
  706. (cur_cpu - outgoing_channel->target_vp));
  707. new_cpu_distance = ((cur_channel->target_vp > cur_cpu) ?
  708. (cur_channel->target_vp - cur_cpu) :
  709. (cur_cpu - cur_channel->target_vp));
  710. if (cpu_distance < new_cpu_distance)
  711. continue;
  712. outgoing_channel = cur_channel;
  713. }
  714. return outgoing_channel;
  715. }
  716. EXPORT_SYMBOL_GPL(vmbus_get_outgoing_channel);
  717. static void invoke_sc_cb(struct vmbus_channel *primary_channel)
  718. {
  719. struct list_head *cur, *tmp;
  720. struct vmbus_channel *cur_channel;
  721. if (primary_channel->sc_creation_callback == NULL)
  722. return;
  723. list_for_each_safe(cur, tmp, &primary_channel->sc_list) {
  724. cur_channel = list_entry(cur, struct vmbus_channel, sc_list);
  725. primary_channel->sc_creation_callback(cur_channel);
  726. }
  727. }
  728. void vmbus_set_sc_create_callback(struct vmbus_channel *primary_channel,
  729. void (*sc_cr_cb)(struct vmbus_channel *new_sc))
  730. {
  731. primary_channel->sc_creation_callback = sc_cr_cb;
  732. }
  733. EXPORT_SYMBOL_GPL(vmbus_set_sc_create_callback);
  734. bool vmbus_are_subchannels_present(struct vmbus_channel *primary)
  735. {
  736. bool ret;
  737. ret = !list_empty(&primary->sc_list);
  738. if (ret) {
  739. /*
  740. * Invoke the callback on sub-channel creation.
  741. * This will present a uniform interface to the
  742. * clients.
  743. */
  744. invoke_sc_cb(primary);
  745. }
  746. return ret;
  747. }
  748. EXPORT_SYMBOL_GPL(vmbus_are_subchannels_present);