// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2026, STMicroelectronics - All Rights Reserved
 */

#include <linux/bus/stm32_firewall.h>
#include <linux/bus/stm32_firewall_device.h>
#include <linux/device.h>
#include <linux/err.h>
#include <linux/kernel.h>
#include <linux/module.h>
#include <linux/of.h>
#include <linux/of_platform.h>
#include <linux/platform_device.h>
#include <linux/pm_runtime.h>
#include <linux/tee_drv.h>
#include <linux/types.h>

enum stm32_dbg_profile {
        PERIPHERAL_DBG_PROFILE  = 0,
        HDP_DBG_PROFILE         = 1,
};

enum stm32_dbg_pta_command {
        /*
         * PTA_CMD_GRANT_DBG_ACCESS - Verify the debug configuration against the given debug profile
         * and grant access or not
         *
         * [in]     value[0].a  Debug profile to grant access to.
         */
        PTA_CMD_GRANT_DBG_ACCESS,
};

/**
 * struct stm32_dbg_bus - OP-TEE based STM32 debug bus private data
 * @dev: STM32 debug bus device.
 * @ctx: OP-TEE context handler.
 */
struct stm32_dbg_bus {
        struct device *dev;
        struct tee_context *ctx;
};

/* Expect at most 1 instance of this driver */
static struct stm32_dbg_bus *stm32_dbg_bus_priv;

static int stm32_dbg_pta_open_session(u32 *id)
{
        struct tee_client_device *dbg_bus_dev = to_tee_client_device(stm32_dbg_bus_priv->dev);
        struct tee_ioctl_open_session_arg sess_arg;
        int ret;

        memset(&sess_arg, 0, sizeof(sess_arg));
        export_uuid(sess_arg.uuid, &dbg_bus_dev->id.uuid);
        sess_arg.clnt_login = TEE_IOCTL_LOGIN_REE_KERNEL;

        ret = tee_client_open_session(stm32_dbg_bus_priv->ctx, &sess_arg, NULL);
        if (ret < 0 || sess_arg.ret) {
                dev_err(stm32_dbg_bus_priv->dev, "Failed opening tee session, err: %#x\n",
                        sess_arg.ret);
                return -EOPNOTSUPP;
        }

        *id = sess_arg.session;

        return 0;
}

static void stm32_dbg_pta_close_session(u32 id)
{
        tee_client_close_session(stm32_dbg_bus_priv->ctx, id);
}

static int stm32_dbg_bus_grant_access(struct stm32_firewall_controller *ctrl, u32 dbg_profile)
{
        struct tee_ioctl_invoke_arg inv_arg = {0};
        struct tee_param param[1] = {0};
        u32 session_id;
        int ret;

        if (dbg_profile != PERIPHERAL_DBG_PROFILE && dbg_profile != HDP_DBG_PROFILE)
                return -EOPNOTSUPP;

        ret = stm32_dbg_pta_open_session(&session_id);
        if (ret)
                return ret;

        inv_arg.func = PTA_CMD_GRANT_DBG_ACCESS;
        inv_arg.session = session_id;
        inv_arg.num_params = 1;
        param[0].attr = TEE_IOCTL_PARAM_ATTR_TYPE_VALUE_INPUT;
        param[0].u.value.a = dbg_profile;

        ret = tee_client_invoke_func(stm32_dbg_bus_priv->ctx, &inv_arg, param);
        if (ret < 0 || inv_arg.ret != 0) {
                dev_dbg(stm32_dbg_bus_priv->dev,
                        "When invoking function, err %x, TEE returns: %x\n", ret, inv_arg.ret);
                if (!ret)
                        ret = -EACCES;
        }

        stm32_dbg_pta_close_session(session_id);

        return ret;
}

/* Implement mandatory release_access ops even if it does nothing*/
static void stm32_dbg_bus_release_access(struct stm32_firewall_controller *ctrl, u32 dbg_profile)
{
}

static int stm32_dbg_bus_plat_probe(struct platform_device *pdev)
{
        struct stm32_firewall_controller *dbg_controller;
        int ret;

        /* Defer if OP-TEE service is not yet available */
        if (!stm32_dbg_bus_priv)
                return -EPROBE_DEFER;

        dbg_controller = devm_kzalloc(&pdev->dev, sizeof(*dbg_controller), GFP_KERNEL);
        if (!dbg_controller)
                return dev_err_probe(&pdev->dev, -ENOMEM, "Couldn't allocate debug controller\n");

        dbg_controller->dev = &pdev->dev;
        dbg_controller->mmio = NULL;
        dbg_controller->name = dev_driver_string(dbg_controller->dev);
        dbg_controller->type = STM32_PERIPHERAL_FIREWALL;
        dbg_controller->grant_access = stm32_dbg_bus_grant_access;
        dbg_controller->release_access = stm32_dbg_bus_release_access;

        ret = stm32_firewall_controller_register(dbg_controller);
        if (ret) {
                dev_err(dbg_controller->dev, "Couldn't register as a firewall controller: %d", ret);
                return ret;
        }

        ret = stm32_firewall_populate_bus(dbg_controller);
        if (ret) {
                dev_err(dbg_controller->dev, "Couldn't populate debug bus: %d", ret);
                stm32_firewall_controller_unregister(dbg_controller);
                return ret;
        }

        pm_runtime_enable(&pdev->dev);

        ret = of_platform_populate(pdev->dev.of_node, NULL, NULL, &pdev->dev);
        if (ret) {
                dev_err(dbg_controller->dev, "Couldn't populate the node: %d", ret);
                stm32_firewall_controller_unregister(dbg_controller);
                return ret;
        }

        return 0;
}

static const struct of_device_id stm32_dbg_bus_of_match[] = {
        { .compatible = "st,stm32mp131-dbg-bus", },
        { .compatible = "st,stm32mp151-dbg-bus", },
        { },
};
MODULE_DEVICE_TABLE(of, stm32_dbg_bus_of_match);

static struct platform_driver stm32_dbg_bus_driver = {
        .probe = stm32_dbg_bus_plat_probe,
        .driver = {
                .name = "stm32-dbg-bus",
                .of_match_table = stm32_dbg_bus_of_match,
        },
};

static int optee_ctx_match(struct tee_ioctl_version_data *ver, const void *data)
{
        return (ver->impl_id == TEE_IMPL_ID_OPTEE);
}

static void stm32_dbg_bus_remove(struct tee_client_device *tee_dev)
{
        tee_client_close_context(stm32_dbg_bus_priv->ctx);
        stm32_dbg_bus_priv = NULL;

        of_platform_depopulate(&tee_dev->dev);
}

static int stm32_dbg_bus_probe(struct tee_client_device *tee_dev)
{
        struct device *dev = &tee_dev->dev;
        struct stm32_dbg_bus *priv;
        int ret = 0;

        if (stm32_dbg_bus_priv)
                return dev_err_probe(dev, -EBUSY,
                                     "A STM32 debug bus device is already initialized\n");

        priv = devm_kzalloc(dev, sizeof(*priv), GFP_KERNEL);
        if (!priv)
                return -ENOMEM;

        /* Open context with TEE driver */
        priv->ctx = tee_client_open_context(NULL, optee_ctx_match, NULL, NULL);
        if (IS_ERR_OR_NULL(priv->ctx))
                return dev_err_probe(dev, PTR_ERR_OR_ZERO(priv->ctx), "Cannot open TEE context\n");

        stm32_dbg_bus_priv = priv;
        stm32_dbg_bus_priv->dev = dev;

        return ret;
}

static const struct tee_client_device_id optee_dbg_bus_id_table[] = {
        {UUID_INIT(0xdd05bc8b, 0x9f3b, 0x49f0,
                   0xb6, 0x49, 0x01, 0xaa, 0x10, 0xc1, 0xc2, 0x10)},
        {}
};

static struct tee_client_driver stm32_optee_dbg_bus_driver = {
        .id_table = optee_dbg_bus_id_table,
        .probe = stm32_dbg_bus_probe,
        .remove = stm32_dbg_bus_remove,
        .driver = {
                .name = "optee_dbg_bus",
        },
};

static void __exit stm32_optee_dbg_bus_driver_exit(void)
{
        platform_driver_unregister(&stm32_dbg_bus_driver);
        tee_client_driver_unregister(&stm32_optee_dbg_bus_driver);
}
module_exit(stm32_optee_dbg_bus_driver_exit);

static int __init stm32_optee_dbg_bus_driver_init(void)
{
        int err;

        err = tee_client_driver_register(&stm32_optee_dbg_bus_driver);
        if (err)
                return err;

        err = platform_driver_register(&stm32_dbg_bus_driver);
        if (err)
                tee_client_driver_unregister(&stm32_optee_dbg_bus_driver);

        return err;
}
module_init(stm32_optee_dbg_bus_driver_init);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Gatien Chevallier <gatien.chevallier@foss.st.com>");
MODULE_DESCRIPTION("OP-TEE based STM32 debug access bus driver");