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 #include <sys/mount.h>
  35
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38 #include <err.h>
  39 #include <string.h>
  40 #include <fts.h>
  41 #include <inttypes.h>
  42
  43 static void usage(int);
  44 static int get_dirsize(char *);
  45 static int get_fslist(void);
  46
  47 static void
  48 usage(int retcode)
  49 {
  50         fprintf(stderr, "usage: vquota check directory\n");
  51         fprintf(stderr, "       vquota lsfs\n");
  52         exit(retcode);
  53 }
  54
  55 static int
  56 get_dirsize(char* dirname)
  57 {
  58         FTS             *fts;
  59         FTSENT          *p;
  60         char*           fts_args[2];
  61         uint64_t        size_of_files = 0;
  62         int             retval = 0;
  63
  64         /* TODO: check directory name sanity */
  65         fts_args[0] = dirname;
  66         fts_args[1] = NULL;
  67
  68         if ((fts = fts_open(fts_args, FTS_PHYSICAL, NULL)) == NULL)
  69                 err(1, "fts_open() failed");
  70
  71         while ((p = fts_read(fts)) != NULL) {
  72                 switch (p->fts_info) {
  73                 /* directories, ignore them */
  74                 case FTS_D:
  75                 case FTS_DC:
  76                 case FTS_DP:
  77                         break;
  78                 /* read errors, warn, continue and flag */
  79                 case FTS_DNR:
  80                 case FTS_ERR:
  81                 case FTS_NS:
  82                         warnx("%s: %s", p->fts_path, strerror(p->fts_errno));
  83                         retval = 1;
  84                         break;
  85                 default:
  86                         size_of_files += p->fts_statp->st_size;
  87                 }
  88         }
  89         fts_close(fts);
  90
  91         printf("%"PRIu64"\n", size_of_files);
  92         return retval;
  93 }
  94
  95 /* print a list of filesystems with accounting enabled */
  96 static int get_fslist(void) {
  97         struct statfs *mntbufp;
  98         int nloc, i;
  99
 100         /* read mount table from kernel */
 101         nloc = getmntinfo(&mntbufp, MNT_NOWAIT|MNT_LOCAL);
 102         if (nloc <= 0) {
 103                 perror("getmntinfo");
 104                 exit(1);
 105         }
 106
 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);
 113         }
 114
 115         return 0;
 116 }
 117
 118 int
 119 main(int argc, char **argv)
 120 {
 121         if (argc < 2)
 122                 usage(1);
 123
 124         if (strcmp(argv[1], "check") == 0) {
 125                 if (argc < 3)
 126                         usage(1);
 127                 return get_dirsize(argv[2]);
 128         }
 129         if (strcmp(argv[1], "lsfs") == 0) {
 130                 return get_fslist();
 131         }
 132
 133         return(0);
 134 }