File: | src/bin/df/df.c |
Warning: | line 176, column 10 Potential leak of memory pointed to by 'mntbuf' |
Press '?' to see keyboard shortcuts
Keyboard shortcuts:
1 | /* $OpenBSD: df.c,v 1.61 2023/01/01 16:31:20 millert Exp $ */ | |||
2 | /* $NetBSD: df.c,v 1.21.2.1 1995/11/01 00:06:11 jtc Exp $ */ | |||
3 | ||||
4 | /* | |||
5 | * Copyright (c) 1980, 1990, 1993, 1994 | |||
6 | * The Regents of the University of California. All rights reserved. | |||
7 | * (c) UNIX System Laboratories, Inc. | |||
8 | * All or some portions of this file are derived from material licensed | |||
9 | * to the University of California by American Telephone and Telegraph | |||
10 | * Co. or Unix System Laboratories, Inc. and are reproduced herein with | |||
11 | * the permission of UNIX System Laboratories, Inc. | |||
12 | * | |||
13 | * Redistribution and use in source and binary forms, with or without | |||
14 | * modification, are permitted provided that the following conditions | |||
15 | * are met: | |||
16 | * 1. Redistributions of source code must retain the above copyright | |||
17 | * notice, this list of conditions and the following disclaimer. | |||
18 | * 2. Redistributions in binary form must reproduce the above copyright | |||
19 | * notice, this list of conditions and the following disclaimer in the | |||
20 | * documentation and/or other materials provided with the distribution. | |||
21 | * 3. Neither the name of the University nor the names of its contributors | |||
22 | * may be used to endorse or promote products derived from this software | |||
23 | * without specific prior written permission. | |||
24 | * | |||
25 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND | |||
26 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE | |||
27 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |||
28 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE | |||
29 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL | |||
30 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS | |||
31 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) | |||
32 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT | |||
33 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY | |||
34 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF | |||
35 | * SUCH DAMAGE. | |||
36 | */ | |||
37 | ||||
38 | #include <sys/stat.h> | |||
39 | #include <sys/mount.h> | |||
40 | ||||
41 | #include <err.h> | |||
42 | #include <errno(*__errno()).h> | |||
43 | #include <fcntl.h> | |||
44 | #include <stdio.h> | |||
45 | #include <stdlib.h> | |||
46 | #include <string.h> | |||
47 | #include <unistd.h> | |||
48 | #include <util.h> | |||
49 | ||||
50 | int bread(int, off_t, void *, int); | |||
51 | static void bsdprint(struct statfs *, long, int); | |||
52 | char *getmntpt(char *); | |||
53 | static void maketypelist(char *); | |||
54 | static int percent(u_int64_t, u_int64_t); | |||
55 | static void posixprint(struct statfs *, long, int); | |||
56 | static void prthuman(struct statfs *sfsp, unsigned long long); | |||
57 | static void prthumanval(long long); | |||
58 | static void prtstat(struct statfs *, int, int, int); | |||
59 | static long regetmntinfo(struct statfs **, long); | |||
60 | static int selected(const char *); | |||
61 | static __dead__attribute__((__noreturn__)) void usage(void); | |||
62 | ||||
63 | extern int e2fs_df(int, char *, struct statfs *); | |||
64 | extern int ffs_df(int, char *, struct statfs *); | |||
65 | static int raw_df(char *, struct statfs *); | |||
66 | ||||
67 | int hflag, iflag, kflag, lflag, nflag, Pflag; | |||
68 | char **typelist = NULL((void *)0); | |||
69 | ||||
70 | int | |||
71 | main(int argc, char *argv[]) | |||
72 | { | |||
73 | struct stat stbuf; | |||
74 | struct statfs *mntbuf; | |||
75 | long mntsize; | |||
76 | int ch, i; | |||
77 | int width, maxwidth; | |||
78 | char *mntpt; | |||
79 | ||||
80 | if (pledge("stdio rpath", NULL((void *)0)) == -1) | |||
| ||||
81 | err(1, "pledge"); | |||
82 | ||||
83 | while ((ch = getopt(argc, argv, "hiklnPt:")) != -1) | |||
84 | switch (ch) { | |||
85 | case 'h': | |||
86 | hflag = 1; | |||
87 | kflag = 0; | |||
88 | break; | |||
89 | case 'i': | |||
90 | iflag = 1; | |||
91 | break; | |||
92 | case 'k': | |||
93 | kflag = 1; | |||
94 | hflag = 0; | |||
95 | break; | |||
96 | case 'l': | |||
97 | lflag = 1; | |||
98 | break; | |||
99 | case 'n': | |||
100 | nflag = 1; | |||
101 | break; | |||
102 | case 'P': | |||
103 | Pflag = 1; | |||
104 | break; | |||
105 | case 't': | |||
106 | if (typelist != NULL((void *)0)) | |||
107 | errx(1, "only one -t option may be specified."); | |||
108 | maketypelist(optarg); | |||
109 | break; | |||
110 | default: | |||
111 | usage(); | |||
112 | } | |||
113 | argc -= optind; | |||
114 | argv += optind; | |||
115 | ||||
116 | if ((iflag || hflag) && Pflag) { | |||
117 | warnx("-h and -i are incompatible with -P"); | |||
118 | usage(); | |||
119 | } | |||
120 | ||||
121 | mntsize = getmntinfo(&mntbuf, MNT_NOWAIT2); | |||
122 | if (mntsize == 0) | |||
123 | err(1, "retrieving information on mounted file systems"); | |||
124 | ||||
125 | if (!*argv) { | |||
126 | mntsize = regetmntinfo(&mntbuf, mntsize); | |||
127 | } else { | |||
128 | mntbuf = calloc(argc, sizeof(struct statfs)); | |||
129 | if (mntbuf == NULL((void *)0)) | |||
130 | err(1, NULL((void *)0)); | |||
131 | mntsize = 0; | |||
132 | for (; *argv; argv++) { | |||
133 | if (stat(*argv, &stbuf) == -1) { | |||
134 | if ((mntpt = getmntpt(*argv)) == 0) { | |||
135 | warn("%s", *argv); | |||
136 | continue; | |||
137 | } | |||
138 | } else if (S_ISCHR(stbuf.st_mode)((stbuf.st_mode & 0170000) == 0020000) || S_ISBLK(stbuf.st_mode)((stbuf.st_mode & 0170000) == 0060000)) { | |||
139 | if (!raw_df(*argv, &mntbuf[mntsize])) | |||
140 | ++mntsize; | |||
141 | continue; | |||
142 | } else | |||
143 | mntpt = *argv; | |||
144 | /* | |||
145 | * Statfs does not take a `wait' flag, so we cannot | |||
146 | * implement nflag here. | |||
147 | */ | |||
148 | if (!statfs(mntpt, &mntbuf[mntsize])) | |||
149 | if (lflag && (mntbuf[mntsize].f_flags & MNT_LOCAL0x00001000) == 0) | |||
150 | warnx("%s is not a local file system", | |||
151 | *argv); | |||
152 | else if (!selected(mntbuf[mntsize].f_fstypename)) | |||
153 | warnx("%s mounted as a %s file system", | |||
154 | *argv, mntbuf[mntsize].f_fstypename); | |||
155 | else | |||
156 | ++mntsize; | |||
157 | else | |||
158 | warn("%s", *argv); | |||
159 | } | |||
160 | } | |||
161 | ||||
162 | if (mntsize
| |||
163 | maxwidth = 11; | |||
164 | for (i = 0; i < mntsize; i++) { | |||
165 | width = strlen(mntbuf[i].f_mntfromname); | |||
166 | if (width > maxwidth) | |||
167 | maxwidth = width; | |||
168 | } | |||
169 | ||||
170 | if (Pflag) | |||
171 | posixprint(mntbuf, mntsize, maxwidth); | |||
172 | else | |||
173 | bsdprint(mntbuf, mntsize, maxwidth); | |||
174 | } | |||
175 | ||||
176 | return (mntsize ? 0 : 1); | |||
| ||||
177 | } | |||
178 | ||||
179 | char * | |||
180 | getmntpt(char *name) | |||
181 | { | |||
182 | long mntsize, i; | |||
183 | struct statfs *mntbuf; | |||
184 | ||||
185 | mntsize = getmntinfo(&mntbuf, MNT_NOWAIT2); | |||
186 | for (i = 0; i < mntsize; i++) { | |||
187 | if (!strcmp(mntbuf[i].f_mntfromname, name)) | |||
188 | return (mntbuf[i].f_mntonname); | |||
189 | } | |||
190 | return (0); | |||
191 | } | |||
192 | ||||
193 | static enum { IN_LIST, NOT_IN_LIST } which; | |||
194 | ||||
195 | static int | |||
196 | selected(const char *type) | |||
197 | { | |||
198 | char **av; | |||
199 | ||||
200 | /* If no type specified, it's always selected. */ | |||
201 | if (typelist == NULL((void *)0)) | |||
202 | return (1); | |||
203 | for (av = typelist; *av != NULL((void *)0); ++av) | |||
204 | if (!strncmp(type, *av, MFSNAMELEN16)) | |||
205 | return (which == IN_LIST ? 1 : 0); | |||
206 | return (which == IN_LIST ? 0 : 1); | |||
207 | } | |||
208 | ||||
209 | static void | |||
210 | maketypelist(char *fslist) | |||
211 | { | |||
212 | int i; | |||
213 | char *nextcp, **av; | |||
214 | ||||
215 | if ((fslist == NULL((void *)0)) || (fslist[0] == '\0')) | |||
216 | errx(1, "empty type list"); | |||
217 | ||||
218 | /* | |||
219 | * XXX | |||
220 | * Note: the syntax is "noxxx,yyy" for no xxx's and | |||
221 | * no yyy's, not the more intuitive "noyyy,noyyy". | |||
222 | */ | |||
223 | if (fslist[0] == 'n' && fslist[1] == 'o') { | |||
224 | fslist += 2; | |||
225 | which = NOT_IN_LIST; | |||
226 | } else | |||
227 | which = IN_LIST; | |||
228 | ||||
229 | /* Count the number of types. */ | |||
230 | for (i = 1, nextcp = fslist; (nextcp = strchr(nextcp, ',')) != NULL((void *)0); i++) | |||
231 | ++nextcp; | |||
232 | ||||
233 | /* Build an array of that many types. */ | |||
234 | if ((av = typelist = calloc(i + 1, sizeof(char *))) == NULL((void *)0)) | |||
235 | err(1, NULL((void *)0)); | |||
236 | av[0] = fslist; | |||
237 | for (i = 1, nextcp = fslist; (nextcp = strchr(nextcp, ',')) != NULL((void *)0); i++) { | |||
238 | *nextcp = '\0'; | |||
239 | av[i] = ++nextcp; | |||
240 | } | |||
241 | /* Terminate the array. */ | |||
242 | av[i] = NULL((void *)0); | |||
243 | } | |||
244 | ||||
245 | /* | |||
246 | * Make a pass over the filesystem info in ``mntbuf'' filtering out | |||
247 | * filesystem types not in ``fsmask'' and possibly re-stating to get | |||
248 | * current (not cached) info. Returns the new count of valid statfs bufs. | |||
249 | */ | |||
250 | static long | |||
251 | regetmntinfo(struct statfs **mntbufp, long mntsize) | |||
252 | { | |||
253 | int i, j; | |||
254 | struct statfs *mntbuf; | |||
255 | ||||
256 | if (!lflag && typelist == NULL((void *)0)) | |||
257 | return (nflag ? mntsize : getmntinfo(mntbufp, MNT_WAIT1)); | |||
258 | ||||
259 | mntbuf = *mntbufp; | |||
260 | j = 0; | |||
261 | for (i = 0; i < mntsize; i++) { | |||
262 | if (lflag && (mntbuf[i].f_flags & MNT_LOCAL0x00001000) == 0) | |||
263 | continue; | |||
264 | if (!selected(mntbuf[i].f_fstypename)) | |||
265 | continue; | |||
266 | if (nflag) | |||
267 | mntbuf[j] = mntbuf[i]; | |||
268 | else | |||
269 | (void)statfs(mntbuf[i].f_mntonname, &mntbuf[j]); | |||
270 | j++; | |||
271 | } | |||
272 | return (j); | |||
273 | } | |||
274 | ||||
275 | /* | |||
276 | * "human-readable" output: use 3 digits max.--put unit suffixes at | |||
277 | * the end. Makes output compact and easy-to-read esp. on huge disks. | |||
278 | * Code moved into libutil; this is now just a wrapper. | |||
279 | */ | |||
280 | static void | |||
281 | prthumanval(long long bytes) | |||
282 | { | |||
283 | char ret[FMT_SCALED_STRSIZE7]; | |||
284 | ||||
285 | if (fmt_scaled(bytes, ret) == -1) { | |||
286 | (void)printf(" %lld", bytes); | |||
287 | return; | |||
288 | } | |||
289 | (void)printf(" %7s", ret); | |||
290 | } | |||
291 | ||||
292 | static void | |||
293 | prthuman(struct statfs *sfsp, unsigned long long used) | |||
294 | { | |||
295 | prthumanval(sfsp->f_blocks * sfsp->f_bsize); | |||
296 | prthumanval(used * sfsp->f_bsize); | |||
297 | prthumanval(sfsp->f_bavail * sfsp->f_bsize); | |||
298 | } | |||
299 | ||||
300 | /* | |||
301 | * Convert statfs returned filesystem size into BLOCKSIZE units. | |||
302 | * Attempts to avoid overflow for large filesystems. | |||
303 | */ | |||
304 | #define fsbtoblk(num, fsbs, bs)(((fsbs) != 0 && (fsbs) < (bs)) ? (num) / ((bs) / ( fsbs)) : (num) * ((fsbs) / (bs))) \ | |||
305 | (((fsbs) != 0 && (fsbs) < (bs)) ? \ | |||
306 | (num) / ((bs) / (fsbs)) : (num) * ((fsbs) / (bs))) | |||
307 | ||||
308 | /* | |||
309 | * Print out status about a filesystem. | |||
310 | */ | |||
311 | static void | |||
312 | prtstat(struct statfs *sfsp, int maxwidth, int headerlen, int blocksize) | |||
313 | { | |||
314 | u_int64_t used, inodes; | |||
315 | int64_t availblks; | |||
316 | ||||
317 | (void)printf("%-*.*s", maxwidth, maxwidth, sfsp->f_mntfromname); | |||
318 | used = sfsp->f_blocks - sfsp->f_bfree; | |||
319 | availblks = sfsp->f_bavail + used; | |||
320 | if (hflag) | |||
321 | prthuman(sfsp, used); | |||
322 | else | |||
323 | (void)printf(" %*llu %9llu %9lld", headerlen, | |||
324 | fsbtoblk(sfsp->f_blocks, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (sfsp->f_blocks) / ((blocksize) / (sfsp-> f_bsize)) : (sfsp->f_blocks) * ((sfsp->f_bsize) / (blocksize ))), | |||
325 | fsbtoblk(used, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (used) / ((blocksize) / (sfsp->f_bsize)) : ( used) * ((sfsp->f_bsize) / (blocksize))), | |||
326 | fsbtoblk(sfsp->f_bavail, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (sfsp->f_bavail) / ((blocksize) / (sfsp-> f_bsize)) : (sfsp->f_bavail) * ((sfsp->f_bsize) / (blocksize )))); | |||
327 | (void)printf(" %5d%%", percent(used, availblks)); | |||
328 | if (iflag) { | |||
329 | inodes = sfsp->f_files; | |||
330 | used = inodes - sfsp->f_ffree; | |||
331 | (void)printf(" %7llu %7llu %5d%% ", used, sfsp->f_ffree, | |||
332 | percent(used, inodes)); | |||
333 | } else | |||
334 | (void)printf(" "); | |||
335 | (void)printf(" %s\n", sfsp->f_mntonname); | |||
336 | } | |||
337 | ||||
338 | /* | |||
339 | * Print in traditional BSD format. | |||
340 | */ | |||
341 | static void | |||
342 | bsdprint(struct statfs *mntbuf, long mntsize, int maxwidth) | |||
343 | { | |||
344 | int i; | |||
345 | char *header; | |||
346 | int headerlen; | |||
347 | long blocksize; | |||
348 | ||||
349 | /* Print the header line */ | |||
350 | if (hflag) { | |||
351 | header = " Size"; | |||
352 | headerlen = strlen(header); | |||
353 | (void)printf("%-*.*s %s Used Avail Capacity", | |||
354 | maxwidth, maxwidth, "Filesystem", header); | |||
355 | } else { | |||
356 | if (kflag) { | |||
357 | blocksize = 1024; | |||
358 | header = "1K-blocks"; | |||
359 | headerlen = strlen(header); | |||
360 | } else | |||
361 | header = getbsize(&headerlen, &blocksize); | |||
362 | (void)printf("%-*.*s %s Used Avail Capacity", | |||
363 | maxwidth, maxwidth, "Filesystem", header); | |||
364 | } | |||
365 | if (iflag) | |||
366 | (void)printf(" iused ifree %%iused"); | |||
367 | (void)printf(" Mounted on\n"); | |||
368 | ||||
369 | ||||
370 | for (i = 0; i < mntsize; i++) | |||
371 | prtstat(&mntbuf[i], maxwidth, headerlen, blocksize); | |||
372 | return; | |||
373 | } | |||
374 | ||||
375 | static int | |||
376 | percent(u_int64_t used, u_int64_t avail) | |||
377 | { | |||
378 | return avail ? (100 * used + (avail - 1)) / avail : 100; | |||
379 | } | |||
380 | ||||
381 | /* | |||
382 | * Print in format defined by POSIX 1002.2, invoke with -P option. | |||
383 | */ | |||
384 | static void | |||
385 | posixprint(struct statfs *mntbuf, long mntsize, int maxwidth) | |||
386 | { | |||
387 | int i; | |||
388 | int blocksize; | |||
389 | char *blockstr; | |||
390 | struct statfs *sfsp; | |||
391 | long long used, avail; | |||
392 | ||||
393 | if (kflag) { | |||
394 | blocksize = 1024; | |||
395 | blockstr = "1024-blocks"; | |||
396 | } else { | |||
397 | blocksize = 512; | |||
398 | blockstr = " 512-blocks"; | |||
399 | } | |||
400 | ||||
401 | (void)printf( | |||
402 | "%-*.*s %s Used Available Capacity Mounted on\n", | |||
403 | maxwidth, maxwidth, "Filesystem", blockstr); | |||
404 | ||||
405 | for (i = 0; i < mntsize; i++) { | |||
406 | sfsp = &mntbuf[i]; | |||
407 | used = sfsp->f_blocks - sfsp->f_bfree; | |||
408 | avail = sfsp->f_bavail + used; | |||
409 | ||||
410 | (void) printf ("%-*.*s %*lld %10lld %11lld %5d%% %s\n", | |||
411 | maxwidth, maxwidth, sfsp->f_mntfromname, | |||
412 | (int)strlen(blockstr), | |||
413 | fsbtoblk(sfsp->f_blocks, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (sfsp->f_blocks) / ((blocksize) / (sfsp-> f_bsize)) : (sfsp->f_blocks) * ((sfsp->f_bsize) / (blocksize ))), | |||
414 | fsbtoblk(used, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (used) / ((blocksize) / (sfsp->f_bsize)) : ( used) * ((sfsp->f_bsize) / (blocksize))), | |||
415 | fsbtoblk(sfsp->f_bavail, sfsp->f_bsize, blocksize)(((sfsp->f_bsize) != 0 && (sfsp->f_bsize) < ( blocksize)) ? (sfsp->f_bavail) / ((blocksize) / (sfsp-> f_bsize)) : (sfsp->f_bavail) * ((sfsp->f_bsize) / (blocksize ))), | |||
416 | percent(used, avail), sfsp->f_mntonname); | |||
417 | } | |||
418 | } | |||
419 | ||||
420 | static int | |||
421 | raw_df(char *file, struct statfs *sfsp) | |||
422 | { | |||
423 | int rfd, ret = -1; | |||
424 | ||||
425 | if ((rfd = open(file, O_RDONLY0x0000)) == -1) { | |||
426 | warn("%s", file); | |||
427 | return (-1); | |||
428 | } | |||
429 | ||||
430 | if (ffs_df(rfd, file, sfsp) == 0) { | |||
431 | ret = 0; | |||
432 | } else if (e2fs_df(rfd, file, sfsp) == 0) { | |||
433 | ret = 0; | |||
434 | } | |||
435 | ||||
436 | close (rfd); | |||
437 | return (ret); | |||
438 | } | |||
439 | ||||
440 | int | |||
441 | bread(int rfd, off_t off, void *buf, int cnt) | |||
442 | { | |||
443 | int nr; | |||
444 | ||||
445 | if ((nr = pread(rfd, buf, cnt, off)) != cnt) { | |||
446 | /* Probably a dismounted disk if errno == EIO. */ | |||
447 | if (errno(*__errno()) != EIO5) | |||
448 | (void)fprintf(stderr(&__sF[2]), "\ndf: %lld: %s\n", | |||
449 | (long long)off, strerror(nr > 0 ? EIO5 : errno(*__errno()))); | |||
450 | return (0); | |||
451 | } | |||
452 | return (1); | |||
453 | } | |||
454 | ||||
455 | static __dead__attribute__((__noreturn__)) void | |||
456 | usage(void) | |||
457 | { | |||
458 | (void)fprintf(stderr(&__sF[2]), | |||
459 | "usage: %s [-hiklnP] [-t type] [[file | file_system] ...]\n", | |||
460 | getprogname()); | |||
461 | exit(1); | |||
462 | } |