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 "list_sort.h"
  40 #include "disk-io.h"
  41
  42
  43 /*
  44  * for btrfs fi show, we maintain a hash of fsids we've already printed.
  45  * This way we don't print dups if a given FS is mounted more than once.
  46  */
  47 #define SEEN_FSID_HASH_SIZE 256
  48
  49 struct seen_fsid {
  50         u8 fsid[BTRFS_FSID_SIZE];
  51         struct seen_fsid *next;
  52 };
  53
  54 static struct seen_fsid *seen_fsid_hash[SEEN_FSID_HASH_SIZE] = {NULL,};
  55
  56 static int add_seen_fsid(u8 *fsid)
  57 {
  58         u8 hash = fsid[0];
  59         int slot = hash % SEEN_FSID_HASH_SIZE;
  60         struct seen_fsid *seen = seen_fsid_hash[slot];
  61         struct seen_fsid *alloc;
  62
  63         if (!seen)
  64                 goto insert;
  65
  66         while (1) {
  67                 if (memcmp(seen->fsid, fsid, BTRFS_FSID_SIZE) == 0)
  68                         return -EEXIST;
  69
  70                 if (!seen->next)
  71                         break;
  72
  73                 seen = seen->next;
  74         }
  75
  76 insert:
  77
  78         alloc = malloc(sizeof(*alloc));
  79         if (!alloc)
  80                 return -ENOMEM;
  81
  82         alloc->next = NULL;
  83         memcpy(alloc->fsid, fsid, BTRFS_FSID_SIZE);
  84
  85         if (seen)
  86                 seen->next = alloc;
  87         else
  88                 seen_fsid_hash[slot] = alloc;
  89
  90         return 0;
  91 }
  92
  93 static void free_seen_fsid(void)
  94 {
  95         int slot;
  96         struct seen_fsid *seen;
  97         struct seen_fsid *next;
  98
  99         for (slot = 0; slot < SEEN_FSID_HASH_SIZE; slot++) {
 100                 seen = seen_fsid_hash[slot];
 101                 while (seen) {
 102                         next = seen->next;
 103                         free(seen);
 104                         seen = next;
 105                 }
 106                 seen_fsid_hash[slot] = NULL;
 107         }
 108 }
 109
 110 static const char * const filesystem_cmd_group_usage[] = {
 111         "btrfs filesystem [<group>] <command> [<args>]",
 112         NULL
 113 };
 114
 115 static const char * const cmd_df_usage[] = {
 116         "btrfs filesystem df <path>",
 117         "Show space usage information for a mount point",
 118         NULL
 119 };
 120
 121 static char *group_type_str(u64 flag)
 122 {
 123         switch (flag & BTRFS_BLOCK_GROUP_TYPE_MASK) {
 124         case BTRFS_BLOCK_GROUP_DATA:
 125                 return "Data";
 126         case BTRFS_BLOCK_GROUP_SYSTEM:
 127                 return "System";
 128         case BTRFS_BLOCK_GROUP_METADATA:
 129                 return "Metadata";
 130         case BTRFS_BLOCK_GROUP_DATA|BTRFS_BLOCK_GROUP_METADATA:
 131                 return "Data+Metadata";
 132         default:
 133                 return "unknown";
 134         }
 135 }
 136
 137 static char *group_profile_str(u64 flag)
 138 {
 139         switch (flag & BTRFS_BLOCK_GROUP_PROFILE_MASK) {
 140         case 0:
 141                 return "single";
 142         case BTRFS_BLOCK_GROUP_RAID0:
 143                 return "RAID0";
 144         case BTRFS_BLOCK_GROUP_RAID1:
 145                 return "RAID1";
 146         case BTRFS_BLOCK_GROUP_RAID5:
 147                 return "RAID5";
 148         case BTRFS_BLOCK_GROUP_RAID6:
 149                 return "RAID6";
 150         case BTRFS_BLOCK_GROUP_DUP:
 151                 return "DUP";
 152         case BTRFS_BLOCK_GROUP_RAID10:
 153                 return "RAID10";
 154         default:
 155                 return "unknown";
 156         }
 157 }
 158
 159 static int get_df(int fd, struct btrfs_ioctl_space_args **sargs_ret)
 160 {
 161         u64 count = 0;
 162         int ret, e;
 163         struct btrfs_ioctl_space_args *sargs;
 164
 165         sargs = malloc(sizeof(struct btrfs_ioctl_space_args));
 166         if (!sargs)
 167                 return -ENOMEM;
 168
 169         sargs->space_slots = 0;
 170         sargs->total_spaces = 0;
 171
 172         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
 173         e = errno;
 174         if (ret) {
 175                 fprintf(stderr, "ERROR: couldn't get space info - %s\n",
 176                         strerror(e));
 177                 free(sargs);
 178                 return -e;
 179         }
 180         /* This really should never happen */
 181         if (!sargs->total_spaces) {
 182                 free(sargs);
 183                 return -ENOENT;
 184         }
 185         count = sargs->total_spaces;
 186         free(sargs);
 187
 188         sargs = malloc(sizeof(struct btrfs_ioctl_space_args) +
 189                         (count * sizeof(struct btrfs_ioctl_space_info)));
 190         if (!sargs)
 191                 return -ENOMEM;
 192
 193         sargs->space_slots = count;
 194         sargs->total_spaces = 0;
 195         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
 196         e = errno;
 197         if (ret) {
 198                 fprintf(stderr, "ERROR: get space info count %llu - %s\n",
 199                                 count, strerror(e));
 200                 free(sargs);
 201                 return -e;
 202         }
 203         *sargs_ret = sargs;
 204         return 0;
 205 }
 206
 207 static void print_df(struct btrfs_ioctl_space_args *sargs)
 208 {
 209         u64 i;
 210         struct btrfs_ioctl_space_info *sp = sargs->spaces;
 211
 212         for (i = 0; i < sargs->total_spaces; i++, sp++) {
 213                 printf("%s, %s: total=%s, used=%s\n",
 214                         group_type_str(sp->flags),
 215                         group_profile_str(sp->flags),
 216                         pretty_size(sp->total_bytes),
 217                         pretty_size(sp->used_bytes));
 218         }
 219 }
 220
 221 static int cmd_df(int argc, char **argv)
 222 {
 223         struct btrfs_ioctl_space_args *sargs = NULL;
 224         int ret;
 225         int fd;
 226         char *path;
 227         DIR  *dirstream = NULL;
 228
 229         if (check_argc_exact(argc, 2))
 230                 usage(cmd_df_usage);
 231
 232         path = argv[1];
 233
 234         fd = open_file_or_dir(path, &dirstream);
 235         if (fd < 0) {
 236                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
 237                 return 1;
 238         }
 239         ret = get_df(fd, &sargs);
 240
 241         if (!ret && sargs) {
 242                 print_df(sargs);
 243                 free(sargs);
 244         } else {
 245                 fprintf(stderr, "ERROR: get_df failed %s\n", strerror(-ret));
 246         }
 247
 248         close_file_or_dir(fd, dirstream);
 249         return !!ret;
 250 }
 251
 252 static int match_search_item_kernel(__u8 *fsid, char *mnt, char *label,
 253                                         char *search)
 254 {
 255         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 256         int search_len = strlen(search);
 257
 258         search_len = min(search_len, BTRFS_UUID_UNPARSED_SIZE);
 259         uuid_unparse(fsid, uuidbuf);
 260         if (!strncmp(uuidbuf, search, search_len))
 261                 return 1;
 262
 263         if (strlen(label) && strcmp(label, search) == 0)
 264                 return 1;
 265
 266         if (strcmp(mnt, search) == 0)
 267                 return 1;
 268
 269         return 0;
 270 }
 271
 272 static int uuid_search(struct btrfs_fs_devices *fs_devices, char *search)
 273 {
 274         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 275         struct list_head *cur;
 276         struct btrfs_device *device;
 277         int search_len = strlen(search);
 278
 279         search_len = min(search_len, BTRFS_UUID_UNPARSED_SIZE);
 280         uuid_unparse(fs_devices->fsid, uuidbuf);
 281         if (!strncmp(uuidbuf, search, search_len))
 282                 return 1;
 283
 284         list_for_each(cur, &fs_devices->devices) {
 285                 device = list_entry(cur, struct btrfs_device, dev_list);
 286                 if ((device->label && strcmp(device->label, search) == 0) ||
 287                     strcmp(device->name, search) == 0)
 288                         return 1;
 289         }
 290         return 0;
 291 }
 292
 293 /*
 294  * Sort devices by devid, ascending
 295  */
 296 static int cmp_device_id(void *priv, struct list_head *a,
 297                 struct list_head *b)
 298 {
 299         const struct btrfs_device *da = list_entry(a, struct btrfs_device,
 300                         dev_list);
 301         const struct btrfs_device *db = list_entry(b, struct btrfs_device,
 302                         dev_list);
 303
 304         return da->devid < db->devid ? -1 :
 305                 da->devid > db->devid ? 1 : 0;
 306 }
 307
 308 static void print_one_uuid(struct btrfs_fs_devices *fs_devices)
 309 {
 310         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 311         struct list_head *cur;
 312         struct btrfs_device *device;
 313         u64 devs_found = 0;
 314         u64 total;
 315
 316         if (add_seen_fsid(fs_devices->fsid))
 317                 return;
 318
 319         uuid_unparse(fs_devices->fsid, uuidbuf);
 320         device = list_entry(fs_devices->devices.next, struct btrfs_device,
 321                             dev_list);
 322         if (device->label && device->label[0])
 323                 printf("Label: '%s' ", device->label);
 324         else
 325                 printf("Label: none ");
 326
 327
 328         total = device->total_devs;
 329         printf(" uuid: %s\n\tTotal devices %llu FS bytes used %s\n", uuidbuf,
 330                (unsigned long long)total,
 331                pretty_size(device->super_bytes_used));
 332
 333         list_sort(NULL, &fs_devices->devices, cmp_device_id);
 334         list_for_each(cur, &fs_devices->devices) {
 335                 device = list_entry(cur, struct btrfs_device, dev_list);
 336
 337                 printf("\tdevid %4llu size %s used %s path %s\n",
 338                        (unsigned long long)device->devid,
 339                        pretty_size(device->total_bytes),
 340                        pretty_size(device->bytes_used), device->name);
 341
 342                 devs_found++;
 343         }
 344         if (devs_found < total) {
 345                 printf("\t*** Some devices missing\n");
 346         }
 347         printf("\n");
 348 }
 349
 350 /* adds up all the used spaces as reported by the space info ioctl
 351  */
 352 static u64 calc_used_bytes(struct btrfs_ioctl_space_args *si)
 353 {
 354         u64 ret = 0;
 355         int i;
 356         for (i = 0; i < si->total_spaces; i++)
 357                 ret += si->spaces[i].used_bytes;
 358         return ret;
 359 }
 360
 361 static int print_one_fs(struct btrfs_ioctl_fs_info_args *fs_info,
 362                 struct btrfs_ioctl_dev_info_args *dev_info,
 363                 struct btrfs_ioctl_space_args *space_info,
 364                 char *label, char *path)
 365 {
 366         int i;
 367         int fd;
 368         int missing = 0;
 369         char uuidbuf[BTRFS_UUID_UNPARSED_SIZE];
 370         struct btrfs_ioctl_dev_info_args *tmp_dev_info;
 371         int ret;
 372
 373         ret = add_seen_fsid(fs_info->fsid);
 374         if (ret == -EEXIST)
 375                 return 0;
 376         else if (ret)
 377                 return ret;
 378
 379         uuid_unparse(fs_info->fsid, uuidbuf);
 380         if (label && strlen(label))
 381                 printf("Label: '%s' ", label);
 382         else
 383                 printf("Label: none ");
 384
 385         printf(" uuid: %s\n\tTotal devices %llu FS bytes used %s\n", uuidbuf,
 386                         fs_info->num_devices,
 387                         pretty_size(calc_used_bytes(space_info)));
 388
 389         for (i = 0; i < fs_info->num_devices; i++) {
 390                 tmp_dev_info = (struct btrfs_ioctl_dev_info_args *)&dev_info[i];
 391
 392                 /* Add check for missing devices even mounted */
 393                 fd = open((char *)tmp_dev_info->path, O_RDONLY);
 394                 if (fd < 0) {
 395                         missing = 1;
 396                         continue;
 397                 }
 398                 close(fd);
 399                 printf("\tdevid %4llu size %s used %s path %s\n",
 400                         tmp_dev_info->devid,
 401                         pretty_size(tmp_dev_info->total_bytes),
 402                         pretty_size(tmp_dev_info->bytes_used),
 403                         tmp_dev_info->path);
 404         }
 405
 406         if (missing)
 407                 printf("\t*** Some devices missing\n");
 408         printf("\n");
 409         return 0;
 410 }
 411
 412 /* This function checks if the given input parameter is
 413  * an uuid or a path
 414  * return -1: some error in the given input
 415  * return 0: unknow input
 416  * return 1: given input is uuid
 417  * return 2: given input is path
 418  */
 419 static int check_arg_type(char *input)
 420 {
 421         uuid_t  out;
 422         char path[PATH_MAX];
 423
 424         if (!input)
 425                 return -EINVAL;
 426
 427         if (realpath(input, path)) {
 428                 if (is_block_device(path) == 1)
 429                         return BTRFS_ARG_BLKDEV;
 430
 431                 if (is_mount_point(path) == 1)
 432                         return BTRFS_ARG_MNTPOINT;
 433
 434                 return BTRFS_ARG_UNKNOWN;
 435         }
 436
 437         if (strlen(input) == (BTRFS_UUID_UNPARSED_SIZE - 1) &&
 438                 !uuid_parse(input, out))
 439                 return BTRFS_ARG_UUID;
 440
 441         return BTRFS_ARG_UNKNOWN;
 442 }
 443
 444 static int btrfs_scan_kernel(void *search)
 445 {
 446         int ret = 0, fd;
 447         int found = 0;
 448         FILE *f;
 449         struct mntent *mnt;
 450         struct btrfs_ioctl_fs_info_args fs_info_arg;
 451         struct btrfs_ioctl_dev_info_args *dev_info_arg = NULL;
 452         struct btrfs_ioctl_space_args *space_info_arg = NULL;
 453         char label[BTRFS_LABEL_SIZE];
 454
 455         f = setmntent("/proc/self/mounts", "r");
 456         if (f == NULL)
 457                 return 1;
 458
 459         memset(label, 0, sizeof(label));
 460         while ((mnt = getmntent(f)) != NULL) {
 461                 if (strcmp(mnt->mnt_type, "btrfs"))
 462                         continue;
 463                 ret = get_fs_info(mnt->mnt_dir, &fs_info_arg,
 464                                 &dev_info_arg);
 465                 if (ret)
 466                         goto out;
 467
 468                 if (get_label_mounted(mnt->mnt_dir, label)) {
 469                         kfree(dev_info_arg);
 470                         goto out;
 471                 }
 472                 if (search && !match_search_item_kernel(fs_info_arg.fsid,
 473                                         mnt->mnt_dir, label, search)) {
 474                         kfree(dev_info_arg);
 475                         continue;
 476                 }
 477
 478                 fd = open(mnt->mnt_dir, O_RDONLY);
 479                 if ((fd != -1) && !get_df(fd, &space_info_arg)) {
 480                         print_one_fs(&fs_info_arg, dev_info_arg,
 481                                         space_info_arg, label, mnt->mnt_dir);
 482                         kfree(space_info_arg);
 483                         memset(label, 0, sizeof(label));
 484                         found = 1;
 485                 }
 486                 if (fd != -1)
 487                         close(fd);
 488                 kfree(dev_info_arg);
 489         }
 490
 491 out:
 492         endmntent(f);
 493         return !found;
 494 }
 495
 496 static int dev_to_fsid(char *dev, __u8 *fsid)
 497 {
 498         struct btrfs_super_block *disk_super;
 499         char *buf;
 500         int ret;
 501         int fd;
 502
 503         buf = malloc(4096);
 504         if (!buf)
 505                 return -ENOMEM;
 506
 507         fd = open(dev, O_RDONLY);
 508         if (fd < 0) {
 509                 ret = -errno;
 510                 free(buf);
 511                 return ret;
 512         }
 513
 514         disk_super = (struct btrfs_super_block *)buf;
 515         ret = btrfs_read_dev_super(fd, disk_super,
 516                         BTRFS_SUPER_INFO_OFFSET);
 517         if (ret)
 518                 goto out;
 519
 520         memcpy(fsid, disk_super->fsid, BTRFS_FSID_SIZE);
 521         ret = 0;
 522
 523 out:
 524         close(fd);
 525         free(buf);
 526         return ret;
 527 }
 528
 529 static const char * const cmd_show_usage[] = {
 530         "btrfs filesystem show [options] [<path>|<uuid>|<device>|label]",
 531         "Show the structure of a filesystem",
 532         "-d|--all-devices   show only disks under /dev containing btrfs filesystem",
 533         "-m|--mounted       show only mounted btrfs",
 534         "If no argument is given, structure of all present filesystems is shown.",
 535         NULL
 536 };
 537
 538 static int cmd_show(int argc, char **argv)
 539 {
 540         struct list_head *all_uuids;
 541         struct btrfs_fs_devices *fs_devices;
 542         struct list_head *cur_uuid;
 543         char *search = NULL;
 544         int ret;
 545         int where = BTRFS_SCAN_LBLKID;
 546         int type = 0;
 547         char mp[BTRFS_PATH_NAME_MAX + 1];
 548         char path[PATH_MAX];
 549         __u8 fsid[BTRFS_FSID_SIZE];
 550         char uuid_buf[37];
 551         int found = 0;
 552
 553         while (1) {
 554                 int long_index;
 555                 static struct option long_options[] = {
 556                         { "all-devices", no_argument, NULL, 'd'},
 557                         { "mounted", no_argument, NULL, 'm'},
 558                         { NULL, no_argument, NULL, 0 },
 559                 };
 560                 int c = getopt_long(argc, argv, "dm", long_options,
 561                                         &long_index);
 562                 if (c < 0)
 563                         break;
 564                 switch (c) {
 565                 case 'd':
 566                         where = BTRFS_SCAN_DEV;
 567                         break;
 568                 case 'm':
 569                         where = BTRFS_SCAN_MOUNTED;
 570                         break;
 571                 default:
 572                         usage(cmd_show_usage);
 573                 }
 574         }
 575
 576         if (check_argc_max(argc, optind + 1))
 577                 usage(cmd_show_usage);
 578
 579         if (argc > optind) {
 580                 search = argv[optind];
 581                 if (strlen(search) == 0)
 582                         usage(cmd_show_usage);
 583                 type = check_arg_type(search);
 584                 /*
 585                  * needs spl handling if input arg is block dev
 586                  * And if input arg is mount-point just print it
 587                  * right away
 588                  */
 589                 if (type == BTRFS_ARG_BLKDEV) {
 590                         if (where == BTRFS_SCAN_DEV) {
 591                                 /* we need to do this because
 592                                  * legacy BTRFS_SCAN_DEV
 593                                  * provides /dev/dm-x paths
 594                                  */
 595                                 if (realpath(search, path))
 596                                         search = path;
 597                         } else {
 598                                 ret = get_btrfs_mount(search,
 599                                                 mp, sizeof(mp));
 600                                 if (!ret) {
 601                                         /* given block dev is mounted*/
 602                                         search = mp;
 603                                         type = BTRFS_ARG_MNTPOINT;
 604                                 } else {
 605                                         ret = dev_to_fsid(search, fsid);
 606                                         if (ret) {
 607                                                 fprintf(stderr,
 608                                                         "ERROR: No btrfs on %s\n",
 609                                                         search);
 610                                                 return 1;
 611                                         }
 612                                         uuid_unparse(fsid, uuid_buf);
 613                                         search = uuid_buf;
 614                                         type = BTRFS_ARG_UUID;
 615                                         goto devs_only;
 616                                 }
 617                         }
 618                 }
 619         }
 620
 621         if (where == BTRFS_SCAN_DEV)
 622                 goto devs_only;
 623
 624         /* show mounted btrfs */
 625         ret = btrfs_scan_kernel(search);
 626         if (search && !ret) {
 627                 /* since search is found we are done */
 628                 goto out;
 629         }
 630
 631         /* shows mounted only */
 632         if (where == BTRFS_SCAN_MOUNTED)
 633                 goto out;
 634
 635 devs_only:
 636         ret = scan_for_btrfs(where, !BTRFS_UPDATE_KERNEL);
 637
 638         if (ret) {
 639                 fprintf(stderr, "ERROR: %d while scanning\n", ret);
 640                 return 1;
 641         }
 642
 643         all_uuids = btrfs_scanned_uuids();
 644         list_for_each(cur_uuid, all_uuids) {
 645                 fs_devices = list_entry(cur_uuid, struct btrfs_fs_devices,
 646                                         list);
 647                 if (search && uuid_search(fs_devices, search) == 0)
 648                         continue;
 649
 650                 print_one_uuid(fs_devices);
 651                 found = 1;
 652         }
 653         if (search && !found)
 654                 ret = 1;
 655
 656         while (!list_empty(all_uuids)) {
 657                 fs_devices = list_entry(all_uuids->next,
 658                                         struct btrfs_fs_devices, list);
 659                 list_del(&fs_devices->list);
 660                 btrfs_close_devices(fs_devices);
 661         }
 662 out:
 663         printf("%s\n", BTRFS_BUILD_VERSION);
 664         free_seen_fsid();
 665         return ret;
 666 }
 667
 668 static const char * const cmd_sync_usage[] = {
 669         "btrfs filesystem sync <path>",
 670         "Force a sync on a filesystem",
 671         NULL
 672 };
 673
 674 static int cmd_sync(int argc, char **argv)
 675 {
 676         int     fd, res, e;
 677         char    *path;
 678         DIR     *dirstream = NULL;
 679
 680         if (check_argc_exact(argc, 2))
 681                 usage(cmd_sync_usage);
 682
 683         path = argv[1];
 684
 685         fd = open_file_or_dir(path, &dirstream);
 686         if (fd < 0) {
 687                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
 688                 return 1;
 689         }
 690
 691         printf("FSSync '%s'\n", path);
 692         res = ioctl(fd, BTRFS_IOC_SYNC);
 693         e = errno;
 694         close_file_or_dir(fd, dirstream);
 695         if( res < 0 ){
 696                 fprintf(stderr, "ERROR: unable to fs-syncing '%s' - %s\n",
 697                         path, strerror(e));
 698                 return 1;
 699         }
 700
 701         return 0;
 702 }
 703
 704 static int parse_compress_type(char *s)
 705 {
 706         if (strcmp(optarg, "zlib") == 0)
 707                 return BTRFS_COMPRESS_ZLIB;
 708         else if (strcmp(optarg, "lzo") == 0)
 709                 return BTRFS_COMPRESS_LZO;
 710         else {
 711                 fprintf(stderr, "Unknown compress type %s\n", s);
 712                 exit(1);
 713         };
 714 }
 715
 716 static const char * const cmd_defrag_usage[] = {
 717         "btrfs filesystem defragment [options] <file>|<dir> [<file>|<dir>...]",
 718         "Defragment a file or a directory",
 719         "",
 720         "-v             be verbose",
 721         "-r             defragment files recursively",
 722         "-c[zlib,lzo]   compress the file while defragmenting",
 723         "-f             flush data to disk immediately after defragmenting",
 724         "-s start       defragment only from byte onward",
 725         "-l len         defragment only up to len bytes",
 726         "-t size        minimal size of file to be considered for defragmenting",
 727         NULL
 728 };
 729
 730 static int do_defrag(int fd, int fancy_ioctl,
 731                 struct btrfs_ioctl_defrag_range_args *range)
 732 {
 733         int ret;
 734
 735         if (!fancy_ioctl)
 736                 ret = ioctl(fd, BTRFS_IOC_DEFRAG, NULL);
 737         else
 738                 ret = ioctl(fd, BTRFS_IOC_DEFRAG_RANGE, range);
 739
 740         return ret;
 741 }
 742
 743 static int defrag_global_fancy_ioctl;
 744 static struct btrfs_ioctl_defrag_range_args defrag_global_range;
 745 static int defrag_global_verbose;
 746 static int defrag_global_errors;
 747 static int defrag_callback(const char *fpath, const struct stat *sb,
 748                 int typeflag, struct FTW *ftwbuf)
 749 {
 750         int ret = 0;
 751         int e = 0;
 752         int fd = 0;
 753
 754         if ((typeflag == FTW_F) && S_ISREG(sb->st_mode)) {
 755                 if (defrag_global_verbose)
 756                         printf("%s\n", fpath);
 757                 fd = open(fpath, O_RDWR);
 758                 e = errno;
 759                 if (fd < 0)
 760                         goto error;
 761                 ret = do_defrag(fd, defrag_global_fancy_ioctl, &defrag_global_range);
 762                 e = errno;
 763                 close(fd);
 764                 if (ret && e == ENOTTY && defrag_global_fancy_ioctl) {
 765                         fprintf(stderr, "ERROR: defrag range ioctl not "
 766                                 "supported in this kernel, please try "
 767                                 "without any options.\n");
 768                         defrag_global_errors++;
 769                         return ENOTTY;
 770                 }
 771                 if (ret)
 772                         goto error;
 773         }
 774         return 0;
 775
 776 error:
 777         fprintf(stderr, "ERROR: defrag failed on %s - %s\n", fpath, strerror(e));
 778         defrag_global_errors++;
 779         return 0;
 780 }
 781
 782 static int cmd_defrag(int argc, char **argv)
 783 {
 784         int fd;
 785         int flush = 0;
 786         u64 start = 0;
 787         u64 len = (u64)-1;
 788         u32 thresh = 0;
 789         int i;
 790         int recursive = 0;
 791         int ret = 0;
 792         struct btrfs_ioctl_defrag_range_args range;
 793         int e = 0;
 794         int compress_type = BTRFS_COMPRESS_NONE;
 795         DIR *dirstream;
 796
 797         defrag_global_errors = 0;
 798         defrag_global_verbose = 0;
 799         defrag_global_errors = 0;
 800         defrag_global_fancy_ioctl = 0;
 801         optind = 1;
 802         while(1) {
 803                 int c = getopt(argc, argv, "vrc::fs:l:t:");
 804                 if (c < 0)
 805                         break;
 806
 807                 switch(c) {
 808                 case 'c':
 809                         compress_type = BTRFS_COMPRESS_ZLIB;
 810                         if (optarg)
 811                                 compress_type = parse_compress_type(optarg);
 812                         defrag_global_fancy_ioctl = 1;
 813                         break;
 814                 case 'f':
 815                         flush = 1;
 816                         defrag_global_fancy_ioctl = 1;
 817                         break;
 818                 case 'v':
 819                         defrag_global_verbose = 1;
 820                         break;
 821                 case 's':
 822                         start = parse_size(optarg);
 823                         defrag_global_fancy_ioctl = 1;
 824                         break;
 825                 case 'l':
 826                         len = parse_size(optarg);
 827                         defrag_global_fancy_ioctl = 1;
 828                         break;
 829                 case 't':
 830                         thresh = parse_size(optarg);
 831                         defrag_global_fancy_ioctl = 1;
 832                         break;
 833                 case 'r':
 834                         recursive = 1;
 835                         break;
 836                 default:
 837                         usage(cmd_defrag_usage);
 838                 }
 839         }
 840
 841         if (check_argc_min(argc - optind, 1))
 842                 usage(cmd_defrag_usage);
 843
 844         memset(&defrag_global_range, 0, sizeof(range));
 845         defrag_global_range.start = start;
 846         defrag_global_range.len = len;
 847         defrag_global_range.extent_thresh = thresh;
 848         if (compress_type) {
 849                 defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_COMPRESS;
 850                 defrag_global_range.compress_type = compress_type;
 851         }
 852         if (flush)
 853                 defrag_global_range.flags |= BTRFS_DEFRAG_RANGE_START_IO;
 854
 855         for (i = optind; i < argc; i++) {
 856                 struct stat st;
 857
 858                 dirstream = NULL;
 859                 fd = open_file_or_dir(argv[i], &dirstream);
 860                 if (fd < 0) {
 861                         fprintf(stderr, "ERROR: failed to open %s - %s\n", argv[i],
 862                                         strerror(errno));
 863                         defrag_global_errors++;
 864                         close_file_or_dir(fd, dirstream);
 865                         continue;
 866                 }
 867                 if (fstat(fd, &st)) {
 868                         fprintf(stderr, "ERROR: failed to stat %s - %s\n",
 869                                         argv[i], strerror(errno));
 870                         defrag_global_errors++;
 871                         close_file_or_dir(fd, dirstream);
 872                         continue;
 873                 }
 874                 if (!(S_ISDIR(st.st_mode) || S_ISREG(st.st_mode))) {
 875                         fprintf(stderr,
 876                             "ERROR: %s is not a directory or a regular file\n",
 877                             argv[i]);
 878                         defrag_global_errors++;
 879                         close_file_or_dir(fd, dirstream);
 880                         continue;
 881                 }
 882                 if (recursive) {
 883                         if (S_ISDIR(st.st_mode)) {
 884                                 ret = nftw(argv[i], defrag_callback, 10,
 885                                                 FTW_MOUNT | FTW_PHYS);
 886                                 if (ret == ENOTTY)
 887                                         exit(1);
 888                                 /* errors are handled in the callback */
 889                                 ret = 0;
 890                         } else {
 891                                 if (defrag_global_verbose)
 892                                         printf("%s\n", argv[i]);
 893                                 ret = do_defrag(fd, defrag_global_fancy_ioctl,
 894                                                 &defrag_global_range);
 895                                 e = errno;
 896                         }
 897                 } else {
 898                         if (defrag_global_verbose)
 899                                 printf("%s\n", argv[i]);
 900                         ret = do_defrag(fd, defrag_global_fancy_ioctl,
 901                                         &defrag_global_range);
 902                         e = errno;
 903                 }
 904                 close_file_or_dir(fd, dirstream);
 905                 if (ret && e == ENOTTY && defrag_global_fancy_ioctl) {
 906                         fprintf(stderr, "ERROR: defrag range ioctl not "
 907                                 "supported in this kernel, please try "
 908                                 "without any options.\n");
 909                         defrag_global_errors++;
 910                         break;
 911                 }
 912                 if (ret) {
 913                         fprintf(stderr, "ERROR: defrag failed on %s - %s\n",
 914                                 argv[i], strerror(e));
 915                         defrag_global_errors++;
 916                 }
 917         }
 918         if (defrag_global_verbose)
 919                 printf("%s\n", BTRFS_BUILD_VERSION);
 920         if (defrag_global_errors)
 921                 fprintf(stderr, "total %d failures\n", defrag_global_errors);
 922
 923         return !!defrag_global_errors;
 924 }
 925
 926 static const char * const cmd_resize_usage[] = {
 927         "btrfs filesystem resize [devid:][+/-]<newsize>[kKmMgGtTpPeE]|[devid:]max <path>",
 928         "Resize a filesystem",
 929         "If 'max' is passed, the filesystem will occupy all available space",
 930         "on the device 'devid'.",
 931         "[kK] means KiB, which denotes 1KiB = 1024B, 1MiB = 1024KiB, etc.",
 932         NULL
 933 };
 934
 935 static int cmd_resize(int argc, char **argv)
 936 {
 937         struct btrfs_ioctl_vol_args     args;
 938         int     fd, res, len, e;
 939         char    *amount, *path;
 940         DIR     *dirstream = NULL;
 941
 942         if (check_argc_exact(argc, 3))
 943                 usage(cmd_resize_usage);
 944
 945         amount = argv[1];
 946         path = argv[2];
 947
 948         len = strlen(amount);
 949         if (len == 0 || len >= BTRFS_VOL_NAME_MAX) {
 950                 fprintf(stderr, "ERROR: size value too long ('%s)\n",
 951                         amount);
 952                 return 1;
 953         }
 954
 955         fd = open_file_or_dir(path, &dirstream);
 956         if (fd < 0) {
 957                 fprintf(stderr, "ERROR: can't access '%s'\n", path);
 958                 return 1;
 959         }
 960
 961         printf("Resize '%s' of '%s'\n", path, amount);
 962         strncpy_null(args.name, amount);
 963         res = ioctl(fd, BTRFS_IOC_RESIZE, &args);
 964         e = errno;
 965         close_file_or_dir(fd, dirstream);
 966         if( res < 0 ){
 967                 fprintf(stderr, "ERROR: unable to resize '%s' - %s\n",
 968                         path, strerror(e));
 969                 return 1;
 970         }
 971         return 0;
 972 }
 973
 974 static const char * const cmd_label_usage[] = {
 975         "btrfs filesystem label [<device>|<mount_point>] [<newlabel>]",
 976         "Get or change the label of a filesystem",
 977         "With one argument, get the label of filesystem on <device>.",
 978         "If <newlabel> is passed, set the filesystem label to <newlabel>.",
 979         NULL
 980 };
 981
 982 static int cmd_label(int argc, char **argv)
 983 {
 984         if (check_argc_min(argc, 2) || check_argc_max(argc, 3))
 985                 usage(cmd_label_usage);
 986
 987         if (argc > 2) {
 988                 return set_label(argv[1], argv[2]);
 989         } else {
 990                 char label[BTRFS_LABEL_SIZE];
 991                 int ret;
 992
 993                 ret = get_label(argv[1], label);
 994                 if (!ret)
 995                         fprintf(stdout, "%s\n", label);
 996
 997                 return ret;
 998         }
 999 }
1000
1001 const struct cmd_group filesystem_cmd_group = {
1002         filesystem_cmd_group_usage, NULL, {
1003                 { "df", cmd_df, cmd_df_usage, NULL, 0 },
1004                 { "show", cmd_show, cmd_show_usage, NULL, 0 },
1005                 { "sync", cmd_sync, cmd_sync_usage, NULL, 0 },
1006                 { "defragment", cmd_defrag, cmd_defrag_usage, NULL, 0 },
1007                 { "balance", cmd_balance, NULL, &balance_cmd_group, 1 },
1008                 { "resize", cmd_resize, cmd_resize_usage, NULL, 0 },
1009                 { "label", cmd_label, cmd_label_usage, NULL, 0 },
1010                 NULL_CMD_STRUCT
1011         }
1012 };
1013
1014 int cmd_filesystem(int argc, char **argv)
1015 {
1016         return handle_command_group(&filesystem_cmd_group, argc, argv);
1017 }