blob: 773a2f25a8f062e794a5ac6b3f4747fab61d0ffb [file] [log] [blame]
Hank Janssen3e7ee492009-07-13 16:02:34 -07001/*
Hank Janssen3e7ee492009-07-13 16:02:34 -07002 * 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>
Hank Janssen3e7ee492009-07-13 16:02:34 -070020 */
Hank Janssen0a466182011-03-29 13:58:47 -070021#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
22
Greg Kroah-Hartman5654e932009-07-14 15:08:20 -070023#include <linux/kernel.h>
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -080024#include <linux/sched.h>
25#include <linux/wait.h>
Greg Kroah-Hartmana0086dc2009-08-17 17:22:08 -070026#include <linux/mm.h>
Tejun Heo5a0e3ad2010-03-24 17:04:11 +090027#include <linux/slab.h>
Hank Janssenc88c4e42010-05-04 15:55:05 -070028#include <linux/module.h>
Greg Kroah-Hartman46a97192011-10-04 12:29:52 -070029#include <linux/hyperv.h>
K. Y. Srinivasan3f335ea2011-05-12 19:34:15 -070030
K. Y. Srinivasan0f2a6612011-05-12 19:34:28 -070031#include "hyperv_vmbus.h"
Hank Janssen3e7ee492009-07-13 16:02:34 -070032
K. Y. Srinivasane3fe0bb2011-02-11 10:00:12 -080033#define NUM_PAGES_SPANNED(addr, len) \
34((PAGE_ALIGN(addr + len) >> PAGE_SHIFT) - (addr >> PAGE_SHIFT))
35
Hank Janssen3e189512010-03-04 22:11:00 +000036/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -070037 * vmbus_setevent- Trigger an event notification on the specified
Hank Janssen3e189512010-03-04 22:11:00 +000038 * channel.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -070039 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -070040static void vmbus_setevent(struct vmbus_channel *channel)
Hank Janssen3e7ee492009-07-13 16:02:34 -070041{
Haiyang Zhang39d70a42010-09-30 10:52:13 -070042 struct hv_monitor_page *monitorpage;
Hank Janssen3e7ee492009-07-13 16:02:34 -070043
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080044 if (channel->offermsg.monitor_allocated) {
Bill Pemberton454f18a2009-07-27 16:47:24 -040045 /* Each u32 represents 32 channels */
Olaf Hering22356582011-03-21 14:41:37 +010046 sync_set_bit(channel->offermsg.child_relid & 31,
Haiyang Zhangda9fcb72011-01-26 12:12:14 -080047 (unsigned long *) vmbus_connection.send_int_page +
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080048 (channel->offermsg.child_relid >> 5));
Hank Janssen3e7ee492009-07-13 16:02:34 -070049
Haiyang Zhangda9fcb72011-01-26 12:12:14 -080050 monitorpage = vmbus_connection.monitor_pages;
Haiyang Zhang39d70a42010-09-30 10:52:13 -070051 monitorpage++; /* Get the child to parent monitor page */
Hank Janssen3e7ee492009-07-13 16:02:34 -070052
Olaf Hering22356582011-03-21 14:41:37 +010053 sync_set_bit(channel->monitor_bit,
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080054 (unsigned long *)&monitorpage->trigger_group
55 [channel->monitor_grp].pending);
Bill Pemberton7c369f42009-07-29 17:00:11 -040056
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -070057 } else {
Haiyang Zhangc6977672011-01-26 12:12:08 -080058 vmbus_set_event(channel->offermsg.child_relid);
Hank Janssen3e7ee492009-07-13 16:02:34 -070059 }
Hank Janssen3e7ee492009-07-13 16:02:34 -070060}
61
Hank Janssen3e189512010-03-04 22:11:00 +000062/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -070063 * vmbus_get_debug_info -Retrieve various channel debug info
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -070064 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -070065void vmbus_get_debug_info(struct vmbus_channel *channel,
Haiyang Zhang39d70a42010-09-30 10:52:13 -070066 struct vmbus_channel_debug_info *debuginfo)
Hank Janssen3e7ee492009-07-13 16:02:34 -070067{
Haiyang Zhang39d70a42010-09-30 10:52:13 -070068 struct hv_monitor_page *monitorpage;
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080069 u8 monitor_group = (u8)channel->offermsg.monitorid / 32;
70 u8 monitor_offset = (u8)channel->offermsg.monitorid % 32;
Hank Janssen3e7ee492009-07-13 16:02:34 -070071
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080072 debuginfo->relid = channel->offermsg.child_relid;
73 debuginfo->state = channel->state;
74 memcpy(&debuginfo->interfacetype,
K. Y. Srinivasan358d2ee2011-08-25 09:48:28 -070075 &channel->offermsg.offer.if_type, sizeof(uuid_le));
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080076 memcpy(&debuginfo->interface_instance,
Haiyang Zhang767dff62011-01-26 12:12:12 -080077 &channel->offermsg.offer.if_instance,
K. Y. Srinivasan358d2ee2011-08-25 09:48:28 -070078 sizeof(uuid_le));
Hank Janssen3e7ee492009-07-13 16:02:34 -070079
Haiyang Zhangda9fcb72011-01-26 12:12:14 -080080 monitorpage = (struct hv_monitor_page *)vmbus_connection.monitor_pages;
Hank Janssen3e7ee492009-07-13 16:02:34 -070081
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080082 debuginfo->monitorid = channel->offermsg.monitorid;
Hank Janssen3e7ee492009-07-13 16:02:34 -070083
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080084 debuginfo->servermonitor_pending =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080085 monitorpage->trigger_group[monitor_group].pending;
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080086 debuginfo->servermonitor_latency =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080087 monitorpage->latency[monitor_group][monitor_offset];
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080088 debuginfo->servermonitor_connectionid =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080089 monitorpage->parameter[monitor_group]
90 [monitor_offset].connectionid.u.id;
Hank Janssen3e7ee492009-07-13 16:02:34 -070091
Haiyang Zhang39d70a42010-09-30 10:52:13 -070092 monitorpage++;
Hank Janssen3e7ee492009-07-13 16:02:34 -070093
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080094 debuginfo->clientmonitor_pending =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080095 monitorpage->trigger_group[monitor_group].pending;
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080096 debuginfo->clientmonitor_latency =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080097 monitorpage->latency[monitor_group][monitor_offset];
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -080098 debuginfo->clientmonitor_connectionid =
Haiyang Zhangf6feebe2010-11-08 14:04:39 -080099 monitorpage->parameter[monitor_group]
100 [monitor_offset].connectionid.u.id;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700101
K. Y. Srinivasana75b61d2011-05-10 07:55:28 -0700102 hv_ringbuffer_get_debuginfo(&channel->inbound, &debuginfo->inbound);
103 hv_ringbuffer_get_debuginfo(&channel->outbound, &debuginfo->outbound);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700104}
105
Hank Janssen3e189512010-03-04 22:11:00 +0000106/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700107 * vmbus_open - Open the specified channel.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700108 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700109int vmbus_open(struct vmbus_channel *newchannel, u32 send_ringbuffer_size,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700110 u32 recv_ringbuffer_size, void *userdata, u32 userdatalen,
111 void (*onchannelcallback)(void *context), void *context)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700112{
K. Y. Srinivasan0987ff62011-08-25 09:48:55 -0700113 struct vmbus_channel_open_channel *open_msg;
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700114 struct vmbus_channel_msginfo *open_info = NULL;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700115 void *in, *out;
Greg Kroah-Hartmandd0813b2009-07-15 14:56:45 -0700116 unsigned long flags;
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700117 int ret, t, err = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700118
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800119 newchannel->onchannel_callback = onchannelcallback;
120 newchannel->channel_callback_context = context;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700121
Bill Pemberton454f18a2009-07-27 16:47:24 -0400122 /* Allocate the ring buffer */
K. Y. Srinivasandf3493e2011-02-11 09:59:00 -0800123 out = (void *)__get_free_pages(GFP_KERNEL|__GFP_ZERO,
124 get_order(send_ringbuffer_size + recv_ringbuffer_size));
125
Bill Pemberton7e052d92010-05-05 15:27:30 -0400126 if (!out)
127 return -ENOMEM;
128
Hank Janssen3e7ee492009-07-13 16:02:34 -0700129
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700130 in = (void *)((unsigned long)out + send_ringbuffer_size);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700131
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800132 newchannel->ringbuffer_pages = out;
133 newchannel->ringbuffer_pagecount = (send_ringbuffer_size +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700134 recv_ringbuffer_size) >> PAGE_SHIFT;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700135
K. Y. Srinivasan72a95cb2011-05-10 07:55:21 -0700136 ret = hv_ringbuffer_init(
137 &newchannel->outbound, out, send_ringbuffer_size);
138
Haiyang Zhangfd4dc882010-05-13 15:56:30 +0000139 if (ret != 0) {
Bill Pemberton3324fb42010-05-05 15:27:49 -0400140 err = ret;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700141 goto error0;
Bill Pemberton3324fb42010-05-05 15:27:49 -0400142 }
Hank Janssen3e7ee492009-07-13 16:02:34 -0700143
K. Y. Srinivasan72a95cb2011-05-10 07:55:21 -0700144 ret = hv_ringbuffer_init(
145 &newchannel->inbound, in, recv_ringbuffer_size);
Haiyang Zhangfd4dc882010-05-13 15:56:30 +0000146 if (ret != 0) {
Bill Pemberton3324fb42010-05-05 15:27:49 -0400147 err = ret;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700148 goto error0;
Bill Pemberton3324fb42010-05-05 15:27:49 -0400149 }
150
Hank Janssen3e7ee492009-07-13 16:02:34 -0700151
Bill Pemberton454f18a2009-07-27 16:47:24 -0400152 /* Establish the gpadl for the ring buffer */
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800153 newchannel->ringbuffer_gpadlhandle = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700154
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700155 ret = vmbus_establish_gpadl(newchannel,
Haiyang Zhang82f8bd42010-11-08 14:04:45 -0800156 newchannel->outbound.ring_buffer,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700157 send_ringbuffer_size +
158 recv_ringbuffer_size,
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800159 &newchannel->ringbuffer_gpadlhandle);
Bill Pembertonb94ef342010-05-05 15:27:37 -0400160
Haiyang Zhangfd4dc882010-05-13 15:56:30 +0000161 if (ret != 0) {
Bill Pembertonb94ef342010-05-05 15:27:37 -0400162 err = ret;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700163 goto error0;
Bill Pembertonb94ef342010-05-05 15:27:37 -0400164 }
Hank Janssen3e7ee492009-07-13 16:02:34 -0700165
Bill Pemberton454f18a2009-07-27 16:47:24 -0400166 /* Create and init the channel open message */
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700167 open_info = kmalloc(sizeof(*open_info) +
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700168 sizeof(struct vmbus_channel_open_channel),
169 GFP_KERNEL);
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700170 if (!open_info) {
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400171 err = -ENOMEM;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700172 goto error0;
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400173 }
Hank Janssen3e7ee492009-07-13 16:02:34 -0700174
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700175 init_completion(&open_info->waitevent);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700176
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700177 open_msg = (struct vmbus_channel_open_channel *)open_info->msg;
K. Y. Srinivasan0987ff62011-08-25 09:48:55 -0700178 open_msg->header.msgtype = CHANNELMSG_OPENCHANNEL;
179 open_msg->openid = newchannel->offermsg.child_relid;
180 open_msg->child_relid = newchannel->offermsg.child_relid;
181 open_msg->ringbuffer_gpadlhandle = newchannel->ringbuffer_gpadlhandle;
182 open_msg->downstream_ringbuffer_pageoffset = send_ringbuffer_size >>
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700183 PAGE_SHIFT;
K. Y. Srinivasan0987ff62011-08-25 09:48:55 -0700184 open_msg->server_contextarea_gpadlhandle = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700185
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700186 if (userdatalen > MAX_USER_DEFINED_BYTES) {
Bill Pembertonc827f942010-05-05 15:27:38 -0400187 err = -EINVAL;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700188 goto error0;
Bill Pembertonc827f942010-05-05 15:27:38 -0400189 }
190
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700191 if (userdatalen)
K. Y. Srinivasan0987ff62011-08-25 09:48:55 -0700192 memcpy(open_msg->userdata, userdata, userdatalen);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700193
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800194 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700195 list_add_tail(&open_info->msglistentry,
Haiyang Zhangda9fcb72011-01-26 12:12:14 -0800196 &vmbus_connection.chn_msg_list);
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800197 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700198
K. Y. Srinivasan0987ff62011-08-25 09:48:55 -0700199 ret = vmbus_post_msg(open_msg,
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700200 sizeof(struct vmbus_channel_open_channel));
Hank Janssen98e08702011-03-29 13:58:44 -0700201
202 if (ret != 0)
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700203 goto error1;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700204
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700205 t = wait_for_completion_timeout(&open_info->waitevent, 5*HZ);
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700206 if (t == 0) {
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -0800207 err = -ETIMEDOUT;
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700208 goto error1;
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -0800209 }
210
Hank Janssen3e7ee492009-07-13 16:02:34 -0700211
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700212 if (open_info->response.open_result.status)
213 err = open_info->response.open_result.status;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700214
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800215 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700216 list_del(&open_info->msglistentry);
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800217 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700218
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700219 kfree(open_info);
Hank Janssen98e08702011-03-29 13:58:44 -0700220 return err;
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400221
K. Y. Srinivasan13925502012-10-12 13:22:42 -0700222error1:
223 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
224 list_del(&open_info->msglistentry);
225 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
226
227error0:
K. Y. Srinivasandf3493e2011-02-11 09:59:00 -0800228 free_pages((unsigned long)out,
229 get_order(send_ringbuffer_size + recv_ringbuffer_size));
K. Y. Srinivasan176fb9e2011-08-25 09:48:57 -0700230 kfree(open_info);
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400231 return err;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700232}
Greg Kroah-Hartman36ceadf2010-10-21 09:58:21 -0700233EXPORT_SYMBOL_GPL(vmbus_open);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700234
Hank Janssen3e189512010-03-04 22:11:00 +0000235/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700236 * create_gpadl_header - Creates a gpadl for the specified buffer
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700237 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700238static int create_gpadl_header(void *kbuffer, u32 size,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700239 struct vmbus_channel_msginfo **msginfo,
240 u32 *messagecount)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700241{
242 int i;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700243 int pagecount;
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700244 unsigned long long pfn;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700245 struct vmbus_channel_gpadl_header *gpadl_header;
246 struct vmbus_channel_gpadl_body *gpadl_body;
247 struct vmbus_channel_msginfo *msgheader;
248 struct vmbus_channel_msginfo *msgbody = NULL;
249 u32 msgsize;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700250
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700251 int pfnsum, pfncount, pfnleft, pfncurr, pfnsize;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700252
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700253 pagecount = size >> PAGE_SHIFT;
254 pfn = virt_to_phys(kbuffer) >> PAGE_SHIFT;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700255
Bill Pemberton454f18a2009-07-27 16:47:24 -0400256 /* do we need a gpadl body msg */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700257 pfnsize = MAX_SIZE_CHANNEL_MESSAGE -
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700258 sizeof(struct vmbus_channel_gpadl_header) -
259 sizeof(struct gpa_range);
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700260 pfncount = pfnsize / sizeof(u64);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700261
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700262 if (pagecount > pfncount) {
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700263 /* we need a gpadl body */
Bill Pemberton454f18a2009-07-27 16:47:24 -0400264 /* fill in the header */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700265 msgsize = sizeof(struct vmbus_channel_msginfo) +
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700266 sizeof(struct vmbus_channel_gpadl_header) +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700267 sizeof(struct gpa_range) + pfncount * sizeof(u64);
268 msgheader = kzalloc(msgsize, GFP_KERNEL);
269 if (!msgheader)
Bill Pembertond1c250b2010-05-05 15:27:35 -0400270 goto nomem;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700271
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800272 INIT_LIST_HEAD(&msgheader->submsglist);
273 msgheader->msgsize = msgsize;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700274
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700275 gpadl_header = (struct vmbus_channel_gpadl_header *)
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800276 msgheader->msg;
277 gpadl_header->rangecount = 1;
278 gpadl_header->range_buflen = sizeof(struct gpa_range) +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700279 pagecount * sizeof(u64);
Haiyang Zhang415f2282011-01-26 12:12:13 -0800280 gpadl_header->range[0].byte_offset = 0;
281 gpadl_header->range[0].byte_count = size;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700282 for (i = 0; i < pfncount; i++)
Haiyang Zhang415f2282011-01-26 12:12:13 -0800283 gpadl_header->range[0].pfn_array[i] = pfn+i;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700284 *msginfo = msgheader;
285 *messagecount = 1;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700286
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700287 pfnsum = pfncount;
288 pfnleft = pagecount - pfncount;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700289
Bill Pemberton454f18a2009-07-27 16:47:24 -0400290 /* how many pfns can we fit */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700291 pfnsize = MAX_SIZE_CHANNEL_MESSAGE -
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700292 sizeof(struct vmbus_channel_gpadl_body);
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700293 pfncount = pfnsize / sizeof(u64);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700294
Bill Pemberton454f18a2009-07-27 16:47:24 -0400295 /* fill in the body */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700296 while (pfnleft) {
297 if (pfnleft > pfncount)
298 pfncurr = pfncount;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700299 else
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700300 pfncurr = pfnleft;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700301
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700302 msgsize = sizeof(struct vmbus_channel_msginfo) +
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700303 sizeof(struct vmbus_channel_gpadl_body) +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700304 pfncurr * sizeof(u64);
305 msgbody = kzalloc(msgsize, GFP_KERNEL);
K. Y. Srinivasanf38cf9c2011-06-06 15:50:06 -0700306
307 if (!msgbody) {
308 struct vmbus_channel_msginfo *pos = NULL;
309 struct vmbus_channel_msginfo *tmp = NULL;
310 /*
311 * Free up all the allocated messages.
312 */
313 list_for_each_entry_safe(pos, tmp,
314 &msgheader->submsglist,
315 msglistentry) {
316
317 list_del(&pos->msglistentry);
318 kfree(pos);
319 }
320
Bill Pembertond1c250b2010-05-05 15:27:35 -0400321 goto nomem;
K. Y. Srinivasanf38cf9c2011-06-06 15:50:06 -0700322 }
323
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800324 msgbody->msgsize = msgsize;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700325 (*messagecount)++;
326 gpadl_body =
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800327 (struct vmbus_channel_gpadl_body *)msgbody->msg;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700328
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700329 /*
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700330 * Gpadl is u32 and we are using a pointer which could
331 * be 64-bit
K. Y. Srinivasanf27df642011-06-06 15:49:56 -0700332 * This is governed by the guest/host protocol and
333 * so the hypervisor gurantees that this is ok.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700334 */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700335 for (i = 0; i < pfncurr; i++)
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800336 gpadl_body->pfn[i] = pfn + pfnsum + i;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700337
Bill Pemberton454f18a2009-07-27 16:47:24 -0400338 /* add to msg header */
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800339 list_add_tail(&msgbody->msglistentry,
340 &msgheader->submsglist);
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700341 pfnsum += pfncurr;
342 pfnleft -= pfncurr;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700343 }
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700344 } else {
Bill Pemberton454f18a2009-07-27 16:47:24 -0400345 /* everything fits in a header */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700346 msgsize = sizeof(struct vmbus_channel_msginfo) +
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700347 sizeof(struct vmbus_channel_gpadl_header) +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700348 sizeof(struct gpa_range) + pagecount * sizeof(u64);
349 msgheader = kzalloc(msgsize, GFP_KERNEL);
350 if (msgheader == NULL)
Kulikov Vasiliye3eb7cd2010-07-16 20:13:51 +0400351 goto nomem;
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800352 msgheader->msgsize = msgsize;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700353
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700354 gpadl_header = (struct vmbus_channel_gpadl_header *)
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800355 msgheader->msg;
356 gpadl_header->rangecount = 1;
357 gpadl_header->range_buflen = sizeof(struct gpa_range) +
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700358 pagecount * sizeof(u64);
Haiyang Zhang415f2282011-01-26 12:12:13 -0800359 gpadl_header->range[0].byte_offset = 0;
360 gpadl_header->range[0].byte_count = size;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700361 for (i = 0; i < pagecount; i++)
Haiyang Zhang415f2282011-01-26 12:12:13 -0800362 gpadl_header->range[0].pfn_array[i] = pfn+i;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700363
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700364 *msginfo = msgheader;
365 *messagecount = 1;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700366 }
367
368 return 0;
Bill Pembertond1c250b2010-05-05 15:27:35 -0400369nomem:
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700370 kfree(msgheader);
371 kfree(msgbody);
Bill Pembertond1c250b2010-05-05 15:27:35 -0400372 return -ENOMEM;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700373}
374
Hank Janssen3e189512010-03-04 22:11:00 +0000375/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700376 * vmbus_establish_gpadl - Estabish a GPADL for the specified buffer
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700377 *
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700378 * @channel: a channel
379 * @kbuffer: from kmalloc
380 * @size: page-size multiple
381 * @gpadl_handle: some funky thing
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700382 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700383int vmbus_establish_gpadl(struct vmbus_channel *channel, void *kbuffer,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700384 u32 size, u32 *gpadl_handle)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700385{
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700386 struct vmbus_channel_gpadl_header *gpadlmsg;
387 struct vmbus_channel_gpadl_body *gpadl_body;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700388 struct vmbus_channel_msginfo *msginfo = NULL;
389 struct vmbus_channel_msginfo *submsginfo;
390 u32 msgcount;
Bill Pemberton53af5452009-09-11 21:46:44 -0400391 struct list_head *curr;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700392 u32 next_gpadl_handle;
Greg Kroah-Hartmandd0813b2009-07-15 14:56:45 -0700393 unsigned long flags;
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400394 int ret = 0;
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700395 int t;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700396
Haiyang Zhangda9fcb72011-01-26 12:12:14 -0800397 next_gpadl_handle = atomic_read(&vmbus_connection.next_gpadl_handle);
398 atomic_inc(&vmbus_connection.next_gpadl_handle);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700399
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700400 ret = create_gpadl_header(kbuffer, size, &msginfo, &msgcount);
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400401 if (ret)
402 return ret;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700403
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700404 init_completion(&msginfo->waitevent);
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400405
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800406 gpadlmsg = (struct vmbus_channel_gpadl_header *)msginfo->msg;
407 gpadlmsg->header.msgtype = CHANNELMSG_GPADL_HEADER;
408 gpadlmsg->child_relid = channel->offermsg.child_relid;
409 gpadlmsg->gpadl = next_gpadl_handle;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700410
Hank Janssen3e7ee492009-07-13 16:02:34 -0700411
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800412 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800413 list_add_tail(&msginfo->msglistentry,
Haiyang Zhangda9fcb72011-01-26 12:12:14 -0800414 &vmbus_connection.chn_msg_list);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700415
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800416 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700417
Haiyang Zhangc6977672011-01-26 12:12:08 -0800418 ret = vmbus_post_msg(gpadlmsg, msginfo->msgsize -
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700419 sizeof(*msginfo));
Hank Janssen98e08702011-03-29 13:58:44 -0700420 if (ret != 0)
K. Y. Srinivasan00d760b2011-06-06 15:50:12 -0700421 goto cleanup;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700422
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700423 if (msgcount > 1) {
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800424 list_for_each(curr, &msginfo->submsglist) {
Bill Pemberton53af5452009-09-11 21:46:44 -0400425
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700426 submsginfo = (struct vmbus_channel_msginfo *)curr;
427 gpadl_body =
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800428 (struct vmbus_channel_gpadl_body *)submsginfo->msg;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700429
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800430 gpadl_body->header.msgtype =
431 CHANNELMSG_GPADL_BODY;
432 gpadl_body->gpadl = next_gpadl_handle;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700433
Haiyang Zhangc6977672011-01-26 12:12:08 -0800434 ret = vmbus_post_msg(gpadl_body,
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800435 submsginfo->msgsize -
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700436 sizeof(*submsginfo));
Haiyang Zhangfd4dc882010-05-13 15:56:30 +0000437 if (ret != 0)
K. Y. Srinivasan00d760b2011-06-06 15:50:12 -0700438 goto cleanup;
Bill Pemberton99259152010-05-05 15:27:36 -0400439
Hank Janssen3e7ee492009-07-13 16:02:34 -0700440 }
441 }
K. Y. Srinivasan40961de2011-06-06 15:50:05 -0700442 t = wait_for_completion_timeout(&msginfo->waitevent, 5*HZ);
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700443 BUG_ON(t == 0);
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -0800444
Hank Janssen3e7ee492009-07-13 16:02:34 -0700445
Bill Pemberton454f18a2009-07-27 16:47:24 -0400446 /* At this point, we received the gpadl created msg */
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800447 *gpadl_handle = gpadlmsg->gpadl;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700448
K. Y. Srinivasan00d760b2011-06-06 15:50:12 -0700449cleanup:
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800450 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800451 list_del(&msginfo->msglistentry);
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800452 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700453
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700454 kfree(msginfo);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700455 return ret;
456}
Greg Kroah-Hartman98873722010-10-21 08:47:43 -0700457EXPORT_SYMBOL_GPL(vmbus_establish_gpadl);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700458
Hank Janssen3e189512010-03-04 22:11:00 +0000459/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700460 * vmbus_teardown_gpadl -Teardown the specified GPADL handle
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700461 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700462int vmbus_teardown_gpadl(struct vmbus_channel *channel, u32 gpadl_handle)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700463{
Greg Kroah-Hartman82250212009-08-26 15:16:04 -0700464 struct vmbus_channel_gpadl_teardown *msg;
Greg Kroah-Hartmanaded71652009-08-18 15:21:19 -0700465 struct vmbus_channel_msginfo *info;
Greg Kroah-Hartmandd0813b2009-07-15 14:56:45 -0700466 unsigned long flags;
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700467 int ret, t;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700468
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700469 info = kmalloc(sizeof(*info) +
470 sizeof(struct vmbus_channel_gpadl_teardown), GFP_KERNEL);
Bill Pembertonc3bf2e22010-05-05 15:27:34 -0400471 if (!info)
472 return -ENOMEM;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700473
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700474 init_completion(&info->waitevent);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700475
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800476 msg = (struct vmbus_channel_gpadl_teardown *)info->msg;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700477
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800478 msg->header.msgtype = CHANNELMSG_GPADL_TEARDOWN;
479 msg->child_relid = channel->offermsg.child_relid;
480 msg->gpadl = gpadl_handle;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700481
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800482 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800483 list_add_tail(&info->msglistentry,
Haiyang Zhangda9fcb72011-01-26 12:12:14 -0800484 &vmbus_connection.chn_msg_list);
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800485 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Haiyang Zhangc6977672011-01-26 12:12:08 -0800486 ret = vmbus_post_msg(msg,
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700487 sizeof(struct vmbus_channel_gpadl_teardown));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700488
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -0800489 BUG_ON(ret != 0);
K. Y. Srinivasan40961de2011-06-06 15:50:05 -0700490 t = wait_for_completion_timeout(&info->waitevent, 5*HZ);
K. Y. Srinivasan9568a192011-05-10 07:55:39 -0700491 BUG_ON(t == 0);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700492
Bill Pemberton454f18a2009-07-27 16:47:24 -0400493 /* Received a torndown response */
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800494 spin_lock_irqsave(&vmbus_connection.channelmsg_lock, flags);
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800495 list_del(&info->msglistentry);
Haiyang Zhang15b2f642011-01-26 12:12:07 -0800496 spin_unlock_irqrestore(&vmbus_connection.channelmsg_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700497
Greg Kroah-Hartman8c69f522009-07-15 12:48:29 -0700498 kfree(info);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700499 return ret;
500}
Greg Kroah-Hartman18726d72010-10-21 08:39:59 -0700501EXPORT_SYMBOL_GPL(vmbus_teardown_gpadl);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700502
Hank Janssen3e189512010-03-04 22:11:00 +0000503/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700504 * vmbus_close - Close the specified channel
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700505 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700506void vmbus_close(struct vmbus_channel *channel)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700507{
Greg Kroah-Hartman82250212009-08-26 15:16:04 -0700508 struct vmbus_channel_close_channel *msg;
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700509 int ret;
K. Y. Srinivasandad76bf2011-08-27 11:31:33 -0700510 unsigned long flags;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700511
Bill Pemberton454f18a2009-07-27 16:47:24 -0400512 /* Stop callback and cancel the timer asap */
K. Y. Srinivasandad76bf2011-08-27 11:31:33 -0700513 spin_lock_irqsave(&channel->inbound_lock, flags);
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800514 channel->onchannel_callback = NULL;
K. Y. Srinivasandad76bf2011-08-27 11:31:33 -0700515 spin_unlock_irqrestore(&channel->inbound_lock, flags);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700516
Bill Pemberton454f18a2009-07-27 16:47:24 -0400517 /* Send a closing message */
Hank Janssen3e7ee492009-07-13 16:02:34 -0700518
K. Y. Srinivasane9a27a92011-06-06 15:49:59 -0700519 msg = &channel->close_msg.msg;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700520
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800521 msg->header.msgtype = CHANNELMSG_CLOSECHANNEL;
522 msg->child_relid = channel->offermsg.child_relid;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700523
Haiyang Zhangc6977672011-01-26 12:12:08 -0800524 ret = vmbus_post_msg(msg, sizeof(struct vmbus_channel_close_channel));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700525
K. Y. Srinivasan0c3b7b22011-02-11 09:59:43 -0800526 BUG_ON(ret != 0);
Bill Pemberton454f18a2009-07-27 16:47:24 -0400527 /* Tear down the gpadl for the channel's ring buffer */
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800528 if (channel->ringbuffer_gpadlhandle)
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700529 vmbus_teardown_gpadl(channel,
Haiyang Zhangc50f7fb2010-11-08 14:04:38 -0800530 channel->ringbuffer_gpadlhandle);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700531
Bill Pemberton454f18a2009-07-27 16:47:24 -0400532 /* Cleanup the ring buffers for this channel */
K. Y. Srinivasan2dba6882011-05-10 07:55:22 -0700533 hv_ringbuffer_cleanup(&channel->outbound);
534 hv_ringbuffer_cleanup(&channel->inbound);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700535
K. Y. Srinivasandf3493e2011-02-11 09:59:00 -0800536 free_pages((unsigned long)channel->ringbuffer_pages,
537 get_order(channel->ringbuffer_pagecount * PAGE_SIZE));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700538
Hank Janssen3e7ee492009-07-13 16:02:34 -0700539
Hank Janssen3e7ee492009-07-13 16:02:34 -0700540}
Greg Kroah-Hartman70bfa302010-10-21 09:52:22 -0700541EXPORT_SYMBOL_GPL(vmbus_close);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700542
Hank Janssenc88c4e42010-05-04 15:55:05 -0700543/**
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700544 * vmbus_sendpacket() - Send the specified buffer on the given channel
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700545 * @channel: Pointer to vmbus_channel structure.
546 * @buffer: Pointer to the buffer you want to receive the data into.
547 * @bufferlen: Maximum size of what the the buffer will hold
548 * @requestid: Identifier of the request
549 * @type: Type of packet that is being send e.g. negotiate, time
Hank Janssenc88c4e42010-05-04 15:55:05 -0700550 * packet etc.
551 *
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700552 * Sends data in @buffer directly to hyper-v via the vmbus
Hank Janssenc88c4e42010-05-04 15:55:05 -0700553 * This will send the data unparsed to hyper-v.
554 *
555 * Mainly used by Hyper-V drivers.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700556 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700557int vmbus_sendpacket(struct vmbus_channel *channel, const void *buffer,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700558 u32 bufferlen, u64 requestid,
559 enum vmbus_packet_type type, u32 flags)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700560{
Greg Kroah-Hartman8dc0a062009-08-27 16:02:36 -0700561 struct vmpacket_descriptor desc;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700562 u32 packetlen = sizeof(struct vmpacket_descriptor) + bufferlen;
Uwe Kleine-König73509682011-01-20 09:32:01 +0100563 u32 packetlen_aligned = ALIGN(packetlen, sizeof(u64));
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700564 struct scatterlist bufferlist[3];
565 u64 aligned_data = 0;
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700566 int ret;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700567
Hank Janssen3e7ee492009-07-13 16:02:34 -0700568
Bill Pemberton454f18a2009-07-27 16:47:24 -0400569 /* Setup the descriptor */
Haiyang Zhang415f2282011-01-26 12:12:13 -0800570 desc.type = type; /* VmbusPacketTypeDataInBand; */
571 desc.flags = flags; /* VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED; */
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700572 /* in 8-bytes granularity */
Haiyang Zhang415f2282011-01-26 12:12:13 -0800573 desc.offset8 = sizeof(struct vmpacket_descriptor) >> 3;
574 desc.len8 = (u16)(packetlen_aligned >> 3);
575 desc.trans_id = requestid;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700576
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700577 sg_init_table(bufferlist, 3);
578 sg_set_buf(&bufferlist[0], &desc, sizeof(struct vmpacket_descriptor));
579 sg_set_buf(&bufferlist[1], buffer, bufferlen);
580 sg_set_buf(&bufferlist[2], &aligned_data,
581 packetlen_aligned - packetlen);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700582
K. Y. Srinivasan633c4dc2011-05-10 07:55:23 -0700583 ret = hv_ringbuffer_write(&channel->outbound, bufferlist, 3);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700584
K. Y. Srinivasandecc49d2011-05-10 07:55:26 -0700585 if (ret == 0 && !hv_get_ringbuffer_interrupt_mask(&channel->outbound))
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700586 vmbus_setevent(channel);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700587
Hank Janssen3e7ee492009-07-13 16:02:34 -0700588 return ret;
589}
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700590EXPORT_SYMBOL(vmbus_sendpacket);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700591
Hank Janssen3e189512010-03-04 22:11:00 +0000592/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700593 * vmbus_sendpacket_pagebuffer - Send a range of single-page buffer
Hank Janssen3e189512010-03-04 22:11:00 +0000594 * packets using a GPADL Direct packet type.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700595 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700596int vmbus_sendpacket_pagebuffer(struct vmbus_channel *channel,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700597 struct hv_page_buffer pagebuffers[],
598 u32 pagecount, void *buffer, u32 bufferlen,
599 u64 requestid)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700600{
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700601 int ret;
602 int i;
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000603 struct vmbus_channel_packet_page_buffer desc;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700604 u32 descsize;
605 u32 packetlen;
606 u32 packetlen_aligned;
607 struct scatterlist bufferlist[3];
608 u64 aligned_data = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700609
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700610 if (pagecount > MAX_PAGE_BUFFER_COUNT)
Bill Pemberton002b53e2010-05-05 15:27:39 -0400611 return -EINVAL;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700612
Hank Janssen3e7ee492009-07-13 16:02:34 -0700613
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700614 /*
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000615 * Adjust the size down since vmbus_channel_packet_page_buffer is the
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700616 * largest size we support
617 */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700618 descsize = sizeof(struct vmbus_channel_packet_page_buffer) -
619 ((MAX_PAGE_BUFFER_COUNT - pagecount) *
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700620 sizeof(struct hv_page_buffer));
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700621 packetlen = descsize + bufferlen;
Uwe Kleine-König73509682011-01-20 09:32:01 +0100622 packetlen_aligned = ALIGN(packetlen, sizeof(u64));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700623
Bill Pemberton454f18a2009-07-27 16:47:24 -0400624 /* Setup the descriptor */
Haiyang Zhang415f2282011-01-26 12:12:13 -0800625 desc.type = VM_PKT_DATA_USING_GPA_DIRECT;
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000626 desc.flags = VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700627 desc.dataoffset8 = descsize >> 3; /* in 8-bytes grandularity */
628 desc.length8 = (u16)(packetlen_aligned >> 3);
629 desc.transactionid = requestid;
630 desc.rangecount = pagecount;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700631
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700632 for (i = 0; i < pagecount; i++) {
Haiyang Zhangca623ad2011-01-26 12:12:11 -0800633 desc.range[i].len = pagebuffers[i].len;
634 desc.range[i].offset = pagebuffers[i].offset;
635 desc.range[i].pfn = pagebuffers[i].pfn;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700636 }
637
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700638 sg_init_table(bufferlist, 3);
639 sg_set_buf(&bufferlist[0], &desc, descsize);
640 sg_set_buf(&bufferlist[1], buffer, bufferlen);
641 sg_set_buf(&bufferlist[2], &aligned_data,
642 packetlen_aligned - packetlen);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700643
K. Y. Srinivasan633c4dc2011-05-10 07:55:23 -0700644 ret = hv_ringbuffer_write(&channel->outbound, bufferlist, 3);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700645
K. Y. Srinivasandecc49d2011-05-10 07:55:26 -0700646 if (ret == 0 && !hv_get_ringbuffer_interrupt_mask(&channel->outbound))
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700647 vmbus_setevent(channel);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700648
Hank Janssen3e7ee492009-07-13 16:02:34 -0700649 return ret;
650}
Greg Kroah-Hartman713efeb2010-10-21 09:29:54 -0700651EXPORT_SYMBOL_GPL(vmbus_sendpacket_pagebuffer);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700652
Hank Janssen3e189512010-03-04 22:11:00 +0000653/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700654 * vmbus_sendpacket_multipagebuffer - Send a multi-page buffer packet
Hank Janssen3e189512010-03-04 22:11:00 +0000655 * using a GPADL Direct packet type.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700656 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700657int vmbus_sendpacket_multipagebuffer(struct vmbus_channel *channel,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700658 struct hv_multipage_buffer *multi_pagebuffer,
659 void *buffer, u32 bufferlen, u64 requestid)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700660{
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700661 int ret;
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000662 struct vmbus_channel_packet_multipage_buffer desc;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700663 u32 descsize;
664 u32 packetlen;
665 u32 packetlen_aligned;
666 struct scatterlist bufferlist[3];
667 u64 aligned_data = 0;
Haiyang Zhangca623ad2011-01-26 12:12:11 -0800668 u32 pfncount = NUM_PAGES_SPANNED(multi_pagebuffer->offset,
669 multi_pagebuffer->len);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700670
Hank Janssen3e7ee492009-07-13 16:02:34 -0700671
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700672 if ((pfncount < 0) || (pfncount > MAX_MULTIPAGE_BUFFER_COUNT))
Bill Pemberton002b53e2010-05-05 15:27:39 -0400673 return -EINVAL;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700674
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700675 /*
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000676 * Adjust the size down since vmbus_channel_packet_multipage_buffer is
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700677 * the largest size we support
678 */
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700679 descsize = sizeof(struct vmbus_channel_packet_multipage_buffer) -
680 ((MAX_MULTIPAGE_BUFFER_COUNT - pfncount) *
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700681 sizeof(u64));
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700682 packetlen = descsize + bufferlen;
Uwe Kleine-König73509682011-01-20 09:32:01 +0100683 packetlen_aligned = ALIGN(packetlen, sizeof(u64));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700684
Hank Janssen3e7ee492009-07-13 16:02:34 -0700685
Bill Pemberton454f18a2009-07-27 16:47:24 -0400686 /* Setup the descriptor */
Haiyang Zhang415f2282011-01-26 12:12:13 -0800687 desc.type = VM_PKT_DATA_USING_GPA_DIRECT;
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000688 desc.flags = VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700689 desc.dataoffset8 = descsize >> 3; /* in 8-bytes grandularity */
690 desc.length8 = (u16)(packetlen_aligned >> 3);
691 desc.transactionid = requestid;
Haiyang Zhang430a8e92010-09-20 21:07:51 +0000692 desc.rangecount = 1;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700693
Haiyang Zhangca623ad2011-01-26 12:12:11 -0800694 desc.range.len = multi_pagebuffer->len;
695 desc.range.offset = multi_pagebuffer->offset;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700696
Haiyang Zhangca623ad2011-01-26 12:12:11 -0800697 memcpy(desc.range.pfn_array, multi_pagebuffer->pfn_array,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700698 pfncount * sizeof(u64));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700699
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700700 sg_init_table(bufferlist, 3);
701 sg_set_buf(&bufferlist[0], &desc, descsize);
702 sg_set_buf(&bufferlist[1], buffer, bufferlen);
703 sg_set_buf(&bufferlist[2], &aligned_data,
704 packetlen_aligned - packetlen);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700705
K. Y. Srinivasan633c4dc2011-05-10 07:55:23 -0700706 ret = hv_ringbuffer_write(&channel->outbound, bufferlist, 3);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700707
K. Y. Srinivasandecc49d2011-05-10 07:55:26 -0700708 if (ret == 0 && !hv_get_ringbuffer_interrupt_mask(&channel->outbound))
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700709 vmbus_setevent(channel);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700710
Hank Janssen3e7ee492009-07-13 16:02:34 -0700711 return ret;
712}
Greg Kroah-Hartman4cb106f2010-10-21 09:23:59 -0700713EXPORT_SYMBOL_GPL(vmbus_sendpacket_multipagebuffer);
Hank Janssenc88c4e42010-05-04 15:55:05 -0700714
715/**
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700716 * vmbus_recvpacket() - Retrieve the user packet on the specified channel
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700717 * @channel: Pointer to vmbus_channel structure.
718 * @buffer: Pointer to the buffer you want to receive the data into.
719 * @bufferlen: Maximum size of what the the buffer will hold
720 * @buffer_actual_len: The actual size of the data after it was received
721 * @requestid: Identifier of the request
Hank Janssenc88c4e42010-05-04 15:55:05 -0700722 *
723 * Receives directly from the hyper-v vmbus and puts the data it received
724 * into Buffer. This will receive the data unparsed from hyper-v.
725 *
726 * Mainly used by Hyper-V drivers.
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700727 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700728int vmbus_recvpacket(struct vmbus_channel *channel, void *buffer,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700729 u32 bufferlen, u32 *buffer_actual_len, u64 *requestid)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700730{
Greg Kroah-Hartman8dc0a062009-08-27 16:02:36 -0700731 struct vmpacket_descriptor desc;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700732 u32 packetlen;
733 u32 userlen;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700734 int ret;
735
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700736 *buffer_actual_len = 0;
737 *requestid = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700738
Hank Janssen3e7ee492009-07-13 16:02:34 -0700739
K. Y. Srinivasana89186c2011-05-10 07:55:24 -0700740 ret = hv_ringbuffer_peek(&channel->inbound, &desc,
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700741 sizeof(struct vmpacket_descriptor));
K. Y. Srinivasandad76bf2011-08-27 11:31:33 -0700742 if (ret != 0)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700743 return 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700744
Haiyang Zhang415f2282011-01-26 12:12:13 -0800745 packetlen = desc.len8 << 3;
746 userlen = packetlen - (desc.offset8 << 3);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700747
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700748 *buffer_actual_len = userlen;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700749
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700750 if (userlen > bufferlen) {
Hank Janssen3e7ee492009-07-13 16:02:34 -0700751
Hank Janssen0a466182011-03-29 13:58:47 -0700752 pr_err("Buffer too small - got %d needs %d\n",
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700753 bufferlen, userlen);
K. Y. Srinivasan926ae5262011-06-06 15:50:11 -0700754 return -ETOOSMALL;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700755 }
756
Haiyang Zhang415f2282011-01-26 12:12:13 -0800757 *requestid = desc.trans_id;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700758
Bill Pemberton454f18a2009-07-27 16:47:24 -0400759 /* Copy over the packet to the user buffer */
K. Y. Srinivasan38397c82011-05-10 07:55:25 -0700760 ret = hv_ringbuffer_read(&channel->inbound, buffer, userlen,
Haiyang Zhang415f2282011-01-26 12:12:13 -0800761 (desc.offset8 << 3));
Hank Janssen3e7ee492009-07-13 16:02:34 -0700762
Hank Janssen3e7ee492009-07-13 16:02:34 -0700763
Hank Janssen3e7ee492009-07-13 16:02:34 -0700764 return 0;
765}
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700766EXPORT_SYMBOL(vmbus_recvpacket);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700767
Hank Janssen3e189512010-03-04 22:11:00 +0000768/*
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700769 * vmbus_recvpacket_raw - Retrieve the raw packet on the specified channel
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700770 */
Haiyang Zhangfff41b22010-10-07 11:40:08 -0700771int vmbus_recvpacket_raw(struct vmbus_channel *channel, void *buffer,
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700772 u32 bufferlen, u32 *buffer_actual_len,
773 u64 *requestid)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700774{
Greg Kroah-Hartman8dc0a062009-08-27 16:02:36 -0700775 struct vmpacket_descriptor desc;
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700776 u32 packetlen;
777 u32 userlen;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700778 int ret;
779
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700780 *buffer_actual_len = 0;
781 *requestid = 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700782
Hank Janssen3e7ee492009-07-13 16:02:34 -0700783
K. Y. Srinivasana89186c2011-05-10 07:55:24 -0700784 ret = hv_ringbuffer_peek(&channel->inbound, &desc,
Greg Kroah-Hartmanf4266e32009-09-01 17:24:57 -0700785 sizeof(struct vmpacket_descriptor));
K. Y. Srinivasandad76bf2011-08-27 11:31:33 -0700786 if (ret != 0)
Hank Janssen3e7ee492009-07-13 16:02:34 -0700787 return 0;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700788
Hank Janssen3e7ee492009-07-13 16:02:34 -0700789
Haiyang Zhang415f2282011-01-26 12:12:13 -0800790 packetlen = desc.len8 << 3;
791 userlen = packetlen - (desc.offset8 << 3);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700792
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700793 *buffer_actual_len = packetlen;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700794
Haiyang Zhang39d70a42010-09-30 10:52:13 -0700795 if (packetlen > bufferlen) {
Hank Janssen0a466182011-03-29 13:58:47 -0700796 pr_err("Buffer too small - needed %d bytes but "
797 "got space for only %d bytes\n",
798 packetlen, bufferlen);
K. Y. Srinivasan3d5cad92011-08-25 09:48:59 -0700799 return -ENOBUFS;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700800 }
801
Haiyang Zhang415f2282011-01-26 12:12:13 -0800802 *requestid = desc.trans_id;
Hank Janssen3e7ee492009-07-13 16:02:34 -0700803
Bill Pemberton454f18a2009-07-27 16:47:24 -0400804 /* Copy over the entire packet to the user buffer */
K. Y. Srinivasan38397c82011-05-10 07:55:25 -0700805 ret = hv_ringbuffer_read(&channel->inbound, buffer, packetlen, 0);
Hank Janssen3e7ee492009-07-13 16:02:34 -0700806
Hank Janssen3e7ee492009-07-13 16:02:34 -0700807 return 0;
808}
Greg Kroah-Hartmanadaee6b2010-10-21 09:09:23 -0700809EXPORT_SYMBOL_GPL(vmbus_recvpacket_raw);