// SPDX-License-Identifier: GPL-2.0+
/*
 * BTRFS filesystem implementation for U-Boot
 *
 * 2017 Marek Behun, CZ.NIC, marek.behun@nic.cz
 */

#include <malloc.h>
#include "ctree.h"
#include "btrfs.h"

/*
 * Resolve the path of ino inside subvolume @root into @path_ret.
 *
 * @path_ret must be at least PATH_MAX size.
 */
static int get_path_in_subvol(struct btrfs_root *root, u64 ino, char *path_ret)
{
	struct btrfs_path path;
	struct btrfs_key key;
	char *tmp;
	u64 cur = ino;
	int ret = 0;

	tmp = malloc(PATH_MAX);
	if (!tmp)
		return -ENOMEM;
	tmp[0] = '\0';

	btrfs_init_path(&path);
	while (cur != BTRFS_FIRST_FREE_OBJECTID) {
		struct btrfs_inode_ref *iref;
		int name_len;

		btrfs_release_path(&path);
		key.objectid = cur;
		key.type = BTRFS_INODE_REF_KEY;
		key.offset = (u64)-1;

		ret = btrfs_search_slot(NULL, root, &key, &path, 0, 0);
		/* Impossible */
		if (ret == 0)
			ret = -EUCLEAN;
		if (ret < 0)
			goto out;
		ret = btrfs_previous_item(root, &path, cur,
					  BTRFS_INODE_REF_KEY);
		if (ret > 0)
			ret = -ENOENT;
		if (ret < 0)
			goto out;

		strncpy(tmp, path_ret, PATH_MAX);
		iref = btrfs_item_ptr(path.nodes[0], path.slots[0],
				      struct btrfs_inode_ref);
		name_len = btrfs_inode_ref_name_len(path.nodes[0],
						    iref);
		if (name_len > BTRFS_NAME_LEN) {
			ret = -ENAMETOOLONG;
			goto out;
		}
		read_extent_buffer(path.nodes[0], path_ret,
				   (unsigned long)(iref + 1), name_len);
		path_ret[name_len] = '/';
		path_ret[name_len + 1] = '\0';
		strncat(path_ret, tmp, PATH_MAX);

		btrfs_item_key_to_cpu(path.nodes[0], &key, path.slots[0]);
		cur = key.offset;
	}
out:
	btrfs_release_path(&path);
	free(tmp);
	return ret;
}

static int get_subvol_name(u64 subvolid, char *name, int max_len)
{
	struct btrfs_root_ref rref;
	struct btrfs_inode_ref iref;
	struct __btrfs_root root;
	u64 dir;
	char tmp[BTRFS_NAME_LEN];
	char *ptr;

	ptr = name + max_len - 1;
	*ptr = '\0';

	while (subvolid != BTRFS_FS_TREE_OBJECTID) {
		subvolid = btrfs_lookup_root_ref(subvolid, &rref, tmp);

		if (subvolid == -1ULL)
			return -1;

		ptr -= rref.name_len + 1;
		if (ptr < name)
			goto too_long;

		memcpy(ptr + 1, tmp, rref.name_len);
		*ptr = '/';

		if (btrfs_find_root(subvolid, &root, NULL))
			return -1;

		dir = rref.dirid;

		while (dir != BTRFS_FIRST_FREE_OBJECTID) {
			dir = __btrfs_lookup_inode_ref(&root, dir, &iref, tmp);

			if (dir == -1ULL)
				return -1;

			ptr -= iref.name_len + 1;
			if (ptr < name)
				goto too_long;

			memcpy(ptr + 1, tmp, iref.name_len);
			*ptr = '/';
		}
	}

	if (ptr == name + max_len - 1) {
		name[0] = '/';
		name[1] = '\0';
	} else {
		memmove(name, ptr, name + max_len - ptr);
	}

	return 0;

too_long:
	printf("%s: subvolume name too long\n", __func__);
	return -1;
}

u64 btrfs_get_default_subvol_objectid(void)
{
	struct btrfs_dir_item item;

	if (__btrfs_lookup_dir_item(&btrfs_info.tree_root,
				  btrfs_info.sb.root_dir_objectid, "default", 7,
				  &item))
		return BTRFS_FS_TREE_OBJECTID;
	return item.location.objectid;
}

static void list_subvols(u64 tree, char *nameptr, int max_name_len, int level)
{
	struct btrfs_key key, *found_key;
	struct __btrfs_path path;
	struct btrfs_root_ref *ref;
	int res;

	key.objectid = tree;
	key.type = BTRFS_ROOT_REF_KEY;
	key.offset = 0;

	if (btrfs_search_tree(&btrfs_info.tree_root, &key, &path))
		return;

	do {
		found_key = btrfs_path_leaf_key(&path);
		if (btrfs_comp_keys_type(&key, found_key))
			break;

		ref = btrfs_path_item_ptr(&path, struct btrfs_root_ref);
		btrfs_root_ref_to_cpu(ref);

		printf("ID %llu parent %llu name ", found_key->offset, tree);
		if (nameptr && !get_subvol_name(found_key->offset, nameptr,
						max_name_len))
			printf("%s\n", nameptr);
		else
			printf("%.*s\n", (int) ref->name_len,
			       (const char *) (ref + 1));

		if (level > 0)
			list_subvols(found_key->offset, nameptr, max_name_len,
				     level - 1);
		else
			printf("%s: Too much recursion, maybe skipping some "
			       "subvolumes\n", __func__);
	} while (!(res = btrfs_next_slot(&path)));

	__btrfs_free_path(&path);
}

void btrfs_list_subvols(void)
{
	char *nameptr = malloc(4096);

	list_subvols(BTRFS_FS_TREE_OBJECTID, nameptr, nameptr ? 4096 : 0, 40);

	if (nameptr)
		free(nameptr);
}