* Copyright (c) 2008 The NetBSD Foundation, Inc.

/* $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;
}