xref: /freebsd/sys/dev/usb/misc/udbp.c (revision 9768746b)
1 /*-
2  * SPDX-License-Identifier: BSD-3-Clause
3  *
4  * Copyright (c) 1996-2000 Whistle Communications, 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, this list of conditions and the following disclaimer.
12  * 2. Redistributions in binary form must reproduce the above copyright
13  *    notice, this list of conditions and the following disclaimer in the
14  *    documentation and/or other materials provided with the distribution.
15  * 3. Neither the name of author nor the names of its
16  *    contributors may be used to endorse or promote products derived
17  *    from this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY NICK HIBMA AND CONTRIBUTORS
20  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
21  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
22  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS
23  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
24  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
25  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
26  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
27  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
28  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29  * POSSIBILITY OF SUCH DAMAGE.
30  *
31  */
32 
33 #include <sys/cdefs.h>
34 __FBSDID("$FreeBSD$");
35 
36 /* Driver for arbitrary double bulk pipe devices.
37  * The driver assumes that there will be the same driver on the other side.
38  *
39  * XXX Some more information on what the framing of the IP packets looks like.
40  *
41  * To take full advantage of bulk transmission, packets should be chosen
42  * between 1k and 5k in size (1k to make sure the sending side starts
43  * streaming, and <5k to avoid overflowing the system with small TDs).
44  */
45 
46 /* probe/attach/detach:
47  *  Connect the driver to the hardware and netgraph
48  *
49  *  The reason we submit a bulk in transfer is that USB does not know about
50  *  interrupts. The bulk transfer continuously polls the device for data.
51  *  While the device has no data available, the device NAKs the TDs. As soon
52  *  as there is data, the transfer happens and the data comes flowing in.
53  *
54  *  In case you were wondering, interrupt transfers happen exactly that way.
55  *  It therefore doesn't make sense to use the interrupt pipe to signal
56  *  'data ready' and then schedule a bulk transfer to fetch it. That would
57  *  incur a 2ms delay at least, without reducing bandwidth requirements.
58  *
59  */
60 
61 #include <sys/stdint.h>
62 #include <sys/stddef.h>
63 #include <sys/param.h>
64 #include <sys/queue.h>
65 #include <sys/types.h>
66 #include <sys/systm.h>
67 #include <sys/kernel.h>
68 #include <sys/bus.h>
69 #include <sys/module.h>
70 #include <sys/lock.h>
71 #include <sys/mutex.h>
72 #include <sys/condvar.h>
73 #include <sys/sysctl.h>
74 #include <sys/sx.h>
75 #include <sys/unistd.h>
76 #include <sys/callout.h>
77 #include <sys/malloc.h>
78 #include <sys/priv.h>
79 
80 #include <dev/usb/usb.h>
81 #include <dev/usb/usbdi.h>
82 #include <dev/usb/usbdi_util.h>
83 #include "usbdevs.h"
84 
85 #define	USB_DEBUG_VAR udbp_debug
86 #include <dev/usb/usb_debug.h>
87 
88 #include <sys/mbuf.h>
89 
90 #include <netgraph/ng_message.h>
91 #include <netgraph/netgraph.h>
92 #include <netgraph/ng_parse.h>
93 #include <netgraph/bluetooth/include/ng_bluetooth.h>
94 
95 #include <dev/usb/misc/udbp.h>
96 
97 #ifdef USB_DEBUG
98 static int udbp_debug = 0;
99 
100 static SYSCTL_NODE(_hw_usb, OID_AUTO, udbp, CTLFLAG_RW | CTLFLAG_MPSAFE, 0,
101     "USB udbp");
102 SYSCTL_INT(_hw_usb_udbp, OID_AUTO, debug, CTLFLAG_RWTUN,
103     &udbp_debug, 0, "udbp debug level");
104 #endif
105 
106 #define	UDBP_TIMEOUT	2000		/* timeout on outbound transfers, in
107 					 * msecs */
108 #define	UDBP_BUFFERSIZE	MCLBYTES	/* maximum number of bytes in one
109 					 * transfer */
110 #define	UDBP_T_WR       0
111 #define	UDBP_T_RD       1
112 #define	UDBP_T_WR_CS    2
113 #define	UDBP_T_RD_CS    3
114 #define	UDBP_T_MAX      4
115 #define	UDBP_Q_MAXLEN   50
116 
117 struct udbp_softc {
118 	struct mtx sc_mtx;
119 	struct ng_bt_mbufq sc_xmitq_hipri;	/* hi-priority transmit queue */
120 	struct ng_bt_mbufq sc_xmitq;	/* low-priority transmit queue */
121 
122 	struct usb_xfer *sc_xfer[UDBP_T_MAX];
123 	node_p	sc_node;		/* back pointer to node */
124 	hook_p	sc_hook;		/* pointer to the hook */
125 	struct mbuf *sc_bulk_in_buffer;
126 
127 	uint32_t sc_packets_in;		/* packets in from downstream */
128 	uint32_t sc_packets_out;	/* packets out towards downstream */
129 
130 	uint8_t	sc_flags;
131 #define	UDBP_FLAG_READ_STALL    0x01	/* read transfer stalled */
132 #define	UDBP_FLAG_WRITE_STALL   0x02	/* write transfer stalled */
133 
134 	uint8_t	sc_name[16];
135 };
136 
137 /* prototypes */
138 
139 static int udbp_modload(module_t mod, int event, void *data);
140 
141 static device_probe_t udbp_probe;
142 static device_attach_t udbp_attach;
143 static device_detach_t udbp_detach;
144 
145 static usb_callback_t udbp_bulk_read_callback;
146 static usb_callback_t udbp_bulk_read_clear_stall_callback;
147 static usb_callback_t udbp_bulk_write_callback;
148 static usb_callback_t udbp_bulk_write_clear_stall_callback;
149 
150 static void	udbp_bulk_read_complete(node_p, hook_p, void *, int);
151 
152 static ng_constructor_t	ng_udbp_constructor;
153 static ng_rcvmsg_t	ng_udbp_rcvmsg;
154 static ng_shutdown_t	ng_udbp_rmnode;
155 static ng_newhook_t	ng_udbp_newhook;
156 static ng_connect_t	ng_udbp_connect;
157 static ng_rcvdata_t	ng_udbp_rcvdata;
158 static ng_disconnect_t	ng_udbp_disconnect;
159 
160 /* Parse type for struct ngudbpstat */
161 static const struct ng_parse_struct_field
162 	ng_udbp_stat_type_fields[] = NG_UDBP_STATS_TYPE_INFO;
163 
164 static const struct ng_parse_type ng_udbp_stat_type = {
165 	&ng_parse_struct_type,
166 	&ng_udbp_stat_type_fields
167 };
168 
169 /* List of commands and how to convert arguments to/from ASCII */
170 static const struct ng_cmdlist ng_udbp_cmdlist[] = {
171 	{
172 		NGM_UDBP_COOKIE,
173 		NGM_UDBP_GET_STATUS,
174 		"getstatus",
175 		NULL,
176 		&ng_udbp_stat_type,
177 	},
178 	{
179 		NGM_UDBP_COOKIE,
180 		NGM_UDBP_SET_FLAG,
181 		"setflag",
182 		&ng_parse_int32_type,
183 		NULL
184 	},
185 	{0}
186 };
187 
188 /* Netgraph node type descriptor */
189 static struct ng_type ng_udbp_typestruct = {
190 	.version = NG_ABI_VERSION,
191 	.name = NG_UDBP_NODE_TYPE,
192 	.constructor = ng_udbp_constructor,
193 	.rcvmsg = ng_udbp_rcvmsg,
194 	.shutdown = ng_udbp_rmnode,
195 	.newhook = ng_udbp_newhook,
196 	.connect = ng_udbp_connect,
197 	.rcvdata = ng_udbp_rcvdata,
198 	.disconnect = ng_udbp_disconnect,
199 	.cmdlist = ng_udbp_cmdlist,
200 };
201 
202 /* USB config */
203 static const struct usb_config udbp_config[UDBP_T_MAX] = {
204 	[UDBP_T_WR] = {
205 		.type = UE_BULK,
206 		.endpoint = UE_ADDR_ANY,
207 		.direction = UE_DIR_OUT,
208 		.bufsize = UDBP_BUFFERSIZE,
209 		.flags = {.pipe_bof = 1,.force_short_xfer = 1,},
210 		.callback = &udbp_bulk_write_callback,
211 		.timeout = UDBP_TIMEOUT,
212 	},
213 
214 	[UDBP_T_RD] = {
215 		.type = UE_BULK,
216 		.endpoint = UE_ADDR_ANY,
217 		.direction = UE_DIR_IN,
218 		.bufsize = UDBP_BUFFERSIZE,
219 		.flags = {.pipe_bof = 1,.short_xfer_ok = 1,},
220 		.callback = &udbp_bulk_read_callback,
221 	},
222 
223 	[UDBP_T_WR_CS] = {
224 		.type = UE_CONTROL,
225 		.endpoint = 0x00,	/* Control pipe */
226 		.direction = UE_DIR_ANY,
227 		.bufsize = sizeof(struct usb_device_request),
228 		.callback = &udbp_bulk_write_clear_stall_callback,
229 		.timeout = 1000,	/* 1 second */
230 		.interval = 50,	/* 50ms */
231 	},
232 
233 	[UDBP_T_RD_CS] = {
234 		.type = UE_CONTROL,
235 		.endpoint = 0x00,	/* Control pipe */
236 		.direction = UE_DIR_ANY,
237 		.bufsize = sizeof(struct usb_device_request),
238 		.callback = &udbp_bulk_read_clear_stall_callback,
239 		.timeout = 1000,	/* 1 second */
240 		.interval = 50,	/* 50ms */
241 	},
242 };
243 
244 static device_method_t udbp_methods[] = {
245 	/* Device interface */
246 	DEVMETHOD(device_probe, udbp_probe),
247 	DEVMETHOD(device_attach, udbp_attach),
248 	DEVMETHOD(device_detach, udbp_detach),
249 
250 	DEVMETHOD_END
251 };
252 
253 static driver_t udbp_driver = {
254 	.name = "udbp",
255 	.methods = udbp_methods,
256 	.size = sizeof(struct udbp_softc),
257 };
258 
259 static const STRUCT_USB_HOST_ID udbp_devs[] = {
260 	{USB_VPI(USB_VENDOR_BELKIN, USB_PRODUCT_BELKIN_F5U258, 0)},
261 	{USB_VPI(USB_VENDOR_NETCHIP, USB_PRODUCT_NETCHIP_TURBOCONNECT, 0)},
262 	{USB_VPI(USB_VENDOR_NETCHIP, USB_PRODUCT_NETCHIP_GADGETZERO, 0)},
263 	{USB_VPI(USB_VENDOR_PROLIFIC, USB_PRODUCT_PROLIFIC_PL2301, 0)},
264 	{USB_VPI(USB_VENDOR_PROLIFIC, USB_PRODUCT_PROLIFIC_PL2302, 0)},
265 	{USB_VPI(USB_VENDOR_PROLIFIC, USB_PRODUCT_PROLIFIC_PL27A1, 0)},
266 	{USB_VPI(USB_VENDOR_ANCHOR, USB_PRODUCT_ANCHOR_EZLINK, 0)},
267 	{USB_VPI(USB_VENDOR_GENESYS, USB_PRODUCT_GENESYS_GL620USB, 0)},
268 };
269 
270 DRIVER_MODULE(udbp, uhub, udbp_driver, udbp_modload, NULL);
271 MODULE_DEPEND(udbp, netgraph, NG_ABI_VERSION, NG_ABI_VERSION, NG_ABI_VERSION);
272 MODULE_DEPEND(udbp, usb, 1, 1, 1);
273 MODULE_VERSION(udbp, 1);
274 USB_PNP_HOST_INFO(udbp_devs);
275 
276 static int
277 udbp_modload(module_t mod, int event, void *data)
278 {
279 	int error;
280 
281 	switch (event) {
282 	case MOD_LOAD:
283 		error = ng_newtype(&ng_udbp_typestruct);
284 		if (error != 0) {
285 			printf("%s: Could not register "
286 			    "Netgraph node type, error=%d\n",
287 			    NG_UDBP_NODE_TYPE, error);
288 		}
289 		break;
290 
291 	case MOD_UNLOAD:
292 		error = ng_rmtype(&ng_udbp_typestruct);
293 		break;
294 
295 	default:
296 		error = EOPNOTSUPP;
297 		break;
298 	}
299 	return (error);
300 }
301 
302 static int
303 udbp_probe(device_t dev)
304 {
305 	struct usb_attach_arg *uaa = device_get_ivars(dev);
306 
307 	if (uaa->usb_mode != USB_MODE_HOST)
308 		return (ENXIO);
309 	if (uaa->info.bConfigIndex != 0)
310 		return (ENXIO);
311 	if (uaa->info.bIfaceIndex != 0)
312 		return (ENXIO);
313 
314 	return (usbd_lookup_id_by_uaa(udbp_devs, sizeof(udbp_devs), uaa));
315 }
316 
317 static int
318 udbp_attach(device_t dev)
319 {
320 	struct usb_attach_arg *uaa = device_get_ivars(dev);
321 	struct udbp_softc *sc = device_get_softc(dev);
322 	int error;
323 
324 	device_set_usb_desc(dev);
325 
326 	snprintf(sc->sc_name, sizeof(sc->sc_name),
327 	    "%s", device_get_nameunit(dev));
328 
329 	mtx_init(&sc->sc_mtx, "udbp lock", NULL, MTX_DEF | MTX_RECURSE);
330 
331 	error = usbd_transfer_setup(uaa->device, &uaa->info.bIfaceIndex,
332 	    sc->sc_xfer, udbp_config, UDBP_T_MAX, sc, &sc->sc_mtx);
333 	if (error) {
334 		DPRINTF("error=%s\n", usbd_errstr(error));
335 		goto detach;
336 	}
337 	NG_BT_MBUFQ_INIT(&sc->sc_xmitq, UDBP_Q_MAXLEN);
338 
339 	NG_BT_MBUFQ_INIT(&sc->sc_xmitq_hipri, UDBP_Q_MAXLEN);
340 
341 	/* create Netgraph node */
342 
343 	if (ng_make_node_common(&ng_udbp_typestruct, &sc->sc_node) != 0) {
344 		printf("%s: Could not create Netgraph node\n",
345 		    sc->sc_name);
346 		sc->sc_node = NULL;
347 		goto detach;
348 	}
349 	/* name node */
350 
351 	if (ng_name_node(sc->sc_node, sc->sc_name) != 0) {
352 		printf("%s: Could not name node\n",
353 		    sc->sc_name);
354 		NG_NODE_UNREF(sc->sc_node);
355 		sc->sc_node = NULL;
356 		goto detach;
357 	}
358 	NG_NODE_SET_PRIVATE(sc->sc_node, sc);
359 
360 	/* the device is now operational */
361 
362 	return (0);			/* success */
363 
364 detach:
365 	udbp_detach(dev);
366 	return (ENOMEM);		/* failure */
367 }
368 
369 static int
370 udbp_detach(device_t dev)
371 {
372 	struct udbp_softc *sc = device_get_softc(dev);
373 
374 	/* destroy Netgraph node */
375 
376 	if (sc->sc_node != NULL) {
377 		NG_NODE_SET_PRIVATE(sc->sc_node, NULL);
378 		ng_rmnode_self(sc->sc_node);
379 		sc->sc_node = NULL;
380 	}
381 	/* free USB transfers, if any */
382 
383 	usbd_transfer_unsetup(sc->sc_xfer, UDBP_T_MAX);
384 
385 	mtx_destroy(&sc->sc_mtx);
386 
387 	/* destroy queues */
388 
389 	NG_BT_MBUFQ_DESTROY(&sc->sc_xmitq);
390 	NG_BT_MBUFQ_DESTROY(&sc->sc_xmitq_hipri);
391 
392 	/* extra check */
393 
394 	if (sc->sc_bulk_in_buffer) {
395 		m_freem(sc->sc_bulk_in_buffer);
396 		sc->sc_bulk_in_buffer = NULL;
397 	}
398 	return (0);			/* success */
399 }
400 
401 static void
402 udbp_bulk_read_callback(struct usb_xfer *xfer, usb_error_t error)
403 {
404 	struct udbp_softc *sc = usbd_xfer_softc(xfer);
405 	struct usb_page_cache *pc;
406 	struct mbuf *m;
407 	int actlen;
408 
409 	usbd_xfer_status(xfer, &actlen, NULL, NULL, NULL);
410 
411 	switch (USB_GET_STATE(xfer)) {
412 	case USB_ST_TRANSFERRED:
413 
414 		/* allocate new mbuf */
415 
416 		MGETHDR(m, M_NOWAIT, MT_DATA);
417 
418 		if (m == NULL) {
419 			goto tr_setup;
420 		}
421 
422 		if (!(MCLGET(m, M_NOWAIT))) {
423 			m_freem(m);
424 			goto tr_setup;
425 		}
426 		m->m_pkthdr.len = m->m_len = actlen;
427 
428 		pc = usbd_xfer_get_frame(xfer, 0);
429 		usbd_copy_out(pc, 0, m->m_data, actlen);
430 
431 		sc->sc_bulk_in_buffer = m;
432 
433 		DPRINTF("received package %d bytes\n", actlen);
434 
435 	case USB_ST_SETUP:
436 tr_setup:
437 		if (sc->sc_bulk_in_buffer) {
438 			ng_send_fn(sc->sc_node, NULL, &udbp_bulk_read_complete, NULL, 0);
439 			return;
440 		}
441 		if (sc->sc_flags & UDBP_FLAG_READ_STALL) {
442 			usbd_transfer_start(sc->sc_xfer[UDBP_T_RD_CS]);
443 			return;
444 		}
445 		usbd_xfer_set_frame_len(xfer, 0, usbd_xfer_max_len(xfer));
446 		usbd_transfer_submit(xfer);
447 		return;
448 
449 	default:			/* Error */
450 		if (error != USB_ERR_CANCELLED) {
451 			/* try to clear stall first */
452 			sc->sc_flags |= UDBP_FLAG_READ_STALL;
453 			usbd_transfer_start(sc->sc_xfer[UDBP_T_RD_CS]);
454 		}
455 		return;
456 	}
457 }
458 
459 static void
460 udbp_bulk_read_clear_stall_callback(struct usb_xfer *xfer, usb_error_t error)
461 {
462 	struct udbp_softc *sc = usbd_xfer_softc(xfer);
463 	struct usb_xfer *xfer_other = sc->sc_xfer[UDBP_T_RD];
464 
465 	if (usbd_clear_stall_callback(xfer, xfer_other)) {
466 		DPRINTF("stall cleared\n");
467 		sc->sc_flags &= ~UDBP_FLAG_READ_STALL;
468 		usbd_transfer_start(xfer_other);
469 	}
470 }
471 
472 static void
473 udbp_bulk_read_complete(node_p node, hook_p hook, void *arg1, int arg2)
474 {
475 	struct udbp_softc *sc = NG_NODE_PRIVATE(node);
476 	struct mbuf *m;
477 	int error;
478 
479 	if (sc == NULL) {
480 		return;
481 	}
482 	mtx_lock(&sc->sc_mtx);
483 
484 	m = sc->sc_bulk_in_buffer;
485 
486 	if (m) {
487 		sc->sc_bulk_in_buffer = NULL;
488 
489 		if ((sc->sc_hook == NULL) ||
490 		    NG_HOOK_NOT_VALID(sc->sc_hook)) {
491 			DPRINTF("No upstream hook\n");
492 			goto done;
493 		}
494 		sc->sc_packets_in++;
495 
496 		NG_SEND_DATA_ONLY(error, sc->sc_hook, m);
497 
498 		m = NULL;
499 	}
500 done:
501 	if (m) {
502 		m_freem(m);
503 	}
504 	/* start USB bulk-in transfer, if not already started */
505 
506 	usbd_transfer_start(sc->sc_xfer[UDBP_T_RD]);
507 
508 	mtx_unlock(&sc->sc_mtx);
509 }
510 
511 static void
512 udbp_bulk_write_callback(struct usb_xfer *xfer, usb_error_t error)
513 {
514 	struct udbp_softc *sc = usbd_xfer_softc(xfer);
515 	struct usb_page_cache *pc;
516 	struct mbuf *m;
517 
518 	switch (USB_GET_STATE(xfer)) {
519 	case USB_ST_TRANSFERRED:
520 
521 		sc->sc_packets_out++;
522 
523 	case USB_ST_SETUP:
524 		if (sc->sc_flags & UDBP_FLAG_WRITE_STALL) {
525 			usbd_transfer_start(sc->sc_xfer[UDBP_T_WR_CS]);
526 			return;
527 		}
528 		/* get next mbuf, if any */
529 
530 		NG_BT_MBUFQ_DEQUEUE(&sc->sc_xmitq_hipri, m);
531 		if (m == NULL) {
532 			NG_BT_MBUFQ_DEQUEUE(&sc->sc_xmitq, m);
533 			if (m == NULL) {
534 				DPRINTF("Data queue is empty\n");
535 				return;
536 			}
537 		}
538 		if (m->m_pkthdr.len > MCLBYTES) {
539 			DPRINTF("truncating large packet "
540 			    "from %d to %d bytes\n", m->m_pkthdr.len,
541 			    MCLBYTES);
542 			m->m_pkthdr.len = MCLBYTES;
543 		}
544 		pc = usbd_xfer_get_frame(xfer, 0);
545 		usbd_m_copy_in(pc, 0, m, 0, m->m_pkthdr.len);
546 
547 		usbd_xfer_set_frame_len(xfer, 0, m->m_pkthdr.len);
548 
549 		DPRINTF("packet out: %d bytes\n", m->m_pkthdr.len);
550 
551 		m_freem(m);
552 
553 		usbd_transfer_submit(xfer);
554 		return;
555 
556 	default:			/* Error */
557 		if (error != USB_ERR_CANCELLED) {
558 			/* try to clear stall first */
559 			sc->sc_flags |= UDBP_FLAG_WRITE_STALL;
560 			usbd_transfer_start(sc->sc_xfer[UDBP_T_WR_CS]);
561 		}
562 		return;
563 	}
564 }
565 
566 static void
567 udbp_bulk_write_clear_stall_callback(struct usb_xfer *xfer, usb_error_t error)
568 {
569 	struct udbp_softc *sc = usbd_xfer_softc(xfer);
570 	struct usb_xfer *xfer_other = sc->sc_xfer[UDBP_T_WR];
571 
572 	if (usbd_clear_stall_callback(xfer, xfer_other)) {
573 		DPRINTF("stall cleared\n");
574 		sc->sc_flags &= ~UDBP_FLAG_WRITE_STALL;
575 		usbd_transfer_start(xfer_other);
576 	}
577 }
578 
579 /***********************************************************************
580  * Start of Netgraph methods
581  **********************************************************************/
582 
583 /*
584  * If this is a device node so this work is done in the attach()
585  * routine and the constructor will return EINVAL as you should not be able
586  * to create nodes that depend on hardware (unless you can add the hardware :)
587  */
588 static int
589 ng_udbp_constructor(node_p node)
590 {
591 	return (EINVAL);
592 }
593 
594 /*
595  * Give our ok for a hook to be added...
596  * If we are not running this might kick a device into life.
597  * Possibly decode information out of the hook name.
598  * Add the hook's private info to the hook structure.
599  * (if we had some). In this example, we assume that there is a
600  * an array of structs, called 'channel' in the private info,
601  * one for each active channel. The private
602  * pointer of each hook points to the appropriate UDBP_hookinfo struct
603  * so that the source of an input packet is easily identified.
604  */
605 static int
606 ng_udbp_newhook(node_p node, hook_p hook, const char *name)
607 {
608 	struct udbp_softc *sc = NG_NODE_PRIVATE(node);
609 	int32_t error = 0;
610 
611 	if (strcmp(name, NG_UDBP_HOOK_NAME)) {
612 		return (EINVAL);
613 	}
614 	mtx_lock(&sc->sc_mtx);
615 
616 	if (sc->sc_hook != NULL) {
617 		error = EISCONN;
618 	} else {
619 		sc->sc_hook = hook;
620 		NG_HOOK_SET_PRIVATE(hook, NULL);
621 	}
622 
623 	mtx_unlock(&sc->sc_mtx);
624 
625 	return (error);
626 }
627 
628 /*
629  * Get a netgraph control message.
630  * Check it is one we understand. If needed, send a response.
631  * We could save the address for an async action later, but don't here.
632  * Always free the message.
633  * The response should be in a malloc'd region that the caller can 'free'.
634  * A response is not required.
635  * Theoretically you could respond defferently to old message types if
636  * the cookie in the header didn't match what we consider to be current
637  * (so that old userland programs could continue to work).
638  */
639 static int
640 ng_udbp_rcvmsg(node_p node, item_p item, hook_p lasthook)
641 {
642 	struct udbp_softc *sc = NG_NODE_PRIVATE(node);
643 	struct ng_mesg *resp = NULL;
644 	int error = 0;
645 	struct ng_mesg *msg;
646 
647 	NGI_GET_MSG(item, msg);
648 	/* Deal with message according to cookie and command */
649 	switch (msg->header.typecookie) {
650 	case NGM_UDBP_COOKIE:
651 		switch (msg->header.cmd) {
652 		case NGM_UDBP_GET_STATUS:
653 			{
654 				struct ngudbpstat *stats;
655 
656 				NG_MKRESPONSE(resp, msg, sizeof(*stats), M_NOWAIT);
657 				if (!resp) {
658 					error = ENOMEM;
659 					break;
660 				}
661 				stats = (struct ngudbpstat *)resp->data;
662 				mtx_lock(&sc->sc_mtx);
663 				stats->packets_in = sc->sc_packets_in;
664 				stats->packets_out = sc->sc_packets_out;
665 				mtx_unlock(&sc->sc_mtx);
666 				break;
667 			}
668 		case NGM_UDBP_SET_FLAG:
669 			if (msg->header.arglen != sizeof(uint32_t)) {
670 				error = EINVAL;
671 				break;
672 			}
673 			DPRINTF("flags = 0x%08x\n",
674 			    *((uint32_t *)msg->data));
675 			break;
676 		default:
677 			error = EINVAL;	/* unknown command */
678 			break;
679 		}
680 		break;
681 	default:
682 		error = EINVAL;		/* unknown cookie type */
683 		break;
684 	}
685 
686 	/* Take care of synchronous response, if any */
687 	NG_RESPOND_MSG(error, node, item, resp);
688 	NG_FREE_MSG(msg);
689 	return (error);
690 }
691 
692 /*
693  * Accept data from the hook and queue it for output.
694  */
695 static int
696 ng_udbp_rcvdata(hook_p hook, item_p item)
697 {
698 	struct udbp_softc *sc = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
699 	struct ng_bt_mbufq *queue_ptr;
700 	struct mbuf *m;
701 	struct ng_tag_prio *ptag;
702 	int error;
703 
704 	if (sc == NULL) {
705 		NG_FREE_ITEM(item);
706 		return (EHOSTDOWN);
707 	}
708 	NGI_GET_M(item, m);
709 	NG_FREE_ITEM(item);
710 
711 	/*
712 	 * Now queue the data for when it can be sent
713 	 */
714 	ptag = (void *)m_tag_locate(m, NGM_GENERIC_COOKIE,
715 	    NG_TAG_PRIO, NULL);
716 
717 	if (ptag && (ptag->priority > NG_PRIO_CUTOFF))
718 		queue_ptr = &sc->sc_xmitq_hipri;
719 	else
720 		queue_ptr = &sc->sc_xmitq;
721 
722 	mtx_lock(&sc->sc_mtx);
723 
724 	if (NG_BT_MBUFQ_FULL(queue_ptr)) {
725 		NG_BT_MBUFQ_DROP(queue_ptr);
726 		NG_FREE_M(m);
727 		error = ENOBUFS;
728 	} else {
729 		NG_BT_MBUFQ_ENQUEUE(queue_ptr, m);
730 		/*
731 		 * start bulk-out transfer, if not already started:
732 		 */
733 		usbd_transfer_start(sc->sc_xfer[UDBP_T_WR]);
734 		error = 0;
735 	}
736 
737 	mtx_unlock(&sc->sc_mtx);
738 
739 	return (error);
740 }
741 
742 /*
743  * Do local shutdown processing..
744  * We are a persistent device, we refuse to go away, and
745  * only remove our links and reset ourself.
746  */
747 static int
748 ng_udbp_rmnode(node_p node)
749 {
750 	struct udbp_softc *sc = NG_NODE_PRIVATE(node);
751 
752 	/* Let old node go */
753 	NG_NODE_SET_PRIVATE(node, NULL);
754 	NG_NODE_UNREF(node);		/* forget it ever existed */
755 
756 	if (sc == NULL) {
757 		goto done;
758 	}
759 	/* Create Netgraph node */
760 	if (ng_make_node_common(&ng_udbp_typestruct, &sc->sc_node) != 0) {
761 		printf("%s: Could not create Netgraph node\n",
762 		    sc->sc_name);
763 		sc->sc_node = NULL;
764 		goto done;
765 	}
766 	/* Name node */
767 	if (ng_name_node(sc->sc_node, sc->sc_name) != 0) {
768 		printf("%s: Could not name Netgraph node\n",
769 		    sc->sc_name);
770 		NG_NODE_UNREF(sc->sc_node);
771 		sc->sc_node = NULL;
772 		goto done;
773 	}
774 	NG_NODE_SET_PRIVATE(sc->sc_node, sc);
775 
776 done:
777 	if (sc) {
778 		mtx_unlock(&sc->sc_mtx);
779 	}
780 	return (0);
781 }
782 
783 /*
784  * This is called once we've already connected a new hook to the other node.
785  * It gives us a chance to balk at the last minute.
786  */
787 static int
788 ng_udbp_connect(hook_p hook)
789 {
790 	struct udbp_softc *sc = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
791 
792 	/* probably not at splnet, force outward queueing */
793 	NG_HOOK_FORCE_QUEUE(NG_HOOK_PEER(hook));
794 
795 	mtx_lock(&sc->sc_mtx);
796 
797 	sc->sc_flags |= (UDBP_FLAG_READ_STALL |
798 	    UDBP_FLAG_WRITE_STALL);
799 
800 	/* start bulk-in transfer */
801 	usbd_transfer_start(sc->sc_xfer[UDBP_T_RD]);
802 
803 	/* start bulk-out transfer */
804 	usbd_transfer_start(sc->sc_xfer[UDBP_T_WR]);
805 
806 	mtx_unlock(&sc->sc_mtx);
807 
808 	return (0);
809 }
810 
811 /*
812  * Dook disconnection
813  *
814  * For this type, removal of the last link destroys the node
815  */
816 static int
817 ng_udbp_disconnect(hook_p hook)
818 {
819 	struct udbp_softc *sc = NG_NODE_PRIVATE(NG_HOOK_NODE(hook));
820 	int error = 0;
821 
822 	if (sc != NULL) {
823 		mtx_lock(&sc->sc_mtx);
824 
825 		if (hook != sc->sc_hook) {
826 			error = EINVAL;
827 		} else {
828 			/* stop bulk-in transfer */
829 			usbd_transfer_stop(sc->sc_xfer[UDBP_T_RD_CS]);
830 			usbd_transfer_stop(sc->sc_xfer[UDBP_T_RD]);
831 
832 			/* stop bulk-out transfer */
833 			usbd_transfer_stop(sc->sc_xfer[UDBP_T_WR_CS]);
834 			usbd_transfer_stop(sc->sc_xfer[UDBP_T_WR]);
835 
836 			/* cleanup queues */
837 			NG_BT_MBUFQ_DRAIN(&sc->sc_xmitq);
838 			NG_BT_MBUFQ_DRAIN(&sc->sc_xmitq_hipri);
839 
840 			if (sc->sc_bulk_in_buffer) {
841 				m_freem(sc->sc_bulk_in_buffer);
842 				sc->sc_bulk_in_buffer = NULL;
843 			}
844 			sc->sc_hook = NULL;
845 		}
846 
847 		mtx_unlock(&sc->sc_mtx);
848 	}
849 	if ((NG_NODE_NUMHOOKS(NG_HOOK_NODE(hook)) == 0)
850 	    && (NG_NODE_IS_VALID(NG_HOOK_NODE(hook))))
851 		ng_rmnode_self(NG_HOOK_NODE(hook));
852 
853 	return (error);
854 }
855