Add a vquota(8) utility
[dragonfly.git] / sbin / vquota / vquota.c
1 /*
2  * Copyright (c) 2011 Fran├žois Tigeot <ftigeot@wolfpond.org>
3  * All rights reserved.
4  * 
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 
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
14  *    distribution.
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.
18  * 
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
30  * SUCH DAMAGE.
31  */
32
33 #include <sys/stat.h>
34
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <err.h>
38 #include <string.h>
39 #include <fts.h>
40
41 static void usage(int);
42 static int get_dirsize( char* );
43
44 static void usage(int retcode) {
45         fprintf( stderr, "usage: vquota check directory\n" );
46         exit(retcode);
47 }
48
49 static int get_dirsize( char* dirname ) {
50         FTS                     *fts;
51         FTSENT          *p;
52         char*           fts_args[2];
53         uint64_t        size_of_files = 0;
54         int                     retval = 0;
55
56         /* TODO: check directory name sanity */
57         fts_args[0] = dirname;
58         fts_args[1] = NULL;
59
60         if ((fts = fts_open( fts_args, FTS_PHYSICAL, NULL)) == NULL)
61                 err(1, "fts_open() failed");
62
63         while ((p = fts_read(fts)) != NULL) {
64                 switch (p->fts_info) {
65                 /* directories, ignore them */
66                 case FTS_D:
67                 case FTS_DC:
68                 case FTS_DP:
69                         break;
70                 /* read errors, warn, continue and flag */
71                 case FTS_DNR:
72                 case FTS_ERR:
73                 case FTS_NS:
74                         warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
75                         retval = 1;
76                         break;
77                 default:
78                         /* files with more than one link, undecided */
79                         // if (p->fts_statp->st_nlink > 1 && linkchk(p))
80                         if (p->fts_statp->st_nlink > 1)
81                                 fprintf( stderr, "%s has more than one link\n", p->fts_name );
82                                 
83                         size_of_files += p->fts_statp->st_size;
84                 }
85         }
86
87         printf( "%lu\n", size_of_files );
88         return retval;
89 }
90
91 int main( int argc, char **argv ) {
92         
93         if (argc < 3)
94                 usage(1);
95         
96         if (strcmp(argv[1], "check") == 0) {
97                 return get_dirsize( argv[2] );
98         }
99
100         return 0;
101 }