drivers/staging/hv/channel.c

   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 #include <linux/kernel.h>
  22 #include <linux/mm.h>
  23 #include <linux/slab.h>
  24 #include <linux/module.h>
  25 #include "osd.h"
  26 #include "logging.h"
  27 #include "vmbus_private.h"
  28
  29 /* Internal routines */
  30 static int create_gpadl_header(
  31         void *kbuffer,  /* must be phys and virt contiguous */
  32         u32 size,       /* page-size multiple */
  33         struct vmbus_channel_msginfo **msginfo,
  34         u32 *messagecount);
  35 static void dump_vmbus_channel(struct vmbus_channel *channel);
  36 static void vmbus_setevent(struct vmbus_channel *channel);
  37
  38
  39 #if 0
  40 static void DumpMonitorPage(struct hv_monitor_page *MonitorPage)
  41 {
  42         int i = 0;
  43         int j = 0;
  44
  45         DPRINT_DBG(VMBUS, "monitorPage - %p, trigger state - %d",
  46                    MonitorPage, MonitorPage->TriggerState);
  47
  48         for (i = 0; i < 4; i++)
  49                 DPRINT_DBG(VMBUS, "trigger group (%d) - %llx", i,
  50                            MonitorPage->TriggerGroup[i].AsUINT64);
  51
  52         for (i = 0; i < 4; i++) {
  53                 for (j = 0; j < 32; j++) {
  54                         DPRINT_DBG(VMBUS, "latency (%d)(%d) - %llx", i, j,
  55                                    MonitorPage->Latency[i][j]);
  56                 }
  57         }
  58         for (i = 0; i < 4; i++) {
  59                 for (j = 0; j < 32; j++) {
  60                         DPRINT_DBG(VMBUS, "param-conn id (%d)(%d) - %d", i, j,
  61                                MonitorPage->Parameter[i][j].ConnectionId.Asu32);
  62                         DPRINT_DBG(VMBUS, "param-flag (%d)(%d) - %d", i, j,
  63                                 MonitorPage->Parameter[i][j].FlagNumber);
  64                 }
  65         }
  66 }
  67 #endif
  68
  69 /*
  70  * vmbus_setevent- Trigger an event notification on the specified
  71  * channel.
  72  */
  73 static void vmbus_setevent(struct vmbus_channel *channel)
  74 {
  75         struct hv_monitor_page *monitorpage;
  76
  77         if (channel->OfferMsg.MonitorAllocated) {
  78                 /* Each u32 represents 32 channels */
  79                 set_bit(channel->OfferMsg.ChildRelId & 31,
  80                         (unsigned long *) gVmbusConnection.SendInterruptPage +
  81                         (channel->OfferMsg.ChildRelId >> 5));
  82
  83                 monitorpage = gVmbusConnection.MonitorPages;
  84                 monitorpage++; /* Get the child to parent monitor page */
  85
  86                 set_bit(channel->MonitorBit,
  87                         (unsigned long *)&monitorpage->TriggerGroup
  88                                         [channel->MonitorGroup].Pending);
  89
  90         } else {
  91                 VmbusSetEvent(channel->OfferMsg.ChildRelId);
  92         }
  93 }
  94
  95 #if 0
  96 static void VmbusChannelClearEvent(struct vmbus_channel *channel)
  97 {
  98         struct hv_monitor_page *monitorPage;
  99
 100         if (Channel->OfferMsg.MonitorAllocated) {
 101                 /* Each u32 represents 32 channels */
 102                 clear_bit(Channel->OfferMsg.ChildRelId & 31,
 103                           (unsigned long *)gVmbusConnection.SendInterruptPage +
 104                           (Channel->OfferMsg.ChildRelId >> 5));
 105
 106                 monitorPage =
 107                         (struct hv_monitor_page *)gVmbusConnection.MonitorPages;
 108                 monitorPage++; /* Get the child to parent monitor page */
 109
 110                 clear_bit(Channel->MonitorBit,
 111                           (unsigned long *)&monitorPage->TriggerGroup
 112                                         [Channel->MonitorGroup].Pending);
 113         }
 114 }
 115
 116 #endif
 117 /*
 118  * vmbus_get_debug_info -Retrieve various channel debug info
 119  */
 120 void vmbus_get_debug_info(struct vmbus_channel *channel,
 121                               struct vmbus_channel_debug_info *debuginfo)
 122 {
 123         struct hv_monitor_page *monitorpage;
 124         u8 monitor_group = (u8)channel->OfferMsg.MonitorId / 32;
 125         u8 monitor_offset = (u8)channel->OfferMsg.MonitorId % 32;
 126         /* u32 monitorBit       = 1 << monitorOffset; */
 127
 128         debuginfo->RelId = channel->OfferMsg.ChildRelId;
 129         debuginfo->State = channel->State;
 130         memcpy(&debuginfo->InterfaceType,
 131                &channel->OfferMsg.Offer.InterfaceType, sizeof(struct hv_guid));
 132         memcpy(&debuginfo->InterfaceInstance,
 133                &channel->OfferMsg.Offer.InterfaceInstance,
 134                sizeof(struct hv_guid));
 135
 136         monitorpage = (struct hv_monitor_page *)gVmbusConnection.MonitorPages;
 137
 138         debuginfo->MonitorId = channel->OfferMsg.MonitorId;
 139
 140         debuginfo->ServerMonitorPending =
 141                         monitorpage->TriggerGroup[monitor_group].Pending;
 142         debuginfo->ServerMonitorLatency =
 143                         monitorpage->Latency[monitor_group][monitor_offset];
 144         debuginfo->ServerMonitorConnectionId =
 145                         monitorpage->Parameter[monitor_group]
 146                                         [monitor_offset].ConnectionId.u.Id;
 147
 148         monitorpage++;
 149
 150         debuginfo->ClientMonitorPending =
 151                         monitorpage->TriggerGroup[monitor_group].Pending;
 152         debuginfo->ClientMonitorLatency =
 153                         monitorpage->Latency[monitor_group][monitor_offset];
 154         debuginfo->ClientMonitorConnectionId =
 155                         monitorpage->Parameter[monitor_group]
 156                                         [monitor_offset].ConnectionId.u.Id;
 157
 158         RingBufferGetDebugInfo(&channel->Inbound, &debuginfo->Inbound);
 159         RingBufferGetDebugInfo(&channel->Outbound, &debuginfo->Outbound);
 160 }
 161
 162 /*
 163  * vmbus_open - Open the specified channel.
 164  */
 165 int vmbus_open(struct vmbus_channel *newchannel, u32 send_ringbuffer_size,
 166                      u32 recv_ringbuffer_size, void *userdata, u32 userdatalen,
 167                      void (*onchannelcallback)(void *context), void *context)
 168 {
 169         struct vmbus_channel_open_channel *openMsg;
 170         struct vmbus_channel_msginfo *openInfo = NULL;
 171         void *in, *out;
 172         unsigned long flags;
 173         int ret, err = 0;
 174
 175         /* Aligned to page size */
 176         /* ASSERT(!(SendRingBufferSize & (PAGE_SIZE - 1))); */
 177         /* ASSERT(!(RecvRingBufferSize & (PAGE_SIZE - 1))); */
 178
 179         newchannel->OnChannelCallback = onchannelcallback;
 180         newchannel->ChannelCallbackContext = context;
 181
 182         /* Allocate the ring buffer */
 183         out = osd_PageAlloc((send_ringbuffer_size + recv_ringbuffer_size)
 184                              >> PAGE_SHIFT);
 185         if (!out)
 186                 return -ENOMEM;
 187
 188         /* ASSERT(((unsigned long)out & (PAGE_SIZE-1)) == 0); */
 189
 190         in = (void *)((unsigned long)out + send_ringbuffer_size);
 191
 192         newchannel->RingBufferPages = out;
 193         newchannel->RingBufferPageCount = (send_ringbuffer_size +
 194                                            recv_ringbuffer_size) >> PAGE_SHIFT;
 195
 196         ret = RingBufferInit(&newchannel->Outbound, out, send_ringbuffer_size);
 197         if (ret != 0) {
 198                 err = ret;
 199                 goto errorout;
 200         }
 201
 202         ret = RingBufferInit(&newchannel->Inbound, in, recv_ringbuffer_size);
 203         if (ret != 0) {
 204                 err = ret;
 205                 goto errorout;
 206         }
 207
 208
 209         /* Establish the gpadl for the ring buffer */
 210         DPRINT_DBG(VMBUS, "Establishing ring buffer's gpadl for channel %p...",
 211                    newchannel);
 212
 213         newchannel->RingBufferGpadlHandle = 0;
 214
 215         ret = vmbus_establish_gpadl(newchannel,
 216                                          newchannel->Outbound.RingBuffer,
 217                                          send_ringbuffer_size +
 218                                          recv_ringbuffer_size,
 219                                          &newchannel->RingBufferGpadlHandle);
 220
 221         if (ret != 0) {
 222                 err = ret;
 223                 goto errorout;
 224         }
 225
 226         DPRINT_DBG(VMBUS, "channel %p <relid %d gpadl 0x%x send ring %p "
 227                    "size %d recv ring %p size %d, downstreamoffset %d>",
 228                    newchannel, newchannel->OfferMsg.ChildRelId,
 229                    newchannel->RingBufferGpadlHandle,
 230                    newchannel->Outbound.RingBuffer,
 231                    newchannel->Outbound.RingSize,
 232                    newchannel->Inbound.RingBuffer,
 233                    newchannel->Inbound.RingSize,
 234                    send_ringbuffer_size);
 235
 236         /* Create and init the channel open message */
 237         openInfo = kmalloc(sizeof(*openInfo) +
 238                            sizeof(struct vmbus_channel_open_channel),
 239                            GFP_KERNEL);
 240         if (!openInfo) {
 241                 err = -ENOMEM;
 242                 goto errorout;
 243         }
 244
 245         openInfo->WaitEvent = osd_WaitEventCreate();
 246         if (!openInfo->WaitEvent) {
 247                 err = -ENOMEM;
 248                 goto errorout;
 249         }
 250
 251         openMsg = (struct vmbus_channel_open_channel *)openInfo->Msg;
 252         openMsg->Header.MessageType = ChannelMessageOpenChannel;
 253         openMsg->OpenId = newchannel->OfferMsg.ChildRelId; /* FIXME */
 254         openMsg->ChildRelId = newchannel->OfferMsg.ChildRelId;
 255         openMsg->RingBufferGpadlHandle = newchannel->RingBufferGpadlHandle;
 256         openMsg->DownstreamRingBufferPageOffset = send_ringbuffer_size >>
 257                                                   PAGE_SHIFT;
 258         openMsg->ServerContextAreaGpadlHandle = 0; /* TODO */
 259
 260         if (userdatalen > MAX_USER_DEFINED_BYTES) {
 261                 err = -EINVAL;
 262                 goto errorout;
 263         }
 264
 265         if (userdatalen)
 266                 memcpy(openMsg->UserData, userdata, userdatalen);
 267
 268         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 269         list_add_tail(&openInfo->MsgListEntry,
 270                       &gVmbusConnection.ChannelMsgList);
 271         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 272
 273         DPRINT_DBG(VMBUS, "Sending channel open msg...");
 274
 275         ret = VmbusPostMessage(openMsg,
 276                                sizeof(struct vmbus_channel_open_channel));
 277         if (ret != 0) {
 278                 DPRINT_ERR(VMBUS, "unable to open channel - %d", ret);
 279                 goto Cleanup;
 280         }
 281
 282         /* FIXME: Need to time-out here */
 283         osd_WaitEventWait(openInfo->WaitEvent);
 284
 285         if (openInfo->Response.OpenResult.Status == 0)
 286                 DPRINT_INFO(VMBUS, "channel <%p> open success!!", newchannel);
 287         else
 288                 DPRINT_INFO(VMBUS, "channel <%p> open failed - %d!!",
 289                             newchannel, openInfo->Response.OpenResult.Status);
 290
 291 Cleanup:
 292         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 293         list_del(&openInfo->MsgListEntry);
 294         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 295
 296         kfree(openInfo->WaitEvent);
 297         kfree(openInfo);
 298         return 0;
 299
 300 errorout:
 301         RingBufferCleanup(&newchannel->Outbound);
 302         RingBufferCleanup(&newchannel->Inbound);
 303         osd_PageFree(out, (send_ringbuffer_size + recv_ringbuffer_size)
 304                      >> PAGE_SHIFT);
 305         kfree(openInfo);
 306         return err;
 307 }
 308
 309 /*
 310  * dump_gpadl_body - Dump the gpadl body message to the console for
 311  * debugging purposes.
 312  */
 313 static void dump_gpadl_body(struct vmbus_channel_gpadl_body *gpadl, u32 len)
 314 {
 315         int i;
 316         int pfncount;
 317
 318         pfncount = (len - sizeof(struct vmbus_channel_gpadl_body)) /
 319                    sizeof(u64);
 320         DPRINT_DBG(VMBUS, "gpadl body - len %d pfn count %d", len, pfncount);
 321
 322         for (i = 0; i < pfncount; i++)
 323                 DPRINT_DBG(VMBUS, "gpadl body  - %d) pfn %llu",
 324                            i, gpadl->Pfn[i]);
 325 }
 326
 327 /*
 328  * dump_gpadl_header - Dump the gpadl header message to the console for
 329  * debugging purposes.
 330  */
 331 static void dump_gpadl_header(struct vmbus_channel_gpadl_header *gpadl)
 332 {
 333         int i, j;
 334         int pagecount;
 335
 336         DPRINT_DBG(VMBUS,
 337                    "gpadl header - relid %d, range count %d, range buflen %d",
 338                    gpadl->ChildRelId, gpadl->RangeCount, gpadl->RangeBufLen);
 339         for (i = 0; i < gpadl->RangeCount; i++) {
 340                 pagecount = gpadl->Range[i].ByteCount >> PAGE_SHIFT;
 341                 pagecount = (pagecount > 26) ? 26 : pagecount;
 342
 343                 DPRINT_DBG(VMBUS, "gpadl range %d - len %d offset %d "
 344                            "page count %d", i, gpadl->Range[i].ByteCount,
 345                            gpadl->Range[i].ByteOffset, pagecount);
 346
 347                 for (j = 0; j < pagecount; j++)
 348                         DPRINT_DBG(VMBUS, "%d) pfn %llu", j,
 349                                    gpadl->Range[i].PfnArray[j]);
 350         }
 351 }
 352
 353 /*
 354  * create_gpadl_header - Creates a gpadl for the specified buffer
 355  */
 356 static int create_gpadl_header(void *kbuffer, u32 size,
 357                                          struct vmbus_channel_msginfo **msginfo,
 358                                          u32 *messagecount)
 359 {
 360         int i;
 361         int pagecount;
 362         unsigned long long pfn;
 363         struct vmbus_channel_gpadl_header *gpadl_header;
 364         struct vmbus_channel_gpadl_body *gpadl_body;
 365         struct vmbus_channel_msginfo *msgheader;
 366         struct vmbus_channel_msginfo *msgbody = NULL;
 367         u32 msgsize;
 368
 369         int pfnsum, pfncount, pfnleft, pfncurr, pfnsize;
 370
 371         /* ASSERT((kbuffer & (PAGE_SIZE-1)) == 0); */
 372         /* ASSERT((Size & (PAGE_SIZE-1)) == 0); */
 373
 374         pagecount = size >> PAGE_SHIFT;
 375         pfn = virt_to_phys(kbuffer) >> PAGE_SHIFT;
 376
 377         /* do we need a gpadl body msg */
 378         pfnsize = MAX_SIZE_CHANNEL_MESSAGE -
 379                   sizeof(struct vmbus_channel_gpadl_header) -
 380                   sizeof(struct gpa_range);
 381         pfncount = pfnsize / sizeof(u64);
 382
 383         if (pagecount > pfncount) {
 384                 /* we need a gpadl body */
 385                 /* fill in the header */
 386                 msgsize = sizeof(struct vmbus_channel_msginfo) +
 387                           sizeof(struct vmbus_channel_gpadl_header) +
 388                           sizeof(struct gpa_range) + pfncount * sizeof(u64);
 389                 msgheader =  kzalloc(msgsize, GFP_KERNEL);
 390                 if (!msgheader)
 391                         goto nomem;
 392
 393                 INIT_LIST_HEAD(&msgheader->SubMsgList);
 394                 msgheader->MessageSize = msgsize;
 395
 396                 gpadl_header = (struct vmbus_channel_gpadl_header *)
 397                         msgheader->Msg;
 398                 gpadl_header->RangeCount = 1;
 399                 gpadl_header->RangeBufLen = sizeof(struct gpa_range) +
 400                                          pagecount * sizeof(u64);
 401                 gpadl_header->Range[0].ByteOffset = 0;
 402                 gpadl_header->Range[0].ByteCount = size;
 403                 for (i = 0; i < pfncount; i++)
 404                         gpadl_header->Range[0].PfnArray[i] = pfn+i;
 405                 *msginfo = msgheader;
 406                 *messagecount = 1;
 407
 408                 pfnsum = pfncount;
 409                 pfnleft = pagecount - pfncount;
 410
 411                 /* how many pfns can we fit */
 412                 pfnsize = MAX_SIZE_CHANNEL_MESSAGE -
 413                           sizeof(struct vmbus_channel_gpadl_body);
 414                 pfncount = pfnsize / sizeof(u64);
 415
 416                 /* fill in the body */
 417                 while (pfnleft) {
 418                         if (pfnleft > pfncount)
 419                                 pfncurr = pfncount;
 420                         else
 421                                 pfncurr = pfnleft;
 422
 423                         msgsize = sizeof(struct vmbus_channel_msginfo) +
 424                                   sizeof(struct vmbus_channel_gpadl_body) +
 425                                   pfncurr * sizeof(u64);
 426                         msgbody = kzalloc(msgsize, GFP_KERNEL);
 427                         /* FIXME: we probably need to more if this fails */
 428                         if (!msgbody)
 429                                 goto nomem;
 430                         msgbody->MessageSize = msgsize;
 431                         (*messagecount)++;
 432                         gpadl_body =
 433                                 (struct vmbus_channel_gpadl_body *)msgbody->Msg;
 434
 435                         /*
 436                          * FIXME:
 437                          * Gpadl is u32 and we are using a pointer which could
 438                          * be 64-bit
 439                          */
 440                         /* gpadl_body->Gpadl = kbuffer; */
 441                         for (i = 0; i < pfncurr; i++)
 442                                 gpadl_body->Pfn[i] = pfn + pfnsum + i;
 443
 444                         /* add to msg header */
 445                         list_add_tail(&msgbody->MsgListEntry,
 446                                       &msgheader->SubMsgList);
 447                         pfnsum += pfncurr;
 448                         pfnleft -= pfncurr;
 449                 }
 450         } else {
 451                 /* everything fits in a header */
 452                 msgsize = sizeof(struct vmbus_channel_msginfo) +
 453                           sizeof(struct vmbus_channel_gpadl_header) +
 454                           sizeof(struct gpa_range) + pagecount * sizeof(u64);
 455                 msgheader = kzalloc(msgsize, GFP_KERNEL);
 456                 if (msgheader == NULL)
 457                         goto nomem;
 458                 msgheader->MessageSize = msgsize;
 459
 460                 gpadl_header = (struct vmbus_channel_gpadl_header *)
 461                         msgheader->Msg;
 462                 gpadl_header->RangeCount = 1;
 463                 gpadl_header->RangeBufLen = sizeof(struct gpa_range) +
 464                                          pagecount * sizeof(u64);
 465                 gpadl_header->Range[0].ByteOffset = 0;
 466                 gpadl_header->Range[0].ByteCount = size;
 467                 for (i = 0; i < pagecount; i++)
 468                         gpadl_header->Range[0].PfnArray[i] = pfn+i;
 469
 470                 *msginfo = msgheader;
 471                 *messagecount = 1;
 472         }
 473
 474         return 0;
 475 nomem:
 476         kfree(msgheader);
 477         kfree(msgbody);
 478         return -ENOMEM;
 479 }
 480
 481 /*
 482  * vmbus_establish_gpadl - Estabish a GPADL for the specified buffer
 483  *
 484  * @channel: a channel
 485  * @kbuffer: from kmalloc
 486  * @size: page-size multiple
 487  * @gpadl_handle: some funky thing
 488  */
 489 int vmbus_establish_gpadl(struct vmbus_channel *channel, void *kbuffer,
 490                                u32 size, u32 *gpadl_handle)
 491 {
 492         struct vmbus_channel_gpadl_header *gpadlmsg;
 493         struct vmbus_channel_gpadl_body *gpadl_body;
 494         /* struct vmbus_channel_gpadl_created *gpadlCreated; */
 495         struct vmbus_channel_msginfo *msginfo = NULL;
 496         struct vmbus_channel_msginfo *submsginfo;
 497         u32 msgcount;
 498         struct list_head *curr;
 499         u32 next_gpadl_handle;
 500         unsigned long flags;
 501         int ret = 0;
 502
 503         next_gpadl_handle = atomic_read(&gVmbusConnection.NextGpadlHandle);
 504         atomic_inc(&gVmbusConnection.NextGpadlHandle);
 505
 506         ret = create_gpadl_header(kbuffer, size, &msginfo, &msgcount);
 507         if (ret)
 508                 return ret;
 509
 510         msginfo->WaitEvent = osd_WaitEventCreate();
 511         if (!msginfo->WaitEvent) {
 512                 ret = -ENOMEM;
 513                 goto Cleanup;
 514         }
 515
 516         gpadlmsg = (struct vmbus_channel_gpadl_header *)msginfo->Msg;
 517         gpadlmsg->Header.MessageType = ChannelMessageGpadlHeader;
 518         gpadlmsg->ChildRelId = channel->OfferMsg.ChildRelId;
 519         gpadlmsg->Gpadl = next_gpadl_handle;
 520
 521         dump_gpadl_header(gpadlmsg);
 522
 523         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 524         list_add_tail(&msginfo->MsgListEntry,
 525                       &gVmbusConnection.ChannelMsgList);
 526
 527         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 528         DPRINT_DBG(VMBUS, "buffer %p, size %d msg cnt %d",
 529                    kbuffer, size, msgcount);
 530
 531         DPRINT_DBG(VMBUS, "Sending GPADL Header - len %zd",
 532                    msginfo->MessageSize - sizeof(*msginfo));
 533
 534         ret = VmbusPostMessage(gpadlmsg, msginfo->MessageSize -
 535                                sizeof(*msginfo));
 536         if (ret != 0) {
 537                 DPRINT_ERR(VMBUS, "Unable to open channel - %d", ret);
 538                 goto Cleanup;
 539         }
 540
 541         if (msgcount > 1) {
 542                 list_for_each(curr, &msginfo->SubMsgList) {
 543
 544                         /* FIXME: should this use list_entry() instead ? */
 545                         submsginfo = (struct vmbus_channel_msginfo *)curr;
 546                         gpadl_body =
 547                              (struct vmbus_channel_gpadl_body *)submsginfo->Msg;
 548
 549                         gpadl_body->Header.MessageType =
 550                                 ChannelMessageGpadlBody;
 551                         gpadl_body->Gpadl = next_gpadl_handle;
 552
 553                         DPRINT_DBG(VMBUS, "Sending GPADL Body - len %zd",
 554                                    submsginfo->MessageSize -
 555                                    sizeof(*submsginfo));
 556
 557                         dump_gpadl_body(gpadl_body, submsginfo->MessageSize -
 558                                       sizeof(*submsginfo));
 559                         ret = VmbusPostMessage(gpadl_body,
 560                                                submsginfo->MessageSize -
 561                                                sizeof(*submsginfo));
 562                         if (ret != 0)
 563                                 goto Cleanup;
 564
 565                 }
 566         }
 567         osd_WaitEventWait(msginfo->WaitEvent);
 568
 569         /* At this point, we received the gpadl created msg */
 570         DPRINT_DBG(VMBUS, "Received GPADL created "
 571                    "(relid %d, status %d handle %x)",
 572                    channel->OfferMsg.ChildRelId,
 573                    msginfo->Response.GpadlCreated.CreationStatus,
 574                    gpadlmsg->Gpadl);
 575
 576         *gpadl_handle = gpadlmsg->Gpadl;
 577
 578 Cleanup:
 579         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 580         list_del(&msginfo->MsgListEntry);
 581         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 582
 583         kfree(msginfo->WaitEvent);
 584         kfree(msginfo);
 585         return ret;
 586 }
 587
 588 /*
 589  * vmbus_teardown_gpadl -Teardown the specified GPADL handle
 590  */
 591 int vmbus_teardown_gpadl(struct vmbus_channel *channel, u32 gpadl_handle)
 592 {
 593         struct vmbus_channel_gpadl_teardown *msg;
 594         struct vmbus_channel_msginfo *info;
 595         unsigned long flags;
 596         int ret;
 597
 598         /* ASSERT(gpadl_handle != 0); */
 599
 600         info = kmalloc(sizeof(*info) +
 601                        sizeof(struct vmbus_channel_gpadl_teardown), GFP_KERNEL);
 602         if (!info)
 603                 return -ENOMEM;
 604
 605         info->WaitEvent = osd_WaitEventCreate();
 606         if (!info->WaitEvent) {
 607                 kfree(info);
 608                 return -ENOMEM;
 609         }
 610
 611         msg = (struct vmbus_channel_gpadl_teardown *)info->Msg;
 612
 613         msg->Header.MessageType = ChannelMessageGpadlTeardown;
 614         msg->ChildRelId = channel->OfferMsg.ChildRelId;
 615         msg->Gpadl = gpadl_handle;
 616
 617         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 618         list_add_tail(&info->MsgListEntry,
 619                       &gVmbusConnection.ChannelMsgList);
 620         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 621
 622         ret = VmbusPostMessage(msg,
 623                                sizeof(struct vmbus_channel_gpadl_teardown));
 624         if (ret != 0) {
 625                 /* TODO: */
 626                 /* something... */
 627         }
 628
 629         osd_WaitEventWait(info->WaitEvent);
 630
 631         /* Received a torndown response */
 632         spin_lock_irqsave(&gVmbusConnection.channelmsg_lock, flags);
 633         list_del(&info->MsgListEntry);
 634         spin_unlock_irqrestore(&gVmbusConnection.channelmsg_lock, flags);
 635
 636         kfree(info->WaitEvent);
 637         kfree(info);
 638         return ret;
 639 }
 640 EXPORT_SYMBOL_GPL(vmbus_teardown_gpadl);
 641
 642 /*
 643  * vmbus_close - Close the specified channel
 644  */
 645 void vmbus_close(struct vmbus_channel *channel)
 646 {
 647         struct vmbus_channel_close_channel *msg;
 648         struct vmbus_channel_msginfo *info;
 649         unsigned long flags;
 650         int ret;
 651
 652         /* Stop callback and cancel the timer asap */
 653         channel->OnChannelCallback = NULL;
 654         del_timer_sync(&channel->poll_timer);
 655
 656         /* Send a closing message */
 657         info = kmalloc(sizeof(*info) +
 658                        sizeof(struct vmbus_channel_close_channel), GFP_KERNEL);
 659         /* FIXME: can't do anything other than return here because the
 660          *        function is void */
 661         if (!info)
 662                 return;
 663
 664         /* info->waitEvent = osd_WaitEventCreate(); */
 665
 666         msg = (struct vmbus_channel_close_channel *)info->Msg;
 667         msg->Header.MessageType = ChannelMessageCloseChannel;
 668         msg->ChildRelId = channel->OfferMsg.ChildRelId;
 669
 670         ret = VmbusPostMessage(msg, sizeof(struct vmbus_channel_close_channel));
 671         if (ret != 0) {
 672                 /* TODO: */
 673                 /* something... */
 674         }
 675
 676         /* Tear down the gpadl for the channel's ring buffer */
 677         if (channel->RingBufferGpadlHandle)
 678                 vmbus_teardown_gpadl(channel,
 679                                           channel->RingBufferGpadlHandle);
 680
 681         /* TODO: Send a msg to release the childRelId */
 682
 683         /* Cleanup the ring buffers for this channel */
 684         RingBufferCleanup(&channel->Outbound);
 685         RingBufferCleanup(&channel->Inbound);
 686
 687         osd_PageFree(channel->RingBufferPages, channel->RingBufferPageCount);
 688
 689         kfree(info);
 690
 691         /*
 692          * If we are closing the channel during an error path in
 693          * opening the channel, don't free the channel since the
 694          * caller will free the channel
 695          */
 696
 697         if (channel->State == CHANNEL_OPEN_STATE) {
 698                 spin_lock_irqsave(&gVmbusConnection.channel_lock, flags);
 699                 list_del(&channel->ListEntry);
 700                 spin_unlock_irqrestore(&gVmbusConnection.channel_lock, flags);
 701
 702                 free_channel(channel);
 703         }
 704 }
 705
 706 /**
 707  * vmbus_sendpacket() - Send the specified buffer on the given channel
 708  * @channel: Pointer to vmbus_channel structure.
 709  * @buffer: Pointer to the buffer you want to receive the data into.
 710  * @bufferlen: Maximum size of what the the buffer will hold
 711  * @requestid: Identifier of the request
 712  * @type: Type of packet that is being send e.g. negotiate, time
 713  * packet etc.
 714  *
 715  * Sends data in @buffer directly to hyper-v via the vmbus
 716  * This will send the data unparsed to hyper-v.
 717  *
 718  * Mainly used by Hyper-V drivers.
 719  */
 720 int vmbus_sendpacket(struct vmbus_channel *channel, const void *buffer,
 721                            u32 bufferlen, u64 requestid,
 722                            enum vmbus_packet_type type, u32 flags)
 723 {
 724         struct vmpacket_descriptor desc;
 725         u32 packetlen = sizeof(struct vmpacket_descriptor) + bufferlen;
 726         u32 packetlen_aligned = ALIGN_UP(packetlen, sizeof(u64));
 727         struct scatterlist bufferlist[3];
 728         u64 aligned_data = 0;
 729         int ret;
 730
 731         DPRINT_DBG(VMBUS, "channel %p buffer %p len %d",
 732                    channel, buffer, bufferlen);
 733
 734         dump_vmbus_channel(channel);
 735
 736         /* ASSERT((packetLenAligned - packetLen) < sizeof(u64)); */
 737
 738         /* Setup the descriptor */
 739         desc.Type = type; /* VmbusPacketTypeDataInBand; */
 740         desc.Flags = flags; /* VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED; */
 741         /* in 8-bytes granularity */
 742         desc.DataOffset8 = sizeof(struct vmpacket_descriptor) >> 3;
 743         desc.Length8 = (u16)(packetlen_aligned >> 3);
 744         desc.TransactionId = requestid;
 745
 746         sg_init_table(bufferlist, 3);
 747         sg_set_buf(&bufferlist[0], &desc, sizeof(struct vmpacket_descriptor));
 748         sg_set_buf(&bufferlist[1], buffer, bufferlen);
 749         sg_set_buf(&bufferlist[2], &aligned_data,
 750                    packetlen_aligned - packetlen);
 751
 752         ret = RingBufferWrite(&channel->Outbound, bufferlist, 3);
 753
 754         /* TODO: We should determine if this is optional */
 755         if (ret == 0 && !GetRingBufferInterruptMask(&channel->Outbound))
 756                 vmbus_setevent(channel);
 757
 758         return ret;
 759 }
 760 EXPORT_SYMBOL(vmbus_sendpacket);
 761
 762 /*
 763  * vmbus_sendpacket_pagebuffer - Send a range of single-page buffer
 764  * packets using a GPADL Direct packet type.
 765  */
 766 int vmbus_sendpacket_pagebuffer(struct vmbus_channel *channel,
 767                                      struct hv_page_buffer pagebuffers[],
 768                                      u32 pagecount, void *buffer, u32 bufferlen,
 769                                      u64 requestid)
 770 {
 771         int ret;
 772         int i;
 773         struct vmbus_channel_packet_page_buffer desc;
 774         u32 descsize;
 775         u32 packetlen;
 776         u32 packetlen_aligned;
 777         struct scatterlist bufferlist[3];
 778         u64 aligned_data = 0;
 779
 780         if (pagecount > MAX_PAGE_BUFFER_COUNT)
 781                 return -EINVAL;
 782
 783         dump_vmbus_channel(channel);
 784
 785         /*
 786          * Adjust the size down since vmbus_channel_packet_page_buffer is the
 787          * largest size we support
 788          */
 789         descsize = sizeof(struct vmbus_channel_packet_page_buffer) -
 790                           ((MAX_PAGE_BUFFER_COUNT - pagecount) *
 791                           sizeof(struct hv_page_buffer));
 792         packetlen = descsize + bufferlen;
 793         packetlen_aligned = ALIGN_UP(packetlen, sizeof(u64));
 794
 795         /* ASSERT((packetLenAligned - packetLen) < sizeof(u64)); */
 796
 797         /* Setup the descriptor */
 798         desc.type = VmbusPacketTypeDataUsingGpaDirect;
 799         desc.flags = VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
 800         desc.dataoffset8 = descsize >> 3; /* in 8-bytes grandularity */
 801         desc.length8 = (u16)(packetlen_aligned >> 3);
 802         desc.transactionid = requestid;
 803         desc.rangecount = pagecount;
 804
 805         for (i = 0; i < pagecount; i++) {
 806                 desc.range[i].Length = pagebuffers[i].Length;
 807                 desc.range[i].Offset = pagebuffers[i].Offset;
 808                 desc.range[i].Pfn        = pagebuffers[i].Pfn;
 809         }
 810
 811         sg_init_table(bufferlist, 3);
 812         sg_set_buf(&bufferlist[0], &desc, descsize);
 813         sg_set_buf(&bufferlist[1], buffer, bufferlen);
 814         sg_set_buf(&bufferlist[2], &aligned_data,
 815                 packetlen_aligned - packetlen);
 816
 817         ret = RingBufferWrite(&channel->Outbound, bufferlist, 3);
 818
 819         /* TODO: We should determine if this is optional */
 820         if (ret == 0 && !GetRingBufferInterruptMask(&channel->Outbound))
 821                 vmbus_setevent(channel);
 822
 823         return ret;
 824 }
 825
 826 /*
 827  * vmbus_sendpacket_multipagebuffer - Send a multi-page buffer packet
 828  * using a GPADL Direct packet type.
 829  */
 830 int vmbus_sendpacket_multipagebuffer(struct vmbus_channel *channel,
 831                                 struct hv_multipage_buffer *multi_pagebuffer,
 832                                 void *buffer, u32 bufferlen, u64 requestid)
 833 {
 834         int ret;
 835         struct vmbus_channel_packet_multipage_buffer desc;
 836         u32 descsize;
 837         u32 packetlen;
 838         u32 packetlen_aligned;
 839         struct scatterlist bufferlist[3];
 840         u64 aligned_data = 0;
 841         u32 pfncount = NUM_PAGES_SPANNED(multi_pagebuffer->Offset,
 842                                          multi_pagebuffer->Length);
 843
 844         dump_vmbus_channel(channel);
 845
 846         DPRINT_DBG(VMBUS, "data buffer - offset %u len %u pfn count %u",
 847                 multi_pagebuffer->Offset,
 848                 multi_pagebuffer->Length, pfncount);
 849
 850         if ((pfncount < 0) || (pfncount > MAX_MULTIPAGE_BUFFER_COUNT))
 851                 return -EINVAL;
 852
 853         /*
 854          * Adjust the size down since vmbus_channel_packet_multipage_buffer is
 855          * the largest size we support
 856          */
 857         descsize = sizeof(struct vmbus_channel_packet_multipage_buffer) -
 858                           ((MAX_MULTIPAGE_BUFFER_COUNT - pfncount) *
 859                           sizeof(u64));
 860         packetlen = descsize + bufferlen;
 861         packetlen_aligned = ALIGN_UP(packetlen, sizeof(u64));
 862
 863         /* ASSERT((packetLenAligned - packetLen) < sizeof(u64)); */
 864
 865         /* Setup the descriptor */
 866         desc.type = VmbusPacketTypeDataUsingGpaDirect;
 867         desc.flags = VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
 868         desc.dataoffset8 = descsize >> 3; /* in 8-bytes grandularity */
 869         desc.length8 = (u16)(packetlen_aligned >> 3);
 870         desc.transactionid = requestid;
 871         desc.rangecount = 1;
 872
 873         desc.range.Length = multi_pagebuffer->Length;
 874         desc.range.Offset = multi_pagebuffer->Offset;
 875
 876         memcpy(desc.range.PfnArray, multi_pagebuffer->PfnArray,
 877                pfncount * sizeof(u64));
 878
 879         sg_init_table(bufferlist, 3);
 880         sg_set_buf(&bufferlist[0], &desc, descsize);
 881         sg_set_buf(&bufferlist[1], buffer, bufferlen);
 882         sg_set_buf(&bufferlist[2], &aligned_data,
 883                 packetlen_aligned - packetlen);
 884
 885         ret = RingBufferWrite(&channel->Outbound, bufferlist, 3);
 886
 887         /* TODO: We should determine if this is optional */
 888         if (ret == 0 && !GetRingBufferInterruptMask(&channel->Outbound))
 889                 vmbus_setevent(channel);
 890
 891         return ret;
 892 }
 893
 894
 895 /**
 896  * vmbus_recvpacket() - Retrieve the user packet on the specified channel
 897  * @channel: Pointer to vmbus_channel structure.
 898  * @buffer: Pointer to the buffer you want to receive the data into.
 899  * @bufferlen: Maximum size of what the the buffer will hold
 900  * @buffer_actual_len: The actual size of the data after it was received
 901  * @requestid: Identifier of the request
 902  *
 903  * Receives directly from the hyper-v vmbus and puts the data it received
 904  * into Buffer. This will receive the data unparsed from hyper-v.
 905  *
 906  * Mainly used by Hyper-V drivers.
 907  */
 908 int vmbus_recvpacket(struct vmbus_channel *channel, void *buffer,
 909                         u32 bufferlen, u32 *buffer_actual_len, u64 *requestid)
 910 {
 911         struct vmpacket_descriptor desc;
 912         u32 packetlen;
 913         u32 userlen;
 914         int ret;
 915         unsigned long flags;
 916
 917         *buffer_actual_len = 0;
 918         *requestid = 0;
 919
 920         spin_lock_irqsave(&channel->inbound_lock, flags);
 921
 922         ret = RingBufferPeek(&channel->Inbound, &desc,
 923                              sizeof(struct vmpacket_descriptor));
 924         if (ret != 0) {
 925                 spin_unlock_irqrestore(&channel->inbound_lock, flags);
 926
 927                 /* DPRINT_DBG(VMBUS, "nothing to read!!"); */
 928                 return 0;
 929         }
 930
 931         /* VmbusChannelClearEvent(Channel); */
 932
 933         packetlen = desc.Length8 << 3;
 934         userlen = packetlen - (desc.DataOffset8 << 3);
 935         /* ASSERT(userLen > 0); */
 936
 937         DPRINT_DBG(VMBUS, "packet received on channel %p relid %d <type %d "
 938                    "flag %d tid %llx pktlen %d datalen %d> ",
 939                    channel, channel->OfferMsg.ChildRelId, desc.Type,
 940                    desc.Flags, desc.TransactionId, packetlen, userlen);
 941
 942         *buffer_actual_len = userlen;
 943
 944         if (userlen > bufferlen) {
 945                 spin_unlock_irqrestore(&channel->inbound_lock, flags);
 946
 947                 DPRINT_ERR(VMBUS, "buffer too small - got %d needs %d",
 948                            bufferlen, userlen);
 949                 return -1;
 950         }
 951
 952         *requestid = desc.TransactionId;
 953
 954         /* Copy over the packet to the user buffer */
 955         ret = RingBufferRead(&channel->Inbound, buffer, userlen,
 956                              (desc.DataOffset8 << 3));
 957
 958         spin_unlock_irqrestore(&channel->inbound_lock, flags);
 959
 960         return 0;
 961 }
 962 EXPORT_SYMBOL(vmbus_recvpacket);
 963
 964 /*
 965  * vmbus_recvpacket_raw - Retrieve the raw packet on the specified channel
 966  */
 967 int vmbus_recvpacket_raw(struct vmbus_channel *channel, void *buffer,
 968                               u32 bufferlen, u32 *buffer_actual_len,
 969                               u64 *requestid)
 970 {
 971         struct vmpacket_descriptor desc;
 972         u32 packetlen;
 973         u32 userlen;
 974         int ret;
 975         unsigned long flags;
 976
 977         *buffer_actual_len = 0;
 978         *requestid = 0;
 979
 980         spin_lock_irqsave(&channel->inbound_lock, flags);
 981
 982         ret = RingBufferPeek(&channel->Inbound, &desc,
 983                              sizeof(struct vmpacket_descriptor));
 984         if (ret != 0) {
 985                 spin_unlock_irqrestore(&channel->inbound_lock, flags);
 986
 987                 /* DPRINT_DBG(VMBUS, "nothing to read!!"); */
 988                 return 0;
 989         }
 990
 991         /* VmbusChannelClearEvent(Channel); */
 992
 993         packetlen = desc.Length8 << 3;
 994         userlen = packetlen - (desc.DataOffset8 << 3);
 995
 996         DPRINT_DBG(VMBUS, "packet received on channel %p relid %d <type %d "
 997                    "flag %d tid %llx pktlen %d datalen %d> ",
 998                    channel, channel->OfferMsg.ChildRelId, desc.Type,
 999                    desc.Flags, desc.TransactionId, packetlen, userlen);
1000
1001         *buffer_actual_len = packetlen;
1002
1003         if (packetlen > bufferlen) {
1004                 spin_unlock_irqrestore(&channel->inbound_lock, flags);
1005
1006                 DPRINT_ERR(VMBUS, "buffer too small - needed %d bytes but "
1007                            "got space for only %d bytes", packetlen, bufferlen);
1008                 return -2;
1009         }
1010
1011         *requestid = desc.TransactionId;
1012
1013         /* Copy over the entire packet to the user buffer */
1014         ret = RingBufferRead(&channel->Inbound, buffer, packetlen, 0);
1015
1016         spin_unlock_irqrestore(&channel->inbound_lock, flags);
1017         return 0;
1018 }
1019
1020 /*
1021  * vmbus_onchannel_event - Channel event callback
1022  */
1023 void vmbus_onchannel_event(struct vmbus_channel *channel)
1024 {
1025         dump_vmbus_channel(channel);
1026         /* ASSERT(Channel->OnChannelCallback); */
1027
1028         channel->OnChannelCallback(channel->ChannelCallbackContext);
1029
1030         mod_timer(&channel->poll_timer, jiffies + usecs_to_jiffies(100));
1031 }
1032
1033 /*
1034  * vmbus_ontimer - Timer event callback
1035  */
1036 void vmbus_ontimer(unsigned long data)
1037 {
1038         struct vmbus_channel *channel = (struct vmbus_channel *)data;
1039
1040         if (channel->OnChannelCallback)
1041                 channel->OnChannelCallback(channel->ChannelCallbackContext);
1042 }
1043
1044 /*
1045  * dump_vmbus_channel- Dump vmbus channel info to the console
1046  */
1047 static void dump_vmbus_channel(struct vmbus_channel *channel)
1048 {
1049         DPRINT_DBG(VMBUS, "Channel (%d)", channel->OfferMsg.ChildRelId);
1050         DumpRingInfo(&channel->Outbound, "Outbound ");
1051         DumpRingInfo(&channel->Inbound, "Inbound ");
1052 }