vquota(8): Silence 'has more than one link' messages.
[dragonfly.git] / sbin / vquota / vquota.c
CommitLineData
6a4c3e18
FT
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>
50c7dea7 40#include <inttypes.h>
6a4c3e18
FT
41
42static void usage(int);
cefc2cb0 43static int get_dirsize(char *);
6a4c3e18 44
cefc2cb0
SW
45static void
46usage(int retcode)
47{
48 fprintf(stderr, "usage: vquota check directory\n");
6a4c3e18
FT
49 exit(retcode);
50}
51
cefc2cb0
SW
52static int
53get_dirsize(char* dirname)
54{
55 FTS *fts;
6a4c3e18
FT
56 FTSENT *p;
57 char* fts_args[2];
58 uint64_t size_of_files = 0;
cefc2cb0 59 int retval = 0;
6a4c3e18
FT
60
61 /* TODO: check directory name sanity */
62 fts_args[0] = dirname;
63 fts_args[1] = NULL;
64
cefc2cb0 65 if ((fts = fts_open(fts_args, FTS_PHYSICAL, NULL)) == NULL)
6a4c3e18
FT
66 err(1, "fts_open() failed");
67
68 while ((p = fts_read(fts)) != NULL) {
69 switch (p->fts_info) {
70 /* directories, ignore them */
71 case FTS_D:
72 case FTS_DC:
73 case FTS_DP:
74 break;
75 /* read errors, warn, continue and flag */
76 case FTS_DNR:
77 case FTS_ERR:
78 case FTS_NS:
79 warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
80 retval = 1;
81 break;
82 default:
44700357 83#if 0
6a4c3e18
FT
84 /* files with more than one link, undecided */
85 // if (p->fts_statp->st_nlink > 1 && linkchk(p))
86 if (p->fts_statp->st_nlink > 1)
cefc2cb0
SW
87 fprintf(stderr, "%s has more than one link\n",
88 p->fts_name );
44700357 89#endif
6a4c3e18
FT
90 size_of_files += p->fts_statp->st_size;
91 }
92 }
a89ecd20 93 fts_close(fts);
6a4c3e18 94
cefc2cb0 95 printf("%"PRIu64"\n", size_of_files);
6a4c3e18
FT
96 return retval;
97}
98
cefc2cb0
SW
99int
100main(int argc, char **argv)
101{
6a4c3e18
FT
102 if (argc < 3)
103 usage(1);
104
105 if (strcmp(argv[1], "check") == 0) {
cefc2cb0 106 return get_dirsize(argv[2]);
44700357
SW
107 } else {
108 usage(1);
6a4c3e18
FT
109 }
110
111 return 0;
112}