/*
 * Copyright (c) 2018-2023, NVIDIA CORPORATION.  All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <linux/io.h>
#include <linux/highmem.h>
#include <linux/platform_device.h>

#include <nvgpu/log.h>
#include <nvgpu/linux/vm.h>
#include <nvgpu/bitops.h>
#include <nvgpu/nvgpu_mem.h>
#include <nvgpu/dma.h>
#include <nvgpu/soc.h>
#include <nvgpu/hw_sim.h>
#include <nvgpu/sim.h>
#include <nvgpu/gk20a.h>

#include "platform_gk20a.h"
#include "os_linux.h"
#include "module.h"

void nvgpu_remove_sim_support_linux(struct gk20a *g)
{
	struct sim_nvgpu_linux *sim_linux;

	if (!g->sim)
		return;

	sim_linux = container_of(g->sim, struct sim_nvgpu_linux, sim);
	if (g->sim->regs) {
		sim_writel(g->sim, sim_config_r(), sim_config_mode_disabled_v());
		iounmap((void __iomem *)g->sim->regs);
		g->sim->regs = 0U;
	}
	nvgpu_kfree(g, sim_linux);
	g->sim = NULL;
}

int nvgpu_init_sim_support_linux(struct gk20a *g,
		struct platform_device *dev)
{
	struct sim_nvgpu_linux *sim_linux;
	void __iomem *addr;
	int err = -ENOMEM;

	if (!nvgpu_platform_is_simulation(g))
		return 0;

	sim_linux = nvgpu_kzalloc(g, sizeof(*sim_linux));
	if (!sim_linux)
		return err;
	g->sim = &sim_linux->sim;
	g->sim->g = g;
	addr = nvgpu_devm_ioremap_resource(dev,
					GK20A_SIM_IORESOURCE_MEM,
					NULL);
	if (IS_ERR(addr)) {
		nvgpu_err(g, "failed to remap gk20a sim regs");
		err = PTR_ERR(addr);
		goto fail;
	}
	g->sim->regs = (uintptr_t)addr;
	sim_linux->remove_support_linux = nvgpu_remove_sim_support_linux;
	return 0;

fail:
	nvgpu_remove_sim_support_linux(g);
	return err;
}

int nvgpu_init_sim_support_linux_igpu_pci(struct gk20a *g,
					struct gk20a_platform *platform)
{
	struct device *dev = dev_from_gk20a(g);
	struct sim_nvgpu_linux *sim_linux;
	void __iomem *addr;
	int err = -ENOMEM;

	if (!nvgpu_platform_is_simulation(g))
		return 0;

	sim_linux = nvgpu_kzalloc(g, sizeof(*sim_linux));
	if (!sim_linux)
		return err;
	g->sim = &sim_linux->sim;
	g->sim->g = g;
	addr = nvgpu_devm_ioremap(dev,
			(resource_size_t)platform->fake_rpc_base,
			(resource_size_t)platform->fake_rpc_size);
	if (IS_ERR(addr)) {
		nvgpu_err(g, "failed to remap gpu sim regs");
		err = PTR_ERR(addr);
		goto fail;
	}
	g->sim->regs = (uintptr_t)addr;
	sim_linux->remove_support_linux = nvgpu_remove_sim_support_linux;
	return 0;

fail:
	nvgpu_remove_sim_support_linux(g);
	return err;
}