btrfs-progs: convert-tests: Add support for custom test scripts
[platform/upstream/btrfs-progs.git] / cmds-device.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 #include <stdio.h>
18 #include <stdlib.h>
19 #include <string.h>
20 #include <unistd.h>
21 #include <fcntl.h>
22 #include <sys/ioctl.h>
23 #include <errno.h>
24 #include <sys/stat.h>
25 #include <getopt.h>
26
27 #include "kerncompat.h"
28 #include "ctree.h"
29 #include "ioctl.h"
30 #include "utils.h"
31 #include "volumes.h"
32 #include "cmds-fi-usage.h"
33
34 #include "commands.h"
35
36 static const char * const device_cmd_group_usage[] = {
37         "btrfs device <command> [<args>]",
38         NULL
39 };
40
41 static const char * const cmd_device_add_usage[] = {
42         "btrfs device add [options] <device> [<device>...] <path>",
43         "Add a device to a filesystem",
44         "-K|--nodiscard    do not perform whole device TRIM",
45         "-f|--force        force overwrite existing filesystem on the disk",
46         NULL
47 };
48
49 static int cmd_device_add(int argc, char **argv)
50 {
51         char    *mntpnt;
52         int i, fdmnt, ret = 0;
53         DIR     *dirstream = NULL;
54         int discard = 1;
55         int force = 0;
56         int last_dev;
57
58         while (1) {
59                 int c;
60                 static const struct option long_options[] = {
61                         { "nodiscard", optional_argument, NULL, 'K'},
62                         { "force", no_argument, NULL, 'f'},
63                         { NULL, 0, NULL, 0}
64                 };
65
66                 c = getopt_long(argc, argv, "Kf", long_options, NULL);
67                 if (c < 0)
68                         break;
69                 switch (c) {
70                 case 'K':
71                         discard = 0;
72                         break;
73                 case 'f':
74                         force = 1;
75                         break;
76                 default:
77                         usage(cmd_device_add_usage);
78                 }
79         }
80
81         if (check_argc_min(argc - optind, 2))
82                 usage(cmd_device_add_usage);
83
84         last_dev = argc - 1;
85         mntpnt = argv[last_dev];
86
87         fdmnt = btrfs_open_dir(mntpnt, &dirstream, 1);
88         if (fdmnt < 0)
89                 return 1;
90
91         for (i = optind; i < last_dev; i++){
92                 struct btrfs_ioctl_vol_args ioctl_args;
93                 int     devfd, res;
94                 u64 dev_block_count = 0;
95                 char *path;
96
97                 res = test_dev_for_mkfs(argv[i], force);
98                 if (res) {
99                         ret++;
100                         continue;
101                 }
102
103                 devfd = open(argv[i], O_RDWR);
104                 if (devfd < 0) {
105                         error("unable to open device '%s'", argv[i]);
106                         ret++;
107                         continue;
108                 }
109
110                 res = btrfs_prepare_device(devfd, argv[i], 1, &dev_block_count,
111                                            0, discard);
112                 close(devfd);
113                 if (res) {
114                         ret++;
115                         goto error_out;
116                 }
117
118                 path = canonicalize_path(argv[i]);
119                 if (!path) {
120                         error("could not canonicalize pathname '%s': %s",
121                                 argv[i], strerror(errno));
122                         ret++;
123                         goto error_out;
124                 }
125
126                 memset(&ioctl_args, 0, sizeof(ioctl_args));
127                 strncpy_null(ioctl_args.name, path);
128                 res = ioctl(fdmnt, BTRFS_IOC_ADD_DEV, &ioctl_args);
129                 if (res < 0) {
130                         error("error adding device '%s': %s",
131                                 path, strerror(errno));
132                         ret++;
133                 }
134                 free(path);
135         }
136
137 error_out:
138         close_file_or_dir(fdmnt, dirstream);
139         return !!ret;
140 }
141
142 static int _cmd_device_remove(int argc, char **argv,
143                 const char * const *usagestr)
144 {
145         char    *mntpnt;
146         int i, fdmnt, ret = 0;
147         DIR     *dirstream = NULL;
148
149         clean_args_no_options(argc, argv, usagestr);
150
151         if (check_argc_min(argc - optind, 2))
152                 usage(usagestr);
153
154         mntpnt = argv[argc - 1];
155
156         fdmnt = btrfs_open_dir(mntpnt, &dirstream, 1);
157         if (fdmnt < 0)
158                 return 1;
159
160         for(i = optind; i < argc - 1; i++) {
161                 struct  btrfs_ioctl_vol_args arg;
162                 struct btrfs_ioctl_vol_args_v2 argv2 = {0};
163                 int is_devid = 0;
164                 int     res;
165
166                 if (string_is_numerical(argv[i])) {
167                         argv2.devid = arg_strtou64(argv[i]);
168                         argv2.flags = BTRFS_DEVICE_SPEC_BY_ID;
169                         is_devid = 1;
170                 } else if (is_block_device(argv[i]) == 1 ||
171                                 strcmp(argv[i], "missing") == 0) {
172                         strncpy_null(argv2.name, argv[i]);
173                 } else {
174                         error("not a block device: %s", argv[i]);
175                         ret++;
176                         continue;
177                 }
178
179                 /*
180                  * Positive values are from BTRFS_ERROR_DEV_*,
181                  * otherwise it's a generic error, one of errnos
182                  */
183                 res = ioctl(fdmnt, BTRFS_IOC_RM_DEV_V2, &argv2);
184
185                 /*
186                  * If BTRFS_IOC_RM_DEV_V2 is not supported we get ENOTTY and if
187                  * argv2.flags includes a flag which kernel doesn't understand then
188                  * we shall get EOPNOTSUPP
189                  */
190                 if (res < 0 && (errno == ENOTTY || errno == EOPNOTSUPP)) {
191                         if (is_devid) {
192                                 error("device delete by id failed: %s",
193                                                         strerror(errno));
194                                 ret++;
195                                 continue;
196                         }
197                         memset(&arg, 0, sizeof(arg));
198                         strncpy_null(arg.name, argv[i]);
199                         res = ioctl(fdmnt, BTRFS_IOC_RM_DEV, &arg);
200                 }
201
202                 if (res) {
203                         const char *msg;
204
205                         if (res > 0)
206                                 msg = btrfs_err_str(res);
207                         else
208                                 msg = strerror(errno);
209                         if (is_devid) {
210                                 error("error removing devid %llu: %s",
211                                         (unsigned long long)argv2.devid, msg);
212                         } else {
213                                 error("error removing device '%s': %s",
214                                         argv[i], msg);
215                         }
216                         ret++;
217                 }
218         }
219
220         close_file_or_dir(fdmnt, dirstream);
221         return !!ret;
222 }
223
224 static const char * const cmd_device_remove_usage[] = {
225         "btrfs device remove <device>|<devid> [<device>|<devid>...] <path>",
226         "Remove a device from a filesystem",
227         NULL
228 };
229
230 static int cmd_device_remove(int argc, char **argv)
231 {
232         return _cmd_device_remove(argc, argv, cmd_device_remove_usage);
233 }
234
235 static const char * const cmd_device_delete_usage[] = {
236         "btrfs device delete <device>|<devid> [<device>|<devid>...] <path>",
237         "Remove a device from a filesystem",
238         NULL
239 };
240
241 static int cmd_device_delete(int argc, char **argv)
242 {
243         return _cmd_device_remove(argc, argv, cmd_device_delete_usage);
244 }
245
246 static const char * const cmd_device_scan_usage[] = {
247         "btrfs device scan [(-d|--all-devices)|<device> [<device>...]]",
248         "Scan devices for a btrfs filesystem",
249         " -d|--all-devices (deprecated)",
250         NULL
251 };
252
253 static int cmd_device_scan(int argc, char **argv)
254 {
255         int i;
256         int devstart;
257         int all = 0;
258         int ret = 0;
259
260         optind = 1;
261         while (1) {
262                 int c;
263                 static const struct option long_options[] = {
264                         { "all-devices", no_argument, NULL, 'd'},
265                         { NULL, 0, NULL, 0}
266                 };
267
268                 c = getopt_long(argc, argv, "d", long_options, NULL);
269                 if (c < 0)
270                         break;
271                 switch (c) {
272                 case 'd':
273                         all = 1;
274                         break;
275                 default:
276                         usage(cmd_device_scan_usage);
277                 }
278         }
279         devstart = optind;
280
281         if (all && check_argc_max(argc - optind, 1))
282                 usage(cmd_device_scan_usage);
283
284         if (all || argc - optind == 0) {
285                 printf("Scanning for Btrfs filesystems\n");
286                 ret = btrfs_scan_lblkid();
287                 error_on(ret, "error %d while scanning", ret);
288                 ret = btrfs_register_all_devices();
289                 error_on(ret, "there are %d errors while registering devices", ret);
290                 goto out;
291         }
292
293         for( i = devstart ; i < argc ; i++ ){
294                 char *path;
295
296                 if (is_block_device(argv[i]) != 1) {
297                         error("not a block device: %s", argv[i]);
298                         ret = 1;
299                         goto out;
300                 }
301                 path = canonicalize_path(argv[i]);
302                 if (!path) {
303                         error("could not canonicalize path '%s': %s",
304                                 argv[i], strerror(errno));
305                         ret = 1;
306                         goto out;
307                 }
308                 printf("Scanning for Btrfs filesystems in '%s'\n", path);
309                 if (btrfs_register_one_device(path) != 0) {
310                         ret = 1;
311                         free(path);
312                         goto out;
313                 }
314                 free(path);
315         }
316
317 out:
318         return !!ret;
319 }
320
321 static const char * const cmd_device_ready_usage[] = {
322         "btrfs device ready <device>",
323         "Check device to see if it has all of its devices in cache for mounting",
324         NULL
325 };
326
327 static int cmd_device_ready(int argc, char **argv)
328 {
329         struct  btrfs_ioctl_vol_args args;
330         int     fd;
331         int     ret;
332         char    *path;
333
334         clean_args_no_options(argc, argv, cmd_device_ready_usage);
335
336         if (check_argc_exact(argc - optind, 1))
337                 usage(cmd_device_ready_usage);
338
339         fd = open("/dev/btrfs-control", O_RDWR);
340         if (fd < 0) {
341                 perror("failed to open /dev/btrfs-control");
342                 return 1;
343         }
344
345         path = canonicalize_path(argv[optind]);
346         if (!path) {
347                 error("could not canonicalize pathname '%s': %s",
348                         argv[optind], strerror(errno));
349                 ret = 1;
350                 goto out;
351         }
352
353         if (is_block_device(path) != 1) {
354                 error("not a block device: %s", path);
355                 ret = 1;
356                 goto out;
357         }
358
359         memset(&args, 0, sizeof(args));
360         strncpy_null(args.name, path);
361         ret = ioctl(fd, BTRFS_IOC_DEVICES_READY, &args);
362         if (ret < 0) {
363                 error("unable to determine if device '%s' is ready for mount: %s",
364                         path, strerror(errno));
365                 ret = 1;
366         }
367
368 out:
369         free(path);
370         close(fd);
371         return ret;
372 }
373
374 static const char * const cmd_device_stats_usage[] = {
375         "btrfs device stats [-z] <path>|<device>",
376         "Show current device IO stats.",
377         "",
378         "-z                     show current stats and reset values to zero",
379         NULL
380 };
381
382 static int cmd_device_stats(int argc, char **argv)
383 {
384         char *dev_path;
385         struct btrfs_ioctl_fs_info_args fi_args;
386         struct btrfs_ioctl_dev_info_args *di_args = NULL;
387         int ret;
388         int fdmnt;
389         int i;
390         int c;
391         int err = 0;
392         __u64 flags = 0;
393         DIR *dirstream = NULL;
394
395         optind = 1;
396         while ((c = getopt(argc, argv, "z")) != -1) {
397                 switch (c) {
398                 case 'z':
399                         flags = BTRFS_DEV_STATS_RESET;
400                         break;
401                 case '?':
402                 default:
403                         usage(cmd_device_stats_usage);
404                 }
405         }
406
407         if (check_argc_exact(argc - optind, 1))
408                 usage(cmd_device_stats_usage);
409
410         dev_path = argv[optind];
411
412         fdmnt = open_path_or_dev_mnt(dev_path, &dirstream, 1);
413         if (fdmnt < 0)
414                 return 1;
415
416         ret = get_fs_info(dev_path, &fi_args, &di_args);
417         if (ret) {
418                 error("getting dev info for devstats failed: %s",
419                         strerror(-ret));
420                 err = 1;
421                 goto out;
422         }
423         if (!fi_args.num_devices) {
424                 error("no devices found");
425                 err = 1;
426                 goto out;
427         }
428
429         for (i = 0; i < fi_args.num_devices; i++) {
430                 struct btrfs_ioctl_get_dev_stats args = {0};
431                 __u8 path[BTRFS_DEVICE_PATH_NAME_MAX + 1];
432
433                 strncpy((char *)path, (char *)di_args[i].path,
434                         BTRFS_DEVICE_PATH_NAME_MAX);
435                 path[BTRFS_DEVICE_PATH_NAME_MAX] = '\0';
436
437                 args.devid = di_args[i].devid;
438                 args.nr_items = BTRFS_DEV_STAT_VALUES_MAX;
439                 args.flags = flags;
440
441                 if (ioctl(fdmnt, BTRFS_IOC_GET_DEV_STATS, &args) < 0) {
442                         error("DEV_STATS ioctl failed on %s: %s",
443                               path, strerror(errno));
444                         err = 1;
445                 } else {
446                         char *canonical_path;
447
448                         canonical_path = canonicalize_path((char *)path);
449
450                         /* No path when device is missing. */
451                         if (!canonical_path) {
452                                 canonical_path = malloc(32);
453                                 if (!canonical_path) {
454                                         error("not enough memory for path buffer");
455                                         goto out;
456                                 }
457                                 snprintf(canonical_path, 32,
458                                          "devid:%llu", args.devid);
459                         }
460
461                         if (args.nr_items >= BTRFS_DEV_STAT_WRITE_ERRS + 1)
462                                 printf("[%s].write_io_errs   %llu\n",
463                                        canonical_path,
464                                        (unsigned long long) args.values[
465                                         BTRFS_DEV_STAT_WRITE_ERRS]);
466                         if (args.nr_items >= BTRFS_DEV_STAT_READ_ERRS + 1)
467                                 printf("[%s].read_io_errs    %llu\n",
468                                        canonical_path,
469                                        (unsigned long long) args.values[
470                                         BTRFS_DEV_STAT_READ_ERRS]);
471                         if (args.nr_items >= BTRFS_DEV_STAT_FLUSH_ERRS + 1)
472                                 printf("[%s].flush_io_errs   %llu\n",
473                                        canonical_path,
474                                        (unsigned long long) args.values[
475                                         BTRFS_DEV_STAT_FLUSH_ERRS]);
476                         if (args.nr_items >= BTRFS_DEV_STAT_CORRUPTION_ERRS + 1)
477                                 printf("[%s].corruption_errs %llu\n",
478                                        canonical_path,
479                                        (unsigned long long) args.values[
480                                         BTRFS_DEV_STAT_CORRUPTION_ERRS]);
481                         if (args.nr_items >= BTRFS_DEV_STAT_GENERATION_ERRS + 1)
482                                 printf("[%s].generation_errs %llu\n",
483                                        canonical_path,
484                                        (unsigned long long) args.values[
485                                         BTRFS_DEV_STAT_GENERATION_ERRS]);
486
487                         free(canonical_path);
488                 }
489         }
490
491 out:
492         free(di_args);
493         close_file_or_dir(fdmnt, dirstream);
494
495         return err;
496 }
497
498 static const char * const cmd_device_usage_usage[] = {
499         "btrfs device usage [options] <path> [<path>..]",
500         "Show detailed information about internal allocations in devices.",
501         HELPINFO_UNITS_SHORT_LONG,
502         NULL
503 };
504
505 static int _cmd_device_usage(int fd, char *path, unsigned unit_mode)
506 {
507         int i;
508         int ret = 0;
509         struct chunk_info *chunkinfo = NULL;
510         struct device_info *devinfo = NULL;
511         int chunkcount = 0;
512         int devcount = 0;
513
514         ret = load_chunk_and_device_info(fd, &chunkinfo, &chunkcount, &devinfo,
515                         &devcount);
516         if (ret)
517                 goto out;
518
519         for (i = 0; i < devcount; i++) {
520                 printf("%s, ID: %llu\n", devinfo[i].path, devinfo[i].devid);
521                 print_device_sizes(fd, &devinfo[i], unit_mode);
522                 print_device_chunks(fd, &devinfo[i], chunkinfo, chunkcount,
523                                 unit_mode);
524                 printf("\n");
525         }
526
527 out:
528         free(devinfo);
529         free(chunkinfo);
530
531         return ret;
532 }
533
534 static int cmd_device_usage(int argc, char **argv)
535 {
536         unsigned unit_mode;
537         int ret = 0;
538         int i;
539
540         unit_mode = get_unit_mode_from_arg(&argc, argv, 1);
541
542         clean_args_no_options(argc, argv, cmd_device_usage_usage);
543
544         if (check_argc_min(argc - optind, 1))
545                 usage(cmd_device_usage_usage);
546
547         for (i = optind; i < argc; i++) {
548                 int fd;
549                 DIR *dirstream = NULL;
550
551                 if (i > 1)
552                         printf("\n");
553
554                 fd = btrfs_open_dir(argv[i], &dirstream, 1);
555                 if (fd < 0) {
556                         ret = 1;
557                         break;
558                 }
559
560                 ret = _cmd_device_usage(fd, argv[i], unit_mode);
561                 close_file_or_dir(fd, dirstream);
562
563                 if (ret)
564                         break;
565         }
566
567         return !!ret;
568 }
569
570 static const char device_cmd_group_info[] =
571 "manage and query devices in the filesystem";
572
573 const struct cmd_group device_cmd_group = {
574         device_cmd_group_usage, device_cmd_group_info, {
575                 { "add", cmd_device_add, cmd_device_add_usage, NULL, 0 },
576                 { "delete", cmd_device_delete, cmd_device_delete_usage, NULL,
577                         CMD_ALIAS },
578                 { "remove", cmd_device_remove, cmd_device_remove_usage, NULL, 0 },
579                 { "scan", cmd_device_scan, cmd_device_scan_usage, NULL, 0 },
580                 { "ready", cmd_device_ready, cmd_device_ready_usage, NULL, 0 },
581                 { "stats", cmd_device_stats, cmd_device_stats_usage, NULL, 0 },
582                 { "usage", cmd_device_usage,
583                         cmd_device_usage_usage, NULL, 0 },
584                 NULL_CMD_STRUCT
585         }
586 };
587
588 int cmd_device(int argc, char **argv)
589 {
590         return handle_command_group(&device_cmd_group, argc, argv);
591 }