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.sun.com/software/products/lustre/docs/GPLv2.pdf
  19  *
  20  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  21  * CA 95054 USA or visit www.sun.com if you need additional information or
  22  * have any questions.
  23  *
  24  * GPL HEADER END
  25  */
  26 /*
  27  * Copyright (c) 2007, 2010, Oracle and/or its affiliates. All rights reserved.
  28  * Use is subject to license terms.
  29  */
  30 /*
  31  * This file is part of Lustre, http://www.lustre.org/
  32  * Lustre is a trademark of Sun Microsystems, Inc.
  33  */
  34
  35 /* for O_DIRECT */
  36 #ifndef _GNU_SOURCE
  37 #define _GNU_SOURCE
  38 #endif
  39
  40 #include<stdio.h>
  41 #include<errno.h>
  42 #include<fcntl.h>
  43 #include<sys/types.h>
  44 #include<sys/stat.h>
  45 #include<unistd.h>
  46 #include<signal.h>
  47 #include<stdlib.h>
  48 #include<string.h>
  49
  50 #define BUFFERSIZE 4096
  51
  52 /* This flag controls termination of the main loop. */
  53 volatile sig_atomic_t keep_going = 1;
  54
  55 /* The signal handler just clears the flag and re-enables itself. */
  56 void catch_alarm(int sig)
  57 {
  58         keep_going = 0;
  59         signal(sig, catch_alarm);
  60 }
  61
  62 int main(int argc, char **argv)
  63 {
  64         char *file;
  65         unsigned char buf[BUFFERSIZE];
  66         int fd, i, rc;
  67         unsigned int test_time;
  68         mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH;
  69
  70         if (argc != 3) {
  71                 printf("Invalid number of arguments.\n");
  72                 printf("Usage %s <file> <test_time_in_seconds>\n", argv[0]);
  73                 return -1;
  74         }
  75
  76         file = argv[1];
  77         test_time = atoi(argv[2]);
  78
  79         /* Establish a handler for SIGALRM signals. */
  80         signal(SIGALRM, catch_alarm);
  81
  82         /* Set an alarm to go off in a little while. */
  83         alarm(test_time);
  84
  85         fd = open(file, O_RDWR|O_TRUNC|O_CREAT|O_SYNC|O_LARGEFILE, mode);
  86         if (fd < 0) {
  87                 fprintf(stderr, "Error: Cannot open file named ");
  88                 perror(file);
  89                 return -1;
  90         }
  91
  92         memset(buf, 1, BUFFERSIZE);
  93         while (keep_going) {
  94                 for (i = 0; i < 1024; i++) {
  95                         rc = write(fd, buf, BUFFERSIZE);
  96                         if (rc < 0) {
  97                                 fprintf(stderr, "Error: Write error ");
  98                                 perror(file);
  99                                 exit(-1);
 100                         } else if (rc != BUFFERSIZE) {
 101                                 fprintf(stderr, "Error: Ddidn't write all data \n");
 102                         }
 103                 }
 104
 105                 if (ftruncate(fd, 0) < 0) {
 106                         fprintf(stderr, "Error: Truncate error ");
 107                         perror(file);
 108                         exit(-1);
 109                 }
 110         }
 111
 112         if (close(fd) < 0) {
 113                 fprintf(stderr, "Error: Cannot close ");
 114                 perror(file);
 115                 return -1;
 116         }
 117
 118         return 0;
 119 }