// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2021 Rafał Miłecki <rafal@milecki.pl>
 */

#include <linux/module.h>
#include <linux/init.h>
#include <linux/of.h>
#include <linux/mtd/mtd.h>
#include <linux/slab.h>
#include <linux/mtd/partitions.h>

#include "ofpart_bcm4908.h"

#define BLPARAMS_FW_OFFSET		"NAND_RFS_OFS"

static long long bcm4908_partitions_fw_offset(void)
{
	struct device_node *root;
	struct property *prop;
	const char *s;

	root = of_find_node_by_path("/");
	if (!root)
		return -ENOENT;

	of_property_for_each_string(root, "brcm_blparms", prop, s) {
		size_t len = strlen(BLPARAMS_FW_OFFSET);
		unsigned long offset;
		int err;

		if (strncmp(s, BLPARAMS_FW_OFFSET, len) || s[len] != '=')
			continue;

		err = kstrtoul(s + len + 1, 0, &offset);
		if (err) {
			pr_err("failed to parse %s\n", s + len + 1);
			of_node_put(root);
			return err;
		}

		of_node_put(root);
		return offset << 10;
	}

	of_node_put(root);
	return -ENOENT;
}

int bcm4908_partitions_post_parse(struct mtd_info *mtd, struct mtd_partition *parts, int nr_parts)
{
	long long fw_offset;
	int i;

	fw_offset = bcm4908_partitions_fw_offset();

	for (i = 0; i < nr_parts; i++) {
		if (of_device_is_compatible(parts[i].of_node, "brcm,bcm4908-firmware")) {
			if (fw_offset < 0 || parts[i].offset == fw_offset)
				parts[i].name = "firmware";
			else
				parts[i].name = "backup";
		}
	}

	return 0;
}