channel_mgmt.c 23 KB

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