channel_mgmt.c 22 KB

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