/*      $NetBSD: uberry.c,v 1.16 2019/12/15 16:48:27 tsutsui Exp $      */

/*-
 * Copyright (c) 2008 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Christos Zoulas.
 *
 * 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.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: uberry.c,v 1.16 2019/12/15 16:48:27 tsutsui Exp $");

#ifdef _KERNEL_OPT
#include "opt_usb.h"
#endif

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/device.h>
#include <sys/ioctl.h>
#include <sys/conf.h>
#include <sys/file.h>
#include <sys/select.h>
#include <sys/proc.h>
#include <sys/vnode.h>
#include <sys/poll.h>
#include <sys/bus.h>

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

#include <dev/usb/usbdevs.h>

#ifdef UBERRY_DEBUG
#define DPRINTF(x)      if (uberrydebug) printf x
#define DPRINTFN(n, x)  if (uberrydebug > n) printf x
int     uberrydebug = 0;
#else
#define DPRINTF(x)
#define DPRINTFN(n, x)
#endif

struct uberry_softc {
        device_t                sc_dev;
        struct usbd_device *    sc_udev;
};

/*
 * Note that we do not attach to USB_PRODUCT_RIM_BLACKBERRY_PEARL_DUAL
 * as we let umass claim the device instead.
 */
static const struct usb_devno uberry_devs[] = {
        { USB_VENDOR_RIM, USB_PRODUCT_RIM_BLACKBERRY },
        { USB_VENDOR_RIM, USB_PRODUCT_RIM_BLACKBERRY_PEARL },
};

#define uberry_lookup(v, p) usb_lookup(uberry_devs, v, p)
#define UBERRY_CONFIG_NO 1

static int      uberry_match(device_t, cfdata_t, void *);
static void     uberry_attach(device_t, device_t, void *);
static int      uberry_detach(device_t, int);

CFATTACH_DECL_NEW(uberry, sizeof(struct uberry_softc), uberry_match,
    uberry_attach, uberry_detach, NULL);

static void
uberry_cmd(struct uberry_softc *sc, uint8_t requestType, uint8_t reqno,
    uint8_t value, uint8_t index, void *data, uint8_t length)
{
        usb_device_request_t req;
        usbd_status err;

        DPRINTF(("berry cmd type=%x, number=%x, value=%d, index=%d, len=%d\n",
            requestType, reqno, value, index, length));
        req.bmRequestType = requestType;
        req.bRequest = reqno;
        USETW(req.wValue, value);
        USETW(req.wIndex, index);
        USETW(req.wLength, length);

        if ((err = usbd_do_request(sc->sc_udev, &req, data)) != 0)
                aprint_error_dev(sc->sc_dev, "sending command failed %d\n",
                    err);
}

static void
uberry_charge(struct uberry_softc *sc)
{
        char dummy[2];
        usbd_status err;

        if (sc->sc_udev->ud_power != USB_MAX_POWER) {
                uberry_cmd(sc, UT_READ | UT_VENDOR, 0xa5, 0, 1, dummy, 2);
                uberry_cmd(sc, UT_WRITE | UT_VENDOR, 0xa2, 0, 1, dummy, 0);
        }

        err = usbd_set_config_no(sc->sc_udev, UBERRY_CONFIG_NO, 1);
        if (err) {
                aprint_error_dev(sc->sc_dev, "failed to set configuration"
                    ", err=%s\n", usbd_errstr(err));
                return;
        }
}

/*
 * Expose both the USB mass storage interface and the database access one
 */
static void
uberry_dual_mode(struct uberry_softc *sc)
{
        char dummy[2];
        usbd_status err;

        uberry_cmd(sc, UT_READ | UT_VENDOR, 0xa9, 1, 1, dummy, 2);

        err = usbd_set_config_no(sc->sc_udev, UBERRY_CONFIG_NO, 1);
        if (err) {
                aprint_error_dev(sc->sc_dev, "failed to set configuration"
                    ", err=%s\n", usbd_errstr(err));
                return;
        }
}


static int
uberry_match(device_t parent, cfdata_t match, void *aux)
{
        struct usb_attach_arg *uaa = aux;

        DPRINTFN(50, ("uberry_match\n"));
        return (uberry_lookup(uaa->uaa_vendor, uaa->uaa_product) != NULL ?
                UMATCH_VENDOR_PRODUCT : UMATCH_NONE);
}

static void
uberry_attach(device_t parent, device_t self, void *aux)
{
        struct uberry_softc *sc = device_private(self);
        struct usb_attach_arg *uaa = aux;
        struct usbd_device *    dev = uaa->uaa_device;
        char                    *devinfop;

        DPRINTFN(10,("uberry_attach: sc=%p\n", sc));

        sc->sc_dev = self;
        sc->sc_udev = dev;

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

        devinfop = usbd_devinfo_alloc(dev, 0);
        aprint_normal_dev(self, "%s\n", devinfop);
        usbd_devinfo_free(devinfop);

        uberry_charge(sc);
        if (uaa->uaa_product == USB_PRODUCT_RIM_BLACKBERRY_PEARL)
                uberry_dual_mode(sc);

        DPRINTFN(10, ("uberry_attach: %p\n", sc->sc_udev));

        if (!pmf_device_register(self, NULL, NULL))
                aprint_error_dev(self, "couldn't establish power handler\n");

        usbd_add_drv_event(USB_EVENT_DRIVER_ATTACH, sc->sc_udev, sc->sc_dev);
        return;
}

static int
uberry_detach(device_t self, int flags)
{
        struct uberry_softc *sc = device_private(self);
        DPRINTF(("uberry_detach: sc=%p flags=%d\n", sc, flags));

        pmf_device_deregister(self);

        usbd_add_drv_event(USB_EVENT_DRIVER_DETACH, sc->sc_udev, sc->sc_dev);

        return 0;
}