Btrfs-progs: fix wrong leaf when checking the trees relationship
[platform/upstream/btrfs-progs.git] / btrfs-list.c
index ab73e65..c53d016 100644 (file)
@@ -183,6 +183,8 @@ static int add_root(struct root_lookup *root_lookup,
        ri->root_id = root_id;
        ri->ref_tree = ref_tree;
        strncpy(ri->name, name, name_len);
+       if (name_len > 0)
+               ri->name[name_len] = 0;
 
        ret = tree_insert(&root_lookup->root, root_id, ref_tree, &ri->rb_node);
        if (ret) {
@@ -199,10 +201,9 @@ static int add_root(struct root_lookup *root_lookup,
  * This can't be called until all the root_info->path fields are filled
  * in by lookup_ino_path
  */
-static int resolve_root(struct root_lookup *rl, struct root_info *ri, int print_parent)
+static int resolve_root(struct root_lookup *rl, struct root_info *ri,
+                       u64 *parent_id, u64 *top_id, char **path)
 {
-       u64 top_id;
-       u64 parent_id = 0;
        char *full_path = NULL;
        int len = 0;
        struct root_info *found;
@@ -211,6 +212,7 @@ static int resolve_root(struct root_lookup *rl, struct root_info *ri, int print_
         * we go backwards from the root_info object and add pathnames
         * from parent directories as we go.
         */
+       *parent_id = 0;
        found = ri;
        while (1) {
                char *tmp;
@@ -234,13 +236,12 @@ static int resolve_root(struct root_lookup *rl, struct root_info *ri, int print_
 
                next = found->ref_tree;
                /* record the first parent */
-               if ( parent_id == 0 ) {
-                       parent_id = next;
-               }
+               if (*parent_id == 0)
+                       *parent_id = next;
 
                /* if the ref_tree refers to ourselves, we're at the top */
                if (next == found->root_id) {
-                       top_id = next;
+                       *top_id = next;
                        break;
                }
 
@@ -250,20 +251,13 @@ static int resolve_root(struct root_lookup *rl, struct root_info *ri, int print_
                 */
                found = tree_search(&rl->root, next);
                if (!found) {
-                       top_id = next;
+                       *top_id = next;
                        break;
                }
        }
-       if (print_parent) {
-               printf("ID %llu parent %llu top level %llu path %s\n",
-                      (unsigned long long)ri->root_id, (unsigned long long)parent_id, (unsigned long long)top_id,
-                       full_path);
-       } else {
-               printf("ID %llu top level %llu path %s\n",
-                      (unsigned long long)ri->root_id, (unsigned long long)top_id,
-                       full_path);
-       }
-       free(full_path);
+
+       *path = full_path;
+
        return 0;
 }
 
@@ -560,10 +554,62 @@ build:
        return full;
 }
 
-int list_subvols(int fd, int print_parent)
+static int get_default_subvolid(int fd, u64 *default_id)
+{
+       struct btrfs_ioctl_search_args args;
+       struct btrfs_ioctl_search_key *sk = &args.key;
+       struct btrfs_ioctl_search_header *sh;
+       u64 found = 0;
+       int ret;
+
+       memset(&args, 0, sizeof(args));
+
+       /*
+        * search for a dir item with a name 'default' in the tree of
+        * tree roots, it should point us to a default root
+        */
+       sk->tree_id = 1;
+
+       /* don't worry about ancient format and request only one item */
+       sk->nr_items = 1;
+
+       sk->max_objectid = BTRFS_ROOT_TREE_DIR_OBJECTID;
+       sk->min_objectid = BTRFS_ROOT_TREE_DIR_OBJECTID;
+       sk->max_type = BTRFS_DIR_ITEM_KEY;
+       sk->min_type = BTRFS_DIR_ITEM_KEY;
+       sk->max_offset = (u64)-1;
+       sk->max_transid = (u64)-1;
+
+       ret = ioctl(fd, BTRFS_IOC_TREE_SEARCH, &args);
+       if (ret < 0)
+               return ret;
+
+       /* the ioctl returns the number of items it found in nr_items */
+       if (sk->nr_items == 0)
+               goto out;
+
+       sh = (struct btrfs_ioctl_search_header *)args.buf;
+
+       if (sh->type == BTRFS_DIR_ITEM_KEY) {
+               struct btrfs_dir_item *di;
+               int name_len;
+               char *name;
+
+               di = (struct btrfs_dir_item *)(sh + 1);
+               name_len = btrfs_stack_dir_name_len(di);
+               name = (char *)(di + 1);
+
+               if (!strncmp("default", name, name_len))
+                       found = btrfs_disk_key_objectid(&di->location);
+       }
+
+out:
+       *default_id = found;
+       return 0;
+}
+
+static int __list_subvol_search(int fd, struct root_lookup *root_lookup)
 {
-       struct root_lookup root_lookup;
-       struct rb_node *n;
        int ret;
        struct btrfs_ioctl_search_args args;
        struct btrfs_ioctl_search_key *sk = &args.key;
@@ -574,10 +620,8 @@ int list_subvols(int fd, int print_parent)
        char *name;
        u64 dir_id;
        int i;
-       int e;
-
-       root_lookup_init(&root_lookup);
 
+       root_lookup_init(root_lookup);
        memset(&args, 0, sizeof(args));
 
        /* search in the tree of tree roots */
@@ -603,12 +647,8 @@ int list_subvols(int fd, int print_parent)
 
        while(1) {
                ret = ioctl(fd, BTRFS_IOC_TREE_SEARCH, &args);
-               e = errno;
-               if (ret < 0) {
-                       fprintf(stderr, "ERROR: can't perform the search - %s\n",
-                               strerror(e));
+               if (ret < 0)
                        return ret;
-               }
                /* the ioctl returns the number of item it found in nr_items */
                if (sk->nr_items == 0)
                        break;
@@ -629,7 +669,7 @@ int list_subvols(int fd, int print_parent)
                                name = (char *)(ref + 1);
                                dir_id = btrfs_stack_root_ref_dirid(ref);
 
-                               add_root(&root_lookup, sh->objectid, sh->offset,
+                               add_root(root_lookup, sh->objectid, sh->offset,
                                         dir_id, name, name_len);
                        }
 
@@ -647,18 +687,25 @@ int list_subvols(int fd, int print_parent)
                /* this iteration is done, step forward one root for the next
                 * ioctl
                 */
-               if (sk->min_objectid < (u64)-1) {
+               if (sk->min_type < BTRFS_ROOT_BACKREF_KEY) {
+                       sk->min_type = BTRFS_ROOT_BACKREF_KEY;
+                       sk->min_offset = 0;
+               } else  if (sk->min_objectid < (u64)-1) {
                        sk->min_objectid++;
                        sk->min_type = BTRFS_ROOT_BACKREF_KEY;
                        sk->min_offset = 0;
                } else
                        break;
        }
-       /*
-        * now we have an rbtree full of root_info objects, but we need to fill
-        * in their path names within the subvol that is referencing each one.
-        */
-       n = rb_first(&root_lookup.root);
+
+       return 0;
+}
+
+static int __list_subvol_fill_paths(int fd, struct root_lookup *root_lookup)
+{
+       struct rb_node *n;
+
+       n = rb_first(&root_lookup->root);
        while (n) {
                struct root_info *entry;
                int ret;
@@ -669,6 +716,50 @@ int list_subvols(int fd, int print_parent)
                n = rb_next(n);
        }
 
+       return 0;
+}
+
+int list_subvols(int fd, int print_parent, int get_default)
+{
+       struct root_lookup root_lookup;
+       struct rb_node *n;
+       u64 default_id;
+       int ret;
+
+       if (get_default) {
+               ret = get_default_subvolid(fd, &default_id);
+               if (ret) {
+                       fprintf(stderr, "ERROR: can't perform the search - %s\n",
+                               strerror(errno));
+                       return ret;
+               }
+               if (default_id == 0) {
+                       fprintf(stderr, "ERROR: 'default' dir item not found\n");
+                       return ret;
+               }
+
+               /* no need to resolve roots if FS_TREE is default */
+               if (default_id == BTRFS_FS_TREE_OBJECTID) {
+                       printf("ID 5 (FS_TREE)\n");
+                       return ret;
+               }
+       }
+
+       ret = __list_subvol_search(fd, &root_lookup);
+       if (ret) {
+               fprintf(stderr, "ERROR: can't perform the search - %s\n",
+                               strerror(errno));
+               return ret;
+       }
+
+       /*
+        * now we have an rbtree full of root_info objects, but we need to fill
+        * in their path names within the subvol that is referencing each one.
+        */
+       ret = __list_subvol_fill_paths(fd, &root_lookup);
+       if (ret < 0)
+               return ret;
+
        /* now that we have all the subvol-relative paths filled in,
         * we have to string the subvols together so that we can get
         * a path all the way back to the FS root
@@ -676,8 +767,29 @@ int list_subvols(int fd, int print_parent)
        n = rb_last(&root_lookup.root);
        while (n) {
                struct root_info *entry;
+               u64 level;
+               u64 parent_id;
+               char *path;
+
                entry = rb_entry(n, struct root_info, rb_node);
-               resolve_root(&root_lookup, entry, print_parent);
+               if (get_default && entry->root_id != default_id) {
+                       n = rb_prev(n);
+                       continue;
+               }
+
+               resolve_root(&root_lookup, entry, &parent_id, &level, &path);
+               if (print_parent) {
+                       printf("ID %llu parent %llu top level %llu path %s\n",
+                               (unsigned long long)entry->root_id,
+                               (unsigned long long)parent_id,
+                               (unsigned long long)level, path);
+               } else {
+                       printf("ID %llu top level %llu path %s\n",
+                               (unsigned long long)entry->root_id,
+                               (unsigned long long)level, path);
+               }
+
+               free(path);
                n = rb_prev(n);
        }
 
@@ -863,3 +975,38 @@ int find_updated_files(int fd, u64 root_id, u64 oldest_gen)
        printf("transid marker was %llu\n", (unsigned long long)max_found);
        return ret;
 }
+
+char *path_for_root(int fd, u64 root)
+{
+       struct root_lookup root_lookup;
+       struct rb_node *n;
+       char *ret_path = NULL;
+       int ret;
+
+       ret = __list_subvol_search(fd, &root_lookup);
+       if (ret < 0)
+               return ERR_PTR(ret);
+
+       ret = __list_subvol_fill_paths(fd, &root_lookup);
+       if (ret < 0)
+               return ERR_PTR(ret);
+
+       n = rb_last(&root_lookup.root);
+       while (n) {
+               struct root_info *entry;
+               u64 parent_id;
+               u64 level;
+               char *path;
+
+               entry = rb_entry(n, struct root_info, rb_node);
+               resolve_root(&root_lookup, entry, &parent_id, &level, &path);
+               if (entry->root_id == root)
+                       ret_path = path;
+               else
+                       free(path);
+
+               n = rb_prev(n);
+       }
+
+       return ret_path;
+}