16bd03b20SKevin Lo /*-
20642eac4SKevin Lo * Copyright (c) 2014, 2017 Kevin Lo
36bd03b20SKevin Lo * All rights reserved.
46bd03b20SKevin Lo *
56bd03b20SKevin Lo * Redistribution and use in source and binary forms, with or without
66bd03b20SKevin Lo * modification, are permitted provided that the following conditions
76bd03b20SKevin Lo * are met:
86bd03b20SKevin Lo * 1. Redistributions of source code must retain the above copyright
96bd03b20SKevin Lo * notice, this list of conditions, and the following disclaimer.
106bd03b20SKevin Lo * 2. Redistributions in binary form must reproduce the above copyright
116bd03b20SKevin Lo * notice, this list of conditions and the following disclaimer in the
126bd03b20SKevin Lo * documentation and/or other materials provided with the distribution.
136bd03b20SKevin Lo *
146bd03b20SKevin Lo * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
156bd03b20SKevin Lo * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
166bd03b20SKevin Lo * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
176bd03b20SKevin Lo * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE FOR
186bd03b20SKevin Lo * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
196bd03b20SKevin Lo * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
206bd03b20SKevin Lo * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
216bd03b20SKevin Lo * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
226bd03b20SKevin Lo * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
236bd03b20SKevin Lo * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
246bd03b20SKevin Lo * SUCH DAMAGE.
256bd03b20SKevin Lo *
266bd03b20SKevin Lo */
276bd03b20SKevin Lo
286bd03b20SKevin Lo #include <sys/cdefs.h>
296bd03b20SKevin Lo #include <sys/stdint.h>
306bd03b20SKevin Lo #include <sys/stddef.h>
316bd03b20SKevin Lo #include <sys/param.h>
326bd03b20SKevin Lo #include <sys/queue.h>
336bd03b20SKevin Lo #include <sys/types.h>
346bd03b20SKevin Lo #include <sys/systm.h>
356bd03b20SKevin Lo #include <sys/kernel.h>
366bd03b20SKevin Lo #include <sys/bus.h>
376bd03b20SKevin Lo #include <sys/module.h>
386bd03b20SKevin Lo #include <sys/lock.h>
396bd03b20SKevin Lo #include <sys/mutex.h>
406bd03b20SKevin Lo #include <sys/condvar.h>
416bd03b20SKevin Lo #include <sys/sysctl.h>
426bd03b20SKevin Lo #include <sys/sx.h>
436bd03b20SKevin Lo #include <sys/unistd.h>
446bd03b20SKevin Lo #include <sys/callout.h>
456bd03b20SKevin Lo #include <sys/malloc.h>
466bd03b20SKevin Lo #include <sys/priv.h>
476bd03b20SKevin Lo #include <sys/conf.h>
486bd03b20SKevin Lo #include <sys/fcntl.h>
496bd03b20SKevin Lo
506bd03b20SKevin Lo #include <dev/usb/usb.h>
516bd03b20SKevin Lo #include <dev/usb/usbdi.h>
526bd03b20SKevin Lo #include <dev/usb/usbhid.h>
536bd03b20SKevin Lo #include "usbdevs.h"
546bd03b20SKevin Lo
556bd03b20SKevin Lo #define USB_DEBUG_VAR usb_debug
566bd03b20SKevin Lo #include <dev/usb/usb_debug.h>
576bd03b20SKevin Lo
586bd03b20SKevin Lo #include <dev/usb/uled_ioctl.h>
596bd03b20SKevin Lo
606bd03b20SKevin Lo struct uled_softc {
616bd03b20SKevin Lo struct usb_fifo_sc sc_fifo;
626bd03b20SKevin Lo struct mtx sc_mtx;
636bd03b20SKevin Lo
646bd03b20SKevin Lo struct usb_device *sc_udev;
656bd03b20SKevin Lo struct uled_color sc_color;
666bd03b20SKevin Lo
676bd03b20SKevin Lo uint8_t sc_state;
686bd03b20SKevin Lo #define ULED_ENABLED 0x01
690642eac4SKevin Lo
700642eac4SKevin Lo int sc_flags;
710642eac4SKevin Lo #define ULED_FLAG_BLINK1 0x0001
726bd03b20SKevin Lo };
736bd03b20SKevin Lo
740642eac4SKevin Lo /* Initial commands. */
750642eac4SKevin Lo static uint8_t blink1[] = { 0x1, 'v', 0, 0, 0, 0, 0, 0 };
760642eac4SKevin Lo static uint8_t dl100b[] = { 0x1f, 0x2, 0, 0x5f, 0, 0, 0x1a, 0x3 };
776bd03b20SKevin Lo
780642eac4SKevin Lo /* Prototypes. */
796bd03b20SKevin Lo static device_probe_t uled_probe;
806bd03b20SKevin Lo static device_attach_t uled_attach;
816bd03b20SKevin Lo static device_detach_t uled_detach;
826bd03b20SKevin Lo
836bd03b20SKevin Lo static usb_fifo_open_t uled_open;
846bd03b20SKevin Lo static usb_fifo_close_t uled_close;
856bd03b20SKevin Lo static usb_fifo_ioctl_t uled_ioctl;
866bd03b20SKevin Lo
876bd03b20SKevin Lo static struct usb_fifo_methods uled_fifo_methods = {
886bd03b20SKevin Lo .f_open = &uled_open,
896bd03b20SKevin Lo .f_close = &uled_close,
906bd03b20SKevin Lo .f_ioctl = &uled_ioctl,
916bd03b20SKevin Lo .basename[0] = "uled",
926bd03b20SKevin Lo };
936bd03b20SKevin Lo
946bd03b20SKevin Lo static usb_error_t uled_ctrl_msg(struct uled_softc *, uint8_t, uint8_t,
950642eac4SKevin Lo uint16_t, uint16_t, void *, uint16_t);
966bd03b20SKevin Lo static int uled_enable(struct uled_softc *);
976bd03b20SKevin Lo
986bd03b20SKevin Lo static device_method_t uled_methods[] = {
996bd03b20SKevin Lo DEVMETHOD(device_probe, uled_probe),
1006bd03b20SKevin Lo DEVMETHOD(device_attach, uled_attach),
1016bd03b20SKevin Lo DEVMETHOD(device_detach, uled_detach),
1026bd03b20SKevin Lo
1036bd03b20SKevin Lo DEVMETHOD_END
1046bd03b20SKevin Lo };
1056bd03b20SKevin Lo
1066bd03b20SKevin Lo static driver_t uled_driver = {
1076bd03b20SKevin Lo .name = "uled",
1086bd03b20SKevin Lo .methods = uled_methods,
1096bd03b20SKevin Lo .size = sizeof(struct uled_softc),
1106bd03b20SKevin Lo };
1116bd03b20SKevin Lo
1126bd03b20SKevin Lo static const STRUCT_USB_HOST_ID uled_devs[] = {
1130642eac4SKevin Lo #define ULED_DEV(v,p,i) { USB_VPI(USB_VENDOR_##v, USB_PRODUCT_##v##_##p, i) }
1140642eac4SKevin Lo ULED_DEV(DREAMLINK, DL100B, 0),
1150642eac4SKevin Lo ULED_DEV(THINGM, BLINK1, ULED_FLAG_BLINK1),
1160642eac4SKevin Lo #undef ULED_DEV
1176bd03b20SKevin Lo };
1186bd03b20SKevin Lo
119bc9372d7SJohn Baldwin DRIVER_MODULE(uled, uhub, uled_driver, NULL, NULL);
120f809f280SWarner Losh MODULE_DEPEND(uled, usb, 1, 1, 1);
121f809f280SWarner Losh MODULE_VERSION(uled, 1);
122f809f280SWarner Losh USB_PNP_HOST_INFO(uled_devs);
123f809f280SWarner Losh
1246bd03b20SKevin Lo static int
uled_probe(device_t dev)1256bd03b20SKevin Lo uled_probe(device_t dev)
1266bd03b20SKevin Lo {
1276bd03b20SKevin Lo struct usb_attach_arg *uaa;
1286bd03b20SKevin Lo
1296bd03b20SKevin Lo uaa = device_get_ivars(dev);
1306bd03b20SKevin Lo if (uaa->usb_mode != USB_MODE_HOST)
1316bd03b20SKevin Lo return (ENXIO);
1326bd03b20SKevin Lo if (uaa->info.bInterfaceClass != UICLASS_HID)
1336bd03b20SKevin Lo return (ENXIO);
1346bd03b20SKevin Lo
1356bd03b20SKevin Lo return (usbd_lookup_id_by_uaa(uled_devs, sizeof(uled_devs), uaa));
1366bd03b20SKevin Lo }
1376bd03b20SKevin Lo
1386bd03b20SKevin Lo static int
uled_attach(device_t dev)1396bd03b20SKevin Lo uled_attach(device_t dev)
1406bd03b20SKevin Lo {
1416bd03b20SKevin Lo struct usb_attach_arg *uaa;
1426bd03b20SKevin Lo struct uled_softc *sc;
1436bd03b20SKevin Lo int unit;
1446bd03b20SKevin Lo usb_error_t error;
1456bd03b20SKevin Lo
1466bd03b20SKevin Lo uaa = device_get_ivars(dev);
1476bd03b20SKevin Lo sc = device_get_softc(dev);
1486bd03b20SKevin Lo unit = device_get_unit(dev);
1490642eac4SKevin Lo sc->sc_flags = USB_GET_DRIVER_INFO(uaa);
1506bd03b20SKevin Lo
1516bd03b20SKevin Lo device_set_usb_desc(dev);
1526bd03b20SKevin Lo mtx_init(&sc->sc_mtx, "uled lock", NULL, MTX_DEF | MTX_RECURSE);
1536bd03b20SKevin Lo
1546bd03b20SKevin Lo sc->sc_udev = uaa->device;
1556bd03b20SKevin Lo
1566bd03b20SKevin Lo error = usb_fifo_attach(uaa->device, sc, &sc->sc_mtx,
1576bd03b20SKevin Lo &uled_fifo_methods, &sc->sc_fifo, unit, -1,
1586bd03b20SKevin Lo uaa->info.bIfaceIndex, UID_ROOT, GID_OPERATOR, 0644);
1596bd03b20SKevin Lo if (error != 0)
1606bd03b20SKevin Lo goto detach;
1616bd03b20SKevin Lo
1626bd03b20SKevin Lo sc->sc_color.red = 0;
1636bd03b20SKevin Lo sc->sc_color.green = 0;
1646bd03b20SKevin Lo sc->sc_color.blue = 0;
1656bd03b20SKevin Lo
1666bd03b20SKevin Lo return (0);
1676bd03b20SKevin Lo
1686bd03b20SKevin Lo detach:
1696bd03b20SKevin Lo uled_detach(dev);
1706bd03b20SKevin Lo return (ENOMEM);
1716bd03b20SKevin Lo }
1726bd03b20SKevin Lo
1736bd03b20SKevin Lo static int
uled_detach(device_t dev)1746bd03b20SKevin Lo uled_detach(device_t dev)
1756bd03b20SKevin Lo {
1766bd03b20SKevin Lo struct uled_softc *sc;
1776bd03b20SKevin Lo
1786bd03b20SKevin Lo sc = device_get_softc(dev);
1796bd03b20SKevin Lo usb_fifo_detach(&sc->sc_fifo);
1806bd03b20SKevin Lo mtx_destroy(&sc->sc_mtx);
1816bd03b20SKevin Lo return (0);
1826bd03b20SKevin Lo }
1836bd03b20SKevin Lo
1846bd03b20SKevin Lo static usb_error_t
uled_ctrl_msg(struct uled_softc * sc,uint8_t rt,uint8_t reqno,uint16_t value,uint16_t index,void * buf,uint16_t buflen)1856bd03b20SKevin Lo uled_ctrl_msg(struct uled_softc *sc, uint8_t rt, uint8_t reqno,
1866bd03b20SKevin Lo uint16_t value, uint16_t index, void *buf, uint16_t buflen)
1876bd03b20SKevin Lo {
1886bd03b20SKevin Lo struct usb_device_request req;
1896bd03b20SKevin Lo
1906bd03b20SKevin Lo req.bmRequestType = rt;
1916bd03b20SKevin Lo req.bRequest = reqno;
1926bd03b20SKevin Lo USETW(req.wValue, value);
1936bd03b20SKevin Lo USETW(req.wIndex, index);
1946bd03b20SKevin Lo USETW(req.wLength, buflen);
1956bd03b20SKevin Lo
1966bd03b20SKevin Lo return (usbd_do_request_flags(sc->sc_udev, &sc->sc_mtx, &req, buf,
1976bd03b20SKevin Lo 0, NULL, 2000));
1986bd03b20SKevin Lo }
1996bd03b20SKevin Lo
2006bd03b20SKevin Lo static int
uled_enable(struct uled_softc * sc)2016bd03b20SKevin Lo uled_enable(struct uled_softc *sc)
2026bd03b20SKevin Lo {
2030642eac4SKevin Lo uint8_t *cmdbuf;
2046bd03b20SKevin Lo int error;
2056bd03b20SKevin Lo
2060642eac4SKevin Lo cmdbuf = (sc->sc_flags & ULED_FLAG_BLINK1) ? blink1 : dl100b;
2070642eac4SKevin Lo
2086bd03b20SKevin Lo sc->sc_state |= ULED_ENABLED;
2096bd03b20SKevin Lo mtx_lock(&sc->sc_mtx);
2106bd03b20SKevin Lo error = uled_ctrl_msg(sc, UT_WRITE_CLASS_INTERFACE, UR_SET_REPORT,
2116bd03b20SKevin Lo 0x200, 0, cmdbuf, sizeof(cmdbuf));
2126bd03b20SKevin Lo mtx_unlock(&sc->sc_mtx);
2136bd03b20SKevin Lo return (error);
2146bd03b20SKevin Lo }
2156bd03b20SKevin Lo
2166bd03b20SKevin Lo static int
uled_open(struct usb_fifo * fifo,int fflags)2176bd03b20SKevin Lo uled_open(struct usb_fifo *fifo, int fflags)
2186bd03b20SKevin Lo {
2196bd03b20SKevin Lo if (fflags & FREAD) {
2206bd03b20SKevin Lo struct uled_softc *sc;
2216bd03b20SKevin Lo int rc;
2226bd03b20SKevin Lo
2236bd03b20SKevin Lo sc = usb_fifo_softc(fifo);
2246bd03b20SKevin Lo if (sc->sc_state & ULED_ENABLED)
2256bd03b20SKevin Lo return (EBUSY);
2266bd03b20SKevin Lo if ((rc = uled_enable(sc)) != 0)
2276bd03b20SKevin Lo return (rc);
2286bd03b20SKevin Lo }
2296bd03b20SKevin Lo return (0);
2306bd03b20SKevin Lo }
2316bd03b20SKevin Lo
2326bd03b20SKevin Lo static void
uled_close(struct usb_fifo * fifo,int fflags)2336bd03b20SKevin Lo uled_close(struct usb_fifo *fifo, int fflags)
2346bd03b20SKevin Lo {
2356bd03b20SKevin Lo if (fflags & FREAD) {
2366bd03b20SKevin Lo struct uled_softc *sc;
2376bd03b20SKevin Lo
2386bd03b20SKevin Lo sc = usb_fifo_softc(fifo);
2396bd03b20SKevin Lo sc->sc_state &= ~ULED_ENABLED;
2406bd03b20SKevin Lo }
2416bd03b20SKevin Lo }
2426bd03b20SKevin Lo
2436bd03b20SKevin Lo static int
uled_ioctl(struct usb_fifo * fifo,u_long cmd,void * addr,int fflags)2446bd03b20SKevin Lo uled_ioctl(struct usb_fifo *fifo, u_long cmd, void *addr, int fflags)
2456bd03b20SKevin Lo {
2466bd03b20SKevin Lo struct uled_softc *sc;
2476bd03b20SKevin Lo struct uled_color color;
2486bd03b20SKevin Lo int error;
2496bd03b20SKevin Lo
2506bd03b20SKevin Lo sc = usb_fifo_softc(fifo);
2516bd03b20SKevin Lo error = 0;
2526bd03b20SKevin Lo
2536bd03b20SKevin Lo mtx_lock(&sc->sc_mtx);
2546bd03b20SKevin Lo
2556bd03b20SKevin Lo switch(cmd) {
2566bd03b20SKevin Lo case ULED_GET_COLOR:
2576bd03b20SKevin Lo *(struct uled_color *)addr = sc->sc_color;
2586bd03b20SKevin Lo break;
2596bd03b20SKevin Lo case ULED_SET_COLOR:
2606bd03b20SKevin Lo color = *(struct uled_color *)addr;
2616bd03b20SKevin Lo uint8_t buf[8];
2626bd03b20SKevin Lo
2636bd03b20SKevin Lo sc->sc_color.red = color.red;
2646bd03b20SKevin Lo sc->sc_color.green = color.green;
2656bd03b20SKevin Lo sc->sc_color.blue = color.blue;
2666bd03b20SKevin Lo
2670642eac4SKevin Lo if (sc->sc_flags & ULED_FLAG_BLINK1) {
2680642eac4SKevin Lo buf[0] = 0x1;
2690642eac4SKevin Lo buf[1] = 'n';
2700642eac4SKevin Lo buf[2] = color.red;
2710642eac4SKevin Lo buf[3] = color.green;
2720642eac4SKevin Lo buf[4] = color.blue;
2730642eac4SKevin Lo buf[5] = buf[6] = buf[7] = 0;
2740642eac4SKevin Lo } else {
2756bd03b20SKevin Lo buf[0] = color.red;
2766bd03b20SKevin Lo buf[1] = color.green;
2776bd03b20SKevin Lo buf[2] = color.blue;
2786bd03b20SKevin Lo buf[3] = buf[4] = buf[5] = 0;
2796bd03b20SKevin Lo buf[6] = 0x1a;
2806bd03b20SKevin Lo buf[7] = 0x05;
2810642eac4SKevin Lo }
2826bd03b20SKevin Lo error = uled_ctrl_msg(sc, UT_WRITE_CLASS_INTERFACE,
2836bd03b20SKevin Lo UR_SET_REPORT, 0x200, 0, buf, sizeof(buf));
2846bd03b20SKevin Lo break;
2856bd03b20SKevin Lo default:
2866bd03b20SKevin Lo error = ENOTTY;
2876bd03b20SKevin Lo break;
2886bd03b20SKevin Lo }
2896bd03b20SKevin Lo
2906bd03b20SKevin Lo mtx_unlock(&sc->sc_mtx);
2916bd03b20SKevin Lo return (error);
2926bd03b20SKevin Lo }
293