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  */
  29
  30 /* for O_DIRECT */
  31 #ifndef _GNU_SOURCE
  32 #define _GNU_SOURCE
  33 #endif
  34
  35 #include<stdio.h>
  36 #include<errno.h>
  37 #include<fcntl.h>
  38 #include<sys/types.h>
  39 #include<sys/stat.h>
  40 #include<unistd.h>
  41 #include<signal.h>
  42 #include<stdlib.h>
  43 #include<string.h>
  44
  45 #define BUFFERSIZE 4096
  46
  47 /* This flag controls termination of the main loop. */
  48 volatile sig_atomic_t keep_going = 1;
  49
  50 /* The signal handler just clears the flag and re-enables itself. */
  51 void catch_alarm(int sig)
  52 {
  53         keep_going = 0;
  54         signal(sig, catch_alarm);
  55 }
  56
  57 int main(int argc, char **argv)
  58 {
  59         char *file;
  60         unsigned char buf[BUFFERSIZE];
  61         int fd, i, rc;
  62         unsigned int test_time;
  63
  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,
  82                   mode);
  83         if (fd < 0) {
  84                 fprintf(stderr, "Error: Cannot open file named ");
  85                 perror(file);
  86                 return -1;
  87         }
  88
  89         memset(buf, 1, BUFFERSIZE);
  90         while (keep_going) {
  91                 for (i = 0; i < 1024; i++) {
  92                         rc = write(fd, buf, BUFFERSIZE);
  93                         if (rc < 0) {
  94                                 fprintf(stderr, "Error: Write error ");
  95                                 perror(file);
  96                                 exit(-1);
  97                         } else if (rc != BUFFERSIZE) {
  98                                 fprintf(stderr,
  99                                         "Error: Ddidn't write all data\n");
 100                         }
 101                 }
 102
 103                 if (ftruncate(fd, 0) < 0) {
 104                         fprintf(stderr, "Error: Truncate error ");
 105                         perror(file);
 106                         exit(-1);
 107                 }
 108         }
 109
 110         if (close(fd) < 0) {
 111                 fprintf(stderr, "Error: Cannot close ");
 112                 perror(file);
 113                 return -1;
 114         }
 115
 116         return 0;
 117 }