// SPDX-License-Identifier: GPL-2.0
/* Copyright (c) 2025 Loongson Technology Corporation Limited. */

#include <linux/device.h>
#include <linux/mfd/loongson-se.h>
#include <linux/platform_device.h>
#include <linux/wait.h>

#include "tpm.h"

struct tpm_loongson_cmd {
        u32 cmd_id;
        u32 data_off;
        u32 data_len;
        u32 pad[5];
};

static int tpm_loongson_recv(struct tpm_chip *chip, u8 *buf, size_t count)
{
        struct loongson_se_engine *tpm_engine = dev_get_drvdata(&chip->dev);
        struct tpm_loongson_cmd *cmd_ret = tpm_engine->command_ret;

        if (cmd_ret->data_len > count)
                return -EIO;

        memcpy(buf, tpm_engine->data_buffer, cmd_ret->data_len);

        return cmd_ret->data_len;
}

static int tpm_loongson_send(struct tpm_chip *chip, u8 *buf, size_t bufsiz, size_t count)
{
        struct loongson_se_engine *tpm_engine = dev_get_drvdata(&chip->dev);
        struct tpm_loongson_cmd *cmd = tpm_engine->command;

        if (count > tpm_engine->buffer_size)
                return -E2BIG;

        cmd->data_len = count;
        memcpy(tpm_engine->data_buffer, buf, count);

        return loongson_se_send_engine_cmd(tpm_engine);
}

static const struct tpm_class_ops tpm_loongson_ops = {
        .flags = TPM_OPS_AUTO_STARTUP,
        .recv = tpm_loongson_recv,
        .send = tpm_loongson_send,
};

static int tpm_loongson_probe(struct platform_device *pdev)
{
        struct loongson_se_engine *tpm_engine;
        struct device *dev = &pdev->dev;
        struct tpm_loongson_cmd *cmd;
        struct tpm_chip *chip;

        tpm_engine = loongson_se_init_engine(dev->parent, SE_ENGINE_TPM);
        if (!tpm_engine)
                return -ENODEV;
        cmd = tpm_engine->command;
        cmd->cmd_id = SE_CMD_TPM;
        cmd->data_off = tpm_engine->buffer_off;

        chip = tpmm_chip_alloc(dev, &tpm_loongson_ops);
        if (IS_ERR(chip))
                return PTR_ERR(chip);
        chip->flags = TPM_CHIP_FLAG_TPM2 | TPM_CHIP_FLAG_IRQ;
        dev_set_drvdata(&chip->dev, tpm_engine);

        return tpm_chip_register(chip);
}

static struct platform_driver tpm_loongson = {
        .probe   = tpm_loongson_probe,
        .driver  = {
                .name  = "tpm_loongson",
        },
};
module_platform_driver(tpm_loongson);

MODULE_ALIAS("platform:tpm_loongson");
MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("Loongson TPM driver");