/*	$NetBSD$  */

/*-
 * Copyright (c) 2005
 *	Damien Bergamini <damien.bergamini@free.fr>
 *
 * 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/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD$");

/*
 * CardBus front-end for the Ralink RT2500 driver.
 */

#include "bpfilter.h"

#include <sys/param.h>
#include <sys/sockio.h>
#include <sys/mbuf.h>
#include <sys/kernel.h>
#include <sys/socket.h>
#include <sys/systm.h>
#include <sys/malloc.h>

#ifdef RAL_LKM
#if __NetBSD_Version__ > 106200000
#include <sys/ksyms.h>
#endif
#include <sys/lkm.h>
#endif

#include <machine/bus.h>
#include <machine/intr.h>

#include <net/if.h>
#include <net/if_dl.h>
#include <net/if_media.h>
#include <net/if_ether.h>

#include <net80211/ieee80211_var.h>
#include <net80211/ieee80211_rssadapt.h>
#include <net80211/ieee80211_radiotap.h>

#include <dev/ic/ralvar.h>

#include <dev/pci/pcireg.h>
#include <dev/pci/pcivar.h>
#include <dev/pci/pcidevs.h>

#include <dev/cardbus/cardbusvar.h>

struct ral_cardbus_softc {
	struct ral_softc	sc_sc;

	/* cardbus specific goo */
	cardbus_devfunc_t	sc_ct;
	cardbustag_t		sc_tag;
	void			*sc_ih;
	bus_size_t		sc_mapsize;
	pcireg_t		sc_bar_val;
	int			sc_intrline;
};

#define PCI_PRODUCT_RALINK_RT2560	0x0201

static int	ral_cardbus_match(struct device *, struct cfdata *, void *);
static void	ral_cardbus_attach(struct device *, struct device *, void *);
static int	ral_cardbus_detach(struct device *, int);

CFATTACH_DECL(ral_cardbus, sizeof (struct ral_cardbus_softc),
    ral_cardbus_match, ral_cardbus_attach, ral_cardbus_detach, ral_activate);

static int	ral_cardbus_enable(struct ral_softc *);
static void	ral_cardbus_disable(struct ral_softc *);
static void	ral_cardbus_power(struct ral_softc *, int);
static void	ral_cardbus_setup(struct ral_cardbus_softc *);

/* 
 * LKM port contributed by iMil <imil@gcu.info> 
 * based on ethfoo / tap
 */
#ifdef RAL_LKM
CFDRIVER_DECL(ral, DV_DULL, NULL);

static int cardbusloc[] = { -1, -1 }; /* device, function */
static struct cfparent cardbusparent = {
	"cardbus", 0, -1
};

static struct cfdata ral_cardbus_cfdata[] = {
	{"ral", "ral_cardbus", 0, FSTATE_STAR, \
	 cardbusloc, 0, &cardbusparent, 0},
	{ 0 }
};

int ral_cardbus_lkmentry(struct lkm_table *, int, int);
static int ral_cardbus_lkmload(struct lkm_table *, int);
static int ral_cardbus_lkmunload(struct lkm_table *, int);

MOD_MISC("ral_cardbus");

int
ral_cardbus_lkmentry(struct lkm_table *lkmtp, int cmd, int ver)
{
	DISPATCH(lkmtp, cmd, ver, ral_cardbus_lkmload,
		 ral_cardbus_lkmunload, lkm_nofunc);
}

extern struct cftablelist allcftables;
struct cftable ral_cardbus_cftable;

static int
ral_cardbus_lkmload(struct lkm_table *lkmtp, int cmd)
{
	int error = 0;

	error = config_cfdriver_attach(&ral_cd);
	if (error) {
		aprint_error("%s: unable to register cfdriver\n",
			     ral_cd.cd_name);
		goto out;
	}

	error = config_cfattach_attach(ral_cd.cd_name, &ral_cardbus_ca);
	if (error) {
		aprint_error("%s: unable to register cfattach\n",
			     ral_cd.cd_name);
		(void)config_cfdriver_detach(&ral_cd);
		goto out;
	}

	ral_cardbus_cftable.ct_cfdata = ral_cardbus_cfdata;
	TAILQ_INSERT_TAIL(&allcftables, &ral_cardbus_cftable, ct_list);

out:
	return error;
}

static int
ral_cardbus_lkmunload(struct lkm_table *lkmtp, int cmd)
{
	int error, i;
	
	for (i = 0; i < ral_cd.cd_ndevs; i++)
		if (ral_cd.cd_devs[i] != NULL &&
		    (error = config_detach(ral_cd.cd_devs[i], 0)) != 0) {
			aprint_error("%s: unable to detach instance\n",
				     ((struct device *)
				      ral_cd.cd_devs[i])->dv_xname);
			return error;
		}

	TAILQ_REMOVE(&allcftables, &ral_cardbus_cftable, ct_list);

	if ((error = config_cfattach_detach(ral_cd.cd_name, 
					    &ral_cardbus_ca)) != 0) {
		aprint_error("%s: unable to deregister cfattach\n",
			     ral_cd.cd_name);
		return error;
	}

	if ((error = config_cfdriver_detach(&ral_cd)) != 0) {
		aprint_error("%s: unable to deregister cfdriver\n",
			     ral_cd.cd_name);
		return error;
	}

	return (0);
}
#endif /* RAL_LKM */

static int
ral_cardbus_match(struct device *parent, struct cfdata *match, void *aux)
{
	struct cardbus_attach_args *ca = aux;

	if (CARDBUS_VENDOR(ca->ca_id) == PCI_VENDOR_RALINK &&
	    CARDBUS_PRODUCT(ca->ca_id) == PCI_PRODUCT_RALINK_RT2560)
		return 1;

	return 0;
}

static void
ral_cardbus_attach(struct device *parent, struct device *self, void *aux)
{
	struct ral_cardbus_softc *csc = (struct ral_cardbus_softc *)self;
	struct ral_softc *sc = &csc->sc_sc;
	struct cardbus_attach_args *ca = aux;
	cardbus_devfunc_t ct = ca->ca_ct;
	bus_addr_t base;
	int error;

	sc->sc_dmat = ca->ca_dmat;
	csc->sc_ct = ct;
	csc->sc_tag = ca->ca_tag;
	csc->sc_intrline = ca->ca_intrline;

	aprint_normal("\n");

	/* power management hooks */
	sc->sc_enable = ral_cardbus_enable;
	sc->sc_disable = ral_cardbus_disable;
	sc->sc_power = ral_cardbus_power;

	/* map control/status registers */
	error = Cardbus_mapreg_map(ct, CARDBUS_BASE0_REG,
	    CARDBUS_MAPREG_TYPE_MEM, 0, &sc->sc_st, &sc->sc_sh, &base,
	    &csc->sc_mapsize);
	if (error != 0) {
		aprint_error("%s: could not map memory space\n",
		    sc->sc_dev.dv_xname);
		return;
	}

#if rbus
#else
	(*cf->cardbus_mem_open)(cc, 0, base, base + csc->sc_mapsize);
#endif

	csc->sc_bar_val = base | CARDBUS_MAPREG_TYPE_MEM;

	/* set up the PCI configuration registers */
	ral_cardbus_setup(csc);

	ral_attach(sc);

	Cardbus_function_disable(ct);
}

static int
ral_cardbus_detach(struct device *self, int flags)
{
	struct ral_cardbus_softc *csc = (struct ral_cardbus_softc *)self;
	struct ral_softc *sc = &csc->sc_sc;
	cardbus_devfunc_t ct = csc->sc_ct;
	cardbus_chipset_tag_t cc = ct->ct_cc;
	cardbus_function_tag_t cf = ct->ct_cf;
	int error;

	error = ral_detach(sc);
	if (error != 0)
		return error;

	/* unhook the interrupt handler */
	if (csc->sc_ih != NULL) {
		cardbus_intr_disestablish(cc, cf, csc->sc_ih);
		csc->sc_ih = NULL;
	}

	/* release bus space and close window */
	Cardbus_mapreg_unmap(ct, CARDBUS_BASE0_REG, sc->sc_st, sc->sc_sh,
	    csc->sc_mapsize);

	return 0;
}

static int
ral_cardbus_enable(struct ral_softc *sc)
{
	struct ral_cardbus_softc *csc = (struct ral_cardbus_softc *)sc;
	cardbus_devfunc_t ct = csc->sc_ct;
	cardbus_chipset_tag_t cc = ct->ct_cc;
	cardbus_function_tag_t cf = ct->ct_cf;

	/* power on the socket */
	Cardbus_function_enable(ct);

	/* setup the PCI configuration registers */
	ral_cardbus_setup(csc);

	/* map and establish the interrupt handler */
	csc->sc_ih = cardbus_intr_establish(cc, cf, csc->sc_intrline, IPL_NET,
	    ral_intr, sc);
	if (csc->sc_ih == NULL) {
		aprint_error("%s: could not establish interrupt\n",
		    sc->sc_dev.dv_xname);
		Cardbus_function_disable(ct);
		return 1;
	}
	aprint_normal("%s: interrupting at %d\n", sc->sc_dev.dv_xname,
	    csc->sc_intrline);

	return 0;
}

static void
ral_cardbus_disable(struct ral_softc *sc)
{
	struct ral_cardbus_softc *csc = (struct ral_cardbus_softc *)sc;
	cardbus_devfunc_t ct = csc->sc_ct;
	cardbus_chipset_tag_t cc = ct->ct_cc;
	cardbus_function_tag_t cf = ct->ct_cf;

	/* unhook the interrupt handler */
	cardbus_intr_disestablish(cc, cf, csc->sc_ih);
	csc->sc_ih = NULL;

	/* power down the socket */
	Cardbus_function_disable(ct);
}

static void
ral_cardbus_power(struct ral_softc *sc, int why)
{
	struct ral_cardbus_softc *csc = (struct ral_cardbus_softc *)sc;

	if (why == PWR_RESUME) {
		/* kick the PCI configuration registers */
		ral_cardbus_setup(csc);
	}
}

static void
ral_cardbus_setup(struct ral_cardbus_softc *csc)
{
	struct ral_softc *sc = &csc->sc_sc;
	cardbus_devfunc_t ct = csc->sc_ct;
	cardbus_chipset_tag_t cc = ct->ct_cc;
	cardbus_function_tag_t cf = ct->ct_cf;
	pcireg_t reg;

#if defined(__NetBSD_Version__) && (__NetBSD_Version__ >= 299001100)
	(void)cardbus_setpowerstate(sc->sc_dev.dv_xname, ct, csc->sc_tag,
	    PCI_PWR_D0);
#else
	int pmreg;

	if (cardbus_get_capability(cc, cf, csc->sc_tag,
		PCI_CAP_PWRMGMT, &pmreg, 0)) {
		reg = cardbus_conf_read(cc, cf, csc->sc_tag, pmreg + 4);
#if 1 /* XXX Probably not right for CardBus. */
		if (reg & 0x03) {
			/*
			 * The card has lost all configuration data in
			 */
			printf("%s: unable to wake up from power state D3\n",  
			       sc->sc_dev.dv_xname);  
			return;
		}
#endif
		if (reg != 0) {  
			printf("%s: waking up from power state D%d\n",  
			       sc->sc_dev.dv_xname, reg);  
			cardbus_conf_write(cc, cf, csc->sc_tag,  
					   pmreg + 4, 0);  
		}  
	}
#endif /* __NetBSD_Version__ */

	/* program the BAR */
	cardbus_conf_write(cc, cf, csc->sc_tag, CARDBUS_BASE0_REG,
	    csc->sc_bar_val);

	/* make sure the right access type is on the cardbus bridge */
	(*cf->cardbus_ctrl)(cc, CARDBUS_MEM_ENABLE);
	(*cf->cardbus_ctrl)(cc, CARDBUS_BM_ENABLE);

	/* enable the appropriate bits in the PCI CSR */
	reg = cardbus_conf_read(cc, cf, csc->sc_tag,
	    CARDBUS_COMMAND_STATUS_REG);
	reg |= CARDBUS_COMMAND_MASTER_ENABLE | CARDBUS_COMMAND_MEM_ENABLE;
	cardbus_conf_write(cc, cf, csc->sc_tag, CARDBUS_COMMAND_STATUS_REG,
	    reg);
}