/*
 * Copyright (c) 2016-2019, NVIDIA CORPORATION. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for
 * more details.
 */
#include <linux/module.h>
#include <linux/device.h>
#include <linux/slab.h>
#include <linux/of_platform.h>
#include <linux/tegra-firmwares.h>
#include <linux/cpu.h>
#include <asm/cpu.h>

static ssize_t tegrafw_read_denver(struct device *dev,
				char *version, size_t size)
{
	char *v = version;
	int i;
	size_t printed = 0;

	for_each_online_cpu(i) {
		struct cpuinfo_arm64 *cpuinfo = &per_cpu(cpu_data, i);
		u32 midr = cpuinfo->reg_midr;

		if (MIDR_IMPLEMENTOR(midr) == ARM_CPU_IMP_NVIDIA) {
			printed = snprintf(v, size, "CPU%d: %u(0x%x) ",
				i, cpuinfo->reg_aidr, cpuinfo->reg_aidr);
			size -= printed;
			v += printed;
		}
	}

	return v - version;
}

#define FIRMWARES_SIZE  10
static struct device *firmwares[FIRMWARES_SIZE];
#define check_out_of_bounds(dev, err) \
	if ((dev) - firmwares >= ARRAY_SIZE(firmwares)) { \
		pr_err("Cannot register 'legacy' firmware info: increase " \
			"firmwares array size"); \
		return (err); \
	}

static int __init tegra_firmwares_init(void)
{
	struct device **dev = firmwares;
	char *versions[] = { "mb1", "mb2", "mb1-bct", "qb", "osl" };
	int v;

	check_out_of_bounds(dev, 0);
	*dev++ = tegrafw_register("MTS", TFW_NORMAL, tegrafw_read_denver, NULL);

	for (v = 0; v < ARRAY_SIZE(versions); v++) {
		check_out_of_bounds(dev, 0);
		*dev++ = tegrafw_register_dt_string(versions[v],
			"/tegra-firmwares", versions[v]);
	}
	return 0;
}

static void __exit tegra_firmwares_exit(void)
{
	struct device **dev = firmwares;

	while (dev - firmwares < ARRAY_SIZE(firmwares))
		tegrafw_unregister(*dev++);
}
module_init(tegra_firmwares_init);
module_exit(tegra_firmwares_exit);
MODULE_LICENSE("GPL");
MODULE_AUTHOR("dmitry pervushin <dpervushin@nvidia.com>");