/*-
 * Copyright 2004 by Peter Grehan. All rights reserved.
 *
 * 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.
 * 3. The name of the author may not be used to endorse or promote products
 *    derived from this software without specific prior written permission.
 *
 * 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>
__FBSDID("$FreeBSD: head/sys/powerpc/powermac/ata_kauai.c 175668 2008-01-26 05:11:09Z julian $");

/*
 * Mac 'Kauai' PCI ATA controller
 */
#include "opt_ata.h"
#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/module.h>
#include <sys/bus.h>
#include <sys/malloc.h>
#include <sys/sema.h>
#include <sys/taskqueue.h>
#include <vm/uma.h>
#include <machine/stdarg.h>
#include <machine/resource.h>
#include <machine/bus.h>
#include <sys/rman.h>
#include <sys/ata.h>
#include <dev/ata/ata-all.h>
#include <ata_if.h>

#include <dev/ofw/openfirm.h>
#include <powerpc/ofw/ofw_pci.h>

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

#include "ata_dbdma.h"

#define  ATA_KAUAI_REGOFFSET	0x2000
#define  ATA_KAUAI_DBDMAOFFSET	0x1000

/*
 * Offset to alt-control register from base
 */
#define  ATA_KAUAI_ALTOFFSET    (ATA_KAUAI_REGOFFSET + 0x160)

/*
 * Define the gap between registers
 */
#define ATA_KAUAI_REGGAP        16

/*
 * PIO and DMA access timing 
 */
#define PIO_CONFIG_REG (ATA_KAUAI_REGOFFSET + 0x200)
#define UDMA_CONFIG_REG (ATA_KAUAI_REGOFFSET + 0x210)

/*
 * Define the kauai pci bus attachment.
 */
static  int  ata_kauai_probe(device_t dev);
static  int  ata_kauai_attach(device_t dev);
static  void ata_kauai_setmode(device_t parent, device_t dev);

static device_method_t ata_kauai_methods[] = {
        /* Device interface */
	DEVMETHOD(device_probe,		ata_kauai_probe),
	DEVMETHOD(device_attach,	ata_kauai_attach),
	DEVMETHOD(device_detach,	bus_generic_detach),
	DEVMETHOD(device_shutdown,	bus_generic_shutdown),
	DEVMETHOD(device_suspend,	bus_generic_suspend),
	DEVMETHOD(device_resume,	bus_generic_resume),

	/* ATA interface */
	DEVMETHOD(ata_setmode,		ata_kauai_setmode),
	{ 0, 0 }
};

struct ata_kauai_softc {
	struct ata_dbdma_channel sc_ch;

	struct resource *sc_memr;

	int shasta;

	uint32_t dmaconf;
	uint32_t pioconf;
};

static driver_t ata_kauai_driver = {
	"ata",
	ata_kauai_methods,
	sizeof(struct ata_kauai_softc),
};

DRIVER_MODULE(ata, pci, ata_kauai_driver, ata_devclass, 0, 0);
MODULE_DEPEND(ata, ata, 1, 1, 1);

/*
 * PCI ID search table
 */
static struct kauai_pci_dev {
        u_int32_t  kpd_devid;
        char    *kpd_desc;
} kauai_pci_devlist[] = {
        { 0x0033106b, "Uninorth2 Kauai ATA Controller" },
        { 0x003b106b, "Intrepid Kauai ATA Controller" },
        { 0x0043106b, "K2 Kauai ATA Controller" },
        { 0x0050106b, "Shasta Kauai ATA Controller" },
        { 0x0069106b, "Intrepid-2 Kauai ATA Controller" },
        { 0, NULL }
};

/*
 * IDE transfer timings
 */
static const u_int pio_timing_kauai[] = {
	0x08000a92,	/* PIO0 */
	0x0800060f,	/* PIO1 */
	0x0800038b,	/* PIO2 */
	0x05000249,	/* PIO3 */
	0x04000148	/* PIO4 */
};
static const u_int pio_timing_shasta[] = {
	0x0a000c97,	/* PIO0 */
	0x07000712,	/* PIO1 */
	0x040003cd,	/* PIO2 */
	0x0400028b,	/* PIO3 */
	0x0400010a	/* PIO4 */
};

static const u_int dma_timing_kauai[] = {
        0x00618000,	/* WDMA0 */
        0x00209000,	/* WDMA1 */
        0x00148000	/* WDMA2 */
};
static const u_int dma_timing_shasta[] = {
        0x00820800,	/* WDMA0 */
        0x0028b000,	/* WDMA1 */
        0x001ca000	/* WDMA2 */
};

static const u_int udma_timing_kauai[] = {
        0x000070c1,	/* UDMA0 */
        0x00005d81,	/* UDMA1 */
        0x00004a61,	/* UDMA2 */
        0x00003a51,	/* UDMA3 */
        0x00002a31,	/* UDMA4 */
        0x00002921	/* UDMA5 */
};
static const u_int udma_timing_shasta[] = {
        0x00035901,	/* UDMA0 */
        0x000348b1,	/* UDMA1 */
        0x00033881,	/* UDMA2 */
        0x00033861,	/* UDMA3 */
        0x00033841,	/* UDMA4 */
        0x00033031,	/* UDMA5 */
        0x00033021	/* UDMA6 */
};

static int
ata_kauai_probe(device_t dev)
{
	struct ata_channel *ch;
	struct ata_kauai_softc *sc;
	u_long startp, countp;
	u_int32_t devid;
	phandle_t node;
	char *compatstring = NULL;
	int i, found, rid, status;

	found = 0;
	devid = pci_get_devid(dev);
        for (i = 0; kauai_pci_devlist[i].kpd_desc != NULL; i++) {
                if (devid == kauai_pci_devlist[i].kpd_devid) {
			found = 1;
                        device_set_desc(dev, kauai_pci_devlist[i].kpd_desc);
		}
	}

	if (!found)
		return (ENXIO);

	node = ofw_pci_find_node(dev);
	sc = device_get_softc(dev);
	bzero(sc, sizeof(struct ata_kauai_softc));
	ch = &sc->sc_ch.sc_ch;

	OF_getprop_alloc(node, "compatible", 1, (void **)&compatstring);
	if (strcmp(compatstring,"shasta-ata") == 0)
		sc->shasta = 1;

	free(compatstring, M_OFWPROP);


	/*
	 * This device seems to ignore writes to the interrupt
	 * config register, resulting in interrupt resources
	 * not being attached. If this is the case, use
	 * Open Firmware to determine the irq, and then attach
	 * the resource. This allows the ATA common code to
	 * allocate the irq.
	 */
	status = bus_get_resource(dev, SYS_RES_IRQ, 0, &startp, &countp);
	if (status == ENOENT) {
		int *irq;
		phandle_t iparent;
		int icells, nintr, i;

		/*
		 * Horrible hack to handle Kauai devices that have their IRQs
		 * set up in an utterly wrong way
		 */
		if (!sc->shasta)
			bus_set_resource(dev, SYS_RES_IRQ, 0, 39, 1);

		/*
		 * For the rest of the interrupts, and the main Shasta
		 * interrupt, get the IRQs from firmware.
		 */
		if (OF_getprop(node, "interrupt-parent", &iparent, 
		    sizeof(iparent)) == sizeof(iparent)) {
			OF_getprop(iparent, "#interrupt-cells", &icells, 
			    sizeof(icells)) ;
		}

		nintr = OF_getprop_alloc(node, "interrupts", sizeof(*irq),
		    (void **)&irq);

		for (i = 0; i < nintr; i += icells)
			bus_set_resource(dev, SYS_RES_IRQ, 
			    i/icells + !sc->shasta, irq[i], 1);

		free(irq, M_OFWPROP);
	}


        rid = PCIR_BARS;
	sc->sc_memr = bus_alloc_resource_any(dev, SYS_RES_MEMORY, &rid, 
	    RF_ACTIVE);
        if (sc->sc_memr == NULL) {
                device_printf(dev, "could not allocate memory\n");
                return (ENXIO);
        }

	/*
	 * Set up the resource vectors
	 */
        for (i = ATA_DATA; i <= ATA_COMMAND; i++) {
                ch->r_io[i].res = sc->sc_memr;
                ch->r_io[i].offset = i*ATA_KAUAI_REGGAP + ATA_KAUAI_REGOFFSET;
        }
        ch->r_io[ATA_CONTROL].res = sc->sc_memr;
        ch->r_io[ATA_CONTROL].offset = ATA_KAUAI_ALTOFFSET;
	ata_default_registers(dev);

        ch->unit = 0;
        ch->flags |= ATA_USE_16BIT;
	ata_generic_hw(dev);

        return (ata_probe(dev));
}

static int
ata_kauai_attach(device_t dev)
{
	struct ata_kauai_softc *sc = device_get_softc(dev);
#if KAUAI_UDMA
	int dbdma_irq_rid = 1;
	struct resource *dbdma_irq;
	void *cookie;
#endif

	pci_enable_busmaster(dev);

	/* Init DMA engine */

	sc->sc_ch.dbdma_rid = 1;
	sc->sc_ch.dbdma_regs = sc->sc_memr;
	sc->sc_ch.dbdma_offset = ATA_KAUAI_DBDMAOFFSET;

	ata_dbdma_dmainit(dev);

	/* Bind to DBDMA interrupt as well */
 
#if KAUAI_UDMA
	if ((dbdma_irq = bus_alloc_resource_any(dev, SYS_RES_IRQ,
	    &dbdma_irq_rid, RF_SHAREABLE | RF_ACTIVE)) != NULL) {
		bus_setup_intr(dev, dbdma_irq, ATA_INTR_FLAGS, NULL,
		    (driver_intr_t *)ata_interrupt, sc,&cookie);
	}
#endif

	/* Set up initial mode */
	if (sc->shasta) {
		sc->pioconf = pio_timing_shasta[4];
		sc->dmaconf = dma_timing_shasta[2];
	} else {
		sc->pioconf = pio_timing_kauai[4];
		sc->dmaconf = dma_timing_kauai[2];
	}

	bus_write_4(sc->sc_memr, PIO_CONFIG_REG, sc->pioconf | sc->dmaconf);

	return ata_attach(dev);
}

static void
ata_kauai_setmode(device_t parent, device_t dev)
{
	struct ata_device *atadev = device_get_softc(dev);
	struct ata_kauai_softc *sc = device_get_softc(parent);
	uint32_t mode;

	mode = atadev->mode = ata_limit_mode(dev,atadev->mode,
	    WDMA2);
#if KAUAI_UDMA
	    (sc->shasta) ? ATA_UDMA6 : ATA_UDMA5);
#endif
	
	if ((mode & ATA_DMA_MASK) == ATA_UDMA0) {
		uint32_t udmaconf = 0;

		if (sc->shasta)
			udmaconf = udma_timing_shasta[mode & ATA_MODE_MASK];
		else
			udmaconf = udma_timing_shasta[mode & ATA_MODE_MASK];

		bus_write_4(sc->sc_memr, UDMA_CONFIG_REG, udmaconf);
	} else if ((mode & ATA_DMA_MASK) == ATA_WDMA0) {
		if (sc->shasta)
			sc->dmaconf = dma_timing_shasta[mode & ATA_MODE_MASK];
		else
			sc->dmaconf = dma_timing_shasta[mode & ATA_MODE_MASK];

		bus_write_4(sc->sc_memr, PIO_CONFIG_REG, 
		    sc->dmaconf | sc->pioconf);
	} else {
		if (sc->shasta)
			sc->pioconf = pio_timing_shasta[(mode & ATA_MODE_MASK)
			    - ATA_PIO0];
		else
			sc->pioconf = pio_timing_kauai[(mode & ATA_MODE_MASK) - 
			    - ATA_PIO0];

		bus_write_4(sc->sc_memr, PIO_CONFIG_REG, 
		    sc->dmaconf | sc->pioconf);
	}
}