/*      $NetBSD: pci_bus_fixup.c,v 1.3 2019/03/01 09:25:59 msaitoh Exp $        */

/*
 * Copyright (c) 1999, by UCHIYAMA Yasushi
 * 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. The name of the developer 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 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 AUTHOR 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. 
 */

/*
 * PCI bus renumbering support.
 */

#include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: pci_bus_fixup.c,v 1.3 2019/03/01 09:25:59 msaitoh Exp $");

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

#include <sys/bus.h>

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

#include <x86/pci/pci_bus_fixup.h>

/* this array lists the parent for each bus number */
int pci_bus_parent[256];

/* this array lists the pcitag to program each bridge */
pcitag_t pci_bus_tag[256];

static void pci_bridge_reset(pci_chipset_tag_t, pcitag_t, void *);

int
pci_bus_fixup(pci_chipset_tag_t pc, int bus)
{
        static int bus_total;
        static int bridge_cnt;
        int device, maxdevs, function, nfuncs, bridge, bus_max, bus_sub;
        const struct pci_quirkdata *qd;
        pcireg_t reg;
        pcitag_t tag;

        bus_max = bus;
        bus_sub = 0;

        if (++bus_total > 256)
                panic("pci_bus_fixup: more than 256 PCI busses?");

        /* Reset bridge configuration on this bus */
        pci_bridge_foreach(pc, bus, bus, pci_bridge_reset, 0);

        maxdevs = pci_bus_maxdevs(pc, bus);

        for (device = 0; device < maxdevs; device++) {
                tag = pci_make_tag(pc, bus, device, 0);
                reg = pci_conf_read(pc, tag, PCI_ID_REG);

                /* Invalid vendor ID value? */
                if (PCI_VENDOR(reg) == PCI_VENDOR_INVALID)
                        continue;
                /* XXX Not invalid, but we've done this ~forever. */
                if (PCI_VENDOR(reg) == 0)
                        continue;

                qd = pci_lookup_quirkdata(PCI_VENDOR(reg), PCI_PRODUCT(reg));

                reg = pci_conf_read(pc, tag, PCI_BHLC_REG);
                if (PCI_HDRTYPE_MULTIFN(reg) ||
                    (qd != NULL &&
                     (qd->quirks & PCI_QUIRK_MULTIFUNCTION) != 0))
                        nfuncs = 8;
                else
                        nfuncs = 1;

                for (function = 0; function < nfuncs; function++) {
                        tag = pci_make_tag(pc, bus, device, function);
                        reg = pci_conf_read(pc, tag, PCI_ID_REG);

                        /* Invalid vendor ID value? */
                        if (PCI_VENDOR(reg) == PCI_VENDOR_INVALID)
                                continue;
                        /* XXX Not invalid, but we've done this ~forever. */
                        if (PCI_VENDOR(reg) == 0)
                                continue;

                        aprint_debug("PCI fixup examining %02x:%02x\n",
                               PCI_VENDOR(reg), PCI_PRODUCT(reg));

                        reg = pci_conf_read(pc, tag, PCI_CLASS_REG);
                        if (PCI_CLASS(reg) == PCI_CLASS_BRIDGE &&
                            (PCI_SUBCLASS(reg) == PCI_SUBCLASS_BRIDGE_PCI ||
                             PCI_SUBCLASS(reg) == PCI_SUBCLASS_BRIDGE_CARDBUS)) {
                                /* Assign the bridge #. */
                                bridge = bridge_cnt++;

                                /* Assign the bridge's secondary bus #. */
                                bus_max++;

                                reg = pci_conf_read(pc, tag,
                                    PCI_BRIDGE_BUS_REG);
                                reg &= 0xff000000;
                                reg |= bus | (bus_max << 8) | (0xff << 16);
                                pci_conf_write(pc, tag, PCI_BRIDGE_BUS_REG,
                                    reg);

                                /* Scan subordinate bus. */
                                bus_sub = pci_bus_fixup(pc, bus_max);

                                /* Configure the bridge. */
                                reg &= 0xff000000;
                                reg |= bus | (bus_max << 8) | (bus_sub << 16);
                                pci_conf_write(pc, tag, PCI_BRIDGE_BUS_REG,
                                    reg);

                                /* record relationship */
                                pci_bus_parent[bus_max]=bus;

                                pci_bus_tag[bus_max]=tag;

                                aprint_debug("PCI bridge %d: primary %d, "
                                    "secondary %d, subordinate %d\n",
                                    bridge, bus, bus_max, bus_sub);

                                /* Next bridge's secondary bus #. */
                                bus_max = (bus_sub > bus_max) ?
                                    bus_sub : bus_max;
                        }
                }
        }

        return (bus_max);       /* last # of subordinate bus */
}

/* Reset bus-bridge configuration */
void
pci_bridge_reset(pci_chipset_tag_t pc, pcitag_t tag, void *ctx)
{
        pcireg_t reg;

        reg = pci_conf_read(pc, tag, PCI_BRIDGE_BUS_REG);
        reg &= 0xff000000;
        reg |= 0x00ffffff;      /* max bus # */
        pci_conf_write(pc, tag, PCI_BRIDGE_BUS_REG, reg);
}