// SPDX-License-Identifier: GPL-2.0
/* Copyright (C) 2020 Maxime Ripard <maxime@cerno.tech> */

#include <linux/device.h>
#include <linux/dma-map-ops.h>
#include <linux/init.h>
#include <linux/notifier.h>
#include <linux/of.h>
#include <linux/platform_device.h>

static const char * const sunxi_mbus_devices[] = {
        /*
         * The display engine virtual devices are not strictly speaking
         * connected to the MBUS, but since DRM will perform all the
         * memory allocations and DMA operations through that device, we
         * need to have the quirk on those devices too.
         */
        "allwinner,sun4i-a10-display-engine",
        "allwinner,sun5i-a10s-display-engine",
        "allwinner,sun5i-a13-display-engine",
        "allwinner,sun6i-a31-display-engine",
        "allwinner,sun6i-a31s-display-engine",
        "allwinner,sun7i-a20-display-engine",
        "allwinner,sun8i-a23-display-engine",
        "allwinner,sun8i-a33-display-engine",
        "allwinner,sun9i-a80-display-engine",

        /*
         * And now we have the regular devices connected to the MBUS
         * (that we know of).
         */
        "allwinner,sun4i-a10-csi1",
        "allwinner,sun4i-a10-display-backend",
        "allwinner,sun4i-a10-display-frontend",
        "allwinner,sun4i-a10-video-engine",
        "allwinner,sun5i-a13-display-backend",
        "allwinner,sun5i-a13-video-engine",
        "allwinner,sun6i-a31-csi",
        "allwinner,sun6i-a31-display-backend",
        "allwinner,sun7i-a20-csi0",
        "allwinner,sun7i-a20-display-backend",
        "allwinner,sun7i-a20-display-frontend",
        "allwinner,sun7i-a20-video-engine",
        "allwinner,sun8i-a23-display-backend",
        "allwinner,sun8i-a23-display-frontend",
        "allwinner,sun8i-a33-display-backend",
        "allwinner,sun8i-a33-display-frontend",
        "allwinner,sun8i-a33-video-engine",
        "allwinner,sun8i-a83t-csi",
        "allwinner,sun8i-h3-csi",
        "allwinner,sun8i-h3-video-engine",
        "allwinner,sun8i-v3s-csi",
        "allwinner,sun9i-a80-display-backend",
        "allwinner,sun50i-a64-csi",
        "allwinner,sun50i-a64-video-engine",
        "allwinner,sun50i-h5-video-engine",
        NULL,
};

static int sunxi_mbus_notifier(struct notifier_block *nb,
                               unsigned long event, void *__dev)
{
        struct device *dev = __dev;
        int ret;

        if (event != BUS_NOTIFY_ADD_DEVICE)
                return NOTIFY_DONE;

        /*
         * Only the devices that need a large memory bandwidth do DMA
         * directly over the memory bus (called MBUS), instead of going
         * through the regular system bus.
         */
        if (!of_device_compatible_match(dev->of_node, sunxi_mbus_devices))
                return NOTIFY_DONE;

        /*
         * Devices with an interconnects property have the MBUS
         * relationship described in their DT and dealt with by
         * of_dma_configure, so we can just skip them.
         *
         * Older DTs or SoCs who are not clearly understood need to set
         * that DMA offset though.
         */
        if (of_property_present(dev->of_node, "interconnects"))
                return NOTIFY_DONE;

        ret = dma_direct_set_offset(dev, PHYS_OFFSET, 0, SZ_4G);
        if (ret)
                dev_err(dev, "Couldn't setup our DMA offset: %d\n", ret);

        return NOTIFY_DONE;
}

static struct notifier_block sunxi_mbus_nb = {
        .notifier_call = sunxi_mbus_notifier,
};

static const char * const sunxi_mbus_platforms[] __initconst = {
        "allwinner,sun4i-a10",
        "allwinner,sun5i-a10s",
        "allwinner,sun5i-a13",
        "allwinner,sun6i-a31",
        "allwinner,sun7i-a20",
        "allwinner,sun8i-a23",
        "allwinner,sun8i-a33",
        "allwinner,sun8i-a83t",
        "allwinner,sun8i-h3",
        "allwinner,sun8i-r40",
        "allwinner,sun8i-v3",
        "allwinner,sun8i-v3s",
        "allwinner,sun9i-a80",
        "allwinner,sun50i-a64",
        "allwinner,sun50i-h5",
        "nextthing,gr8",
        NULL,
};

static int __init sunxi_mbus_init(void)
{
        if (!of_device_compatible_match(of_root, sunxi_mbus_platforms))
                return 0;

        bus_register_notifier(&platform_bus_type, &sunxi_mbus_nb);
        return 0;
}
arch_initcall(sunxi_mbus_init);