1250199Sgrehan/*-
2250199Sgrehan * Copyright (c) 2009-2012 Microsoft Corp.
3250199Sgrehan * Copyright (c) 2012 NetApp Inc.
4250199Sgrehan * Copyright (c) 2012 Citrix Inc.
5250199Sgrehan * All rights reserved.
6250199Sgrehan *
7250199Sgrehan * Redistribution and use in source and binary forms, with or without
8250199Sgrehan * modification, are permitted provided that the following conditions
9250199Sgrehan * are met:
10250199Sgrehan * 1. Redistributions of source code must retain the above copyright
11250199Sgrehan *    notice unmodified, this list of conditions, and the following
12250199Sgrehan *    disclaimer.
13250199Sgrehan * 2. Redistributions in binary form must reproduce the above copyright
14250199Sgrehan *    notice, this list of conditions and the following disclaimer in the
15250199Sgrehan *    documentation and/or other materials provided with the distribution.
16250199Sgrehan *
17250199Sgrehan * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18250199Sgrehan * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19250199Sgrehan * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20250199Sgrehan * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21250199Sgrehan * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22250199Sgrehan * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23250199Sgrehan * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24250199Sgrehan * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25250199Sgrehan * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26250199Sgrehan * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27250199Sgrehan */
28250199Sgrehan
29256276Sdim#include <sys/cdefs.h>
30256276Sdim__FBSDID("$FreeBSD$");
31256276Sdim
32250199Sgrehan#include <sys/param.h>
33250199Sgrehan#include <sys/malloc.h>
34250199Sgrehan#include <sys/systm.h>
35250199Sgrehan#include <sys/mbuf.h>
36250199Sgrehan#include <sys/lock.h>
37250199Sgrehan#include <sys/mutex.h>
38250199Sgrehan#include <machine/bus.h>
39250199Sgrehan#include <vm/vm.h>
40250199Sgrehan#include <vm/vm_param.h>
41250199Sgrehan#include <vm/pmap.h>
42250199Sgrehan
43250199Sgrehan#include "hv_vmbus_priv.h"
44250199Sgrehan
45250199Sgrehanstatic int 	vmbus_channel_create_gpadl_header(
46250199Sgrehan			/* must be phys and virt contiguous*/
47250199Sgrehan			void*				contig_buffer,
48250199Sgrehan			/* page-size multiple */
49250199Sgrehan			uint32_t 			size,
50250199Sgrehan			hv_vmbus_channel_msg_info**	msg_info,
51250199Sgrehan			uint32_t*			message_count);
52250199Sgrehan
53250199Sgrehanstatic void 	vmbus_channel_set_event(hv_vmbus_channel* channel);
54250199Sgrehan
55250199Sgrehan/**
56250199Sgrehan *  @brief Trigger an event notification on the specified channel
57250199Sgrehan */
58250199Sgrehanstatic void
59250199Sgrehanvmbus_channel_set_event(hv_vmbus_channel *channel)
60250199Sgrehan{
61250199Sgrehan	hv_vmbus_monitor_page *monitor_page;
62250199Sgrehan
63250199Sgrehan	if (channel->offer_msg.monitor_allocated) {
64250199Sgrehan		/* Each uint32_t represents 32 channels */
65250199Sgrehan		synch_set_bit((channel->offer_msg.child_rel_id & 31),
66250199Sgrehan			((uint32_t *)hv_vmbus_g_connection.send_interrupt_page
67250199Sgrehan				+ ((channel->offer_msg.child_rel_id >> 5))));
68250199Sgrehan
69250199Sgrehan		monitor_page = (hv_vmbus_monitor_page *)
70250199Sgrehan			hv_vmbus_g_connection.monitor_pages;
71250199Sgrehan
72250199Sgrehan		monitor_page++; /* Get the child to parent monitor page */
73250199Sgrehan
74250199Sgrehan		synch_set_bit(channel->monitor_bit,
75250199Sgrehan			(uint32_t *)&monitor_page->
76256276Sdim				trigger_group[channel->monitor_group].u.pending);
77250199Sgrehan	} else {
78250199Sgrehan		hv_vmbus_set_event(channel->offer_msg.child_rel_id);
79250199Sgrehan	}
80250199Sgrehan
81250199Sgrehan}
82250199Sgrehan
83250199Sgrehan/**
84250199Sgrehan * @brief Open the specified channel
85250199Sgrehan */
86250199Sgrehanint
87250199Sgrehanhv_vmbus_channel_open(
88250199Sgrehan	hv_vmbus_channel*		new_channel,
89250199Sgrehan	uint32_t			send_ring_buffer_size,
90250199Sgrehan	uint32_t			recv_ring_buffer_size,
91250199Sgrehan	void*				user_data,
92250199Sgrehan	uint32_t			user_data_len,
93250199Sgrehan	hv_vmbus_pfn_channel_callback	pfn_on_channel_callback,
94250199Sgrehan	void* 				context)
95250199Sgrehan{
96250199Sgrehan
97250199Sgrehan	int ret = 0;
98250199Sgrehan	void *in, *out;
99250199Sgrehan	hv_vmbus_channel_open_channel*	open_msg;
100250199Sgrehan	hv_vmbus_channel_msg_info* 	open_info;
101250199Sgrehan
102250199Sgrehan	new_channel->on_channel_callback = pfn_on_channel_callback;
103250199Sgrehan	new_channel->channel_callback_context = context;
104250199Sgrehan
105250199Sgrehan	/* Allocate the ring buffer */
106250199Sgrehan	out = contigmalloc((send_ring_buffer_size + recv_ring_buffer_size),
107256352Sgrehan	    M_DEVBUF, M_ZERO, 0UL, BUS_SPACE_MAXADDR, PAGE_SIZE, 0);
108250199Sgrehan	KASSERT(out != NULL,
109250199Sgrehan	    ("Error VMBUS: contigmalloc failed to allocate Ring Buffer!"));
110250199Sgrehan	if (out == NULL)
111256352Sgrehan		return (ENOMEM);
112250199Sgrehan
113250199Sgrehan	in = ((uint8_t *) out + send_ring_buffer_size);
114250199Sgrehan
115250199Sgrehan	new_channel->ring_buffer_pages = out;
116256352Sgrehan	new_channel->ring_buffer_page_count = (send_ring_buffer_size +
117256352Sgrehan	    recv_ring_buffer_size) >> PAGE_SHIFT;
118256352Sgrehan	new_channel->ring_buffer_size = send_ring_buffer_size +
119256352Sgrehan	    recv_ring_buffer_size;
120250199Sgrehan
121250199Sgrehan	hv_vmbus_ring_buffer_init(
122250199Sgrehan		&new_channel->outbound,
123250199Sgrehan		out,
124250199Sgrehan		send_ring_buffer_size);
125250199Sgrehan
126250199Sgrehan	hv_vmbus_ring_buffer_init(
127250199Sgrehan		&new_channel->inbound,
128250199Sgrehan		in,
129250199Sgrehan		recv_ring_buffer_size);
130250199Sgrehan
131250199Sgrehan	/**
132250199Sgrehan	 * Establish the gpadl for the ring buffer
133250199Sgrehan	 */
134250199Sgrehan	new_channel->ring_buffer_gpadl_handle = 0;
135250199Sgrehan
136250199Sgrehan	ret = hv_vmbus_channel_establish_gpadl(new_channel,
137250199Sgrehan		new_channel->outbound.ring_buffer,
138250199Sgrehan		send_ring_buffer_size + recv_ring_buffer_size,
139250199Sgrehan		&new_channel->ring_buffer_gpadl_handle);
140250199Sgrehan
141250199Sgrehan	/**
142250199Sgrehan	 * Create and init the channel open message
143250199Sgrehan	 */
144250199Sgrehan	open_info = (hv_vmbus_channel_msg_info*) malloc(
145250199Sgrehan		sizeof(hv_vmbus_channel_msg_info) +
146250199Sgrehan			sizeof(hv_vmbus_channel_open_channel),
147250199Sgrehan		M_DEVBUF,
148250199Sgrehan		M_NOWAIT);
149250199Sgrehan	KASSERT(open_info != NULL,
150250199Sgrehan	    ("Error VMBUS: malloc failed to allocate Open Channel message!"));
151250199Sgrehan
152250199Sgrehan	if (open_info == NULL)
153250199Sgrehan		return (ENOMEM);
154250199Sgrehan
155250199Sgrehan	sema_init(&open_info->wait_sema, 0, "Open Info Sema");
156250199Sgrehan
157250199Sgrehan	open_msg = (hv_vmbus_channel_open_channel*) open_info->msg;
158250199Sgrehan	open_msg->header.message_type = HV_CHANNEL_MESSAGE_OPEN_CHANNEL;
159250199Sgrehan	open_msg->open_id = new_channel->offer_msg.child_rel_id;
160250199Sgrehan	open_msg->child_rel_id = new_channel->offer_msg.child_rel_id;
161250199Sgrehan	open_msg->ring_buffer_gpadl_handle =
162250199Sgrehan		new_channel->ring_buffer_gpadl_handle;
163250199Sgrehan	open_msg->downstream_ring_buffer_page_offset = send_ring_buffer_size
164250199Sgrehan		>> PAGE_SHIFT;
165250199Sgrehan	open_msg->server_context_area_gpadl_handle = 0;
166250199Sgrehan
167250199Sgrehan	if (user_data_len)
168250199Sgrehan		memcpy(open_msg->user_data, user_data, user_data_len);
169250199Sgrehan
170250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
171250199Sgrehan	TAILQ_INSERT_TAIL(
172250199Sgrehan		&hv_vmbus_g_connection.channel_msg_anchor,
173250199Sgrehan		open_info,
174250199Sgrehan		msg_list_entry);
175250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
176250199Sgrehan
177250199Sgrehan	ret = hv_vmbus_post_message(
178250199Sgrehan		open_msg, sizeof(hv_vmbus_channel_open_channel));
179250199Sgrehan
180250199Sgrehan	if (ret != 0)
181250199Sgrehan	    goto cleanup;
182250199Sgrehan
183250199Sgrehan	ret = sema_timedwait(&open_info->wait_sema, 500); /* KYS 5 seconds */
184250199Sgrehan
185250199Sgrehan	if (ret)
186250199Sgrehan	    goto cleanup;
187250199Sgrehan
188250199Sgrehan	if (open_info->response.open_result.status == 0) {
189250199Sgrehan	    if(bootverbose)
190250199Sgrehan		printf("VMBUS: channel <%p> open success.\n", new_channel);
191250199Sgrehan	} else {
192250199Sgrehan	    if(bootverbose)
193250199Sgrehan		printf("Error VMBUS: channel <%p> open failed - %d!\n",
194250199Sgrehan			new_channel, open_info->response.open_result.status);
195250199Sgrehan	}
196250199Sgrehan
197250199Sgrehan	cleanup:
198250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
199250199Sgrehan	TAILQ_REMOVE(
200250199Sgrehan		&hv_vmbus_g_connection.channel_msg_anchor,
201250199Sgrehan		open_info,
202250199Sgrehan		msg_list_entry);
203250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
204250199Sgrehan	sema_destroy(&open_info->wait_sema);
205250199Sgrehan	free(open_info, M_DEVBUF);
206250199Sgrehan
207250199Sgrehan	return (ret);
208250199Sgrehan}
209250199Sgrehan
210250199Sgrehan/**
211250199Sgrehan * @brief Create a gpadl for the specified buffer
212250199Sgrehan */
213250199Sgrehanstatic int
214250199Sgrehanvmbus_channel_create_gpadl_header(
215250199Sgrehan	void*				contig_buffer,
216250199Sgrehan	uint32_t			size,	/* page-size multiple */
217250199Sgrehan	hv_vmbus_channel_msg_info**	msg_info,
218250199Sgrehan	uint32_t*			message_count)
219250199Sgrehan{
220250199Sgrehan	int				i;
221250199Sgrehan	int				page_count;
222250199Sgrehan	unsigned long long 		pfn;
223250199Sgrehan	uint32_t			msg_size;
224250199Sgrehan	hv_vmbus_channel_gpadl_header*	gpa_header;
225250199Sgrehan	hv_vmbus_channel_gpadl_body*	gpadl_body;
226250199Sgrehan	hv_vmbus_channel_msg_info*	msg_header;
227250199Sgrehan	hv_vmbus_channel_msg_info*	msg_body;
228250199Sgrehan
229250199Sgrehan	int pfnSum, pfnCount, pfnLeft, pfnCurr, pfnSize;
230250199Sgrehan
231250199Sgrehan	page_count = size >> PAGE_SHIFT;
232250199Sgrehan	pfn = hv_get_phys_addr(contig_buffer) >> PAGE_SHIFT;
233250199Sgrehan
234250199Sgrehan	/*do we need a gpadl body msg */
235250199Sgrehan	pfnSize = HV_MAX_SIZE_CHANNEL_MESSAGE
236250199Sgrehan	    - sizeof(hv_vmbus_channel_gpadl_header)
237250199Sgrehan	    - sizeof(hv_gpa_range);
238250199Sgrehan	pfnCount = pfnSize / sizeof(uint64_t);
239250199Sgrehan
240250199Sgrehan	if (page_count > pfnCount) { /* if(we need a gpadl body)	*/
241250199Sgrehan	    /* fill in the header		*/
242250199Sgrehan	    msg_size = sizeof(hv_vmbus_channel_msg_info)
243250199Sgrehan		+ sizeof(hv_vmbus_channel_gpadl_header)
244250199Sgrehan		+ sizeof(hv_gpa_range)
245250199Sgrehan		+ pfnCount * sizeof(uint64_t);
246250199Sgrehan	    msg_header = malloc(msg_size, M_DEVBUF, M_NOWAIT | M_ZERO);
247250199Sgrehan	    KASSERT(
248250199Sgrehan		msg_header != NULL,
249250199Sgrehan		("Error VMBUS: malloc failed to allocate Gpadl Message!"));
250250199Sgrehan	    if (msg_header == NULL)
251250199Sgrehan		return (ENOMEM);
252250199Sgrehan
253250199Sgrehan	    TAILQ_INIT(&msg_header->sub_msg_list_anchor);
254250199Sgrehan	    msg_header->message_size = msg_size;
255250199Sgrehan
256250199Sgrehan	    gpa_header = (hv_vmbus_channel_gpadl_header*) msg_header->msg;
257250199Sgrehan	    gpa_header->range_count = 1;
258250199Sgrehan	    gpa_header->range_buf_len = sizeof(hv_gpa_range)
259250199Sgrehan		+ page_count * sizeof(uint64_t);
260250199Sgrehan	    gpa_header->range[0].byte_offset = 0;
261250199Sgrehan	    gpa_header->range[0].byte_count = size;
262250199Sgrehan	    for (i = 0; i < pfnCount; i++) {
263250199Sgrehan		gpa_header->range[0].pfn_array[i] = pfn + i;
264250199Sgrehan	    }
265250199Sgrehan	    *msg_info = msg_header;
266250199Sgrehan	    *message_count = 1;
267250199Sgrehan
268250199Sgrehan	    pfnSum = pfnCount;
269250199Sgrehan	    pfnLeft = page_count - pfnCount;
270250199Sgrehan
271250199Sgrehan	    /*
272250199Sgrehan	     *  figure out how many pfns we can fit
273250199Sgrehan	     */
274250199Sgrehan	    pfnSize = HV_MAX_SIZE_CHANNEL_MESSAGE
275250199Sgrehan		- sizeof(hv_vmbus_channel_gpadl_body);
276250199Sgrehan	    pfnCount = pfnSize / sizeof(uint64_t);
277250199Sgrehan
278250199Sgrehan	    /*
279250199Sgrehan	     * fill in the body
280250199Sgrehan	     */
281250199Sgrehan	    while (pfnLeft) {
282250199Sgrehan		if (pfnLeft > pfnCount) {
283250199Sgrehan		    pfnCurr = pfnCount;
284250199Sgrehan		} else {
285250199Sgrehan		    pfnCurr = pfnLeft;
286250199Sgrehan		}
287250199Sgrehan
288250199Sgrehan		msg_size = sizeof(hv_vmbus_channel_msg_info) +
289250199Sgrehan		    sizeof(hv_vmbus_channel_gpadl_body) +
290250199Sgrehan		    pfnCurr * sizeof(uint64_t);
291250199Sgrehan		msg_body = malloc(msg_size, M_DEVBUF, M_NOWAIT | M_ZERO);
292250199Sgrehan		KASSERT(
293250199Sgrehan		    msg_body != NULL,
294250199Sgrehan		    ("Error VMBUS: malloc failed to allocate Gpadl msg_body!"));
295250199Sgrehan		if (msg_body == NULL)
296250199Sgrehan		    return (ENOMEM);
297250199Sgrehan
298250199Sgrehan		msg_body->message_size = msg_size;
299250199Sgrehan		(*message_count)++;
300250199Sgrehan		gpadl_body =
301250199Sgrehan		    (hv_vmbus_channel_gpadl_body*) msg_body->msg;
302250199Sgrehan		/*
303250199Sgrehan		 * gpadl_body->gpadl = kbuffer;
304250199Sgrehan		 */
305250199Sgrehan		for (i = 0; i < pfnCurr; i++) {
306250199Sgrehan		    gpadl_body->pfn[i] = pfn + pfnSum + i;
307250199Sgrehan		}
308250199Sgrehan
309250199Sgrehan		TAILQ_INSERT_TAIL(
310250199Sgrehan		    &msg_header->sub_msg_list_anchor,
311250199Sgrehan		    msg_body,
312250199Sgrehan		    msg_list_entry);
313250199Sgrehan		pfnSum += pfnCurr;
314250199Sgrehan		pfnLeft -= pfnCurr;
315250199Sgrehan	    }
316250199Sgrehan	} else { /* else everything fits in a header */
317250199Sgrehan
318250199Sgrehan	    msg_size = sizeof(hv_vmbus_channel_msg_info) +
319250199Sgrehan		sizeof(hv_vmbus_channel_gpadl_header) +
320250199Sgrehan		sizeof(hv_gpa_range) +
321250199Sgrehan		page_count * sizeof(uint64_t);
322250199Sgrehan	    msg_header = malloc(msg_size, M_DEVBUF, M_NOWAIT | M_ZERO);
323250199Sgrehan	    KASSERT(
324250199Sgrehan		msg_header != NULL,
325250199Sgrehan		("Error VMBUS: malloc failed to allocate Gpadl Message!"));
326250199Sgrehan	    if (msg_header == NULL)
327250199Sgrehan		return (ENOMEM);
328250199Sgrehan
329250199Sgrehan	    msg_header->message_size = msg_size;
330250199Sgrehan
331250199Sgrehan	    gpa_header = (hv_vmbus_channel_gpadl_header*) msg_header->msg;
332250199Sgrehan	    gpa_header->range_count = 1;
333250199Sgrehan	    gpa_header->range_buf_len = sizeof(hv_gpa_range) +
334250199Sgrehan		page_count * sizeof(uint64_t);
335250199Sgrehan	    gpa_header->range[0].byte_offset = 0;
336250199Sgrehan	    gpa_header->range[0].byte_count = size;
337250199Sgrehan	    for (i = 0; i < page_count; i++) {
338250199Sgrehan		gpa_header->range[0].pfn_array[i] = pfn + i;
339250199Sgrehan	    }
340250199Sgrehan
341250199Sgrehan	    *msg_info = msg_header;
342250199Sgrehan	    *message_count = 1;
343250199Sgrehan	}
344250199Sgrehan
345250199Sgrehan	return (0);
346250199Sgrehan}
347250199Sgrehan
348250199Sgrehan/**
349250199Sgrehan * @brief Establish a GPADL for the specified buffer
350250199Sgrehan */
351250199Sgrehanint
352250199Sgrehanhv_vmbus_channel_establish_gpadl(
353250199Sgrehan	hv_vmbus_channel*	channel,
354250199Sgrehan	void*			contig_buffer,
355250199Sgrehan	uint32_t		size, /* page-size multiple */
356250199Sgrehan	uint32_t*		gpadl_handle)
357250199Sgrehan
358250199Sgrehan{
359250199Sgrehan	int ret = 0;
360250199Sgrehan	hv_vmbus_channel_gpadl_header*	gpadl_msg;
361250199Sgrehan	hv_vmbus_channel_gpadl_body*	gpadl_body;
362250199Sgrehan	hv_vmbus_channel_msg_info*	msg_info;
363250199Sgrehan	hv_vmbus_channel_msg_info*	sub_msg_info;
364250199Sgrehan	uint32_t			msg_count;
365250199Sgrehan	hv_vmbus_channel_msg_info*	curr;
366250199Sgrehan	uint32_t			next_gpadl_handle;
367250199Sgrehan
368250199Sgrehan	next_gpadl_handle = hv_vmbus_g_connection.next_gpadl_handle;
369250199Sgrehan	atomic_add_int((int*) &hv_vmbus_g_connection.next_gpadl_handle, 1);
370250199Sgrehan
371250199Sgrehan	ret = vmbus_channel_create_gpadl_header(
372250199Sgrehan		contig_buffer, size, &msg_info, &msg_count);
373250199Sgrehan
374250199Sgrehan	if(ret != 0) { /* if(allocation failed) return immediately */
375250199Sgrehan	    /* reverse atomic_add_int above */
376250199Sgrehan	    atomic_subtract_int((int*)
377250199Sgrehan		    &hv_vmbus_g_connection.next_gpadl_handle, 1);
378250199Sgrehan	    return ret;
379250199Sgrehan	}
380250199Sgrehan
381250199Sgrehan	sema_init(&msg_info->wait_sema, 0, "Open Info Sema");
382250199Sgrehan	gpadl_msg = (hv_vmbus_channel_gpadl_header*) msg_info->msg;
383250199Sgrehan	gpadl_msg->header.message_type = HV_CHANNEL_MESSAGEL_GPADL_HEADER;
384250199Sgrehan	gpadl_msg->child_rel_id = channel->offer_msg.child_rel_id;
385250199Sgrehan	gpadl_msg->gpadl = next_gpadl_handle;
386250199Sgrehan
387250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
388250199Sgrehan	TAILQ_INSERT_TAIL(
389250199Sgrehan		&hv_vmbus_g_connection.channel_msg_anchor,
390250199Sgrehan		msg_info,
391250199Sgrehan		msg_list_entry);
392250199Sgrehan
393250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
394250199Sgrehan
395250199Sgrehan	ret = hv_vmbus_post_message(
396250199Sgrehan		gpadl_msg,
397250199Sgrehan		msg_info->message_size -
398250199Sgrehan		    (uint32_t) sizeof(hv_vmbus_channel_msg_info));
399250199Sgrehan
400250199Sgrehan	if (ret != 0)
401250199Sgrehan	    goto cleanup;
402250199Sgrehan
403250199Sgrehan	if (msg_count > 1) {
404250199Sgrehan	    TAILQ_FOREACH(curr,
405250199Sgrehan		    &msg_info->sub_msg_list_anchor, msg_list_entry) {
406250199Sgrehan		sub_msg_info = curr;
407250199Sgrehan		gpadl_body =
408250199Sgrehan		    (hv_vmbus_channel_gpadl_body*) sub_msg_info->msg;
409250199Sgrehan
410250199Sgrehan		gpadl_body->header.message_type =
411250199Sgrehan		    HV_CHANNEL_MESSAGE_GPADL_BODY;
412250199Sgrehan		gpadl_body->gpadl = next_gpadl_handle;
413250199Sgrehan
414250199Sgrehan		ret = hv_vmbus_post_message(
415250199Sgrehan			gpadl_body,
416250199Sgrehan			sub_msg_info->message_size
417250199Sgrehan			    - (uint32_t) sizeof(hv_vmbus_channel_msg_info));
418250199Sgrehan		 /* if (the post message failed) give up and clean up */
419250199Sgrehan		if(ret != 0)
420250199Sgrehan		    goto cleanup;
421250199Sgrehan	    }
422250199Sgrehan	}
423250199Sgrehan
424250199Sgrehan	ret = sema_timedwait(&msg_info->wait_sema, 500); /* KYS 5 seconds*/
425250199Sgrehan	if (ret != 0)
426250199Sgrehan	    goto cleanup;
427250199Sgrehan
428250199Sgrehan	*gpadl_handle = gpadl_msg->gpadl;
429250199Sgrehan
430250199Sgrehancleanup:
431250199Sgrehan
432250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
433250199Sgrehan	TAILQ_REMOVE(&hv_vmbus_g_connection.channel_msg_anchor,
434250199Sgrehan		msg_info, msg_list_entry);
435250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
436250199Sgrehan
437250199Sgrehan	sema_destroy(&msg_info->wait_sema);
438250199Sgrehan	free(msg_info, M_DEVBUF);
439250199Sgrehan
440250199Sgrehan	return (ret);
441250199Sgrehan}
442250199Sgrehan
443250199Sgrehan/**
444250199Sgrehan * @brief Teardown the specified GPADL handle
445250199Sgrehan */
446250199Sgrehanint
447250199Sgrehanhv_vmbus_channel_teardown_gpdal(
448250199Sgrehan	hv_vmbus_channel*	channel,
449250199Sgrehan	uint32_t		gpadl_handle)
450250199Sgrehan{
451250199Sgrehan	int					ret = 0;
452250199Sgrehan	hv_vmbus_channel_gpadl_teardown*	msg;
453250199Sgrehan	hv_vmbus_channel_msg_info*		info;
454250199Sgrehan
455250199Sgrehan	info = (hv_vmbus_channel_msg_info *)
456250199Sgrehan		malloc(	sizeof(hv_vmbus_channel_msg_info) +
457250199Sgrehan			sizeof(hv_vmbus_channel_gpadl_teardown),
458250199Sgrehan				M_DEVBUF, M_NOWAIT);
459250199Sgrehan	KASSERT(info != NULL,
460250199Sgrehan	    ("Error VMBUS: malloc failed to allocate Gpadl Teardown Msg!"));
461250199Sgrehan	if (info == NULL) {
462250199Sgrehan	    ret = ENOMEM;
463250199Sgrehan	    goto cleanup;
464250199Sgrehan	}
465250199Sgrehan
466250199Sgrehan	sema_init(&info->wait_sema, 0, "Open Info Sema");
467250199Sgrehan
468250199Sgrehan	msg = (hv_vmbus_channel_gpadl_teardown*) info->msg;
469250199Sgrehan
470250199Sgrehan	msg->header.message_type = HV_CHANNEL_MESSAGE_GPADL_TEARDOWN;
471250199Sgrehan	msg->child_rel_id = channel->offer_msg.child_rel_id;
472250199Sgrehan	msg->gpadl = gpadl_handle;
473250199Sgrehan
474250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
475250199Sgrehan	TAILQ_INSERT_TAIL(&hv_vmbus_g_connection.channel_msg_anchor,
476250199Sgrehan			info, msg_list_entry);
477250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
478250199Sgrehan
479250199Sgrehan	ret = hv_vmbus_post_message(msg,
480250199Sgrehan			sizeof(hv_vmbus_channel_gpadl_teardown));
481250199Sgrehan	if (ret != 0)
482250199Sgrehan	    goto cleanup;
483250199Sgrehan
484250199Sgrehan	ret = sema_timedwait(&info->wait_sema, 500); /* KYS 5 seconds */
485250199Sgrehan
486250199Sgrehancleanup:
487250199Sgrehan	/*
488250199Sgrehan	 * Received a torndown response
489250199Sgrehan	 */
490250199Sgrehan	mtx_lock_spin(&hv_vmbus_g_connection.channel_msg_lock);
491250199Sgrehan	TAILQ_REMOVE(&hv_vmbus_g_connection.channel_msg_anchor,
492250199Sgrehan			info, msg_list_entry);
493250199Sgrehan	mtx_unlock_spin(&hv_vmbus_g_connection.channel_msg_lock);
494250199Sgrehan	sema_destroy(&info->wait_sema);
495250199Sgrehan	free(info, M_DEVBUF);
496250199Sgrehan
497250199Sgrehan	return (ret);
498250199Sgrehan}
499250199Sgrehan
500250199Sgrehan/**
501250199Sgrehan * @brief Close the specified channel
502250199Sgrehan */
503250199Sgrehanvoid
504250199Sgrehanhv_vmbus_channel_close(hv_vmbus_channel *channel)
505250199Sgrehan{
506250199Sgrehan	int ret = 0;
507250199Sgrehan	hv_vmbus_channel_close_channel* msg;
508250199Sgrehan	hv_vmbus_channel_msg_info* info;
509250199Sgrehan
510250199Sgrehan	mtx_lock(&channel->inbound_lock);
511250199Sgrehan	channel->on_channel_callback = NULL;
512250199Sgrehan	mtx_unlock(&channel->inbound_lock);
513250199Sgrehan
514250199Sgrehan	/**
515250199Sgrehan	 * Send a closing message
516250199Sgrehan	 */
517250199Sgrehan	info = (hv_vmbus_channel_msg_info *)
518250199Sgrehan		malloc(	sizeof(hv_vmbus_channel_msg_info) +
519250199Sgrehan			sizeof(hv_vmbus_channel_close_channel),
520250199Sgrehan				M_DEVBUF, M_NOWAIT);
521250199Sgrehan	KASSERT(info != NULL, ("VMBUS: malloc failed hv_vmbus_channel_close!"));
522250199Sgrehan	if(info == NULL)
523250199Sgrehan	    return;
524250199Sgrehan
525250199Sgrehan	msg = (hv_vmbus_channel_close_channel*) info->msg;
526250199Sgrehan	msg->header.message_type = HV_CHANNEL_MESSAGE_CLOSE_CHANNEL;
527250199Sgrehan	msg->child_rel_id = channel->offer_msg.child_rel_id;
528250199Sgrehan
529250199Sgrehan	ret = hv_vmbus_post_message(
530250199Sgrehan		msg, sizeof(hv_vmbus_channel_close_channel));
531250199Sgrehan
532250199Sgrehan	/* Tear down the gpadl for the channel's ring buffer */
533250199Sgrehan	if (channel->ring_buffer_gpadl_handle) {
534250199Sgrehan		hv_vmbus_channel_teardown_gpdal(channel,
535250199Sgrehan			channel->ring_buffer_gpadl_handle);
536250199Sgrehan	}
537250199Sgrehan
538250199Sgrehan	/* TODO: Send a msg to release the childRelId */
539250199Sgrehan
540250199Sgrehan	/* cleanup the ring buffers for this channel */
541250199Sgrehan	hv_ring_buffer_cleanup(&channel->outbound);
542250199Sgrehan	hv_ring_buffer_cleanup(&channel->inbound);
543250199Sgrehan
544256352Sgrehan	contigfree(channel->ring_buffer_pages, channel->ring_buffer_size,
545256352Sgrehan	    M_DEVBUF);
546250199Sgrehan
547250199Sgrehan	free(info, M_DEVBUF);
548250199Sgrehan
549250199Sgrehan	/*
550250199Sgrehan	 *  If we are closing the channel during an error path in
551250199Sgrehan	 *  opening the channel, don't free the channel
552250199Sgrehan	 *  since the caller will free the channel
553250199Sgrehan	 */
554250199Sgrehan	if (channel->state == HV_CHANNEL_OPEN_STATE) {
555250199Sgrehan		mtx_lock_spin(&hv_vmbus_g_connection.channel_lock);
556250199Sgrehan		TAILQ_REMOVE(
557250199Sgrehan			&hv_vmbus_g_connection.channel_anchor,
558250199Sgrehan			channel,
559250199Sgrehan			list_entry);
560250199Sgrehan		mtx_unlock_spin(&hv_vmbus_g_connection.channel_lock);
561250199Sgrehan
562250199Sgrehan		hv_vmbus_free_vmbus_channel(channel);
563250199Sgrehan	}
564250199Sgrehan
565250199Sgrehan}
566250199Sgrehan
567250199Sgrehan/**
568250199Sgrehan * @brief Send the specified buffer on the given channel
569250199Sgrehan */
570250199Sgrehanint
571250199Sgrehanhv_vmbus_channel_send_packet(
572250199Sgrehan	hv_vmbus_channel*	channel,
573250199Sgrehan	void*			buffer,
574250199Sgrehan	uint32_t		buffer_len,
575250199Sgrehan	uint64_t		request_id,
576250199Sgrehan	hv_vmbus_packet_type	type,
577250199Sgrehan	uint32_t		flags)
578250199Sgrehan{
579250199Sgrehan	int			ret = 0;
580250199Sgrehan	hv_vm_packet_descriptor	desc;
581250199Sgrehan	uint32_t		packet_len;
582250199Sgrehan	uint64_t		aligned_data;
583250199Sgrehan	uint32_t		packet_len_aligned;
584250199Sgrehan	hv_vmbus_sg_buffer_list	buffer_list[3];
585250199Sgrehan
586250199Sgrehan	packet_len = sizeof(hv_vm_packet_descriptor) + buffer_len;
587250199Sgrehan	packet_len_aligned = HV_ALIGN_UP(packet_len, sizeof(uint64_t));
588250199Sgrehan	aligned_data = 0;
589250199Sgrehan
590250199Sgrehan	/* Setup the descriptor */
591250199Sgrehan	desc.type = type;   /* HV_VMBUS_PACKET_TYPE_DATA_IN_BAND;             */
592250199Sgrehan	desc.flags = flags; /* HV_VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED */
593250199Sgrehan			    /* in 8-bytes granularity */
594250199Sgrehan	desc.data_offset8 = sizeof(hv_vm_packet_descriptor) >> 3;
595250199Sgrehan	desc.length8 = (uint16_t) (packet_len_aligned >> 3);
596250199Sgrehan	desc.transaction_id = request_id;
597250199Sgrehan
598250199Sgrehan	buffer_list[0].data = &desc;
599250199Sgrehan	buffer_list[0].length = sizeof(hv_vm_packet_descriptor);
600250199Sgrehan
601250199Sgrehan	buffer_list[1].data = buffer;
602250199Sgrehan	buffer_list[1].length = buffer_len;
603250199Sgrehan
604250199Sgrehan	buffer_list[2].data = &aligned_data;
605250199Sgrehan	buffer_list[2].length = packet_len_aligned - packet_len;
606250199Sgrehan
607250199Sgrehan	ret = hv_ring_buffer_write(&channel->outbound, buffer_list, 3);
608250199Sgrehan
609250199Sgrehan	/* TODO: We should determine if this is optional */
610250199Sgrehan	if (ret == 0
611250199Sgrehan		&& !hv_vmbus_get_ring_buffer_interrupt_mask(
612250199Sgrehan			&channel->outbound)) {
613250199Sgrehan		vmbus_channel_set_event(channel);
614250199Sgrehan	}
615250199Sgrehan
616250199Sgrehan	return (ret);
617250199Sgrehan}
618250199Sgrehan
619250199Sgrehan/**
620250199Sgrehan * @brief Send a range of single-page buffer packets using
621250199Sgrehan * a GPADL Direct packet type
622250199Sgrehan */
623250199Sgrehanint
624250199Sgrehanhv_vmbus_channel_send_packet_pagebuffer(
625250199Sgrehan	hv_vmbus_channel*	channel,
626250199Sgrehan	hv_vmbus_page_buffer	page_buffers[],
627250199Sgrehan	uint32_t		page_count,
628250199Sgrehan	void*			buffer,
629250199Sgrehan	uint32_t		buffer_len,
630250199Sgrehan	uint64_t		request_id)
631250199Sgrehan{
632250199Sgrehan
633250199Sgrehan	int					ret = 0;
634250199Sgrehan	int					i = 0;
635250199Sgrehan	uint32_t				packet_len;
636250199Sgrehan	uint32_t				packetLen_aligned;
637250199Sgrehan	hv_vmbus_sg_buffer_list			buffer_list[3];
638250199Sgrehan	hv_vmbus_channel_packet_page_buffer	desc;
639250199Sgrehan	uint32_t				descSize;
640250199Sgrehan	uint64_t				alignedData = 0;
641250199Sgrehan
642250199Sgrehan	if (page_count > HV_MAX_PAGE_BUFFER_COUNT)
643250199Sgrehan		return (EINVAL);
644250199Sgrehan
645250199Sgrehan	/*
646250199Sgrehan	 * Adjust the size down since hv_vmbus_channel_packet_page_buffer
647250199Sgrehan	 *  is the largest size we support
648250199Sgrehan	 */
649250199Sgrehan	descSize = sizeof(hv_vmbus_channel_packet_page_buffer) -
650250199Sgrehan			((HV_MAX_PAGE_BUFFER_COUNT - page_count) *
651250199Sgrehan			sizeof(hv_vmbus_page_buffer));
652250199Sgrehan	packet_len = descSize + buffer_len;
653250199Sgrehan	packetLen_aligned = HV_ALIGN_UP(packet_len, sizeof(uint64_t));
654250199Sgrehan
655250199Sgrehan	/* Setup the descriptor */
656250199Sgrehan	desc.type = HV_VMBUS_PACKET_TYPE_DATA_USING_GPA_DIRECT;
657250199Sgrehan	desc.flags = HV_VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
658250199Sgrehan	desc.data_offset8 = descSize >> 3; /* in 8-bytes granularity */
659250199Sgrehan	desc.length8 = (uint16_t) (packetLen_aligned >> 3);
660250199Sgrehan	desc.transaction_id = request_id;
661250199Sgrehan	desc.range_count = page_count;
662250199Sgrehan
663250199Sgrehan	for (i = 0; i < page_count; i++) {
664250199Sgrehan		desc.range[i].length = page_buffers[i].length;
665250199Sgrehan		desc.range[i].offset = page_buffers[i].offset;
666250199Sgrehan		desc.range[i].pfn = page_buffers[i].pfn;
667250199Sgrehan	}
668250199Sgrehan
669250199Sgrehan	buffer_list[0].data = &desc;
670250199Sgrehan	buffer_list[0].length = descSize;
671250199Sgrehan
672250199Sgrehan	buffer_list[1].data = buffer;
673250199Sgrehan	buffer_list[1].length = buffer_len;
674250199Sgrehan
675250199Sgrehan	buffer_list[2].data = &alignedData;
676250199Sgrehan	buffer_list[2].length = packetLen_aligned - packet_len;
677250199Sgrehan
678250199Sgrehan	ret = hv_ring_buffer_write(&channel->outbound, buffer_list, 3);
679250199Sgrehan
680250199Sgrehan	/* TODO: We should determine if this is optional */
681250199Sgrehan	if (ret == 0 &&
682250199Sgrehan		!hv_vmbus_get_ring_buffer_interrupt_mask(&channel->outbound)) {
683250199Sgrehan		vmbus_channel_set_event(channel);
684250199Sgrehan	}
685250199Sgrehan
686250199Sgrehan	return (ret);
687250199Sgrehan}
688250199Sgrehan
689250199Sgrehan/**
690250199Sgrehan * @brief Send a multi-page buffer packet using a GPADL Direct packet type
691250199Sgrehan */
692250199Sgrehanint
693250199Sgrehanhv_vmbus_channel_send_packet_multipagebuffer(
694250199Sgrehan	hv_vmbus_channel*		channel,
695250199Sgrehan	hv_vmbus_multipage_buffer*	multi_page_buffer,
696250199Sgrehan	void*				buffer,
697250199Sgrehan	uint32_t			buffer_len,
698250199Sgrehan	uint64_t			request_id)
699250199Sgrehan{
700250199Sgrehan
701250199Sgrehan	int			ret = 0;
702250199Sgrehan	uint32_t		desc_size;
703250199Sgrehan	uint32_t		packet_len;
704250199Sgrehan	uint32_t		packet_len_aligned;
705250199Sgrehan	uint32_t		pfn_count;
706250199Sgrehan	uint64_t		aligned_data = 0;
707250199Sgrehan	hv_vmbus_sg_buffer_list	buffer_list[3];
708250199Sgrehan	hv_vmbus_channel_packet_multipage_buffer desc;
709250199Sgrehan
710250199Sgrehan	pfn_count =
711250199Sgrehan	    HV_NUM_PAGES_SPANNED(
712250199Sgrehan		    multi_page_buffer->offset,
713250199Sgrehan		    multi_page_buffer->length);
714250199Sgrehan
715250199Sgrehan	if ((pfn_count == 0) || (pfn_count > HV_MAX_MULTIPAGE_BUFFER_COUNT))
716250199Sgrehan	    return (EINVAL);
717250199Sgrehan	/*
718250199Sgrehan	 * Adjust the size down since hv_vmbus_channel_packet_multipage_buffer
719250199Sgrehan	 * is the largest size we support
720250199Sgrehan	 */
721250199Sgrehan	desc_size =
722250199Sgrehan	    sizeof(hv_vmbus_channel_packet_multipage_buffer) -
723250199Sgrehan		    ((HV_MAX_MULTIPAGE_BUFFER_COUNT - pfn_count) *
724250199Sgrehan			sizeof(uint64_t));
725250199Sgrehan	packet_len = desc_size + buffer_len;
726250199Sgrehan	packet_len_aligned = HV_ALIGN_UP(packet_len, sizeof(uint64_t));
727250199Sgrehan
728250199Sgrehan	/*
729250199Sgrehan	 * Setup the descriptor
730250199Sgrehan	 */
731250199Sgrehan	desc.type = HV_VMBUS_PACKET_TYPE_DATA_USING_GPA_DIRECT;
732250199Sgrehan	desc.flags = HV_VMBUS_DATA_PACKET_FLAG_COMPLETION_REQUESTED;
733250199Sgrehan	desc.data_offset8 = desc_size >> 3; /* in 8-bytes granularity */
734250199Sgrehan	desc.length8 = (uint16_t) (packet_len_aligned >> 3);
735250199Sgrehan	desc.transaction_id = request_id;
736250199Sgrehan	desc.range_count = 1;
737250199Sgrehan
738250199Sgrehan	desc.range.length = multi_page_buffer->length;
739250199Sgrehan	desc.range.offset = multi_page_buffer->offset;
740250199Sgrehan
741250199Sgrehan	memcpy(desc.range.pfn_array, multi_page_buffer->pfn_array,
742250199Sgrehan		pfn_count * sizeof(uint64_t));
743250199Sgrehan
744250199Sgrehan	buffer_list[0].data = &desc;
745250199Sgrehan	buffer_list[0].length = desc_size;
746250199Sgrehan
747250199Sgrehan	buffer_list[1].data = buffer;
748250199Sgrehan	buffer_list[1].length = buffer_len;
749250199Sgrehan
750250199Sgrehan	buffer_list[2].data = &aligned_data;
751250199Sgrehan	buffer_list[2].length = packet_len_aligned - packet_len;
752250199Sgrehan
753250199Sgrehan	ret = hv_ring_buffer_write(&channel->outbound, buffer_list, 3);
754250199Sgrehan
755250199Sgrehan	/* TODO: We should determine if this is optional */
756250199Sgrehan	if (ret == 0 &&
757250199Sgrehan	    !hv_vmbus_get_ring_buffer_interrupt_mask(&channel->outbound)) {
758250199Sgrehan	    vmbus_channel_set_event(channel);
759250199Sgrehan	}
760250199Sgrehan
761250199Sgrehan	return (ret);
762250199Sgrehan}
763250199Sgrehan
764250199Sgrehan/**
765250199Sgrehan * @brief Retrieve the user packet on the specified channel
766250199Sgrehan */
767250199Sgrehanint
768250199Sgrehanhv_vmbus_channel_recv_packet(
769250199Sgrehan	hv_vmbus_channel*	channel,
770250199Sgrehan	void*			Buffer,
771250199Sgrehan	uint32_t		buffer_len,
772250199Sgrehan	uint32_t*		buffer_actual_len,
773250199Sgrehan	uint64_t*		request_id)
774250199Sgrehan{
775250199Sgrehan	int			ret;
776250199Sgrehan	uint32_t		user_len;
777250199Sgrehan	uint32_t		packet_len;
778250199Sgrehan	hv_vm_packet_descriptor	desc;
779250199Sgrehan
780250199Sgrehan	*buffer_actual_len = 0;
781250199Sgrehan	*request_id = 0;
782250199Sgrehan
783250199Sgrehan	ret = hv_ring_buffer_peek(&channel->inbound, &desc,
784250199Sgrehan		sizeof(hv_vm_packet_descriptor));
785250199Sgrehan	if (ret != 0)
786250199Sgrehan		return (0);
787250199Sgrehan
788250199Sgrehan	packet_len = desc.length8 << 3;
789250199Sgrehan	user_len = packet_len - (desc.data_offset8 << 3);
790250199Sgrehan
791250199Sgrehan	*buffer_actual_len = user_len;
792250199Sgrehan
793250199Sgrehan	if (user_len > buffer_len)
794250199Sgrehan		return (EINVAL);
795250199Sgrehan
796250199Sgrehan	*request_id = desc.transaction_id;
797250199Sgrehan
798250199Sgrehan	/* Copy over the packet to the user buffer */
799250199Sgrehan	ret = hv_ring_buffer_read(&channel->inbound, Buffer, user_len,
800250199Sgrehan		(desc.data_offset8 << 3));
801250199Sgrehan
802250199Sgrehan	return (0);
803250199Sgrehan}
804250199Sgrehan
805250199Sgrehan/**
806250199Sgrehan * @brief Retrieve the raw packet on the specified channel
807250199Sgrehan */
808250199Sgrehanint
809250199Sgrehanhv_vmbus_channel_recv_packet_raw(
810250199Sgrehan	hv_vmbus_channel*	channel,
811250199Sgrehan	void*			buffer,
812250199Sgrehan	uint32_t		buffer_len,
813250199Sgrehan	uint32_t*		buffer_actual_len,
814250199Sgrehan	uint64_t*		request_id)
815250199Sgrehan{
816250199Sgrehan	int		ret;
817250199Sgrehan	uint32_t	packetLen;
818250199Sgrehan	uint32_t	userLen;
819250199Sgrehan	hv_vm_packet_descriptor	desc;
820250199Sgrehan
821250199Sgrehan	*buffer_actual_len = 0;
822250199Sgrehan	*request_id = 0;
823250199Sgrehan
824250199Sgrehan	ret = hv_ring_buffer_peek(
825250199Sgrehan		&channel->inbound, &desc,
826250199Sgrehan		sizeof(hv_vm_packet_descriptor));
827250199Sgrehan
828250199Sgrehan	if (ret != 0)
829250199Sgrehan	    return (0);
830250199Sgrehan
831250199Sgrehan	packetLen = desc.length8 << 3;
832250199Sgrehan	userLen = packetLen - (desc.data_offset8 << 3);
833250199Sgrehan
834250199Sgrehan	*buffer_actual_len = packetLen;
835250199Sgrehan
836250199Sgrehan	if (packetLen > buffer_len)
837250199Sgrehan	    return (ENOBUFS);
838250199Sgrehan
839250199Sgrehan	*request_id = desc.transaction_id;
840250199Sgrehan
841250199Sgrehan	/* Copy over the entire packet to the user buffer */
842250199Sgrehan	ret = hv_ring_buffer_read(&channel->inbound, buffer, packetLen, 0);
843250199Sgrehan
844250199Sgrehan	return (0);
845250199Sgrehan}
846