e7436842777971f935ccdd8aa4a5ecd6e4a43056
[platform/upstream/btrfs-progs.git] / cmds-fi-usage.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 <sys/ioctl.h>
22 #include <errno.h>
23 #include <stdarg.h>
24 #include <getopt.h>
25
26 #include "utils.h"
27 #include "kerncompat.h"
28 #include "ctree.h"
29 #include "string-table.h"
30 #include "cmds-fi-usage.h"
31 #include "commands.h"
32
33 #include "version.h"
34 #include "help.h"
35
36 /*
37  * Add the chunk info to the chunk_info list
38  */
39 static int add_info_to_list(struct chunk_info **info_ptr,
40                         int *info_count,
41                         struct btrfs_chunk *chunk)
42 {
43
44         u64 type = btrfs_stack_chunk_type(chunk);
45         u64 size = btrfs_stack_chunk_length(chunk);
46         int num_stripes = btrfs_stack_chunk_num_stripes(chunk);
47         int j;
48
49         for (j = 0 ; j < num_stripes ; j++) {
50                 int i;
51                 struct chunk_info *p = NULL;
52                 struct btrfs_stripe *stripe;
53                 u64    devid;
54
55                 stripe = btrfs_stripe_nr(chunk, j);
56                 devid = btrfs_stack_stripe_devid(stripe);
57
58                 for (i = 0 ; i < *info_count ; i++)
59                         if ((*info_ptr)[i].type == type &&
60                             (*info_ptr)[i].devid == devid &&
61                             (*info_ptr)[i].num_stripes == num_stripes ) {
62                                 p = (*info_ptr) + i;
63                                 break;
64                         }
65
66                 if (!p) {
67                         int tmp = sizeof(struct btrfs_chunk) * (*info_count + 1);
68                         struct chunk_info *res = realloc(*info_ptr, tmp);
69
70                         if (!res) {
71                                 free(*info_ptr);
72                                 error("not enough memory");
73                                 return -ENOMEM;
74                         }
75
76                         *info_ptr = res;
77                         p = res + *info_count;
78                         (*info_count)++;
79
80                         p->devid = devid;
81                         p->type = type;
82                         p->size = 0;
83                         p->num_stripes = num_stripes;
84                 }
85
86                 p->size += size;
87
88         }
89
90         return 0;
91
92 }
93
94 /*
95  *  Helper to sort the chunk type
96  */
97 static int cmp_chunk_block_group(u64 f1, u64 f2)
98 {
99
100         u64 mask;
101
102         if ((f1 & BTRFS_BLOCK_GROUP_TYPE_MASK) ==
103                 (f2 & BTRFS_BLOCK_GROUP_TYPE_MASK))
104                         mask = BTRFS_BLOCK_GROUP_PROFILE_MASK;
105         else if (f2 & BTRFS_BLOCK_GROUP_SYSTEM)
106                         return -1;
107         else if (f1 & BTRFS_BLOCK_GROUP_SYSTEM)
108                         return +1;
109         else
110                         mask = BTRFS_BLOCK_GROUP_TYPE_MASK;
111
112         if ((f1 & mask) > (f2 & mask))
113                 return +1;
114         else if ((f1 & mask) < (f2 & mask))
115                 return -1;
116         else
117                 return 0;
118 }
119
120 /*
121  * Helper to sort the chunk
122  */
123 static int cmp_chunk_info(const void *a, const void *b)
124 {
125         return cmp_chunk_block_group(
126                 ((struct chunk_info *)a)->type,
127                 ((struct chunk_info *)b)->type);
128 }
129
130 static int load_chunk_info(int fd, struct chunk_info **info_ptr, int *info_count)
131 {
132         int ret;
133         struct btrfs_ioctl_search_args args;
134         struct btrfs_ioctl_search_key *sk = &args.key;
135         struct btrfs_ioctl_search_header *sh;
136         unsigned long off = 0;
137         int i, e;
138
139         memset(&args, 0, sizeof(args));
140
141         /*
142          * there may be more than one ROOT_ITEM key if there are
143          * snapshots pending deletion, we have to loop through
144          * them.
145          */
146         sk->tree_id = BTRFS_CHUNK_TREE_OBJECTID;
147
148         sk->min_objectid = 0;
149         sk->max_objectid = (u64)-1;
150         sk->max_type = 0;
151         sk->min_type = (u8)-1;
152         sk->min_offset = 0;
153         sk->max_offset = (u64)-1;
154         sk->min_transid = 0;
155         sk->max_transid = (u64)-1;
156         sk->nr_items = 4096;
157
158         while (1) {
159                 ret = ioctl(fd, BTRFS_IOC_TREE_SEARCH, &args);
160                 e = errno;
161                 if (e == EPERM)
162                         return -e;
163
164                 if (ret < 0) {
165                         error("cannot look up chunk tree info: %s",
166                                 strerror(e));
167                         return 1;
168                 }
169                 /* the ioctl returns the number of item it found in nr_items */
170
171                 if (sk->nr_items == 0)
172                         break;
173
174                 off = 0;
175                 for (i = 0; i < sk->nr_items; i++) {
176                         struct btrfs_chunk *item;
177                         sh = (struct btrfs_ioctl_search_header *)(args.buf +
178                                                                   off);
179
180                         off += sizeof(*sh);
181                         item = (struct btrfs_chunk *)(args.buf + off);
182
183                         ret = add_info_to_list(info_ptr, info_count, item);
184                         if (ret) {
185                                 *info_ptr = NULL;
186                                 return 1;
187                         }
188
189                         off += btrfs_search_header_len(sh);
190
191                         sk->min_objectid = btrfs_search_header_objectid(sh);
192                         sk->min_type = btrfs_search_header_type(sh);
193                         sk->min_offset = btrfs_search_header_offset(sh)+1;
194
195                 }
196                 if (!sk->min_offset)    /* overflow */
197                         sk->min_type++;
198                 else
199                         continue;
200
201                 if (!sk->min_type)
202                         sk->min_objectid++;
203                  else
204                         continue;
205
206                 if (!sk->min_objectid)
207                         break;
208         }
209
210         qsort(*info_ptr, *info_count, sizeof(struct chunk_info),
211                 cmp_chunk_info);
212
213         return 0;
214 }
215
216 /*
217  * Helper to sort the struct btrfs_ioctl_space_info
218  */
219 static int cmp_btrfs_ioctl_space_info(const void *a, const void *b)
220 {
221         return cmp_chunk_block_group(
222                 ((struct btrfs_ioctl_space_info *)a)->flags,
223                 ((struct btrfs_ioctl_space_info *)b)->flags);
224 }
225
226 /*
227  * This function load all the information about the space usage
228  */
229 static struct btrfs_ioctl_space_args *load_space_info(int fd, char *path)
230 {
231         struct btrfs_ioctl_space_args *sargs = NULL, *sargs_orig = NULL;
232         int ret, count;
233
234         sargs_orig = sargs = calloc(1, sizeof(struct btrfs_ioctl_space_args));
235         if (!sargs) {
236                 error("not enough memory");
237                 return NULL;
238         }
239
240         sargs->space_slots = 0;
241         sargs->total_spaces = 0;
242
243         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
244         if (ret < 0) {
245                 error("cannot get space info on '%s': %s", path,
246                         strerror(errno));
247                 free(sargs);
248                 return NULL;
249         }
250         if (!sargs->total_spaces) {
251                 free(sargs);
252                 printf("No chunks found\n");
253                 return NULL;
254         }
255
256         count = sargs->total_spaces;
257
258         sargs = realloc(sargs, sizeof(struct btrfs_ioctl_space_args) +
259                         (count * sizeof(struct btrfs_ioctl_space_info)));
260         if (!sargs) {
261                 free(sargs_orig);
262                 error("not enough memory");
263                 return NULL;
264         }
265
266         sargs->space_slots = count;
267         sargs->total_spaces = 0;
268
269         ret = ioctl(fd, BTRFS_IOC_SPACE_INFO, sargs);
270         if (ret < 0) {
271                 error("cannot get space info with %u slots: %s",
272                         count, strerror(errno));
273                 free(sargs);
274                 return NULL;
275         }
276
277         qsort(&(sargs->spaces), count, sizeof(struct btrfs_ioctl_space_info),
278                 cmp_btrfs_ioctl_space_info);
279
280         return sargs;
281 }
282
283 /*
284  * This function computes the space occupied by a *single* RAID5/RAID6 chunk.
285  * The computation is performed on the basis of the number of stripes
286  * which compose the chunk, which could be different from the number of devices
287  * if a disk is added later.
288  */
289 static void get_raid56_used(int fd, struct chunk_info *chunks, int chunkcount,
290                 u64 *raid5_used, u64 *raid6_used)
291 {
292         struct chunk_info *info_ptr = chunks;
293         *raid5_used = 0;
294         *raid6_used = 0;
295
296         while (chunkcount-- > 0) {
297                 if (info_ptr->type & BTRFS_BLOCK_GROUP_RAID5)
298                         (*raid5_used) += info_ptr->size / (info_ptr->num_stripes - 1);
299                 if (info_ptr->type & BTRFS_BLOCK_GROUP_RAID6)
300                         (*raid6_used) += info_ptr->size / (info_ptr->num_stripes - 2);
301                 info_ptr++;
302         }
303 }
304
305 #define MIN_UNALOCATED_THRESH   SZ_16M
306 static int print_filesystem_usage_overall(int fd, struct chunk_info *chunkinfo,
307                 int chunkcount, struct device_info *devinfo, int devcount,
308                 char *path, unsigned unit_mode)
309 {
310         struct btrfs_ioctl_space_args *sargs = NULL;
311         int i;
312         int ret = 0;
313         int width = 10;         /* default 10 for human units */
314         /*
315          * r_* prefix is for raw data
316          * l_* is for logical
317          */
318         u64 r_total_size = 0;   /* filesystem size, sum of device sizes */
319         u64 r_total_chunks = 0; /* sum of chunks sizes on disk(s) */
320         u64 r_total_used = 0;
321         u64 r_total_unused = 0;
322         u64 r_total_missing = 0;        /* sum of missing devices size */
323         u64 r_data_used = 0;
324         u64 r_data_chunks = 0;
325         u64 l_data_chunks = 0;
326         u64 r_metadata_used = 0;
327         u64 r_metadata_chunks = 0;
328         u64 l_metadata_chunks = 0;
329         u64 r_system_used = 0;
330         u64 r_system_chunks = 0;
331         double data_ratio;
332         double metadata_ratio;
333         /* logical */
334         u64 raid5_used = 0;
335         u64 raid6_used = 0;
336         u64 l_global_reserve = 0;
337         u64 l_global_reserve_used = 0;
338         u64 free_estimated = 0;
339         u64 free_min = 0;
340         int max_data_ratio = 1;
341         int mixed = 0;
342
343         sargs = load_space_info(fd, path);
344         if (!sargs) {
345                 ret = 1;
346                 goto exit;
347         }
348
349         r_total_size = 0;
350         for (i = 0; i < devcount; i++) {
351                 r_total_size += devinfo[i].size;
352                 if (!devinfo[i].device_size)
353                         r_total_missing += devinfo[i].size;
354         }
355
356         if (r_total_size == 0) {
357                 error("cannot get space info on '%s': %s",
358                         path, strerror(errno));
359
360                 ret = 1;
361                 goto exit;
362         }
363         get_raid56_used(fd, chunkinfo, chunkcount, &raid5_used, &raid6_used);
364
365         for (i = 0; i < sargs->total_spaces; i++) {
366                 int ratio;
367                 u64 flags = sargs->spaces[i].flags;
368
369                 /*
370                  * The raid5/raid6 ratio depends by the stripes number
371                  * used by every chunk. It is computed separately
372                  */
373                 if (flags & BTRFS_BLOCK_GROUP_RAID0)
374                         ratio = 1;
375                 else if (flags & BTRFS_BLOCK_GROUP_RAID1)
376                         ratio = 2;
377                 else if (flags & BTRFS_BLOCK_GROUP_RAID5)
378                         ratio = 0;
379                 else if (flags & BTRFS_BLOCK_GROUP_RAID6)
380                         ratio = 0;
381                 else if (flags & BTRFS_BLOCK_GROUP_DUP)
382                         ratio = 2;
383                 else if (flags & BTRFS_BLOCK_GROUP_RAID10)
384                         ratio = 2;
385                 else
386                         ratio = 1;
387
388                 if (!ratio)
389                         warning("RAID56 detected, not implemented");
390
391                 if (ratio > max_data_ratio)
392                         max_data_ratio = ratio;
393
394                 if (flags & BTRFS_SPACE_INFO_GLOBAL_RSV) {
395                         l_global_reserve = sargs->spaces[i].total_bytes;
396                         l_global_reserve_used = sargs->spaces[i].used_bytes;
397                 }
398                 if ((flags & (BTRFS_BLOCK_GROUP_DATA | BTRFS_BLOCK_GROUP_METADATA))
399                     == (BTRFS_BLOCK_GROUP_DATA | BTRFS_BLOCK_GROUP_METADATA)) {
400                         mixed = 1;
401                 }
402                 if (flags & BTRFS_BLOCK_GROUP_DATA) {
403                         r_data_used += sargs->spaces[i].used_bytes * ratio;
404                         r_data_chunks += sargs->spaces[i].total_bytes * ratio;
405                         l_data_chunks += sargs->spaces[i].total_bytes;
406                 }
407                 if (flags & BTRFS_BLOCK_GROUP_METADATA) {
408                         r_metadata_used += sargs->spaces[i].used_bytes * ratio;
409                         r_metadata_chunks += sargs->spaces[i].total_bytes * ratio;
410                         l_metadata_chunks += sargs->spaces[i].total_bytes;
411                 }
412                 if (flags & BTRFS_BLOCK_GROUP_SYSTEM) {
413                         r_system_used += sargs->spaces[i].used_bytes * ratio;
414                         r_system_chunks += sargs->spaces[i].total_bytes * ratio;
415                 }
416         }
417
418         r_total_chunks = r_data_chunks + r_system_chunks;
419         r_total_used = r_data_used + r_system_used;
420         if (!mixed) {
421                 r_total_chunks += r_metadata_chunks;
422                 r_total_used += r_metadata_used;
423         }
424         r_total_unused = r_total_size - r_total_chunks;
425
426         /* Raw / Logical = raid factor, >= 1 */
427         data_ratio = (double)r_data_chunks / l_data_chunks;
428         if (mixed)
429                 metadata_ratio = data_ratio;
430         else
431                 metadata_ratio = (double)r_metadata_chunks / l_metadata_chunks;
432
433 #if 0
434         /* add the raid5/6 allocated space */
435         total_chunks += raid5_used + raid6_used;
436 #endif
437
438         /*
439          * We're able to fill at least DATA for the unused space
440          *
441          * With mixed raid levels, this gives a rough estimate but more
442          * accurate than just counting the logical free space
443          * (l_data_chunks - l_data_used)
444          *
445          * In non-mixed case there's no difference.
446          */
447         free_estimated = (r_data_chunks - r_data_used) / data_ratio;
448         /*
449          * For mixed-bg the metadata are left out in calculations thus global
450          * reserve would be lost. Part of it could be permanently allocated,
451          * we have to subtract the used bytes so we don't go under zero free.
452          */
453         if (mixed)
454                 free_estimated -= l_global_reserve - l_global_reserve_used;
455         free_min = free_estimated;
456
457         /* Chop unallocatable space */
458         /* FIXME: must be applied per device */
459         if (r_total_unused >= MIN_UNALOCATED_THRESH) {
460                 free_estimated += r_total_unused / data_ratio;
461                 /* Match the calculation of 'df', use the highest raid ratio */
462                 free_min += r_total_unused / max_data_ratio;
463         }
464
465         if (unit_mode != UNITS_HUMAN)
466                 width = 18;
467
468         printf("Overall:\n");
469
470         printf("    Device size:\t\t%*s\n", width,
471                 pretty_size_mode(r_total_size, unit_mode));
472         printf("    Device allocated:\t\t%*s\n", width,
473                 pretty_size_mode(r_total_chunks, unit_mode));
474         printf("    Device unallocated:\t\t%*s\n", width,
475                 pretty_size_mode(r_total_unused, unit_mode | UNITS_NEGATIVE));
476         printf("    Device missing:\t\t%*s\n", width,
477                 pretty_size_mode(r_total_missing, unit_mode));
478         printf("    Used:\t\t\t%*s\n", width,
479                 pretty_size_mode(r_total_used, unit_mode));
480         printf("    Free (estimated):\t\t%*s\t(",
481                 width,
482                 pretty_size_mode(free_estimated, unit_mode));
483         printf("min: %s)\n", pretty_size_mode(free_min, unit_mode));
484         printf("    Data ratio:\t\t\t%*.2f\n",
485                 width, data_ratio);
486         printf("    Metadata ratio:\t\t%*.2f\n",
487                 width, metadata_ratio);
488         printf("    Global reserve:\t\t%*s\t(used: %s)\n", width,
489                 pretty_size_mode(l_global_reserve, unit_mode),
490                 pretty_size_mode(l_global_reserve_used, unit_mode));
491
492 exit:
493
494         if (sargs)
495                 free(sargs);
496
497         return ret;
498 }
499
500 /*
501  *  Helper to sort the device_info structure
502  */
503 static int cmp_device_info(const void *a, const void *b)
504 {
505         return strcmp(((struct device_info *)a)->path,
506                         ((struct device_info *)b)->path);
507 }
508
509 /*
510  *  This function loads the device_info structure and put them in an array
511  */
512 static int load_device_info(int fd, struct device_info **device_info_ptr,
513                            int *device_info_count)
514 {
515         int ret, i, ndevs;
516         struct btrfs_ioctl_fs_info_args fi_args;
517         struct btrfs_ioctl_dev_info_args dev_info;
518         struct device_info *info;
519
520         *device_info_count = 0;
521         *device_info_ptr = NULL;
522
523         ret = ioctl(fd, BTRFS_IOC_FS_INFO, &fi_args);
524         if (ret < 0) {
525                 if (errno == EPERM)
526                         return -errno;
527                 error("cannot get filesystem info: %s",
528                                 strerror(errno));
529                 return 1;
530         }
531
532         info = calloc(fi_args.num_devices, sizeof(struct device_info));
533         if (!info) {
534                 error("not enough memory");
535                 return 1;
536         }
537
538         for (i = 0, ndevs = 0 ; i <= fi_args.max_id ; i++) {
539                 if (ndevs >= fi_args.num_devices) {
540                         error("unexpected number of devices: %d >= %llu", ndevs,
541                                 (unsigned long long)fi_args.num_devices);
542                         goto out;
543                 }
544                 memset(&dev_info, 0, sizeof(dev_info));
545                 ret = get_device_info(fd, i, &dev_info);
546
547                 if (ret == -ENODEV)
548                         continue;
549                 if (ret) {
550                         error("cannot get info about device devid=%d", i);
551                         goto out;
552                 }
553
554                 info[ndevs].devid = dev_info.devid;
555                 if (!dev_info.path[0]) {
556                         strcpy(info[ndevs].path, "missing");
557                 } else {
558                         strcpy(info[ndevs].path, (char *)dev_info.path);
559                         info[ndevs].device_size =
560                                 get_partition_size((char *)dev_info.path);
561                 }
562                 info[ndevs].size = dev_info.total_bytes;
563                 ++ndevs;
564         }
565
566         if (ndevs != fi_args.num_devices) {
567                 error("unexpected number of devices: %d != %llu", ndevs,
568                                 (unsigned long long)fi_args.num_devices);
569                 goto out;
570         }
571
572         qsort(info, fi_args.num_devices,
573                 sizeof(struct device_info), cmp_device_info);
574
575         *device_info_count = fi_args.num_devices;
576         *device_info_ptr = info;
577
578         return 0;
579
580 out:
581         free(info);
582         return ret;
583 }
584
585 int load_chunk_and_device_info(int fd, struct chunk_info **chunkinfo,
586                 int *chunkcount, struct device_info **devinfo, int *devcount)
587 {
588         int ret;
589
590         ret = load_chunk_info(fd, chunkinfo, chunkcount);
591         if (ret == -EPERM) {
592                 warning(
593 "cannot read detailed chunk info, RAID5/6 numbers will be incorrect, run as root");
594         } else if (ret) {
595                 return ret;
596         }
597
598         ret = load_device_info(fd, devinfo, devcount);
599         if (ret == -EPERM) {
600                 warning(
601                 "cannot get filesystem info from ioctl(FS_INFO), run as root");
602                 ret = 0;
603         }
604
605         return ret;
606 }
607
608 /*
609  *  This function computes the size of a chunk in a disk
610  */
611 static u64 calc_chunk_size(struct chunk_info *ci)
612 {
613         if (ci->type & BTRFS_BLOCK_GROUP_RAID0)
614                 return ci->size / ci->num_stripes;
615         else if (ci->type & BTRFS_BLOCK_GROUP_RAID1)
616                 return ci->size ;
617         else if (ci->type & BTRFS_BLOCK_GROUP_DUP)
618                 return ci->size ;
619         else if (ci->type & BTRFS_BLOCK_GROUP_RAID5)
620                 return ci->size / (ci->num_stripes -1);
621         else if (ci->type & BTRFS_BLOCK_GROUP_RAID6)
622                 return ci->size / (ci->num_stripes -2);
623         else if (ci->type & BTRFS_BLOCK_GROUP_RAID10)
624                 return ci->size / ci->num_stripes;
625         return ci->size;
626 }
627
628 /*
629  *  This function print the results of the command "btrfs fi usage"
630  *  in tabular format
631  */
632 static void _cmd_filesystem_usage_tabular(unsigned unit_mode,
633                                         struct btrfs_ioctl_space_args *sargs,
634                                         struct chunk_info *chunks_info_ptr,
635                                         int chunks_info_count,
636                                         struct device_info *device_info_ptr,
637                                         int device_info_count)
638 {
639         int i;
640         u64 total_unused = 0;
641         struct string_table *matrix = NULL;
642         int  ncols, nrows;
643         int col;
644         int unallocated_col;
645         int spaceinfos_col;
646         const int vhdr_skip = 3;        /* amount of vertical header space */
647
648         /* id, path, unallocated */
649         ncols = 3;
650         spaceinfos_col = 2;
651         /* Properly count the real space infos */
652         for (i = 0; i < sargs->total_spaces; i++) {
653                 if (sargs->spaces[i].flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
654                         continue;
655                 ncols++;
656         }
657
658         /* 2 for header, empty line, devices, ===, total, used */
659         nrows = vhdr_skip + device_info_count + 1 + 2;
660
661         matrix = table_create(ncols, nrows);
662         if (!matrix) {
663                 error("not enough memory");
664                 return;
665         }
666
667         /*
668          * We have to skip the global block reserve everywhere as it's an
669          * artificial blockgroup
670          */
671
672         /* header */
673         for (i = 0, col = spaceinfos_col; i < sargs->total_spaces; i++) {
674                 u64 flags = sargs->spaces[i].flags;
675
676                 if (flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
677                         continue;
678
679                 table_printf(matrix, col, 0, "<%s",
680                                 btrfs_group_type_str(flags));
681                 table_printf(matrix, col, 1, "<%s",
682                                 btrfs_group_profile_str(flags));
683                 col++;
684         }
685         unallocated_col = col;
686
687         table_printf(matrix, 0, 1, "<Id");
688         table_printf(matrix, 1, 1, "<Path");
689         table_printf(matrix, unallocated_col, 1, "<Unallocated");
690
691         /* body */
692         for (i = 0; i < device_info_count; i++) {
693                 int k;
694                 char *p;
695
696                 u64  total_allocated = 0, unused;
697
698                 p = strrchr(device_info_ptr[i].path, '/');
699                 if (!p)
700                         p = device_info_ptr[i].path;
701                 else
702                         p++;
703
704                 table_printf(matrix, 0, vhdr_skip + i, ">%llu",
705                                 device_info_ptr[i].devid);
706                 table_printf(matrix, 1, vhdr_skip + i, "<%s",
707                                 device_info_ptr[i].path);
708
709                 for (col = spaceinfos_col, k = 0; k < sargs->total_spaces; k++) {
710                         u64     flags = sargs->spaces[k].flags;
711                         u64 devid = device_info_ptr[i].devid;
712                         int     j;
713                         u64 size = 0;
714
715                         if (flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
716                                 continue;
717
718                         for (j = 0 ; j < chunks_info_count ; j++) {
719                                 if (chunks_info_ptr[j].type != flags )
720                                                 continue;
721                                 if (chunks_info_ptr[j].devid != devid)
722                                                 continue;
723
724                                 size += calc_chunk_size(chunks_info_ptr+j);
725                         }
726
727                         if (size)
728                                 table_printf(matrix, col, vhdr_skip+ i,
729                                         ">%s", pretty_size_mode(size, unit_mode));
730                         else
731                                 table_printf(matrix, col, vhdr_skip + i, ">-");
732
733                         total_allocated += size;
734                         col++;
735                 }
736
737                 unused = get_partition_size(device_info_ptr[i].path)
738                                 - total_allocated;
739
740                 table_printf(matrix, unallocated_col, vhdr_skip + i, ">%s",
741                         pretty_size_mode(unused, unit_mode | UNITS_NEGATIVE));
742                 total_unused += unused;
743
744         }
745
746         for (i = 0; i < spaceinfos_col; i++) {
747                 table_printf(matrix, i, vhdr_skip - 1, "*-");
748                 table_printf(matrix, i, vhdr_skip + device_info_count, "*-");
749         }
750
751         for (i = 0, col = spaceinfos_col; i < sargs->total_spaces; i++) {
752                 if (sargs->spaces[i].flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
753                         continue;
754
755                 table_printf(matrix, col, vhdr_skip - 1, "*-");
756                 table_printf(matrix, col, vhdr_skip + device_info_count, "*-");
757                 col++;
758         }
759         /* One for Unallocated */
760         table_printf(matrix, col, vhdr_skip - 1, "*-");
761         table_printf(matrix, col, vhdr_skip + device_info_count, "*-");
762
763         /* footer */
764         table_printf(matrix, 1, vhdr_skip + device_info_count + 1, "<Total");
765         for (i = 0, col = spaceinfos_col; i < sargs->total_spaces; i++) {
766                 if (sargs->spaces[i].flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
767                         continue;
768
769                 table_printf(matrix, col++, vhdr_skip + device_info_count + 1,
770                         ">%s",
771                         pretty_size_mode(sargs->spaces[i].total_bytes, unit_mode));
772         }
773
774         table_printf(matrix, unallocated_col, vhdr_skip + device_info_count + 1,
775                 ">%s",
776                 pretty_size_mode(total_unused, unit_mode | UNITS_NEGATIVE));
777
778         table_printf(matrix, 1, vhdr_skip + device_info_count + 2, "<Used");
779         for (i = 0, col = spaceinfos_col; i < sargs->total_spaces; i++) {
780                 if (sargs->spaces[i].flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
781                         continue;
782
783                 table_printf(matrix, col++, vhdr_skip + device_info_count + 2,
784                         ">%s",
785                         pretty_size_mode(sargs->spaces[i].used_bytes, unit_mode));
786         }
787
788         table_dump(matrix);
789         table_free(matrix);
790 }
791
792 /*
793  *  This function prints the unused space per every disk
794  */
795 static void print_unused(struct chunk_info *info_ptr,
796                           int info_count,
797                           struct device_info *device_info_ptr,
798                           int device_info_count,
799                           unsigned unit_mode)
800 {
801         int i;
802         for (i = 0; i < device_info_count; i++) {
803                 int     j;
804                 u64     total = 0;
805
806                 for (j = 0; j < info_count; j++)
807                         if (info_ptr[j].devid == device_info_ptr[i].devid)
808                                 total += calc_chunk_size(info_ptr+j);
809
810                 printf("   %s\t%10s\n",
811                         device_info_ptr[i].path,
812                         pretty_size_mode(device_info_ptr[i].size - total,
813                                 unit_mode));
814         }
815 }
816
817 /*
818  *  This function prints the allocated chunk per every disk
819  */
820 static void print_chunk_device(u64 chunk_type,
821                                 struct chunk_info *chunks_info_ptr,
822                                 int chunks_info_count,
823                                 struct device_info *device_info_ptr,
824                                 int device_info_count,
825                                 unsigned unit_mode)
826 {
827         int i;
828
829         for (i = 0; i < device_info_count; i++) {
830                 int     j;
831                 u64     total = 0;
832
833                 for (j = 0; j < chunks_info_count; j++) {
834
835                         if (chunks_info_ptr[j].type != chunk_type)
836                                 continue;
837                         if (chunks_info_ptr[j].devid != device_info_ptr[i].devid)
838                                 continue;
839
840                         total += calc_chunk_size(&(chunks_info_ptr[j]));
841                         //total += chunks_info_ptr[j].size;
842                 }
843
844                 if (total > 0)
845                         printf("   %s\t%10s\n",
846                                 device_info_ptr[i].path,
847                                 pretty_size_mode(total, unit_mode));
848         }
849 }
850
851 /*
852  *  This function print the results of the command "btrfs fi usage"
853  *  in linear format
854  */
855 static void _cmd_filesystem_usage_linear(unsigned unit_mode,
856                                         struct btrfs_ioctl_space_args *sargs,
857                                         struct chunk_info *info_ptr,
858                                         int info_count,
859                                         struct device_info *device_info_ptr,
860                                         int device_info_count)
861 {
862         int i;
863
864         for (i = 0; i < sargs->total_spaces; i++) {
865                 const char *description;
866                 const char *r_mode;
867                 u64 flags = sargs->spaces[i].flags;
868
869                 if (flags & BTRFS_SPACE_INFO_GLOBAL_RSV)
870                         continue;
871
872                 description = btrfs_group_type_str(flags);
873                 r_mode = btrfs_group_profile_str(flags);
874
875                 printf("%s,%s: Size:%s, ",
876                         description,
877                         r_mode,
878                         pretty_size_mode(sargs->spaces[i].total_bytes,
879                                 unit_mode));
880                 printf("Used:%s\n",
881                         pretty_size_mode(sargs->spaces[i].used_bytes, unit_mode));
882                 print_chunk_device(flags, info_ptr, info_count,
883                                 device_info_ptr, device_info_count, unit_mode);
884                 printf("\n");
885         }
886
887         printf("Unallocated:\n");
888         print_unused(info_ptr, info_count, device_info_ptr, device_info_count,
889                         unit_mode | UNITS_NEGATIVE);
890 }
891
892 static int print_filesystem_usage_by_chunk(int fd,
893                 struct chunk_info *chunkinfo, int chunkcount,
894                 struct device_info *devinfo, int devcount,
895                 char *path, unsigned unit_mode, int tabular)
896 {
897         struct btrfs_ioctl_space_args *sargs;
898         int ret = 0;
899
900         if (!chunkinfo)
901                 return 0;
902
903         sargs = load_space_info(fd, path);
904         if (!sargs) {
905                 ret = 1;
906                 goto out;
907         }
908
909         if (tabular)
910                 _cmd_filesystem_usage_tabular(unit_mode, sargs, chunkinfo,
911                                 chunkcount, devinfo, devcount);
912         else
913                 _cmd_filesystem_usage_linear(unit_mode, sargs, chunkinfo,
914                                 chunkcount, devinfo, devcount);
915
916         free(sargs);
917 out:
918         return ret;
919 }
920
921 const char * const cmd_filesystem_usage_usage[] = {
922         "btrfs filesystem usage [options] <path> [<path>..]",
923         "Show detailed information about internal filesystem usage .",
924         HELPINFO_UNITS_SHORT_LONG,
925         "-T                 show data in tabular format",
926         NULL
927 };
928
929 int cmd_filesystem_usage(int argc, char **argv)
930 {
931         int ret = 0;
932         unsigned unit_mode;
933         int i;
934         int more_than_one = 0;
935         int tabular = 0;
936
937         unit_mode = get_unit_mode_from_arg(&argc, argv, 1);
938
939         while (1) {
940                 int c;
941
942                 c = getopt(argc, argv, "T");
943                 if (c < 0)
944                         break;
945
946                 switch (c) {
947                 case 'T':
948                         tabular = 1;
949                         break;
950                 default:
951                         usage(cmd_filesystem_usage_usage);
952                 }
953         }
954
955         if (check_argc_min(argc - optind, 1))
956                 usage(cmd_filesystem_usage_usage);
957
958         for (i = optind; i < argc; i++) {
959                 int fd;
960                 DIR *dirstream = NULL;
961                 struct chunk_info *chunkinfo = NULL;
962                 struct device_info *devinfo = NULL;
963                 int chunkcount = 0;
964                 int devcount = 0;
965
966                 fd = btrfs_open_dir(argv[i], &dirstream, 1);
967                 if (fd < 0) {
968                         ret = 1;
969                         goto out;
970                 }
971                 if (more_than_one)
972                         printf("\n");
973
974                 ret = load_chunk_and_device_info(fd, &chunkinfo, &chunkcount,
975                                 &devinfo, &devcount);
976                 if (ret)
977                         goto cleanup;
978
979                 ret = print_filesystem_usage_overall(fd, chunkinfo, chunkcount,
980                                 devinfo, devcount, argv[i], unit_mode);
981                 if (ret)
982                         goto cleanup;
983                 printf("\n");
984                 ret = print_filesystem_usage_by_chunk(fd, chunkinfo, chunkcount,
985                                 devinfo, devcount, argv[i], unit_mode, tabular);
986 cleanup:
987                 close_file_or_dir(fd, dirstream);
988                 free(chunkinfo);
989                 free(devinfo);
990
991                 if (ret)
992                         goto out;
993                 more_than_one = 1;
994         }
995
996 out:
997         return !!ret;
998 }
999
1000 void print_device_chunks(int fd, struct device_info *devinfo,
1001                 struct chunk_info *chunks_info_ptr,
1002                 int chunks_info_count, unsigned unit_mode)
1003 {
1004         int i;
1005         u64 allocated = 0;
1006
1007         for (i = 0 ; i < chunks_info_count ; i++) {
1008                 const char *description;
1009                 const char *r_mode;
1010                 u64 flags;
1011                 u64 size;
1012
1013                 if (chunks_info_ptr[i].devid != devinfo->devid)
1014                         continue;
1015
1016                 flags = chunks_info_ptr[i].type;
1017
1018                 description = btrfs_group_type_str(flags);
1019                 r_mode = btrfs_group_profile_str(flags);
1020                 size = calc_chunk_size(chunks_info_ptr+i);
1021                 printf("   %s,%s:%*s%10s\n",
1022                         description,
1023                         r_mode,
1024                         (int)(20 - strlen(description) - strlen(r_mode)), "",
1025                         pretty_size_mode(size, unit_mode));
1026
1027                 allocated += size;
1028
1029         }
1030         printf("   Unallocated: %*s%10s\n",
1031                 (int)(20 - strlen("Unallocated")), "",
1032                 pretty_size_mode(devinfo->size - allocated,
1033                         unit_mode | UNITS_NEGATIVE));
1034 }
1035
1036 void print_device_sizes(int fd, struct device_info *devinfo, unsigned unit_mode)
1037 {
1038         printf("   Device size: %*s%10s\n",
1039                 (int)(20 - strlen("Device size")), "",
1040                 pretty_size_mode(devinfo->device_size, unit_mode));
1041         printf("   Device slack: %*s%10s\n",
1042                 (int)(20 - strlen("Device slack")), "",
1043                 pretty_size_mode(devinfo->device_size - devinfo->size,
1044                         unit_mode));
1045 }