tests/util/df/getmntinfo.c

   1 /*      $NetBSD: getmntinfo.c,v 1.2 2007/12/03 18:49:24 riz Exp $       */
   2 /*
   3  * Copyright (c) 2007 The NetBSD Foundation, Inc.
   4  * All rights reserved.
   5  *
   6  * Redistribution and use in source and binary forms, with or without
   7  * modification, are permitted provided that the following conditions
   8  * are met:
   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 the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  16  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  17  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  18  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  19  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  20  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  21  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  22  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  23  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  24  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  25  * POSSIBILITY OF SUCH DAMAGE.
  26  */
  27
  28 #include <sys/param.h>
  29 #include <sys/ucred.h>
  30 #include <sys/mount.h>
  31
  32 #include <err.h>
  33 #include <stdlib.h>
  34 #include <string.h>
  35
  36 #define KB              * 1024
  37 #define MB              * 1024 KB
  38 #define GB              * 1024 MB
  39
  40 static struct statvfs *getnewstatvfs(void);
  41 static void other_variants(const struct statvfs *, const int *, int,
  42     const int *, int);
  43 static void setup_filer(void);
  44 static void setup_ld0g(void);
  45 static void setup_strpct(void);
  46
  47 static struct statvfs *allstatvfs;
  48 static int sftotal, sfused;
  49
  50 struct statvfs *
  51 getnewstatvfs(void)
  52 {
  53
  54         if (sftotal == sfused) {
  55                 sftotal = sftotal ? sftotal * 2 : 1;
  56                 allstatvfs = realloc(allstatvfs,
  57                     sftotal * sizeof(struct statvfs));
  58                 if (allstatvfs == NULL)
  59                         err(EXIT_FAILURE, "realloc");
  60         }
  61
  62         return (&allstatvfs[sfused++]);
  63 }
  64
  65 void
  66 other_variants(const struct statvfs *tmpl, const int *minfree, int minfreecnt,
  67     const int *consumed, int consumedcnt)
  68 {
  69         int64_t total, used;
  70         struct statvfs *sf;
  71         int i, j;
  72
  73         for (i = 0; i < minfreecnt; i++)
  74                 for (j = 0; j < consumedcnt; j++) {
  75                         sf = getnewstatvfs();
  76                         *sf = *tmpl;
  77                         total = (int64_t)(u_long)sf->f_blocks * sf->f_bsize;
  78                         used =  total * consumed[j] / 100;
  79                         sf->f_bfree = (total - used) / sf->f_bsize;
  80                         sf->f_bavail = (total * (100 - minfree[i]) / 100 -
  81                             used) / (int)sf->f_bsize;
  82                         sf->f_bresvd = sf->f_bfree - sf->f_bavail;
  83                 }
  84 }
  85
  86 /*
  87  * Parameter taken from:
  88  * http://mail-index.NetBSD.org/tech-userlevel/2004/03/24/0001.html
  89  */
  90 void
  91 setup_filer(void)
  92 {
  93         static const struct statvfs tmpl = {
  94 #define BSIZE   512
  95 #define TOTAL   1147ULL GB
  96 #define USED    132ULL MB
  97                 .f_bsize = BSIZE,
  98                 .f_frsize = BSIZE,
  99                 .f_blocks = TOTAL / BSIZE,
 100                 .f_bfree = (TOTAL - USED) / BSIZE,
 101                 .f_bavail = (TOTAL - USED) / BSIZE,
 102                 .f_bresvd = 0,
 103                 .f_mntfromname = "filer:/",
 104                 .f_mntonname = "/filer",
 105 #undef USED
 106 #undef TOTAL
 107 #undef BSIZE
 108         };
 109         static const int minfree[] = { 0, 5, 10, 15, };
 110         static const int consumed[] = { 0, 20, 60, 95, 100 };
 111
 112         *getnewstatvfs() = tmpl;
 113         other_variants(&tmpl, minfree, sizeof(minfree) / sizeof(minfree[0]),
 114             consumed, sizeof(consumed) / sizeof(consumed[0]));
 115 }
 116
 117 /*
 118  * Parameter taken from:
 119  * http://mail-index.NetBSD.org/current-users/2004/03/01/0038.html
 120  */
 121 void
 122 setup_ld0g(void)
 123 {
 124         static const struct statvfs tmpl = {
 125 #define BSIZE   4096                    /* Guess */
 126 #define TOTAL   1308726116ULL KB
 127 #define USED    17901268ULL KB
 128 #define AVAIL   1225388540ULL KB
 129                 .f_bsize = BSIZE,
 130                 .f_frsize = BSIZE,
 131                 .f_blocks = TOTAL / BSIZE,
 132                 .f_bfree = (TOTAL - USED) / BSIZE,
 133                 .f_bavail = AVAIL / BSIZE,
 134                 .f_bresvd = (TOTAL - USED) / BSIZE - AVAIL / BSIZE,
 135                 .f_mntfromname = "/dev/ld0g",
 136                 .f_mntonname = "/anon-root",
 137 #undef AVAIL
 138 #undef USED
 139 #undef TOTAL
 140 #undef BSIZE
 141         };
 142         static const int minfree[] = { 0, 5, 10, 15, };
 143         static const int consumed[] = { 0, 20, 60, 95, 100 };
 144
 145         *getnewstatvfs() = tmpl;
 146         other_variants(&tmpl, minfree, sizeof(minfree) / sizeof(minfree[0]),
 147             consumed, sizeof(consumed) / sizeof(consumed[0]));
 148 }
 149
 150 /*
 151  * Test of strpct() with huge number.
 152  */
 153 void
 154 setup_strpct(void)
 155 {
 156         static const struct statvfs tmpl = {
 157 #define BSIZE   4096                    /* Guess */
 158 #define TOTAL   0x4ffffffffULL KB
 159 #define USED    (TOTAL / 2)
 160 #define AVAIL   (TOTAL / 2)
 161                 .f_bsize = BSIZE,
 162                 .f_frsize = BSIZE,
 163                 .f_blocks = TOTAL / BSIZE,
 164                 .f_bfree = (TOTAL - USED) / BSIZE,
 165                 .f_bavail = AVAIL / BSIZE,
 166                 .f_bresvd = (TOTAL - USED) / BSIZE - AVAIL / BSIZE,
 167                 .f_mntfromname = "/dev/strpct",
 168                 .f_mntonname = "/strpct",
 169 #undef AVAIL
 170 #undef USED
 171 #undef TOTAL
 172 #undef BSIZE
 173         };
 174
 175         *getnewstatvfs() = tmpl;
 176 }
 177
 178 /*
 179  * Parameter taken from:
 180  * http://www.netbsd.org/cgi-bin/query-pr-single.pl?number=23600
 181  */
 182 void
 183 setup_pr23600(void)
 184 {
 185         static const struct statvfs tmpl = {
 186 #define BSIZE   512
 187 #define TOTAL   20971376ULL
 188 #define USED    5719864ULL
 189 #define AVAIL   15251512ULL
 190                 .f_bsize = BSIZE,
 191                 .f_frsize = BSIZE,
 192                 .f_blocks = TOTAL,
 193                 .f_bfree = TOTAL - USED,
 194                 .f_bavail = AVAIL,
 195                 .f_bresvd = TOTAL - USED - AVAIL,
 196                 .f_mntfromname = "/dev/wd0e",
 197                 .f_mntonname = "/mount/windows/C",
 198 #undef AVAIL
 199 #undef USED
 200 #undef TOTAL
 201 #undef BSIZE
 202         };
 203
 204         *getnewstatvfs() = tmpl;
 205 }
 206
 207 int
 208 getmntinfo(struct statvfs **mntbuf, int flags)
 209 {
 210
 211         setup_filer();
 212         setup_ld0g();
 213         setup_strpct();
 214         setup_pr23600();
 215
 216         *mntbuf = allstatvfs;
 217         return (sfused);
 218 }