btrfs-progs: read_tree_block() and read_node_slot() cleanup.
[platform/upstream/btrfs-progs.git] / btrfs-debug-tree.c
1 /*
2  * Copyright (C) 2007 Oracle.  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 <uuid/uuid.h>
23 #include "kerncompat.h"
24 #include "radix-tree.h"
25 #include "ctree.h"
26 #include "disk-io.h"
27 #include "print-tree.h"
28 #include "transaction.h"
29 #include "utils.h"
30
31 static int print_usage(void)
32 {
33         fprintf(stderr, "usage: btrfs-debug-tree [-e] [-d] [-r] [-R] [-u]\n");
34         fprintf(stderr, "                        [-b block_num ] device\n");
35         fprintf(stderr, "\t-e : print detailed extents info\n");
36         fprintf(stderr, "\t-d : print info of btrfs device and root tree dirs"
37                     " only\n");
38         fprintf(stderr, "\t-r : print info of roots only\n");
39         fprintf(stderr, "\t-R : print info of roots and root backups\n");
40         fprintf(stderr, "\t-u : print info of uuid tree only\n");
41         fprintf(stderr, "\t-b block_num : print info of the specified block"
42                     " only\n");
43         fprintf(stderr,
44                 "\t-t tree_id : print only the tree with the given id\n");
45         fprintf(stderr, "%s\n", PACKAGE_STRING);
46         exit(1);
47 }
48
49 static void print_extents(struct btrfs_root *root, struct extent_buffer *eb)
50 {
51         int i;
52         u32 nr;
53         u32 size;
54
55         if (!eb)
56                 return;
57
58         if (btrfs_is_leaf(eb)) {
59                 btrfs_print_leaf(root, eb);
60                 return;
61         }
62
63         size = btrfs_level_size(root, btrfs_header_level(eb) - 1);
64         nr = btrfs_header_nritems(eb);
65         for (i = 0; i < nr; i++) {
66                 struct extent_buffer *next = read_tree_block(root,
67                                              btrfs_node_blockptr(eb, i),
68                                              size,
69                                              btrfs_node_ptr_generation(eb, i));
70                 if (!extent_buffer_uptodate(next))
71                         continue;
72                 if (btrfs_is_leaf(next) &&
73                     btrfs_header_level(eb) != 1)
74                         BUG();
75                 if (btrfs_header_level(next) !=
76                         btrfs_header_level(eb) - 1)
77                         BUG();
78                 print_extents(root, next);
79                 free_extent_buffer(next);
80         }
81 }
82
83 static void print_old_roots(struct btrfs_super_block *super)
84 {
85         struct btrfs_root_backup *backup;
86         int i;
87
88         for (i = 0; i < BTRFS_NUM_BACKUP_ROOTS; i++) {
89                 backup = super->super_roots + i;
90                 printf("btrfs root backup slot %d\n", i);
91                 printf("\ttree root gen %llu block %llu\n",
92                        (unsigned long long)btrfs_backup_tree_root_gen(backup),
93                        (unsigned long long)btrfs_backup_tree_root(backup));
94
95                 printf("\t\textent root gen %llu block %llu\n",
96                        (unsigned long long)btrfs_backup_extent_root_gen(backup),
97                        (unsigned long long)btrfs_backup_extent_root(backup));
98
99                 printf("\t\tchunk root gen %llu block %llu\n",
100                        (unsigned long long)btrfs_backup_chunk_root_gen(backup),
101                        (unsigned long long)btrfs_backup_chunk_root(backup));
102
103                 printf("\t\tdevice root gen %llu block %llu\n",
104                        (unsigned long long)btrfs_backup_dev_root_gen(backup),
105                        (unsigned long long)btrfs_backup_dev_root(backup));
106
107                 printf("\t\tcsum root gen %llu block %llu\n",
108                        (unsigned long long)btrfs_backup_csum_root_gen(backup),
109                        (unsigned long long)btrfs_backup_csum_root(backup));
110
111                 printf("\t\tfs root gen %llu block %llu\n",
112                        (unsigned long long)btrfs_backup_fs_root_gen(backup),
113                        (unsigned long long)btrfs_backup_fs_root(backup));
114
115                 printf("\t\t%llu used %llu total %llu devices\n",
116                        (unsigned long long)btrfs_backup_bytes_used(backup),
117                        (unsigned long long)btrfs_backup_total_bytes(backup),
118                        (unsigned long long)btrfs_backup_num_devices(backup));
119         }
120 }
121
122 int main(int ac, char **av)
123 {
124         struct btrfs_root *root;
125         struct btrfs_fs_info *info;
126         struct btrfs_path path;
127         struct btrfs_key key;
128         struct btrfs_root_item ri;
129         struct extent_buffer *leaf;
130         struct btrfs_disk_key disk_key;
131         struct btrfs_key found_key;
132         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
133         int ret;
134         int slot;
135         int extent_only = 0;
136         int device_only = 0;
137         int uuid_tree_only = 0;
138         int roots_only = 0;
139         int root_backups = 0;
140         u64 block_only = 0;
141         struct btrfs_root *tree_root_scan;
142         u64 tree_id = 0;
143
144         radix_tree_init();
145
146         while(1) {
147                 int c;
148                 c = getopt(ac, av, "deb:rRut:");
149                 if (c < 0)
150                         break;
151                 switch(c) {
152                         case 'e':
153                                 extent_only = 1;
154                                 break;
155                         case 'd':
156                                 device_only = 1;
157                                 break;
158                         case 'r':
159                                 roots_only = 1;
160                                 break;
161                         case 'u':
162                                 uuid_tree_only = 1;
163                                 break;
164                         case 'R':
165                                 roots_only = 1;
166                                 root_backups = 1;
167                                 break;
168                         case 'b':
169                                 block_only = arg_strtou64(optarg);
170                                 break;
171                         case 't':
172                                 tree_id = arg_strtou64(optarg);
173                                 break;
174                         default:
175                                 print_usage();
176                 }
177         }
178         set_argv0(av);
179         ac = ac - optind;
180         if (check_argc_exact(ac, 1))
181                 print_usage();
182
183         ret = check_arg_type(av[optind]);
184         if (ret != BTRFS_ARG_BLKDEV && ret != BTRFS_ARG_REG) {
185                 fprintf(stderr, "'%s' is not a block device or regular file\n",
186                         av[optind]);
187                 exit(1);
188         }
189
190         info = open_ctree_fs_info(av[optind], 0, 0, OPEN_CTREE_PARTIAL);
191         if (!info) {
192                 fprintf(stderr, "unable to open %s\n", av[optind]);
193                 exit(1);
194         }
195
196         root = info->fs_root;
197         if (!root) {
198                 fprintf(stderr, "unable to open %s\n", av[optind]);
199                 exit(1);
200         }
201
202         if (block_only) {
203                 leaf = read_tree_block(root,
204                                       block_only,
205                                       root->leafsize, 0);
206
207                 if (extent_buffer_uptodate(leaf) &&
208                     btrfs_header_level(leaf) != 0) {
209                         free_extent_buffer(leaf);
210                         leaf = NULL;
211                 }
212
213                 if (!leaf) {
214                         leaf = read_tree_block(root,
215                                               block_only,
216                                               root->nodesize, 0);
217                 }
218                 if (!extent_buffer_uptodate(leaf)) {
219                         fprintf(stderr, "failed to read %llu\n",
220                                 (unsigned long long)block_only);
221                         goto close_root;
222                 }
223                 btrfs_print_tree(root, leaf, 0);
224                 goto close_root;
225         }
226
227         if (!(extent_only || uuid_tree_only || tree_id)) {
228                 if (roots_only) {
229                         printf("root tree: %llu level %d\n",
230                              (unsigned long long)info->tree_root->node->start,
231                              btrfs_header_level(info->tree_root->node));
232                         printf("chunk tree: %llu level %d\n",
233                              (unsigned long long)info->chunk_root->node->start,
234                              btrfs_header_level(info->chunk_root->node));
235                 } else {
236                         if (info->tree_root->node) {
237                                 printf("root tree\n");
238                                 btrfs_print_tree(info->tree_root,
239                                                  info->tree_root->node, 1);
240                         }
241
242                         if (info->chunk_root->node) {
243                                 printf("chunk tree\n");
244                                 btrfs_print_tree(info->chunk_root,
245                                                  info->chunk_root->node, 1);
246                         }
247                 }
248         }
249         tree_root_scan = info->tree_root;
250
251         btrfs_init_path(&path);
252 again:
253         if (!extent_buffer_uptodate(tree_root_scan->node))
254                 goto no_node;
255
256         key.offset = 0;
257         key.objectid = 0;
258         btrfs_set_key_type(&key, BTRFS_ROOT_ITEM_KEY);
259         ret = btrfs_search_slot(NULL, tree_root_scan, &key, &path, 0, 0);
260         BUG_ON(ret < 0);
261         while(1) {
262                 leaf = path.nodes[0];
263                 slot = path.slots[0];
264                 if (slot >= btrfs_header_nritems(leaf)) {
265                         ret = btrfs_next_leaf(tree_root_scan, &path);
266                         if (ret != 0)
267                                 break;
268                         leaf = path.nodes[0];
269                         slot = path.slots[0];
270                 }
271                 btrfs_item_key(leaf, &disk_key, path.slots[0]);
272                 btrfs_disk_key_to_cpu(&found_key, &disk_key);
273                 if (btrfs_key_type(&found_key) == BTRFS_ROOT_ITEM_KEY) {
274                         unsigned long offset;
275                         struct extent_buffer *buf;
276                         int skip = extent_only | device_only | uuid_tree_only;
277
278                         offset = btrfs_item_ptr_offset(leaf, slot);
279                         read_extent_buffer(leaf, &ri, offset, sizeof(ri));
280                         buf = read_tree_block(tree_root_scan,
281                                               btrfs_root_bytenr(&ri),
282                                               btrfs_level_size(tree_root_scan,
283                                                         btrfs_root_level(&ri)),
284                                               0);
285                         if (!extent_buffer_uptodate(buf))
286                                 goto next;
287                         if (tree_id && found_key.objectid != tree_id)
288                                 goto next;
289
290                         switch(found_key.objectid) {
291                         case BTRFS_ROOT_TREE_OBJECTID:
292                                 if (!skip)
293                                         printf("root");
294                                 break;
295                         case BTRFS_EXTENT_TREE_OBJECTID:
296                                 if (!device_only && !uuid_tree_only)
297                                         skip = 0;
298                                 if (!skip)
299                                         printf("extent");
300                                 break;
301                         case BTRFS_CHUNK_TREE_OBJECTID:
302                                 if (!skip) {
303                                         printf("chunk");
304                                 }
305                                 break;
306                         case BTRFS_DEV_TREE_OBJECTID:
307                                 if (!uuid_tree_only)
308                                         skip = 0;
309                                 if (!skip)
310                                         printf("device");
311                                 break;
312                         case BTRFS_FS_TREE_OBJECTID:
313                                 if (!skip) {
314                                         printf("fs");
315                                 }
316                                 break;
317                         case BTRFS_ROOT_TREE_DIR_OBJECTID:
318                                 skip = 0;
319                                 printf("directory");
320                                 break;
321                         case BTRFS_CSUM_TREE_OBJECTID:
322                                 if (!skip) {
323                                         printf("checksum");
324                                 }
325                                 break;
326                         case BTRFS_ORPHAN_OBJECTID:
327                                 if (!skip) {
328                                         printf("orphan");
329                                 }
330                                 break;
331                         case BTRFS_TREE_LOG_OBJECTID:
332                                 if (!skip) {
333                                         printf("log");
334                                 }
335                                 break;
336                         case BTRFS_TREE_LOG_FIXUP_OBJECTID:
337                                 if (!skip) {
338                                         printf("log fixup");
339                                 }
340                                 break;
341                         case BTRFS_TREE_RELOC_OBJECTID:
342                                 if (!skip) {
343                                         printf("reloc");
344                                 }
345                                 break;
346                         case BTRFS_DATA_RELOC_TREE_OBJECTID:
347                                 if (!skip) {
348                                         printf("data reloc");
349                                 }
350                                 break;
351                         case BTRFS_EXTENT_CSUM_OBJECTID:
352                                 if (!skip) {
353                                         printf("extent checksum");
354                                 }
355                                 break;
356                         case BTRFS_QUOTA_TREE_OBJECTID:
357                                 if (!skip) {
358                                         printf("quota");
359                                 }
360                                 break;
361                         case BTRFS_UUID_TREE_OBJECTID:
362                                 if (!extent_only && !device_only)
363                                         skip = 0;
364                                 if (!skip)
365                                         printf("uuid");
366                                 break;
367                         case BTRFS_MULTIPLE_OBJECTIDS:
368                                 if (!skip) {
369                                         printf("multiple");
370                                 }
371                                 break;
372                         default:
373                                 if (!skip) {
374                                         printf("file");
375                                 }
376                         }
377                         if (extent_only && !skip) {
378                                 print_extents(tree_root_scan, buf);
379                         } else if (!skip) {
380                                 printf(" tree ");
381                                 btrfs_print_key(&disk_key);
382                                 if (roots_only) {
383                                         printf(" %llu level %d\n",
384                                                (unsigned long long)buf->start,
385                                                btrfs_header_level(buf));
386                                 } else {
387                                         printf(" \n");
388                                         btrfs_print_tree(tree_root_scan, buf, 1);
389                                 }
390                         }
391                         free_extent_buffer(buf);
392                 }
393 next:
394                 path.slots[0]++;
395         }
396 no_node:
397         btrfs_release_path(&path);
398
399         if (tree_root_scan == info->tree_root &&
400             info->log_root_tree) {
401                 tree_root_scan = info->log_root_tree;
402                 goto again;
403         }
404
405         if (extent_only || device_only || uuid_tree_only)
406                 goto close_root;
407
408         if (root_backups)
409                 print_old_roots(info->super_copy);
410
411         printf("total bytes %llu\n",
412                (unsigned long long)btrfs_super_total_bytes(info->super_copy));
413         printf("bytes used %llu\n",
414                (unsigned long long)btrfs_super_bytes_used(info->super_copy));
415         uuidbuf[BTRFS_UUID_UNPARSED_SIZE - 1] = '\0';
416         uuid_unparse(info->super_copy->fsid, uuidbuf);
417         printf("uuid %s\n", uuidbuf);
418         printf("%s\n", PACKAGE_STRING);
419 close_root:
420         return close_ctree(root);
421 }