/*      $NetBSD: cleanalot_async.c,v 1.7 2008/04/28 20:23:06 martin Exp $       */

/*-
 * Copyright (c) 2006 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by Konrad E. Schroder <perseant@hhhh.org>.
 *
 * 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 <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>

void dirname(int n, char *s)
{
        if (n == 0) {
                strcat(s, "/0");
                mkdir(s);
        }
        while (n) {
                sprintf(s + strlen(s), "/%x", n & 0xf);
                n >>= 4;
                mkdir(s);
        }
}

/*
 * Write a file, creating the directory if necessary.
 */
int write_file(int gen, int n, int plex, char *buf, int size)
{
        FILE *fp;
        char s[1024], *t;
        int r;

        sprintf(s, "dir_%x_%x", plex, gen);
        dirname(n, s);
        strcat(s, ".file");

        /* printf("write file %d.%d.%x: %s\n", gen, plex, n, s); */

        fp = fopen(s, "wb");
        if (fp == NULL)
                return 0;
        if (size)
        r = fwrite(buf, size, 1, fp);
        else
                r = 1;
        fclose(fp);

        return r;
}

int write_dirs(int gen, int size, int plex)
{
        int i, j, tot;
        char s[1024];
        char *buf;

        /* Create all base dirs */
        for (i = 0; i < plex; i++) {
                sprintf(s, "dir_%x_%x", i, gen);
                if (mkdir(s, 0700) != 0)
                        return 0;
        }

        /* Write files */
        if (size) {
        buf = malloc(size);
        if (buf == NULL)
                return 0;
        }
        tot = 0;
        for (i = 0; ; i++) {
                for (j = 0; j < plex; j++) {
                        if (write_file(gen, i, j, buf, size) == 0) {
                                if (size)
                                free(buf);
                                return tot;
                        }
                        ++tot;
                }
        }
        /* NOTREACHED */
}

int main(int argc, char **argv)
{
        int c, i, j;
        int bs = 0;
        int count = 0;
        int plex = 2;
        char cmd[1024];

        bs = -1;
        while((c = getopt(argc, argv, "b:n:p:")) != -1) {
                switch(c) {
                    case 'b':
                        bs = atoi(optarg);
                        break;
                    case 'n':
                        count = atoi(optarg);
                        break;
                    case 'p':
                        plex = atoi(optarg);
                        break;
                    default:
                        exit(1);
                }
        }

        /*
         * Process old-style, non-flag parameters
         */
        if (count == 0) {
                if (argv[optind] != NULL)
                        count = atoi(argv[optind]);
        }
        if (bs < 0 && getenv("BS") != NULL)
                bs = atoi(getenv("BS"));
        if (bs < 0)
                bs = 16384;
        if (plex == 0)
                plex = 2;

        for (i = 0; count == 0 || i < count; i++) {
                if (count)
                        printf("::: begin iteration %d/%d\n", i, count);
                else
                        printf("::: begin iteration %d\n", i);

                for (j = 0; ; j++) {
                        if ((c = write_dirs(j, bs, plex)) == 0)
                                break;
                        printf("%d: %d files of size %d\n", j, c, bs);
                        sprintf(cmd, "rm -rf dir_%x_%x", plex - 1, j);
                        system(cmd);
                        sync();
                }
                system("df -k .");
                printf("remove files\n");
                system("rm -rf dir_*");
                system("df -k .");
        }
}