vmbus_chan.c revision 308517
1/*-
2 * Copyright (c) 2009-2012,2016 Microsoft Corp.
3 * Copyright (c) 2012 NetApp Inc.
4 * Copyright (c) 2012 Citrix Inc.
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice unmodified, this list of conditions, and the following
12 *    disclaimer.
13 * 2. Redistributions in binary form must reproduce the above copyright
14 *    notice, this list of conditions and the following disclaimer in the
15 *    documentation and/or other materials provided with the distribution.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
18 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
19 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
20 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
21 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
22 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
26 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
27 */
28
29#include <sys/cdefs.h>
30__FBSDID("$FreeBSD: stable/11/sys/dev/hyperv/vmbus/vmbus_chan.c 308517 2016-11-11 08:43:34Z sephe $");
31
32#include <sys/param.h>
33#include <sys/kernel.h>
34#include <sys/lock.h>
35#include <sys/malloc.h>
36#include <sys/mutex.h>
37#include <sys/smp.h>
38#include <sys/sysctl.h>
39#include <sys/systm.h>
40
41#include <machine/atomic.h>
42
43#include <dev/hyperv/include/hyperv_busdma.h>
44#include <dev/hyperv/vmbus/hyperv_var.h>
45#include <dev/hyperv/vmbus/vmbus_reg.h>
46#include <dev/hyperv/vmbus/vmbus_var.h>
47#include <dev/hyperv/vmbus/vmbus_brvar.h>
48#include <dev/hyperv/vmbus/vmbus_chanvar.h>
49
50static void			vmbus_chan_update_evtflagcnt(
51				    struct vmbus_softc *,
52				    const struct vmbus_channel *);
53static void			vmbus_chan_close_internal(
54				    struct vmbus_channel *);
55static int			vmbus_chan_sysctl_mnf(SYSCTL_HANDLER_ARGS);
56static void			vmbus_chan_sysctl_create(
57				    struct vmbus_channel *);
58static struct vmbus_channel	*vmbus_chan_alloc(struct vmbus_softc *);
59static void			vmbus_chan_free(struct vmbus_channel *);
60static int			vmbus_chan_add(struct vmbus_channel *);
61static void			vmbus_chan_cpu_default(struct vmbus_channel *);
62static int			vmbus_chan_release(struct vmbus_channel *);
63static void			vmbus_chan_set_chmap(struct vmbus_channel *);
64static void			vmbus_chan_clear_chmap(struct vmbus_channel *);
65
66static void			vmbus_chan_ins_prilist(struct vmbus_softc *,
67				    struct vmbus_channel *);
68static void			vmbus_chan_rem_prilist(struct vmbus_softc *,
69				    struct vmbus_channel *);
70static void			vmbus_chan_ins_list(struct vmbus_softc *,
71				    struct vmbus_channel *);
72static void			vmbus_chan_rem_list(struct vmbus_softc *,
73				    struct vmbus_channel *);
74static void			vmbus_chan_ins_sublist(struct vmbus_channel *,
75				    struct vmbus_channel *);
76static void			vmbus_chan_rem_sublist(struct vmbus_channel *,
77				    struct vmbus_channel *);
78
79static void			vmbus_chan_task(void *, int);
80static void			vmbus_chan_task_nobatch(void *, int);
81static void			vmbus_chan_clrchmap_task(void *, int);
82static void			vmbus_prichan_attach_task(void *, int);
83static void			vmbus_subchan_attach_task(void *, int);
84static void			vmbus_prichan_detach_task(void *, int);
85static void			vmbus_subchan_detach_task(void *, int);
86
87static void			vmbus_chan_msgproc_choffer(struct vmbus_softc *,
88				    const struct vmbus_message *);
89static void			vmbus_chan_msgproc_chrescind(
90				    struct vmbus_softc *,
91				    const struct vmbus_message *);
92
93/*
94 * Vmbus channel message processing.
95 */
96static const vmbus_chanmsg_proc_t
97vmbus_chan_msgprocs[VMBUS_CHANMSG_TYPE_MAX] = {
98	VMBUS_CHANMSG_PROC(CHOFFER,	vmbus_chan_msgproc_choffer),
99	VMBUS_CHANMSG_PROC(CHRESCIND,	vmbus_chan_msgproc_chrescind),
100
101	VMBUS_CHANMSG_PROC_WAKEUP(CHOPEN_RESP),
102	VMBUS_CHANMSG_PROC_WAKEUP(GPADL_CONNRESP),
103	VMBUS_CHANMSG_PROC_WAKEUP(GPADL_DISCONNRESP)
104};
105
106/*
107 * Notify host that there are data pending on our TX bufring.
108 */
109static __inline void
110vmbus_chan_signal_tx(const struct vmbus_channel *chan)
111{
112	atomic_set_long(chan->ch_evtflag, chan->ch_evtflag_mask);
113	if (chan->ch_txflags & VMBUS_CHAN_TXF_HASMNF)
114		atomic_set_int(chan->ch_montrig, chan->ch_montrig_mask);
115	else
116		hypercall_signal_event(chan->ch_monprm_dma.hv_paddr);
117}
118
119static void
120vmbus_chan_ins_prilist(struct vmbus_softc *sc, struct vmbus_channel *chan)
121{
122
123	mtx_assert(&sc->vmbus_prichan_lock, MA_OWNED);
124	if (atomic_testandset_int(&chan->ch_stflags,
125	    VMBUS_CHAN_ST_ONPRIL_SHIFT))
126		panic("channel is already on the prilist");
127	TAILQ_INSERT_TAIL(&sc->vmbus_prichans, chan, ch_prilink);
128}
129
130static void
131vmbus_chan_rem_prilist(struct vmbus_softc *sc, struct vmbus_channel *chan)
132{
133
134	mtx_assert(&sc->vmbus_prichan_lock, MA_OWNED);
135	if (atomic_testandclear_int(&chan->ch_stflags,
136	    VMBUS_CHAN_ST_ONPRIL_SHIFT) == 0)
137		panic("channel is not on the prilist");
138	TAILQ_REMOVE(&sc->vmbus_prichans, chan, ch_prilink);
139}
140
141static void
142vmbus_chan_ins_sublist(struct vmbus_channel *prichan,
143    struct vmbus_channel *chan)
144{
145
146	mtx_assert(&prichan->ch_subchan_lock, MA_OWNED);
147
148	if (atomic_testandset_int(&chan->ch_stflags,
149	    VMBUS_CHAN_ST_ONSUBL_SHIFT))
150		panic("channel is already on the sublist");
151	TAILQ_INSERT_TAIL(&prichan->ch_subchans, chan, ch_sublink);
152
153	/* Bump sub-channel count. */
154	prichan->ch_subchan_cnt++;
155}
156
157static void
158vmbus_chan_rem_sublist(struct vmbus_channel *prichan,
159    struct vmbus_channel *chan)
160{
161
162	mtx_assert(&prichan->ch_subchan_lock, MA_OWNED);
163
164	KASSERT(prichan->ch_subchan_cnt > 0,
165	    ("invalid subchan_cnt %d", prichan->ch_subchan_cnt));
166	prichan->ch_subchan_cnt--;
167
168	if (atomic_testandclear_int(&chan->ch_stflags,
169	    VMBUS_CHAN_ST_ONSUBL_SHIFT) == 0)
170		panic("channel is not on the sublist");
171	TAILQ_REMOVE(&prichan->ch_subchans, chan, ch_sublink);
172}
173
174static void
175vmbus_chan_ins_list(struct vmbus_softc *sc, struct vmbus_channel *chan)
176{
177
178	mtx_assert(&sc->vmbus_chan_lock, MA_OWNED);
179	if (atomic_testandset_int(&chan->ch_stflags,
180	    VMBUS_CHAN_ST_ONLIST_SHIFT))
181		panic("channel is already on the list");
182	TAILQ_INSERT_TAIL(&sc->vmbus_chans, chan, ch_link);
183}
184
185static void
186vmbus_chan_rem_list(struct vmbus_softc *sc, struct vmbus_channel *chan)
187{
188
189	mtx_assert(&sc->vmbus_chan_lock, MA_OWNED);
190	if (atomic_testandclear_int(&chan->ch_stflags,
191	    VMBUS_CHAN_ST_ONLIST_SHIFT) == 0)
192		panic("channel is not on the list");
193	TAILQ_REMOVE(&sc->vmbus_chans, chan, ch_link);
194}
195
196static int
197vmbus_chan_sysctl_mnf(SYSCTL_HANDLER_ARGS)
198{
199	struct vmbus_channel *chan = arg1;
200	int mnf = 0;
201
202	if (chan->ch_txflags & VMBUS_CHAN_TXF_HASMNF)
203		mnf = 1;
204	return sysctl_handle_int(oidp, &mnf, 0, req);
205}
206
207static void
208vmbus_chan_sysctl_create(struct vmbus_channel *chan)
209{
210	struct sysctl_oid *ch_tree, *chid_tree, *br_tree;
211	struct sysctl_ctx_list *ctx;
212	uint32_t ch_id;
213	char name[16];
214
215	/*
216	 * Add sysctl nodes related to this channel to this
217	 * channel's sysctl ctx, so that they can be destroyed
218	 * independently upon close of this channel, which can
219	 * happen even if the device is not detached.
220	 */
221	ctx = &chan->ch_sysctl_ctx;
222	sysctl_ctx_init(ctx);
223
224	/*
225	 * Create dev.NAME.UNIT.channel tree.
226	 */
227	ch_tree = SYSCTL_ADD_NODE(ctx,
228	    SYSCTL_CHILDREN(device_get_sysctl_tree(chan->ch_dev)),
229	    OID_AUTO, "channel", CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "");
230	if (ch_tree == NULL)
231		return;
232
233	/*
234	 * Create dev.NAME.UNIT.channel.CHANID tree.
235	 */
236	if (VMBUS_CHAN_ISPRIMARY(chan))
237		ch_id = chan->ch_id;
238	else
239		ch_id = chan->ch_prichan->ch_id;
240	snprintf(name, sizeof(name), "%d", ch_id);
241	chid_tree = SYSCTL_ADD_NODE(ctx, SYSCTL_CHILDREN(ch_tree),
242	    OID_AUTO, name, CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "");
243	if (chid_tree == NULL)
244		return;
245
246	if (!VMBUS_CHAN_ISPRIMARY(chan)) {
247		/*
248		 * Create dev.NAME.UNIT.channel.CHANID.sub tree.
249		 */
250		ch_tree = SYSCTL_ADD_NODE(ctx, SYSCTL_CHILDREN(chid_tree),
251		    OID_AUTO, "sub", CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "");
252		if (ch_tree == NULL)
253			return;
254
255		/*
256		 * Create dev.NAME.UNIT.channel.CHANID.sub.SUBIDX tree.
257		 *
258		 * NOTE:
259		 * chid_tree is changed to this new sysctl tree.
260		 */
261		snprintf(name, sizeof(name), "%d", chan->ch_subidx);
262		chid_tree = SYSCTL_ADD_NODE(ctx, SYSCTL_CHILDREN(ch_tree),
263		    OID_AUTO, name, CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "");
264		if (chid_tree == NULL)
265			return;
266
267		SYSCTL_ADD_UINT(ctx, SYSCTL_CHILDREN(chid_tree), OID_AUTO,
268		    "chanid", CTLFLAG_RD, &chan->ch_id, 0, "channel id");
269	}
270
271	SYSCTL_ADD_UINT(ctx, SYSCTL_CHILDREN(chid_tree), OID_AUTO,
272	    "cpu", CTLFLAG_RD, &chan->ch_cpuid, 0, "owner CPU id");
273	SYSCTL_ADD_PROC(ctx, SYSCTL_CHILDREN(chid_tree), OID_AUTO,
274	    "mnf", CTLTYPE_INT | CTLFLAG_RD | CTLFLAG_MPSAFE,
275	    chan, 0, vmbus_chan_sysctl_mnf, "I",
276	    "has monitor notification facilities");
277
278	br_tree = SYSCTL_ADD_NODE(ctx, SYSCTL_CHILDREN(chid_tree), OID_AUTO,
279	    "br", CTLFLAG_RD | CTLFLAG_MPSAFE, 0, "");
280	if (br_tree != NULL) {
281		/*
282		 * Create sysctl tree for RX bufring.
283		 */
284		vmbus_br_sysctl_create(ctx, br_tree, &chan->ch_rxbr.rxbr, "rx");
285		/*
286		 * Create sysctl tree for TX bufring.
287		 */
288		vmbus_br_sysctl_create(ctx, br_tree, &chan->ch_txbr.txbr, "tx");
289	}
290}
291
292int
293vmbus_chan_open(struct vmbus_channel *chan, int txbr_size, int rxbr_size,
294    const void *udata, int udlen, vmbus_chan_callback_t cb, void *cbarg)
295{
296	struct vmbus_chan_br cbr;
297	int error;
298
299	/*
300	 * Allocate the TX+RX bufrings.
301	 */
302	KASSERT(chan->ch_bufring == NULL, ("bufrings are allocated"));
303	chan->ch_bufring = hyperv_dmamem_alloc(bus_get_dma_tag(chan->ch_dev),
304	    PAGE_SIZE, 0, txbr_size + rxbr_size, &chan->ch_bufring_dma,
305	    BUS_DMA_WAITOK);
306	if (chan->ch_bufring == NULL) {
307		device_printf(chan->ch_dev, "bufring allocation failed\n");
308		return (ENOMEM);
309	}
310
311	cbr.cbr = chan->ch_bufring;
312	cbr.cbr_paddr = chan->ch_bufring_dma.hv_paddr;
313	cbr.cbr_txsz = txbr_size;
314	cbr.cbr_rxsz = rxbr_size;
315
316	error = vmbus_chan_open_br(chan, &cbr, udata, udlen, cb, cbarg);
317	if (error) {
318		hyperv_dmamem_free(&chan->ch_bufring_dma, chan->ch_bufring);
319		chan->ch_bufring = NULL;
320	}
321	return (error);
322}
323
324int
325vmbus_chan_open_br(struct vmbus_channel *chan, const struct vmbus_chan_br *cbr,
326    const void *udata, int udlen, vmbus_chan_callback_t cb, void *cbarg)
327{
328	struct vmbus_softc *sc = chan->ch_vmbus;
329	const struct vmbus_chanmsg_chopen_resp *resp;
330	const struct vmbus_message *msg;
331	struct vmbus_chanmsg_chopen *req;
332	struct vmbus_msghc *mh;
333	uint32_t status;
334	int error, txbr_size, rxbr_size;
335	task_fn_t *task_fn;
336	uint8_t *br;
337
338	if (udlen > VMBUS_CHANMSG_CHOPEN_UDATA_SIZE) {
339		device_printf(sc->vmbus_dev,
340		    "invalid udata len %d for chan%u\n", udlen, chan->ch_id);
341		return EINVAL;
342	}
343
344	br = cbr->cbr;
345	txbr_size = cbr->cbr_txsz;
346	rxbr_size = cbr->cbr_rxsz;
347	KASSERT((txbr_size & PAGE_MASK) == 0,
348	    ("send bufring size is not multiple page"));
349	KASSERT((rxbr_size & PAGE_MASK) == 0,
350	    ("recv bufring size is not multiple page"));
351	KASSERT((cbr->cbr_paddr & PAGE_MASK) == 0,
352	    ("bufring is not page aligned"));
353
354	/*
355	 * Zero out the TX/RX bufrings, in case that they were used before.
356	 */
357	memset(br, 0, txbr_size + rxbr_size);
358
359	if (atomic_testandset_int(&chan->ch_stflags,
360	    VMBUS_CHAN_ST_OPENED_SHIFT))
361		panic("double-open chan%u", chan->ch_id);
362
363	chan->ch_cb = cb;
364	chan->ch_cbarg = cbarg;
365
366	vmbus_chan_update_evtflagcnt(sc, chan);
367
368	chan->ch_tq = VMBUS_PCPU_GET(chan->ch_vmbus, event_tq, chan->ch_cpuid);
369	if (chan->ch_flags & VMBUS_CHAN_FLAG_BATCHREAD)
370		task_fn = vmbus_chan_task;
371	else
372		task_fn = vmbus_chan_task_nobatch;
373	TASK_INIT(&chan->ch_task, 0, task_fn, chan);
374
375	/* TX bufring comes first */
376	vmbus_txbr_setup(&chan->ch_txbr, br, txbr_size);
377	/* RX bufring immediately follows TX bufring */
378	vmbus_rxbr_setup(&chan->ch_rxbr, br + txbr_size, rxbr_size);
379
380	/* Create sysctl tree for this channel */
381	vmbus_chan_sysctl_create(chan);
382
383	/*
384	 * Connect the bufrings, both RX and TX, to this channel.
385	 */
386	error = vmbus_chan_gpadl_connect(chan, cbr->cbr_paddr,
387	    txbr_size + rxbr_size, &chan->ch_bufring_gpadl);
388	if (error) {
389		device_printf(sc->vmbus_dev,
390		    "failed to connect bufring GPADL to chan%u\n", chan->ch_id);
391		goto failed;
392	}
393
394	/*
395	 * Install this channel, before it is opened, but after everything
396	 * else has been setup.
397	 */
398	vmbus_chan_set_chmap(chan);
399
400	/*
401	 * Open channel w/ the bufring GPADL on the target CPU.
402	 */
403	mh = vmbus_msghc_get(sc, sizeof(*req));
404	if (mh == NULL) {
405		device_printf(sc->vmbus_dev,
406		    "can not get msg hypercall for chopen(chan%u)\n",
407		    chan->ch_id);
408		error = ENXIO;
409		goto failed;
410	}
411
412	req = vmbus_msghc_dataptr(mh);
413	req->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_CHOPEN;
414	req->chm_chanid = chan->ch_id;
415	req->chm_openid = chan->ch_id;
416	req->chm_gpadl = chan->ch_bufring_gpadl;
417	req->chm_vcpuid = chan->ch_vcpuid;
418	req->chm_txbr_pgcnt = txbr_size >> PAGE_SHIFT;
419	if (udlen > 0)
420		memcpy(req->chm_udata, udata, udlen);
421
422	error = vmbus_msghc_exec(sc, mh);
423	if (error) {
424		device_printf(sc->vmbus_dev,
425		    "chopen(chan%u) msg hypercall exec failed: %d\n",
426		    chan->ch_id, error);
427		vmbus_msghc_put(sc, mh);
428		goto failed;
429	}
430
431	msg = vmbus_msghc_wait_result(sc, mh);
432	resp = (const struct vmbus_chanmsg_chopen_resp *)msg->msg_data;
433	status = resp->chm_status;
434
435	vmbus_msghc_put(sc, mh);
436
437	if (status == 0) {
438		if (bootverbose) {
439			device_printf(sc->vmbus_dev, "chan%u opened\n",
440			    chan->ch_id);
441		}
442		return 0;
443	}
444
445	device_printf(sc->vmbus_dev, "failed to open chan%u\n", chan->ch_id);
446	error = ENXIO;
447
448failed:
449	vmbus_chan_clear_chmap(chan);
450	if (chan->ch_bufring_gpadl) {
451		vmbus_chan_gpadl_disconnect(chan, chan->ch_bufring_gpadl);
452		chan->ch_bufring_gpadl = 0;
453	}
454	atomic_clear_int(&chan->ch_stflags, VMBUS_CHAN_ST_OPENED);
455	return error;
456}
457
458int
459vmbus_chan_gpadl_connect(struct vmbus_channel *chan, bus_addr_t paddr,
460    int size, uint32_t *gpadl0)
461{
462	struct vmbus_softc *sc = chan->ch_vmbus;
463	struct vmbus_msghc *mh;
464	struct vmbus_chanmsg_gpadl_conn *req;
465	const struct vmbus_message *msg;
466	size_t reqsz;
467	uint32_t gpadl, status;
468	int page_count, range_len, i, cnt, error;
469	uint64_t page_id;
470
471	/*
472	 * Preliminary checks.
473	 */
474
475	KASSERT((size & PAGE_MASK) == 0,
476	    ("invalid GPA size %d, not multiple page size", size));
477	page_count = size >> PAGE_SHIFT;
478
479	KASSERT((paddr & PAGE_MASK) == 0,
480	    ("GPA is not page aligned %jx", (uintmax_t)paddr));
481	page_id = paddr >> PAGE_SHIFT;
482
483	range_len = __offsetof(struct vmbus_gpa_range, gpa_page[page_count]);
484	/*
485	 * We don't support multiple GPA ranges.
486	 */
487	if (range_len > UINT16_MAX) {
488		device_printf(sc->vmbus_dev, "GPA too large, %d pages\n",
489		    page_count);
490		return EOPNOTSUPP;
491	}
492
493	/*
494	 * Allocate GPADL id.
495	 */
496	gpadl = vmbus_gpadl_alloc(sc);
497	*gpadl0 = gpadl;
498
499	/*
500	 * Connect this GPADL to the target channel.
501	 *
502	 * NOTE:
503	 * Since each message can only hold small set of page
504	 * addresses, several messages may be required to
505	 * complete the connection.
506	 */
507	if (page_count > VMBUS_CHANMSG_GPADL_CONN_PGMAX)
508		cnt = VMBUS_CHANMSG_GPADL_CONN_PGMAX;
509	else
510		cnt = page_count;
511	page_count -= cnt;
512
513	reqsz = __offsetof(struct vmbus_chanmsg_gpadl_conn,
514	    chm_range.gpa_page[cnt]);
515	mh = vmbus_msghc_get(sc, reqsz);
516	if (mh == NULL) {
517		device_printf(sc->vmbus_dev,
518		    "can not get msg hypercall for gpadl->chan%u\n",
519		    chan->ch_id);
520		return EIO;
521	}
522
523	req = vmbus_msghc_dataptr(mh);
524	req->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_GPADL_CONN;
525	req->chm_chanid = chan->ch_id;
526	req->chm_gpadl = gpadl;
527	req->chm_range_len = range_len;
528	req->chm_range_cnt = 1;
529	req->chm_range.gpa_len = size;
530	req->chm_range.gpa_ofs = 0;
531	for (i = 0; i < cnt; ++i)
532		req->chm_range.gpa_page[i] = page_id++;
533
534	error = vmbus_msghc_exec(sc, mh);
535	if (error) {
536		device_printf(sc->vmbus_dev,
537		    "gpadl->chan%u msg hypercall exec failed: %d\n",
538		    chan->ch_id, error);
539		vmbus_msghc_put(sc, mh);
540		return error;
541	}
542
543	while (page_count > 0) {
544		struct vmbus_chanmsg_gpadl_subconn *subreq;
545
546		if (page_count > VMBUS_CHANMSG_GPADL_SUBCONN_PGMAX)
547			cnt = VMBUS_CHANMSG_GPADL_SUBCONN_PGMAX;
548		else
549			cnt = page_count;
550		page_count -= cnt;
551
552		reqsz = __offsetof(struct vmbus_chanmsg_gpadl_subconn,
553		    chm_gpa_page[cnt]);
554		vmbus_msghc_reset(mh, reqsz);
555
556		subreq = vmbus_msghc_dataptr(mh);
557		subreq->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_GPADL_SUBCONN;
558		subreq->chm_gpadl = gpadl;
559		for (i = 0; i < cnt; ++i)
560			subreq->chm_gpa_page[i] = page_id++;
561
562		vmbus_msghc_exec_noresult(mh);
563	}
564	KASSERT(page_count == 0, ("invalid page count %d", page_count));
565
566	msg = vmbus_msghc_wait_result(sc, mh);
567	status = ((const struct vmbus_chanmsg_gpadl_connresp *)
568	    msg->msg_data)->chm_status;
569
570	vmbus_msghc_put(sc, mh);
571
572	if (status != 0) {
573		device_printf(sc->vmbus_dev, "gpadl->chan%u failed: "
574		    "status %u\n", chan->ch_id, status);
575		return EIO;
576	} else {
577		if (bootverbose) {
578			device_printf(sc->vmbus_dev, "gpadl->chan%u "
579			    "succeeded\n", chan->ch_id);
580		}
581	}
582	return 0;
583}
584
585/*
586 * Disconnect the GPA from the target channel
587 */
588int
589vmbus_chan_gpadl_disconnect(struct vmbus_channel *chan, uint32_t gpadl)
590{
591	struct vmbus_softc *sc = chan->ch_vmbus;
592	struct vmbus_msghc *mh;
593	struct vmbus_chanmsg_gpadl_disconn *req;
594	int error;
595
596	mh = vmbus_msghc_get(sc, sizeof(*req));
597	if (mh == NULL) {
598		device_printf(sc->vmbus_dev,
599		    "can not get msg hypercall for gpa x->chan%u\n",
600		    chan->ch_id);
601		return EBUSY;
602	}
603
604	req = vmbus_msghc_dataptr(mh);
605	req->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_GPADL_DISCONN;
606	req->chm_chanid = chan->ch_id;
607	req->chm_gpadl = gpadl;
608
609	error = vmbus_msghc_exec(sc, mh);
610	if (error) {
611		device_printf(sc->vmbus_dev,
612		    "gpa x->chan%u msg hypercall exec failed: %d\n",
613		    chan->ch_id, error);
614		vmbus_msghc_put(sc, mh);
615		return error;
616	}
617
618	vmbus_msghc_wait_result(sc, mh);
619	/* Discard result; no useful information */
620	vmbus_msghc_put(sc, mh);
621
622	return 0;
623}
624
625static void
626vmbus_chan_clrchmap_task(void *xchan, int pending __unused)
627{
628	struct vmbus_channel *chan = xchan;
629
630	critical_enter();
631	chan->ch_vmbus->vmbus_chmap[chan->ch_id] = NULL;
632	critical_exit();
633}
634
635static void
636vmbus_chan_clear_chmap(struct vmbus_channel *chan)
637{
638	struct task chmap_task;
639
640	TASK_INIT(&chmap_task, 0, vmbus_chan_clrchmap_task, chan);
641	taskqueue_enqueue(chan->ch_tq, &chmap_task);
642	taskqueue_drain(chan->ch_tq, &chmap_task);
643}
644
645static void
646vmbus_chan_set_chmap(struct vmbus_channel *chan)
647{
648	__compiler_membar();
649	chan->ch_vmbus->vmbus_chmap[chan->ch_id] = chan;
650}
651
652static void
653vmbus_chan_close_internal(struct vmbus_channel *chan)
654{
655	struct vmbus_softc *sc = chan->ch_vmbus;
656	struct vmbus_msghc *mh;
657	struct vmbus_chanmsg_chclose *req;
658	int error;
659
660	/* TODO: stringent check */
661	atomic_clear_int(&chan->ch_stflags, VMBUS_CHAN_ST_OPENED);
662
663	/*
664	 * Free this channel's sysctl tree attached to its device's
665	 * sysctl tree.
666	 */
667	sysctl_ctx_free(&chan->ch_sysctl_ctx);
668
669	/*
670	 * NOTE:
671	 * Order is critical.  This channel _must_ be uninstalled first,
672	 * else the channel task may be enqueued by the IDT after it has
673	 * been drained.
674	 */
675	vmbus_chan_clear_chmap(chan);
676	taskqueue_drain(chan->ch_tq, &chan->ch_task);
677	chan->ch_tq = NULL;
678
679	/*
680	 * Close this channel.
681	 */
682	mh = vmbus_msghc_get(sc, sizeof(*req));
683	if (mh == NULL) {
684		device_printf(sc->vmbus_dev,
685		    "can not get msg hypercall for chclose(chan%u)\n",
686		    chan->ch_id);
687		return;
688	}
689
690	req = vmbus_msghc_dataptr(mh);
691	req->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_CHCLOSE;
692	req->chm_chanid = chan->ch_id;
693
694	error = vmbus_msghc_exec_noresult(mh);
695	vmbus_msghc_put(sc, mh);
696
697	if (error) {
698		device_printf(sc->vmbus_dev,
699		    "chclose(chan%u) msg hypercall exec failed: %d\n",
700		    chan->ch_id, error);
701		return;
702	} else if (bootverbose) {
703		device_printf(sc->vmbus_dev, "close chan%u\n", chan->ch_id);
704	}
705
706	/*
707	 * Disconnect the TX+RX bufrings from this channel.
708	 */
709	if (chan->ch_bufring_gpadl) {
710		vmbus_chan_gpadl_disconnect(chan, chan->ch_bufring_gpadl);
711		chan->ch_bufring_gpadl = 0;
712	}
713
714	/*
715	 * Destroy the TX+RX bufrings.
716	 */
717	if (chan->ch_bufring != NULL) {
718		hyperv_dmamem_free(&chan->ch_bufring_dma, chan->ch_bufring);
719		chan->ch_bufring = NULL;
720	}
721}
722
723/*
724 * Caller should make sure that all sub-channels have
725 * been added to 'chan' and all to-be-closed channels
726 * are not being opened.
727 */
728void
729vmbus_chan_close(struct vmbus_channel *chan)
730{
731	int subchan_cnt;
732
733	if (!VMBUS_CHAN_ISPRIMARY(chan)) {
734		/*
735		 * Sub-channel is closed when its primary channel
736		 * is closed; done.
737		 */
738		return;
739	}
740
741	/*
742	 * Close all sub-channels, if any.
743	 */
744	subchan_cnt = chan->ch_subchan_cnt;
745	if (subchan_cnt > 0) {
746		struct vmbus_channel **subchan;
747		int i;
748
749		subchan = vmbus_subchan_get(chan, subchan_cnt);
750		for (i = 0; i < subchan_cnt; ++i)
751			vmbus_chan_close_internal(subchan[i]);
752		vmbus_subchan_rel(subchan, subchan_cnt);
753	}
754
755	/* Then close the primary channel. */
756	vmbus_chan_close_internal(chan);
757}
758
759void
760vmbus_chan_intr_drain(struct vmbus_channel *chan)
761{
762
763	taskqueue_drain(chan->ch_tq, &chan->ch_task);
764}
765
766int
767vmbus_chan_send(struct vmbus_channel *chan, uint16_t type, uint16_t flags,
768    void *data, int dlen, uint64_t xactid)
769{
770	struct vmbus_chanpkt pkt;
771	int pktlen, pad_pktlen, hlen, error;
772	uint64_t pad = 0;
773	struct iovec iov[3];
774	boolean_t send_evt;
775
776	hlen = sizeof(pkt);
777	pktlen = hlen + dlen;
778	pad_pktlen = VMBUS_CHANPKT_TOTLEN(pktlen);
779	KASSERT(pad_pktlen <= vmbus_txbr_maxpktsz(&chan->ch_txbr),
780	    ("invalid packet size %d", pad_pktlen));
781
782	pkt.cp_hdr.cph_type = type;
783	pkt.cp_hdr.cph_flags = flags;
784	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_hlen, hlen);
785	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_tlen, pad_pktlen);
786	pkt.cp_hdr.cph_xactid = xactid;
787
788	iov[0].iov_base = &pkt;
789	iov[0].iov_len = hlen;
790	iov[1].iov_base = data;
791	iov[1].iov_len = dlen;
792	iov[2].iov_base = &pad;
793	iov[2].iov_len = pad_pktlen - pktlen;
794
795	error = vmbus_txbr_write(&chan->ch_txbr, iov, 3, &send_evt);
796	if (!error && send_evt)
797		vmbus_chan_signal_tx(chan);
798	return error;
799}
800
801int
802vmbus_chan_send_sglist(struct vmbus_channel *chan,
803    struct vmbus_gpa sg[], int sglen, void *data, int dlen, uint64_t xactid)
804{
805	struct vmbus_chanpkt_sglist pkt;
806	int pktlen, pad_pktlen, hlen, error;
807	struct iovec iov[4];
808	boolean_t send_evt;
809	uint64_t pad = 0;
810
811	hlen = __offsetof(struct vmbus_chanpkt_sglist, cp_gpa[sglen]);
812	pktlen = hlen + dlen;
813	pad_pktlen = VMBUS_CHANPKT_TOTLEN(pktlen);
814	KASSERT(pad_pktlen <= vmbus_txbr_maxpktsz(&chan->ch_txbr),
815	    ("invalid packet size %d", pad_pktlen));
816
817	pkt.cp_hdr.cph_type = VMBUS_CHANPKT_TYPE_GPA;
818	pkt.cp_hdr.cph_flags = VMBUS_CHANPKT_FLAG_RC;
819	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_hlen, hlen);
820	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_tlen, pad_pktlen);
821	pkt.cp_hdr.cph_xactid = xactid;
822	pkt.cp_rsvd = 0;
823	pkt.cp_gpa_cnt = sglen;
824
825	iov[0].iov_base = &pkt;
826	iov[0].iov_len = sizeof(pkt);
827	iov[1].iov_base = sg;
828	iov[1].iov_len = sizeof(struct vmbus_gpa) * sglen;
829	iov[2].iov_base = data;
830	iov[2].iov_len = dlen;
831	iov[3].iov_base = &pad;
832	iov[3].iov_len = pad_pktlen - pktlen;
833
834	error = vmbus_txbr_write(&chan->ch_txbr, iov, 4, &send_evt);
835	if (!error && send_evt)
836		vmbus_chan_signal_tx(chan);
837	return error;
838}
839
840int
841vmbus_chan_send_prplist(struct vmbus_channel *chan,
842    struct vmbus_gpa_range *prp, int prp_cnt, void *data, int dlen,
843    uint64_t xactid)
844{
845	struct vmbus_chanpkt_prplist pkt;
846	int pktlen, pad_pktlen, hlen, error;
847	struct iovec iov[4];
848	boolean_t send_evt;
849	uint64_t pad = 0;
850
851	hlen = __offsetof(struct vmbus_chanpkt_prplist,
852	    cp_range[0].gpa_page[prp_cnt]);
853	pktlen = hlen + dlen;
854	pad_pktlen = VMBUS_CHANPKT_TOTLEN(pktlen);
855	KASSERT(pad_pktlen <= vmbus_txbr_maxpktsz(&chan->ch_txbr),
856	    ("invalid packet size %d", pad_pktlen));
857
858	pkt.cp_hdr.cph_type = VMBUS_CHANPKT_TYPE_GPA;
859	pkt.cp_hdr.cph_flags = VMBUS_CHANPKT_FLAG_RC;
860	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_hlen, hlen);
861	VMBUS_CHANPKT_SETLEN(pkt.cp_hdr.cph_tlen, pad_pktlen);
862	pkt.cp_hdr.cph_xactid = xactid;
863	pkt.cp_rsvd = 0;
864	pkt.cp_range_cnt = 1;
865
866	iov[0].iov_base = &pkt;
867	iov[0].iov_len = sizeof(pkt);
868	iov[1].iov_base = prp;
869	iov[1].iov_len = __offsetof(struct vmbus_gpa_range, gpa_page[prp_cnt]);
870	iov[2].iov_base = data;
871	iov[2].iov_len = dlen;
872	iov[3].iov_base = &pad;
873	iov[3].iov_len = pad_pktlen - pktlen;
874
875	error = vmbus_txbr_write(&chan->ch_txbr, iov, 4, &send_evt);
876	if (!error && send_evt)
877		vmbus_chan_signal_tx(chan);
878	return error;
879}
880
881int
882vmbus_chan_recv(struct vmbus_channel *chan, void *data, int *dlen0,
883    uint64_t *xactid)
884{
885	struct vmbus_chanpkt_hdr pkt;
886	int error, dlen, hlen;
887
888	error = vmbus_rxbr_peek(&chan->ch_rxbr, &pkt, sizeof(pkt));
889	if (error)
890		return (error);
891
892	if (__predict_false(pkt.cph_hlen < VMBUS_CHANPKT_HLEN_MIN)) {
893		device_printf(chan->ch_dev, "invalid hlen %u\n",
894		    pkt.cph_hlen);
895		/* XXX this channel is dead actually. */
896		return (EIO);
897	}
898	if (__predict_false(pkt.cph_hlen > pkt.cph_tlen)) {
899		device_printf(chan->ch_dev, "invalid hlen %u and tlen %u\n",
900		    pkt.cph_hlen, pkt.cph_tlen);
901		/* XXX this channel is dead actually. */
902		return (EIO);
903	}
904
905	hlen = VMBUS_CHANPKT_GETLEN(pkt.cph_hlen);
906	dlen = VMBUS_CHANPKT_GETLEN(pkt.cph_tlen) - hlen;
907
908	if (*dlen0 < dlen) {
909		/* Return the size of this packet's data. */
910		*dlen0 = dlen;
911		return (ENOBUFS);
912	}
913
914	*xactid = pkt.cph_xactid;
915	*dlen0 = dlen;
916
917	/* Skip packet header */
918	error = vmbus_rxbr_read(&chan->ch_rxbr, data, dlen, hlen);
919	KASSERT(!error, ("vmbus_rxbr_read failed"));
920
921	return (0);
922}
923
924int
925vmbus_chan_recv_pkt(struct vmbus_channel *chan,
926    struct vmbus_chanpkt_hdr *pkt0, int *pktlen0)
927{
928	struct vmbus_chanpkt_hdr pkt;
929	int error, pktlen;
930
931	error = vmbus_rxbr_peek(&chan->ch_rxbr, &pkt, sizeof(pkt));
932	if (error)
933		return (error);
934
935	if (__predict_false(pkt.cph_hlen < VMBUS_CHANPKT_HLEN_MIN)) {
936		device_printf(chan->ch_dev, "invalid hlen %u\n",
937		    pkt.cph_hlen);
938		/* XXX this channel is dead actually. */
939		return (EIO);
940	}
941	if (__predict_false(pkt.cph_hlen > pkt.cph_tlen)) {
942		device_printf(chan->ch_dev, "invalid hlen %u and tlen %u\n",
943		    pkt.cph_hlen, pkt.cph_tlen);
944		/* XXX this channel is dead actually. */
945		return (EIO);
946	}
947
948	pktlen = VMBUS_CHANPKT_GETLEN(pkt.cph_tlen);
949	if (*pktlen0 < pktlen) {
950		/* Return the size of this packet. */
951		*pktlen0 = pktlen;
952		return (ENOBUFS);
953	}
954	*pktlen0 = pktlen;
955
956	/* Include packet header */
957	error = vmbus_rxbr_read(&chan->ch_rxbr, pkt0, pktlen, 0);
958	KASSERT(!error, ("vmbus_rxbr_read failed"));
959
960	return (0);
961}
962
963static void
964vmbus_chan_task(void *xchan, int pending __unused)
965{
966	struct vmbus_channel *chan = xchan;
967	vmbus_chan_callback_t cb = chan->ch_cb;
968	void *cbarg = chan->ch_cbarg;
969
970	/*
971	 * Optimize host to guest signaling by ensuring:
972	 * 1. While reading the channel, we disable interrupts from
973	 *    host.
974	 * 2. Ensure that we process all posted messages from the host
975	 *    before returning from this callback.
976	 * 3. Once we return, enable signaling from the host. Once this
977	 *    state is set we check to see if additional packets are
978	 *    available to read. In this case we repeat the process.
979	 *
980	 * NOTE: Interrupt has been disabled in the ISR.
981	 */
982	for (;;) {
983		uint32_t left;
984
985		cb(chan, cbarg);
986
987		left = vmbus_rxbr_intr_unmask(&chan->ch_rxbr);
988		if (left == 0) {
989			/* No more data in RX bufring; done */
990			break;
991		}
992		vmbus_rxbr_intr_mask(&chan->ch_rxbr);
993	}
994}
995
996static void
997vmbus_chan_task_nobatch(void *xchan, int pending __unused)
998{
999	struct vmbus_channel *chan = xchan;
1000
1001	chan->ch_cb(chan, chan->ch_cbarg);
1002}
1003
1004static __inline void
1005vmbus_event_flags_proc(struct vmbus_softc *sc, volatile u_long *event_flags,
1006    int flag_cnt)
1007{
1008	int f;
1009
1010	for (f = 0; f < flag_cnt; ++f) {
1011		uint32_t chid_base;
1012		u_long flags;
1013		int chid_ofs;
1014
1015		if (event_flags[f] == 0)
1016			continue;
1017
1018		flags = atomic_swap_long(&event_flags[f], 0);
1019		chid_base = f << VMBUS_EVTFLAG_SHIFT;
1020
1021		while ((chid_ofs = ffsl(flags)) != 0) {
1022			struct vmbus_channel *chan;
1023
1024			--chid_ofs; /* NOTE: ffsl is 1-based */
1025			flags &= ~(1UL << chid_ofs);
1026
1027			chan = sc->vmbus_chmap[chid_base + chid_ofs];
1028			if (__predict_false(chan == NULL)) {
1029				/* Channel is closed. */
1030				continue;
1031			}
1032			__compiler_membar();
1033
1034			if (chan->ch_flags & VMBUS_CHAN_FLAG_BATCHREAD)
1035				vmbus_rxbr_intr_mask(&chan->ch_rxbr);
1036			taskqueue_enqueue(chan->ch_tq, &chan->ch_task);
1037		}
1038	}
1039}
1040
1041void
1042vmbus_event_proc(struct vmbus_softc *sc, int cpu)
1043{
1044	struct vmbus_evtflags *eventf;
1045
1046	/*
1047	 * On Host with Win8 or above, the event page can be checked directly
1048	 * to get the id of the channel that has the pending interrupt.
1049	 */
1050	eventf = VMBUS_PCPU_GET(sc, event_flags, cpu) + VMBUS_SINT_MESSAGE;
1051	vmbus_event_flags_proc(sc, eventf->evt_flags,
1052	    VMBUS_PCPU_GET(sc, event_flags_cnt, cpu));
1053}
1054
1055void
1056vmbus_event_proc_compat(struct vmbus_softc *sc, int cpu)
1057{
1058	struct vmbus_evtflags *eventf;
1059
1060	eventf = VMBUS_PCPU_GET(sc, event_flags, cpu) + VMBUS_SINT_MESSAGE;
1061	if (atomic_testandclear_long(&eventf->evt_flags[0], 0)) {
1062		vmbus_event_flags_proc(sc, sc->vmbus_rx_evtflags,
1063		    VMBUS_CHAN_MAX_COMPAT >> VMBUS_EVTFLAG_SHIFT);
1064	}
1065}
1066
1067static void
1068vmbus_chan_update_evtflagcnt(struct vmbus_softc *sc,
1069    const struct vmbus_channel *chan)
1070{
1071	volatile int *flag_cnt_ptr;
1072	int flag_cnt;
1073
1074	flag_cnt = (chan->ch_id / VMBUS_EVTFLAG_LEN) + 1;
1075	flag_cnt_ptr = VMBUS_PCPU_PTR(sc, event_flags_cnt, chan->ch_cpuid);
1076
1077	for (;;) {
1078		int old_flag_cnt;
1079
1080		old_flag_cnt = *flag_cnt_ptr;
1081		if (old_flag_cnt >= flag_cnt)
1082			break;
1083		if (atomic_cmpset_int(flag_cnt_ptr, old_flag_cnt, flag_cnt)) {
1084			if (bootverbose) {
1085				device_printf(sc->vmbus_dev,
1086				    "channel%u update cpu%d flag_cnt to %d\n",
1087				    chan->ch_id, chan->ch_cpuid, flag_cnt);
1088			}
1089			break;
1090		}
1091	}
1092}
1093
1094static struct vmbus_channel *
1095vmbus_chan_alloc(struct vmbus_softc *sc)
1096{
1097	struct vmbus_channel *chan;
1098
1099	chan = malloc(sizeof(*chan), M_DEVBUF, M_WAITOK | M_ZERO);
1100
1101	chan->ch_monprm = hyperv_dmamem_alloc(bus_get_dma_tag(sc->vmbus_dev),
1102	    HYPERCALL_PARAM_ALIGN, 0, sizeof(struct hyperv_mon_param),
1103	    &chan->ch_monprm_dma, BUS_DMA_WAITOK | BUS_DMA_ZERO);
1104	if (chan->ch_monprm == NULL) {
1105		device_printf(sc->vmbus_dev, "monprm alloc failed\n");
1106		free(chan, M_DEVBUF);
1107		return NULL;
1108	}
1109
1110	chan->ch_vmbus = sc;
1111	mtx_init(&chan->ch_subchan_lock, "vmbus subchan", NULL, MTX_DEF);
1112	TAILQ_INIT(&chan->ch_subchans);
1113	vmbus_rxbr_init(&chan->ch_rxbr);
1114	vmbus_txbr_init(&chan->ch_txbr);
1115
1116	return chan;
1117}
1118
1119static void
1120vmbus_chan_free(struct vmbus_channel *chan)
1121{
1122
1123	KASSERT(TAILQ_EMPTY(&chan->ch_subchans) && chan->ch_subchan_cnt == 0,
1124	    ("still owns sub-channels"));
1125	KASSERT((chan->ch_stflags &
1126	    (VMBUS_CHAN_ST_OPENED |
1127	     VMBUS_CHAN_ST_ONPRIL |
1128	     VMBUS_CHAN_ST_ONSUBL |
1129	     VMBUS_CHAN_ST_ONLIST)) == 0, ("free busy channel"));
1130	hyperv_dmamem_free(&chan->ch_monprm_dma, chan->ch_monprm);
1131	mtx_destroy(&chan->ch_subchan_lock);
1132	vmbus_rxbr_deinit(&chan->ch_rxbr);
1133	vmbus_txbr_deinit(&chan->ch_txbr);
1134	free(chan, M_DEVBUF);
1135}
1136
1137static int
1138vmbus_chan_add(struct vmbus_channel *newchan)
1139{
1140	struct vmbus_softc *sc = newchan->ch_vmbus;
1141	struct vmbus_channel *prichan;
1142
1143	if (newchan->ch_id == 0) {
1144		/*
1145		 * XXX
1146		 * Chan0 will neither be processed nor should be offered;
1147		 * skip it.
1148		 */
1149		device_printf(sc->vmbus_dev, "got chan0 offer, discard\n");
1150		return EINVAL;
1151	} else if (newchan->ch_id >= VMBUS_CHAN_MAX) {
1152		device_printf(sc->vmbus_dev, "invalid chan%u offer\n",
1153		    newchan->ch_id);
1154		return EINVAL;
1155	}
1156
1157	if (bootverbose) {
1158		device_printf(sc->vmbus_dev, "chan%u subidx%u offer\n",
1159		    newchan->ch_id, newchan->ch_subidx);
1160	}
1161
1162	mtx_lock(&sc->vmbus_prichan_lock);
1163	TAILQ_FOREACH(prichan, &sc->vmbus_prichans, ch_prilink) {
1164		/*
1165		 * Sub-channel will have the same type GUID and instance
1166		 * GUID as its primary channel.
1167		 */
1168		if (memcmp(&prichan->ch_guid_type, &newchan->ch_guid_type,
1169		    sizeof(struct hyperv_guid)) == 0 &&
1170		    memcmp(&prichan->ch_guid_inst, &newchan->ch_guid_inst,
1171		    sizeof(struct hyperv_guid)) == 0)
1172			break;
1173	}
1174	if (VMBUS_CHAN_ISPRIMARY(newchan)) {
1175		if (prichan == NULL) {
1176			/* Install the new primary channel */
1177			vmbus_chan_ins_prilist(sc, newchan);
1178			mtx_unlock(&sc->vmbus_prichan_lock);
1179			goto done;
1180		} else {
1181			mtx_unlock(&sc->vmbus_prichan_lock);
1182			device_printf(sc->vmbus_dev, "duplicated primary "
1183			    "chan%u\n", newchan->ch_id);
1184			return EINVAL;
1185		}
1186	} else { /* Sub-channel */
1187		if (prichan == NULL) {
1188			mtx_unlock(&sc->vmbus_prichan_lock);
1189			device_printf(sc->vmbus_dev, "no primary chan for "
1190			    "chan%u\n", newchan->ch_id);
1191			return EINVAL;
1192		}
1193		/*
1194		 * Found the primary channel for this sub-channel and
1195		 * move on.
1196		 *
1197		 * XXX refcnt prichan
1198		 */
1199	}
1200	mtx_unlock(&sc->vmbus_prichan_lock);
1201
1202	/*
1203	 * This is a sub-channel; link it with the primary channel.
1204	 */
1205	KASSERT(!VMBUS_CHAN_ISPRIMARY(newchan),
1206	    ("new channel is not sub-channel"));
1207	KASSERT(prichan != NULL, ("no primary channel"));
1208
1209	newchan->ch_prichan = prichan;
1210	newchan->ch_dev = prichan->ch_dev;
1211
1212	mtx_lock(&prichan->ch_subchan_lock);
1213	vmbus_chan_ins_sublist(prichan, newchan);
1214	mtx_unlock(&prichan->ch_subchan_lock);
1215	/*
1216	 * Notify anyone that is interested in this sub-channel,
1217	 * after this sub-channel is setup.
1218	 */
1219	wakeup(prichan);
1220done:
1221	/*
1222	 * Hook this channel up for later rescind.
1223	 */
1224	mtx_lock(&sc->vmbus_chan_lock);
1225	vmbus_chan_ins_list(sc, newchan);
1226	mtx_unlock(&sc->vmbus_chan_lock);
1227	return 0;
1228}
1229
1230void
1231vmbus_chan_cpu_set(struct vmbus_channel *chan, int cpu)
1232{
1233	KASSERT(cpu >= 0 && cpu < mp_ncpus, ("invalid cpu %d", cpu));
1234
1235	if (chan->ch_vmbus->vmbus_version == VMBUS_VERSION_WS2008 ||
1236	    chan->ch_vmbus->vmbus_version == VMBUS_VERSION_WIN7) {
1237		/* Only cpu0 is supported */
1238		cpu = 0;
1239	}
1240
1241	chan->ch_cpuid = cpu;
1242	chan->ch_vcpuid = VMBUS_PCPU_GET(chan->ch_vmbus, vcpuid, cpu);
1243
1244	if (bootverbose) {
1245		printf("vmbus_chan%u: assigned to cpu%u [vcpu%u]\n",
1246		    chan->ch_id, chan->ch_cpuid, chan->ch_vcpuid);
1247	}
1248}
1249
1250void
1251vmbus_chan_cpu_rr(struct vmbus_channel *chan)
1252{
1253	static uint32_t vmbus_chan_nextcpu;
1254	int cpu;
1255
1256	cpu = atomic_fetchadd_int(&vmbus_chan_nextcpu, 1) % mp_ncpus;
1257	vmbus_chan_cpu_set(chan, cpu);
1258}
1259
1260static void
1261vmbus_chan_cpu_default(struct vmbus_channel *chan)
1262{
1263	/*
1264	 * By default, pin the channel to cpu0.  Devices having
1265	 * special channel-cpu mapping requirement should call
1266	 * vmbus_chan_cpu_{set,rr}().
1267	 */
1268	vmbus_chan_cpu_set(chan, 0);
1269}
1270
1271static void
1272vmbus_chan_msgproc_choffer(struct vmbus_softc *sc,
1273    const struct vmbus_message *msg)
1274{
1275	const struct vmbus_chanmsg_choffer *offer;
1276	struct vmbus_channel *chan;
1277	task_fn_t *detach_fn, *attach_fn;
1278	int error;
1279
1280	offer = (const struct vmbus_chanmsg_choffer *)msg->msg_data;
1281
1282	chan = vmbus_chan_alloc(sc);
1283	if (chan == NULL) {
1284		device_printf(sc->vmbus_dev, "allocate chan%u failed\n",
1285		    offer->chm_chanid);
1286		return;
1287	}
1288
1289	chan->ch_id = offer->chm_chanid;
1290	chan->ch_subidx = offer->chm_subidx;
1291	chan->ch_guid_type = offer->chm_chtype;
1292	chan->ch_guid_inst = offer->chm_chinst;
1293
1294	/* Batch reading is on by default */
1295	chan->ch_flags |= VMBUS_CHAN_FLAG_BATCHREAD;
1296
1297	chan->ch_monprm->mp_connid = VMBUS_CONNID_EVENT;
1298	if (sc->vmbus_version != VMBUS_VERSION_WS2008)
1299		chan->ch_monprm->mp_connid = offer->chm_connid;
1300
1301	if (offer->chm_flags1 & VMBUS_CHOFFER_FLAG1_HASMNF) {
1302		int trig_idx;
1303
1304		/*
1305		 * Setup MNF stuffs.
1306		 */
1307		chan->ch_txflags |= VMBUS_CHAN_TXF_HASMNF;
1308
1309		trig_idx = offer->chm_montrig / VMBUS_MONTRIG_LEN;
1310		if (trig_idx >= VMBUS_MONTRIGS_MAX)
1311			panic("invalid monitor trigger %u", offer->chm_montrig);
1312		chan->ch_montrig =
1313		    &sc->vmbus_mnf2->mnf_trigs[trig_idx].mt_pending;
1314
1315		chan->ch_montrig_mask =
1316		    1 << (offer->chm_montrig % VMBUS_MONTRIG_LEN);
1317	}
1318
1319	/*
1320	 * Setup event flag.
1321	 */
1322	chan->ch_evtflag =
1323	    &sc->vmbus_tx_evtflags[chan->ch_id >> VMBUS_EVTFLAG_SHIFT];
1324	chan->ch_evtflag_mask = 1UL << (chan->ch_id & VMBUS_EVTFLAG_MASK);
1325
1326	/*
1327	 * Setup attach and detach tasks.
1328	 */
1329	if (VMBUS_CHAN_ISPRIMARY(chan)) {
1330		chan->ch_mgmt_tq = sc->vmbus_devtq;
1331		attach_fn = vmbus_prichan_attach_task;
1332		detach_fn = vmbus_prichan_detach_task;
1333	} else {
1334		chan->ch_mgmt_tq = sc->vmbus_subchtq;
1335		attach_fn = vmbus_subchan_attach_task;
1336		detach_fn = vmbus_subchan_detach_task;
1337	}
1338	TASK_INIT(&chan->ch_attach_task, 0, attach_fn, chan);
1339	TASK_INIT(&chan->ch_detach_task, 0, detach_fn, chan);
1340
1341	/* Select default cpu for this channel. */
1342	vmbus_chan_cpu_default(chan);
1343
1344	error = vmbus_chan_add(chan);
1345	if (error) {
1346		device_printf(sc->vmbus_dev, "add chan%u failed: %d\n",
1347		    chan->ch_id, error);
1348		vmbus_chan_free(chan);
1349		return;
1350	}
1351	taskqueue_enqueue(chan->ch_mgmt_tq, &chan->ch_attach_task);
1352}
1353
1354static void
1355vmbus_chan_msgproc_chrescind(struct vmbus_softc *sc,
1356    const struct vmbus_message *msg)
1357{
1358	const struct vmbus_chanmsg_chrescind *note;
1359	struct vmbus_channel *chan;
1360
1361	note = (const struct vmbus_chanmsg_chrescind *)msg->msg_data;
1362	if (note->chm_chanid > VMBUS_CHAN_MAX) {
1363		device_printf(sc->vmbus_dev, "invalid rescinded chan%u\n",
1364		    note->chm_chanid);
1365		return;
1366	}
1367
1368	if (bootverbose) {
1369		device_printf(sc->vmbus_dev, "chan%u rescinded\n",
1370		    note->chm_chanid);
1371	}
1372
1373	/*
1374	 * Find and remove the target channel from the channel list.
1375	 */
1376	mtx_lock(&sc->vmbus_chan_lock);
1377	TAILQ_FOREACH(chan, &sc->vmbus_chans, ch_link) {
1378		if (chan->ch_id == note->chm_chanid)
1379			break;
1380	}
1381	if (chan == NULL) {
1382		mtx_unlock(&sc->vmbus_chan_lock);
1383		device_printf(sc->vmbus_dev, "chan%u is not offered\n",
1384		    note->chm_chanid);
1385		return;
1386	}
1387	vmbus_chan_rem_list(sc, chan);
1388	mtx_unlock(&sc->vmbus_chan_lock);
1389
1390	if (VMBUS_CHAN_ISPRIMARY(chan)) {
1391		/*
1392		 * The target channel is a primary channel; remove the
1393		 * target channel from the primary channel list now,
1394		 * instead of later, so that it will not be found by
1395		 * other sub-channel offers, which are processed in
1396		 * this thread.
1397		 */
1398		mtx_lock(&sc->vmbus_prichan_lock);
1399		vmbus_chan_rem_prilist(sc, chan);
1400		mtx_unlock(&sc->vmbus_prichan_lock);
1401	}
1402
1403	/* Detach the target channel. */
1404	taskqueue_enqueue(chan->ch_mgmt_tq, &chan->ch_detach_task);
1405}
1406
1407static int
1408vmbus_chan_release(struct vmbus_channel *chan)
1409{
1410	struct vmbus_softc *sc = chan->ch_vmbus;
1411	struct vmbus_chanmsg_chfree *req;
1412	struct vmbus_msghc *mh;
1413	int error;
1414
1415	mh = vmbus_msghc_get(sc, sizeof(*req));
1416	if (mh == NULL) {
1417		device_printf(sc->vmbus_dev, "can not get msg hypercall for "
1418		    "chfree(chan%u)\n", chan->ch_id);
1419		return (ENXIO);
1420	}
1421
1422	req = vmbus_msghc_dataptr(mh);
1423	req->chm_hdr.chm_type = VMBUS_CHANMSG_TYPE_CHFREE;
1424	req->chm_chanid = chan->ch_id;
1425
1426	error = vmbus_msghc_exec_noresult(mh);
1427	vmbus_msghc_put(sc, mh);
1428
1429	if (error) {
1430		device_printf(sc->vmbus_dev, "chfree(chan%u) failed: %d",
1431		    chan->ch_id, error);
1432	} else {
1433		if (bootverbose) {
1434			device_printf(sc->vmbus_dev, "chan%u freed\n",
1435			    chan->ch_id);
1436		}
1437	}
1438	return (error);
1439}
1440
1441static void
1442vmbus_prichan_detach_task(void *xchan, int pending __unused)
1443{
1444	struct vmbus_channel *chan = xchan;
1445
1446	KASSERT(VMBUS_CHAN_ISPRIMARY(chan),
1447	    ("chan%u is not primary channel", chan->ch_id));
1448
1449	/* Delete and detach the device associated with this channel. */
1450	vmbus_delete_child(chan);
1451
1452	/* Release this channel (back to vmbus). */
1453	vmbus_chan_release(chan);
1454
1455	/* Free this channel's resource. */
1456	vmbus_chan_free(chan);
1457}
1458
1459static void
1460vmbus_subchan_detach_task(void *xchan, int pending __unused)
1461{
1462	struct vmbus_channel *chan = xchan;
1463	struct vmbus_channel *pri_chan = chan->ch_prichan;
1464
1465	KASSERT(!VMBUS_CHAN_ISPRIMARY(chan),
1466	    ("chan%u is primary channel", chan->ch_id));
1467
1468	/* Release this channel (back to vmbus). */
1469	vmbus_chan_release(chan);
1470
1471	/* Unlink from its primary channel's sub-channel list. */
1472	mtx_lock(&pri_chan->ch_subchan_lock);
1473	vmbus_chan_rem_sublist(pri_chan, chan);
1474	mtx_unlock(&pri_chan->ch_subchan_lock);
1475	/* Notify anyone that is waiting for this sub-channel to vanish. */
1476	wakeup(pri_chan);
1477
1478	/* Free this channel's resource. */
1479	vmbus_chan_free(chan);
1480}
1481
1482static void
1483vmbus_prichan_attach_task(void *xchan, int pending __unused)
1484{
1485
1486	/*
1487	 * Add device for this primary channel.
1488	 */
1489	vmbus_add_child(xchan);
1490}
1491
1492static void
1493vmbus_subchan_attach_task(void *xchan __unused, int pending __unused)
1494{
1495
1496	/* Nothing */
1497}
1498
1499void
1500vmbus_chan_destroy_all(struct vmbus_softc *sc)
1501{
1502
1503	/*
1504	 * Detach all devices and destroy the corresponding primary
1505	 * channels.
1506	 */
1507	for (;;) {
1508		struct vmbus_channel *chan;
1509
1510		mtx_lock(&sc->vmbus_chan_lock);
1511		TAILQ_FOREACH(chan, &sc->vmbus_chans, ch_link) {
1512			if (VMBUS_CHAN_ISPRIMARY(chan))
1513				break;
1514		}
1515		if (chan == NULL) {
1516			/* No more primary channels; done. */
1517			mtx_unlock(&sc->vmbus_chan_lock);
1518			break;
1519		}
1520		vmbus_chan_rem_list(sc, chan);
1521		mtx_unlock(&sc->vmbus_chan_lock);
1522
1523		mtx_lock(&sc->vmbus_prichan_lock);
1524		vmbus_chan_rem_prilist(sc, chan);
1525		mtx_unlock(&sc->vmbus_prichan_lock);
1526
1527		taskqueue_enqueue(chan->ch_mgmt_tq, &chan->ch_detach_task);
1528	}
1529}
1530
1531/*
1532 * The channel whose vcpu binding is closest to the currect vcpu will
1533 * be selected.
1534 * If no multi-channel, always select primary channel.
1535 */
1536struct vmbus_channel *
1537vmbus_chan_cpu2chan(struct vmbus_channel *prichan, int cpu)
1538{
1539	struct vmbus_channel *sel, *chan;
1540	uint32_t vcpu, sel_dist;
1541
1542	KASSERT(cpu >= 0 && cpu < mp_ncpus, ("invalid cpuid %d", cpu));
1543	if (TAILQ_EMPTY(&prichan->ch_subchans))
1544		return prichan;
1545
1546	vcpu = VMBUS_PCPU_GET(prichan->ch_vmbus, vcpuid, cpu);
1547
1548#define CHAN_VCPU_DIST(ch, vcpu)		\
1549	(((ch)->ch_vcpuid > (vcpu)) ?		\
1550	 ((ch)->ch_vcpuid - (vcpu)) : ((vcpu) - (ch)->ch_vcpuid))
1551
1552#define CHAN_SELECT(ch)				\
1553do {						\
1554	sel = ch;				\
1555	sel_dist = CHAN_VCPU_DIST(ch, vcpu);	\
1556} while (0)
1557
1558	CHAN_SELECT(prichan);
1559
1560	mtx_lock(&prichan->ch_subchan_lock);
1561	TAILQ_FOREACH(chan, &prichan->ch_subchans, ch_sublink) {
1562		uint32_t dist;
1563
1564		KASSERT(chan->ch_stflags & VMBUS_CHAN_ST_OPENED,
1565		    ("chan%u is not opened", chan->ch_id));
1566
1567		if (chan->ch_vcpuid == vcpu) {
1568			/* Exact match; done */
1569			CHAN_SELECT(chan);
1570			break;
1571		}
1572
1573		dist = CHAN_VCPU_DIST(chan, vcpu);
1574		if (sel_dist <= dist) {
1575			/* Far or same distance; skip */
1576			continue;
1577		}
1578
1579		/* Select the closer channel. */
1580		CHAN_SELECT(chan);
1581	}
1582	mtx_unlock(&prichan->ch_subchan_lock);
1583
1584#undef CHAN_SELECT
1585#undef CHAN_VCPU_DIST
1586
1587	return sel;
1588}
1589
1590struct vmbus_channel **
1591vmbus_subchan_get(struct vmbus_channel *pri_chan, int subchan_cnt)
1592{
1593	struct vmbus_channel **ret, *chan;
1594	int i;
1595
1596	KASSERT(subchan_cnt > 0, ("invalid sub-channel count %d", subchan_cnt));
1597
1598	ret = malloc(subchan_cnt * sizeof(struct vmbus_channel *), M_TEMP,
1599	    M_WAITOK);
1600
1601	mtx_lock(&pri_chan->ch_subchan_lock);
1602
1603	while (pri_chan->ch_subchan_cnt < subchan_cnt)
1604		mtx_sleep(pri_chan, &pri_chan->ch_subchan_lock, 0, "subch", 0);
1605
1606	i = 0;
1607	TAILQ_FOREACH(chan, &pri_chan->ch_subchans, ch_sublink) {
1608		/* TODO: refcnt chan */
1609		ret[i] = chan;
1610
1611		++i;
1612		if (i == subchan_cnt)
1613			break;
1614	}
1615	KASSERT(i == subchan_cnt, ("invalid subchan count %d, should be %d",
1616	    pri_chan->ch_subchan_cnt, subchan_cnt));
1617
1618	mtx_unlock(&pri_chan->ch_subchan_lock);
1619
1620	return ret;
1621}
1622
1623void
1624vmbus_subchan_rel(struct vmbus_channel **subchan, int subchan_cnt __unused)
1625{
1626
1627	free(subchan, M_TEMP);
1628}
1629
1630void
1631vmbus_subchan_drain(struct vmbus_channel *pri_chan)
1632{
1633	mtx_lock(&pri_chan->ch_subchan_lock);
1634	while (pri_chan->ch_subchan_cnt > 0)
1635		mtx_sleep(pri_chan, &pri_chan->ch_subchan_lock, 0, "dsubch", 0);
1636	mtx_unlock(&pri_chan->ch_subchan_lock);
1637}
1638
1639void
1640vmbus_chan_msgproc(struct vmbus_softc *sc, const struct vmbus_message *msg)
1641{
1642	vmbus_chanmsg_proc_t msg_proc;
1643	uint32_t msg_type;
1644
1645	msg_type = ((const struct vmbus_chanmsg_hdr *)msg->msg_data)->chm_type;
1646	KASSERT(msg_type < VMBUS_CHANMSG_TYPE_MAX,
1647	    ("invalid message type %u", msg_type));
1648
1649	msg_proc = vmbus_chan_msgprocs[msg_type];
1650	if (msg_proc != NULL)
1651		msg_proc(sc, msg);
1652}
1653
1654void
1655vmbus_chan_set_readbatch(struct vmbus_channel *chan, bool on)
1656{
1657	if (!on)
1658		chan->ch_flags &= ~VMBUS_CHAN_FLAG_BATCHREAD;
1659	else
1660		chan->ch_flags |= VMBUS_CHAN_FLAG_BATCHREAD;
1661}
1662
1663uint32_t
1664vmbus_chan_id(const struct vmbus_channel *chan)
1665{
1666	return chan->ch_id;
1667}
1668
1669uint32_t
1670vmbus_chan_subidx(const struct vmbus_channel *chan)
1671{
1672	return chan->ch_subidx;
1673}
1674
1675bool
1676vmbus_chan_is_primary(const struct vmbus_channel *chan)
1677{
1678	if (VMBUS_CHAN_ISPRIMARY(chan))
1679		return true;
1680	else
1681		return false;
1682}
1683
1684const struct hyperv_guid *
1685vmbus_chan_guid_inst(const struct vmbus_channel *chan)
1686{
1687	return &chan->ch_guid_inst;
1688}
1689
1690int
1691vmbus_chan_prplist_nelem(int br_size, int prpcnt_max, int dlen_max)
1692{
1693	int elem_size;
1694
1695	elem_size = __offsetof(struct vmbus_chanpkt_prplist,
1696	    cp_range[0].gpa_page[prpcnt_max]);
1697	elem_size += dlen_max;
1698	elem_size = VMBUS_CHANPKT_TOTLEN(elem_size);
1699
1700	return (vmbus_br_nelem(br_size, elem_size));
1701}
1702
1703bool
1704vmbus_chan_tx_empty(const struct vmbus_channel *chan)
1705{
1706
1707	return (vmbus_txbr_empty(&chan->ch_txbr));
1708}
1709
1710bool
1711vmbus_chan_rx_empty(const struct vmbus_channel *chan)
1712{
1713
1714	return (vmbus_rxbr_empty(&chan->ch_rxbr));
1715}
1716
1717void
1718vmbus_chan_run_task(struct vmbus_channel *chan, struct task *task)
1719{
1720
1721	taskqueue_enqueue(chan->ch_tq, task);
1722	taskqueue_drain(chan->ch_tq, task);
1723}
1724
1725struct taskqueue *
1726vmbus_chan_mgmt_tq(const struct vmbus_channel *chan)
1727{
1728
1729	return (chan->ch_mgmt_tq);
1730}
1731