/* $NetBSD: jh7110_pcie.c,v 1.2 2025/01/09 10:39:01 skrll Exp $ */

/*-
 * Copyright (c) 2024 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Nick Hudson
 *
 * 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 NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``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 FOUNDATION OR CONTRIBUTORS
 * 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: jh7110_pcie.c,v 1.2 2025/01/09 10:39:01 skrll Exp $");

#include <sys/param.h>

#include <sys/bitops.h>
#include <sys/kmem.h>

#include <dev/fdt/fdtvar.h>
#include <dev/fdt/syscon.h>

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

#include <riscv/fdt/pcihost_fdtvar.h>

struct jh7110_pcie_irq {
	struct jh7110_pcie_softc *
				jpi_sc;
	void			*jpi_arg;
	int			(*jpi_fn)(void *);
	int			jpi_mpsafe;
};

struct jh7110_pcie_softc {
	bus_space_tag_t		sc_bst;

	struct pcihost_softc	sc_phsc;

	bus_space_handle_t	sc_apb_bsh;
	bus_addr_t		sc_apb_addr;
	bus_size_t		sc_apb_size;

	bus_space_handle_t	sc_cfg_bsh;
	bus_addr_t		sc_cfg_addr;
	bus_size_t		sc_cfg_size;

	// syscon
	const struct syscon *	sc_syscon;
	bus_size_t		sc_stg_base;

	struct fdtbus_gpio_pin *sc_perst_gpio;

	// # pins
	struct jh7110_pcie_irq	*sc_irq[PCI_INTERRUPT_PIN_MAX];
};

#define	RD4(sc, reg)							      \
    bus_space_read_4((sc)->sc_bst, (sc)->sc_apb_bsh, (reg))
#define	WR4(sc, reg, val)						      \
    bus_space_write_4((sc)->sc_bst, (sc)->sc_apb_bsh, (reg), (val))

#define SET4(sc, off, mask)						      \
    WR4((sc), (off), RD4((sc), (off)) | (mask))
#define CLR4(sc, off, mask)						      \
    WR4((sc), (off), RD4((sc), (off)) & ~(mask))
#define UPD4(sc, off, clr, set)						      \
    WR4((sc), (off), (RD4((sc), (off)) & ~(clr)) | (set))

#define JH7110_PCIE0_CFG_BASE	0x940000000UL
#define JH7110_PCIE1_CFG_BASE	0x9c0000000UL

/* PLDA register definitions */
#define PLDA_GEN_SETTINGS			0x80
#define  PLDA_GEN_RP_ENABLE			1
#define PLDA_PCI_IDS				0x9c
#define  PLDA_PCI_IDS_REVISION_MASK		__BITS( 7, 0)
#define  PLDA_PCI_IDS_CLASSCODE_MASK		__BITS(31, 8)
#define PLDA_MISC				0xb4
#define  PLDA_MISC_PHYFUNC_DISABLE		__BIT(15)
#define PLDA_WINROM				0xfc
#define  PLDA_WINROM_PREF64SUPPORT		__BIT(3)

#define PLDA_IMASK_LOCAL			0x180
#define  PLDA_IMASK_INT_INTA			__BIT(24)
#define  PLDA_IMASK_INT_INTB			__BIT(25)
#define  PLDA_IMASK_INT_INTC			__BIT(26)
#define  PLDA_IMASK_INT_INTD			__BIT(27)
#define  PLDA_IMASK_INT_INTX			__BITS(27, 24)
#define  PLDA_IMASK_INT_MSI			__BIT(28)
#define PLDA_ISTATUS_LOCAL			0x184
#define  PLDA_ISTATUS_INT_INTA			__BIT(24)
#define  PLDA_ISTATUS_INT_INTB			__BIT(25)
#define  PLDA_ISTATUS_INT_INTC			__BIT(26)
#define  PLDA_ISTATUS_INT_INTD			__BIT(27)
#define  PLDA_ISTATUS_INT_INTX			__BITS(27, 24)
#define  PLDA_ISTATUS_INT_MSI			__BIT(28)
#define PLDA_IMASK_HOST				0x188
#define PLDA_ISTATUS_HOST			0x18c
#define PLDA_IMSI_ADDR				0x190
#define PLDA_ISTATUS_MSI			0x194
#define PLDA_PMSG_SUPPORT_RX			0x3f0
#define  PLDA_PMSG_LTR_SUPPORT			__BIT(2)

/* PCIe Master table init defines */
#define PLDA_ATR0_PCIE_WIN0_SRCADDR_PARAM	0x600
#define  PLDA_ATR0_PCIE_ATR_SIZE		0x25
#define  PLDA_ATR0_PCIE_ATR_SIZE_SHIFT		1
#define PLDA_ATR0_PCIE_WIN0_SRC_ADDR		0x604
#define PLDA_ATR0_PCIE_WIN0_TRSL_ADDR_LSB	0x608
#define PLDA_ATR0_PCIE_WIN0_TRSL_ADDR_UDW	0x60c
#define PLDA_ATR0_PCIE_WIN0_TRSL_PARAM		0x610

#define PLDA_ATR_AXI4_SLV0_SRC_ADDR_LO(n)	(0x800 + (n) * 0x20)
#define  PLDA_ATR_SIZE_SRC_MASK			__BITS(31, 12)
#define  PLDA_ATR_SIZE_MASK			__BITS(6, 1)
#define  PLDA_ATR_IMPL				__BIT(0)
#define PLDA_ATR_AXI4_SLV0_SRC_ADDR_HI(n)	(0x804 + (n) * 0x20)
#define PLDA_ATR_AXI4_SLV0_TRSL_ADDR_LO(n)	(0x808 + (n) * 0x20)
#define PLDA_ATR_AXI4_SLV0_TRSL_ADDR_HI(n)	(0x80c + (n) * 0x20)
#define PLDA_ATR_AXI4_SLV0_TRSL_PARAM(n)	(0x810 + (n) * 0x20)
#define  PLDA_TRSL_ID_PCIE_RX_TX		0
#define  PLDA_TRSL_ID_PCIE_CONFIG		1

#define PCIE_FUNC_NUM				4

/* system control */
#define STG_SYSCON_PCIE0_BASE			0x0048
#define STG_SYSCON_PCIE1_BASE			0x01f8

#define STG_SYSCON_AR_OFFSET			0x0078
#define STG_SYSCON_AXI4_SLVL_AR_MASK		__BITS(22, 8)
#define  STG_SYSCON_AXI4_SLVL_PHY_AR_MASK	__BITS(20,17)
#define  STG_SYSCON_AXI4_SLVL_PHY_AR(x)		\
    __SHIFTIN((x), STG_SYSCON_AXI4_SLVL_PHY_AR_MASK)

#define STG_SYSCON_AW_OFFSET			0x007c
#define STG_SYSCON_CLKREQ			__BIT(22)
#define STG_SYSCON_CKREF_SRC_MASK		__BITS(19, 18)
#define STG_SYSCON_AXI4_SLVL_AW_MASK		__BITS(14,  0)
#define  STG_SYSCON_AXI4_SLVL_PHY_AW_MASK	__BITS(12,  9)
#define  STG_SYSCON_AXI4_SLVL_PHY_AW(x)		\
    __SHIFTIN((x), STG_SYSCON_AXI4_SLVL_PHY_AW_MASK)

#define STG_SYSCON_RP_NEP_OFFSET		0x00e8
#define STG_SYSCON_K_RP_NEP			__BIT(8)

#define STG_SYSCON_LNKSTA_OFFSET		0x0170
#define DATA_LINK_ACTIVE			__BIT(5)

#define ECAM_BUS_MASK				__BITS(27, 20)
#define ECAM_DEV_MASK				__BITS(19, 15)
#define ECAM_FUNC_MASK				__BITS(14, 12)
#define ECAM_OFFSET_MASK			__BITS(11,  0)

static int
jh7110_pcie_bus_maxdevs(void *v, int bus)
{
	struct pcihost_softc * const phsc = v;

	if (bus >= phsc->sc_bus_min || bus <= phsc->sc_bus_max)
		return 1;
	return 0;
}

static pcitag_t
jh7110_pcie_make_tag(void *v, int bus, int dev, int fn)
{
//	struct pcihost_softc * const phsc = v;

	/* Return ECAM address. */
	return
	    __SHIFTIN(bus, ECAM_BUS_MASK) |
	    __SHIFTIN(dev, ECAM_DEV_MASK) |
	    __SHIFTIN(fn, ECAM_FUNC_MASK) |
	    0;
}

static void
jh7110_pcie_decompose_tag(void *v, pcitag_t tag,
    int *busp, int *devp, int *fnp)
{
//	struct pcihost_softc * const phsc = v;

	if (busp != NULL)
		*busp = __SHIFTOUT(tag, ECAM_BUS_MASK);
	if (devp != NULL)
		*devp = __SHIFTOUT(tag, ECAM_DEV_MASK);
	if (fnp != NULL)
		*fnp = __SHIFTOUT(tag, ECAM_FUNC_MASK);
}

static bool
jh7110_pcie_conf_ok(struct jh7110_pcie_softc *sc,
    int bus, int dev, int fn, int offset)
{

	/* Only one device on root port and the first subordinate port. */
	if (bus < 2 && dev < 1)
		return true;

	return false;
}

static pcireg_t
jh7110_pcie_conf_read(void *v, pcitag_t tag, int offset)
{
	struct pcihost_softc * const phsc = v;
	struct jh7110_pcie_softc * const sc =
	    container_of(phsc, struct jh7110_pcie_softc, sc_phsc);
	int bus, dev, fn;

	KASSERT(offset >= 0);
	KASSERT(offset < PCI_EXTCONF_SIZE);

	jh7110_pcie_decompose_tag(phsc, tag, &bus, &dev, &fn);

	if (!jh7110_pcie_conf_ok(sc, bus, dev, fn, offset))
		return 0xffffffff;

	bus_size_t reg =
	    __SHIFTIN(bus, ECAM_BUS_MASK) |
	    __SHIFTIN(dev, ECAM_DEV_MASK) |
	    __SHIFTIN(fn, ECAM_FUNC_MASK) |
	    offset;

	return bus_space_read_4(sc->sc_bst, sc->sc_cfg_bsh, reg);
}

static void
jh7110_pcie_conf_write(void *v, pcitag_t tag, int offset, pcireg_t data)
{
	struct pcihost_softc * const phsc = v;
	struct jh7110_pcie_softc * const sc =
	    container_of(phsc, struct jh7110_pcie_softc, sc_phsc);
	int bus, dev, fn;

	KASSERT(offset >= 0);
	KASSERT(offset < PCI_EXTCONF_SIZE);

	jh7110_pcie_decompose_tag(phsc, tag, &bus, &dev, &fn);

	if (!jh7110_pcie_conf_ok(sc, bus, dev, fn, offset))
		return;

	bus_size_t reg =
	    __SHIFTIN(bus, ECAM_BUS_MASK) |
	    __SHIFTIN(dev, ECAM_DEV_MASK) |
	    __SHIFTIN(fn, ECAM_FUNC_MASK) |
	    offset;

	bus_space_write_4(sc->sc_bst, sc->sc_cfg_bsh, reg, data);
}

/* INTx interrupt controller */
static void *
jh7110_pcie_intx_establish(device_t dev, u_int *specifier, int ipl, int flags,
    int (*func)(void *), void *arg, const char *xname)
{
	struct jh7110_pcie_softc * const sc = device_private(dev);
	const u_int mpsafe = (flags & FDT_INTR_MPSAFE) ? IST_MPSAFE : 0;
	const u_int pin = be32toh(specifier[0]) - 1;

	KASSERT((RD4(sc, PLDA_IMASK_LOCAL) & (PLDA_IMASK_INT_INTA << pin)) == 0);

	struct jh7110_pcie_irq *jpi = sc->sc_irq[pin];
	if (jpi == NULL) {
		jpi = kmem_alloc(sizeof(*jpi), KM_SLEEP);
		jpi->jpi_sc = sc;
		jpi->jpi_fn = func;
		jpi->jpi_arg = arg;
		jpi->jpi_mpsafe = mpsafe;

		sc->sc_irq[pin] = jpi;
	} else {
		device_printf(dev, "shared interrupts not supported\n");
		return NULL;
	}

	/* Unmask the interrupt. */
	SET4(sc, PLDA_IMASK_LOCAL, (PLDA_IMASK_INT_INTA << pin));

	return jpi;
}


static void
jh7110_pcie_intx_disestablish(device_t dev, void *ih)
{
	struct jh7110_pcie_softc * const sc = device_private(dev);
	struct pcihost_softc * const phsc = &sc->sc_phsc;

	device_printf(dev, "%s\n", __func__);

	fdtbus_intr_disestablish(phsc->sc_phandle, ih);
}

static bool
jh7110_pcie_intx_string(device_t dev, u_int *specifier, char *buf,
    size_t buflen)
{
	struct jh7110_pcie_softc * const sc = device_private(dev);
	struct pcihost_softc * const phsc = &sc->sc_phsc;

	fdtbus_intr_str(phsc->sc_phandle, 0, buf, buflen);

	return true;
}

static int
jh7110_pcie_intx_intr(struct jh7110_pcie_softc *sc, uint32_t status)
{
	int handled = 0;
	u_int pin;

	CTASSERT(__arraycount(sc->sc_irq) == 4);
	for (pin = 0; pin < __arraycount(sc->sc_irq); pin++) {
		if ((status & (PLDA_IMASK_INT_INTA << pin)) == 0)
			continue;

		struct jh7110_pcie_irq *jpi = sc->sc_irq[pin];

		if (jpi == NULL)
			continue;

		if (!jpi->jpi_mpsafe)
			KERNEL_LOCK(1, NULL);
		handled |= jpi->jpi_fn(jpi->jpi_arg);
		if (!jpi->jpi_mpsafe)
			KERNEL_UNLOCK_ONE(NULL);
	}

	return handled;
}


static int
jh7110_pcie_intr(void *v)
{
	struct jh7110_pcie_softc * const sc = v;
	int handled = 0;

	uint32_t status = RD4(sc, PLDA_ISTATUS_LOCAL);
	if (status == 0)
		return 0;

	if (status & PLDA_ISTATUS_INT_INTX)
		handled |= jh7110_pcie_intx_intr(sc, status);

	WR4(sc, PLDA_ISTATUS_LOCAL, status);

	return handled;
}

static struct fdtbus_interrupt_controller_func jh7110_pcie_intxfuncs = {
	.establish = jh7110_pcie_intx_establish,
	.disestablish = jh7110_pcie_intx_disestablish,
	.intrstr = jh7110_pcie_intx_string,
};

#define SCRD4(sc, off)		syscon_read_4((sc), (off))
#define SCWR4(sc, off, val)	syscon_write_4((sc), (off), (val))

#define SCSET4(sc, off, mask)						      \
    SCWR4((sc), (off), SCRD4((sc), (off)) | (mask))
#define SCCLR4(sc, off, mask)						      \
    SCWR4((sc), (off), SCRD4((sc), (off)) & ~(mask))
#define SCUPD4(sc, off, clr, set)					      \
    SCWR4((sc), (off), (SCRD4((sc), (off)) & ~(clr)) | (set))


static int
jh7110_pcie_host_init(struct jh7110_pcie_softc *sc)
{
	struct pcihost_softc * const phsc = &sc->sc_phsc;

	syscon_lock(sc->sc_syscon);
	SCSET4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_RP_NEP_OFFSET,
	    STG_SYSCON_K_RP_NEP);

	SCUPD4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AW_OFFSET,
	    STG_SYSCON_CKREF_SRC_MASK,
	    __SHIFTIN(2, STG_SYSCON_CKREF_SRC_MASK));

	SCSET4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AW_OFFSET,
	    STG_SYSCON_CLKREQ);

	/* enable clocks */
	struct clk *clk;
	fdtbus_clock_assign(phsc->sc_phandle);
	for (u_int c = 0;
	    (clk = fdtbus_clock_get_index(phsc->sc_phandle, c)) != NULL;
	    c++) {
		if (clk_enable(clk) != 0) {
			aprint_error_dev(phsc->sc_dev,
			    "couldn't enable clock #%d\n", c);
			return ENXIO;
		}
	}
	/* de-assert resets */
	struct fdtbus_reset *rst;
	for (u_int r = 0;
	    (rst = fdtbus_reset_get_index(phsc->sc_phandle, r)) != NULL;
	    r++) {
		if (fdtbus_reset_deassert(rst) != 0) {
			aprint_error_dev(phsc->sc_dev,
			    "couldn't de-assert reset #%d\n", r);
			return ENXIO;
		}
	}

	fdtbus_gpio_write(sc->sc_perst_gpio, 1);

	/* Disable additional functions. */
	for (u_int i = 1; i < PCIE_FUNC_NUM; i++) {

		SCUPD4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AR_OFFSET,
		    STG_SYSCON_AXI4_SLVL_AR_MASK,
		    __SHIFTIN(i, STG_SYSCON_AXI4_SLVL_PHY_AR_MASK));

		SCUPD4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AW_OFFSET,
		    STG_SYSCON_AXI4_SLVL_AW_MASK,
		    __SHIFTIN(i, STG_SYSCON_AXI4_SLVL_PHY_AW_MASK));

		SET4(sc, PLDA_MISC, PLDA_MISC_PHYFUNC_DISABLE);
	}

	SCCLR4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AR_OFFSET,
	    STG_SYSCON_AXI4_SLVL_AR_MASK);

	SCCLR4(sc->sc_syscon, sc->sc_stg_base + STG_SYSCON_AW_OFFSET,
	    STG_SYSCON_AXI4_SLVL_AW_MASK);

	/* Configure controller as root port. */
	UPD4(sc, PLDA_GEN_SETTINGS, PLDA_GEN_RP_ENABLE, PLDA_GEN_RP_ENABLE);
	uint64_t base_addr = 0;

#define CONFIG_SPACE_ADDR_OFFSET 0x1000

	SET4(sc, CONFIG_SPACE_ADDR_OFFSET + 0x10, BUS_ADDR_LO32(base_addr));
	SET4(sc, CONFIG_SPACE_ADDR_OFFSET + 0x14, BUS_ADDR_HI32(base_addr));

	/* Configure as PCI bridge. */
	UPD4(sc, PLDA_PCI_IDS,
	    PLDA_PCI_IDS_CLASSCODE_MASK,
	    PCI_CLASS_CODE(PCI_CLASS_BRIDGE,
		PCI_SUBCLASS_BRIDGE_PCI,
		PCI_INTERFACE_BRIDGE_PCI_PCI));

	/* Enable prefetchable memory windows. */
	SET4(sc, PLDA_WINROM, PLDA_WINROM_PREF64SUPPORT);

	/* Disable LTR message forwarding. */
	CLR4(sc, PLDA_PMSG_SUPPORT_RX, PLDA_PMSG_LTR_SUPPORT);

	/*
	 * PERST# must remain asserted for at least 100us after the
	 * reference clock becomes stable.  But also has to remain
	 * active at least 100ms after power up.  Since we may have
	 * just powered on the device, play it safe and use 100ms.
	 */
	delay(100 * 1000);

	/* Deassert PERST#. */
	fdtbus_gpio_write(sc->sc_perst_gpio, 0);

	/* Wait for link to come up. */
	uint32_t reg;
	for (int timo = 100; timo > 0; timo--) {
		reg = SCRD4(sc->sc_syscon,
		    sc->sc_stg_base + STG_SYSCON_LNKSTA_OFFSET);
		if (reg & DATA_LINK_ACTIVE)
			break;
		delay(1000);
	}

	syscon_unlock(sc->sc_syscon);

	if ((reg & DATA_LINK_ACTIVE) == 0) {
		aprint_error_dev(phsc->sc_dev, "link not up\n");
		    return ENXIO;
	}

	return 0;
}

static int
jh7110_pcie_atr_init(struct jh7110_pcie_softc *sc)
{
	const uint32_t *ranges;
	bus_addr_t phyaddr;	// PADDR?
	bus_addr_t pciaddr;
	bus_size_t size;

	WR4(sc, PLDA_ATR_AXI4_SLV0_SRC_ADDR_LO(0),
	    PLDA_ATR_IMPL |
	    __SHIFTIN(ilog2(sc->sc_cfg_size) - 1, PLDA_ATR_SIZE_MASK) |
	    BUS_ADDR_LO32(sc->sc_cfg_addr));
	WR4(sc, PLDA_ATR_AXI4_SLV0_SRC_ADDR_HI(0), BUS_ADDR_HI32(sc->sc_cfg_addr));
	WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_ADDR_LO(0), 0);
	WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_ADDR_HI(0), 0);
	WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_PARAM(0), PLDA_TRSL_ID_PCIE_CONFIG);

	struct pcihost_softc * const phsc = &sc->sc_phsc;
	int ranges_len;
	ranges = fdtbus_get_prop(phsc->sc_phandle, "ranges", &ranges_len);
	if (ranges == NULL) {
		aprint_error_dev(phsc->sc_dev,
		    "couldn't find 'ranges' property\n");
		return ENXIO;
	}
	const int ranges_cells = ranges_len / sizeof(uint32_t);

	for (u_int i = 0, n = 1; i < ranges_cells && n < 8; i += 7, n++) {
		pciaddr =
		    __SHIFTIN(be32toh(ranges[i + 1]), __BITS(63, 32)) |
		    __SHIFTIN(be32toh(ranges[i + 2]), __BITS(31,  0));
		phyaddr =
		    __SHIFTIN(be32toh(ranges[i + 3]), __BITS(63, 32)) |
		    __SHIFTIN(be32toh(ranges[i + 4]), __BITS(31,  0));
		size = be32toh(ranges[i + 6]);

		WR4(sc, PLDA_ATR_AXI4_SLV0_SRC_ADDR_LO(n),
		    PLDA_ATR_IMPL |
		    __SHIFTIN(ilog2(size) /* - 1 */, PLDA_ATR_SIZE_MASK) |
		    BUS_ADDR_LO32(phyaddr));
		WR4(sc, PLDA_ATR_AXI4_SLV0_SRC_ADDR_HI(n), BUS_ADDR_HI32(phyaddr));
		WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_ADDR_LO(n), BUS_ADDR_LO32(pciaddr));
		WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_ADDR_HI(n), BUS_ADDR_HI32(pciaddr));
		WR4(sc, PLDA_ATR_AXI4_SLV0_TRSL_PARAM(n), PLDA_TRSL_ID_PCIE_RX_TX);
	}

	uint32_t val;
	val = RD4(sc, PLDA_ATR0_PCIE_WIN0_SRCADDR_PARAM);
	val |= (PLDA_ATR0_PCIE_ATR_SIZE << PLDA_ATR0_PCIE_ATR_SIZE_SHIFT);
	WR4(sc, PLDA_ATR0_PCIE_WIN0_SRCADDR_PARAM, val);
	WR4(sc, PLDA_ATR0_PCIE_WIN0_SRC_ADDR, 0);

	return 0;
}

/* Compat string(s) */
static const struct device_compatible_entry compat_data[] = {
	{ .compat = "starfive,jh7110-pcie" },
	DEVICE_COMPAT_EOL
};


static int
jh7110_pcie_match(device_t parent, cfdata_t cf, void *aux)
{
	struct fdt_attach_args * const faa = aux;

	return of_compatible_match(faa->faa_phandle, compat_data);
}

static void
jh7110_pcie_attach(device_t parent, device_t self, void *aux)
{
	struct jh7110_pcie_softc * const sc = device_private(self);
	struct pcihost_softc * const phsc = &sc->sc_phsc;
	struct fdt_attach_args * const faa = aux;
	const int phandle = faa->faa_phandle;
	int error;

	sc->sc_bst = faa->faa_bst;

	phsc->sc_dev = self;
	phsc->sc_bst = faa->faa_bst;
	phsc->sc_pci_bst = faa->faa_bst;
	phsc->sc_dmat = faa->faa_dmat;
	phsc->sc_phandle = faa->faa_phandle;

	/* Handle old and new binding names. */
	if (fdtbus_get_reg_byname(phandle, "apb",
	    &sc->sc_apb_addr, &sc->sc_apb_size) != 0) {
		aprint_error(": couldn't get apb registers\n");
		return;
	}
	if (fdtbus_get_reg_byname(phandle, "cfg",
	    &sc->sc_cfg_addr, &sc->sc_cfg_size) != 0) {
		aprint_error(": couldn't get cfg registers\n");
		return;
	}

	const int mapflags = 0 /*BUS_SPACE_MAP_NONPOSTED*/;
	error = bus_space_map(sc->sc_bst, sc->sc_apb_addr,
	    sc->sc_apb_size, mapflags, &sc->sc_apb_bsh);
	if (error) {
		aprint_error(": can't map APB registers\n");
		return;
	}
	error = bus_space_map(sc->sc_bst, sc->sc_cfg_addr,
	    sc->sc_cfg_size, mapflags, &sc->sc_cfg_bsh);
	if (error) {
		aprint_error(": can't map CFG registers\n");
		return;
	}
	int len;
	const char *stgsyscon = "starfive,stg-syscon";
	const u_int *stgsyscon_data =
	    fdtbus_get_prop(phandle, stgsyscon, &len);
	if (stgsyscon_data == NULL) {
		aprint_error(": couldn't get '%s' property\n", stgsyscon);
		return;
	}

	if (len != 1 * sizeof(uint32_t)) {
		aprint_error(": incorrect '%s' data (len = %u)\n", stgsyscon,
		    len);
		return;
	}
	int syscon_phandle =
	    fdtbus_get_phandle_from_native(be32dec(&stgsyscon_data[0]));

	sc->sc_syscon = fdtbus_syscon_lookup(syscon_phandle);
	if (sc->sc_syscon == NULL) {
		aprint_error(": couldn't get '%s' (%d)\n", stgsyscon,
		    syscon_phandle);
		return;
	}

	sc->sc_perst_gpio = fdtbus_gpio_acquire(phandle,
	    "perst-gpios", GPIO_PIN_OUTPUT);
	if (sc->sc_perst_gpio == NULL) {
		aprint_error(": couldn't get 'perst-gpios'");
		return;
	}

	switch (sc->sc_cfg_addr) {
	case JH7110_PCIE0_CFG_BASE:
		sc->sc_stg_base = STG_SYSCON_PCIE0_BASE;
		break;
	case JH7110_PCIE1_CFG_BASE:
		sc->sc_stg_base = STG_SYSCON_PCIE1_BASE;
		break;
	default:
		aprint_error(": unknown controller at 0x%lx\n",
		    sc->sc_cfg_addr);
		return;
	}

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

	error = jh7110_pcie_host_init(sc);
	if (error) {
		/* error already printed */
		return;
	}

	/* Configure Address Translation. */
	error = jh7110_pcie_atr_init(sc);
	if (error) {

	}

	/* Mask and acknowledge all interrupts. */
	WR4(sc, PLDA_IMASK_LOCAL, 0);
	WR4(sc, PLDA_ISTATUS_LOCAL, 0xffffffff);

	char intrstr[128];
	if (!fdtbus_intr_str(phandle, 0, intrstr, sizeof(intrstr))) {
		aprint_error(": failed to decode interrupt\n");
		return;
	}
	void *ih = fdtbus_intr_establish_xname(phandle, 0, IPL_VM,
	    FDT_INTR_MPSAFE, jh7110_pcie_intr, sc, device_xname(self));
	if (ih == NULL) {
		aprint_error_dev(self, "failed to establish interrupt on %s\n",
		    intrstr);
		// XXXNH unwind
		return;
	}
	aprint_normal_dev(self, "interrupting on %s\n", intrstr);

	fdtbus_register_interrupt_controller(self,
	    OF_child(phsc->sc_phandle), &jh7110_pcie_intxfuncs);

	phsc->sc_type = PCIHOST_ECAM;
	pcihost_init(&phsc->sc_pc, phsc);

	phsc->sc_pc.pc_bus_maxdevs = jh7110_pcie_bus_maxdevs;
	phsc->sc_pc.pc_make_tag = jh7110_pcie_make_tag;
	phsc->sc_pc.pc_decompose_tag = jh7110_pcie_decompose_tag;
	phsc->sc_pc.pc_conf_read = jh7110_pcie_conf_read;
	phsc->sc_pc.pc_conf_write = jh7110_pcie_conf_write;

	pcihost_init2(phsc);
}

CFATTACH_DECL_NEW(jh7110_pcie, sizeof(struct jh7110_pcie_softc),
	jh7110_pcie_match, jh7110_pcie_attach, NULL, NULL);