// SPDX-License-Identifier: (GPL-2.0+ OR MIT)
/*
 * Copyright (c) 2023 Amlogic, Inc. All rights reserved.
 */

#include <env.h>
#include <memalign.h>
#include <errno.h>
#include <amlogic/storage.h>
#include <asm/global_data.h>
#include <common.h>
#include <env_internal.h>
#include <search.h>

DECLARE_GLOBAL_DATA_PTR;

#ifdef CONFIG_CMD_SAVEENV
static int env_storage_save(void)
{
	if (store_get_type() == BOOT_NONE) {
		printf("env_storage: must init before save\n");
		return -ENOENT;
	}
	ALLOC_CACHE_ALIGN_BUFFER(env_t, env_new, 1);
	if (env_export(env_new)) {
		printf("env_storage: export failed\n");
		return -EINVAL;
	}

	if (store_rsv_write(RSV_ENV, CONFIG_ENV_SIZE, (void *)env_new)) {
		printf("env_storage: write failed\n");
		return -EIO;
	}

	return 0;
}
#endif /* CONFIG_CMD_SAVEENV */

static int env_storage_load(void)
{
	if (store_get_type() == BOOT_NONE) {
		printf("env_storage: must init before load\n");
		return -ENOENT;
	}
	ALLOC_CACHE_ALIGN_BUFFER(char, buf, CONFIG_ENV_SIZE);

	if (store_rsv_read(RSV_ENV, CONFIG_ENV_SIZE, (void *)buf)) {
		env_set_default("!env_storage: read failed", 0);
		return -EIO;
	}
	else if (env_import(buf, 1, H_EXTERNAL)) {
		env_set_default("!env_storage: import failed", 0);
		return -EINVAL;
	}

	char *res = NULL;
	ssize_t len = hexport_r(&env_htab, '\n', H_HIDE_DOT, &res, 0, 0, NULL);

	printf("env used size: %zu / CONFIG_ENV_SIZE=%d\n", (size_t)len, CONFIG_ENV_SIZE);
	free(res);

	if (len <= 1)
		env_set_default("## Resetting to default environment\n",
			H_INTERACTIVE);

	return 0;
}


U_BOOT_ENV_LOCATION(storage) = {
	.location	= ENVL_STORAGE,
	ENV_NAME("STORAGE")
	.load		= env_storage_load,
#ifdef CONFIG_CMD_SAVEENV
	.save		= env_save_ptr(env_storage_save),
#endif
};