/*	$OpenBSD: km.c,v 1.8 2013/09/17 13:42:34 kettenis Exp $	*/

/*
 * Copyright (c) 2008/2010 Constantine A. Murenin <cnst+dfly@bugmail.mojo.ru>
 *
 * 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 DISCLAIMS 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.
 */

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/bus.h>
#include <sys/sensors.h>

#include <machine/md_var.h>
#include <machine/specialreg.h>

#include <bus/pci/pcivar.h>
#include "pcidevs.h"


/*
 * AMD Family 10h/11h/14h/15h Processors, Function 3 -- Miscellaneous Control
 */

/* Function 3 Registers */
#define KM_REP_TEMP_CONTR_R	0xa4
#define KM_THERMTRIP_STAT_R	0xe4
#define KM_NORTHBRIDGE_CAP_R	0xe8
#define KM_CPUID_FAMILY_MODEL_R	0xfc

/* Operations on Reported Temperature Control Register */
#define KM_GET_CURTMP(r)	(((r) >> 21) & 0x7ff)

/* Operations on Thermtrip Status Register */
#define KM_GET_DIODEOFFSET(r)	(((r) >> 8) & 0x7f)


struct km_softc {
	device_t		sc_dev;
	struct ksensor		sc_sensor;
	struct ksensordev	sc_sensordev;
};

static void	km_identify(driver_t *, device_t);
static int	km_probe(device_t);
static int	km_attach(device_t);
static int	km_detach(device_t);
static void	km_refresh(void *);

static device_method_t km_methods[] = {
	DEVMETHOD(device_identify,	km_identify),
	DEVMETHOD(device_probe,		km_probe),
	DEVMETHOD(device_attach,	km_attach),
	DEVMETHOD(device_detach,	km_detach),
	{ NULL, NULL }
};

static driver_t km_driver = {
	"km",
	km_methods,
	sizeof(struct km_softc)
};

static devclass_t km_devclass;

DRIVER_MODULE(km, hostb, km_driver, km_devclass, NULL, NULL);


static void
km_identify(driver_t *driver, device_t parent)
{
	if (km_probe(parent) == ENXIO)
		return;
	if (device_find_child(parent, driver->name, -1) != NULL)
		return;
	device_add_child(parent, driver->name, -1);
}

static int
km_probe(device_t dev)
{
	char *desc;

	if (pci_get_vendor(dev) != PCI_VENDOR_AMD)
		return ENXIO;

	switch (pci_get_device(dev)) {
	case PCI_PRODUCT_AMD_AMD64_F10_MISC:
		desc = "AMD Family 10h temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F11_MISC:
		desc = "AMD Family 11h temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F14_MISC:
		if (CPUID_TO_FAMILY(cpu_id) == 0x12)
			desc = "AMD Family 12h temperature sensor";
		else
			desc = "AMD Family 14h temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F15_0x_MISC:
		desc = "AMD Family 15/0xh temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F15_1x_MISC:
		desc = "AMD Family 15/1xh temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F15_3x_MISC:
		desc = "AMD Family 15/3xh temperature sensor";
		break;
	case PCI_PRODUCT_AMD_AMD64_F16_MISC:
		desc = "AMD Family 16h temperature sensor";
		break;
	default:
		return ENXIO;
	}

	if (device_get_desc(dev) == NULL)
		device_set_desc(dev, desc);
	return 0;
}

static int
km_attach(device_t dev)
{
	struct km_softc	*sc;

	sc = device_get_softc(dev);
	sc->sc_dev = dev;

	strlcpy(sc->sc_sensordev.xname, device_get_nameunit(dev),
	    sizeof(sc->sc_sensordev.xname));

	sc->sc_sensor.type = SENSOR_TEMP;
	sensor_attach(&sc->sc_sensordev, &sc->sc_sensor);

	sensor_task_register(sc, km_refresh, 5);

	sensordev_install(&sc->sc_sensordev);
	return 0;
}

static int
km_detach(device_t dev)
{
	struct km_softc	*sc = device_get_softc(dev);

	sensordev_deinstall(&sc->sc_sensordev);
	sensor_task_unregister(sc);
	return 0;
}

static void
km_refresh(void *arg)
{
	struct km_softc	*sc = arg;
	struct ksensor	*s = &sc->sc_sensor;
	uint32_t	r;
	int		c;

	r = pci_read_config(sc->sc_dev, KM_REP_TEMP_CONTR_R, 4);
	c = KM_GET_CURTMP(r);
	s->value = c * 125000 + 273150000;
}