/* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
 * vim:expandtab:shiftwidth=8:tabstop=8:
 *
 * GPL HEADER START
 *
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 only,
 * as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License version 2 for more details (a copy is included
 * in the LICENSE file that accompanied this code).
 *
 * You should have received a copy of the GNU General Public License
 * version 2 along with this program; If not, see
 * http://www.sun.com/software/products/lustre/docs/GPLv2.pdf
 *
 * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
 * CA 95054 USA or visit www.sun.com if you need additional information or
 * have any questions.
 *
 * GPL HEADER END
 */
/*
 * Copyright  2008 Sun Microsystems, Inc. All rights reserved
 * Use is subject to license terms.
 */
/*
 * This file is part of Lustre, http://www.lustre.org/
 * Lustre is a trademark of Sun Microsystems, Inc.
 */

/* for O_DIRECT */
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif

#include<stdio.h>
#include<errno.h>
#include<fcntl.h>
#include<sys/types.h>
#include<sys/stat.h>
#include<unistd.h>
#include<signal.h>
#include<stdlib.h>
#include<string.h>

#define BUFFERSIZE 4096

/* This flag controls termination of the main loop. */
volatile sig_atomic_t keep_going = 1;

/* The signal handler just clears the flag and re-enables itself. */
void catch_alarm(int sig)
{
	keep_going = 0;
	signal(sig, catch_alarm);
}

int main(int argc, char **argv)
{
        char *file;
        unsigned char buf[BUFFERSIZE];
        int fd, i, rc;
	unsigned int test_time;
        mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH;

        if (argc != 3) {
                printf("Invalid number of arguments.\n");
                printf("Usage %s <file> <test_time_in_seconds>\n", argv[0]);
                return -1;
        }

        file = argv[1];
        test_time = atoi(argv[2]);

	/* Establish a handler for SIGALRM signals. */
	signal(SIGALRM, catch_alarm);

       	/* Set an alarm to go off in a little while. */
       	alarm(test_time);

        fd = open(file, O_RDWR|O_TRUNC|O_CREAT|O_SYNC|O_LARGEFILE, mode);
        if (fd < 0) {
                fprintf(stderr, "Error: Cannot open file named ");
                perror(file);
                return -1;
        }

        memset(buf, 1, BUFFERSIZE);
	while (keep_going) {
                for (i = 0; i < 1024; i++) {
		        rc = write(fd, buf, BUFFERSIZE);
			if (rc < 0) {
                                fprintf(stderr, "Error: Write error ");
                                perror(file);
				exit(-1);
                        } else if (rc != BUFFERSIZE) {
                                fprintf(stderr, "Error: Ddidn't write all data \n");
			}
                }

                if (ftruncate(fd, 0) < 0) {
                        fprintf(stderr, "Error: Truncate error ");
                        perror(file);
			exit(-1);
                }
        }

        if (close(fd) < 0) {
                fprintf(stderr, "Error: Cannot close ");
                perror(file);
                return -1;
        }

        return 0;
}