// SPDX-License-Identifier: GPL-2.0-only
/*
 * Copyright (C) 2024 Allied Telesis
 */

#include <linux/errno.h>
#include <linux/libfdt.h>
#include <linux/printk.h>
#include <linux/types.h>

#include <asm/fw/fw.h>
#include <asm/machine.h>

static __init int realtek_add_initrd(void *fdt)
{
	int node, err;
	u32 start, size;

	node = fdt_path_offset(fdt, "/chosen");
	if (node < 0) {
		pr_err("/chosen node not found\n");
		return -ENOENT;
	}

	start = fw_getenvl("initrd_start");
	size = fw_getenvl("initrd_size");

	if (start == 0 && size == 0)
		return 0;

	pr_info("Adding initrd info from environment\n");

	err = fdt_setprop_u32(fdt, node, "linux,initrd-start", start);
	if (err) {
		pr_err("unable to set initrd-start: %d\n", err);
		return err;
	}

	err = fdt_setprop_u32(fdt, node, "linux,initrd-end", start + size);
	if (err) {
		pr_err("unable to set initrd-end: %d\n", err);
		return err;
	}

	return 0;
}

static const struct mips_fdt_fixup realtek_fdt_fixups[] __initconst = {
	{ realtek_add_initrd, "add initrd" },
	{},
};

static __init const void *realtek_fixup_fdt(const void *fdt, const void *match_data)
{
	static unsigned char fdt_buf[16 << 10] __initdata;
	int err;

	if (fdt_check_header(fdt))
		panic("Corrupt DT");

	fw_init_cmdline();

	err = apply_mips_fdt_fixups(fdt_buf, sizeof(fdt_buf), fdt, realtek_fdt_fixups);
	if (err)
		panic("Unable to fixup FDT: %d", err);

	return fdt_buf;

}

static const struct of_device_id realtek_of_match[] __initconst = {
	{ .compatible = "realtek,rtl9302-soc" },
	{}
};

MIPS_MACHINE(realtek) = {
	.matches = realtek_of_match,
	.fixup_fdt = realtek_fixup_fdt,
};