btrfs-progs: check: fix missing newlines
[platform/upstream/btrfs-progs.git] / cmds-device.c
index c2f3a40..fa0830f 100644 (file)
@@ -49,7 +49,7 @@ static const char * const cmd_device_add_usage[] = {
 static int cmd_device_add(int argc, char **argv)
 {
        char    *mntpnt;
-       int     i, fdmnt, ret=0, e;
+       int i, fdmnt, ret = 0;
        DIR     *dirstream = NULL;
        int discard = 1;
        int force = 0;
@@ -102,13 +102,14 @@ static int cmd_device_add(int argc, char **argv)
 
                devfd = open(argv[i], O_RDWR);
                if (devfd < 0) {
-                       fprintf(stderr, "ERROR: Unable to open device '%s'\n", argv[i]);
+                       error("unable to open device '%s'", argv[i]);
                        ret++;
                        continue;
                }
 
-               res = btrfs_prepare_device(devfd, argv[i], 1, &dev_block_count,
-                                          0, discard);
+               res = btrfs_prepare_device(devfd, argv[i], &dev_block_count, 0,
+                               PREP_DEVICE_ZERO_END | PREP_DEVICE_VERBOSE |
+                               (discard ? PREP_DEVICE_DISCARD : 0));
                close(devfd);
                if (res) {
                        ret++;
@@ -117,8 +118,7 @@ static int cmd_device_add(int argc, char **argv)
 
                path = canonicalize_path(argv[i]);
                if (!path) {
-                       fprintf(stderr,
-                               "ERROR: Could not canonicalize pathname '%s': %s\n",
+                       error("could not canonicalize pathname '%s': %s",
                                argv[i], strerror(errno));
                        ret++;
                        goto error_out;
@@ -127,10 +127,9 @@ static int cmd_device_add(int argc, char **argv)
                memset(&ioctl_args, 0, sizeof(ioctl_args));
                strncpy_null(ioctl_args.name, path);
                res = ioctl(fdmnt, BTRFS_IOC_ADD_DEV, &ioctl_args);
-               e = errno;
                if (res < 0) {
-                       fprintf(stderr, "ERROR: error adding the device '%s' - %s\n",
-                               path, strerror(e));
+                       error("error adding device '%s': %s",
+                               path, strerror(errno));
                        ret++;
                }
                free(path);
@@ -145,10 +144,12 @@ static int _cmd_device_remove(int argc, char **argv,
                const char * const *usagestr)
 {
        char    *mntpnt;
-       int     i, fdmnt, ret=0, e;
+       int i, fdmnt, ret = 0;
        DIR     *dirstream = NULL;
 
-       if (check_argc_min(argc, 3))
+       clean_args_no_options(argc, argv, usagestr);
+
+       if (check_argc_min(argc - optind, 2))
                usage(usagestr);
 
        mntpnt = argv[argc - 1];
@@ -157,30 +158,62 @@ static int _cmd_device_remove(int argc, char **argv,
        if (fdmnt < 0)
                return 1;
 
-       for(i=1 ; i < argc - 1; i++ ){
+       for(i = optind; i < argc - 1; i++) {
                struct  btrfs_ioctl_vol_args arg;
+               struct btrfs_ioctl_vol_args_v2 argv2 = {0};
+               int is_devid = 0;
                int     res;
 
-               if (is_block_device(argv[i]) != 1) {
-                       fprintf(stderr,
-                               "ERROR: %s is not a block device\n", argv[i]);
+               if (string_is_numerical(argv[i])) {
+                       argv2.devid = arg_strtou64(argv[i]);
+                       argv2.flags = BTRFS_DEVICE_SPEC_BY_ID;
+                       is_devid = 1;
+               } else if (is_block_device(argv[i]) == 1 ||
+                               strcmp(argv[i], "missing") == 0) {
+                       strncpy_null(argv2.name, argv[i]);
+               } else {
+                       error("not a block device: %s", argv[i]);
                        ret++;
                        continue;
                }
-               memset(&arg, 0, sizeof(arg));
-               strncpy_null(arg.name, argv[i]);
-               res = ioctl(fdmnt, BTRFS_IOC_RM_DEV, &arg);
-               e = errno;
+
+               /*
+                * Positive values are from BTRFS_ERROR_DEV_*,
+                * otherwise it's a generic error, one of errnos
+                */
+               res = ioctl(fdmnt, BTRFS_IOC_RM_DEV_V2, &argv2);
+
+               /*
+                * If BTRFS_IOC_RM_DEV_V2 is not supported we get ENOTTY and if
+                * argv2.flags includes a flag which kernel doesn't understand then
+                * we shall get EOPNOTSUPP
+                */
+               if (res < 0 && (errno == ENOTTY || errno == EOPNOTSUPP)) {
+                       if (is_devid) {
+                               error("device delete by id failed: %s",
+                                                       strerror(errno));
+                               ret++;
+                               continue;
+                       }
+                       memset(&arg, 0, sizeof(arg));
+                       strncpy_null(arg.name, argv[i]);
+                       res = ioctl(fdmnt, BTRFS_IOC_RM_DEV, &arg);
+               }
+
                if (res) {
                        const char *msg;
 
                        if (res > 0)
                                msg = btrfs_err_str(res);
                        else
-                               msg = strerror(e);
-                       fprintf(stderr,
-                               "ERROR: error removing the device '%s' - %s\n",
-                               argv[i], msg);
+                               msg = strerror(errno);
+                       if (is_devid) {
+                               error("error removing devid %llu: %s",
+                                       (unsigned long long)argv2.devid, msg);
+                       } else {
+                               error("error removing device '%s': %s",
+                                       argv[i], msg);
+                       }
                        ret++;
                }
        }
@@ -190,7 +223,7 @@ static int _cmd_device_remove(int argc, char **argv,
 }
 
 static const char * const cmd_device_remove_usage[] = {
-       "btrfs device remove <device> [<device>...] <path>",
+       "btrfs device remove <device>|<devid> [<device>|<devid>...] <path>",
        "Remove a device from a filesystem",
        NULL
 };
@@ -201,7 +234,7 @@ static int cmd_device_remove(int argc, char **argv)
 }
 
 static const char * const cmd_device_delete_usage[] = {
-       "btrfs device delete <device> [<device>...] <path>",
+       "btrfs device delete <device>|<devid> [<device>|<devid>...] <path>",
        "Remove a device from a filesystem",
        NULL
 };
@@ -221,11 +254,10 @@ static const char * const cmd_device_scan_usage[] = {
 static int cmd_device_scan(int argc, char **argv)
 {
        int i;
-       int devstart = 1;
+       int devstart;
        int all = 0;
        int ret = 0;
 
-       optind = 1;
        while (1) {
                int c;
                static const struct option long_options[] = {
@@ -244,18 +276,17 @@ static int cmd_device_scan(int argc, char **argv)
                        usage(cmd_device_scan_usage);
                }
        }
+       devstart = optind;
 
-       if (all && check_argc_max(argc, 2))
+       if (all && check_argc_max(argc - optind, 1))
                usage(cmd_device_scan_usage);
 
-       if (all || argc == 1) {
+       if (all || argc - optind == 0) {
                printf("Scanning for Btrfs filesystems\n");
-               ret = btrfs_scan_lblkid();
-               if (ret)
-                       fprintf(stderr, "ERROR: error %d while scanning\n", ret);
+               ret = btrfs_scan_devices();
+               error_on(ret, "error %d while scanning", ret);
                ret = btrfs_register_all_devices();
-               if (ret)
-                       fprintf(stderr, "ERROR: error %d while registering\n", ret);
+               error_on(ret, "there are %d errors while registering devices", ret);
                goto out;
        }
 
@@ -263,15 +294,13 @@ static int cmd_device_scan(int argc, char **argv)
                char *path;
 
                if (is_block_device(argv[i]) != 1) {
-                       fprintf(stderr,
-                               "ERROR: %s is not a block device\n", argv[i]);
+                       error("not a block device: %s", argv[i]);
                        ret = 1;
                        goto out;
                }
                path = canonicalize_path(argv[i]);
                if (!path) {
-                       fprintf(stderr,
-                               "ERROR: Could not canonicalize path '%s': %s\n",
+                       error("could not canonicalize path '%s': %s",
                                argv[i], strerror(errno));
                        ret = 1;
                        goto out;
@@ -302,7 +331,9 @@ static int cmd_device_ready(int argc, char **argv)
        int     ret;
        char    *path;
 
-       if (check_argc_min(argc, 2))
+       clean_args_no_options(argc, argv, cmd_device_ready_usage);
+
+       if (check_argc_exact(argc - optind, 1))
                usage(cmd_device_ready_usage);
 
        fd = open("/dev/btrfs-control", O_RDWR);
@@ -311,18 +342,16 @@ static int cmd_device_ready(int argc, char **argv)
                return 1;
        }
 
-       path = canonicalize_path(argv[argc - 1]);
+       path = canonicalize_path(argv[optind]);
        if (!path) {
-               fprintf(stderr,
-                       "ERROR: Could not canonicalize pathname '%s': %s\n",
-                       argv[argc - 1], strerror(errno));
+               error("could not canonicalize pathname '%s': %s",
+                       argv[optind], strerror(errno));
                ret = 1;
                goto out;
        }
 
        if (is_block_device(path) != 1) {
-               fprintf(stderr,
-                       "ERROR: %s is not a block device\n", path);
+               error("not a block device: %s", path);
                ret = 1;
                goto out;
        }
@@ -331,9 +360,8 @@ static int cmd_device_ready(int argc, char **argv)
        strncpy_null(args.name, path);
        ret = ioctl(fd, BTRFS_IOC_DEVICES_READY, &args);
        if (ret < 0) {
-               fprintf(stderr, "ERROR: unable to determine if the device '%s'"
-                       " is ready for mounting - %s\n", path,
-                       strerror(errno));
+               error("unable to determine if device '%s' is ready for mount: %s",
+                       path, strerror(errno));
                ret = 1;
        }
 
@@ -364,7 +392,6 @@ static int cmd_device_stats(int argc, char **argv)
        __u64 flags = 0;
        DIR *dirstream = NULL;
 
-       optind = 1;
        while ((c = getopt(argc, argv, "z")) != -1) {
                switch (c) {
                case 'z':
@@ -376,8 +403,7 @@ static int cmd_device_stats(int argc, char **argv)
                }
        }
 
-       argc = argc - optind;
-       if (check_argc_exact(argc, 1))
+       if (check_argc_exact(argc - optind, 1))
                usage(cmd_device_stats_usage);
 
        dev_path = argv[optind];
@@ -388,13 +414,13 @@ static int cmd_device_stats(int argc, char **argv)
 
        ret = get_fs_info(dev_path, &fi_args, &di_args);
        if (ret) {
-               fprintf(stderr, "ERROR: getting dev info for devstats failed: "
-                               "%s\n", strerror(-ret));
+               error("getting dev info for devstats failed: %s",
+                       strerror(-ret));
                err = 1;
                goto out;
        }
        if (!fi_args.num_devices) {
-               fprintf(stderr, "ERROR: no devices found\n");
+               error("no devices found");
                err = 1;
                goto out;
        }
@@ -412,15 +438,25 @@ static int cmd_device_stats(int argc, char **argv)
                args.flags = flags;
 
                if (ioctl(fdmnt, BTRFS_IOC_GET_DEV_STATS, &args) < 0) {
-                       fprintf(stderr,
-                               "ERROR: ioctl(BTRFS_IOC_GET_DEV_STATS) on %s failed: %s\n",
-                               path, strerror(errno));
+                       error("DEV_STATS ioctl failed on %s: %s",
+                             path, strerror(errno));
                        err = 1;
                } else {
                        char *canonical_path;
 
                        canonical_path = canonicalize_path((char *)path);
 
+                       /* No path when device is missing. */
+                       if (!canonical_path) {
+                               canonical_path = malloc(32);
+                               if (!canonical_path) {
+                                       error("not enough memory for path buffer");
+                                       goto out;
+                               }
+                               snprintf(canonical_path, 32,
+                                        "devid:%llu", args.devid);
+                       }
+
                        if (args.nr_items >= BTRFS_DEV_STAT_WRITE_ERRS + 1)
                                printf("[%s].write_io_errs   %llu\n",
                                       canonical_path,
@@ -461,7 +497,7 @@ out:
 static const char * const cmd_device_usage_usage[] = {
        "btrfs device usage [options] <path> [<path>..]",
        "Show detailed information about internal allocations in devices.",
-       HELPINFO_OUTPUT_UNIT_DF,
+       HELPINFO_UNITS_SHORT_LONG,
        NULL
 };
 
@@ -498,35 +534,35 @@ static int cmd_device_usage(int argc, char **argv)
 {
        unsigned unit_mode;
        int ret = 0;
-       int more_than_one = 0;
        int i;
 
        unit_mode = get_unit_mode_from_arg(&argc, argv, 1);
 
-       if (check_argc_min(argc, 2) || argv[1][0] == '-')
+       clean_args_no_options(argc, argv, cmd_device_usage_usage);
+
+       if (check_argc_min(argc - optind, 1))
                usage(cmd_device_usage_usage);
 
-       for (i = 1; i < argc; i++) {
+       for (i = optind; i < argc; i++) {
                int fd;
                DIR *dirstream = NULL;
 
-               if (more_than_one)
+               if (i > 1)
                        printf("\n");
 
                fd = btrfs_open_dir(argv[i], &dirstream, 1);
                if (fd < 0) {
                        ret = 1;
-                       goto out;
+                       break;
                }
 
                ret = _cmd_device_usage(fd, argv[i], unit_mode);
                close_file_or_dir(fd, dirstream);
 
                if (ret)
-                       goto out;
-               more_than_one = 1;
+                       break;
        }
-out:
+
        return !!ret;
 }