1 | /* vi: set sw=4 ts=4: */
|
---|
2 | /*
|
---|
3 | * Mini df implementation for busybox
|
---|
4 | *
|
---|
5 | * Copyright (C) 1999-2004 by Erik Andersen <andersen@codepoet.org>
|
---|
6 | * based on original code by (I think) Bruce Perens <bruce@pixar.com>.
|
---|
7 | *
|
---|
8 | * Licensed under GPLv2 or later, see file LICENSE in this source tree.
|
---|
9 | */
|
---|
10 |
|
---|
11 | /* BB_AUDIT SUSv3 _NOT_ compliant -- option -t missing. */
|
---|
12 | /* http://www.opengroup.org/onlinepubs/007904975/utilities/df.html */
|
---|
13 |
|
---|
14 | /* Mar 16, 2003 Manuel Novoa III (mjn3@codepoet.org)
|
---|
15 | *
|
---|
16 | * Size reduction. Removed floating point dependency. Added error checking
|
---|
17 | * on output. Output stats on 0-sized filesystems if specifically listed on
|
---|
18 | * the command line. Properly round *-blocks, Used, and Available quantities.
|
---|
19 | *
|
---|
20 | * Aug 28, 2008 Bernhard Reutner-Fischer
|
---|
21 | *
|
---|
22 | * Implement -P and -B; better coreutils compat; cleanup
|
---|
23 | */
|
---|
24 |
|
---|
25 | #include <mntent.h>
|
---|
26 | #include <sys/vfs.h>
|
---|
27 | #include "libbb.h"
|
---|
28 | #include "unicode.h"
|
---|
29 |
|
---|
30 | #if !ENABLE_FEATURE_HUMAN_READABLE
|
---|
31 | static unsigned long kscale(unsigned long b, unsigned long bs)
|
---|
32 | {
|
---|
33 | return (b * (unsigned long long) bs + 1024/2) / 1024;
|
---|
34 | }
|
---|
35 | #endif
|
---|
36 |
|
---|
37 | int df_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
|
---|
38 | int df_main(int argc UNUSED_PARAM, char **argv)
|
---|
39 | {
|
---|
40 | unsigned long blocks_used;
|
---|
41 | unsigned blocks_percent_used;
|
---|
42 | unsigned long df_disp_hr = 1024;
|
---|
43 | int status = EXIT_SUCCESS;
|
---|
44 | unsigned opt;
|
---|
45 | FILE *mount_table;
|
---|
46 | struct mntent *mount_entry;
|
---|
47 | struct statfs s;
|
---|
48 |
|
---|
49 | enum {
|
---|
50 | OPT_KILO = (1 << 0),
|
---|
51 | OPT_POSIX = (1 << 1),
|
---|
52 | OPT_ALL = (1 << 2) * ENABLE_FEATURE_DF_FANCY,
|
---|
53 | OPT_INODE = (1 << 3) * ENABLE_FEATURE_DF_FANCY,
|
---|
54 | OPT_BSIZE = (1 << 4) * ENABLE_FEATURE_DF_FANCY,
|
---|
55 | OPT_HUMAN = (1 << (2 + 3*ENABLE_FEATURE_DF_FANCY)) * ENABLE_FEATURE_HUMAN_READABLE,
|
---|
56 | OPT_MEGA = (1 << (3 + 3*ENABLE_FEATURE_DF_FANCY)) * ENABLE_FEATURE_HUMAN_READABLE,
|
---|
57 | };
|
---|
58 | const char *disp_units_hdr = NULL;
|
---|
59 | char *chp;
|
---|
60 |
|
---|
61 | init_unicode();
|
---|
62 |
|
---|
63 | #if ENABLE_FEATURE_HUMAN_READABLE && ENABLE_FEATURE_DF_FANCY
|
---|
64 | opt_complementary = "k-mB:m-Bk:B-km";
|
---|
65 | #elif ENABLE_FEATURE_HUMAN_READABLE
|
---|
66 | opt_complementary = "k-m:m-k";
|
---|
67 | #endif
|
---|
68 | opt = getopt32(argv, "kP"
|
---|
69 | IF_FEATURE_DF_FANCY("aiB:")
|
---|
70 | IF_FEATURE_HUMAN_READABLE("hm")
|
---|
71 | IF_FEATURE_DF_FANCY(, &chp));
|
---|
72 | if (opt & OPT_MEGA)
|
---|
73 | df_disp_hr = 1024*1024;
|
---|
74 |
|
---|
75 | if (opt & OPT_BSIZE)
|
---|
76 | df_disp_hr = xatoul_range(chp, 1, ULONG_MAX); /* disallow 0 */
|
---|
77 |
|
---|
78 | /* From the manpage of df from coreutils-6.10:
|
---|
79 | Disk space is shown in 1K blocks by default, unless the environment
|
---|
80 | variable POSIXLY_CORRECT is set, in which case 512-byte blocks are used.
|
---|
81 | */
|
---|
82 | if (getenv("POSIXLY_CORRECT")) /* TODO - a new libbb function? */
|
---|
83 | df_disp_hr = 512;
|
---|
84 |
|
---|
85 | if (opt & OPT_HUMAN) {
|
---|
86 | df_disp_hr = 0;
|
---|
87 | disp_units_hdr = " Size";
|
---|
88 | }
|
---|
89 | if (opt & OPT_INODE)
|
---|
90 | disp_units_hdr = " Inodes";
|
---|
91 |
|
---|
92 | if (disp_units_hdr == NULL) {
|
---|
93 | #if ENABLE_FEATURE_HUMAN_READABLE
|
---|
94 | disp_units_hdr = xasprintf("%s-blocks",
|
---|
95 | /* print df_disp_hr, show no fractionals,
|
---|
96 | * use suffixes if OPT_POSIX is set in opt */
|
---|
97 | make_human_readable_str(df_disp_hr, 0, !!(opt & OPT_POSIX))
|
---|
98 | );
|
---|
99 | #else
|
---|
100 | disp_units_hdr = xasprintf("%lu-blocks", df_disp_hr);
|
---|
101 | #endif
|
---|
102 | }
|
---|
103 | printf("Filesystem %-15sUsed Available %s Mounted on\n",
|
---|
104 | disp_units_hdr, (opt & OPT_POSIX) ? "Capacity" : "Use%");
|
---|
105 |
|
---|
106 | mount_table = NULL;
|
---|
107 | argv += optind;
|
---|
108 | if (!argv[0]) {
|
---|
109 | mount_table = setmntent(bb_path_mtab_file, "r");
|
---|
110 | if (!mount_table)
|
---|
111 | bb_perror_msg_and_die(bb_path_mtab_file);
|
---|
112 | }
|
---|
113 |
|
---|
114 | while (1) {
|
---|
115 | const char *device;
|
---|
116 | const char *mount_point;
|
---|
117 |
|
---|
118 | if (mount_table) {
|
---|
119 | mount_entry = getmntent(mount_table);
|
---|
120 | if (!mount_entry) {
|
---|
121 | endmntent(mount_table);
|
---|
122 | break;
|
---|
123 | }
|
---|
124 | } else {
|
---|
125 | mount_point = *argv++;
|
---|
126 | if (!mount_point)
|
---|
127 | break;
|
---|
128 | mount_entry = find_mount_point(mount_point, 1);
|
---|
129 | if (!mount_entry) {
|
---|
130 | bb_error_msg("%s: can't find mount point", mount_point);
|
---|
131 | set_error:
|
---|
132 | status = EXIT_FAILURE;
|
---|
133 | continue;
|
---|
134 | }
|
---|
135 | }
|
---|
136 |
|
---|
137 | device = mount_entry->mnt_fsname;
|
---|
138 | mount_point = mount_entry->mnt_dir;
|
---|
139 |
|
---|
140 | if (statfs(mount_point, &s) != 0) {
|
---|
141 | bb_simple_perror_msg(mount_point);
|
---|
142 | goto set_error;
|
---|
143 | }
|
---|
144 |
|
---|
145 | if ((s.f_blocks > 0) || !mount_table || (opt & OPT_ALL)) {
|
---|
146 | if (opt & OPT_INODE) {
|
---|
147 | s.f_blocks = s.f_files;
|
---|
148 | s.f_bavail = s.f_bfree = s.f_ffree;
|
---|
149 | s.f_bsize = 1;
|
---|
150 |
|
---|
151 | if (df_disp_hr)
|
---|
152 | df_disp_hr = 1;
|
---|
153 | }
|
---|
154 | blocks_used = s.f_blocks - s.f_bfree;
|
---|
155 | blocks_percent_used = 0;
|
---|
156 | if (blocks_used + s.f_bavail) {
|
---|
157 | blocks_percent_used = (blocks_used * 100ULL
|
---|
158 | + (blocks_used + s.f_bavail)/2
|
---|
159 | ) / (blocks_used + s.f_bavail);
|
---|
160 | }
|
---|
161 |
|
---|
162 | /* GNU coreutils 6.10 skips certain mounts, try to be compatible. */
|
---|
163 | if (strcmp(device, "rootfs") == 0)
|
---|
164 | continue;
|
---|
165 |
|
---|
166 | #ifdef WHY_WE_DO_IT_FOR_DEV_ROOT_ONLY
|
---|
167 | if (strcmp(device, "/dev/root") == 0) {
|
---|
168 | /* Adjusts device to be the real root device,
|
---|
169 | * or leaves device alone if it can't find it */
|
---|
170 | device = find_block_device("/");
|
---|
171 | if (!device) {
|
---|
172 | goto set_error;
|
---|
173 | }
|
---|
174 | }
|
---|
175 | #endif
|
---|
176 |
|
---|
177 | #if ENABLE_UNICODE_SUPPORT
|
---|
178 | {
|
---|
179 | uni_stat_t uni_stat;
|
---|
180 | char *uni_dev = unicode_conv_to_printable(&uni_stat, device);
|
---|
181 | if (uni_stat.unicode_width > 20) {
|
---|
182 | printf("%s\n%20s", uni_dev, "");
|
---|
183 | } else {
|
---|
184 | printf("%s%*s", uni_dev, 20 - (int)uni_stat.unicode_width, "");
|
---|
185 | }
|
---|
186 | free(uni_dev);
|
---|
187 | }
|
---|
188 | #else
|
---|
189 | if (printf("\n%-20s" + 1, device) > 20)
|
---|
190 | printf("\n%-20s", "");
|
---|
191 | #endif
|
---|
192 |
|
---|
193 | #if ENABLE_FEATURE_HUMAN_READABLE
|
---|
194 | printf(" %9s ",
|
---|
195 | /* f_blocks x f_bsize / df_disp_hr, show one fractional,
|
---|
196 | * use suffixes if df_disp_hr == 0 */
|
---|
197 | make_human_readable_str(s.f_blocks, s.f_bsize, df_disp_hr));
|
---|
198 |
|
---|
199 | printf(" %9s " + 1,
|
---|
200 | /* EXPR x f_bsize / df_disp_hr, show one fractional,
|
---|
201 | * use suffixes if df_disp_hr == 0 */
|
---|
202 | make_human_readable_str((s.f_blocks - s.f_bfree),
|
---|
203 | s.f_bsize, df_disp_hr));
|
---|
204 |
|
---|
205 | printf("%9s %3u%% %s\n",
|
---|
206 | /* f_bavail x f_bsize / df_disp_hr, show one fractional,
|
---|
207 | * use suffixes if df_disp_hr == 0 */
|
---|
208 | make_human_readable_str(s.f_bavail, s.f_bsize, df_disp_hr),
|
---|
209 | blocks_percent_used, mount_point);
|
---|
210 | #else
|
---|
211 | printf(" %9lu %9lu %9lu %3u%% %s\n",
|
---|
212 | kscale(s.f_blocks, s.f_bsize),
|
---|
213 | kscale(s.f_blocks - s.f_bfree, s.f_bsize),
|
---|
214 | kscale(s.f_bavail, s.f_bsize),
|
---|
215 | blocks_percent_used, mount_point);
|
---|
216 | #endif
|
---|
217 | }
|
---|
218 | }
|
---|
219 |
|
---|
220 | return status;
|
---|
221 | }
|
---|