/*
 * (C) Copyright 2021 Rockchip Electronics Co., Ltd
 *
 * SPDX-License-Identifier:     GPL-2.0+
 */

#include <common.h>
#include <boot_rkimg.h>
#include <memalign.h>
#include <android_avb/rk_avb_ops_user.h>

#define CONFIG_SET_DFU_ALT_BUF_LEN	(SZ_1K)
static char *get_dfu_alt(char *interface, char *devstr)
{
	struct blk_desc *dev_desc;
	char *alt_boot;
#ifdef CONFIG_ANDROID_AB
	char current_slot[3] = {0};
#endif

	dev_desc = rockchip_get_bootdev();
	if (!dev_desc) {
		printf("%s: dev_desc is NULL!\n", __func__);
		return NULL;
	}

	switch (dev_desc->if_type) {
#ifdef CONFIG_DFU_MMC
	case IF_TYPE_MMC:
		alt_boot = DFU_ALT_BOOT_EMMC;
		break;
#endif
#ifdef CONFIG_DFU_MTD
	case IF_TYPE_MTD:
#ifdef CONFIG_ANDROID_AB
		rk_avb_get_current_slot(current_slot);

		if (!strcmp(current_slot, "_a")) {
			alt_boot = DFU_ALT_BOOT_MTD_B;
			printf("\ncurrent boot from slot A!\n");
		} else if (!strcmp(current_slot, "_b")) {
			alt_boot = DFU_ALT_BOOT_MTD_A;
			printf("\ncurrent boot from slot B!\n");
		} else {
			return NULL;
		}
#else
		alt_boot = DFU_ALT_BOOT_MTD;
#endif
		break;
#endif /* CONFIG_DFU_MTD */
	default:
		printf("[dfu ERROR]:Boot device type is invalid!\n");
		return NULL;
	}

	return alt_boot;
}

void set_dfu_alt_info(char *interface, char *devstr)
{
	size_t buf_size = CONFIG_SET_DFU_ALT_BUF_LEN;
	ALLOC_CACHE_ALIGN_BUFFER(char, buf, buf_size);
	char *alt_info = "Settings not found!";
	char *status = "error!\n";
	char *alt_setting;
	int offset = 0;

	puts("DFU alt info setting: ");

	alt_setting = get_dfu_alt(interface, devstr);
	if (alt_setting) {
		env_set("dfu_alt_boot", alt_setting);
		offset = snprintf(buf, buf_size, "%s", alt_setting);
	}

	if (offset) {
		alt_info = buf;
		status = "done\n";
	}

	env_set("dfu_alt_info", alt_info);
	puts(status);
}