// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2015 Broadcom Corporation
 */

#include <linux/kernel.h>
#include <linux/pci.h>
#include <linux/clk.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/interrupt.h>
#include <linux/platform_device.h>
#include <linux/of_address.h>
#include <linux/of_pci.h>
#include <linux/of_platform.h>
#include <linux/phy/phy.h>

#include "../pci.h"
#include "pcie-iproc.h"

static const struct of_device_id iproc_pcie_of_match_table[] = {
        {
                .compatible = "brcm,iproc-pcie",
                .data = (int *)IPROC_PCIE_PAXB,
        }, {
                .compatible = "brcm,iproc-pcie-paxb-v2",
                .data = (int *)IPROC_PCIE_PAXB_V2,
        }, {
                .compatible = "brcm,iproc-pcie-paxc",
                .data = (int *)IPROC_PCIE_PAXC,
        }, {
                .compatible = "brcm,iproc-pcie-paxc-v2",
                .data = (int *)IPROC_PCIE_PAXC_V2,
        },
        { /* sentinel */ }
};
MODULE_DEVICE_TABLE(of, iproc_pcie_of_match_table);

static int iproc_pltfm_pcie_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct iproc_pcie *pcie;
        struct device_node *np = dev->of_node;
        struct resource reg;
        struct pci_host_bridge *bridge;
        int ret;

        bridge = devm_pci_alloc_host_bridge(dev, sizeof(*pcie));
        if (!bridge)
                return -ENOMEM;

        pcie = pci_host_bridge_priv(bridge);

        pcie->dev = dev;
        pcie->type = (uintptr_t)of_device_get_match_data(dev);

        ret = of_address_to_resource(np, 0, &reg);
        if (ret < 0) {
                dev_err(dev, "unable to obtain controller resources\n");
                return ret;
        }

        pcie->base = devm_pci_remap_cfgspace(dev, reg.start,
                                             resource_size(&reg));
        if (!pcie->base) {
                dev_err(dev, "unable to map controller registers\n");
                return -ENOMEM;
        }
        pcie->base_addr = reg.start;

        if (of_property_read_bool(np, "brcm,pcie-ob")) {
                u32 val;

                ret = of_property_read_u32(np, "brcm,pcie-ob-axi-offset",
                                           &val);
                if (ret) {
                        dev_err(dev,
                                "missing brcm,pcie-ob-axi-offset property\n");
                        return ret;
                }
                pcie->ob.axi_offset = val;
                pcie->need_ob_cfg = true;
        }

        /*
         * DT nodes are not used by all platforms that use the iProc PCIe
         * core driver. For platforms that require explicit inbound mapping
         * configuration, "dma-ranges" would have been present in DT
         */
        pcie->need_ib_cfg = of_property_read_bool(np, "dma-ranges");

        /* PHY use is optional */
        pcie->phy = devm_phy_optional_get(dev, "pcie-phy");
        if (IS_ERR(pcie->phy))
                return PTR_ERR(pcie->phy);

        /* PAXC doesn't support legacy IRQs, skip mapping */
        switch (pcie->type) {
        case IPROC_PCIE_PAXC:
        case IPROC_PCIE_PAXC_V2:
                pcie->map_irq = NULL;
                break;
        default:
                break;
        }

        ret = iproc_pcie_setup(pcie, &bridge->windows);
        if (ret) {
                dev_err(dev, "PCIe controller setup failed\n");
                return ret;
        }

        platform_set_drvdata(pdev, pcie);
        return 0;
}

static void iproc_pltfm_pcie_remove(struct platform_device *pdev)
{
        struct iproc_pcie *pcie = platform_get_drvdata(pdev);

        iproc_pcie_remove(pcie);
}

static void iproc_pltfm_pcie_shutdown(struct platform_device *pdev)
{
        struct iproc_pcie *pcie = platform_get_drvdata(pdev);

        iproc_pcie_shutdown(pcie);
}

static struct platform_driver iproc_pltfm_pcie_driver = {
        .driver = {
                .name = "iproc-pcie",
                .of_match_table = of_match_ptr(iproc_pcie_of_match_table),
        },
        .probe = iproc_pltfm_pcie_probe,
        .remove = iproc_pltfm_pcie_remove,
        .shutdown = iproc_pltfm_pcie_shutdown,
};
module_platform_driver(iproc_pltfm_pcie_driver);

MODULE_AUTHOR("Ray Jui <rjui@broadcom.com>");
MODULE_DESCRIPTION("Broadcom iPROC PCIe platform driver");
MODULE_LICENSE("GPL v2");