drivers/staging/hv/connection.c

   1 /*
   2  *
   3  * Copyright (c) 2009, Microsoft Corporation.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms and conditions of the GNU General Public License,
   7  * version 2, as published by the Free Software Foundation.
   8  *
   9  * This program is distributed in the hope it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
  12  * more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along with
  15  * this program; if not, write to the Free Software Foundation, Inc., 59 Temple
  16  * Place - Suite 330, Boston, MA 02111-1307 USA.
  17  *
  18  * Authors:
  19  *   Haiyang Zhang <haiyangz@microsoft.com>
  20  *   Hank Janssen  <hjanssen@microsoft.com>
  21  *
  22  */
  23 #include <linux/kernel.h>
  24 #include <linux/sched.h>
  25 #include <linux/wait.h>
  26 #include <linux/mm.h>
  27 #include <linux/slab.h>
  28 #include <linux/vmalloc.h>
  29 #include "osd.h"
  30 #include "logging.h"
  31 #include "vmbus_private.h"
  32
  33
  34 struct vmbus_connection vmbus_connection = {
  35         .conn_state             = DISCONNECTED,
  36         .next_gpadl_handle      = ATOMIC_INIT(0xE1E10),
  37 };
  38
  39 /*
  40  * vmbus_connect - Sends a connect request on the partition service connection
  41  */
  42 int vmbus_connect(void)
  43 {
  44         int ret = 0;
  45         struct vmbus_channel_msginfo *msginfo = NULL;
  46         struct vmbus_channel_initiate_contact *msg;
  47         unsigned long flags;
  48
  49         /* Make sure we are not connecting or connected */
  50         if (vmbus_connection.conn_state != DISCONNECTED)
  51                 return -1;
  52
  53         /* Initialize the vmbus connection */
  54         vmbus_connection.conn_state = CONNECTING;
  55         vmbus_connection.work_queue = create_workqueue("hv_vmbus_con");
  56         if (!vmbus_connection.work_queue) {
  57                 ret = -1;
  58                 goto Cleanup;
  59         }
  60
  61         INIT_LIST_HEAD(&vmbus_connection.chn_msg_list);
  62         spin_lock_init(&vmbus_connection.channelmsg_lock);
  63
  64         INIT_LIST_HEAD(&vmbus_connection.chn_list);
  65         spin_lock_init(&vmbus_connection.channel_lock);
  66
  67         /*
  68          * Setup the vmbus event connection for channel interrupt
  69          * abstraction stuff
  70          */
  71         vmbus_connection.int_page =
  72         (void *)__get_free_pages(GFP_KERNEL|__GFP_ZERO, 0);
  73         if (vmbus_connection.int_page == NULL) {
  74                 ret = -1;
  75                 goto Cleanup;
  76         }
  77
  78         vmbus_connection.recv_int_page = vmbus_connection.int_page;
  79         vmbus_connection.send_int_page =
  80                 (void *)((unsigned long)vmbus_connection.int_page +
  81                         (PAGE_SIZE >> 1));
  82
  83         /*
  84          * Setup the monitor notification facility. The 1st page for
  85          * parent->child and the 2nd page for child->parent
  86          */
  87         vmbus_connection.monitor_pages =
  88         (void *)__get_free_pages((GFP_KERNEL|__GFP_ZERO), 1);
  89         if (vmbus_connection.monitor_pages == NULL) {
  90                 ret = -1;
  91                 goto Cleanup;
  92         }
  93
  94         msginfo = kzalloc(sizeof(*msginfo) +
  95                           sizeof(struct vmbus_channel_initiate_contact),
  96                           GFP_KERNEL);
  97         if (msginfo == NULL) {
  98                 ret = -ENOMEM;
  99                 goto Cleanup;
 100         }
 101
 102         init_waitqueue_head(&msginfo->waitevent);
 103
 104         msg = (struct vmbus_channel_initiate_contact *)msginfo->msg;
 105
 106         msg->header.msgtype = CHANNELMSG_INITIATE_CONTACT;
 107         msg->vmbus_version_requested = VMBUS_REVISION_NUMBER;
 108         msg->interrupt_page = virt_to_phys(vmbus_connection.int_page);
 109         msg->monitor_page1 = virt_to_phys(vmbus_connection.monitor_pages);
 110         msg->monitor_page2 = virt_to_phys(
 111                         (void *)((unsigned long)vmbus_connection.monitor_pages +
 112                                  PAGE_SIZE));
 113
 114         /*
 115          * Add to list before we send the request since we may
 116          * receive the response before returning from this routine
 117          */
 118         spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
 119         list_add_tail(&msginfo->msglistentry,
 120                       &vmbus_connection.chn_msg_list);
 121
 122         spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
 123
 124         DPRINT_DBG(VMBUS, "Vmbus connection - interrupt pfn %llx, "
 125                    "monitor1 pfn %llx,, monitor2 pfn %llx",
 126                    msg->interrupt_page, msg->monitor_page1, msg->monitor_page2);
 127
 128         DPRINT_DBG(VMBUS, "Sending channel initiate msg...");
 129         ret = vmbus_post_msg(msg,
 130                                sizeof(struct vmbus_channel_initiate_contact));
 131         if (ret != 0) {
 132                 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
 133                 list_del(&msginfo->msglistentry);
 134                 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock,
 135                                         flags);
 136                 goto Cleanup;
 137         }
 138
 139         /* Wait for the connection response */
 140         msginfo->wait_condition = 0;
 141         wait_event_timeout(msginfo->waitevent, msginfo->wait_condition,
 142                         msecs_to_jiffies(1000));
 143         if (msginfo->wait_condition == 0) {
 144                 spin_lock_irqsave(&vmbus_connection.channelmsg_lock,
 145                                 flags);
 146                 list_del(&msginfo->msglistentry);
 147                 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock,
 148                                         flags);
 149                 ret = -ETIMEDOUT;
 150                 goto Cleanup;
 151         }
 152
 153         spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
 154         list_del(&msginfo->msglistentry);
 155         spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
 156
 157         /* Check if successful */
 158         if (msginfo->response.version_response.version_supported) {
 159                 DPRINT_INFO(VMBUS, "Vmbus connected!!");
 160                 vmbus_connection.conn_state = CONNECTED;
 161
 162         } else {
 163                 DPRINT_ERR(VMBUS, "Vmbus connection failed!!..."
 164                            "current version (%d) not supported",
 165                            VMBUS_REVISION_NUMBER);
 166                 ret = -1;
 167                 goto Cleanup;
 168         }
 169
 170         kfree(msginfo);
 171         return 0;
 172
 173 Cleanup:
 174         vmbus_connection.conn_state = DISCONNECTED;
 175
 176         if (vmbus_connection.work_queue)
 177                 destroy_workqueue(vmbus_connection.work_queue);
 178
 179         if (vmbus_connection.int_page) {
 180                 free_pages((unsigned long)vmbus_connection.int_page, 0);
 181                 vmbus_connection.int_page = NULL;
 182         }
 183
 184         if (vmbus_connection.monitor_pages) {
 185                 free_pages((unsigned long)vmbus_connection.monitor_pages, 1);
 186                 vmbus_connection.monitor_pages = NULL;
 187         }
 188
 189         if (msginfo) {
 190                 kfree(msginfo);
 191         }
 192
 193         return ret;
 194 }
 195
 196 /*
 197  * vmbus_disconnect -
 198  * Sends a disconnect request on the partition service connection
 199  */
 200 int vmbus_disconnect(void)
 201 {
 202         int ret = 0;
 203         struct vmbus_channel_message_header *msg;
 204
 205         /* Make sure we are connected */
 206         if (vmbus_connection.conn_state != CONNECTED)
 207                 return -1;
 208
 209         msg = kzalloc(sizeof(struct vmbus_channel_message_header), GFP_KERNEL);
 210         if (!msg)
 211                 return -ENOMEM;
 212
 213         msg->msgtype = CHANNELMSG_UNLOAD;
 214
 215         ret = vmbus_post_msg(msg,
 216                                sizeof(struct vmbus_channel_message_header));
 217         if (ret != 0)
 218                 goto Cleanup;
 219
 220         free_pages((unsigned long)vmbus_connection.int_page, 0);
 221         free_pages((unsigned long)vmbus_connection.monitor_pages, 1);
 222
 223         /* TODO: iterate thru the msg list and free up */
 224         destroy_workqueue(vmbus_connection.work_queue);
 225
 226         vmbus_connection.conn_state = DISCONNECTED;
 227
 228         DPRINT_INFO(VMBUS, "Vmbus disconnected!!");
 229
 230 Cleanup:
 231         kfree(msg);
 232         return ret;
 233 }
 234
 235 /*
 236  * relid2channel - Get the channel object given its
 237  * child relative id (ie channel id)
 238  */
 239 struct vmbus_channel *relid2channel(u32 relid)
 240 {
 241         struct vmbus_channel *channel;
 242         struct vmbus_channel *found_channel  = NULL;
 243         unsigned long flags;
 244
 245         spin_lock_irqsave(&vmbus_connection.channel_lock, flags);
 246         list_for_each_entry(channel, &vmbus_connection.chn_list, listentry) {
 247                 if (channel->offermsg.child_relid == relid) {
 248                         found_channel = channel;
 249                         break;
 250                 }
 251         }
 252         spin_unlock_irqrestore(&vmbus_connection.channel_lock, flags);
 253
 254         return found_channel;
 255 }
 256
 257 /*
 258  * process_chn_event - Process a channel event notification
 259  */
 260 static void process_chn_event(void *context)
 261 {
 262         struct vmbus_channel *channel;
 263         u32 relid = (u32)(unsigned long)context;
 264
 265         /* ASSERT(relId > 0); */
 266
 267         /*
 268          * Find the channel based on this relid and invokes the
 269          * channel callback to process the event
 270          */
 271         channel = relid2channel(relid);
 272
 273         if (channel) {
 274                 vmbus_onchannel_event(channel);
 275                 /*
 276                  * WorkQueueQueueWorkItem(channel->dataWorkQueue,
 277                  *                        vmbus_onchannel_event,
 278                  *                        (void*)channel);
 279                  */
 280         } else {
 281                 DPRINT_ERR(VMBUS, "channel not found for relid - %d.", relid);
 282         }
 283 }
 284
 285 /*
 286  * vmbus_on_event - Handler for events
 287  */
 288 void vmbus_on_event(void)
 289 {
 290         int dword;
 291         int maxdword = MAX_NUM_CHANNELS_SUPPORTED >> 5;
 292         int bit;
 293         int relid;
 294         u32 *recv_int_page = vmbus_connection.recv_int_page;
 295
 296         /* Check events */
 297         if (recv_int_page) {
 298                 for (dword = 0; dword < maxdword; dword++) {
 299                         if (recv_int_page[dword]) {
 300                                 for (bit = 0; bit < 32; bit++) {
 301                                         if (test_and_clear_bit(bit,
 302                                                 (unsigned long *)
 303                                                 &recv_int_page[dword])) {
 304                                                 relid = (dword << 5) + bit;
 305                                                 DPRINT_DBG(VMBUS, "event detected for relid - %d", relid);
 306
 307                                                 if (relid == 0) {
 308                                                         /* special case - vmbus channel protocol msg */
 309                                                         DPRINT_DBG(VMBUS, "invalid relid - %d", relid);
 310                                                         continue;
 311                                                 } else {
 312                                                         /* QueueWorkItem(VmbusProcessEvent, (void*)relid); */
 313                                                         /* ret = WorkQueueQueueWorkItem(gVmbusConnection.workQueue, VmbusProcessChannelEvent, (void*)relid); */
 314                                                 process_chn_event((void *)
 315                                                 (unsigned long)relid);
 316                                                 }
 317                                         }
 318                                 }
 319                         }
 320                  }
 321         }
 322         return;
 323 }
 324
 325 /*
 326  * vmbus_post_msg - Send a msg on the vmbus's message connection
 327  */
 328 int vmbus_post_msg(void *buffer, size_t buflen)
 329 {
 330         union hv_connection_id conn_id;
 331
 332         conn_id.asu32 = 0;
 333         conn_id.u.id = VMBUS_MESSAGE_CONNECTION_ID;
 334         return hv_post_message(conn_id, 1, buffer, buflen);
 335 }
 336
 337 /*
 338  * vmbus_set_event - Send an event notification to the parent
 339  */
 340 int vmbus_set_event(u32 child_relid)
 341 {
 342         /* Each u32 represents 32 channels */
 343         set_bit(child_relid & 31,
 344                 (unsigned long *)vmbus_connection.send_int_page +
 345                 (child_relid >> 5));
 346
 347         return hv_signal_event();
 348 }