lustre/tests/write_time_limit.c

   1 /*
   2  * GPL HEADER START
   3  *
   4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License version 2 only,
   8  * as published by the Free Software Foundation.
   9  *
  10  * This program is distributed in the hope that it will be useful, but
  11  * WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  13  * General Public License version 2 for more details (a copy is included
  14  * in the LICENSE file that accompanied this code).
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * version 2 along with this program; If not, see
  18  * http://www.gnu.org/licenses/gpl-2.0.html
  19  *
  20  * GPL HEADER END
  21  */
  22 /*
  23  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  */
  26 /*
  27  * This file is part of Lustre, http://www.lustre.org/
  28  * Lustre is a trademark of Sun Microsystems, Inc.
  29  */
  30
  31 /* for O_DIRECT */
  32 #ifndef _GNU_SOURCE
  33 #define _GNU_SOURCE
  34 #endif
  35
  36 #include<stdio.h>
  37 #include<errno.h>
  38 #include<fcntl.h>
  39 #include<sys/types.h>
  40 #include<sys/stat.h>
  41 #include<unistd.h>
  42 #include<signal.h>
  43 #include<stdlib.h>
  44 #include<string.h>
  45
  46 #define BUFFERSIZE 4096
  47
  48 /* This flag controls termination of the main loop. */
  49 volatile sig_atomic_t keep_going = 1;
  50
  51 /* The signal handler just clears the flag and re-enables itself. */
  52 void catch_alarm(int sig)
  53 {
  54         keep_going = 0;
  55         signal(sig, catch_alarm);
  56 }
  57
  58 int main(int argc, char **argv)
  59 {
  60         char *file;
  61         unsigned char buf[BUFFERSIZE];
  62         int fd, i, rc;
  63         unsigned int test_time;
  64
  65         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH;
  66
  67         if (argc != 3) {
  68                 printf("Invalid number of arguments.\n");
  69                 printf("Usage %s <file> <test_time_in_seconds>\n", argv[0]);
  70                 return -1;
  71         }
  72
  73         file = argv[1];
  74         test_time = atoi(argv[2]);
  75
  76         /* Establish a handler for SIGALRM signals. */
  77         signal(SIGALRM, catch_alarm);
  78
  79         /* Set an alarm to go off in a little while. */
  80         alarm(test_time);
  81
  82         fd = open(file, O_RDWR | O_TRUNC | O_CREAT | O_SYNC | O_LARGEFILE,
  83                   mode);
  84         if (fd < 0) {
  85                 fprintf(stderr, "Error: Cannot open file named ");
  86                 perror(file);
  87                 return -1;
  88         }
  89
  90         memset(buf, 1, BUFFERSIZE);
  91         while (keep_going) {
  92                 for (i = 0; i < 1024; i++) {
  93                         rc = write(fd, buf, BUFFERSIZE);
  94                         if (rc < 0) {
  95                                 fprintf(stderr, "Error: Write error ");
  96                                 perror(file);
  97                                 exit(-1);
  98                         } else if (rc != BUFFERSIZE) {
  99                                 fprintf(stderr,
 100                                         "Error: Ddidn't write all data\n");
 101                         }
 102                 }
 103
 104                 if (ftruncate(fd, 0) < 0) {
 105                         fprintf(stderr, "Error: Truncate error ");
 106                         perror(file);
 107                         exit(-1);
 108                 }
 109         }
 110
 111         if (close(fd) < 0) {
 112                 fprintf(stderr, "Error: Cannot close ");
 113                 perror(file);
 114                 return -1;
 115         }
 116
 117         return 0;
 118 }