/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */
/*
 * Copyright 2008 Sun Microsystems, Inc.  All rights reserved.
 * Use is subject to license terms.
 * Copyright 2022 Oxide Computer Company
 */

/*
 * Determine the PCI configuration mechanism recommended by the BIOS.
 */

#include <sys/types.h>
#include <sys/sunddi.h>
#include <sys/pci_impl.h>
#include <sys/ddi_subrdefs.h>
#include <sys/bootconf.h>
#include <sys/psw.h>
#include <sys/modctl.h>
#include <sys/errno.h>
#include <sys/pci.h>
#include <sys/pci_cfgspace.h>
#include <sys/reboot.h>
#include <sys/pci_cfgspace_impl.h>
#include <sys/mutex.h>
#include <sys/plat/pci_prd.h>

extern int pci_boot_debug;
extern int pci_boot_maxbus;

/*
 * Interface routines
 */
void pci_enumerate(int);
void pci_setup_tree(void);
void pci_reprogram(void);
dev_info_t *pci_boot_bus_to_dip(uint32_t);

static struct modlmisc modlmisc = {
        &mod_miscops, "PCI BIOS interface"
};

static struct modlinkage modlinkage = {
        MODREV_1, (void *)&modlmisc, NULL
};

static pci_prd_upcalls_t pci_upcalls = {
        .pru_bus2dip_f = pci_boot_bus_to_dip
};

int
_init(void)
{
        int     err;

        if ((err = pci_prd_init(&pci_upcalls)) != 0) {
                return (err);
        }

        if ((err = mod_install(&modlinkage)) != 0) {
                pci_prd_fini();
                return (err);
        }

        impl_bus_add_probe(pci_enumerate);
        return (0);
}

int
_fini(void)
{
        int     err;

        if ((err = mod_remove(&modlinkage)) != 0)
                return (err);

        impl_bus_delete_probe(pci_enumerate);
        pci_prd_fini();
        return (0);
}

int
_info(struct modinfo *modinfop)
{
        return (mod_info(&modlinkage, modinfop));
}


/*
 * This function is invoked twice: first time, with reprogram=0 to
 * set up the PCI portion of the device tree. The second time is
 * for reprogramming devices not set up by the BIOS.
 */
void
pci_enumerate(int reprogram)
{
        extern void add_pci_fixes(void);
        extern void undo_pci_fixes(void);

        /*
         * On our first pass through here actually determine what the maximum
         * bus that we should use is.
         */
        if (reprogram == 0) {
                pci_boot_maxbus = pci_prd_max_bus();
        }

        add_pci_fixes();

        if (reprogram) {
                pci_reprogram();
                undo_pci_fixes();
                return;
        }

        /* setup device tree */
        pci_setup_tree();
        undo_pci_fixes();
}