107b30a49SStefan Behrens /* 207b30a49SStefan Behrens * Copyright (C) STRATO AG 2013. All rights reserved. 307b30a49SStefan Behrens * 407b30a49SStefan Behrens * This program is free software; you can redistribute it and/or 507b30a49SStefan Behrens * modify it under the terms of the GNU General Public 607b30a49SStefan Behrens * License v2 as published by the Free Software Foundation. 707b30a49SStefan Behrens * 807b30a49SStefan Behrens * This program is distributed in the hope that it will be useful, 907b30a49SStefan Behrens * but WITHOUT ANY WARRANTY; without even the implied warranty of 1007b30a49SStefan Behrens * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 1107b30a49SStefan Behrens * General Public License for more details. 1207b30a49SStefan Behrens * 1307b30a49SStefan Behrens * You should have received a copy of the GNU General Public 1407b30a49SStefan Behrens * License along with this program; if not, write to the 1507b30a49SStefan Behrens * Free Software Foundation, Inc., 59 Temple Place - Suite 330, 1607b30a49SStefan Behrens * Boston, MA 021110-1307, USA. 1707b30a49SStefan Behrens */ 1807b30a49SStefan Behrens #include <linux/uuid.h> 1907b30a49SStefan Behrens #include <asm/unaligned.h> 2007b30a49SStefan Behrens #include "ctree.h" 2107b30a49SStefan Behrens #include "transaction.h" 2207b30a49SStefan Behrens #include "disk-io.h" 2307b30a49SStefan Behrens #include "print-tree.h" 2407b30a49SStefan Behrens 2507b30a49SStefan Behrens 2607b30a49SStefan Behrens static void btrfs_uuid_to_key(u8 *uuid, u8 type, struct btrfs_key *key) 2707b30a49SStefan Behrens { 2807b30a49SStefan Behrens key->type = type; 2907b30a49SStefan Behrens key->objectid = get_unaligned_le64(uuid); 3007b30a49SStefan Behrens key->offset = get_unaligned_le64(uuid + sizeof(u64)); 3107b30a49SStefan Behrens } 3207b30a49SStefan Behrens 3307b30a49SStefan Behrens /* return -ENOENT for !found, < 0 for errors, or 0 if an item was found */ 3407b30a49SStefan Behrens static int btrfs_uuid_tree_lookup(struct btrfs_root *uuid_root, u8 *uuid, 3507b30a49SStefan Behrens u8 type, u64 subid) 3607b30a49SStefan Behrens { 3707b30a49SStefan Behrens int ret; 3807b30a49SStefan Behrens struct btrfs_path *path = NULL; 3907b30a49SStefan Behrens struct extent_buffer *eb; 4007b30a49SStefan Behrens int slot; 4107b30a49SStefan Behrens u32 item_size; 4207b30a49SStefan Behrens unsigned long offset; 4307b30a49SStefan Behrens struct btrfs_key key; 4407b30a49SStefan Behrens 4507b30a49SStefan Behrens if (WARN_ON_ONCE(!uuid_root)) { 4607b30a49SStefan Behrens ret = -ENOENT; 4707b30a49SStefan Behrens goto out; 4807b30a49SStefan Behrens } 4907b30a49SStefan Behrens 5007b30a49SStefan Behrens path = btrfs_alloc_path(); 5107b30a49SStefan Behrens if (!path) { 5207b30a49SStefan Behrens ret = -ENOMEM; 5307b30a49SStefan Behrens goto out; 5407b30a49SStefan Behrens } 5507b30a49SStefan Behrens 5607b30a49SStefan Behrens btrfs_uuid_to_key(uuid, type, &key); 5707b30a49SStefan Behrens ret = btrfs_search_slot(NULL, uuid_root, &key, path, 0, 0); 5807b30a49SStefan Behrens if (ret < 0) { 5907b30a49SStefan Behrens goto out; 6007b30a49SStefan Behrens } else if (ret > 0) { 6107b30a49SStefan Behrens ret = -ENOENT; 6207b30a49SStefan Behrens goto out; 6307b30a49SStefan Behrens } 6407b30a49SStefan Behrens 6507b30a49SStefan Behrens eb = path->nodes[0]; 6607b30a49SStefan Behrens slot = path->slots[0]; 6707b30a49SStefan Behrens item_size = btrfs_item_size_nr(eb, slot); 6807b30a49SStefan Behrens offset = btrfs_item_ptr_offset(eb, slot); 6907b30a49SStefan Behrens ret = -ENOENT; 7007b30a49SStefan Behrens 7107b30a49SStefan Behrens if (!IS_ALIGNED(item_size, sizeof(u64))) { 72efe120a0SFrank Holton btrfs_warn(uuid_root->fs_info, "uuid item with illegal size %lu!", 7307b30a49SStefan Behrens (unsigned long)item_size); 7407b30a49SStefan Behrens goto out; 7507b30a49SStefan Behrens } 7607b30a49SStefan Behrens while (item_size) { 7707b30a49SStefan Behrens __le64 data; 7807b30a49SStefan Behrens 7907b30a49SStefan Behrens read_extent_buffer(eb, &data, offset, sizeof(data)); 8007b30a49SStefan Behrens if (le64_to_cpu(data) == subid) { 8107b30a49SStefan Behrens ret = 0; 8207b30a49SStefan Behrens break; 8307b30a49SStefan Behrens } 8407b30a49SStefan Behrens offset += sizeof(data); 8507b30a49SStefan Behrens item_size -= sizeof(data); 8607b30a49SStefan Behrens } 8707b30a49SStefan Behrens 8807b30a49SStefan Behrens out: 8907b30a49SStefan Behrens btrfs_free_path(path); 9007b30a49SStefan Behrens return ret; 9107b30a49SStefan Behrens } 9207b30a49SStefan Behrens 9307b30a49SStefan Behrens int btrfs_uuid_tree_add(struct btrfs_trans_handle *trans, 9407b30a49SStefan Behrens struct btrfs_root *uuid_root, u8 *uuid, u8 type, 9507b30a49SStefan Behrens u64 subid_cpu) 9607b30a49SStefan Behrens { 9707b30a49SStefan Behrens int ret; 9807b30a49SStefan Behrens struct btrfs_path *path = NULL; 9907b30a49SStefan Behrens struct btrfs_key key; 10007b30a49SStefan Behrens struct extent_buffer *eb; 10107b30a49SStefan Behrens int slot; 10207b30a49SStefan Behrens unsigned long offset; 10307b30a49SStefan Behrens __le64 subid_le; 10407b30a49SStefan Behrens 10507b30a49SStefan Behrens ret = btrfs_uuid_tree_lookup(uuid_root, uuid, type, subid_cpu); 10607b30a49SStefan Behrens if (ret != -ENOENT) 10707b30a49SStefan Behrens return ret; 10807b30a49SStefan Behrens 10907b30a49SStefan Behrens if (WARN_ON_ONCE(!uuid_root)) { 11007b30a49SStefan Behrens ret = -EINVAL; 11107b30a49SStefan Behrens goto out; 11207b30a49SStefan Behrens } 11307b30a49SStefan Behrens 11407b30a49SStefan Behrens btrfs_uuid_to_key(uuid, type, &key); 11507b30a49SStefan Behrens 11607b30a49SStefan Behrens path = btrfs_alloc_path(); 11707b30a49SStefan Behrens if (!path) { 11807b30a49SStefan Behrens ret = -ENOMEM; 11907b30a49SStefan Behrens goto out; 12007b30a49SStefan Behrens } 12107b30a49SStefan Behrens 12207b30a49SStefan Behrens ret = btrfs_insert_empty_item(trans, uuid_root, path, &key, 12307b30a49SStefan Behrens sizeof(subid_le)); 12407b30a49SStefan Behrens if (ret >= 0) { 12507b30a49SStefan Behrens /* Add an item for the type for the first time */ 12607b30a49SStefan Behrens eb = path->nodes[0]; 12707b30a49SStefan Behrens slot = path->slots[0]; 12807b30a49SStefan Behrens offset = btrfs_item_ptr_offset(eb, slot); 12907b30a49SStefan Behrens } else if (ret == -EEXIST) { 13007b30a49SStefan Behrens /* 13107b30a49SStefan Behrens * An item with that type already exists. 13207b30a49SStefan Behrens * Extend the item and store the new subid at the end. 13307b30a49SStefan Behrens */ 13407b30a49SStefan Behrens btrfs_extend_item(uuid_root, path, sizeof(subid_le)); 13507b30a49SStefan Behrens eb = path->nodes[0]; 13607b30a49SStefan Behrens slot = path->slots[0]; 13707b30a49SStefan Behrens offset = btrfs_item_ptr_offset(eb, slot); 13807b30a49SStefan Behrens offset += btrfs_item_size_nr(eb, slot) - sizeof(subid_le); 13907b30a49SStefan Behrens } else if (ret < 0) { 140efe120a0SFrank Holton btrfs_warn(uuid_root->fs_info, "insert uuid item failed %d " 141efe120a0SFrank Holton "(0x%016llx, 0x%016llx) type %u!", 14207b30a49SStefan Behrens ret, (unsigned long long)key.objectid, 14307b30a49SStefan Behrens (unsigned long long)key.offset, type); 14407b30a49SStefan Behrens goto out; 14507b30a49SStefan Behrens } 14607b30a49SStefan Behrens 14707b30a49SStefan Behrens ret = 0; 14807b30a49SStefan Behrens subid_le = cpu_to_le64(subid_cpu); 14907b30a49SStefan Behrens write_extent_buffer(eb, &subid_le, offset, sizeof(subid_le)); 15007b30a49SStefan Behrens btrfs_mark_buffer_dirty(eb); 15107b30a49SStefan Behrens 15207b30a49SStefan Behrens out: 15307b30a49SStefan Behrens btrfs_free_path(path); 15407b30a49SStefan Behrens return ret; 15507b30a49SStefan Behrens } 15607b30a49SStefan Behrens 15707b30a49SStefan Behrens int btrfs_uuid_tree_rem(struct btrfs_trans_handle *trans, 15807b30a49SStefan Behrens struct btrfs_root *uuid_root, u8 *uuid, u8 type, 15907b30a49SStefan Behrens u64 subid) 16007b30a49SStefan Behrens { 16107b30a49SStefan Behrens int ret; 16207b30a49SStefan Behrens struct btrfs_path *path = NULL; 16307b30a49SStefan Behrens struct btrfs_key key; 16407b30a49SStefan Behrens struct extent_buffer *eb; 16507b30a49SStefan Behrens int slot; 16607b30a49SStefan Behrens unsigned long offset; 16707b30a49SStefan Behrens u32 item_size; 16807b30a49SStefan Behrens unsigned long move_dst; 16907b30a49SStefan Behrens unsigned long move_src; 17007b30a49SStefan Behrens unsigned long move_len; 17107b30a49SStefan Behrens 17207b30a49SStefan Behrens if (WARN_ON_ONCE(!uuid_root)) { 17307b30a49SStefan Behrens ret = -EINVAL; 17407b30a49SStefan Behrens goto out; 17507b30a49SStefan Behrens } 17607b30a49SStefan Behrens 17707b30a49SStefan Behrens btrfs_uuid_to_key(uuid, type, &key); 17807b30a49SStefan Behrens 17907b30a49SStefan Behrens path = btrfs_alloc_path(); 18007b30a49SStefan Behrens if (!path) { 18107b30a49SStefan Behrens ret = -ENOMEM; 18207b30a49SStefan Behrens goto out; 18307b30a49SStefan Behrens } 18407b30a49SStefan Behrens 18507b30a49SStefan Behrens ret = btrfs_search_slot(trans, uuid_root, &key, path, -1, 1); 18607b30a49SStefan Behrens if (ret < 0) { 187efe120a0SFrank Holton btrfs_warn(uuid_root->fs_info, "error %d while searching for uuid item!", 18807b30a49SStefan Behrens ret); 18907b30a49SStefan Behrens goto out; 19007b30a49SStefan Behrens } 19107b30a49SStefan Behrens if (ret > 0) { 19207b30a49SStefan Behrens ret = -ENOENT; 19307b30a49SStefan Behrens goto out; 19407b30a49SStefan Behrens } 19507b30a49SStefan Behrens 19607b30a49SStefan Behrens eb = path->nodes[0]; 19707b30a49SStefan Behrens slot = path->slots[0]; 19807b30a49SStefan Behrens offset = btrfs_item_ptr_offset(eb, slot); 19907b30a49SStefan Behrens item_size = btrfs_item_size_nr(eb, slot); 20007b30a49SStefan Behrens if (!IS_ALIGNED(item_size, sizeof(u64))) { 201efe120a0SFrank Holton btrfs_warn(uuid_root->fs_info, "uuid item with illegal size %lu!", 20207b30a49SStefan Behrens (unsigned long)item_size); 20307b30a49SStefan Behrens ret = -ENOENT; 20407b30a49SStefan Behrens goto out; 20507b30a49SStefan Behrens } 20607b30a49SStefan Behrens while (item_size) { 20707b30a49SStefan Behrens __le64 read_subid; 20807b30a49SStefan Behrens 20907b30a49SStefan Behrens read_extent_buffer(eb, &read_subid, offset, sizeof(read_subid)); 21007b30a49SStefan Behrens if (le64_to_cpu(read_subid) == subid) 21107b30a49SStefan Behrens break; 21207b30a49SStefan Behrens offset += sizeof(read_subid); 21307b30a49SStefan Behrens item_size -= sizeof(read_subid); 21407b30a49SStefan Behrens } 21507b30a49SStefan Behrens 21607b30a49SStefan Behrens if (!item_size) { 21707b30a49SStefan Behrens ret = -ENOENT; 21807b30a49SStefan Behrens goto out; 21907b30a49SStefan Behrens } 22007b30a49SStefan Behrens 22107b30a49SStefan Behrens item_size = btrfs_item_size_nr(eb, slot); 22207b30a49SStefan Behrens if (item_size == sizeof(subid)) { 22307b30a49SStefan Behrens ret = btrfs_del_item(trans, uuid_root, path); 22407b30a49SStefan Behrens goto out; 22507b30a49SStefan Behrens } 22607b30a49SStefan Behrens 22707b30a49SStefan Behrens move_dst = offset; 22807b30a49SStefan Behrens move_src = offset + sizeof(subid); 22907b30a49SStefan Behrens move_len = item_size - (move_src - btrfs_item_ptr_offset(eb, slot)); 23007b30a49SStefan Behrens memmove_extent_buffer(eb, move_dst, move_src, move_len); 23107b30a49SStefan Behrens btrfs_truncate_item(uuid_root, path, item_size - sizeof(subid), 1); 23207b30a49SStefan Behrens 23307b30a49SStefan Behrens out: 23407b30a49SStefan Behrens btrfs_free_path(path); 23507b30a49SStefan Behrens return ret; 23607b30a49SStefan Behrens } 23770f80175SStefan Behrens 23870f80175SStefan Behrens static int btrfs_uuid_iter_rem(struct btrfs_root *uuid_root, u8 *uuid, u8 type, 23970f80175SStefan Behrens u64 subid) 24070f80175SStefan Behrens { 24170f80175SStefan Behrens struct btrfs_trans_handle *trans; 24270f80175SStefan Behrens int ret; 24370f80175SStefan Behrens 24470f80175SStefan Behrens /* 1 - for the uuid item */ 24570f80175SStefan Behrens trans = btrfs_start_transaction(uuid_root, 1); 24670f80175SStefan Behrens if (IS_ERR(trans)) { 24770f80175SStefan Behrens ret = PTR_ERR(trans); 24870f80175SStefan Behrens goto out; 24970f80175SStefan Behrens } 25070f80175SStefan Behrens 25170f80175SStefan Behrens ret = btrfs_uuid_tree_rem(trans, uuid_root, uuid, type, subid); 25270f80175SStefan Behrens btrfs_end_transaction(trans, uuid_root); 25370f80175SStefan Behrens 25470f80175SStefan Behrens out: 25570f80175SStefan Behrens return ret; 25670f80175SStefan Behrens } 25770f80175SStefan Behrens 25870f80175SStefan Behrens int btrfs_uuid_tree_iterate(struct btrfs_fs_info *fs_info, 25970f80175SStefan Behrens int (*check_func)(struct btrfs_fs_info *, u8 *, u8, 26070f80175SStefan Behrens u64)) 26170f80175SStefan Behrens { 26270f80175SStefan Behrens struct btrfs_root *root = fs_info->uuid_root; 26370f80175SStefan Behrens struct btrfs_key key; 26470f80175SStefan Behrens struct btrfs_path *path; 26570f80175SStefan Behrens int ret = 0; 26670f80175SStefan Behrens struct extent_buffer *leaf; 26770f80175SStefan Behrens int slot; 26870f80175SStefan Behrens u32 item_size; 26970f80175SStefan Behrens unsigned long offset; 27070f80175SStefan Behrens 27170f80175SStefan Behrens path = btrfs_alloc_path(); 27270f80175SStefan Behrens if (!path) { 27370f80175SStefan Behrens ret = -ENOMEM; 27470f80175SStefan Behrens goto out; 27570f80175SStefan Behrens } 27670f80175SStefan Behrens 27770f80175SStefan Behrens key.objectid = 0; 27870f80175SStefan Behrens key.type = 0; 27970f80175SStefan Behrens key.offset = 0; 28070f80175SStefan Behrens 28170f80175SStefan Behrens again_search_slot: 28270f80175SStefan Behrens path->keep_locks = 1; 2836174d3cbSFilipe David Borba Manana ret = btrfs_search_forward(root, &key, path, 0); 28470f80175SStefan Behrens if (ret) { 28570f80175SStefan Behrens if (ret > 0) 28670f80175SStefan Behrens ret = 0; 28770f80175SStefan Behrens goto out; 28870f80175SStefan Behrens } 28970f80175SStefan Behrens 29070f80175SStefan Behrens while (1) { 29170f80175SStefan Behrens cond_resched(); 29270f80175SStefan Behrens leaf = path->nodes[0]; 29370f80175SStefan Behrens slot = path->slots[0]; 29470f80175SStefan Behrens btrfs_item_key_to_cpu(leaf, &key, slot); 29570f80175SStefan Behrens 29670f80175SStefan Behrens if (key.type != BTRFS_UUID_KEY_SUBVOL && 29770f80175SStefan Behrens key.type != BTRFS_UUID_KEY_RECEIVED_SUBVOL) 29870f80175SStefan Behrens goto skip; 29970f80175SStefan Behrens 30070f80175SStefan Behrens offset = btrfs_item_ptr_offset(leaf, slot); 30170f80175SStefan Behrens item_size = btrfs_item_size_nr(leaf, slot); 30270f80175SStefan Behrens if (!IS_ALIGNED(item_size, sizeof(u64))) { 303efe120a0SFrank Holton btrfs_warn(fs_info, "uuid item with illegal size %lu!", 30470f80175SStefan Behrens (unsigned long)item_size); 30570f80175SStefan Behrens goto skip; 30670f80175SStefan Behrens } 30770f80175SStefan Behrens while (item_size) { 30870f80175SStefan Behrens u8 uuid[BTRFS_UUID_SIZE]; 30970f80175SStefan Behrens __le64 subid_le; 31070f80175SStefan Behrens u64 subid_cpu; 31170f80175SStefan Behrens 31270f80175SStefan Behrens put_unaligned_le64(key.objectid, uuid); 31370f80175SStefan Behrens put_unaligned_le64(key.offset, uuid + sizeof(u64)); 31470f80175SStefan Behrens read_extent_buffer(leaf, &subid_le, offset, 31570f80175SStefan Behrens sizeof(subid_le)); 31670f80175SStefan Behrens subid_cpu = le64_to_cpu(subid_le); 31770f80175SStefan Behrens ret = check_func(fs_info, uuid, key.type, subid_cpu); 31870f80175SStefan Behrens if (ret < 0) 31970f80175SStefan Behrens goto out; 32070f80175SStefan Behrens if (ret > 0) { 32170f80175SStefan Behrens btrfs_release_path(path); 32270f80175SStefan Behrens ret = btrfs_uuid_iter_rem(root, uuid, key.type, 32370f80175SStefan Behrens subid_cpu); 32470f80175SStefan Behrens if (ret == 0) { 32570f80175SStefan Behrens /* 32670f80175SStefan Behrens * this might look inefficient, but the 32770f80175SStefan Behrens * justification is that it is an 32870f80175SStefan Behrens * exception that check_func returns 1, 32970f80175SStefan Behrens * and that in the regular case only one 33070f80175SStefan Behrens * entry per UUID exists. 33170f80175SStefan Behrens */ 33270f80175SStefan Behrens goto again_search_slot; 33370f80175SStefan Behrens } 33470f80175SStefan Behrens if (ret < 0 && ret != -ENOENT) 33570f80175SStefan Behrens goto out; 33670f80175SStefan Behrens } 33770f80175SStefan Behrens item_size -= sizeof(subid_le); 33870f80175SStefan Behrens offset += sizeof(subid_le); 33970f80175SStefan Behrens } 34070f80175SStefan Behrens 34170f80175SStefan Behrens skip: 34270f80175SStefan Behrens ret = btrfs_next_item(root, path); 34370f80175SStefan Behrens if (ret == 0) 34470f80175SStefan Behrens continue; 34570f80175SStefan Behrens else if (ret > 0) 34670f80175SStefan Behrens ret = 0; 34770f80175SStefan Behrens break; 34870f80175SStefan Behrens } 34970f80175SStefan Behrens 35070f80175SStefan Behrens out: 35170f80175SStefan Behrens btrfs_free_path(path); 35270f80175SStefan Behrens if (ret) 353efe120a0SFrank Holton btrfs_warn(fs_info, "btrfs_uuid_tree_iterate failed %d", ret); 35470f80175SStefan Behrens return 0; 35570f80175SStefan Behrens } 356