/*      $NetBSD: getmntinfo.c,v 1.1 2012/03/17 16:33:11 jruoho Exp $    */
/*
 * Copyright (c) 2007 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

#include <sys/param.h>
#include <sys/ucred.h>
#include <sys/mount.h>

#include <err.h>
#include <stdlib.h>
#include <string.h>

#define KB              * 1024
#define MB              * 1024 KB
#define GB              * 1024 MB

static struct statvfs *getnewstatvfs(void);
static void other_variants(const struct statvfs *, const int *, int,
    const int *, int);
static void setup_filer(void);
static void setup_ld0g(void);
static void setup_strpct(void);

static struct statvfs *allstatvfs;
static int sftotal, sfused;

struct statvfs *
getnewstatvfs(void)
{

        if (sftotal == sfused) {
                sftotal = sftotal ? sftotal * 2 : 1;
                allstatvfs = realloc(allstatvfs,
                    sftotal * sizeof(struct statvfs));
                if (allstatvfs == NULL)
                        err(EXIT_FAILURE, "realloc");
        }

        return (&allstatvfs[sfused++]);
}

void
other_variants(const struct statvfs *tmpl, const int *minfree, int minfreecnt,
    const int *consumed, int consumedcnt)
{
        int64_t total, used;
        struct statvfs *sf;
        int i, j;

        for (i = 0; i < minfreecnt; i++)
                for (j = 0; j < consumedcnt; j++) {
                        sf = getnewstatvfs();
                        *sf = *tmpl;
                        total = (int64_t)(u_long)sf->f_blocks * sf->f_bsize;
                        used =  total * consumed[j] / 100;
                        sf->f_bfree = (total - used) / sf->f_bsize;
                        sf->f_bavail = (total * (100 - minfree[i]) / 100 -
                            used) / (int)sf->f_bsize;
                        sf->f_bresvd = sf->f_bfree - sf->f_bavail;
                }
}

/*
 * Parameter taken from:
 * http://mail-index.NetBSD.org/tech-userlevel/2004/03/24/0001.html
 */
void
setup_filer(void)
{
        static const struct statvfs tmpl = {
#define BSIZE   512
#define TOTAL   1147ULL GB
#define USED    132ULL MB
                .f_bsize = BSIZE,
                .f_frsize = BSIZE,
                .f_blocks = TOTAL / BSIZE,
                .f_bfree = (TOTAL - USED) / BSIZE,
                .f_bavail = (TOTAL - USED) / BSIZE,
                .f_bresvd = 0,
                .f_mntfromname = "filer:/",
                .f_mntonname = "/filer",
#undef USED
#undef TOTAL
#undef BSIZE
        };
        static const int minfree[] = { 0, 5, 10, 15, };
        static const int consumed[] = { 0, 20, 60, 95, 100 };

        *getnewstatvfs() = tmpl;
        other_variants(&tmpl, minfree, sizeof(minfree) / sizeof(minfree[0]),
            consumed, sizeof(consumed) / sizeof(consumed[0]));
}

/*
 * Parameter taken from:
 * http://mail-index.NetBSD.org/current-users/2004/03/01/0038.html
 */
void
setup_ld0g(void)
{
        static const struct statvfs tmpl = {
#define BSIZE   4096                    /* Guess */
#define TOTAL   1308726116ULL KB
#define USED    17901268ULL KB
#define AVAIL   1225388540ULL KB
                .f_bsize = BSIZE,
                .f_frsize = BSIZE,
                .f_blocks = TOTAL / BSIZE,
                .f_bfree = (TOTAL - USED) / BSIZE,
                .f_bavail = AVAIL / BSIZE,
                .f_bresvd = (TOTAL - USED) / BSIZE - AVAIL / BSIZE,
                .f_mntfromname = "/dev/ld0g",
                .f_mntonname = "/anon-root",
#undef AVAIL
#undef USED
#undef TOTAL
#undef BSIZE
        };
        static const int minfree[] = { 0, 5, 10, 15, };
        static const int consumed[] = { 0, 20, 60, 95, 100 };

        *getnewstatvfs() = tmpl;
        other_variants(&tmpl, minfree, sizeof(minfree) / sizeof(minfree[0]),
            consumed, sizeof(consumed) / sizeof(consumed[0]));
}

/*
 * Test of strpct() with huge number.
 */
void
setup_strpct(void)
{
        static const struct statvfs tmpl = {
#define BSIZE   4096                    /* Guess */
#define TOTAL   0x4ffffffffULL KB
#define USED    (TOTAL / 2)
#define AVAIL   (TOTAL / 2)
                .f_bsize = BSIZE,
                .f_frsize = BSIZE,
                .f_blocks = TOTAL / BSIZE,
                .f_bfree = (TOTAL - USED) / BSIZE,
                .f_bavail = AVAIL / BSIZE,
                .f_bresvd = (TOTAL - USED) / BSIZE - AVAIL / BSIZE,
                .f_mntfromname = "/dev/strpct",
                .f_mntonname = "/strpct",
#undef AVAIL
#undef USED
#undef TOTAL
#undef BSIZE
        };

        *getnewstatvfs() = tmpl;
}

/*
 * Parameter taken from:
 * http://www.netbsd.org/cgi-bin/query-pr-single.pl?number=23600
 */
static void
setup_pr23600(void)
{
        static const struct statvfs tmpl = {
#define BSIZE   512
#define TOTAL   20971376ULL
#define USED    5719864ULL
#define AVAIL   15251512ULL
                .f_bsize = BSIZE,
                .f_frsize = BSIZE,
                .f_blocks = TOTAL,
                .f_bfree = TOTAL - USED,
                .f_bavail = AVAIL,
                .f_bresvd = TOTAL - USED - AVAIL,
                .f_mntfromname = "/dev/wd0e",
                .f_mntonname = "/mount/windows/C",
#undef AVAIL
#undef USED
#undef TOTAL
#undef BSIZE
        };

        *getnewstatvfs() = tmpl;
}

int
getmntinfo(struct statvfs **mntbuf, int flags)
{

        setup_filer();
        setup_ld0g();
        setup_strpct();
        setup_pr23600();

        *mntbuf = allstatvfs;
        return (sfused);
}