/*	$NetBSD: rdcide.c,v 1.8 2014/07/08 18:01:26 msaitoh Exp $	*/

/*
 * Copyright (c) 2011 Manuel Bouyer.
 *
 * 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 AUTHOR ``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 AUTHOR 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: rdcide.c,v 1.8 2014/07/08 18:01:26 msaitoh Exp $");

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

#include <dev/pci/pcivar.h>
#include <dev/pci/pcidevs.h>
#include <dev/pci/pciidereg.h>
#include <dev/pci/pciidevar.h>
#include <dev/pci/rdcide_reg.h>

static void rdcide_chip_map(struct pciide_softc *,
    const struct pci_attach_args *);
static void rdcide_setup_channel(struct ata_channel *);

static bool rdcide_resume(device_t, const pmf_qual_t *);
static bool rdcide_suspend(device_t, const pmf_qual_t *);
static int  rdcide_match(device_t, cfdata_t, void *);
static void rdcide_attach(device_t, device_t, void *);

static const struct pciide_product_desc pciide_intel_products[] =  {
	{ PCI_PRODUCT_RDC_R1011_IDE,
	  0,
	  "RDC R1011 IDE controller",
	  rdcide_chip_map,
	},
	{ PCI_PRODUCT_RDC_R1012_IDE,
	  0,
	  "RDC R1012 IDE controller",
	  rdcide_chip_map,
	},
};

CFATTACH_DECL_NEW(rdcide, sizeof(struct pciide_softc),
    rdcide_match, rdcide_attach, NULL, NULL);

static int
rdcide_match(device_t parent, cfdata_t match, void *aux)
{
	struct pci_attach_args *pa = aux;

	if (PCI_VENDOR(pa->pa_id) == PCI_VENDOR_RDC) {
		if (pciide_lookup_product(pa->pa_id, pciide_intel_products))
			return (2);
	}
	return (0);
}

static void
rdcide_attach(device_t parent, device_t self, void *aux)
{
	struct pci_attach_args *pa = aux;
	struct pciide_softc *sc = device_private(self);

	sc->sc_wdcdev.sc_atac.atac_dev = self;

	pciide_common_attach(sc, pa,
	    pciide_lookup_product(pa->pa_id, pciide_intel_products));

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

static bool
rdcide_resume(device_t dv, const pmf_qual_t *qual)
{
	struct pciide_softc *sc = device_private(dv);

	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_PATR,
	    sc->sc_pm_reg[0]);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_PSD1ATR,
	    sc->sc_pm_reg[1]);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_UDCCR,
	    sc->sc_pm_reg[2]);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_IIOCR,
	    sc->sc_pm_reg[3]);

	return true;
}

static bool
rdcide_suspend(device_t dv, const pmf_qual_t *qual)
{
	struct pciide_softc *sc = device_private(dv);

	sc->sc_pm_reg[0] = pci_conf_read(sc->sc_pc, sc->sc_tag,
	    RDCIDE_PATR);
	sc->sc_pm_reg[1] = pci_conf_read(sc->sc_pc, sc->sc_tag,
	    RDCIDE_PSD1ATR);
	sc->sc_pm_reg[2] = pci_conf_read(sc->sc_pc, sc->sc_tag,
	    RDCIDE_UDCCR);
	sc->sc_pm_reg[3] = pci_conf_read(sc->sc_pc, sc->sc_tag,
	    RDCIDE_IIOCR);

	return true;
}

static void
rdcide_chip_map(struct pciide_softc *sc, const struct pci_attach_args *pa)
{
	struct pciide_channel *cp;
	int channel;
	u_int32_t patr;
	pcireg_t interface = PCI_INTERFACE(pa->pa_class);

	if (pciide_chipen(sc, pa) == 0)
		return;

	aprint_verbose_dev(sc->sc_wdcdev.sc_atac.atac_dev,
	    "bus-master DMA support present");
	pciide_mapreg_dma(sc, pa);
	aprint_verbose("\n");
	sc->sc_wdcdev.sc_atac.atac_cap |= ATAC_CAP_DATA16 | ATAC_CAP_DATA32;
	if (sc->sc_dma_ok) {
		sc->sc_wdcdev.sc_atac.atac_cap |= ATAC_CAP_DMA | ATAC_CAP_UDMA;
		sc->sc_wdcdev.irqack = pciide_irqack;
		sc->sc_wdcdev.dma_init = pciide_dma_init;
	}
	sc->sc_wdcdev.sc_atac.atac_pio_cap = 4;
	sc->sc_wdcdev.sc_atac.atac_dma_cap = 2;
	sc->sc_wdcdev.sc_atac.atac_udma_cap = 5;
	sc->sc_wdcdev.sc_atac.atac_set_modes = rdcide_setup_channel;
	sc->sc_wdcdev.sc_atac.atac_channels = sc->wdc_chanarray;
	sc->sc_wdcdev.sc_atac.atac_nchannels = PCIIDE_NUM_CHANNELS;
	sc->sc_wdcdev.wdc_maxdrives = 2;

	ATADEBUG_PRINT(("rdcide_setup_chip: old PATR=0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PATR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", PSD1ATR=0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PSD1ATR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", UDCCR 0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_UDCCR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", IIOCR 0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_IIOCR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT(("\n"), DEBUG_PROBE);

	wdc_allocate_regs(&sc->sc_wdcdev);

	for (channel = 0; channel < sc->sc_wdcdev.sc_atac.atac_nchannels;
	     channel++) {
		cp = &sc->pciide_channels[channel];
		if (pciide_chansetup(sc, channel, interface) == 0)
			continue;
		patr = pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PATR);
		if ((patr & RDCIDE_PATR_EN(channel)) == 0) {
			aprint_normal_dev(sc->sc_wdcdev.sc_atac.atac_dev,
			    "%s channel ignored (disabled)\n", cp->name);
			cp->ata_channel.ch_flags |= ATACH_DISABLED;
			continue;
		}
		pciide_mapchan(pa, cp, interface, pciide_pci_intr);
	}
	ATADEBUG_PRINT(("rdcide_setup_chip: PATR=0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PATR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", PSD1ATR=0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PSD1ATR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", UDCCR 0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_UDCCR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT((", IIOCR 0x%x",
	    pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_IIOCR)),
	    DEBUG_PROBE);
	ATADEBUG_PRINT(("\n"), DEBUG_PROBE);

}

static void
rdcide_setup_channel(struct ata_channel *chp)
{
	u_int8_t drive;
	u_int32_t patr, psd1atr, udccr, iiocr;
	struct pciide_channel *cp = CHAN_TO_PCHAN(chp);
	struct pciide_softc *sc = CHAN_TO_PCIIDE(chp);
	struct ata_drive_datas *drvp = cp->ata_channel.ch_drive;

	patr = pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PATR);
	psd1atr = pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_PSD1ATR);
	udccr = pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_UDCCR);
	iiocr = pci_conf_read(sc->sc_pc, sc->sc_tag, RDCIDE_IIOCR);

	/* setup DMA */
	pciide_channel_dma_setup(cp);

	/* clear modes */
	patr = patr & (RDCIDE_PATR_EN(0) | RDCIDE_PATR_EN(1));
	psd1atr &= ~RDCIDE_PSD1ATR_SETUP_MASK(chp->ch_channel);
	psd1atr &= ~RDCIDE_PSD1ATR_HOLD_MASK(chp->ch_channel);
	for (drive = 0; drive < 2; drive++) {
		udccr &= ~RDCIDE_UDCCR_EN(chp->ch_channel, drive);
		udccr &= ~RDCIDE_UDCCR_TIM_MASK(chp->ch_channel, drive);
		iiocr &= ~RDCIDE_IIOCR_CLK_MASK(chp->ch_channel, drive);
	}
	/* now setup modes */
	for (drive = 0; drive < 2; drive++) {
		if (drvp[drive].drive_type == ATA_DRIVET_NONE)
			continue;
		if (drvp[drive].drive_type == ATA_DRIVET_ATAPI)
			patr |= RDCIDE_PATR_ATA(chp->ch_channel, drive);
		if (drive == 0) {
			patr |= RDCIDE_PATR_SETUP(
			    rdcide_setup[drvp[drive].PIO_mode],
			    chp->ch_channel);
			patr |= RDCIDE_PATR_HOLD(
			    rdcide_hold[drvp[drive].PIO_mode],
			    chp->ch_channel);
		} else {
			patr |= RDCIDE_PATR_DEV1_TEN(chp->ch_channel);
			psd1atr |= RDCIDE_PSD1ATR_SETUP(
			    rdcide_setup[drvp[drive].PIO_mode],
			    chp->ch_channel);
			psd1atr |= RDCIDE_PSD1ATR_HOLD(
			    rdcide_hold[drvp[drive].PIO_mode],
			    chp->ch_channel);
		}
		if (drvp[drive].PIO_mode > 0) {
			patr |= RDCIDE_PATR_FTIM(chp->ch_channel, drive);
			patr |= RDCIDE_PATR_IORDY(chp->ch_channel, drive);
		}
		if (drvp[drive].drive_flags & ATA_DRIVE_DMA) {
			patr |= RDCIDE_PATR_DMAEN(chp->ch_channel, drive);
		}
		if ((drvp[drive].drive_flags & ATA_DRIVE_UDMA) == 0)
			continue;

		if ((iiocr & RDCIDE_IIOCR_CABLE(chp->ch_channel, drive)) == 0
		    && drvp[drive].UDMA_mode > 2)
			drvp[drive].UDMA_mode = 2;
		udccr |= RDCIDE_UDCCR_EN(chp->ch_channel, drive);
		udccr |= RDCIDE_UDCCR_TIM(
		    rdcide_udmatim[drvp[drive].UDMA_mode],
		    chp->ch_channel, drive);
		iiocr |= RDCIDE_IIOCR_CLK(
		    rdcide_udmaclk[drvp[drive].UDMA_mode],
		    chp->ch_channel, drive);
	}

	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_PATR, patr);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_PSD1ATR, psd1atr);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_UDCCR, udccr);
	pci_conf_write(sc->sc_pc, sc->sc_tag, RDCIDE_IIOCR, iiocr);
}