1 /* $NetBSD: ugen.c,v 1.168 2021/09/26 01:16:09 thorpej Exp $ */ 2 3 /* 4 * Copyright (c) 1998, 2004 The NetBSD Foundation, Inc. 5 * All rights reserved. 6 * 7 * This code is derived from software contributed to The NetBSD Foundation 8 * by Lennart Augustsson (lennart@augustsson.net) at 9 * Carlstedt Research & Technology. 10 * 11 * Copyright (c) 2006 BBN Technologies Corp. All rights reserved. 12 * Effort sponsored in part by the Defense Advanced Research Projects 13 * Agency (DARPA) and the Department of the Interior National Business 14 * Center under agreement number NBCHC050166. 15 * 16 * Redistribution and use in source and binary forms, with or without 17 * modification, are permitted provided that the following conditions 18 * are met: 19 * 1. Redistributions of source code must retain the above copyright 20 * notice, this list of conditions and the following disclaimer. 21 * 2. Redistributions in binary form must reproduce the above copyright 22 * notice, this list of conditions and the following disclaimer in the 23 * documentation and/or other materials provided with the distribution. 24 * 25 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS 26 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED 27 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 28 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS 29 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 30 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 31 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 32 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 33 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 34 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 35 * POSSIBILITY OF SUCH DAMAGE. 36 */ 37 38 39 #include <sys/cdefs.h> 40 __KERNEL_RCSID(0, "$NetBSD: ugen.c,v 1.168 2021/09/26 01:16:09 thorpej Exp $"); 41 42 #ifdef _KERNEL_OPT 43 #include "opt_compat_netbsd.h" 44 #include "opt_usb.h" 45 #endif 46 47 #include <sys/param.h> 48 #include <sys/systm.h> 49 #include <sys/kernel.h> 50 #include <sys/kmem.h> 51 #include <sys/device.h> 52 #include <sys/ioctl.h> 53 #include <sys/conf.h> 54 #include <sys/tty.h> 55 #include <sys/file.h> 56 #include <sys/select.h> 57 #include <sys/proc.h> 58 #include <sys/vnode.h> 59 #include <sys/poll.h> 60 #include <sys/compat_stub.h> 61 #include <sys/module.h> 62 #include <sys/rbtree.h> 63 64 #include <dev/usb/usb.h> 65 #include <dev/usb/usbdi.h> 66 #include <dev/usb/usbdi_util.h> 67 68 #include "ioconf.h" 69 70 #ifdef UGEN_DEBUG 71 #define DPRINTF(x) if (ugendebug) printf x 72 #define DPRINTFN(n,x) if (ugendebug>(n)) printf x 73 int ugendebug = 0; 74 #else 75 #define DPRINTF(x) 76 #define DPRINTFN(n,x) 77 #endif 78 79 #define UGEN_CHUNK 128 /* chunk size for read */ 80 #define UGEN_IBSIZE 1020 /* buffer size */ 81 #define UGEN_BBSIZE 1024 82 83 #define UGEN_NISOREQS 4 /* number of outstanding xfer requests */ 84 #define UGEN_NISORFRMS 8 /* number of transactions per req */ 85 #define UGEN_NISOFRAMES (UGEN_NISORFRMS * UGEN_NISOREQS) 86 87 #define UGEN_BULK_RA_WB_BUFSIZE 16384 /* default buffer size */ 88 #define UGEN_BULK_RA_WB_BUFMAX (1 << 20) /* maximum allowed buffer */ 89 90 struct isoreq { 91 struct ugen_endpoint *sce; 92 struct usbd_xfer *xfer; 93 void *dmabuf; 94 uint16_t sizes[UGEN_NISORFRMS]; 95 }; 96 97 struct ugen_endpoint { 98 struct ugen_softc *sc; 99 usb_endpoint_descriptor_t *edesc; 100 struct usbd_interface *iface; 101 int state; 102 #define UGEN_SHORT_OK 0x04 /* short xfers are OK */ 103 #define UGEN_BULK_RA 0x08 /* in bulk read-ahead mode */ 104 #define UGEN_BULK_WB 0x10 /* in bulk write-behind mode */ 105 #define UGEN_RA_WB_STOP 0x20 /* RA/WB xfer is stopped (buffer full/empty) */ 106 struct usbd_pipe *pipeh; 107 struct clist q; 108 u_char *ibuf; /* start of buffer (circular for isoc) */ 109 u_char *fill; /* location for input (isoc) */ 110 u_char *limit; /* end of circular buffer (isoc) */ 111 u_char *cur; /* current read location (isoc) */ 112 uint32_t timeout; 113 uint32_t ra_wb_bufsize; /* requested size for RA/WB buffer */ 114 uint32_t ra_wb_reqsize; /* requested xfer length for RA/WB */ 115 uint32_t ra_wb_used; /* how much is in buffer */ 116 uint32_t ra_wb_xferlen; /* current xfer length for RA/WB */ 117 struct usbd_xfer *ra_wb_xfer; 118 struct isoreq isoreqs[UGEN_NISOREQS]; 119 /* Keep these last; we don't overwrite them in ugen_set_config() */ 120 #define UGEN_ENDPOINT_NONZERO_CRUFT offsetof(struct ugen_endpoint, rsel) 121 struct selinfo rsel; 122 kcondvar_t cv; 123 }; 124 125 struct ugen_softc { 126 device_t sc_dev; /* base device */ 127 struct usbd_device *sc_udev; 128 struct rb_node sc_node; 129 unsigned sc_unit; 130 131 kmutex_t sc_lock; 132 kcondvar_t sc_detach_cv; 133 134 char sc_is_open[USB_MAX_ENDPOINTS]; 135 struct ugen_endpoint sc_endpoints[USB_MAX_ENDPOINTS][2]; 136 #define OUT 0 137 #define IN 1 138 139 int sc_refcnt; 140 char sc_buffer[UGEN_BBSIZE]; 141 u_char sc_dying; 142 u_char sc_attached; 143 }; 144 145 static struct { 146 kmutex_t lock; 147 rb_tree_t tree; 148 } ugenif __cacheline_aligned; 149 150 static int 151 compare_ugen(void *cookie, const void *vsca, const void *vscb) 152 { 153 const struct ugen_softc *sca = vsca; 154 const struct ugen_softc *scb = vscb; 155 156 if (sca->sc_unit < scb->sc_unit) 157 return -1; 158 if (sca->sc_unit > scb->sc_unit) 159 return +1; 160 return 0; 161 } 162 163 static int 164 compare_ugen_key(void *cookie, const void *vsc, const void *vk) 165 { 166 const struct ugen_softc *sc = vsc; 167 const unsigned *k = vk; 168 169 if (sc->sc_unit < *k) 170 return -1; 171 if (sc->sc_unit > *k) 172 return +1; 173 return 0; 174 } 175 176 static const rb_tree_ops_t ugenif_tree_ops = { 177 .rbto_compare_nodes = compare_ugen, 178 .rbto_compare_key = compare_ugen_key, 179 .rbto_node_offset = offsetof(struct ugen_softc, sc_node), 180 }; 181 182 static void 183 ugenif_get_unit(struct ugen_softc *sc) 184 { 185 struct ugen_softc *sc0; 186 unsigned i; 187 188 mutex_enter(&ugenif.lock); 189 for (i = 0, sc0 = RB_TREE_MIN(&ugenif.tree); 190 sc0 != NULL && i == sc0->sc_unit; 191 i++, sc0 = RB_TREE_NEXT(&ugenif.tree, sc0)) 192 KASSERT(i < UINT_MAX); 193 KASSERT(rb_tree_find_node(&ugenif.tree, &i) == NULL); 194 sc->sc_unit = i; 195 sc0 = rb_tree_insert_node(&ugenif.tree, sc); 196 KASSERT(sc0 == sc); 197 KASSERT(rb_tree_find_node(&ugenif.tree, &i) == sc); 198 mutex_exit(&ugenif.lock); 199 } 200 201 static void 202 ugenif_put_unit(struct ugen_softc *sc) 203 { 204 205 mutex_enter(&ugenif.lock); 206 KASSERT(rb_tree_find_node(&ugenif.tree, &sc->sc_unit) == sc); 207 rb_tree_remove_node(&ugenif.tree, sc); 208 sc->sc_unit = -1; 209 mutex_exit(&ugenif.lock); 210 } 211 212 static struct ugen_softc * 213 ugenif_acquire(unsigned unit) 214 { 215 struct ugen_softc *sc; 216 217 mutex_enter(&ugenif.lock); 218 sc = rb_tree_find_node(&ugenif.tree, &unit); 219 if (sc == NULL) 220 goto out; 221 mutex_enter(&sc->sc_lock); 222 if (sc->sc_dying) { 223 mutex_exit(&sc->sc_lock); 224 sc = NULL; 225 goto out; 226 } 227 KASSERT(sc->sc_refcnt < INT_MAX); 228 sc->sc_refcnt++; 229 mutex_exit(&sc->sc_lock); 230 out: mutex_exit(&ugenif.lock); 231 232 return sc; 233 } 234 235 static void 236 ugenif_release(struct ugen_softc *sc) 237 { 238 239 mutex_enter(&sc->sc_lock); 240 if (--sc->sc_refcnt < 0) 241 cv_broadcast(&sc->sc_detach_cv); 242 mutex_exit(&sc->sc_lock); 243 } 244 245 static dev_type_open(ugenopen); 246 static dev_type_close(ugenclose); 247 static dev_type_read(ugenread); 248 static dev_type_write(ugenwrite); 249 static dev_type_ioctl(ugenioctl); 250 static dev_type_poll(ugenpoll); 251 static dev_type_kqfilter(ugenkqfilter); 252 253 const struct cdevsw ugen_cdevsw = { 254 .d_open = ugenopen, 255 .d_close = ugenclose, 256 .d_read = ugenread, 257 .d_write = ugenwrite, 258 .d_ioctl = ugenioctl, 259 .d_stop = nostop, 260 .d_tty = notty, 261 .d_poll = ugenpoll, 262 .d_mmap = nommap, 263 .d_kqfilter = ugenkqfilter, 264 .d_discard = nodiscard, 265 .d_flag = D_OTHER, 266 }; 267 268 Static void ugenintr(struct usbd_xfer *, void *, 269 usbd_status); 270 Static void ugen_isoc_rintr(struct usbd_xfer *, void *, 271 usbd_status); 272 Static void ugen_bulkra_intr(struct usbd_xfer *, void *, 273 usbd_status); 274 Static void ugen_bulkwb_intr(struct usbd_xfer *, void *, 275 usbd_status); 276 Static int ugen_do_read(struct ugen_softc *, int, struct uio *, int); 277 Static int ugen_do_write(struct ugen_softc *, int, struct uio *, int); 278 Static int ugen_do_ioctl(struct ugen_softc *, int, u_long, 279 void *, int, struct lwp *); 280 Static int ugen_set_config(struct ugen_softc *, int, int); 281 Static usb_config_descriptor_t *ugen_get_cdesc(struct ugen_softc *, 282 int, int *); 283 Static usbd_status ugen_set_interface(struct ugen_softc *, int, int); 284 Static int ugen_get_alt_index(struct ugen_softc *, int); 285 Static void ugen_clear_endpoints(struct ugen_softc *); 286 287 #define UGENUNIT(n) ((minor(n) >> 4) & 0xf) 288 #define UGENENDPOINT(n) (minor(n) & 0xf) 289 #define UGENDEV(u, e) (makedev(0, ((u) << 4) | (e))) 290 291 static int ugenif_match(device_t, cfdata_t, void *); 292 static void ugenif_attach(device_t, device_t, void *); 293 static int ugen_match(device_t, cfdata_t, void *); 294 static void ugen_attach(device_t, device_t, void *); 295 static int ugen_detach(device_t, int); 296 static int ugen_activate(device_t, enum devact); 297 298 CFATTACH_DECL_NEW(ugen, sizeof(struct ugen_softc), ugen_match, 299 ugen_attach, ugen_detach, ugen_activate); 300 CFATTACH_DECL_NEW(ugenif, sizeof(struct ugen_softc), ugenif_match, 301 ugenif_attach, ugen_detach, ugen_activate); 302 303 /* toggle to control attach priority. -1 means "let autoconf decide" */ 304 int ugen_override = -1; 305 306 static int 307 ugen_match(device_t parent, cfdata_t match, void *aux) 308 { 309 struct usb_attach_arg *uaa = aux; 310 int override; 311 312 if (ugen_override != -1) 313 override = ugen_override; 314 else 315 override = match->cf_flags & 1; 316 317 if (override) 318 return UMATCH_HIGHEST; 319 else if (uaa->uaa_usegeneric) 320 return UMATCH_GENERIC; 321 else 322 return UMATCH_NONE; 323 } 324 325 static int 326 ugenif_match(device_t parent, cfdata_t match, void *aux) 327 { 328 /* Assume that they knew what they configured! (see ugenif(4)) */ 329 return UMATCH_HIGHEST; 330 } 331 332 static void 333 ugen_attach(device_t parent, device_t self, void *aux) 334 { 335 struct usb_attach_arg *uaa = aux; 336 struct usbif_attach_arg uiaa; 337 338 memset(&uiaa, 0, sizeof(uiaa)); 339 uiaa.uiaa_port = uaa->uaa_port; 340 uiaa.uiaa_vendor = uaa->uaa_vendor; 341 uiaa.uiaa_product = uaa->uaa_product; 342 uiaa.uiaa_release = uaa->uaa_release; 343 uiaa.uiaa_device = uaa->uaa_device; 344 uiaa.uiaa_configno = -1; 345 uiaa.uiaa_ifaceno = -1; 346 347 ugenif_attach(parent, self, &uiaa); 348 } 349 350 static void 351 ugenif_attach(device_t parent, device_t self, void *aux) 352 { 353 struct ugen_softc *sc = device_private(self); 354 struct usbif_attach_arg *uiaa = aux; 355 struct usbd_device *udev; 356 char *devinfop; 357 usbd_status err; 358 int i, dir, conf; 359 360 aprint_naive("\n"); 361 aprint_normal("\n"); 362 363 mutex_init(&sc->sc_lock, MUTEX_DEFAULT, IPL_SOFTUSB); 364 cv_init(&sc->sc_detach_cv, "ugendet"); 365 366 devinfop = usbd_devinfo_alloc(uiaa->uiaa_device, 0); 367 aprint_normal_dev(self, "%s\n", devinfop); 368 usbd_devinfo_free(devinfop); 369 370 sc->sc_dev = self; 371 sc->sc_udev = udev = uiaa->uiaa_device; 372 373 for (i = 0; i < USB_MAX_ENDPOINTS; i++) { 374 for (dir = OUT; dir <= IN; dir++) { 375 struct ugen_endpoint *sce; 376 377 sce = &sc->sc_endpoints[i][dir]; 378 selinit(&sce->rsel); 379 cv_init(&sce->cv, "ugensce"); 380 } 381 } 382 383 if (!pmf_device_register(self, NULL, NULL)) 384 aprint_error_dev(self, "couldn't establish power handler\n"); 385 386 if (uiaa->uiaa_ifaceno < 0) { 387 /* 388 * If we attach the whole device, 389 * set configuration index 0, the default one. 390 */ 391 err = usbd_set_config_index(udev, 0, 0); 392 if (err) { 393 aprint_error_dev(self, 394 "setting configuration index 0 failed\n"); 395 return; 396 } 397 } 398 399 /* Get current configuration */ 400 conf = usbd_get_config_descriptor(udev)->bConfigurationValue; 401 402 /* Set up all the local state for this configuration. */ 403 err = ugen_set_config(sc, conf, uiaa->uiaa_ifaceno < 0); 404 if (err) { 405 aprint_error_dev(self, "setting configuration %d failed\n", 406 conf); 407 return; 408 } 409 410 ugenif_get_unit(sc); 411 usbd_add_drv_event(USB_EVENT_DRIVER_ATTACH, sc->sc_udev, sc->sc_dev); 412 sc->sc_attached = 1; 413 } 414 415 Static void 416 ugen_clear_endpoints(struct ugen_softc *sc) 417 { 418 419 /* Clear out the old info, but leave the selinfo and cv initialised. */ 420 for (int i = 0; i < USB_MAX_ENDPOINTS; i++) { 421 for (int dir = OUT; dir <= IN; dir++) { 422 struct ugen_endpoint *sce = &sc->sc_endpoints[i][dir]; 423 memset(sce, 0, UGEN_ENDPOINT_NONZERO_CRUFT); 424 } 425 } 426 } 427 428 Static int 429 ugen_set_config(struct ugen_softc *sc, int configno, int chkopen) 430 { 431 struct usbd_device *dev = sc->sc_udev; 432 usb_config_descriptor_t *cdesc; 433 struct usbd_interface *iface; 434 usb_endpoint_descriptor_t *ed; 435 struct ugen_endpoint *sce; 436 uint8_t niface, nendpt; 437 int ifaceno, endptno, endpt; 438 usbd_status err; 439 int dir; 440 441 DPRINTFN(1,("ugen_set_config: %s to configno %d, sc=%p\n", 442 device_xname(sc->sc_dev), configno, sc)); 443 444 KASSERT(KERNEL_LOCKED_P()); /* sc_is_open */ 445 446 if (chkopen) { 447 /* 448 * We start at 1, not 0, because we don't care whether the 449 * control endpoint is open or not. It is always present. 450 */ 451 for (endptno = 1; endptno < USB_MAX_ENDPOINTS; endptno++) 452 if (sc->sc_is_open[endptno]) { 453 DPRINTFN(1, 454 ("ugen_set_config: %s - endpoint %d is open\n", 455 device_xname(sc->sc_dev), endptno)); 456 return USBD_IN_USE; 457 } 458 459 /* Prevent opening while we're setting the config. */ 460 for (endptno = 1; endptno < USB_MAX_ENDPOINTS; endptno++) { 461 KASSERT(!sc->sc_is_open[endptno]); 462 sc->sc_is_open[endptno] = 1; 463 } 464 } 465 466 /* Avoid setting the current value. */ 467 cdesc = usbd_get_config_descriptor(dev); 468 if (!cdesc || cdesc->bConfigurationValue != configno) { 469 err = usbd_set_config_no(dev, configno, 1); 470 if (err) 471 goto out; 472 } 473 474 ugen_clear_endpoints(sc); 475 476 err = usbd_interface_count(dev, &niface); 477 if (err) 478 goto out; 479 480 for (ifaceno = 0; ifaceno < niface; ifaceno++) { 481 DPRINTFN(1,("ugen_set_config: ifaceno %d\n", ifaceno)); 482 err = usbd_device2interface_handle(dev, ifaceno, &iface); 483 if (err) 484 goto out; 485 err = usbd_endpoint_count(iface, &nendpt); 486 if (err) 487 goto out; 488 for (endptno = 0; endptno < nendpt; endptno++) { 489 ed = usbd_interface2endpoint_descriptor(iface,endptno); 490 KASSERT(ed != NULL); 491 endpt = ed->bEndpointAddress; 492 dir = UE_GET_DIR(endpt) == UE_DIR_IN ? IN : OUT; 493 sce = &sc->sc_endpoints[UE_GET_ADDR(endpt)][dir]; 494 DPRINTFN(1,("ugen_set_config: endptno %d, endpt=0x%02x" 495 "(%d,%d), sce=%p\n", 496 endptno, endpt, UE_GET_ADDR(endpt), 497 UE_GET_DIR(endpt), sce)); 498 sce->sc = sc; 499 sce->edesc = ed; 500 sce->iface = iface; 501 } 502 } 503 err = USBD_NORMAL_COMPLETION; 504 505 out: if (chkopen) { 506 /* 507 * Allow open again now that we're done trying to set 508 * the config. 509 */ 510 for (endptno = 1; endptno < USB_MAX_ENDPOINTS; endptno++) { 511 KASSERT(sc->sc_is_open[endptno]); 512 sc->sc_is_open[endptno] = 0; 513 } 514 } 515 return err; 516 } 517 518 static int 519 ugenopen(dev_t dev, int flag, int mode, struct lwp *l) 520 { 521 struct ugen_softc *sc; 522 int unit = UGENUNIT(dev); 523 int endpt = UGENENDPOINT(dev); 524 usb_endpoint_descriptor_t *edesc; 525 struct ugen_endpoint *sce; 526 int dir, isize; 527 usbd_status err; 528 struct usbd_xfer *xfer; 529 int i, j; 530 int error; 531 int opened; 532 533 KASSERT(KERNEL_LOCKED_P()); /* sc_is_open */ 534 535 if ((sc = ugenif_acquire(unit)) == NULL) 536 return ENXIO; 537 538 DPRINTFN(5, ("ugenopen: flag=%d, mode=%d, unit=%d endpt=%d\n", 539 flag, mode, unit, endpt)); 540 541 /* The control endpoint allows multiple opens. */ 542 if (endpt == USB_CONTROL_ENDPOINT) { 543 opened = sc->sc_is_open[USB_CONTROL_ENDPOINT] = 1; 544 error = 0; 545 goto out; 546 } 547 548 if (sc->sc_is_open[endpt]) { 549 error = EBUSY; 550 goto out; 551 } 552 opened = sc->sc_is_open[endpt] = 1; 553 554 /* Make sure there are pipes for all directions. */ 555 for (dir = OUT; dir <= IN; dir++) { 556 if (flag & (dir == OUT ? FWRITE : FREAD)) { 557 sce = &sc->sc_endpoints[endpt][dir]; 558 if (sce->edesc == NULL) { 559 error = ENXIO; 560 goto out; 561 } 562 } 563 } 564 565 /* Actually open the pipes. */ 566 /* XXX Should back out properly if it fails. */ 567 for (dir = OUT; dir <= IN; dir++) { 568 if (!(flag & (dir == OUT ? FWRITE : FREAD))) 569 continue; 570 sce = &sc->sc_endpoints[endpt][dir]; 571 sce->state = 0; 572 sce->timeout = USBD_NO_TIMEOUT; 573 DPRINTFN(5, ("ugenopen: sc=%p, endpt=%d, dir=%d, sce=%p\n", 574 sc, endpt, dir, sce)); 575 edesc = sce->edesc; 576 switch (edesc->bmAttributes & UE_XFERTYPE) { 577 case UE_INTERRUPT: 578 if (dir == OUT) { 579 err = usbd_open_pipe(sce->iface, 580 edesc->bEndpointAddress, 0, &sce->pipeh); 581 if (err) { 582 error = EIO; 583 goto out; 584 } 585 break; 586 } 587 isize = UGETW(edesc->wMaxPacketSize); 588 if (isize == 0) { /* shouldn't happen */ 589 error = EINVAL; 590 goto out; 591 } 592 sce->ibuf = kmem_alloc(isize, KM_SLEEP); 593 DPRINTFN(5, ("ugenopen: intr endpt=%d,isize=%d\n", 594 endpt, isize)); 595 if (clalloc(&sce->q, UGEN_IBSIZE, 0) == -1) { 596 kmem_free(sce->ibuf, isize); 597 sce->ibuf = NULL; 598 error = ENOMEM; 599 goto out; 600 } 601 err = usbd_open_pipe_intr(sce->iface, 602 edesc->bEndpointAddress, 603 USBD_SHORT_XFER_OK, &sce->pipeh, sce, 604 sce->ibuf, isize, ugenintr, 605 USBD_DEFAULT_INTERVAL); 606 if (err) { 607 clfree(&sce->q); 608 kmem_free(sce->ibuf, isize); 609 sce->ibuf = NULL; 610 error = EIO; 611 goto out; 612 } 613 DPRINTFN(5, ("ugenopen: interrupt open done\n")); 614 break; 615 case UE_BULK: 616 err = usbd_open_pipe(sce->iface, 617 edesc->bEndpointAddress, 0, &sce->pipeh); 618 if (err) { 619 error = EIO; 620 goto out; 621 } 622 sce->ra_wb_bufsize = UGEN_BULK_RA_WB_BUFSIZE; 623 /* 624 * Use request size for non-RA/WB transfers 625 * as the default. 626 */ 627 sce->ra_wb_reqsize = UGEN_BBSIZE; 628 break; 629 case UE_ISOCHRONOUS: 630 if (dir == OUT) { 631 error = EINVAL; 632 goto out; 633 } 634 isize = UGETW(edesc->wMaxPacketSize); 635 if (isize == 0) { /* shouldn't happen */ 636 error = EINVAL; 637 goto out; 638 } 639 sce->ibuf = kmem_alloc(isize * UGEN_NISOFRAMES, 640 KM_SLEEP); 641 sce->cur = sce->fill = sce->ibuf; 642 sce->limit = sce->ibuf + isize * UGEN_NISOFRAMES; 643 DPRINTFN(5, ("ugenopen: isoc endpt=%d, isize=%d\n", 644 endpt, isize)); 645 err = usbd_open_pipe(sce->iface, 646 edesc->bEndpointAddress, 0, &sce->pipeh); 647 if (err) { 648 kmem_free(sce->ibuf, isize * UGEN_NISOFRAMES); 649 sce->ibuf = NULL; 650 error = EIO; 651 goto out; 652 } 653 for (i = 0; i < UGEN_NISOREQS; ++i) { 654 sce->isoreqs[i].sce = sce; 655 err = usbd_create_xfer(sce->pipeh, 656 isize * UGEN_NISORFRMS, 0, UGEN_NISORFRMS, 657 &xfer); 658 if (err) 659 goto bad; 660 sce->isoreqs[i].xfer = xfer; 661 sce->isoreqs[i].dmabuf = usbd_get_buffer(xfer); 662 for (j = 0; j < UGEN_NISORFRMS; ++j) 663 sce->isoreqs[i].sizes[j] = isize; 664 usbd_setup_isoc_xfer(xfer, &sce->isoreqs[i], 665 sce->isoreqs[i].sizes, UGEN_NISORFRMS, 0, 666 ugen_isoc_rintr); 667 (void)usbd_transfer(xfer); 668 } 669 DPRINTFN(5, ("ugenopen: isoc open done\n")); 670 break; 671 bad: 672 while (--i >= 0) { /* implicit buffer free */ 673 usbd_destroy_xfer(sce->isoreqs[i].xfer); 674 sce->isoreqs[i].xfer = NULL; 675 } 676 usbd_close_pipe(sce->pipeh); 677 sce->pipeh = NULL; 678 kmem_free(sce->ibuf, isize * UGEN_NISOFRAMES); 679 sce->ibuf = NULL; 680 error = ENOMEM; 681 goto out; 682 case UE_CONTROL: 683 sce->timeout = USBD_DEFAULT_TIMEOUT; 684 error = EINVAL; 685 goto out; 686 } 687 } 688 error = 0; 689 out: if (error && opened) 690 sc->sc_is_open[endpt] = 0; 691 ugenif_release(sc); 692 return error; 693 } 694 695 static void 696 ugen_do_close(struct ugen_softc *sc, int flag, int endpt) 697 { 698 struct ugen_endpoint *sce; 699 int dir; 700 int i; 701 702 KASSERT(KERNEL_LOCKED_P()); /* sc_is_open */ 703 704 if (!sc->sc_is_open[endpt]) 705 goto out; 706 707 if (endpt == USB_CONTROL_ENDPOINT) { 708 DPRINTFN(5, ("ugenclose: close control\n")); 709 goto out; 710 } 711 712 for (dir = OUT; dir <= IN; dir++) { 713 if (!(flag & (dir == OUT ? FWRITE : FREAD))) 714 continue; 715 sce = &sc->sc_endpoints[endpt][dir]; 716 if (sce->pipeh == NULL) 717 continue; 718 DPRINTFN(5, ("ugenclose: endpt=%d dir=%d sce=%p\n", 719 endpt, dir, sce)); 720 721 usbd_abort_pipe(sce->pipeh); 722 723 int isize = UGETW(sce->edesc->wMaxPacketSize); 724 int msize = 0; 725 726 switch (sce->edesc->bmAttributes & UE_XFERTYPE) { 727 case UE_INTERRUPT: 728 ndflush(&sce->q, sce->q.c_cc); 729 clfree(&sce->q); 730 msize = isize; 731 break; 732 case UE_ISOCHRONOUS: 733 for (i = 0; i < UGEN_NISOREQS; ++i) { 734 usbd_destroy_xfer(sce->isoreqs[i].xfer); 735 sce->isoreqs[i].xfer = NULL; 736 } 737 msize = isize * UGEN_NISOFRAMES; 738 break; 739 case UE_BULK: 740 if (sce->state & (UGEN_BULK_RA | UGEN_BULK_WB)) { 741 usbd_destroy_xfer(sce->ra_wb_xfer); 742 sce->ra_wb_xfer = NULL; 743 msize = sce->ra_wb_bufsize; 744 } 745 break; 746 default: 747 break; 748 } 749 usbd_close_pipe(sce->pipeh); 750 sce->pipeh = NULL; 751 if (sce->ibuf != NULL) { 752 kmem_free(sce->ibuf, msize); 753 sce->ibuf = NULL; 754 } 755 } 756 757 out: sc->sc_is_open[endpt] = 0; 758 for (dir = OUT; dir <= IN; dir++) { 759 sce = &sc->sc_endpoints[endpt][dir]; 760 KASSERT(sce->pipeh == NULL); 761 KASSERT(sce->ibuf == NULL); 762 KASSERT(sce->ra_wb_xfer == NULL); 763 for (i = 0; i < UGEN_NISOREQS; i++) 764 KASSERT(sce->isoreqs[i].xfer == NULL); 765 } 766 } 767 768 static int 769 ugenclose(dev_t dev, int flag, int mode, struct lwp *l) 770 { 771 int endpt = UGENENDPOINT(dev); 772 struct ugen_softc *sc; 773 774 DPRINTFN(5, ("ugenclose: flag=%d, mode=%d, unit=%d, endpt=%d\n", 775 flag, mode, UGENUNIT(dev), endpt)); 776 777 KASSERT(KERNEL_LOCKED_P()); /* ugen_do_close */ 778 779 if ((sc = ugenif_acquire(UGENUNIT(dev))) == NULL) 780 return ENXIO; 781 782 KASSERT(sc->sc_is_open[endpt]); 783 ugen_do_close(sc, flag, endpt); 784 KASSERT(!sc->sc_is_open[endpt]); 785 786 ugenif_release(sc); 787 788 return 0; 789 } 790 791 Static int 792 ugen_do_read(struct ugen_softc *sc, int endpt, struct uio *uio, int flag) 793 { 794 struct ugen_endpoint *sce = &sc->sc_endpoints[endpt][IN]; 795 uint32_t n, tn; 796 struct usbd_xfer *xfer; 797 usbd_status err; 798 int error = 0; 799 800 DPRINTFN(5, ("%s: ugenread: %d\n", device_xname(sc->sc_dev), endpt)); 801 802 if (endpt == USB_CONTROL_ENDPOINT) 803 return ENODEV; 804 805 KASSERT(sce->edesc); 806 KASSERT(sce->pipeh); 807 808 switch (sce->edesc->bmAttributes & UE_XFERTYPE) { 809 case UE_INTERRUPT: 810 /* Block until activity occurred. */ 811 mutex_enter(&sc->sc_lock); 812 while (sce->q.c_cc == 0) { 813 if (flag & IO_NDELAY) { 814 mutex_exit(&sc->sc_lock); 815 return EWOULDBLOCK; 816 } 817 DPRINTFN(5, ("ugenread: sleep on %p\n", sce)); 818 /* "ugenri" */ 819 error = cv_timedwait_sig(&sce->cv, &sc->sc_lock, 820 mstohz(sce->timeout)); 821 DPRINTFN(5, ("ugenread: woke, error=%d\n", error)); 822 if (sc->sc_dying) 823 error = EIO; 824 if (error) 825 break; 826 } 827 mutex_exit(&sc->sc_lock); 828 829 /* Transfer as many chunks as possible. */ 830 while (sce->q.c_cc > 0 && uio->uio_resid > 0 && !error) { 831 n = uimin(sce->q.c_cc, uio->uio_resid); 832 if (n > sizeof(sc->sc_buffer)) 833 n = sizeof(sc->sc_buffer); 834 835 /* Remove a small chunk from the input queue. */ 836 q_to_b(&sce->q, sc->sc_buffer, n); 837 DPRINTFN(5, ("ugenread: got %d chars\n", n)); 838 839 /* Copy the data to the user process. */ 840 error = uiomove(sc->sc_buffer, n, uio); 841 if (error) 842 break; 843 } 844 break; 845 case UE_BULK: 846 if (sce->state & UGEN_BULK_RA) { 847 DPRINTFN(5, ("ugenread: BULK_RA req: %zd used: %d\n", 848 uio->uio_resid, sce->ra_wb_used)); 849 xfer = sce->ra_wb_xfer; 850 851 mutex_enter(&sc->sc_lock); 852 if (sce->ra_wb_used == 0 && flag & IO_NDELAY) { 853 mutex_exit(&sc->sc_lock); 854 return EWOULDBLOCK; 855 } 856 while (uio->uio_resid > 0 && !error) { 857 while (sce->ra_wb_used == 0) { 858 DPRINTFN(5, 859 ("ugenread: sleep on %p\n", 860 sce)); 861 /* "ugenrb" */ 862 error = cv_timedwait_sig(&sce->cv, 863 &sc->sc_lock, mstohz(sce->timeout)); 864 DPRINTFN(5, 865 ("ugenread: woke, error=%d\n", 866 error)); 867 if (sc->sc_dying) 868 error = EIO; 869 if (error) 870 break; 871 } 872 873 /* Copy data to the process. */ 874 while (uio->uio_resid > 0 875 && sce->ra_wb_used > 0) { 876 n = uimin(uio->uio_resid, 877 sce->ra_wb_used); 878 n = uimin(n, sce->limit - sce->cur); 879 error = uiomove(sce->cur, n, uio); 880 if (error) 881 break; 882 sce->cur += n; 883 sce->ra_wb_used -= n; 884 if (sce->cur == sce->limit) 885 sce->cur = sce->ibuf; 886 } 887 888 /* 889 * If the transfers stopped because the 890 * buffer was full, restart them. 891 */ 892 if (sce->state & UGEN_RA_WB_STOP && 893 sce->ra_wb_used < sce->limit - sce->ibuf) { 894 n = (sce->limit - sce->ibuf) 895 - sce->ra_wb_used; 896 usbd_setup_xfer(xfer, sce, NULL, 897 uimin(n, sce->ra_wb_xferlen), 898 0, USBD_NO_TIMEOUT, 899 ugen_bulkra_intr); 900 sce->state &= ~UGEN_RA_WB_STOP; 901 err = usbd_transfer(xfer); 902 if (err != USBD_IN_PROGRESS) 903 /* 904 * The transfer has not been 905 * queued. Setting STOP 906 * will make us try 907 * again at the next read. 908 */ 909 sce->state |= UGEN_RA_WB_STOP; 910 } 911 } 912 mutex_exit(&sc->sc_lock); 913 break; 914 } 915 error = usbd_create_xfer(sce->pipeh, UGEN_BBSIZE, 916 0, 0, &xfer); 917 if (error) 918 return error; 919 while ((n = uimin(UGEN_BBSIZE, uio->uio_resid)) != 0) { 920 DPRINTFN(1, ("ugenread: start transfer %d bytes\n",n)); 921 tn = n; 922 err = usbd_bulk_transfer(xfer, sce->pipeh, 923 sce->state & UGEN_SHORT_OK ? USBD_SHORT_XFER_OK : 0, 924 sce->timeout, sc->sc_buffer, &tn); 925 if (err) { 926 if (err == USBD_INTERRUPTED) 927 error = EINTR; 928 else if (err == USBD_TIMEOUT) 929 error = ETIMEDOUT; 930 else 931 error = EIO; 932 break; 933 } 934 DPRINTFN(1, ("ugenread: got %d bytes\n", tn)); 935 error = uiomove(sc->sc_buffer, tn, uio); 936 if (error || tn < n) 937 break; 938 } 939 usbd_destroy_xfer(xfer); 940 break; 941 case UE_ISOCHRONOUS: 942 mutex_enter(&sc->sc_lock); 943 while (sce->cur == sce->fill) { 944 if (flag & IO_NDELAY) { 945 mutex_exit(&sc->sc_lock); 946 return EWOULDBLOCK; 947 } 948 /* "ugenri" */ 949 DPRINTFN(5, ("ugenread: sleep on %p\n", sce)); 950 error = cv_timedwait_sig(&sce->cv, &sc->sc_lock, 951 mstohz(sce->timeout)); 952 DPRINTFN(5, ("ugenread: woke, error=%d\n", error)); 953 if (sc->sc_dying) 954 error = EIO; 955 if (error) 956 break; 957 } 958 959 while (sce->cur != sce->fill && uio->uio_resid > 0 && !error) { 960 if(sce->fill > sce->cur) 961 n = uimin(sce->fill - sce->cur, uio->uio_resid); 962 else 963 n = uimin(sce->limit - sce->cur, uio->uio_resid); 964 965 DPRINTFN(5, ("ugenread: isoc got %d chars\n", n)); 966 967 /* Copy the data to the user process. */ 968 error = uiomove(sce->cur, n, uio); 969 if (error) 970 break; 971 sce->cur += n; 972 if (sce->cur >= sce->limit) 973 sce->cur = sce->ibuf; 974 } 975 mutex_exit(&sc->sc_lock); 976 break; 977 978 979 default: 980 return ENXIO; 981 } 982 return error; 983 } 984 985 static int 986 ugenread(dev_t dev, struct uio *uio, int flag) 987 { 988 int endpt = UGENENDPOINT(dev); 989 struct ugen_softc *sc; 990 int error; 991 992 if ((sc = ugenif_acquire(UGENUNIT(dev))) == NULL) 993 return ENXIO; 994 error = ugen_do_read(sc, endpt, uio, flag); 995 ugenif_release(sc); 996 997 return error; 998 } 999 1000 Static int 1001 ugen_do_write(struct ugen_softc *sc, int endpt, struct uio *uio, 1002 int flag) 1003 { 1004 struct ugen_endpoint *sce = &sc->sc_endpoints[endpt][OUT]; 1005 uint32_t n; 1006 int error = 0; 1007 uint32_t tn; 1008 char *dbuf; 1009 struct usbd_xfer *xfer; 1010 usbd_status err; 1011 1012 DPRINTFN(5, ("%s: ugenwrite: %d\n", device_xname(sc->sc_dev), endpt)); 1013 1014 if (endpt == USB_CONTROL_ENDPOINT) 1015 return ENODEV; 1016 1017 KASSERT(sce->edesc); 1018 KASSERT(sce->pipeh); 1019 1020 switch (sce->edesc->bmAttributes & UE_XFERTYPE) { 1021 case UE_BULK: 1022 if (sce->state & UGEN_BULK_WB) { 1023 DPRINTFN(5, ("ugenwrite: BULK_WB req: %zd used: %d\n", 1024 uio->uio_resid, sce->ra_wb_used)); 1025 xfer = sce->ra_wb_xfer; 1026 1027 mutex_enter(&sc->sc_lock); 1028 if (sce->ra_wb_used == sce->limit - sce->ibuf && 1029 flag & IO_NDELAY) { 1030 mutex_exit(&sc->sc_lock); 1031 return EWOULDBLOCK; 1032 } 1033 while (uio->uio_resid > 0 && !error) { 1034 while (sce->ra_wb_used == 1035 sce->limit - sce->ibuf) { 1036 DPRINTFN(5, 1037 ("ugenwrite: sleep on %p\n", 1038 sce)); 1039 /* "ugenwb" */ 1040 error = cv_timedwait_sig(&sce->cv, 1041 &sc->sc_lock, mstohz(sce->timeout)); 1042 DPRINTFN(5, 1043 ("ugenwrite: woke, error=%d\n", 1044 error)); 1045 if (sc->sc_dying) 1046 error = EIO; 1047 if (error) 1048 break; 1049 } 1050 1051 /* Copy data from the process. */ 1052 while (uio->uio_resid > 0 && 1053 sce->ra_wb_used < sce->limit - sce->ibuf) { 1054 n = uimin(uio->uio_resid, 1055 (sce->limit - sce->ibuf) 1056 - sce->ra_wb_used); 1057 n = uimin(n, sce->limit - sce->fill); 1058 error = uiomove(sce->fill, n, uio); 1059 if (error) 1060 break; 1061 sce->fill += n; 1062 sce->ra_wb_used += n; 1063 if (sce->fill == sce->limit) 1064 sce->fill = sce->ibuf; 1065 } 1066 1067 /* 1068 * If the transfers stopped because the 1069 * buffer was empty, restart them. 1070 */ 1071 if (sce->state & UGEN_RA_WB_STOP && 1072 sce->ra_wb_used > 0) { 1073 dbuf = (char *)usbd_get_buffer(xfer); 1074 n = uimin(sce->ra_wb_used, 1075 sce->ra_wb_xferlen); 1076 tn = uimin(n, sce->limit - sce->cur); 1077 memcpy(dbuf, sce->cur, tn); 1078 dbuf += tn; 1079 if (n - tn > 0) 1080 memcpy(dbuf, sce->ibuf, 1081 n - tn); 1082 usbd_setup_xfer(xfer, sce, NULL, n, 1083 0, USBD_NO_TIMEOUT, 1084 ugen_bulkwb_intr); 1085 sce->state &= ~UGEN_RA_WB_STOP; 1086 err = usbd_transfer(xfer); 1087 if (err != USBD_IN_PROGRESS) 1088 /* 1089 * The transfer has not been 1090 * queued. Setting STOP 1091 * will make us try again 1092 * at the next read. 1093 */ 1094 sce->state |= UGEN_RA_WB_STOP; 1095 } 1096 } 1097 mutex_exit(&sc->sc_lock); 1098 break; 1099 } 1100 error = usbd_create_xfer(sce->pipeh, UGEN_BBSIZE, 1101 0, 0, &xfer); 1102 if (error) 1103 return error; 1104 while ((n = uimin(UGEN_BBSIZE, uio->uio_resid)) != 0) { 1105 error = uiomove(sc->sc_buffer, n, uio); 1106 if (error) 1107 break; 1108 DPRINTFN(1, ("ugenwrite: transfer %d bytes\n", n)); 1109 err = usbd_bulk_transfer(xfer, sce->pipeh, 0, sce->timeout, 1110 sc->sc_buffer, &n); 1111 if (err) { 1112 if (err == USBD_INTERRUPTED) 1113 error = EINTR; 1114 else if (err == USBD_TIMEOUT) 1115 error = ETIMEDOUT; 1116 else 1117 error = EIO; 1118 break; 1119 } 1120 } 1121 usbd_destroy_xfer(xfer); 1122 break; 1123 case UE_INTERRUPT: 1124 error = usbd_create_xfer(sce->pipeh, 1125 UGETW(sce->edesc->wMaxPacketSize), 0, 0, &xfer); 1126 if (error) 1127 return error; 1128 while ((n = uimin(UGETW(sce->edesc->wMaxPacketSize), 1129 uio->uio_resid)) != 0) { 1130 error = uiomove(sc->sc_buffer, n, uio); 1131 if (error) 1132 break; 1133 DPRINTFN(1, ("ugenwrite: transfer %d bytes\n", n)); 1134 err = usbd_intr_transfer(xfer, sce->pipeh, 0, 1135 sce->timeout, sc->sc_buffer, &n); 1136 if (err) { 1137 if (err == USBD_INTERRUPTED) 1138 error = EINTR; 1139 else if (err == USBD_TIMEOUT) 1140 error = ETIMEDOUT; 1141 else 1142 error = EIO; 1143 break; 1144 } 1145 } 1146 usbd_destroy_xfer(xfer); 1147 break; 1148 default: 1149 return ENXIO; 1150 } 1151 return error; 1152 } 1153 1154 static int 1155 ugenwrite(dev_t dev, struct uio *uio, int flag) 1156 { 1157 int endpt = UGENENDPOINT(dev); 1158 struct ugen_softc *sc; 1159 int error; 1160 1161 if ((sc = ugenif_acquire(UGENUNIT(dev))) == NULL) 1162 return ENXIO; 1163 error = ugen_do_write(sc, endpt, uio, flag); 1164 ugenif_release(sc); 1165 1166 return error; 1167 } 1168 1169 static int 1170 ugen_activate(device_t self, enum devact act) 1171 { 1172 struct ugen_softc *sc = device_private(self); 1173 1174 switch (act) { 1175 case DVACT_DEACTIVATE: 1176 sc->sc_dying = 1; 1177 return 0; 1178 default: 1179 return EOPNOTSUPP; 1180 } 1181 } 1182 1183 static int 1184 ugen_detach(device_t self, int flags) 1185 { 1186 struct ugen_softc *sc = device_private(self); 1187 struct ugen_endpoint *sce; 1188 int i, dir; 1189 int maj, mn; 1190 1191 DPRINTF(("ugen_detach: sc=%p flags=%d\n", sc, flags)); 1192 1193 KASSERT(KERNEL_LOCKED_P()); /* sc_is_open */ 1194 1195 /* 1196 * Fail if we're not forced to detach and userland has any 1197 * endpoints open. 1198 */ 1199 if ((flags & DETACH_FORCE) == 0) { 1200 for (i = 0; i < USB_MAX_ENDPOINTS; i++) { 1201 if (sc->sc_is_open[i]) 1202 return EBUSY; 1203 } 1204 } 1205 1206 /* Prevent new users. Prevent suspend/resume. */ 1207 sc->sc_dying = 1; 1208 pmf_device_deregister(self); 1209 1210 /* 1211 * If we never finished attaching, skip nixing endpoints and 1212 * users because there aren't any. 1213 */ 1214 if (!sc->sc_attached) 1215 goto out; 1216 1217 /* Abort all pipes. */ 1218 for (i = 0; i < USB_MAX_ENDPOINTS; i++) { 1219 for (dir = OUT; dir <= IN; dir++) { 1220 sce = &sc->sc_endpoints[i][dir]; 1221 if (sce->pipeh) 1222 usbd_abort_pipe(sce->pipeh); 1223 } 1224 } 1225 1226 /* 1227 * Wait for users to drain. Before this point there can be no 1228 * more I/O operations started because we set sc_dying; after 1229 * this, there can be no more I/O operations in progress, so it 1230 * will be safe to free things. 1231 */ 1232 mutex_enter(&sc->sc_lock); 1233 if (--sc->sc_refcnt >= 0) { 1234 /* Wake everyone */ 1235 for (i = 0; i < USB_MAX_ENDPOINTS; i++) { 1236 for (dir = OUT; dir <= IN; dir++) 1237 cv_broadcast(&sc->sc_endpoints[i][dir].cv); 1238 } 1239 /* Wait for processes to go away. */ 1240 do { 1241 cv_wait(&sc->sc_detach_cv, &sc->sc_lock); 1242 } while (sc->sc_refcnt >= 0); 1243 } 1244 mutex_exit(&sc->sc_lock); 1245 1246 /* locate the major number */ 1247 maj = cdevsw_lookup_major(&ugen_cdevsw); 1248 1249 /* 1250 * Nuke the vnodes for any open instances (calls ugenclose, but 1251 * with no effect because we already set sc_dying). 1252 */ 1253 mn = sc->sc_unit * USB_MAX_ENDPOINTS; 1254 vdevgone(maj, mn, mn + USB_MAX_ENDPOINTS - 1, VCHR); 1255 1256 /* Actually close any lingering pipes. */ 1257 for (i = 0; i < USB_MAX_ENDPOINTS; i++) 1258 ugen_do_close(sc, FREAD|FWRITE, i); 1259 1260 usbd_add_drv_event(USB_EVENT_DRIVER_DETACH, sc->sc_udev, sc->sc_dev); 1261 ugenif_put_unit(sc); 1262 1263 out: for (i = 0; i < USB_MAX_ENDPOINTS; i++) { 1264 for (dir = OUT; dir <= IN; dir++) { 1265 sce = &sc->sc_endpoints[i][dir]; 1266 seldestroy(&sce->rsel); 1267 cv_destroy(&sce->cv); 1268 } 1269 } 1270 1271 cv_destroy(&sc->sc_detach_cv); 1272 mutex_destroy(&sc->sc_lock); 1273 1274 return 0; 1275 } 1276 1277 Static void 1278 ugenintr(struct usbd_xfer *xfer, void *addr, usbd_status status) 1279 { 1280 struct ugen_endpoint *sce = addr; 1281 struct ugen_softc *sc = sce->sc; 1282 uint32_t count; 1283 u_char *ibuf; 1284 1285 if (status == USBD_CANCELLED) 1286 return; 1287 1288 if (status != USBD_NORMAL_COMPLETION) { 1289 DPRINTF(("ugenintr: status=%d\n", status)); 1290 if (status == USBD_STALLED) 1291 usbd_clear_endpoint_stall_async(sce->pipeh); 1292 return; 1293 } 1294 1295 usbd_get_xfer_status(xfer, NULL, NULL, &count, NULL); 1296 ibuf = sce->ibuf; 1297 1298 DPRINTFN(5, ("ugenintr: xfer=%p status=%d count=%d\n", 1299 xfer, status, count)); 1300 DPRINTFN(5, (" data = %02x %02x %02x\n", 1301 ibuf[0], ibuf[1], ibuf[2])); 1302 1303 mutex_enter(&sc->sc_lock); 1304 (void)b_to_q(ibuf, count, &sce->q); 1305 cv_signal(&sce->cv); 1306 mutex_exit(&sc->sc_lock); 1307 selnotify(&sce->rsel, 0, 0); 1308 } 1309 1310 Static void 1311 ugen_isoc_rintr(struct usbd_xfer *xfer, void *addr, 1312 usbd_status status) 1313 { 1314 struct isoreq *req = addr; 1315 struct ugen_endpoint *sce = req->sce; 1316 struct ugen_softc *sc = sce->sc; 1317 uint32_t count, n; 1318 int i, isize; 1319 1320 /* Return if we are aborting. */ 1321 if (status == USBD_CANCELLED) 1322 return; 1323 1324 usbd_get_xfer_status(xfer, NULL, NULL, &count, NULL); 1325 DPRINTFN(5,("ugen_isoc_rintr: xfer %ld, count=%d\n", 1326 (long)(req - sce->isoreqs), count)); 1327 1328 mutex_enter(&sc->sc_lock); 1329 1330 /* throw away oldest input if the buffer is full */ 1331 if (sce->fill < sce->cur && sce->cur <= sce->fill + count) { 1332 sce->cur += count; 1333 if (sce->cur >= sce->limit) 1334 sce->cur = sce->ibuf + (sce->limit - sce->cur); 1335 DPRINTFN(5, ("ugen_isoc_rintr: throwing away %d bytes\n", 1336 count)); 1337 } 1338 1339 isize = UGETW(sce->edesc->wMaxPacketSize); 1340 for (i = 0; i < UGEN_NISORFRMS; i++) { 1341 uint32_t actlen = req->sizes[i]; 1342 char const *tbuf = (char const *)req->dmabuf + isize * i; 1343 1344 /* copy data to buffer */ 1345 while (actlen > 0) { 1346 n = uimin(actlen, sce->limit - sce->fill); 1347 memcpy(sce->fill, tbuf, n); 1348 1349 tbuf += n; 1350 actlen -= n; 1351 sce->fill += n; 1352 if (sce->fill == sce->limit) 1353 sce->fill = sce->ibuf; 1354 } 1355 1356 /* setup size for next transfer */ 1357 req->sizes[i] = isize; 1358 } 1359 1360 usbd_setup_isoc_xfer(xfer, req, req->sizes, UGEN_NISORFRMS, 0, 1361 ugen_isoc_rintr); 1362 (void)usbd_transfer(xfer); 1363 1364 cv_signal(&sce->cv); 1365 mutex_exit(&sc->sc_lock); 1366 selnotify(&sce->rsel, 0, 0); 1367 } 1368 1369 Static void 1370 ugen_bulkra_intr(struct usbd_xfer *xfer, void *addr, 1371 usbd_status status) 1372 { 1373 struct ugen_endpoint *sce = addr; 1374 struct ugen_softc *sc = sce->sc; 1375 uint32_t count, n; 1376 char const *tbuf; 1377 usbd_status err; 1378 1379 /* Return if we are aborting. */ 1380 if (status == USBD_CANCELLED) 1381 return; 1382 1383 if (status != USBD_NORMAL_COMPLETION) { 1384 DPRINTF(("ugen_bulkra_intr: status=%d\n", status)); 1385 sce->state |= UGEN_RA_WB_STOP; 1386 if (status == USBD_STALLED) 1387 usbd_clear_endpoint_stall_async(sce->pipeh); 1388 return; 1389 } 1390 1391 usbd_get_xfer_status(xfer, NULL, NULL, &count, NULL); 1392 1393 mutex_enter(&sc->sc_lock); 1394 1395 /* Keep track of how much is in the buffer. */ 1396 sce->ra_wb_used += count; 1397 1398 /* Copy data to buffer. */ 1399 tbuf = (char const *)usbd_get_buffer(sce->ra_wb_xfer); 1400 n = uimin(count, sce->limit - sce->fill); 1401 memcpy(sce->fill, tbuf, n); 1402 tbuf += n; 1403 count -= n; 1404 sce->fill += n; 1405 if (sce->fill == sce->limit) 1406 sce->fill = sce->ibuf; 1407 if (count > 0) { 1408 memcpy(sce->fill, tbuf, count); 1409 sce->fill += count; 1410 } 1411 1412 /* Set up the next request if necessary. */ 1413 n = (sce->limit - sce->ibuf) - sce->ra_wb_used; 1414 if (n > 0) { 1415 usbd_setup_xfer(xfer, sce, NULL, uimin(n, sce->ra_wb_xferlen), 0, 1416 USBD_NO_TIMEOUT, ugen_bulkra_intr); 1417 err = usbd_transfer(xfer); 1418 if (err != USBD_IN_PROGRESS) { 1419 printf("usbd_bulkra_intr: error=%d\n", err); 1420 /* 1421 * The transfer has not been queued. Setting STOP 1422 * will make us try again at the next read. 1423 */ 1424 sce->state |= UGEN_RA_WB_STOP; 1425 } 1426 } 1427 else 1428 sce->state |= UGEN_RA_WB_STOP; 1429 1430 cv_signal(&sce->cv); 1431 mutex_exit(&sc->sc_lock); 1432 selnotify(&sce->rsel, 0, 0); 1433 } 1434 1435 Static void 1436 ugen_bulkwb_intr(struct usbd_xfer *xfer, void *addr, 1437 usbd_status status) 1438 { 1439 struct ugen_endpoint *sce = addr; 1440 struct ugen_softc *sc = sce->sc; 1441 uint32_t count, n; 1442 char *tbuf; 1443 usbd_status err; 1444 1445 /* Return if we are aborting. */ 1446 if (status == USBD_CANCELLED) 1447 return; 1448 1449 if (status != USBD_NORMAL_COMPLETION) { 1450 DPRINTF(("ugen_bulkwb_intr: status=%d\n", status)); 1451 sce->state |= UGEN_RA_WB_STOP; 1452 if (status == USBD_STALLED) 1453 usbd_clear_endpoint_stall_async(sce->pipeh); 1454 return; 1455 } 1456 1457 usbd_get_xfer_status(xfer, NULL, NULL, &count, NULL); 1458 1459 mutex_enter(&sc->sc_lock); 1460 1461 /* Keep track of how much is in the buffer. */ 1462 sce->ra_wb_used -= count; 1463 1464 /* Update buffer pointers. */ 1465 sce->cur += count; 1466 if (sce->cur >= sce->limit) 1467 sce->cur = sce->ibuf + (sce->cur - sce->limit); 1468 1469 /* Set up next request if necessary. */ 1470 if (sce->ra_wb_used > 0) { 1471 /* copy data from buffer */ 1472 tbuf = (char *)usbd_get_buffer(sce->ra_wb_xfer); 1473 count = uimin(sce->ra_wb_used, sce->ra_wb_xferlen); 1474 n = uimin(count, sce->limit - sce->cur); 1475 memcpy(tbuf, sce->cur, n); 1476 tbuf += n; 1477 if (count - n > 0) 1478 memcpy(tbuf, sce->ibuf, count - n); 1479 1480 usbd_setup_xfer(xfer, sce, NULL, count, 0, USBD_NO_TIMEOUT, 1481 ugen_bulkwb_intr); 1482 err = usbd_transfer(xfer); 1483 if (err != USBD_IN_PROGRESS) { 1484 printf("usbd_bulkwb_intr: error=%d\n", err); 1485 /* 1486 * The transfer has not been queued. Setting STOP 1487 * will make us try again at the next write. 1488 */ 1489 sce->state |= UGEN_RA_WB_STOP; 1490 } 1491 } 1492 else 1493 sce->state |= UGEN_RA_WB_STOP; 1494 1495 cv_signal(&sce->cv); 1496 mutex_exit(&sc->sc_lock); 1497 selnotify(&sce->rsel, 0, 0); 1498 } 1499 1500 Static usbd_status 1501 ugen_set_interface(struct ugen_softc *sc, int ifaceidx, int altno) 1502 { 1503 struct usbd_interface *iface; 1504 usb_endpoint_descriptor_t *ed; 1505 usbd_status err; 1506 struct ugen_endpoint *sce; 1507 uint8_t niface, nendpt, endptno, endpt; 1508 int dir; 1509 1510 DPRINTFN(15, ("ugen_set_interface %d %d\n", ifaceidx, altno)); 1511 1512 err = usbd_interface_count(sc->sc_udev, &niface); 1513 if (err) 1514 return err; 1515 if (ifaceidx < 0 || ifaceidx >= niface) 1516 return USBD_INVAL; 1517 1518 err = usbd_device2interface_handle(sc->sc_udev, ifaceidx, &iface); 1519 if (err) 1520 return err; 1521 err = usbd_endpoint_count(iface, &nendpt); 1522 if (err) 1523 return err; 1524 1525 /* change setting */ 1526 err = usbd_set_interface(iface, altno); 1527 if (err) 1528 return err; 1529 1530 err = usbd_endpoint_count(iface, &nendpt); 1531 if (err) 1532 return err; 1533 1534 ugen_clear_endpoints(sc); 1535 1536 for (endptno = 0; endptno < nendpt; endptno++) { 1537 ed = usbd_interface2endpoint_descriptor(iface,endptno); 1538 KASSERT(ed != NULL); 1539 endpt = ed->bEndpointAddress; 1540 dir = UE_GET_DIR(endpt) == UE_DIR_IN ? IN : OUT; 1541 sce = &sc->sc_endpoints[UE_GET_ADDR(endpt)][dir]; 1542 sce->sc = sc; 1543 sce->edesc = ed; 1544 sce->iface = iface; 1545 } 1546 return 0; 1547 } 1548 1549 /* Retrieve a complete descriptor for a certain device and index. */ 1550 Static usb_config_descriptor_t * 1551 ugen_get_cdesc(struct ugen_softc *sc, int index, int *lenp) 1552 { 1553 usb_config_descriptor_t *cdesc, *tdesc, cdescr; 1554 int len; 1555 usbd_status err; 1556 1557 if (index == USB_CURRENT_CONFIG_INDEX) { 1558 tdesc = usbd_get_config_descriptor(sc->sc_udev); 1559 if (tdesc == NULL) 1560 return NULL; 1561 len = UGETW(tdesc->wTotalLength); 1562 if (lenp) 1563 *lenp = len; 1564 cdesc = kmem_alloc(len, KM_SLEEP); 1565 memcpy(cdesc, tdesc, len); 1566 DPRINTFN(5,("ugen_get_cdesc: current, len=%d\n", len)); 1567 } else { 1568 err = usbd_get_config_desc(sc->sc_udev, index, &cdescr); 1569 if (err) 1570 return 0; 1571 len = UGETW(cdescr.wTotalLength); 1572 DPRINTFN(5,("ugen_get_cdesc: index=%d, len=%d\n", index, len)); 1573 if (lenp) 1574 *lenp = len; 1575 cdesc = kmem_alloc(len, KM_SLEEP); 1576 err = usbd_get_config_desc_full(sc->sc_udev, index, cdesc, len); 1577 if (err) { 1578 kmem_free(cdesc, len); 1579 return 0; 1580 } 1581 } 1582 return cdesc; 1583 } 1584 1585 Static int 1586 ugen_get_alt_index(struct ugen_softc *sc, int ifaceidx) 1587 { 1588 struct usbd_interface *iface; 1589 usbd_status err; 1590 1591 err = usbd_device2interface_handle(sc->sc_udev, ifaceidx, &iface); 1592 if (err) 1593 return -1; 1594 return usbd_get_interface_altindex(iface); 1595 } 1596 1597 Static int 1598 ugen_do_ioctl(struct ugen_softc *sc, int endpt, u_long cmd, 1599 void *addr, int flag, struct lwp *l) 1600 { 1601 struct ugen_endpoint *sce; 1602 usbd_status err; 1603 struct usbd_interface *iface; 1604 struct usb_config_desc *cd; 1605 usb_config_descriptor_t *cdesc; 1606 struct usb_interface_desc *id; 1607 usb_interface_descriptor_t *idesc; 1608 struct usb_endpoint_desc *ed; 1609 usb_endpoint_descriptor_t *edesc; 1610 struct usb_alt_interface *ai; 1611 struct usb_string_desc *si; 1612 uint8_t conf, alt; 1613 int cdesclen; 1614 int error; 1615 int dir; 1616 1617 KASSERT(KERNEL_LOCKED_P()); /* ugen_set_config */ 1618 1619 DPRINTFN(5, ("ugenioctl: cmd=%08lx\n", cmd)); 1620 1621 switch (cmd) { 1622 case FIONBIO: 1623 /* All handled in the upper FS layer. */ 1624 return 0; 1625 case USB_SET_SHORT_XFER: 1626 if (endpt == USB_CONTROL_ENDPOINT) 1627 return EINVAL; 1628 /* This flag only affects read */ 1629 sce = &sc->sc_endpoints[endpt][IN]; 1630 if (sce == NULL || sce->pipeh == NULL) 1631 return EINVAL; 1632 if (*(int *)addr) 1633 sce->state |= UGEN_SHORT_OK; 1634 else 1635 sce->state &= ~UGEN_SHORT_OK; 1636 return 0; 1637 case USB_SET_TIMEOUT: 1638 for (dir = OUT; dir <= IN; dir++) { 1639 sce = &sc->sc_endpoints[endpt][dir]; 1640 if (sce == NULL) 1641 return EINVAL; 1642 1643 sce->timeout = *(int *)addr; 1644 } 1645 return 0; 1646 case USB_SET_BULK_RA: 1647 if (endpt == USB_CONTROL_ENDPOINT) 1648 return EINVAL; 1649 sce = &sc->sc_endpoints[endpt][IN]; 1650 if (sce == NULL || sce->pipeh == NULL) 1651 return EINVAL; 1652 edesc = sce->edesc; 1653 if ((edesc->bmAttributes & UE_XFERTYPE) != UE_BULK) 1654 return EINVAL; 1655 1656 if (*(int *)addr) { 1657 /* Only turn RA on if it's currently off. */ 1658 if (sce->state & UGEN_BULK_RA) 1659 return 0; 1660 KASSERT(sce->ra_wb_xfer == NULL); 1661 KASSERT(sce->ibuf == NULL); 1662 1663 if (sce->ra_wb_bufsize == 0 || sce->ra_wb_reqsize == 0) 1664 /* shouldn't happen */ 1665 return EINVAL; 1666 error = usbd_create_xfer(sce->pipeh, 1667 sce->ra_wb_reqsize, 0, 0, &sce->ra_wb_xfer); 1668 if (error) 1669 return error; 1670 sce->ra_wb_xferlen = sce->ra_wb_reqsize; 1671 sce->ibuf = kmem_alloc(sce->ra_wb_bufsize, KM_SLEEP); 1672 sce->fill = sce->cur = sce->ibuf; 1673 sce->limit = sce->ibuf + sce->ra_wb_bufsize; 1674 sce->ra_wb_used = 0; 1675 sce->state |= UGEN_BULK_RA; 1676 sce->state &= ~UGEN_RA_WB_STOP; 1677 /* Now start reading. */ 1678 usbd_setup_xfer(sce->ra_wb_xfer, sce, NULL, 1679 uimin(sce->ra_wb_xferlen, sce->ra_wb_bufsize), 1680 0, USBD_NO_TIMEOUT, ugen_bulkra_intr); 1681 err = usbd_transfer(sce->ra_wb_xfer); 1682 if (err != USBD_IN_PROGRESS) { 1683 sce->state &= ~UGEN_BULK_RA; 1684 kmem_free(sce->ibuf, sce->ra_wb_bufsize); 1685 sce->ibuf = NULL; 1686 usbd_destroy_xfer(sce->ra_wb_xfer); 1687 sce->ra_wb_xfer = NULL; 1688 return EIO; 1689 } 1690 } else { 1691 /* Only turn RA off if it's currently on. */ 1692 if (!(sce->state & UGEN_BULK_RA)) 1693 return 0; 1694 1695 sce->state &= ~UGEN_BULK_RA; 1696 usbd_abort_pipe(sce->pipeh); 1697 usbd_destroy_xfer(sce->ra_wb_xfer); 1698 sce->ra_wb_xfer = NULL; 1699 /* 1700 * XXX Discard whatever's in the buffer, but we 1701 * should keep it around and drain the buffer 1702 * instead. 1703 */ 1704 kmem_free(sce->ibuf, sce->ra_wb_bufsize); 1705 sce->ibuf = NULL; 1706 } 1707 return 0; 1708 case USB_SET_BULK_WB: 1709 if (endpt == USB_CONTROL_ENDPOINT) 1710 return EINVAL; 1711 sce = &sc->sc_endpoints[endpt][OUT]; 1712 if (sce == NULL || sce->pipeh == NULL) 1713 return EINVAL; 1714 edesc = sce->edesc; 1715 if ((edesc->bmAttributes & UE_XFERTYPE) != UE_BULK) 1716 return EINVAL; 1717 1718 if (*(int *)addr) { 1719 /* Only turn WB on if it's currently off. */ 1720 if (sce->state & UGEN_BULK_WB) 1721 return 0; 1722 KASSERT(sce->ra_wb_xfer == NULL); 1723 KASSERT(sce->ibuf == NULL); 1724 1725 if (sce->ra_wb_bufsize == 0 || sce->ra_wb_reqsize == 0) 1726 /* shouldn't happen */ 1727 return EINVAL; 1728 error = usbd_create_xfer(sce->pipeh, sce->ra_wb_reqsize, 1729 0, 0, &sce->ra_wb_xfer); 1730 /* XXX check error??? */ 1731 sce->ra_wb_xferlen = sce->ra_wb_reqsize; 1732 sce->ibuf = kmem_alloc(sce->ra_wb_bufsize, KM_SLEEP); 1733 sce->fill = sce->cur = sce->ibuf; 1734 sce->limit = sce->ibuf + sce->ra_wb_bufsize; 1735 sce->ra_wb_used = 0; 1736 sce->state |= UGEN_BULK_WB | UGEN_RA_WB_STOP; 1737 } else { 1738 /* Only turn WB off if it's currently on. */ 1739 if (!(sce->state & UGEN_BULK_WB)) 1740 return 0; 1741 1742 sce->state &= ~UGEN_BULK_WB; 1743 /* 1744 * XXX Discard whatever's in the buffer, but we 1745 * should keep it around and keep writing to 1746 * drain the buffer instead. 1747 */ 1748 usbd_abort_pipe(sce->pipeh); 1749 usbd_destroy_xfer(sce->ra_wb_xfer); 1750 sce->ra_wb_xfer = NULL; 1751 kmem_free(sce->ibuf, sce->ra_wb_bufsize); 1752 sce->ibuf = NULL; 1753 } 1754 return 0; 1755 case USB_SET_BULK_RA_OPT: 1756 case USB_SET_BULK_WB_OPT: 1757 { 1758 struct usb_bulk_ra_wb_opt *opt; 1759 1760 if (endpt == USB_CONTROL_ENDPOINT) 1761 return EINVAL; 1762 opt = (struct usb_bulk_ra_wb_opt *)addr; 1763 if (cmd == USB_SET_BULK_RA_OPT) 1764 sce = &sc->sc_endpoints[endpt][IN]; 1765 else 1766 sce = &sc->sc_endpoints[endpt][OUT]; 1767 if (sce == NULL || sce->pipeh == NULL) 1768 return EINVAL; 1769 if (opt->ra_wb_buffer_size < 1 || 1770 opt->ra_wb_buffer_size > UGEN_BULK_RA_WB_BUFMAX || 1771 opt->ra_wb_request_size < 1 || 1772 opt->ra_wb_request_size > opt->ra_wb_buffer_size) 1773 return EINVAL; 1774 /* 1775 * XXX These changes do not take effect until the 1776 * next time RA/WB mode is enabled but they ought to 1777 * take effect immediately. 1778 */ 1779 sce->ra_wb_bufsize = opt->ra_wb_buffer_size; 1780 sce->ra_wb_reqsize = opt->ra_wb_request_size; 1781 return 0; 1782 } 1783 default: 1784 break; 1785 } 1786 1787 if (endpt != USB_CONTROL_ENDPOINT) 1788 return EINVAL; 1789 1790 switch (cmd) { 1791 #ifdef UGEN_DEBUG 1792 case USB_SETDEBUG: 1793 ugendebug = *(int *)addr; 1794 break; 1795 #endif 1796 case USB_GET_CONFIG: 1797 err = usbd_get_config(sc->sc_udev, &conf); 1798 if (err) 1799 return EIO; 1800 *(int *)addr = conf; 1801 break; 1802 case USB_SET_CONFIG: 1803 if (!(flag & FWRITE)) 1804 return EPERM; 1805 err = ugen_set_config(sc, *(int *)addr, 1); 1806 switch (err) { 1807 case USBD_NORMAL_COMPLETION: 1808 break; 1809 case USBD_IN_USE: 1810 return EBUSY; 1811 default: 1812 return EIO; 1813 } 1814 break; 1815 case USB_GET_ALTINTERFACE: 1816 ai = (struct usb_alt_interface *)addr; 1817 err = usbd_device2interface_handle(sc->sc_udev, 1818 ai->uai_interface_index, &iface); 1819 if (err) 1820 return EINVAL; 1821 idesc = usbd_get_interface_descriptor(iface); 1822 if (idesc == NULL) 1823 return EIO; 1824 ai->uai_alt_no = idesc->bAlternateSetting; 1825 break; 1826 case USB_SET_ALTINTERFACE: 1827 if (!(flag & FWRITE)) 1828 return EPERM; 1829 ai = (struct usb_alt_interface *)addr; 1830 err = usbd_device2interface_handle(sc->sc_udev, 1831 ai->uai_interface_index, &iface); 1832 if (err) 1833 return EINVAL; 1834 err = ugen_set_interface(sc, ai->uai_interface_index, 1835 ai->uai_alt_no); 1836 if (err) 1837 return EINVAL; 1838 break; 1839 case USB_GET_NO_ALT: 1840 ai = (struct usb_alt_interface *)addr; 1841 cdesc = ugen_get_cdesc(sc, ai->uai_config_index, &cdesclen); 1842 if (cdesc == NULL) 1843 return EINVAL; 1844 idesc = usbd_find_idesc(cdesc, ai->uai_interface_index, 0); 1845 if (idesc == NULL) { 1846 kmem_free(cdesc, cdesclen); 1847 return EINVAL; 1848 } 1849 ai->uai_alt_no = usbd_get_no_alts(cdesc, 1850 idesc->bInterfaceNumber); 1851 kmem_free(cdesc, cdesclen); 1852 break; 1853 case USB_GET_DEVICE_DESC: 1854 *(usb_device_descriptor_t *)addr = 1855 *usbd_get_device_descriptor(sc->sc_udev); 1856 break; 1857 case USB_GET_CONFIG_DESC: 1858 cd = (struct usb_config_desc *)addr; 1859 cdesc = ugen_get_cdesc(sc, cd->ucd_config_index, &cdesclen); 1860 if (cdesc == NULL) 1861 return EINVAL; 1862 cd->ucd_desc = *cdesc; 1863 kmem_free(cdesc, cdesclen); 1864 break; 1865 case USB_GET_INTERFACE_DESC: 1866 id = (struct usb_interface_desc *)addr; 1867 cdesc = ugen_get_cdesc(sc, id->uid_config_index, &cdesclen); 1868 if (cdesc == NULL) 1869 return EINVAL; 1870 if (id->uid_config_index == USB_CURRENT_CONFIG_INDEX && 1871 id->uid_alt_index == USB_CURRENT_ALT_INDEX) 1872 alt = ugen_get_alt_index(sc, id->uid_interface_index); 1873 else 1874 alt = id->uid_alt_index; 1875 idesc = usbd_find_idesc(cdesc, id->uid_interface_index, alt); 1876 if (idesc == NULL) { 1877 kmem_free(cdesc, cdesclen); 1878 return EINVAL; 1879 } 1880 id->uid_desc = *idesc; 1881 kmem_free(cdesc, cdesclen); 1882 break; 1883 case USB_GET_ENDPOINT_DESC: 1884 ed = (struct usb_endpoint_desc *)addr; 1885 cdesc = ugen_get_cdesc(sc, ed->ued_config_index, &cdesclen); 1886 if (cdesc == NULL) 1887 return EINVAL; 1888 if (ed->ued_config_index == USB_CURRENT_CONFIG_INDEX && 1889 ed->ued_alt_index == USB_CURRENT_ALT_INDEX) 1890 alt = ugen_get_alt_index(sc, ed->ued_interface_index); 1891 else 1892 alt = ed->ued_alt_index; 1893 edesc = usbd_find_edesc(cdesc, ed->ued_interface_index, 1894 alt, ed->ued_endpoint_index); 1895 if (edesc == NULL) { 1896 kmem_free(cdesc, cdesclen); 1897 return EINVAL; 1898 } 1899 ed->ued_desc = *edesc; 1900 kmem_free(cdesc, cdesclen); 1901 break; 1902 case USB_GET_FULL_DESC: 1903 { 1904 int len; 1905 struct iovec iov; 1906 struct uio uio; 1907 struct usb_full_desc *fd = (struct usb_full_desc *)addr; 1908 1909 cdesc = ugen_get_cdesc(sc, fd->ufd_config_index, &cdesclen); 1910 if (cdesc == NULL) 1911 return EINVAL; 1912 len = cdesclen; 1913 if (len > fd->ufd_size) 1914 len = fd->ufd_size; 1915 iov.iov_base = (void *)fd->ufd_data; 1916 iov.iov_len = len; 1917 uio.uio_iov = &iov; 1918 uio.uio_iovcnt = 1; 1919 uio.uio_resid = len; 1920 uio.uio_offset = 0; 1921 uio.uio_rw = UIO_READ; 1922 uio.uio_vmspace = l->l_proc->p_vmspace; 1923 error = uiomove((void *)cdesc, len, &uio); 1924 kmem_free(cdesc, cdesclen); 1925 return error; 1926 } 1927 case USB_GET_STRING_DESC: { 1928 int len; 1929 si = (struct usb_string_desc *)addr; 1930 err = usbd_get_string_desc(sc->sc_udev, si->usd_string_index, 1931 si->usd_language_id, &si->usd_desc, &len); 1932 if (err) 1933 return EINVAL; 1934 break; 1935 } 1936 case USB_DO_REQUEST: 1937 { 1938 struct usb_ctl_request *ur = (void *)addr; 1939 int len = UGETW(ur->ucr_request.wLength); 1940 struct iovec iov; 1941 struct uio uio; 1942 void *ptr = 0; 1943 usbd_status xerr; 1944 1945 error = 0; 1946 1947 if (!(flag & FWRITE)) 1948 return EPERM; 1949 /* Avoid requests that would damage the bus integrity. */ 1950 if ((ur->ucr_request.bmRequestType == UT_WRITE_DEVICE && 1951 ur->ucr_request.bRequest == UR_SET_ADDRESS) || 1952 (ur->ucr_request.bmRequestType == UT_WRITE_DEVICE && 1953 ur->ucr_request.bRequest == UR_SET_CONFIG) || 1954 (ur->ucr_request.bmRequestType == UT_WRITE_INTERFACE && 1955 ur->ucr_request.bRequest == UR_SET_INTERFACE)) 1956 return EINVAL; 1957 1958 if (len < 0 || len > 32767) 1959 return EINVAL; 1960 if (len != 0) { 1961 iov.iov_base = (void *)ur->ucr_data; 1962 iov.iov_len = len; 1963 uio.uio_iov = &iov; 1964 uio.uio_iovcnt = 1; 1965 uio.uio_resid = len; 1966 uio.uio_offset = 0; 1967 uio.uio_rw = 1968 ur->ucr_request.bmRequestType & UT_READ ? 1969 UIO_READ : UIO_WRITE; 1970 uio.uio_vmspace = l->l_proc->p_vmspace; 1971 ptr = kmem_alloc(len, KM_SLEEP); 1972 if (uio.uio_rw == UIO_WRITE) { 1973 error = uiomove(ptr, len, &uio); 1974 if (error) 1975 goto ret; 1976 } 1977 } 1978 sce = &sc->sc_endpoints[endpt][IN]; 1979 xerr = usbd_do_request_flags(sc->sc_udev, &ur->ucr_request, 1980 ptr, ur->ucr_flags, &ur->ucr_actlen, sce->timeout); 1981 if (xerr) { 1982 error = EIO; 1983 goto ret; 1984 } 1985 if (len != 0) { 1986 if (uio.uio_rw == UIO_READ) { 1987 size_t alen = uimin(len, ur->ucr_actlen); 1988 error = uiomove(ptr, alen, &uio); 1989 if (error) 1990 goto ret; 1991 } 1992 } 1993 ret: 1994 if (ptr) 1995 kmem_free(ptr, len); 1996 return error; 1997 } 1998 case USB_GET_DEVICEINFO: 1999 usbd_fill_deviceinfo(sc->sc_udev, 2000 (struct usb_device_info *)addr, 0); 2001 break; 2002 case USB_GET_DEVICEINFO_OLD: 2003 { 2004 int ret; 2005 MODULE_HOOK_CALL(usb_subr_fill_30_hook, 2006 (sc->sc_udev, (struct usb_device_info_old *)addr, 0, 2007 usbd_devinfo_vp, usbd_printBCD), 2008 enosys(), ret); 2009 if (ret == 0) 2010 return 0; 2011 return EINVAL; 2012 } 2013 default: 2014 return EINVAL; 2015 } 2016 return 0; 2017 } 2018 2019 static int 2020 ugenioctl(dev_t dev, u_long cmd, void *addr, int flag, struct lwp *l) 2021 { 2022 int endpt = UGENENDPOINT(dev); 2023 struct ugen_softc *sc; 2024 int error; 2025 2026 if ((sc = ugenif_acquire(UGENUNIT(dev))) == 0) 2027 return ENXIO; 2028 error = ugen_do_ioctl(sc, endpt, cmd, addr, flag, l); 2029 ugenif_release(sc); 2030 2031 return error; 2032 } 2033 2034 static int 2035 ugenpoll(dev_t dev, int events, struct lwp *l) 2036 { 2037 struct ugen_softc *sc; 2038 struct ugen_endpoint *sce_in, *sce_out; 2039 int revents = 0; 2040 2041 if ((sc = ugenif_acquire(UGENUNIT(dev))) == NULL) 2042 return POLLHUP; 2043 2044 if (UGENENDPOINT(dev) == USB_CONTROL_ENDPOINT) { 2045 revents |= POLLERR; 2046 goto out; 2047 } 2048 2049 sce_in = &sc->sc_endpoints[UGENENDPOINT(dev)][IN]; 2050 sce_out = &sc->sc_endpoints[UGENENDPOINT(dev)][OUT]; 2051 KASSERT(sce_in->edesc || sce_out->edesc); 2052 KASSERT(sce_in->pipeh || sce_out->pipeh); 2053 2054 mutex_enter(&sc->sc_lock); 2055 if (sce_in && sce_in->pipeh && (events & (POLLIN | POLLRDNORM))) 2056 switch (sce_in->edesc->bmAttributes & UE_XFERTYPE) { 2057 case UE_INTERRUPT: 2058 if (sce_in->q.c_cc > 0) 2059 revents |= events & (POLLIN | POLLRDNORM); 2060 else 2061 selrecord(l, &sce_in->rsel); 2062 break; 2063 case UE_ISOCHRONOUS: 2064 if (sce_in->cur != sce_in->fill) 2065 revents |= events & (POLLIN | POLLRDNORM); 2066 else 2067 selrecord(l, &sce_in->rsel); 2068 break; 2069 case UE_BULK: 2070 if (sce_in->state & UGEN_BULK_RA) { 2071 if (sce_in->ra_wb_used > 0) 2072 revents |= events & 2073 (POLLIN | POLLRDNORM); 2074 else 2075 selrecord(l, &sce_in->rsel); 2076 break; 2077 } 2078 /* 2079 * We have no easy way of determining if a read will 2080 * yield any data or a write will happen. 2081 * Pretend they will. 2082 */ 2083 revents |= events & (POLLIN | POLLRDNORM); 2084 break; 2085 default: 2086 break; 2087 } 2088 if (sce_out && sce_out->pipeh && (events & (POLLOUT | POLLWRNORM))) 2089 switch (sce_out->edesc->bmAttributes & UE_XFERTYPE) { 2090 case UE_INTERRUPT: 2091 case UE_ISOCHRONOUS: 2092 /* XXX unimplemented */ 2093 break; 2094 case UE_BULK: 2095 if (sce_out->state & UGEN_BULK_WB) { 2096 if (sce_out->ra_wb_used < 2097 sce_out->limit - sce_out->ibuf) 2098 revents |= events & 2099 (POLLOUT | POLLWRNORM); 2100 else 2101 selrecord(l, &sce_out->rsel); 2102 break; 2103 } 2104 /* 2105 * We have no easy way of determining if a read will 2106 * yield any data or a write will happen. 2107 * Pretend they will. 2108 */ 2109 revents |= events & (POLLOUT | POLLWRNORM); 2110 break; 2111 default: 2112 break; 2113 } 2114 2115 mutex_exit(&sc->sc_lock); 2116 2117 out: ugenif_release(sc); 2118 return revents; 2119 } 2120 2121 static void 2122 filt_ugenrdetach(struct knote *kn) 2123 { 2124 struct ugen_endpoint *sce = kn->kn_hook; 2125 struct ugen_softc *sc = sce->sc; 2126 2127 mutex_enter(&sc->sc_lock); 2128 selremove_knote(&sce->rsel, kn); 2129 mutex_exit(&sc->sc_lock); 2130 } 2131 2132 static int 2133 filt_ugenread_intr(struct knote *kn, long hint) 2134 { 2135 struct ugen_endpoint *sce = kn->kn_hook; 2136 struct ugen_softc *sc = sce->sc; 2137 int ret; 2138 2139 mutex_enter(&sc->sc_lock); 2140 if (sc->sc_dying) { 2141 ret = 0; 2142 } else { 2143 kn->kn_data = sce->q.c_cc; 2144 ret = kn->kn_data > 0; 2145 } 2146 mutex_exit(&sc->sc_lock); 2147 2148 return ret; 2149 } 2150 2151 static int 2152 filt_ugenread_isoc(struct knote *kn, long hint) 2153 { 2154 struct ugen_endpoint *sce = kn->kn_hook; 2155 struct ugen_softc *sc = sce->sc; 2156 int ret; 2157 2158 mutex_enter(&sc->sc_lock); 2159 if (sc->sc_dying) { 2160 ret = 0; 2161 } else if (sce->cur == sce->fill) { 2162 ret = 0; 2163 } else if (sce->cur < sce->fill) { 2164 kn->kn_data = sce->fill - sce->cur; 2165 ret = 1; 2166 } else { 2167 kn->kn_data = (sce->limit - sce->cur) + 2168 (sce->fill - sce->ibuf); 2169 ret = 1; 2170 } 2171 mutex_exit(&sc->sc_lock); 2172 2173 return ret; 2174 } 2175 2176 static int 2177 filt_ugenread_bulk(struct knote *kn, long hint) 2178 { 2179 struct ugen_endpoint *sce = kn->kn_hook; 2180 struct ugen_softc *sc = sce->sc; 2181 int ret; 2182 2183 mutex_enter(&sc->sc_lock); 2184 if (sc->sc_dying) { 2185 ret = 0; 2186 } else if (!(sce->state & UGEN_BULK_RA)) { 2187 /* 2188 * We have no easy way of determining if a read will 2189 * yield any data or a write will happen. 2190 * So, emulate "seltrue". 2191 */ 2192 ret = filt_seltrue(kn, hint); 2193 } else if (sce->ra_wb_used == 0) { 2194 ret = 0; 2195 } else { 2196 kn->kn_data = sce->ra_wb_used; 2197 ret = 1; 2198 } 2199 mutex_exit(&sc->sc_lock); 2200 2201 return ret; 2202 } 2203 2204 static int 2205 filt_ugenwrite_bulk(struct knote *kn, long hint) 2206 { 2207 struct ugen_endpoint *sce = kn->kn_hook; 2208 struct ugen_softc *sc = sce->sc; 2209 int ret; 2210 2211 mutex_enter(&sc->sc_lock); 2212 if (sc->sc_dying) { 2213 ret = 0; 2214 } else if (!(sce->state & UGEN_BULK_WB)) { 2215 /* 2216 * We have no easy way of determining if a read will 2217 * yield any data or a write will happen. 2218 * So, emulate "seltrue". 2219 */ 2220 ret = filt_seltrue(kn, hint); 2221 } else if (sce->ra_wb_used == sce->limit - sce->ibuf) { 2222 ret = 0; 2223 } else { 2224 kn->kn_data = (sce->limit - sce->ibuf) - sce->ra_wb_used; 2225 ret = 1; 2226 } 2227 mutex_exit(&sc->sc_lock); 2228 2229 return ret; 2230 } 2231 2232 static const struct filterops ugenread_intr_filtops = { 2233 .f_flags = FILTEROP_ISFD, 2234 .f_attach = NULL, 2235 .f_detach = filt_ugenrdetach, 2236 .f_event = filt_ugenread_intr, 2237 }; 2238 2239 static const struct filterops ugenread_isoc_filtops = { 2240 .f_flags = FILTEROP_ISFD, 2241 .f_attach = NULL, 2242 .f_detach = filt_ugenrdetach, 2243 .f_event = filt_ugenread_isoc, 2244 }; 2245 2246 static const struct filterops ugenread_bulk_filtops = { 2247 .f_flags = FILTEROP_ISFD, 2248 .f_attach = NULL, 2249 .f_detach = filt_ugenrdetach, 2250 .f_event = filt_ugenread_bulk, 2251 }; 2252 2253 static const struct filterops ugenwrite_bulk_filtops = { 2254 .f_flags = FILTEROP_ISFD, 2255 .f_attach = NULL, 2256 .f_detach = filt_ugenrdetach, 2257 .f_event = filt_ugenwrite_bulk, 2258 }; 2259 2260 static int 2261 ugenkqfilter(dev_t dev, struct knote *kn) 2262 { 2263 struct ugen_softc *sc; 2264 struct ugen_endpoint *sce; 2265 struct selinfo *sip; 2266 int error; 2267 2268 if ((sc = ugenif_acquire(UGENUNIT(dev))) == NULL) 2269 return ENXIO; 2270 2271 if (UGENENDPOINT(dev) == USB_CONTROL_ENDPOINT) { 2272 error = ENODEV; 2273 goto out; 2274 } 2275 2276 switch (kn->kn_filter) { 2277 case EVFILT_READ: 2278 sce = &sc->sc_endpoints[UGENENDPOINT(dev)][IN]; 2279 if (sce == NULL) { 2280 error = EINVAL; 2281 goto out; 2282 } 2283 2284 sip = &sce->rsel; 2285 switch (sce->edesc->bmAttributes & UE_XFERTYPE) { 2286 case UE_INTERRUPT: 2287 kn->kn_fop = &ugenread_intr_filtops; 2288 break; 2289 case UE_ISOCHRONOUS: 2290 kn->kn_fop = &ugenread_isoc_filtops; 2291 break; 2292 case UE_BULK: 2293 kn->kn_fop = &ugenread_bulk_filtops; 2294 break; 2295 default: 2296 error = EINVAL; 2297 goto out; 2298 } 2299 break; 2300 2301 case EVFILT_WRITE: 2302 sce = &sc->sc_endpoints[UGENENDPOINT(dev)][OUT]; 2303 if (sce == NULL) { 2304 error = EINVAL; 2305 goto out; 2306 } 2307 2308 sip = &sce->rsel; 2309 switch (sce->edesc->bmAttributes & UE_XFERTYPE) { 2310 case UE_INTERRUPT: 2311 case UE_ISOCHRONOUS: 2312 /* XXX poll doesn't support this */ 2313 error = EINVAL; 2314 goto out; 2315 2316 case UE_BULK: 2317 kn->kn_fop = &ugenwrite_bulk_filtops; 2318 break; 2319 default: 2320 error = EINVAL; 2321 goto out; 2322 } 2323 break; 2324 2325 default: 2326 error = EINVAL; 2327 goto out; 2328 } 2329 2330 kn->kn_hook = sce; 2331 2332 mutex_enter(&sc->sc_lock); 2333 selrecord_knote(sip, kn); 2334 mutex_exit(&sc->sc_lock); 2335 2336 error = 0; 2337 2338 out: ugenif_release(sc); 2339 return error; 2340 } 2341 2342 MODULE(MODULE_CLASS_DRIVER, ugen, NULL); 2343 2344 static int 2345 ugen_modcmd(modcmd_t cmd, void *aux) 2346 { 2347 2348 switch (cmd) { 2349 case MODULE_CMD_INIT: 2350 mutex_init(&ugenif.lock, MUTEX_DEFAULT, IPL_NONE); 2351 rb_tree_init(&ugenif.tree, &ugenif_tree_ops); 2352 return 0; 2353 default: 2354 return ENOTTY; 2355 } 2356 } 2357