/* $NetBSD: auvitek.c,v 1.13 2022/03/13 12:49:36 riastradh Exp $ */

/*-
* Copyright (c) 2010 Jared D. McNeill <[email protected]>
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
* ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
* TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
* POSSIBILITY OF SUCH DAMAGE.
*/

/*
* Auvitek AU0828 USB controller
*/

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: auvitek.c,v 1.13 2022/03/13 12:49:36 riastradh Exp $");

#include <sys/types.h>
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/device.h>
#include <sys/conf.h>
#include <sys/bus.h>
#include <sys/module.h>

#include <dev/usb/usb.h>
#include <dev/usb/usbdi.h>
#include <dev/usb/usbdi_util.h>
#include <dev/usb/usbdevs.h>

#include <dev/usb/auvitekreg.h>
#include <dev/usb/auvitekvar.h>

static int      auvitek_match(device_t, cfdata_t, void *);
static void     auvitek_attach(device_t, device_t, void *);
static int      auvitek_detach(device_t, int);
static int      auvitek_rescan(device_t, const char *, const int *);
static void     auvitek_childdet(device_t, device_t);
static int      auvitek_activate(device_t, enum devact);

CFATTACH_DECL2_NEW(auvitek, sizeof(struct auvitek_softc),
   auvitek_match, auvitek_attach, auvitek_detach, auvitek_activate,
   auvitek_rescan, auvitek_childdet);

static const struct {
       uint16_t                vendor;
       uint16_t                product;
       const char *            name;
       enum auvitek_board      board;
} auvitek_devices[] = {
       { 0x2040, 0x7200,
         "WinTV HVR-950Q", AUVITEK_BOARD_HVR_950Q },
       { 0x2040, 0x7240,
         "WinTV HVR-850", AUVITEK_BOARD_HVR_850 },
};

static int
auvitek_match(device_t parent, cfdata_t match, void *opaque)
{
       struct usb_attach_arg *uaa = opaque;
       unsigned int i;

       for (i = 0; i < __arraycount(auvitek_devices); i++) {
               if (auvitek_devices[i].vendor == uaa->uaa_vendor &&
                   auvitek_devices[i].product == uaa->uaa_product)
                       return UMATCH_VENDOR_PRODUCT;
       }

       return UMATCH_NONE;
}

static void
auvitek_attach(device_t parent, device_t self, void *opaque)
{
       struct auvitek_softc *sc = device_private(self);
       struct usb_attach_arg *uaa = opaque;
       struct usbd_device *dev = uaa->uaa_device;
       usb_endpoint_descriptor_t *ed;
       usbd_status err;
       unsigned int i;
       uint8_t nep;

       aprint_naive("\n");
       aprint_normal(": AU0828\n");

       sc->sc_dev = self;
       sc->sc_udev = dev;
       sc->sc_uport = uaa->uaa_port;

       for (i = 0; i < __arraycount(auvitek_devices); i++) {
               if (auvitek_devices[i].vendor == uaa->uaa_vendor &&
                   auvitek_devices[i].product == uaa->uaa_product)
                       break;
       }
       KASSERT(i != __arraycount(auvitek_devices));
       sc->sc_descr = auvitek_devices[i].name;
       sc->sc_board = auvitek_devices[i].board;

       sc->sc_dying = sc->sc_running = 0;

       mutex_init(&sc->sc_subdev_lock, MUTEX_DEFAULT, IPL_NONE);

       err = usbd_set_config_index(dev, 0, 1);
       if (err) {
               aprint_error_dev(self, "couldn't set config index: %s\n",
                   usbd_errstr(err));
               return;
       }
       err = usbd_device2interface_handle(dev, 0, &sc->sc_isoc_iface);
       if (err) {
               aprint_error_dev(self, "couldn't get interface handle: %s\n",
                   usbd_errstr(err));
               return;
       }
       err = usbd_device2interface_handle(dev, 3, &sc->sc_bulk_iface);
       if (err) {
               aprint_error_dev(self, "couldn't get interface handle: %s\n",
                   usbd_errstr(err));
               return;
       }

       sc->sc_ax.ax_sc = sc->sc_ab.ab_sc = sc;
       sc->sc_ax.ax_endpt = sc->sc_ab.ab_endpt = -1;

       err = usbd_set_interface(sc->sc_isoc_iface, AUVITEK_XFER_ALTNO);
       if (err) {
               aprint_error_dev(self, "couldn't set interface: %s\n",
                   usbd_errstr(err));
               return;
       }

       nep = 0;
       usbd_endpoint_count(sc->sc_isoc_iface, &nep);
       for (i = 0; i < nep; i++) {
               int dir, type;

               ed = usbd_interface2endpoint_descriptor(sc->sc_isoc_iface, i);
               if (ed == NULL) {
                       aprint_error_dev(self,
                           "couldn't read endpoint descriptor %d\n", i);
                       continue;
               }

               dir = UE_GET_DIR(ed->bEndpointAddress);
               type = UE_GET_XFERTYPE(ed->bmAttributes);

               if (dir == UE_DIR_IN && type == UE_ISOCHRONOUS &&
                   sc->sc_ax.ax_endpt == -1) {
                       sc->sc_ax.ax_endpt = ed->bEndpointAddress;
                       sc->sc_ax.ax_maxpktlen =
                           UE_GET_SIZE(UGETW(ed->wMaxPacketSize)) *
                           (UE_GET_TRANS(UGETW(ed->wMaxPacketSize)) + 1);
               }
       }

       err = usbd_set_interface(sc->sc_isoc_iface, 0);
       if (err) {
               aprint_error_dev(self, "couldn't set interface: %s\n",
                   usbd_errstr(err));
               return;
       }

       if (sc->sc_ax.ax_endpt == -1) {
               aprint_error_dev(self, "couldn't find isoc endpoint\n");
               sc->sc_dying = 1;
               return;
       }
       if (sc->sc_ax.ax_maxpktlen == 0) {
               aprint_error_dev(self, "couldn't determine packet length\n");
               sc->sc_dying = 1;
               return;
       }

       aprint_debug_dev(self, "isoc endpoint 0x%02x size %d\n",
           sc->sc_ax.ax_endpt, sc->sc_ax.ax_maxpktlen);

       nep = 0;
       usbd_endpoint_count(sc->sc_bulk_iface, &nep);
       for (i = 0; i < nep; i++) {
               int dir, type;

               ed = usbd_interface2endpoint_descriptor(sc->sc_bulk_iface, i);
               if (ed == NULL) {
                       aprint_error_dev(self,
                           "couldn't read endpoint descriptor %d\n", i);
                       continue;
               }

               dir = UE_GET_DIR(ed->bEndpointAddress);
               type = UE_GET_XFERTYPE(ed->bmAttributes);

               if (dir == UE_DIR_IN && type == UE_BULK &&
                   sc->sc_ab.ab_endpt == -1) {
                       sc->sc_ab.ab_endpt = ed->bEndpointAddress;
               }
       }

       if (sc->sc_ab.ab_endpt == -1) {
               aprint_error_dev(self, "couldn't find bulk endpoint\n");
               sc->sc_dying = 1;
               return;
       }

       aprint_debug_dev(self, "bulk endpoint 0x%02x size %d\n",
           sc->sc_ab.ab_endpt, AUVITEK_BULK_BUFLEN);

       auvitek_board_init(sc);

       auvitek_i2c_attach(sc);

       sc->sc_au8522 = au8522_open(self, &sc->sc_i2c, 0x8e >> 1,
           auvitek_board_get_if_frequency(sc));
       if (sc->sc_au8522 == NULL) {
               aprint_error_dev(sc->sc_dev, "couldn't initialize decoder\n");
               sc->sc_dying = 1;
               return;
       }

       config_mountroot(self, auvitek_attach_tuner);

       auvitek_video_attach(sc);
       auvitek_audio_attach(sc);
       auvitek_dtv_attach(sc);
}

void
auvitek_attach_tuner(device_t self)
{
       struct auvitek_softc *sc = device_private(self);

       mutex_enter(&sc->sc_subdev_lock);
       if (sc->sc_xc5k == NULL) {
               sc->sc_xc5k = xc5k_open(sc->sc_dev, &sc->sc_i2c, 0xc2 >> 1,
                   auvitek_board_tuner_reset, sc,
                   auvitek_board_get_if_frequency(sc),
                   FE_ATSC);
       }
       mutex_exit(&sc->sc_subdev_lock);
}

static int
auvitek_detach(device_t self, int flags)
{
       struct auvitek_softc *sc = device_private(self);
       int error;

       sc->sc_dying = 1;

       error = config_detach_children(self, flags);
       if (error) {
               /*
                * XXX Should ask autoconf to block open with
                * .d_cfdriver until we're done, instead of setting
                * this and then rolling it back.
                */
               sc->sc_dying = 0;
               return error;
       }

       pmf_device_deregister(self);

       auvitek_dtv_detach(sc, flags);
       auvitek_audio_detach(sc, flags);
       auvitek_video_detach(sc, flags);

       if (sc->sc_xc5k)
               xc5k_close(sc->sc_xc5k);
       if (sc->sc_au8522)
               au8522_close(sc->sc_au8522);

       auvitek_i2c_detach(sc, flags);

       mutex_destroy(&sc->sc_subdev_lock);

       return 0;
}

int
auvitek_activate(device_t self, enum devact act)
{
       struct auvitek_softc *sc = device_private(self);

       switch (act) {
       case DVACT_DEACTIVATE:
               sc->sc_dying = 1;
               return 0;
       default:
               return 0;
       }
}

static int
auvitek_rescan(device_t self, const char *ifattr, const int *locs)
{
       struct auvitek_softc *sc = device_private(self);

       auvitek_video_rescan(sc, ifattr, locs);
       auvitek_dtv_rescan(sc, ifattr, locs);
       auvitek_i2c_rescan(sc, ifattr, locs);

       return 0;
}

static void
auvitek_childdet(device_t self, device_t child)
{
       struct auvitek_softc *sc = device_private(self);

       auvitek_video_childdet(sc, child);
       auvitek_audio_childdet(sc, child);
       auvitek_dtv_childdet(sc, child);
}

uint8_t
auvitek_read_1(struct auvitek_softc *sc, uint16_t reg)
{
       usb_device_request_t req;
       usbd_status err;
       int actlen;
       uint8_t data;

       req.bmRequestType = UT_READ_VENDOR_DEVICE;
       req.bRequest = AU0828_CMD_REQUEST_IN;
       USETW(req.wValue, 0);
       USETW(req.wIndex, reg);
       USETW(req.wLength, sizeof(data));

       KERNEL_LOCK(1, curlwp);
       err = usbd_do_request_flags(sc->sc_udev, &req, &data, 0,
           &actlen, USBD_DEFAULT_TIMEOUT);
       KERNEL_UNLOCK_ONE(curlwp);

       if (err)
               printf("%s: read failed: %s\n", device_xname(sc->sc_dev),
                   usbd_errstr(err));

       return data;
}

void
auvitek_write_1(struct auvitek_softc *sc, uint16_t reg, uint8_t data)
{
       usb_device_request_t req;
       usbd_status err;
       int actlen;

       req.bmRequestType = UT_WRITE_VENDOR_DEVICE;
       req.bRequest = AU0828_CMD_REQUEST_OUT;
       USETW(req.wValue, data);
       USETW(req.wIndex, reg);
       USETW(req.wLength, 0);

       KERNEL_LOCK(1, curlwp);
       err = usbd_do_request_flags(sc->sc_udev, &req, NULL, 0,
           &actlen, USBD_DEFAULT_TIMEOUT);
       KERNEL_UNLOCK_ONE(curlwp);

       if (err)
               printf("%s: write failed: %s\n", device_xname(sc->sc_dev),
                   usbd_errstr(err));
}

MODULE(MODULE_CLASS_DRIVER, auvitek, "au8522,xc5k");

#ifdef _MODULE
#include "ioconf.c"
#endif

static int
auvitek_modcmd(modcmd_t cmd, void *opaque)
{
       switch (cmd) {
       case MODULE_CMD_INIT:
#ifdef _MODULE
               return config_init_component(cfdriver_ioconf_auvitek,
                   cfattach_ioconf_auvitek, cfdata_ioconf_auvitek);
#else
               return 0;
#endif
       case MODULE_CMD_FINI:
#ifdef _MODULE
               return config_fini_component(cfdriver_ioconf_auvitek,
                   cfattach_ioconf_auvitek, cfdata_ioconf_auvitek);
#else
               return 0;
#endif
       default:
               return ENOTTY;
       }
}