lustre/tests/iopentest2.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  */
   4
   5 #include <sys/types.h>
   6 #include <sys/stat.h>
   7 #include <fcntl.h>
   8 #include <stdio.h>
   9 #include <unistd.h>
  10 #include <limits.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <libgen.h>
  14 #include <errno.h>
  15
  16 const char *progname;
  17 const char usage_fmt[] = "Usage: %s <mountpoint>\n";
  18 #define INAME_LEN (PATH_MAX + 1)
  19
  20 #define CHECK_IT(exp, pstr) \
  21 if (!(exp)) { \
  22     fprintf(stderr, "%s: at %s:%d: ", progname, __FILE__, __LINE__); \
  23     perror((pstr)); \
  24     exit(1); \
  25 }
  26
  27 #define CHECK_SNPRINTF(rc, len) \
  28     CHECK_IT((rc) > 0 && (rc) <= (len), "snprintf() failed")
  29
  30 static char *get_iname(char *fname, const char *mtpt)
  31 {
  32         char *iname;
  33         int fd, rc;
  34         struct stat buf;
  35
  36         iname = malloc(INAME_LEN);
  37         CHECK_IT(iname, "malloc() failed");
  38
  39         fd = open(fname, O_CREAT, 0644);
  40         if (fd < 0 && errno != EISDIR) {
  41                 fprintf(stderr, "%s:%d: open(%s) failed: %s\n", __FILE__,
  42                         __LINE__, fname, strerror(errno));
  43                 exit(1);
  44         }
  45
  46         if (fd >= 0)
  47                 close(fd);
  48
  49         rc = stat(fname, &buf);
  50         if (rc != 0) {
  51                 fprintf(stderr, "%s:%d: stat(%s) failed: %s\n", __FILE__,
  52                         __LINE__, fname, strerror(errno));
  53                 exit(1);
  54         }
  55
  56         rc = snprintf(iname, INAME_LEN,
  57                       "%s/__iopen__/%lu", mtpt, (unsigned long)buf.st_ino);
  58         CHECK_SNPRINTF(rc, INAME_LEN);
  59
  60         return iname;
  61 }
  62
  63 int main(int argc, char *argv[])
  64 {
  65         char *fname, *mtpt, *pname;
  66         char *fname_iname, *dir;
  67         char *dir_iname = NULL, *foo = NULL, *bar = NULL;
  68         int rc, fd, i, thread = 0;
  69
  70         pname = strdup(argv[0]);
  71         progname = basename(argv[0]);
  72
  73         if (argc != 2) {
  74                 fprintf(stderr, usage_fmt, progname);
  75                 return 1;
  76         }
  77
  78         for (i = 1; i <= 10; i++) {
  79                 rc = fork();
  80                 if (rc < 0) {
  81                         fprintf(stderr, "error: %s: #%d - %s\n", argv[0], i,
  82                                 strerror(rc = errno));
  83                         break;
  84                 } else if (rc == 0) {
  85                         thread = i;
  86                         break;
  87                 }
  88                 printf("%s: thread #%d (PID %d) started\n", argv[0], i, rc);
  89                 rc = 0;
  90         }
  91
  92         if (thread != 0) {
  93                 mtpt  = argv[1];
  94                 fname = malloc(INAME_LEN);
  95                 CHECK_IT(fname, "malloc() failed");
  96
  97                 rc = snprintf(fname, INAME_LEN,
  98                               "%s/%d", mtpt, getpid());
  99                 CHECK_SNPRINTF(rc, INAME_LEN);
 100
 101                 rc = mkdir(fname, 0644);
 102                 if (rc != 0) {
 103                         fprintf(stderr, "%s:%d: mkdir(%s) failed: %s\n",
 104                                 __FILE__, __LINE__, fname, strerror(errno));
 105                         exit(1);
 106                 }
 107
 108                 fname_iname = get_iname(fname, mtpt);
 109
 110                 dir = malloc(INAME_LEN);
 111                 CHECK_IT(dir, "malloc() failed");
 112
 113                 rc = snprintf(dir, INAME_LEN, "%s/dir", fname_iname);
 114                 CHECK_SNPRINTF(rc, INAME_LEN);
 115
 116                 foo = malloc(INAME_LEN);
 117                 CHECK_IT(foo, "malloc() failed");
 118
 119                 bar = malloc(INAME_LEN);
 120                 CHECK_IT(bar, "malloc() failed");
 121
 122                 for (i = 0; i < 1000; i++) {
 123                         rc = mkdir(dir, 0644);
 124                         if (rc != 0) {
 125                                 fprintf(stderr, "%s:%d: mkdir(%s) failed: %s\n",
 126                                         __FILE__, __LINE__, dir,
 127                                         strerror(errno));
 128                                 exit(1);
 129                         }
 130
 131                         dir_iname = get_iname(dir, mtpt);
 132
 133                         rc = snprintf(foo, INAME_LEN, "%s/bar", dir_iname);
 134                         CHECK_SNPRINTF(rc, INAME_LEN);
 135
 136                         rc = snprintf(bar, INAME_LEN, "%s/bar", dir_iname);
 137                         CHECK_SNPRINTF(rc, INAME_LEN);
 138
 139                         fd = open(foo, O_CREAT, 0644);
 140                         if (fd < 0) {
 141                                 fprintf(stderr, "%s:%d: open(%s) failed: %s\n",
 142                                         __FILE__, __LINE__, foo,
 143                                         strerror(errno));
 144                                 exit(1);
 145                         }
 146                         rc = close(fd);
 147                         if (rc != 0) {
 148                                 fprintf(stderr, "%s:%d: close() failed: %s\n",
 149                                         __FILE__, __LINE__, strerror(errno));
 150                                 exit(1);
 151                         }
 152
 153                         rc = rename(foo, bar);
 154                         if (rc != 0) {
 155                                 fprintf(stderr, "%s:%d: rename(%s, %s) failed: "
 156                                         "%s\n", __FILE__, __LINE__, foo, bar,
 157                                         strerror(errno));
 158                                 exit(1);
 159                         }
 160
 161                         rc = unlink(bar);
 162                         if (rc != 0) {
 163                                 fprintf(stderr, "%s:%d: unlink(%s) failed: "
 164                                         "%s\n", __FILE__, __LINE__, bar,
 165                                         strerror(errno));
 166                                 exit(1);
 167                         }
 168                         rc = rmdir(dir);
 169                         if (rc != 0) {
 170                                 fprintf(stderr, "%s:%d: rmdir(%s) failed: %s\n",
 171                                         __FILE__, __LINE__, dir,
 172                                         strerror(errno));
 173                                 exit(1);
 174                         }
 175
 176                         free(dir_iname);
 177                 }
 178         }
 179         return 0;
 180 }