2 * Copyright (c) 2011 François Tigeot <ftigeot@wolfpond.org>
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in
13 * the documentation and/or other materials provided with the
15 * 3. Neither the name of The DragonFly Project nor the names of its
16 * contributors may be used to endorse or promote products derived
17 * from this software without specific, prior written permission.
19 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
20 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
21 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
22 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
23 * COPYRIGHT HOLDERS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
24 * INCIDENTAL, SPECIAL, EXEMPLARY OR CONSEQUENTIAL DAMAGES (INCLUDING,
25 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
26 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
27 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
28 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
29 * OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
34 #include <sys/mount.h>
43 static void usage(int);
44 static int get_dirsize(char *);
45 static int get_fslist(void);
50 fprintf(stderr, "usage: vquota check directory\n");
51 fprintf(stderr, " vquota lsfs\n");
56 get_dirsize(char* dirname)
61 uint64_t size_of_files = 0;
64 /* TODO: check directory name sanity */
65 fts_args[0] = dirname;
68 if ((fts = fts_open(fts_args, FTS_PHYSICAL, NULL)) == NULL)
69 err(1, "fts_open() failed");
71 while ((p = fts_read(fts)) != NULL) {
72 switch (p->fts_info) {
73 /* directories, ignore them */
78 /* read errors, warn, continue and flag */
82 warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
86 size_of_files += p->fts_statp->st_size;
91 printf("%"PRIu64"\n", size_of_files);
95 /* print a list of filesystems with accounting enabled */
96 static int get_fslist(void) {
97 struct statfs *mntbufp;
100 /* read mount table from kernel */
101 nloc = getmntinfo(&mntbufp, MNT_NOWAIT|MNT_LOCAL);
103 perror("getmntinfo");
107 /* iterate mounted filesystems */
108 for (i=0; i<nloc; i++) {
109 /* vfs accounting enabled on this one ? */
110 if (mntbufp[i].f_flags & MNT_ACCOUNTING)
111 printf("%s on %s\n", mntbufp[i].f_mntfromname,
112 mntbufp[i].f_mntonname);
119 main(int argc, char **argv)
124 if (strcmp(argv[1], "check") == 0) {
127 return get_dirsize(argv[2]);
129 if (strcmp(argv[1], "lsfs") == 0) {