/*      $NetBSD: 9994_f.c,v 1.3 2008/04/28 20:23:06 martin Exp $        */

/*-
 * Copyright (c) 2002 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * This code is derived from software contributed to The NetBSD Foundation
 * by IWAMOTO Toshihiro <iwamoto@sat.t.u-tokyo.ac.jp> and 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.
 */

/*
 * Regression test based on test program submutted with NetBSD PR #9994.
 * Two files are opened, one in the current directory and another in a
 * control directory (in another filesystem).  Random read, write, and
 * truncate operations are performed on both files.
 *
 * An error return indicates that an operation failed on one of the files
 * (control *or* test, either one).
 *
 * A zero return indicates success; the two files should be identical.
 * (This program does not test to make sure that they are identical.)
 */

#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <sys/types.h>
#include <unistd.h>

#define CONTROL_ERROR 1
#define TEST_ERROR    2

int main(int argc, char **argv)
{
        int bsize, quiet, usepid;
        char *hoge, *buf;
        char prop[] = "/\b-\b\\\b|\b";
        int testfd, controlfd;
        int c, i, j, k, n, p;
        char *ctldir;
        unsigned long rseed;
        const char *prog = getprogname();

        bsize = 8192;
        n = 10000;
        ctldir = "/var/tmp";
        quiet = 0;
        usepid = 0;
        rseed = time(0);

        while ((c = getopt(argc, argv, "b:c:n:pqs:")) != -1) {
                switch(c) {
                    case 'b':
                            bsize = atoi(optarg);
                            break;
                    case 'c':
                            ctldir = optarg;
                            break;
                    case 'n':
                            n = atoi(optarg);
                            break;
                    case 'p':
                            ++usepid;
                            break;
                    case 'q':
                            ++quiet;
                            break;
                    case 's':
                            rseed = strtoul(optarg, NULL, 0);
                            break;
                    default:
                            errx(1, "usage: %s [-b bsize] [-c control-dir] [-n count] [-pq] [-s randseed]", prog);
                            /* NOTREACHED */
                }
        }

        srandom(rseed);
        hoge = (char *)malloc(bsize);
        buf = (char *)malloc(bsize);

        for(i = 0; i < bsize; i++)
                hoge[i] = random() & 0xff;

        /* Set up test and control file descriptors */

        if (usepid)
                sprintf(buf, "test.%d", getpid());
        else
                sprintf(buf, "test");
        unlink(buf);
        testfd = open(buf, O_RDWR | O_CREAT, 0644);
        if (testfd < 0) {
                perror("open");
                exit(TEST_ERROR);
        }

        if (usepid)
                sprintf(buf, "%s/control.%d", ctldir, getpid());
        else
                sprintf(buf, "%s/control", ctldir);
        unlink(buf);
        controlfd = open(buf, O_RDWR | O_CREAT, 0644);
        if (controlfd < 0) {
                perror("open");
                exit(CONTROL_ERROR);
        }

        for(i = 0, p = 0; i < n; i++) {
                if (!quiet) {
                        /* Draw propeller */
                        p = (p + 1) % 4;
                        write(1, prop + 2 * p, 2);
                }

                j = random() % 3;
                k = random() & 0x3fffff;
                if (j != 2) {
                        if (lseek(testfd, k, SEEK_SET) < 0) {
                                perror("read");
                                exit(TEST_ERROR);
                        }
                        if (lseek(controlfd, k, SEEK_SET) < 0) {
                                perror("read");
                                exit(CONTROL_ERROR);
                        }
                }

                switch(j) {
                    case 0:
                            if (read(testfd, buf, bsize) < 0) {
                                    perror("read");
                                    exit(TEST_ERROR);
                            }
                            if (read(controlfd, buf, bsize) < 0) {
                                    perror("read");
                                    exit(CONTROL_ERROR);
                            }
                            break;
                    case 1:
                            if (write(testfd, hoge, bsize) < 0) {
                                    perror("write");
                                    exit(TEST_ERROR);
                            }
                            if (write(controlfd, hoge, bsize) < 0) {
                                    perror("write");
                                    exit(CONTROL_ERROR);
                            }
                            break;
                    case 2:
                            if (ftruncate(testfd, k) < 0) {
                                    perror("write");
                                    exit(TEST_ERROR);
                            }
                            if (ftruncate(controlfd, k) < 0) {
                                    perror("write");
                                    exit(CONTROL_ERROR);
                            }
                            break;
                }
                j = random() % (100 * 1000);
                usleep(j);
        }
        exit(0);
}