/*	$NetBSD: uthum.c,v 1.23 2022/03/28 12:44:17 riastradh Exp $   */
/*	$OpenBSD: uthum.c,v 1.6 2010/01/03 18:43:02 deraadt Exp $   */

/*
 * Copyright (c) 2009 Yojiro UO <yuo@nui.org>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

/*
 * Driver for TEMPer and TEMPerHUM HID
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: uthum.c,v 1.23 2022/03/28 12:44:17 riastradh Exp $");

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

#include <sys/param.h>
#include <sys/proc.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/device.h>
#include <sys/conf.h>

#include <dev/sysmon/sysmonvar.h>

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

#ifdef UTHUM_DEBUG
int	uthumdebug = 0;
#define DPRINTFN(n, x)	do { if (uthumdebug > (n)) printf x; } while (0)
#else
#define DPRINTFN(n, x)
#endif

#define DPRINTF(x) DPRINTFN(0, x)


/* TEMPerHUM */
#define CMD_DEVTYPE 0x52 /* XXX */
#define CMD_GETDATA 0x48 /* XXX */
#define CMD_GETTEMP 0x54 /* XXX */
static const uint8_t cmd_start[8] =
	{ 0x0a, 0x0b, 0x0c, 0x0d, 0x00, 0x00, 0x02, 0x00 };
static const uint8_t cmd_end[8] =
	{ 0x0a, 0x0b, 0x0c, 0x0d, 0x00, 0x00, 0x01, 0x00 };

/* sensors */
#define UTHUM_TEMP		0
#define UTHUM_HUMIDITY		1
#define UTHUM_MAX_SENSORS	2

#define UTHUM_TYPE_UNKNOWN	0
#define UTHUM_TYPE_SHT1x	1
#define UTHUM_TYPE_TEMPER	2

struct uthum_softc {
	device_t		 sc_dev;
	struct uhidev *		 sc_hdev;
	struct usbd_device *	 sc_udev;
	u_char			 sc_dying;
	uint16_t		 sc_flag;
	int			 sc_sensortype;

	/* uhidev parameters */
	size_t			 sc_flen;	/* feature report length */
	size_t			 sc_olen;	/* output report length */

	/* sensor framework */
	struct sysmon_envsys		 *sc_sme;
	envsys_data_t			 sc_sensor[UTHUM_MAX_SENSORS];

	uint8_t			 sc_num_sensors;
};


static const struct usb_devno uthum_devs[] = {
	/* XXX: various TEMPer variants using same VID/PID */
	{ USB_VENDOR_TENX, USB_PRODUCT_TENX_TEMPER},
};
#define uthum_lookup(v, p) usb_lookup(uthum_devs, v, p)

static int uthum_match(device_t, cfdata_t, void *);
static void uthum_attach(device_t, device_t, void *);
static int uthum_detach(device_t, int);
static int uthum_activate(device_t, enum devact);

static int uthum_read_data(struct uthum_softc *, uint8_t, uint8_t *, size_t, int);
static int uthum_check_sensortype(struct uthum_softc *);
static int uthum_temper_temp(uint8_t, uint8_t);
static int uthum_sht1x_temp(uint8_t, uint8_t);
static int uthum_sht1x_rh(unsigned int, int);

static void uthum_refresh(struct sysmon_envsys *, envsys_data_t *);


CFATTACH_DECL_NEW(uthum, sizeof(struct uthum_softc), uthum_match, uthum_attach,
    uthum_detach, uthum_activate);

static int
uthum_match(device_t parent, cfdata_t match, void *aux)
{
	struct uhidev_attach_arg *uha = aux;

	return uthum_lookup(uha->uiaa->uiaa_vendor, uha->uiaa->uiaa_product)
	    != NULL ? UMATCH_VENDOR_PRODUCT : UMATCH_NONE;
}

static void
uthum_attach(device_t parent, device_t self, void *aux)
{
	struct uthum_softc *sc = device_private(self);
	struct uhidev_attach_arg *uha = aux;
	struct usbd_device *dev = uha->uiaa->uiaa_device;
	int size, repid;
	void *desc;

	sc->sc_dev = self;
	sc->sc_hdev = uha->parent;
	sc->sc_udev = dev;
	sc->sc_num_sensors = 0;

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

	uhidev_get_report_desc(uha->parent, &desc, &size);
	repid = uha->reportid;
	sc->sc_olen = hid_report_size(desc, size, hid_output, repid);
	sc->sc_flen = hid_report_size(desc, size, hid_feature, repid);

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

	if (sc->sc_flen < 32) {
		/* not sensor interface, just attach */
		return;
	}

	sc->sc_sensortype = uthum_check_sensortype(sc);

	/* attach sensor */
	sc->sc_sme = sysmon_envsys_create();
	sc->sc_sme->sme_name = device_xname(self);

	switch (sc->sc_sensortype) {
	case UTHUM_TYPE_SHT1x:
		(void)strlcpy(sc->sc_sensor[UTHUM_TEMP].desc, "temp",
		    sizeof(sc->sc_sensor[UTHUM_TEMP].desc));
		sc->sc_sensor[UTHUM_TEMP].units = ENVSYS_STEMP;
		sc->sc_sensor[UTHUM_TEMP].state = ENVSYS_SINVALID;

		(void)strlcpy(sc->sc_sensor[UTHUM_HUMIDITY].desc,
		    "relative humidity",
		    sizeof(sc->sc_sensor[UTHUM_HUMIDITY].desc));
		sc->sc_sensor[UTHUM_HUMIDITY].units = ENVSYS_INTEGER;
		sc->sc_sensor[UTHUM_HUMIDITY].value_cur = 0;
		sc->sc_sensor[UTHUM_HUMIDITY].state = ENVSYS_SINVALID;

		sysmon_envsys_sensor_attach(sc->sc_sme,
		    &sc->sc_sensor[UTHUM_TEMP]);
		sysmon_envsys_sensor_attach(sc->sc_sme,
		    &sc->sc_sensor[UTHUM_HUMIDITY]);
		sc->sc_num_sensors = 2;
		DPRINTF(("sensor type: SHT1x\n"));
		break;
	case UTHUM_TYPE_TEMPER:
		(void)strlcpy(sc->sc_sensor[UTHUM_TEMP].desc, "temp",
		    sizeof(sc->sc_sensor[UTHUM_TEMP].desc));
		sc->sc_sensor[UTHUM_TEMP].units = ENVSYS_STEMP;
		sc->sc_sensor[UTHUM_TEMP].state = ENVSYS_SINVALID;

		sysmon_envsys_sensor_attach(sc->sc_sme,
		    &sc->sc_sensor[UTHUM_TEMP]);
		sc->sc_num_sensors = 1;
		DPRINTF(("sensor type: TEMPer\n"));
		break;
	case UTHUM_TYPE_UNKNOWN:
		DPRINTF(("sensor type: unknown, give up to attach sensors\n"));
	default:
		break;
	}

	if (sc->sc_num_sensors > 0) {
		sc->sc_sme->sme_cookie = sc;
		sc->sc_sme->sme_refresh = uthum_refresh;

		if (sysmon_envsys_register(sc->sc_sme)) {
			sysmon_envsys_destroy(sc->sc_sme);
			sc->sc_sme = NULL;
			aprint_error_dev(self,
			    "unable to register with sysmon\n");
		}
	} else {
		sysmon_envsys_destroy(sc->sc_sme);
		sc->sc_sme = NULL;
	}

	DPRINTF(("uthum_attach: complete\n"));
}

static int
uthum_detach(device_t self, int flags)
{
	struct uthum_softc *sc = device_private(self);
	int rv = 0;

	sc->sc_dying = 1;

	if (sc->sc_sme != NULL)
		sysmon_envsys_unregister(sc->sc_sme);

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

	return rv;
}

static int
uthum_activate(device_t self, enum devact act)
{
	struct uthum_softc *sc = device_private(self);

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

static int
uthum_read_data(struct uthum_softc *sc, uint8_t target_cmd, uint8_t *buf,
	size_t len, int need_delay)
{
	int i;
	uint8_t cmdbuf[32], report[256];
	size_t olen, flen;

	/* if return buffer is null, do nothing */
	if ((buf == NULL) || len == 0)
		return 0;

	olen = uimin(sc->sc_olen, sizeof(cmdbuf));

	/* issue query */
	memset(cmdbuf, 0, sizeof(cmdbuf));
	memcpy(cmdbuf, cmd_start, sizeof(cmd_start));
	if (uhidev_set_report(sc->sc_hdev, UHID_OUTPUT_REPORT,
	    cmdbuf, olen))
		return EIO;

	memset(cmdbuf, 0, sizeof(cmdbuf));
	cmdbuf[0] = target_cmd;
	if (uhidev_set_report(sc->sc_hdev, UHID_OUTPUT_REPORT,
	    cmdbuf, olen))
		return EIO;

	memset(cmdbuf, 0, sizeof(cmdbuf));
	for (i = 0; i < 7; i++) {
		if (uhidev_set_report(sc->sc_hdev, UHID_OUTPUT_REPORT,
		    cmdbuf, olen))
			return EIO;
	}
	memcpy(cmdbuf, cmd_end, sizeof(cmd_end));
	if (uhidev_set_report(sc->sc_hdev, UHID_OUTPUT_REPORT,
	    cmdbuf, olen))
		return EIO;

	/* wait if required */
	if (need_delay > 1)
		kpause("uthum", false, (need_delay*hz+999)/1000 + 1, NULL);

	/* get answer */
	flen = uimin(sc->sc_flen, sizeof(report));
	if (uhidev_get_report(sc->sc_hdev, UHID_FEATURE_REPORT,
	    report, flen))
		return EIO;
	memcpy(buf, report, len);
	return 0;
}

static int
uthum_check_sensortype(struct uthum_softc *sc)
{
	uint8_t buf[8];
	static uint8_t sht1x_sig0[] =
	    { 0x57, 0x5a, 0x13, 0x00, 0x14, 0x00, 0x53, 0x00 };
	static uint8_t sht1x_sig1[] =
	    { 0x57, 0x5a, 0x14, 0x00, 0x14, 0x00, 0x53, 0x00 };
	static uint8_t temper_sig[] =
	    { 0x57, 0x58, 0x14, 0x00, 0x14, 0x00, 0x53, 0x00 };

	if (uthum_read_data(sc, CMD_DEVTYPE, buf, sizeof(buf), 0) != 0) {
		DPRINTF(("uthum: read fail\n"));
		return UTHUM_TYPE_UNKNOWN;
	}

	/*
	 * currently we have not enough information about the return value,
	 * therefore, compare full bytes.
	 * TEMPerHUM HID (SHT1x version) will return:
	 *   { 0x57, 0x5a, 0x13, 0x00, 0x14, 0x00, 0x53, 0x00 }
	 *   { 0x57, 0x5a, 0x14, 0x00, 0x14, 0x00, 0x53, 0x00 }
	 * TEMPer HID (TEMPer version) will return:
	 *   { 0x57, 0x58, 0x14, 0x00, 0x14, 0x00, 0x53, 0x00 }
	 */
	DPRINTF(("uthum: device signature: "
	    "0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x\n",
	    buf[0], buf[1], buf[2], buf[3], buf[4], buf[5], buf[6], buf[7]));
	if (0 == memcmp(buf, sht1x_sig0, sizeof(sht1x_sig0)))
		return UTHUM_TYPE_SHT1x;
	if (0 == memcmp(buf, sht1x_sig1, sizeof(sht1x_sig1)))
		return UTHUM_TYPE_SHT1x;
	if (0 == memcmp(buf, temper_sig, sizeof(temper_sig)))
		return UTHUM_TYPE_TEMPER;

	return UTHUM_TYPE_UNKNOWN;
}


static void
uthum_refresh(struct sysmon_envsys *sme, envsys_data_t *edata)
{
	struct uthum_softc *sc = sme->sme_cookie;
	uint8_t buf[8];
	unsigned int humidity_tick;
	int temp, rh;

	switch (sc->sc_sensortype) {
	case UTHUM_TYPE_SHT1x:
		if (uthum_read_data(sc, CMD_GETDATA, buf, sizeof(buf), 1000)
		    != 0) {
			DPRINTF(("uthum: data read fail\n"));
			sc->sc_sensor[UTHUM_TEMP].state = ENVSYS_SINVALID;
			sc->sc_sensor[UTHUM_HUMIDITY].state = ENVSYS_SINVALID;
			return;
		}
		DPRINTF(("%s: read SHT1x data "
		    "0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x\n",
		    sc->sc_sme->sme_name, buf[0], buf[1], buf[2], buf[3],
		    buf[4], buf[5], buf[6], buf[7]));

		humidity_tick = (buf[2] * 256 + buf[3]) & 0x0fff;

		temp = uthum_sht1x_temp(buf[0], buf[1]);
		rh = uthum_sht1x_rh(humidity_tick, temp);

		sc->sc_sensor[UTHUM_HUMIDITY].value_cur = rh / 1000;
		sc->sc_sensor[UTHUM_HUMIDITY].state = ENVSYS_SVALID;
		break;
	case UTHUM_TYPE_TEMPER:
		if (uthum_read_data(sc, CMD_GETTEMP, buf, sizeof(buf), 0) != 0) {
			DPRINTF(("uthum: data read fail\n"));
			sc->sc_sensor[UTHUM_TEMP].state = ENVSYS_SINVALID;
			return;
		}
		DPRINTF(("%s: read TEMPER data "
		    "0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x 0x%0x\n",
		    sc->sc_sme->sme_name, buf[0], buf[1], buf[2], buf[3],
		    buf[4], buf[5], buf[6], buf[7]));
		temp = uthum_temper_temp(buf[0], buf[1]);
		break;
	default:
		/* do nothing */
		return;
	}

	sc->sc_sensor[UTHUM_TEMP].value_cur = (temp * 10000) + 273150000;
	sc->sc_sensor[UTHUM_TEMP].state = ENVSYS_SVALID;
}

/* return C-degree * 100 value */
static int
uthum_temper_temp(uint8_t msb, uint8_t lsb)
{
	int val;

	val = (msb << 8) | lsb;
	if (val >= 32768) {
		val = val - 65536;
	}
	val = (val * 100) >> 8;
	return val;
}

/* return C-degree * 100 value */
static int
uthum_sht1x_temp(uint8_t msb, uint8_t lsb)
{
	int val;

	val = ((msb << 8) + lsb) - 4096;
	return val;
}

/* return %RH * 1000 */
static int
uthum_sht1x_rh(unsigned int ticks, int temp)
{
	int rh_l, rh;

	rh_l = (-40000 + 405 * ticks) - ((7 * ticks * ticks) / 250);
	rh = ((temp - 2500) * (1 + (ticks >> 7)) + rh_l) / 10;
	return rh;
}