cmds-filesystem.c

   1 /*
   2  * This program is free software; you can redistribute it and/or
   3  * modify it under the terms of the GNU General Public
   4  * License v2 as published by the Free Software Foundation.
   5  *
   6  * This program is distributed in the hope that it will be useful,
   7  * but WITHOUT ANY WARRANTY; without even the implied warranty of
   8  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   9  * General Public License for more details.
  10  *
  11  * You should have received a copy of the GNU General Public
  12  * License along with this program; if not, write to the
  13  * Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  14  * Boston, MA 021110-1307, USA.
  15  */
  16
  17 #define _XOPEN_SOURCE 500
  18 #include <stdio.h>
  19 #include <stdlib.h>
  20 #include <string.h>
  21 #include <unistd.h>
  22 #include <sys/ioctl.h>
  23 #include <errno.h>
  24 #include <uuid/uuid.h>
  25 #include <ctype.h>
  26 #include <fcntl.h>
  27 #include <ftw.h>
  28 #include <mntent.h>
  29 #include <linux/limits.h>
  30 #include <getopt.h>
  31
  32 #include "kerncompat.h"
  33 #include "ctree.h"
  34 #include "ioctl.h"
  35 #include "utils.h"
  36 #include "volumes.h"
  37 #include "version.h"
  38 #include "commands.h"
  39 #include "cmds-fi-disk_usage.h"
  40 #include "list_sort.h"
  41 #include "disk-io.h"
  42
  43
  44 /*
  45  * for btrfs fi show, we maintain a hash of fsids we've already printed.
  46  * This way we don't print dups if a given FS is mounted more than once.
  47  */
  48 #define SEEN_FSID_HASH_SIZE 256
  49
  50 struct seen_fsid {
  51         u8 fsid[BTRFS_FSID_SIZE];
  52         struct seen_fsid *next;
  53 };
  54
  55 static struct seen_fsid *seen_fsid_hash[SEEN_FSID_HASH_SIZE] = {NULL,};
  56
  57 static int is_seen_fsid(u8 *fsid)
  58 {
  59         u8 hash = fsid[0];
  60         int slot = hash % SEEN_FSID_HASH_SIZE;
  61         struct seen_fsid *seen = seen_fsid_hash[slot];
  62
  63         return seen ? 1 : 0;
  64 }
  65
  66 static int add_seen_fsid(u8 *fsid)
  67 {
  68         u8 hash = fsid[0];
  69         int slot = hash % SEEN_FSID_HASH_SIZE;
  70         struct seen_fsid *seen = seen_fsid_hash[slot];
  71         struct seen_fsid *alloc;
  72
  73         if (!seen)
  74                 goto insert;
  75
  76         while (1) {
  77                 if (memcmp(seen->fsid, fsid, BTRFS_FSID_SIZE) == 0)
  78                         return -EEXIST;
  79
  80                 if (!seen->next)
  81                         break;
  82
  83                 seen = seen->next;
  84         }
  85
  86 insert:
  87
  88         alloc = malloc(sizeof(*alloc));
  89         if (!alloc)
  90                 return -ENOMEM;
  91
  92         alloc->next = NULL;
  93         memcpy(alloc->fsid, fsid, BTRFS_FSID_SIZE);
  94
  95         if (seen)
  96                 seen->next = alloc;
  97         else
  98                 seen_fsid_hash[slot] = alloc;
  99
 100         return 0;
 101 }
 102
 103 static void free_seen_fsid(void)
 104 {
 105         int slot;
 106         struct seen_fsid *seen;
 107         struct seen_fsid *next;
 108
 109         for (slot = 0; slot < SEEN_FSID_HASH_SIZE; slot++) {
 110                 seen = seen_fsid_hash[slot];
 111                 while (seen) {
 112                         next = seen->next;
 113                         free(seen);
 114                         seen = next;
 115                 }
 116                 seen_fsid_hash[slot] = NULL;
 117         }
 118 }
 119
 120 static const char * const filesystem_cmd_group_usage[] = {
 121         "btrfs filesystem [<group>] <command> [<args>]",
 122         NULL
 123 };
 124
 125 static const char * const cmd_filesystem_df_usage[] = {
 126        "btrfs filesystem df [options] <path>",
 127        "Show space usage information for a mount point",
 128         "-b|--raw           raw numbers in bytes",
 129         "-h                 human friendly numbers, base 1024 (default)",
 130         "-H                 human friendly numbers, base 1000",
 131         "--iec              use 1024 as a base (KiB, MiB, GiB, TiB)",
 132         "--si               use 1000 as a base (kB, MB, GB, TB)",
 133         "-k|--kbytes        show sizes in KiB, or kB with --si",
 134         "-m|--mbytes        show sizes in MiB, or MB with --si",
 135         "-g|--gbytes        show sizes in GiB, or GB with --si",
 136         "-t|--tbytes        show sizes in TiB, or TB with --si",
 137        NULL
 138 };
 139
 140 static int get_df(int fd, struct btrfs_ioctl_space_args **sargs_ret)
 141 {
 142         u64 count = 0;
 143         int ret, e;
 144         struct btrfs_ioctl_space_args *sargs;
 145
 146         sargs = malloc(sizeof(struct btrfs_ioctl_space_args));
 147         if (!sargs)
 148                 return -ENOMEM;
 149
 150         sargs->space_slots = 0;
 151         sargs->total_spaces = 0;
 152
 153         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
 154         e = errno;
 155         if (ret) {
 156                 fprintf(stderr, "ERROR: couldn't get space info - %s\n",
 157                         strerror(e));
 158                 free(sargs);
 159                 return -e;
 160         }
 161         /* This really should never happen */
 162         if (!sargs->total_spaces) {
 163                 free(sargs);
 164                 return -ENOENT;
 165         }
 166         count = sargs->total_spaces;
 167         free(sargs);
 168
 169         sargs = malloc(sizeof(struct btrfs_ioctl_space_args) +
 170                         (count * sizeof(struct btrfs_ioctl_space_info)));
 171         if (!sargs)
 172                 return -ENOMEM;
 173
 174         sargs->space_slots = count;
 175         sargs->total_spaces = 0;
 176         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
 177         e = errno;
 178         if (ret) {
 179                 fprintf(stderr, "ERROR: get space info count %llu - %s\n",
 180                                 count, strerror(e));
 181                 free(sargs);
 182                 return -e;
 183         }
 184         *sargs_ret = sargs;
 185         return 0;
 186 }
 187
 188 static void print_df(struct btrfs_ioctl_space_args *sargs, unsigned unit_mode)
 189 {
 190         u64 i;
 191         struct btrfs_ioctl_space_info *sp = sargs->spaces;
 192
 193         for (i = 0; i < sargs->total_spaces; i++, sp++) {
 194                 printf("%s, %s: total=%s, used=%s\n",
 195                         btrfs_group_type_str(sp->flags),
 196                         btrfs_group_profile_str(sp->flags),
 197                         pretty_size_mode(sp->total_bytes, unit_mode),
 198                         pretty_size_mode(sp->used_bytes, unit_mode));
 199         }
 200 }
 201
 202 static int cmd_filesystem_df(int argc, char **argv)
 203 {
 204         struct btrfs_ioctl_space_args *sargs = NULL;
 205         int ret;
 206         int fd;
 207         char *path;
 208         DIR *dirstream = NULL;
 209         unsigned unit_mode = UNITS_DEFAULT;
 210
 211         while (1) {
 212                 int long_index;
 213                 static const struct option long_options[] = {
 214                         { "raw", no_argument, NULL, 'b'},
 215                         { "kbytes", no_argument, NULL, 'k'},
 216                         { "mbytes", no_argument, NULL, 'm'},
 217                         { "gbytes", no_argument, NULL, 'g'},
 218                         { "tbytes", no_argument, NULL, 't'},
 219                         { "si", no_argument, NULL, GETOPT_VAL_SI},
 220                         { "iec", no_argument, NULL, GETOPT_VAL_IEC},
 221                 };
 222                 int c = getopt_long(argc, argv, "bhHkmgt", long_options,
 223                                         &long_index);
 224                 if (c < 0)
 225                         break;
 226                 switch (c) {
 227                 case 'b':
 228                         unit_mode = UNITS_RAW;
 229                         break;
 230                 case 'k':
 231                         units_set_base(&unit_mode, UNITS_KBYTES);
 232                         break;
 233                 case 'm':
 234                         units_set_base(&unit_mode, UNITS_MBYTES);
 235                         break;
 236                 case 'g':
 237                         units_set_base(&unit_mode, UNITS_GBYTES);
 238                         break;
 239                 case 't':
 240                         units_set_base(&unit_mode, UNITS_TBYTES);
 241                         break;
 242                 case 'h':
 243                         unit_mode = UNITS_HUMAN_BINARY;
 244                         break;
 245                 case 'H':
 246                         unit_mode = UNITS_HUMAN_DECIMAL;
 247                         break;
 248                 case GETOPT_VAL_SI:
 249                         units_set_mode(&unit_mode, UNITS_DECIMAL);
 250                         break;
 251                 case GETOPT_VAL_IEC:
 252                         units_set_mode(&unit_mode, UNITS_BINARY);
 253                         break;
 254                 default:
 255                         usage(cmd_filesystem_df_usage);
 256                 }
 257         }
 258
 259         if (check_argc_exact(argc, optind + 1))
 260                 usage(cmd_filesystem_df_usage);
 261
 262         path = argv[optind];
 263
 264         fd = open_file_or_dir(path, &dirstream);
 265         if (fd < 0) {
 266                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
 267                 return 1;
 268         }
 269         ret = get_df(fd, &sargs);
 270
 271         if (ret == 0) {
 272                 print_df(sargs, unit_mode);
 273                 free(sargs);
 274         } else {
 275                 fprintf(stderr, "ERROR: get_df failed %s\n", strerror(-ret));
 276         }
 277
 278         close_file_or_dir(fd, dirstream);
 279         return !!ret;
 280 }
 281
 282 static int match_search_item_kernel(__u8 *fsid, char *mnt, char *label,
 283                                         char *search)
 284 {
 285         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 286         int search_len = strlen(search);
 287
 288         search_len = min(search_len, BTRFS_UUID_UNPARSED_SIZE);
 289         uuid_unparse(fsid, uuidbuf);
 290         if (!strncmp(uuidbuf, search, search_len))
 291                 return 1;
 292
 293         if (strlen(label) && strcmp(label, search) == 0)
 294                 return 1;
 295
 296         if (strcmp(mnt, search) == 0)
 297                 return 1;
 298
 299         return 0;
 300 }
 301
 302 static int uuid_search(struct btrfs_fs_devices *fs_devices, char *search)
 303 {
 304         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 305         struct list_head *cur;
 306         struct btrfs_device *device;
 307         int search_len = strlen(search);
 308
 309         search_len = min(search_len, BTRFS_UUID_UNPARSED_SIZE);
 310         uuid_unparse(fs_devices->fsid, uuidbuf);
 311         if (!strncmp(uuidbuf, search, search_len))
 312                 return 1;
 313
 314         list_for_each(cur, &fs_devices->devices) {
 315                 device = list_entry(cur, struct btrfs_device, dev_list);
 316                 if ((device->label && strcmp(device->label, search) == 0) ||
 317                     strcmp(device->name, search) == 0)
 318                         return 1;
 319         }
 320         return 0;
 321 }
 322
 323 /*
 324  * Sort devices by devid, ascending
 325  */
 326 static int cmp_device_id(void *priv, struct list_head *a,
 327                 struct list_head *b)
 328 {
 329         const struct btrfs_device *da = list_entry(a, struct btrfs_device,
 330                         dev_list);
 331         const struct btrfs_device *db = list_entry(b, struct btrfs_device,
 332                         dev_list);
 333
 334         return da->devid < db->devid ? -1 :
 335                 da->devid > db->devid ? 1 : 0;
 336 }
 337
 338 static void splice_device_list(struct list_head *seed_devices,
 339                                struct list_head *all_devices)
 340 {
 341         struct btrfs_device *in_all, *next_all;
 342         struct btrfs_device *in_seed, *next_seed;
 343
 344         list_for_each_entry_safe(in_all, next_all, all_devices, dev_list) {
 345                 list_for_each_entry_safe(in_seed, next_seed, seed_devices,
 346                                                                 dev_list) {
 347                         if (in_all->devid == in_seed->devid) {
 348                                 /*
 349                                  * When do dev replace in a sprout fs
 350                                  * to a dev in its seed fs, the replacing
 351                                  * dev will reside in the sprout fs and
 352                                  * the replaced dev will still exist
 353                                  * in the seed fs.
 354                                  * So pick the latest one when showing
 355                                  * the sprout fs.
 356                                  */
 357                                 if (in_all->generation
 358                                                 < in_seed->generation) {
 359                                         list_del(&in_all->dev_list);
 360                                         free(in_all);
 361                                 } else if (in_all->generation
 362                                                 > in_seed->generation) {
 363                                         list_del(&in_seed->dev_list);
 364                                         free(in_seed);
 365                                 }
 366                                 break;
 367                         }
 368                 }
 369         }
 370
 371         list_splice(seed_devices, all_devices);
 372 }
 373
 374 static void print_devices(struct btrfs_fs_devices *fs_devices,
 375                           u64 *devs_found)
 376 {
 377         struct btrfs_device *device;
 378         struct btrfs_fs_devices *cur_fs;
 379         struct list_head *all_devices;
 380
 381         all_devices = &fs_devices->devices;
 382         cur_fs = fs_devices->seed;
 383         /* add all devices of seed fs to the fs to be printed */
 384         while (cur_fs) {
 385                 splice_device_list(&cur_fs->devices, all_devices);
 386                 cur_fs = cur_fs->seed;
 387         }
 388
 389         list_sort(NULL, all_devices, cmp_device_id);
 390         list_for_each_entry(device, all_devices, dev_list) {
 391                 printf("\tdevid %4llu size %s used %s path %s\n",
 392                        (unsigned long long)device->devid,
 393                        pretty_size(device->total_bytes),
 394                        pretty_size(device->bytes_used), device->name);
 395
 396                 (*devs_found)++;
 397         }
 398 }
 399
 400 static void print_one_uuid(struct btrfs_fs_devices *fs_devices)
 401 {
 402         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 403         struct btrfs_device *device;
 404         u64 devs_found = 0;
 405         u64 total;
 406
 407         if (add_seen_fsid(fs_devices->fsid))
 408                 return;
 409
 410         uuid_unparse(fs_devices->fsid, uuidbuf);
 411         device = list_entry(fs_devices->devices.next, struct btrfs_device,
 412                             dev_list);
 413         if (device->label && device->label[0])
 414                 printf("Label: '%s' ", device->label);
 415         else
 416                 printf("Label: none ");
 417
 418         total = device->total_devs;
 419         printf(" uuid: %s\n\tTotal devices %llu FS bytes used %s\n", uuidbuf,
 420                (unsigned long long)total,
 421                pretty_size(device->super_bytes_used));
 422
 423         print_devices(fs_devices, &devs_found);
 424
 425         if (devs_found < total) {
 426                 printf("\t*** Some devices missing\n");
 427         }
 428         printf("\n");
 429 }
 430
 431 /* adds up all the used spaces as reported by the space info ioctl
 432  */
 433 static u64 calc_used_bytes(struct btrfs_ioctl_space_args *si)
 434 {
 435         u64 ret = 0;
 436         int i;
 437         for (i = 0; i < si->total_spaces; i++)
 438                 ret += si->spaces[i].used_bytes;
 439         return ret;
 440 }
 441
 442 static int print_one_fs(struct btrfs_ioctl_fs_info_args *fs_info,
 443                 struct btrfs_ioctl_dev_info_args *dev_info,
 444                 struct btrfs_ioctl_space_args *space_info,
 445                 char *label, char *path)
 446 {
 447         int i;
 448         int fd;
 449         int missing = 0;
 450         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 451         struct btrfs_ioctl_dev_info_args *tmp_dev_info;
 452         int ret;
 453
 454         ret = add_seen_fsid(fs_info->fsid);
 455         if (ret == -EEXIST)
 456                 return 0;
 457         else if (ret)
 458                 return ret;
 459
 460         uuid_unparse(fs_info->fsid, uuidbuf);
 461         if (label && strlen(label))
 462                 printf("Label: '%s' ", label);
 463         else
 464                 printf("Label: none ");
 465
 466         printf(" uuid: %s\n\tTotal devices %llu FS bytes used %s\n", uuidbuf,
 467                         fs_info->num_devices,
 468                         pretty_size(calc_used_bytes(space_info)));
 469
 470         for (i = 0; i < fs_info->num_devices; i++) {
 471                 char *canonical_path;
 472
 473                 tmp_dev_info = (struct btrfs_ioctl_dev_info_args *)&dev_info[i];
 474                 canonical_path = canonicalize_path((char *)tmp_dev_info->path);
 475
 476                 /* Add check for missing devices even mounted */
 477                 fd = open((char *)tmp_dev_info->path, O_RDONLY);
 478                 if (fd < 0) {
 479                         missing = 1;
 480                         continue;
 481                 }
 482                 close(fd);
 483                 printf("\tdevid %4llu size %s used %s path %s\n",
 484                         tmp_dev_info->devid,
 485                         pretty_size(tmp_dev_info->total_bytes),
 486                         pretty_size(tmp_dev_info->bytes_used),
 487                         canonical_path);
 488
 489                 free(canonical_path);
 490         }
 491
 492         if (missing)
 493                 printf("\t*** Some devices missing\n");
 494         printf("\n");
 495         return 0;
 496 }
 497
 498 /* This function checks if the given input parameter is
 499  * an uuid or a path
 500  * return -1: some error in the given input
 501  * return 0: unknow input
 502  * return 1: given input is uuid
 503  * return 2: given input is path
 504  */
 505 static int check_arg_type(char *input)
 506 {
 507         uuid_t  out;
 508         char path[PATH_MAX];
 509
 510         if (!input)
 511                 return -EINVAL;
 512
 513         if (realpath(input, path)) {
 514                 if (is_block_device(path) == 1)
 515                         return BTRFS_ARG_BLKDEV;
 516
 517                 if (is_mount_point(path) == 1)
 518                         return BTRFS_ARG_MNTPOINT;
 519
 520                 return BTRFS_ARG_UNKNOWN;
 521         }
 522
 523         if (strlen(input) == (BTRFS_UUID_UNPARSED_SIZE - 1) &&
 524                 !uuid_parse(input, out))
 525                 return BTRFS_ARG_UUID;
 526
 527         return BTRFS_ARG_UNKNOWN;
 528 }
 529
 530 static int btrfs_scan_kernel(void *search)
 531 {
 532         int ret = 0, fd;
 533         int found = 0;
 534         FILE *f;
 535         struct mntent *mnt;
 536         struct btrfs_ioctl_fs_info_args fs_info_arg;
 537         struct btrfs_ioctl_dev_info_args *dev_info_arg = NULL;
 538         struct btrfs_ioctl_space_args *space_info_arg = NULL;
 539         char label[BTRFS_LABEL_SIZE];
 540
 541         f = setmntent("/proc/self/mounts", "r");
 542         if (f == NULL)
 543                 return 1;
 544
 545         memset(label, 0, sizeof(label));
 546         while ((mnt = getmntent(f)) != NULL) {
 547                 if (strcmp(mnt->mnt_type, "btrfs"))
 548                         continue;
 549                 ret = get_fs_info(mnt->mnt_dir, &fs_info_arg,
 550                                 &dev_info_arg);
 551                 if (ret)
 552                         goto out;
 553
 554                 if (get_label_mounted(mnt->mnt_dir, label)) {
 555                         kfree(dev_info_arg);
 556                         goto out;
 557                 }
 558                 if (search && !match_search_item_kernel(fs_info_arg.fsid,
 559                                         mnt->mnt_dir, label, search)) {
 560                         kfree(dev_info_arg);
 561                         continue;
 562                 }
 563
 564                 fd = open(mnt->mnt_dir, O_RDONLY);
 565                 if ((fd != -1) && !get_df(fd, &space_info_arg)) {
 566                         print_one_fs(&fs_info_arg, dev_info_arg,
 567                                         space_info_arg, label, mnt->mnt_dir);
 568                         kfree(space_info_arg);
 569                         memset(label, 0, sizeof(label));
 570                         found = 1;
 571                 }
 572                 if (fd != -1)
 573                         close(fd);
 574                 kfree(dev_info_arg);
 575         }
 576
 577 out:
 578         endmntent(f);
 579         return !found;
 580 }
 581
 582 static int dev_to_fsid(char *dev, __u8 *fsid)
 583 {
 584         struct btrfs_super_block *disk_super;
 585         char *buf;
 586         int ret;
 587         int fd;
 588
 589         buf = malloc(4096);
 590         if (!buf)
 591                 return -ENOMEM;
 592
 593         fd = open(dev, O_RDONLY);
 594         if (fd < 0) {
 595                 ret = -errno;
 596                 free(buf);
 597                 return ret;
 598         }
 599
 600         disk_super = (struct btrfs_super_block *)buf;
 601         ret = btrfs_read_dev_super(fd, disk_super,
 602                                    BTRFS_SUPER_INFO_OFFSET, 0);
 603         if (ret)
 604                 goto out;
 605
 606         memcpy(fsid, disk_super->fsid, BTRFS_FSID_SIZE);
 607         ret = 0;
 608
 609 out:
 610         close(fd);
 611         free(buf);
 612         return ret;
 613 }
 614
 615 static void free_fs_devices(struct btrfs_fs_devices *fs_devices)
 616 {
 617         struct btrfs_fs_devices *cur_seed, *next_seed;
 618         struct btrfs_device *device;
 619
 620         while (!list_empty(&fs_devices->devices)) {
 621                 device = list_entry(fs_devices->devices.next,
 622                                         struct btrfs_device, dev_list);
 623                 list_del(&device->dev_list);
 624
 625                 free(device->name);
 626                 free(device->label);
 627                 free(device);
 628         }
 629
 630         /* free seed fs chain */
 631         cur_seed = fs_devices->seed;
 632         fs_devices->seed = NULL;
 633         while (cur_seed) {
 634                 next_seed = cur_seed->seed;
 635                 free(cur_seed);
 636
 637                 cur_seed = next_seed;
 638         }
 639
 640         list_del(&fs_devices->list);
 641         free(fs_devices);
 642 }
 643
 644 static int copy_device(struct btrfs_device *dst,
 645                        struct btrfs_device *src)
 646 {
 647         dst->devid = src->devid;
 648         memcpy(dst->uuid, src->uuid, BTRFS_UUID_SIZE);
 649         if (src->name == NULL)
 650                 dst->name = NULL;
 651         else {
 652                 dst->name = strdup(src->name);
 653                 if (!dst->name)
 654                         return -ENOMEM;
 655         }
 656         if (src->label == NULL)
 657                 dst->label = NULL;
 658         else {
 659                 dst->label = strdup(src->label);
 660                 if (!dst->label) {
 661                         free(dst->name);
 662                         return -ENOMEM;
 663                 }
 664         }
 665         dst->total_devs = src->total_devs;
 666         dst->super_bytes_used = src->super_bytes_used;
 667         dst->total_bytes = src->total_bytes;
 668         dst->bytes_used = src->bytes_used;
 669         dst->generation = src->generation;
 670
 671         return 0;
 672 }
 673
 674 static int copy_fs_devices(struct btrfs_fs_devices *dst,
 675                            struct btrfs_fs_devices *src)
 676 {
 677         struct btrfs_device *cur_dev, *dev_copy;
 678         int ret = 0;
 679
 680         memcpy(dst->fsid, src->fsid, BTRFS_FSID_SIZE);
 681         INIT_LIST_HEAD(&dst->devices);
 682         dst->seed = NULL;
 683
 684         list_for_each_entry(cur_dev, &src->devices, dev_list) {
 685                 dev_copy = malloc(sizeof(*dev_copy));
 686                 if (!dev_copy) {
 687                         ret = -ENOMEM;
 688                         break;
 689                 }
 690
 691                 ret = copy_device(dev_copy, cur_dev);
 692                 if (ret) {
 693                         free(dev_copy);
 694                         break;
 695                 }
 696
 697                 list_add(&dev_copy->dev_list, &dst->devices);
 698                 dev_copy->fs_devices = dst;
 699         }
 700
 701         return ret;
 702 }
 703
 704 static int find_and_copy_seed(struct btrfs_fs_devices *seed,
 705                               struct btrfs_fs_devices *copy,
 706                               struct list_head *fs_uuids) {
 707         struct btrfs_fs_devices *cur_fs;
 708
 709         list_for_each_entry(cur_fs, fs_uuids, list)
 710                 if (!memcmp(seed->fsid, cur_fs->fsid, BTRFS_FSID_SIZE))
 711                         return copy_fs_devices(copy, cur_fs);
 712
 713         return 1;
 714 }
 715
 716 static int has_seed_devices(struct btrfs_fs_devices *fs_devices)
 717 {
 718         struct btrfs_device *device;
 719         int dev_cnt_total, dev_cnt = 0;
 720
 721         device = list_first_entry(&fs_devices->devices, struct btrfs_device,
 722                                   dev_list);
 723
 724         dev_cnt_total = device->total_devs;
 725
 726         list_for_each_entry(device, &fs_devices->devices, dev_list)
 727                 dev_cnt++;
 728
 729         return dev_cnt_total != dev_cnt;
 730 }
 731
 732 static int search_umounted_fs_uuids(struct list_head *all_uuids,
 733                                     char *search)
 734 {
 735         struct btrfs_fs_devices *cur_fs, *fs_copy;
 736         struct list_head *fs_uuids;
 737         int ret = 0;
 738
 739         fs_uuids = btrfs_scanned_uuids();
 740
 741         /*
 742          * The fs_uuids list is global, and open_ctree_* will
 743          * modify it, make a private copy here
 744          */
 745         list_for_each_entry(cur_fs, fs_uuids, list) {
 746                 /* don't bother handle all fs, if search target specified */
 747                 if (search) {
 748                         if (uuid_search(cur_fs, search) == 0)
 749                                 continue;
 750                         ret = 1;
 751                 }
 752
 753                 /* skip all fs already shown as mounted fs */
 754                 if (is_seen_fsid(cur_fs->fsid))
 755                         continue;
 756
 757                 fs_copy = malloc(sizeof(*fs_copy));
 758                 if (!fs_copy) {
 759                         ret = -ENOMEM;
 760                         goto out;
 761                 }
 762
 763                 ret = copy_fs_devices(fs_copy, cur_fs);
 764                 if (ret) {
 765                         free(fs_copy);
 766                         goto out;
 767                 }
 768
 769                 list_add(&fs_copy->list, all_uuids);
 770         }
 771
 772 out:
 773         return ret;
 774 }
 775
 776 static int map_seed_devices(struct list_head *all_uuids)
 777 {
 778         struct btrfs_fs_devices *cur_fs, *cur_seed;
 779         struct btrfs_fs_devices *seed_copy;
 780         struct btrfs_fs_devices *opened_fs;
 781         struct btrfs_device *device;
 782         struct btrfs_fs_info *fs_info;
 783         struct list_head *fs_uuids;
 784         int ret = 0;
 785
 786         fs_uuids = btrfs_scanned_uuids();
 787
 788         list_for_each_entry(cur_fs, all_uuids, list) {
 789                 device = list_first_entry(&cur_fs->devices,
 790                                                 struct btrfs_device, dev_list);
 791                 if (!device)
 792                         continue;
 793
 794                 /* skip fs without seeds */
 795                 if (!has_seed_devices(cur_fs))
 796                         continue;
 797
 798                 /*
 799                  * open_ctree_* detects seed/sprout mapping
 800                  */
 801                 fs_info = open_ctree_fs_info(device->name, 0, 0,
 802                                                 OPEN_CTREE_PARTIAL);
 803                 if (!fs_info)
 804                         continue;
 805
 806                 /*
 807                  * copy the seed chain under the opened fs
 808                  */
 809                 opened_fs = fs_info->fs_devices;
 810                 cur_seed = cur_fs;
 811                 while (opened_fs->seed) {
 812                         seed_copy = malloc(sizeof(*seed_copy));
 813                         if (!seed_copy) {
 814                                 ret = -ENOMEM;
 815                                 goto fail_out;
 816                         }
 817                         ret = find_and_copy_seed(opened_fs->seed, seed_copy,
 818                                                  fs_uuids);
 819                         if (ret) {
 820                                 free(seed_copy);
 821                                 goto fail_out;
 822                         }
 823
 824                         cur_seed->seed = seed_copy;
 825
 826                         opened_fs = opened_fs->seed;
 827                         cur_seed = cur_seed->seed;
 828                 }
 829
 830                 close_ctree(fs_info->chunk_root);
 831         }
 832
 833 out:
 834         return ret;
 835 fail_out:
 836         close_ctree(fs_info->chunk_root);
 837         goto out;
 838 }
 839
 840 static const char * const cmd_show_usage[] = {
 841         "btrfs filesystem show [options] [<path>|<uuid>|<device>|label]",
 842         "Show the structure of a filesystem",
 843         "-d|--all-devices   show only disks under /dev containing btrfs filesystem",
 844         "-m|--mounted       show only mounted btrfs",
 845         "If no argument is given, structure of all present filesystems is shown.",
 846         NULL
 847 };
 848
 849 static int cmd_show(int argc, char **argv)
 850 {
 851         LIST_HEAD(all_uuids);
 852         struct btrfs_fs_devices *fs_devices;
 853         char *search = NULL;
 854         int ret;
 855         /* default, search both kernel and udev */
 856         int where = -1;
 857         int type = 0;
 858         char mp[BTRFS_PATH_NAME_MAX + 1];
 859         char path[PATH_MAX];
 860         __u8 fsid[BTRFS_FSID_SIZE];
 861         char uuid_buf[BTRFS_UUID_UNPARSED_SIZE];
 862         int found = 0;
 863
 864         while (1) {
 865                 int long_index;
 866                 static struct option long_options[] = {
 867                         { "all-devices", no_argument, NULL, 'd'},
 868                         { "mounted", no_argument, NULL, 'm'},
 869                         { NULL, no_argument, NULL, 0 },
 870                 };
 871                 int c = getopt_long(argc, argv, "dm", long_options,
 872                                         &long_index);
 873                 if (c < 0)
 874                         break;
 875                 switch (c) {
 876                 case 'd':
 877                         where = BTRFS_SCAN_LBLKID;
 878                         break;
 879                 case 'm':
 880                         where = BTRFS_SCAN_MOUNTED;
 881                         break;
 882                 default:
 883                         usage(cmd_show_usage);
 884                 }
 885         }
 886
 887         if (check_argc_max(argc, optind + 1))
 888                 usage(cmd_show_usage);
 889
 890         if (argc > optind) {
 891                 search = argv[optind];
 892                 if (strlen(search) == 0)
 893                         usage(cmd_show_usage);
 894                 type = check_arg_type(search);
 895
 896                 /*
 897                  * For search is a device:
 898                  *     realpath do /dev/mapper/XX => /dev/dm-X
 899                  *     which is required by BTRFS_SCAN_DEV
 900                  * For search is a mountpoint:
 901                  *     realpath do  /mnt/btrfs/  => /mnt/btrfs
 902                  *     which shall be recognized by btrfs_scan_kernel()
 903                  */
 904                 if (!realpath(search, path)) {
 905                         fprintf(stderr, "ERROR: Could not show %s: %s\n",
 906                                         search, strerror(errno));
 907                         return 1;
 908                 }
 909
 910                 search = path;
 911
 912                 /*
 913                  * Needs special handling if input arg is block dev And if
 914                  * input arg is mount-point just print it right away
 915                  */
 916                 if (type == BTRFS_ARG_BLKDEV && where != BTRFS_SCAN_LBLKID) {
 917                         ret = get_btrfs_mount(search, mp, sizeof(mp));
 918                         if (!ret) {
 919                                 /* given block dev is mounted */
 920                                 search = mp;
 921                                 type = BTRFS_ARG_MNTPOINT;
 922                         } else {
 923                                 ret = dev_to_fsid(search, fsid);
 924                                 if (ret) {
 925                                         fprintf(stderr,
 926                                                 "ERROR: No btrfs on %s\n",
 927                                                 search);
 928                                         return 1;
 929                                 }
 930                                 uuid_unparse(fsid, uuid_buf);
 931                                 search = uuid_buf;
 932                                 type = BTRFS_ARG_UUID;
 933                                 goto devs_only;
 934                         }
 935                 }
 936         }
 937
 938         if (where == BTRFS_SCAN_LBLKID)
 939                 goto devs_only;
 940
 941         /* show mounted btrfs */
 942         ret = btrfs_scan_kernel(search);
 943         if (search && !ret) {
 944                 /* since search is found we are done */
 945                 goto out;
 946         }
 947
 948         /* shows mounted only */
 949         if (where == BTRFS_SCAN_MOUNTED)
 950                 goto out;
 951
 952 devs_only:
 953         ret = btrfs_scan_lblkid();
 954
 955         if (ret) {
 956                 fprintf(stderr, "ERROR: %d while scanning\n", ret);
 957                 return 1;
 958         }
 959
 960         found = search_umounted_fs_uuids(&all_uuids, search);
 961         if (found < 0) {
 962                 fprintf(stderr,
 963                         "ERROR: %d while searching target device\n", ret);
 964                 return 1;
 965         }
 966
 967         /*
 968          * The seed/sprout mapping are not detected yet,
 969          * do mapping build for all umounted fs
 970          */
 971         ret = map_seed_devices(&all_uuids);
 972         if (ret) {
 973                 fprintf(stderr,
 974                         "ERROR: %d while mapping seed devices\n", ret);
 975                 return 1;
 976         }
 977
 978         list_for_each_entry(fs_devices, &all_uuids, list)
 979                 print_one_uuid(fs_devices);
 980
 981         if (search && !found)
 982                 ret = 1;
 983
 984         while (!list_empty(&all_uuids)) {
 985                 fs_devices = list_entry(all_uuids.next,
 986                                         struct btrfs_fs_devices, list);
 987                 free_fs_devices(fs_devices);
 988         }
 989 out:
 990         printf("%s\n", BTRFS_BUILD_VERSION);
 991         free_seen_fsid();
 992         return ret;
 993 }
 994
 995 static const char * const cmd_sync_usage[] = {
 996         "btrfs filesystem sync <path>",
 997         "Force a sync on a filesystem",
 998         NULL
 999 };
1000
1001 static int cmd_sync(int argc, char **argv)
1002 {
1003         int     fd, res, e;
1004         char    *path;
1005         DIR     *dirstream = NULL;
1006
1007         if (check_argc_exact(argc, 2))
1008                 usage(cmd_sync_usage);
1009
1010         path = argv[1];
1011
1012         fd = open_file_or_dir(path, &dirstream);
1013         if (fd < 0) {
1014                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
1015                 return 1;
1016         }
1017
1018         printf("FSSync '%s'\n", path);
1019         res = ioctl(fd, BTRFS_IOC_SYNC);
1020         e = errno;
1021         close_file_or_dir(fd, dirstream);
1022         if( res < 0 ){
1023                 fprintf(stderr, "ERROR: unable to fs-syncing '%s' - %s\n",
1024                         path, strerror(e));
1025                 return 1;
1026         }
1027
1028         return 0;
1029 }
1030
1031 static int parse_compress_type(char *s)
1032 {
1033         if (strcmp(optarg, "zlib") == 0)
1034                 return BTRFS_COMPRESS_ZLIB;
1035         else if (strcmp(optarg, "lzo") == 0)
1036                 return BTRFS_COMPRESS_LZO;
1037         else {
1038                 fprintf(stderr, "Unknown compress type %s\n", s);
1039                 exit(1);
1040         };
1041 }
1042
1043 static const char * const cmd_defrag_usage[] = {
1044         "btrfs filesystem defragment [options] <file>|<dir> [<file>|<dir>...]",
1045         "Defragment a file or a directory",
1046         "",
1047         "-v             be verbose",
1048         "-r             defragment files recursively",
1049         "-c[zlib,lzo]   compress the file while defragmenting",
1050         "-f             flush data to disk immediately after defragmenting",
1051         "-s start       defragment only from byte onward",
1052         "-l len         defragment only up to len bytes",
1053         "-t size        minimal size of file to be considered for defragmenting",
1054         NULL
1055 };
1056
1057 static int do_defrag(int fd, int fancy_ioctl,
1058                 struct btrfs_ioctl_defrag_range_args *range)
1059 {
1060         int ret;
1061
1062         if (!fancy_ioctl)
1063                 ret = ioctl(fd, BTRFS_IOC_DEFRAG, NULL);
1064         else
1065                 ret = ioctl(fd, BTRFS_IOC_DEFRAG_RANGE, range);
1066
1067         return ret;
1068 }
1069
1070 static int defrag_global_fancy_ioctl;
1071 static struct btrfs_ioctl_defrag_range_args defrag_global_range;
1072 static int defrag_global_verbose;
1073 static int defrag_global_errors;
1074 static int defrag_callback(const char *fpath, const struct stat *sb,
1075                 int typeflag, struct FTW *ftwbuf)
1076 {
1077         int ret = 0;
1078         int e = 0;
1079         int fd = 0;
1080
1081         if ((typeflag == FTW_F) && S_ISREG(sb->st_mode)) {
1082                 if (defrag_global_verbose)
1083                         printf("%s\n", fpath);
1084                 fd = open(fpath, O_RDWR);
1085                 e = errno;
1086                 if (fd < 0)
1087                         goto error;
1088                 ret = do_defrag(fd, defrag_global_fancy_ioctl, &defrag_global_range);
1089                 e = errno;
1090                 close(fd);
1091                 if (ret && e == ENOTTY && defrag_global_fancy_ioctl) {
1092                         fprintf(stderr, "ERROR: defrag range ioctl not "
1093                                 "supported in this kernel, please try "
1094                                 "without any options.\n");
1095                         defrag_global_errors++;
1096                         return ENOTTY;
1097                 }
1098                 if (ret)
1099                         goto error;
1100         }
1101         return 0;
1102
1103 error:
1104         fprintf(stderr, "ERROR: defrag failed on %s - %s\n", fpath, strerror(e));
1105         defrag_global_errors++;
1106         return 0;
1107 }
1108
1109 static int cmd_defrag(int argc, char **argv)
1110 {
1111         int fd;
1112         int flush = 0;
1113         u64 start = 0;
1114         u64 len = (u64)-1;
1115         u32 thresh = 0;
1116         int i;
1117         int recursive = 0;
1118         int ret = 0;
1119         struct btrfs_ioctl_defrag_range_args range;
1120         int e = 0;
1121         int compress_type = BTRFS_COMPRESS_NONE;
1122         DIR *dirstream;
1123
1124         defrag_global_errors = 0;
1125         defrag_global_verbose = 0;
1126         defrag_global_errors = 0;
1127         defrag_global_fancy_ioctl = 0;
1128         optind = 1;
1129         while(1) {
1130                 int c = getopt(argc, argv, "vrc::fs:l:t:");
1131                 if (c < 0)
1132                         break;
1133
1134                 switch(c) {
1135                 case 'c':
1136                         compress_type = BTRFS_COMPRESS_ZLIB;
1137                         if (optarg)
1138                                 compress_type = parse_compress_type(optarg);
1139                         defrag_global_fancy_ioctl = 1;
1140                         break;
1141                 case 'f':
1142                         flush = 1;
1143                         defrag_global_fancy_ioctl = 1;
1144                         break;
1145                 case 'v':
1146                         defrag_global_verbose = 1;
1147                         break;
1148                 case 's':
1149                         start = parse_size(optarg);
1150                         defrag_global_fancy_ioctl = 1;
1151                         break;
1152                 case 'l':
1153                         len = parse_size(optarg);
1154                         defrag_global_fancy_ioctl = 1;
1155                         break;
1156                 case 't':
1157                         thresh = parse_size(optarg);
1158                         defrag_global_fancy_ioctl = 1;
1159                         break;
1160                 case 'r':
1161                         recursive = 1;
1162                         break;
1163                 default:
1164                         usage(cmd_defrag_usage);
1165                 }
1166         }
1167
1168         if (check_argc_min(argc - optind, 1))
1169                 usage(cmd_defrag_usage);
1170
1171         memset(&defrag_global_range, 0, sizeof(range));
1172         defrag_global_range.start = start;
1173         defrag_global_range.len = len;
1174         defrag_global_range.extent_thresh = thresh;
1175         if (compress_type) {
1176                 defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_COMPRESS;
1177                 defrag_global_range.compress_type = compress_type;
1178         }
1179         if (flush)
1180                 defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_START_IO;
1181
1182         for (i = optind; i < argc; i++) {
1183                 struct stat st;
1184
1185                 dirstream = NULL;
1186                 fd = open_file_or_dir(argv[i], &dirstream);
1187                 if (fd < 0) {
1188                         fprintf(stderr, "ERROR: failed to open %s - %s\n", argv[i],
1189                                         strerror(errno));
1190                         defrag_global_errors++;
1191                         close_file_or_dir(fd, dirstream);
1192                         continue;
1193                 }
1194                 if (fstat(fd, &st)) {
1195                         fprintf(stderr, "ERROR: failed to stat %s - %s\n",
1196                                         argv[i], strerror(errno));
1197                         defrag_global_errors++;
1198                         close_file_or_dir(fd, dirstream);
1199                         continue;
1200                 }
1201                 if (!(S_ISDIR(st.st_mode) || S_ISREG(st.st_mode))) {
1202                         fprintf(stderr,
1203                             "ERROR: %s is not a directory or a regular file\n",
1204                             argv[i]);
1205                         defrag_global_errors++;
1206                         close_file_or_dir(fd, dirstream);
1207                         continue;
1208                 }
1209                 if (recursive) {
1210                         if (S_ISDIR(st.st_mode)) {
1211                                 ret = nftw(argv[i], defrag_callback, 10,
1212                                                 FTW_MOUNT | FTW_PHYS);
1213                                 if (ret == ENOTTY)
1214                                         exit(1);
1215                                 /* errors are handled in the callback */
1216                                 ret = 0;
1217                         } else {
1218                                 if (defrag_global_verbose)
1219                                         printf("%s\n", argv[i]);
1220                                 ret = do_defrag(fd, defrag_global_fancy_ioctl,
1221                                                 &defrag_global_range);
1222                                 e = errno;
1223                         }
1224                 } else {
1225                         if (defrag_global_verbose)
1226                                 printf("%s\n", argv[i]);
1227                         ret = do_defrag(fd, defrag_global_fancy_ioctl,
1228                                         &defrag_global_range);
1229                         e = errno;
1230                 }
1231                 close_file_or_dir(fd, dirstream);
1232                 if (ret && e == ENOTTY && defrag_global_fancy_ioctl) {
1233                         fprintf(stderr, "ERROR: defrag range ioctl not "
1234                                 "supported in this kernel, please try "
1235                                 "without any options.\n");
1236                         defrag_global_errors++;
1237                         break;
1238                 }
1239                 if (ret) {
1240                         fprintf(stderr, "ERROR: defrag failed on %s - %s\n",
1241                                 argv[i], strerror(e));
1242                         defrag_global_errors++;
1243                 }
1244         }
1245         if (defrag_global_verbose)
1246                 printf("%s\n", BTRFS_BUILD_VERSION);
1247         if (defrag_global_errors)
1248                 fprintf(stderr, "total %d failures\n", defrag_global_errors);
1249
1250         return !!defrag_global_errors;
1251 }
1252
1253 static const char * const cmd_resize_usage[] = {
1254         "btrfs filesystem resize [devid:][+/-]<newsize>[kKmMgGtTpPeE]|[devid:]max <path>",
1255         "Resize a filesystem",
1256         "If 'max' is passed, the filesystem will occupy all available space",
1257         "on the device 'devid'.",
1258         "[kK] means KiB, which denotes 1KiB = 1024B, 1MiB = 1024KiB, etc.",
1259         NULL
1260 };
1261
1262 static int cmd_resize(int argc, char **argv)
1263 {
1264         struct btrfs_ioctl_vol_args     args;
1265         int     fd, res, len, e;
1266         char    *amount, *path;
1267         DIR     *dirstream = NULL;
1268
1269         if (check_argc_exact(argc, 3))
1270                 usage(cmd_resize_usage);
1271
1272         amount = argv[1];
1273         path = argv[2];
1274
1275         len = strlen(amount);
1276         if (len == 0 || len >= BTRFS_VOL_NAME_MAX) {
1277                 fprintf(stderr, "ERROR: size value too long ('%s)\n",
1278                         amount);
1279                 return 1;
1280         }
1281
1282         fd = open_file_or_dir(path, &dirstream);
1283         if (fd < 0) {
1284                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
1285                 return 1;
1286         }
1287
1288         printf("Resize '%s' of '%s'\n", path, amount);
1289         strncpy_null(args.name, amount);
1290         res = ioctl(fd, BTRFS_IOC_RESIZE, &args);
1291         e = errno;
1292         close_file_or_dir(fd, dirstream);
1293         if( res < 0 ){
1294                 fprintf(stderr, "ERROR: unable to resize '%s' - %s\n",
1295                         path, strerror(e));
1296                 return 1;
1297         }
1298         return 0;
1299 }
1300
1301 static const char * const cmd_label_usage[] = {
1302         "btrfs filesystem label [<device>|<mount_point>] [<newlabel>]",
1303         "Get or change the label of a filesystem",
1304         "With one argument, get the label of filesystem on <device>.",
1305         "If <newlabel> is passed, set the filesystem label to <newlabel>.",
1306         NULL
1307 };
1308
1309 static int cmd_label(int argc, char **argv)
1310 {
1311         if (check_argc_min(argc, 2) || check_argc_max(argc, 3))
1312                 usage(cmd_label_usage);
1313
1314         if (argc > 2) {
1315                 return set_label(argv[1], argv[2]);
1316         } else {
1317                 char label[BTRFS_LABEL_SIZE];
1318                 int ret;
1319
1320                 ret = get_label(argv[1], label);
1321                 if (!ret)
1322                         fprintf(stdout, "%s\n", label);
1323
1324                 return ret;
1325         }
1326 }
1327
1328 const struct cmd_group filesystem_cmd_group = {
1329         filesystem_cmd_group_usage, NULL, {
1330                 { "df", cmd_filesystem_df, cmd_filesystem_df_usage, NULL, 0 },
1331                 { "show", cmd_show, cmd_show_usage, NULL, 0 },
1332                 { "sync", cmd_sync, cmd_sync_usage, NULL, 0 },
1333                 { "defragment", cmd_defrag, cmd_defrag_usage, NULL, 0 },
1334                 { "balance", cmd_balance, NULL, &balance_cmd_group, 1 },
1335                 { "resize", cmd_resize, cmd_resize_usage, NULL, 0 },
1336                 { "label", cmd_label, cmd_label_usage, NULL, 0 },
1337                 { "usage", cmd_filesystem_usage,
1338                         cmd_filesystem_usage_usage, NULL, 0 },
1339
1340                 NULL_CMD_STRUCT
1341         }
1342 };
1343
1344 int cmd_filesystem(int argc, char **argv)
1345 {
1346         return handle_command_group(&filesystem_cmd_group, argc, argv);
1347 }