// SPDX-License-Identifier: GPL-2.0-only
/*
 * Nintendo Wii and Wii U OTP driver
 *
 * This is a driver exposing the OTP of a Nintendo Wii or Wii U console.
 *
 * This memory contains common and per-console keys, signatures and
 * related data required to access peripherals.
 *
 * Based on reversed documentation from https://wiiubrew.org/wiki/Hardware/OTP
 *
 * Copyright (C) 2021 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
 */

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

#define HW_OTPCMD  0
#define HW_OTPDATA 4
#define OTP_READ   0x80000000
#define BANK_SIZE  128
#define WORD_SIZE  4

struct nintendo_otp_priv {
        void __iomem *regs;
};

struct nintendo_otp_devtype_data {
        const char *name;
        unsigned int num_banks;
};

static const struct nintendo_otp_devtype_data hollywood_otp_data = {
        .name = "wii-otp",
        .num_banks = 1,
};

static const struct nintendo_otp_devtype_data latte_otp_data = {
        .name = "wiiu-otp",
        .num_banks = 8,
};

static int nintendo_otp_reg_read(void *context,
                                 unsigned int reg, void *_val, size_t bytes)
{
        struct nintendo_otp_priv *priv = context;
        u32 *val = _val;
        int words = bytes / WORD_SIZE;
        u32 bank, addr;

        while (words--) {
                bank = (reg / BANK_SIZE) << 8;
                addr = (reg / WORD_SIZE) % (BANK_SIZE / WORD_SIZE);
                iowrite32be(OTP_READ | bank | addr, priv->regs + HW_OTPCMD);
                *val++ = ioread32be(priv->regs + HW_OTPDATA);
                reg += WORD_SIZE;
        }

        return 0;
}

static const struct of_device_id nintendo_otp_of_table[] = {
        { .compatible = "nintendo,hollywood-otp", .data = &hollywood_otp_data },
        { .compatible = "nintendo,latte-otp", .data = &latte_otp_data },
        {/* sentinel */},
};
MODULE_DEVICE_TABLE(of, nintendo_otp_of_table);

static int nintendo_otp_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        const struct of_device_id *of_id =
                of_match_device(nintendo_otp_of_table, dev);
        struct nvmem_device *nvmem;
        struct nintendo_otp_priv *priv;

        struct nvmem_config config = {
                .stride = WORD_SIZE,
                .word_size = WORD_SIZE,
                .reg_read = nintendo_otp_reg_read,
                .read_only = true,
                .root_only = true,
        };

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

        priv->regs = devm_platform_ioremap_resource(pdev, 0);
        if (IS_ERR(priv->regs))
                return PTR_ERR(priv->regs);

        if (of_id->data) {
                const struct nintendo_otp_devtype_data *data = of_id->data;
                config.name = data->name;
                config.size = data->num_banks * BANK_SIZE;
        }

        config.dev = dev;
        config.priv = priv;

        nvmem = devm_nvmem_register(dev, &config);

        return PTR_ERR_OR_ZERO(nvmem);
}

static struct platform_driver nintendo_otp_driver = {
        .probe = nintendo_otp_probe,
        .driver = {
                .name = "nintendo-otp",
                .of_match_table = nintendo_otp_of_table,
        },
};
module_platform_driver(nintendo_otp_driver);
MODULE_AUTHOR("Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>");
MODULE_DESCRIPTION("Nintendo Wii and Wii U OTP driver");
MODULE_LICENSE("GPL v2");