/*-
 * SPDX-License-Identifier: BSD-2-Clause
 *
 * Copyright (c) 2023 Beckhoff Automation GmbH & Co. KG
 *
 * 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 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.
 */

#include <sys/cdefs.h>

#include <sys/param.h>
#include <sys/systm.h>
#include <sys/kernel.h>
#include <sys/module.h>
#include <sys/malloc.h>
#include <sys/bus.h>
#include <sys/cpu.h>
#include <machine/bus.h>

#include <dev/fdt/simplebus.h>

#include <dev/ofw/openfirm.h>
#include <dev/ofw/ofw_bus.h>
#include <dev/ofw/ofw_bus_subr.h>

#include <dev/psci/smccc.h>

#include <dev/firmware/xilinx/pm_defs.h>

#include "zynqmp_firmware_if.h"

enum {
        IOCTL_GET_RPU_OPER_MODE = 0,
        IOCTL_SET_RPU_OPER_MODE = 1,
        IOCTL_RPU_BOOT_ADDR_CONFIG = 2,
        IOCTL_TCM_COMB_CONFIG = 3,
        IOCTL_SET_TAPDELAY_BYPASS = 4,
        IOCTL_SET_SGMII_MODE = 5,
        IOCTL_SD_DLL_RESET = 6,
        IOCTL_SET_SD_TAPDELAY = 7,
         /* Ioctl for clock driver */
        IOCTL_SET_PLL_FRAC_MODE = 8,
        IOCTL_GET_PLL_FRAC_MODE = 9,
        IOCTL_SET_PLL_FRAC_DATA = 10,
        IOCTL_GET_PLL_FRAC_DATA = 11,
        IOCTL_WRITE_GGS = 12,
        IOCTL_READ_GGS = 13,
        IOCTL_WRITE_PGGS = 14,
        IOCTL_READ_PGGS = 15,
        /* IOCTL for ULPI reset */
        IOCTL_ULPI_RESET = 16,
        /* Set healthy bit value */
        IOCTL_SET_BOOT_HEALTH_STATUS = 17,
        IOCTL_AFI = 18,
        /* Probe counter read/write */
        IOCTL_PROBE_COUNTER_READ = 19,
        IOCTL_PROBE_COUNTER_WRITE = 20,
        IOCTL_OSPI_MUX_SELECT = 21,
        /* IOCTL for USB power request */
        IOCTL_USB_SET_STATE = 22,
        /* IOCTL to get last reset reason */
        IOCTL_GET_LAST_RESET_REASON = 23,
        /* AI engine NPI ISR clear */
        IOCTL_AIE_ISR_CLEAR = 24,
        /* Register SGI to ATF */
        IOCTL_REGISTER_SGI = 25,
};

typedef int (*zynqmp_callfn_t)(register_t, register_t, register_t, uint32_t *payload);

struct zynqmp_firmware_softc {
        struct simplebus_softc  sc;
        device_t                dev;
        zynqmp_callfn_t         callfn;
};

/* SMC calling methods */
#define PM_SIP_SVC      0xC2000000

static int
zynqmp_call_smc(uint32_t id, uint32_t a0, uint32_t a1, uint32_t a2, uint32_t a3, uint32_t *payload, bool ignore_error)
{
        struct arm_smccc_res res;
        uint64_t args[3];

        args[0] = id | PM_SIP_SVC;
        args[1] = ((uint64_t)a1 << 32) | a0;
        args[2] = ((uint64_t)a3 << 32) | a2;
        arm_smccc_invoke_smc(args[0], args[1], args[2], &res);
        if (payload != NULL) {
                payload[0] = res.a0 & 0xFFFFFFFF;
                payload[1] = res.a0 >> 32;
                payload[2] = res.a1 & 0xFFFFFFFF;
                payload[3] = res.a1 >> 32;
                if (!ignore_error && payload[0] != PM_RET_SUCCESS) {
                        printf("%s: fail %x\n", __func__, payload[0]);
                        return (EINVAL);
                }
        }
        return (0);
}

/* Firmware methods */
static int
zynqmp_get_api_version(struct zynqmp_firmware_softc *sc)
{
        uint32_t payload[4];
        int rv;

        rv = zynqmp_call_smc(PM_GET_API_VERSION, 0, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        device_printf(sc->dev, "API version = %d.%d\n",
            payload[1] >> 16, payload[1] & 0xFFFF);
out:
        return (rv);
}

static int
zynqmp_get_chipid(struct zynqmp_firmware_softc *sc)
{
        uint32_t payload[4];
        int rv;

        rv = zynqmp_call_smc(PM_GET_CHIPID, 0, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        device_printf(sc->dev, "ID Code = %x Version = %x\n",
            payload[1], payload[2]);
out:
        return (rv);
}

static int
zynqmp_get_trustzone_version(struct zynqmp_firmware_softc *sc)
{
        uint32_t payload[4];
        int rv;

        rv = zynqmp_call_smc(PM_GET_TRUSTZONE_VERSION, 0, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        device_printf(sc->dev, "Trustzone Version = %x\n",
            payload[1]);
out:
        return (rv);
}

/* zynqmp_firmware methods */
static int
zynqmp_firmware_clock_enable(device_t dev, uint32_t clkid)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_ENABLE, clkid, 0, 0, 0, payload, false);
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
        return (rv);
}

static int
zynqmp_firmware_clock_disable(device_t dev, uint32_t clkid)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_DISABLE, clkid, 0, 0, 0, payload, false);
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
        return (rv);
}

static int
zynqmp_firmware_clock_getstate(device_t dev, uint32_t clkid, bool *enabled)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_GETSTATE, clkid, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *enabled = payload[1] == 1 ? true : false;

out:
        return (rv);
}

static int
zynqmp_firmware_clock_setdivider(device_t dev, uint32_t clkid, uint32_t div)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_SETDIVIDER, clkid, div, 0, 0, payload, false);
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
        return (rv);
}

static int
zynqmp_firmware_clock_getdivider(device_t dev, uint32_t clkid, uint32_t *div)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_GETDIVIDER, clkid, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *div = payload[1];

out:
        return (rv);
}

static int
zynqmp_firmware_clock_setparent(device_t dev, uint32_t clkid, uint32_t parentid)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_SETPARENT, clkid, parentid, 0, 0, payload, false);
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
        return (rv);
}

static int
zynqmp_firmware_clock_getparent(device_t dev, uint32_t clkid, uint32_t *parentid)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_CLOCK_GETPARENT, clkid, 0, 0, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *parentid = payload[1];
out:
        return (rv);
}

static int
zynqmp_firmware_pll_get_mode(device_t dev, uint32_t pllid, uint32_t *mode)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_IOCTL, 0, IOCTL_GET_PLL_FRAC_MODE, pllid, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *mode = payload[1];
out:
        return (rv);
}

static int
zynqmp_firmware_pll_get_frac_data(device_t dev, uint32_t pllid, uint32_t *data)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_IOCTL, 0, IOCTL_GET_PLL_FRAC_DATA, pllid, 0, payload, false);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *data = payload[1];
out:
        return (rv);
}

static int
zynqmp_firmware_clock_get_fixedfactor(device_t dev, uint32_t clkid, uint32_t *mult, uint32_t *div)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_QUERY_DATA, PM_QID_CLOCK_GET_FIXEDFACTOR_PARAMS, clkid, 0, 0, payload, true);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                goto out;
        }
        *mult = payload[1];
        *div = payload[2];
out:
        return (rv);
}

static int
zynqmp_firmware_query_data(device_t dev, uint32_t qid, uint32_t arg1, uint32_t arg2, uint32_t arg3, uint32_t *data)
{
        struct zynqmp_firmware_softc *sc;
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_QUERY_DATA, qid, arg1, arg2, arg3, data, true);
        /*
         * PM_QID_CLOCK_GET_NAME always success and if the clock name couldn't
         * be found the clock name will be all null byte
         */
        if (qid == 1)
                rv = 0;
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
        return (rv);
}

static int
zynqmp_firmware_reset_assert(device_t dev, uint32_t resetid, bool enable)
{
        struct zynqmp_firmware_softc *sc;
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_RESET_ASSERT, resetid, enable, 0, 0, NULL, true);
        if (rv != 0)
                device_printf(sc->dev, "SMC Call fail %d\n", rv);

        return (rv);
}

static int
zynqmp_firmware_reset_get_status(device_t dev, uint32_t resetid, bool *status)
{
        struct zynqmp_firmware_softc *sc;
        uint32_t payload[4];
        int rv;

        sc = device_get_softc(dev);
        rv = zynqmp_call_smc(PM_RESET_GET_STATUS, resetid, 0, 0, 0, payload, true);
        if (rv != 0) {
                device_printf(sc->dev, "SMC Call fail %d\n", rv);
                return (rv);
        }
        *status = payload[1];

        return (rv);
}

/* Simplebus methods */
static struct simplebus_devinfo *
zynqmp_firmware_setup_dinfo(device_t dev, phandle_t node,
    struct simplebus_devinfo *di)
{
        struct simplebus_softc *sc;
        struct simplebus_devinfo *ndi;

        sc = device_get_softc(dev);
        if (di == NULL)
                ndi = malloc(sizeof(*ndi), M_DEVBUF, M_WAITOK | M_ZERO);
        else
                ndi = di;
        if (ofw_bus_gen_setup_devinfo(&ndi->obdinfo, node) != 0) {
                if (di == NULL)
                        free(ndi, M_DEVBUF);
                return (NULL);
        }

        /* reg resources is from the parent but interrupts is on the node itself */
        resource_list_init(&ndi->rl);
        ofw_bus_reg_to_rl(dev, OF_parent(node), sc->acells, sc->scells, &ndi->rl);
        ofw_bus_intr_to_rl(dev, node, &ndi->rl, NULL);

        return (ndi);
}

static device_t
zynqmp_firmware_add_device(device_t dev, phandle_t node, u_int order,
    const char *name, int unit, struct simplebus_devinfo *di)
{
        struct simplebus_devinfo *ndi;
        device_t cdev;

        if ((ndi = zynqmp_firmware_setup_dinfo(dev, node, di)) == NULL)
                return (NULL);
        cdev = device_add_child_ordered(dev, order, name, unit);
        if (cdev == NULL) {
                device_printf(dev, "<%s>: device_add_child failed\n",
                    ndi->obdinfo.obd_name);
                resource_list_free(&ndi->rl);
                ofw_bus_gen_destroy_devinfo(&ndi->obdinfo);
                if (di == NULL)
                        free(ndi, M_DEVBUF);
                return (NULL);
        }
        device_set_ivars(cdev, ndi);

        return(cdev);
}

static int
zynqmp_firmware_probe(device_t dev)
{

        if (!ofw_bus_status_okay(dev))
                return (ENXIO);
        if (!ofw_bus_is_compatible(dev, "xlnx,zynqmp-firmware"))
                return (ENXIO);
        device_set_desc(dev, "ZynqMP Firmware");
        return (0);
}

static int
zynqmp_firmware_attach(device_t dev)
{
        struct zynqmp_firmware_softc *sc;
        phandle_t node, child;
        device_t cdev;

        sc = device_get_softc(dev);
        sc->dev = dev;

        if (bootverbose) {
                zynqmp_get_api_version(sc);
                zynqmp_get_chipid(sc);
                zynqmp_get_trustzone_version(sc);
        }

        /* Attach children */
        node = ofw_bus_get_node(dev);
        for (child = OF_child(node); child != 0; child = OF_peer(child)) {
                cdev = zynqmp_firmware_add_device(dev, child, 0, NULL, -1, NULL);
                if (cdev != NULL)
                        device_probe_and_attach(cdev);
        }

        bus_attach_children(dev);
        return (0);
}

static device_method_t zynqmp_firmware_methods[] = {
        /* device_if */
        DEVMETHOD(device_probe,         zynqmp_firmware_probe),
        DEVMETHOD(device_attach,        zynqmp_firmware_attach),

        /* zynqmp_firmware_if */
        DEVMETHOD(zynqmp_firmware_clock_enable, zynqmp_firmware_clock_enable),
        DEVMETHOD(zynqmp_firmware_clock_disable, zynqmp_firmware_clock_disable),
        DEVMETHOD(zynqmp_firmware_clock_getstate, zynqmp_firmware_clock_getstate),
        DEVMETHOD(zynqmp_firmware_clock_setdivider, zynqmp_firmware_clock_setdivider),
        DEVMETHOD(zynqmp_firmware_clock_getdivider, zynqmp_firmware_clock_getdivider),
        DEVMETHOD(zynqmp_firmware_clock_setparent, zynqmp_firmware_clock_setparent),
        DEVMETHOD(zynqmp_firmware_clock_getparent, zynqmp_firmware_clock_getparent),
        DEVMETHOD(zynqmp_firmware_pll_get_mode, zynqmp_firmware_pll_get_mode),
        DEVMETHOD(zynqmp_firmware_pll_get_frac_data, zynqmp_firmware_pll_get_frac_data),
        DEVMETHOD(zynqmp_firmware_clock_get_fixedfactor, zynqmp_firmware_clock_get_fixedfactor),
        DEVMETHOD(zynqmp_firmware_query_data, zynqmp_firmware_query_data),
        DEVMETHOD(zynqmp_firmware_reset_assert, zynqmp_firmware_reset_assert),
        DEVMETHOD(zynqmp_firmware_reset_get_status, zynqmp_firmware_reset_get_status),

        DEVMETHOD_END
};

DEFINE_CLASS_1(zynqmp_firmware, zynqmp_firmware_driver, zynqmp_firmware_methods,
  sizeof(struct zynqmp_firmware_softc), simplebus_driver);

EARLY_DRIVER_MODULE(zynqmp_firmware, simplebus, zynqmp_firmware_driver, 0, 0,
    BUS_PASS_BUS + BUS_PASS_ORDER_LATE);
MODULE_VERSION(zynqmp_firmware, 1);