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         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH;
  65
  66         if (argc != 3) {
  67                 printf("Invalid number of arguments.\n");
  68                 printf("Usage %s <file> <test_time_in_seconds>\n", argv[0]);
  69                 return -1;
  70         }
  71
  72         file = argv[1];
  73         test_time = atoi(argv[2]);
  74
  75         /* Establish a handler for SIGALRM signals. */
  76         signal(SIGALRM, catch_alarm);
  77
  78         /* Set an alarm to go off in a little while. */
  79         alarm(test_time);
  80
  81         fd = open(file, O_RDWR|O_TRUNC|O_CREAT|O_SYNC|O_LARGEFILE, mode);
  82         if (fd < 0) {
  83                 fprintf(stderr, "Error: Cannot open file named ");
  84                 perror(file);
  85                 return -1;
  86         }
  87
  88         memset(buf, 1, BUFFERSIZE);
  89         while (keep_going) {
  90                 for (i = 0; i < 1024; i++) {
  91                         rc = write(fd, buf, BUFFERSIZE);
  92                         if (rc < 0) {
  93                                 fprintf(stderr, "Error: Write error ");
  94                                 perror(file);
  95                                 exit(-1);
  96                         } else if (rc != BUFFERSIZE) {
  97                                 fprintf(stderr, "Error: Ddidn't write all data \n");
  98                         }
  99                 }
 100
 101                 if (ftruncate(fd, 0) < 0) {
 102                         fprintf(stderr, "Error: Truncate error ");
 103                         perror(file);
 104                         exit(-1);
 105                 }
 106         }
 107
 108         if (close(fd) < 0) {
 109                 fprintf(stderr, "Error: Cannot close ");
 110                 perror(file);
 111                 return -1;
 112         }
 113
 114         return 0;
 115 }