// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright 2023-2025 NXP
 */

#include <linux/device.h>
#include <linux/io.h>
#include <linux/module.h>
#include <linux/nvmem-provider.h>
#include <linux/of.h>
#include <linux/of_device.h>
#include <linux/platform_device.h>

struct s32g_ocotp_priv {
        struct device *dev;
        void __iomem *base;
};

static int s32g_ocotp_read(void *context, unsigned int offset,
                            void *val, size_t bytes)
{
        struct s32g_ocotp_priv *s32g_data = context;
        u32 *dst = val;

        while (bytes >= sizeof(u32)) {
                *dst++ = ioread32(s32g_data->base + offset);

                bytes -= sizeof(u32);
                offset += sizeof(u32);
        }

        return 0;
}

static struct nvmem_keepout s32g_keepouts[] = {
        { .start = 0,   .end = 520 },
        { .start = 540, .end = 564 },
        { .start = 596, .end = 664 },
        { .start = 668, .end = 676 },
        { .start = 684, .end = 732 },
        { .start = 744, .end = 864 },
        { .start = 908, .end = 924 },
        { .start = 928, .end = 936 },
        { .start = 948, .end = 964 },
        { .start = 968, .end = 976 },
        { .start = 984, .end = 1012 },
};

static struct nvmem_config s32g_ocotp_nvmem_config = {
        .name = "s32g-ocotp",
        .add_legacy_fixed_of_cells = true,
        .read_only = true,
        .word_size = 4,
        .reg_read = s32g_ocotp_read,
        .keepout = s32g_keepouts,
        .nkeepout = ARRAY_SIZE(s32g_keepouts),
};

static const struct of_device_id ocotp_of_match[] = {
        { .compatible = "nxp,s32g2-ocotp" },
        { /* sentinel */ }
};

static int s32g_ocotp_probe(struct platform_device *pdev)
{
        struct s32g_ocotp_priv *s32g_data;
        struct device *dev = &pdev->dev;
        struct nvmem_device *nvmem;
        struct resource *res;

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

        s32g_data->base = devm_platform_get_and_ioremap_resource(pdev, 0, &res);
        if (IS_ERR(s32g_data->base))
                return dev_err_probe(dev, PTR_ERR(s32g_data->base),
                                     "Cannot map OCOTP device.\n");

        s32g_data->dev = dev;
        s32g_ocotp_nvmem_config.dev = dev;
        s32g_ocotp_nvmem_config.priv = s32g_data;
        s32g_ocotp_nvmem_config.size = resource_size(res);

        nvmem = devm_nvmem_register(dev, &s32g_ocotp_nvmem_config);

        return PTR_ERR_OR_ZERO(nvmem);
}

static struct platform_driver s32g_ocotp_driver = {
        .probe = s32g_ocotp_probe,
        .driver = {
                .name = "s32g-ocotp",
                .of_match_table = ocotp_of_match,
        },
};
module_platform_driver(s32g_ocotp_driver);
MODULE_AUTHOR("NXP");
MODULE_DESCRIPTION("S32G OCOTP driver");
MODULE_LICENSE("GPL");