Patchs from Jeff Garzik <jgarzik@mandrakesoft.com> to cleanup
[oweals/busybox.git] / coreutils / df.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Mini df implementation for busybox
4  *
5  * Copyright (C) 1999,2000,2001 by Lineo, inc.
6  * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
7  * based on original code by (I think) Bruce Perens <bruce@pixar.com>.
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
17  * General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22  *
23  */
24
25 #include <stdio.h>
26 #include <stdlib.h>
27 #include <string.h>
28 #include <mntent.h>
29 #include <sys/vfs.h>
30 #include <getopt.h>
31 #include "busybox.h"
32
33 extern const char mtab_file[];  /* Defined in utility.c */
34 #ifdef BB_FEATURE_HUMAN_READABLE
35 static unsigned long df_disp_hr = KILOBYTE; 
36 #endif
37
38 static int do_df(char *device, const char *mount_point)
39 {
40         struct statfs s;
41         long blocks_used;
42         long blocks_percent_used;
43 #ifdef BB_FEATURE_HUMAN_READABLE
44         long base;
45 #endif  
46
47         if (statfs(mount_point, &s) != 0) {
48                 perror_msg("%s", mount_point);
49                 return FALSE;
50         }
51
52         if (s.f_blocks > 0) {
53                 blocks_used = s.f_blocks - s.f_bfree;
54                 if(blocks_used == 0)
55                         blocks_percent_used = 0;
56                 else {
57                         blocks_percent_used = (long)
58                           (blocks_used * 100.0 / (blocks_used + s.f_bavail) + 0.5);
59                 }
60                 if (strcmp(device, "/dev/root") == 0) {
61                         /* Adjusts device to be the real root device,
62                          * or leaves device alone if it can't find it */
63                         find_real_root_device_name( device);
64                 }
65 #ifdef BB_FEATURE_HUMAN_READABLE
66                 switch (df_disp_hr) {
67                         case MEGABYTE:
68                                 base = KILOBYTE;
69                                 break;
70                         case KILOBYTE:
71                                 base = 1;
72                                 break;
73                         default:
74                                 base = 0;
75                 }
76                 printf("%-20s %9s ", device,
77                            make_human_readable_str(s.f_blocks * (s.f_bsize/KILOBYTE), base));
78                 printf("%9s ",
79                            make_human_readable_str((s.f_blocks - s.f_bfree) * (s.f_bsize/KILOBYTE), base));
80                 printf("%9s %3ld%% %s\n",
81                            make_human_readable_str(s.f_bavail * (s.f_bsize/KILOBYTE), base),
82                            blocks_percent_used, mount_point);
83 #else
84                 printf("%-20s %9ld %9ld %9ld %3ld%% %s\n",
85                                 device,
86                                 (long) (s.f_blocks * (s.f_bsize / KILOBYTE)),
87                                 (long) ((s.f_blocks - s.f_bfree) * (s.f_bsize / KILOBYTE)),
88                                 (long) (s.f_bavail * (s.f_bsize / KILOBYTE)),
89                                 blocks_percent_used, mount_point);
90 #endif
91         }
92
93         return TRUE;
94 }
95
96 extern int df_main(int argc, char **argv)
97 {
98         int status = EXIT_SUCCESS;
99         int opt = 0;
100         int i = 0;
101         char disp_units_hdr[80] = "1k-blocks"; /* default display is kilobytes */
102
103         while ((opt = getopt(argc, argv, "k"
104 #ifdef BB_FEATURE_HUMAN_READABLE
105         "hm"
106 #endif
107 )) > 0)
108         {
109                 switch (opt) {
110 #ifdef BB_FEATURE_HUMAN_READABLE
111                         case 'h':
112                                 df_disp_hr = 0;
113                                 strcpy(disp_units_hdr, "     Size");
114                                 break;
115                         case 'm':
116                                 df_disp_hr = MEGABYTE;
117                                 strcpy(disp_units_hdr, "1M-blocks");
118                                 break;
119 #endif
120                         case 'k':
121                                 /* default display is kilobytes */
122                                 break;
123                         default:
124                                           show_usage();
125                 }
126         }
127
128         printf("%-20s %-14s %s %s %s %s\n", "Filesystem", disp_units_hdr,
129                "Used", "Available", "Use%", "Mounted on");
130
131         if(optind < argc) {
132                 struct mntent *mount_entry;
133                 for(i = optind; i < argc; i++)
134                 {
135                         if ((mount_entry = find_mount_point(argv[i], mtab_file)) == 0) {
136                                 error_msg("%s: can't find mount point.", argv[i]);
137                                 status = EXIT_FAILURE;
138                         } else if (!do_df(mount_entry->mnt_fsname, mount_entry->mnt_dir))
139                                 status = EXIT_FAILURE;
140                 }
141         } else {
142                 FILE *mount_table;
143                 struct mntent *mount_entry;
144
145                 mount_table = setmntent(mtab_file, "r");
146                 if (mount_table == 0) {
147                         perror_msg("%s", mtab_file);
148                         return EXIT_FAILURE;
149                 }
150
151                 while ((mount_entry = getmntent(mount_table))) {
152                         if (!do_df(mount_entry->mnt_fsname, mount_entry->mnt_dir))
153                                 status = EXIT_FAILURE;
154                 }
155                 endmntent(mount_table);
156         }
157
158         return status;
159 }
160
161 /*
162 Local Variables:
163 c-file-style: "linux"
164 c-basic-offset: 4
165 tab-width: 4
166 End:
167 */