btrfs-find-root.c

   1 /*
   2  * Copyright (C) 2011 Red Hat.  All rights reserved.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public
   6  * License v2 as published by the Free Software Foundation.
   7  *
   8  * This program is distributed in the hope that it will be useful,
   9  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  10  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11  * General Public License for more details.
  12  *
  13  * You should have received a copy of the GNU General Public
  14  * License along with this program; if not, write to the
  15  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  16  * Boston, MA 021110-1307, USA.
  17  */
  18
  19 #include <stdio.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <fcntl.h>
  23 #include <sys/stat.h>
  24 #include <zlib.h>
  25 #include <getopt.h>
  26
  27 #include "kerncompat.h"
  28 #include "ctree.h"
  29 #include "disk-io.h"
  30 #include "print-tree.h"
  31 #include "transaction.h"
  32 #include "list.h"
  33 #include "volumes.h"
  34 #include "utils.h"
  35 #include "crc32c.h"
  36 #include "extent-cache.h"
  37 #include "find-root.h"
  38
  39 static void usage(void)
  40 {
  41         fprintf(stderr, "Usage: find-roots [-a] [-o search_objectid] "
  42                 "[ -g search_generation ] [ -l search_level ] <device>\n");
  43 }
  44
  45 /*
  46  * Get reliable generation and level for given root.
  47  *
  48  * We have two sources of gen/level: superblock and tree root.
  49  * superblock include the following level:
  50  *   Root, chunk, log
  51  * and the following generations:
  52  *   Root, chunk, uuid
  53  * Other gen/leven can only be read from its btrfs_tree_root if possible.
  54  *
  55  * Currently we only believe things from superblock.
  56  */
  57 static void get_root_gen_and_level(u64 objectid, struct btrfs_fs_info *fs_info,
  58                                    u64 *ret_gen, u8 *ret_level)
  59 {
  60         struct btrfs_super_block *super = fs_info->super_copy;
  61         u64 gen = (u64)-1;
  62         u8 level = (u8)-1;
  63
  64         switch (objectid) {
  65         case BTRFS_ROOT_TREE_OBJECTID:
  66                 level = btrfs_super_root_level(super);
  67                 gen = btrfs_super_generation(super);
  68                 break;
  69         case BTRFS_CHUNK_TREE_OBJECTID:
  70                 level = btrfs_super_chunk_root_level(super);
  71                 gen = btrfs_super_chunk_root_generation(super);
  72                 printf("Search for chunk root is not supported yet\n");
  73                 break;
  74         case BTRFS_TREE_LOG_OBJECTID:
  75                 level = btrfs_super_log_root_level(super);
  76                 gen = btrfs_super_log_root_transid(super);
  77                 break;
  78         case BTRFS_UUID_TREE_OBJECTID:
  79                 gen = btrfs_super_uuid_tree_generation(super);
  80                 break;
  81         }
  82         if (gen != (u64)-1) {
  83                 printf("Superblock thinks the generation is %llu\n", gen);
  84                 if (ret_gen)
  85                         *ret_gen = gen;
  86         } else {
  87                 printf("Superblock doesn't contain generation info for root %llu\n",
  88                        objectid);
  89         }
  90         if (level != (u8)-1) {
  91                 printf("Superblock thinks the level is %u\n", level);
  92                 if (ret_level)
  93                         *ret_level = level;
  94         } else {
  95                 printf("Superblock doesn't contain the level info for root %llu\n",
  96                        objectid);
  97         }
  98 }
  99
 100 static void print_one_result(struct cache_extent *tree_block,
 101                              u8 level, u64 generation,
 102                              struct btrfs_find_root_filter *filter)
 103 {
 104         int unsure = 0;
 105
 106         if (filter->match_gen == (u64)-1 || filter->match_level == (u8)-1)
 107                 unsure = 1;
 108         printf("Well block %llu(gen: %llu level: %u) seems good, ",
 109                tree_block->start, generation, level);
 110         if (unsure)
 111                 printf("but we are unsure about the correct generation/level\n");
 112         else
 113                 printf("but generation/level doesn't match, want gen: %llu level: %u\n",
 114                        filter->match_gen, filter->match_level);
 115 }
 116
 117 static void print_find_root_result(struct cache_tree *result,
 118                                    struct btrfs_find_root_filter *filter)
 119 {
 120         struct btrfs_find_root_gen_cache *gen_cache;
 121         struct cache_extent *cache;
 122         struct cache_extent *tree_block;
 123         u64 generation = 0;
 124         u8 level = 0;
 125
 126         for (cache = last_cache_extent(result);
 127              cache; cache = prev_cache_extent(cache)) {
 128                 gen_cache = container_of(cache,
 129                                 struct btrfs_find_root_gen_cache, cache);
 130                 level = gen_cache->highest_level;
 131                 generation = cache->start;
 132                 if (level == filter->match_level &&
 133                     generation == filter->match_gen)
 134                         continue;
 135                 for (tree_block = last_cache_extent(&gen_cache->eb_tree);
 136                      tree_block; tree_block = prev_cache_extent(tree_block))
 137                         print_one_result(tree_block, level, generation, filter);
 138         }
 139 }
 140
 141 int main(int argc, char **argv)
 142 {
 143         struct btrfs_root *root;
 144         struct btrfs_find_root_filter filter = {0};
 145         struct cache_tree result;
 146         struct cache_extent *found;
 147         int ret;
 148
 149         /* Default to search root tree */
 150         filter.objectid = BTRFS_ROOT_TREE_OBJECTID;
 151         filter.match_gen = (u64)-1;
 152         filter.match_level = (u8)-1;
 153         while (1) {
 154                 static const struct option long_options[] = {
 155                         { "help", no_argument, NULL, GETOPT_VAL_HELP},
 156                         { NULL, 0, NULL, 0 }
 157                 };
 158                 int c = getopt_long(argc, argv, "al:o:g:", long_options, NULL);
 159
 160                 if (c < 0)
 161                         break;
 162
 163                 switch (c) {
 164                 case 'a':
 165                         filter.search_all = 1;
 166                         break;
 167                 case 'o':
 168                         filter.objectid = arg_strtou64(optarg);
 169                         break;
 170                 case 'g':
 171                         filter.generation = arg_strtou64(optarg);
 172                         break;
 173                 case 'l':
 174                         filter.level = arg_strtou64(optarg);
 175                         break;
 176                 case GETOPT_VAL_HELP:
 177                 default:
 178                         usage();
 179                         exit(c != GETOPT_VAL_HELP);
 180                 }
 181         }
 182
 183         set_argv0(argv);
 184         argc = argc - optind;
 185         if (check_argc_min(argc, 1)) {
 186                 usage();
 187                 exit(1);
 188         }
 189
 190         root = open_ctree(argv[optind], 0, OPEN_CTREE_CHUNK_ROOT_ONLY);
 191         if (!root) {
 192                 fprintf(stderr, "Open ctree failed\n");
 193                 exit(1);
 194         }
 195         cache_tree_init(&result);
 196
 197         get_root_gen_and_level(filter.objectid, root->fs_info,
 198                                &filter.match_gen, &filter.match_level);
 199         ret = btrfs_find_root_search(root, &filter, &result, &found);
 200         if (ret < 0) {
 201                 fprintf(stderr, "Fail to search the tree root: %s\n",
 202                         strerror(-ret));
 203                 goto out;
 204         }
 205         if (ret > 0) {
 206                 printf("Found tree root at %llu gen %llu level %u\n",
 207                        found->start, filter.match_gen, filter.match_level);
 208                 ret = 0;
 209         }
 210         print_find_root_result(&result, &filter);
 211 out:
 212         btrfs_find_root_free(&result);
 213         close_ctree(root);
 214         return ret;
 215 }