lustre/tests/writeme.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) 2002, 2010, Oracle and/or its affiliates. All rights reserved.
  24  * Use is subject to license terms.
  25  *
  26  * Copyright (c) 2011, Intel Corporation.
  27  */
  28 /*
  29  * This file is part of Lustre, http://www.lustre.org/
  30  * Lustre is a trademark of Sun Microsystems, Inc.
  31  */
  32
  33 #include <fcntl.h>
  34 #include <unistd.h>
  35 #include <stdlib.h>
  36 #include <stdio.h>
  37 #include <string.h>
  38 #include <stdbool.h>
  39
  40 void usage(char *prog)
  41 {
  42         printf("usage: %s [-s] [-b <bytes>] filename\n", prog);
  43         exit(1);
  44 }
  45
  46 int main(int argc, char **argv)
  47 {
  48         bool limit_write = false, do_sync = false;
  49         int c, per_write, fd, rc;
  50         unsigned long bytes = 0;
  51         char buf[4096];
  52         char *endptr = NULL;
  53
  54         while ((c = getopt(argc, argv, "sb:")) != -1) {
  55                 switch (c) {
  56                 case 's':
  57                         do_sync = true;
  58                         break;
  59                 case 'b':
  60                         limit_write = true;
  61                         bytes = strtoul(optarg, &endptr, 10);
  62                         if (endptr != NULL && *endptr != '\0')
  63                                 usage(argv[0]);
  64                         break;
  65                 default:
  66                         usage(argv[0]);
  67                         break;
  68                 }
  69         }
  70
  71         if (argc - optind != 1)
  72                 usage(argv[0]);
  73
  74         memset(buf, 0, 4096);
  75         fd = open(argv[optind], O_RDWR | O_CREAT, 0600);
  76         if (fd == -1) {
  77                 printf("Error opening %s\n", argv[1]);
  78                 exit(1);
  79         }
  80
  81         /* Even 0 bytes, write at least once */
  82         if (limit_write) {
  83                 do {
  84                         per_write = bytes > 4096 ? 4096 : bytes;
  85                         rc = write(fd, buf, per_write);
  86                         if (rc > 0)
  87                                 bytes -= rc;
  88                         else if (rc < 0)
  89                                 break;
  90                 } while (bytes > 0);
  91
  92                 return rc >= 0 ? 0 : rc;
  93         }
  94
  95         for (rc = 0; ;) {
  96                 sprintf(buf, "write %d\n", rc);
  97                 rc = write(fd, buf, sizeof(buf));
  98                 if (do_sync)
  99                         sync();
 100                 sleep(1);
 101         }
 102
 103         return 0;
 104 }