btrfs-progs: tree-stats: use on-stack path buffer in calc_root_size
[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], &dev_block_count, 0,
111                                 PREP_DEVICE_ZERO_END | PREP_DEVICE_VERBOSE |
112                                 (discard ? PREP_DEVICE_DISCARD : 0));
113                 close(devfd);
114                 if (res) {
115                         ret++;
116                         goto error_out;
117                 }
118
119                 path = canonicalize_path(argv[i]);
120                 if (!path) {
121                         error("could not canonicalize pathname '%s': %s",
122                                 argv[i], strerror(errno));
123                         ret++;
124                         goto error_out;
125                 }
126
127                 memset(&ioctl_args, 0, sizeof(ioctl_args));
128                 strncpy_null(ioctl_args.name, path);
129                 res = ioctl(fdmnt, BTRFS_IOC_ADD_DEV, &ioctl_args);
130                 if (res < 0) {
131                         error("error adding device '%s': %s",
132                                 path, strerror(errno));
133                         ret++;
134                 }
135                 free(path);
136         }
137
138 error_out:
139         close_file_or_dir(fdmnt, dirstream);
140         return !!ret;
141 }
142
143 static int _cmd_device_remove(int argc, char **argv,
144                 const char * const *usagestr)
145 {
146         char    *mntpnt;
147         int i, fdmnt, ret = 0;
148         DIR     *dirstream = NULL;
149
150         clean_args_no_options(argc, argv, usagestr);
151
152         if (check_argc_min(argc - optind, 2))
153                 usage(usagestr);
154
155         mntpnt = argv[argc - 1];
156
157         fdmnt = btrfs_open_dir(mntpnt, &dirstream, 1);
158         if (fdmnt < 0)
159                 return 1;
160
161         for(i = optind; i < argc - 1; i++) {
162                 struct  btrfs_ioctl_vol_args arg;
163                 struct btrfs_ioctl_vol_args_v2 argv2 = {0};
164                 int is_devid = 0;
165                 int     res;
166
167                 if (string_is_numerical(argv[i])) {
168                         argv2.devid = arg_strtou64(argv[i]);
169                         argv2.flags = BTRFS_DEVICE_SPEC_BY_ID;
170                         is_devid = 1;
171                 } else if (is_block_device(argv[i]) == 1 ||
172                                 strcmp(argv[i], "missing") == 0) {
173                         strncpy_null(argv2.name, argv[i]);
174                 } else {
175                         error("not a block device: %s", argv[i]);
176                         ret++;
177                         continue;
178                 }
179
180                 /*
181                  * Positive values are from BTRFS_ERROR_DEV_*,
182                  * otherwise it's a generic error, one of errnos
183                  */
184                 res = ioctl(fdmnt, BTRFS_IOC_RM_DEV_V2, &argv2);
185
186                 /*
187                  * If BTRFS_IOC_RM_DEV_V2 is not supported we get ENOTTY and if
188                  * argv2.flags includes a flag which kernel doesn't understand then
189                  * we shall get EOPNOTSUPP
190                  */
191                 if (res < 0 && (errno == ENOTTY || errno == EOPNOTSUPP)) {
192                         if (is_devid) {
193                                 error("device delete by id failed: %s",
194                                                         strerror(errno));
195                                 ret++;
196                                 continue;
197                         }
198                         memset(&arg, 0, sizeof(arg));
199                         strncpy_null(arg.name, argv[i]);
200                         res = ioctl(fdmnt, BTRFS_IOC_RM_DEV, &arg);
201                 }
202
203                 if (res) {
204                         const char *msg;
205
206                         if (res > 0)
207                                 msg = btrfs_err_str(res);
208                         else
209                                 msg = strerror(errno);
210                         if (is_devid) {
211                                 error("error removing devid %llu: %s",
212                                         (unsigned long long)argv2.devid, msg);
213                         } else {
214                                 error("error removing device '%s': %s",
215                                         argv[i], msg);
216                         }
217                         ret++;
218                 }
219         }
220
221         close_file_or_dir(fdmnt, dirstream);
222         return !!ret;
223 }
224
225 static const char * const cmd_device_remove_usage[] = {
226         "btrfs device remove <device>|<devid> [<device>|<devid>...] <path>",
227         "Remove a device from a filesystem",
228         NULL
229 };
230
231 static int cmd_device_remove(int argc, char **argv)
232 {
233         return _cmd_device_remove(argc, argv, cmd_device_remove_usage);
234 }
235
236 static const char * const cmd_device_delete_usage[] = {
237         "btrfs device delete <device>|<devid> [<device>|<devid>...] <path>",
238         "Remove a device from a filesystem",
239         NULL
240 };
241
242 static int cmd_device_delete(int argc, char **argv)
243 {
244         return _cmd_device_remove(argc, argv, cmd_device_delete_usage);
245 }
246
247 static const char * const cmd_device_scan_usage[] = {
248         "btrfs device scan [(-d|--all-devices)|<device> [<device>...]]",
249         "Scan devices for a btrfs filesystem",
250         " -d|--all-devices (deprecated)",
251         NULL
252 };
253
254 static int cmd_device_scan(int argc, char **argv)
255 {
256         int i;
257         int devstart;
258         int all = 0;
259         int ret = 0;
260
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_devices();
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         while ((c = getopt(argc, argv, "z")) != -1) {
396                 switch (c) {
397                 case 'z':
398                         flags = BTRFS_DEV_STATS_RESET;
399                         break;
400                 case '?':
401                 default:
402                         usage(cmd_device_stats_usage);
403                 }
404         }
405
406         if (check_argc_exact(argc - optind, 1))
407                 usage(cmd_device_stats_usage);
408
409         dev_path = argv[optind];
410
411         fdmnt = open_path_or_dev_mnt(dev_path, &dirstream, 1);
412         if (fdmnt < 0)
413                 return 1;
414
415         ret = get_fs_info(dev_path, &fi_args, &di_args);
416         if (ret) {
417                 error("getting dev info for devstats failed: %s",
418                         strerror(-ret));
419                 err = 1;
420                 goto out;
421         }
422         if (!fi_args.num_devices) {
423                 error("no devices found");
424                 err = 1;
425                 goto out;
426         }
427
428         for (i = 0; i < fi_args.num_devices; i++) {
429                 struct btrfs_ioctl_get_dev_stats args = {0};
430                 __u8 path[BTRFS_DEVICE_PATH_NAME_MAX + 1];
431
432                 strncpy((char *)path, (char *)di_args[i].path,
433                         BTRFS_DEVICE_PATH_NAME_MAX);
434                 path[BTRFS_DEVICE_PATH_NAME_MAX] = '\0';
435
436                 args.devid = di_args[i].devid;
437                 args.nr_items = BTRFS_DEV_STAT_VALUES_MAX;
438                 args.flags = flags;
439
440                 if (ioctl(fdmnt, BTRFS_IOC_GET_DEV_STATS, &args) < 0) {
441                         error("DEV_STATS ioctl failed on %s: %s",
442                               path, strerror(errno));
443                         err = 1;
444                 } else {
445                         char *canonical_path;
446
447                         canonical_path = canonicalize_path((char *)path);
448
449                         /* No path when device is missing. */
450                         if (!canonical_path) {
451                                 canonical_path = malloc(32);
452                                 if (!canonical_path) {
453                                         error("not enough memory for path buffer");
454                                         goto out;
455                                 }
456                                 snprintf(canonical_path, 32,
457                                          "devid:%llu", args.devid);
458                         }
459
460                         if (args.nr_items >= BTRFS_DEV_STAT_WRITE_ERRS + 1)
461                                 printf("[%s].write_io_errs   %llu\n",
462                                        canonical_path,
463                                        (unsigned long long) args.values[
464                                         BTRFS_DEV_STAT_WRITE_ERRS]);
465                         if (args.nr_items >= BTRFS_DEV_STAT_READ_ERRS + 1)
466                                 printf("[%s].read_io_errs    %llu\n",
467                                        canonical_path,
468                                        (unsigned long long) args.values[
469                                         BTRFS_DEV_STAT_READ_ERRS]);
470                         if (args.nr_items >= BTRFS_DEV_STAT_FLUSH_ERRS + 1)
471                                 printf("[%s].flush_io_errs   %llu\n",
472                                        canonical_path,
473                                        (unsigned long long) args.values[
474                                         BTRFS_DEV_STAT_FLUSH_ERRS]);
475                         if (args.nr_items >= BTRFS_DEV_STAT_CORRUPTION_ERRS + 1)
476                                 printf("[%s].corruption_errs %llu\n",
477                                        canonical_path,
478                                        (unsigned long long) args.values[
479                                         BTRFS_DEV_STAT_CORRUPTION_ERRS]);
480                         if (args.nr_items >= BTRFS_DEV_STAT_GENERATION_ERRS + 1)
481                                 printf("[%s].generation_errs %llu\n",
482                                        canonical_path,
483                                        (unsigned long long) args.values[
484                                         BTRFS_DEV_STAT_GENERATION_ERRS]);
485
486                         free(canonical_path);
487                 }
488         }
489
490 out:
491         free(di_args);
492         close_file_or_dir(fdmnt, dirstream);
493
494         return err;
495 }
496
497 static const char * const cmd_device_usage_usage[] = {
498         "btrfs device usage [options] <path> [<path>..]",
499         "Show detailed information about internal allocations in devices.",
500         HELPINFO_UNITS_SHORT_LONG,
501         NULL
502 };
503
504 static int _cmd_device_usage(int fd, char *path, unsigned unit_mode)
505 {
506         int i;
507         int ret = 0;
508         struct chunk_info *chunkinfo = NULL;
509         struct device_info *devinfo = NULL;
510         int chunkcount = 0;
511         int devcount = 0;
512
513         ret = load_chunk_and_device_info(fd, &chunkinfo, &chunkcount, &devinfo,
514                         &devcount);
515         if (ret)
516                 goto out;
517
518         for (i = 0; i < devcount; i++) {
519                 printf("%s, ID: %llu\n", devinfo[i].path, devinfo[i].devid);
520                 print_device_sizes(fd, &devinfo[i], unit_mode);
521                 print_device_chunks(fd, &devinfo[i], chunkinfo, chunkcount,
522                                 unit_mode);
523                 printf("\n");
524         }
525
526 out:
527         free(devinfo);
528         free(chunkinfo);
529
530         return ret;
531 }
532
533 static int cmd_device_usage(int argc, char **argv)
534 {
535         unsigned unit_mode;
536         int ret = 0;
537         int i;
538
539         unit_mode = get_unit_mode_from_arg(&argc, argv, 1);
540
541         clean_args_no_options(argc, argv, cmd_device_usage_usage);
542
543         if (check_argc_min(argc - optind, 1))
544                 usage(cmd_device_usage_usage);
545
546         for (i = optind; i < argc; i++) {
547                 int fd;
548                 DIR *dirstream = NULL;
549
550                 if (i > 1)
551                         printf("\n");
552
553                 fd = btrfs_open_dir(argv[i], &dirstream, 1);
554                 if (fd < 0) {
555                         ret = 1;
556                         break;
557                 }
558
559                 ret = _cmd_device_usage(fd, argv[i], unit_mode);
560                 close_file_or_dir(fd, dirstream);
561
562                 if (ret)
563                         break;
564         }
565
566         return !!ret;
567 }
568
569 static const char device_cmd_group_info[] =
570 "manage and query devices in the filesystem";
571
572 const struct cmd_group device_cmd_group = {
573         device_cmd_group_usage, device_cmd_group_info, {
574                 { "add", cmd_device_add, cmd_device_add_usage, NULL, 0 },
575                 { "delete", cmd_device_delete, cmd_device_delete_usage, NULL,
576                         CMD_ALIAS },
577                 { "remove", cmd_device_remove, cmd_device_remove_usage, NULL, 0 },
578                 { "scan", cmd_device_scan, cmd_device_scan_usage, NULL, 0 },
579                 { "ready", cmd_device_ready, cmd_device_ready_usage, NULL, 0 },
580                 { "stats", cmd_device_stats, cmd_device_stats_usage, NULL, 0 },
581                 { "usage", cmd_device_usage,
582                         cmd_device_usage_usage, NULL, 0 },
583                 NULL_CMD_STRUCT
584         }
585 };
586
587 int cmd_device(int argc, char **argv)
588 {
589         return handle_command_group(&device_cmd_group, argc, argv);
590 }