lustre/tests/openfile.c

   1 /* -*- mode: c; c-basic-offset: 8; indent-tabs-mode: nil; -*-
   2  * vim:expandtab:shiftwidth=8:tabstop=8:
   3  */
   4
   5 #if 0
   6 #define DEBUG
   7 #endif
   8
   9 /* for O_DIRECTORY and O_DIRECT */
  10 #ifndef _GNU_SOURCE
  11 #define _GNU_SOURCE
  12 #endif
  13
  14 #include <stdio.h>
  15 #include <sys/types.h>
  16 #include <sys/stat.h>
  17 #include <fcntl.h>
  18 #include <errno.h>
  19 #include <string.h>
  20 #include <stdlib.h>
  21 #include <unistd.h>
  22 #include <lustre/lustre_user.h>
  23
  24 typedef struct flag_mapping {
  25        const char *string;
  26        const int  flag;
  27 } FLAG_MAPPING;
  28
  29 FLAG_MAPPING flag_table[] = {
  30        {"O_RDONLY", O_RDONLY},
  31        {"O_WRONLY", O_WRONLY},
  32        {"O_RDWR", O_RDWR},
  33        {"O_CREAT", O_CREAT},
  34        {"O_EXCL", O_EXCL},
  35        {"O_NOCTTY", O_NOCTTY},
  36        {"O_TRUNC", O_TRUNC},
  37        {"O_APPEND", O_APPEND},
  38        {"O_NONBLOCK", O_NONBLOCK},
  39        {"O_NDELAY", O_NDELAY},
  40        {"O_SYNC", O_SYNC},
  41 #ifdef O_DIRECT
  42        {"O_DIRECT", O_DIRECT},
  43 #endif
  44        {"O_LARGEFILE", O_LARGEFILE},
  45        {"O_DIRECTORY", O_DIRECTORY},
  46        {"O_NOFOLLOW", O_NOFOLLOW},
  47        {"O_LOV_DELAY_CREATE", O_LOV_DELAY_CREATE},
  48        {"", -1}
  49 };
  50
  51 void Usage_and_abort(void)
  52 {
  53        fprintf(stderr, "Usage: openfile -f flags [ -m mode ] filename \n");
  54        fprintf(stderr, "e.g. openfile -f O_RDWR:O_CREAT -m 0755 /etc/passwd\n");
  55        exit(-1);
  56 }
  57
  58 int main(int argc, char** argv)
  59 {
  60         int    fd;
  61         int    flags=0;
  62         mode_t mode=0644;
  63         char*  fname=NULL;
  64         int    mode_set=0;
  65         int    flag_set=0;
  66         int    c;
  67         int    save_errno;
  68         int    print_usage=0;
  69         char*  cloned_flags = NULL;
  70
  71         if (argc == 1)
  72                 Usage_and_abort();
  73
  74         while ((c = getopt (argc, argv, "f:m:")) != -1) {
  75                 switch (c) {
  76                 case 'f': {
  77                         char *tmp;
  78
  79                         cloned_flags = (char *)malloc(strlen(optarg)+1);
  80                         if (cloned_flags == NULL) {
  81                                 fprintf(stderr, "Insufficient memory.\n");
  82                                 save_errno = -1;
  83                                 goto out;
  84                         }
  85
  86                         strncpy(cloned_flags, optarg, strlen(optarg)+1);
  87                         flags = atoi(cloned_flags);
  88                         if (flags > 0) {
  89                                 flag_set = 1;
  90 #ifdef DEBUG
  91                                 printf("flags = %d\n",flags);
  92 #endif
  93                                 break;
  94                         } else
  95                                 flags = 0;
  96
  97                         for (tmp = strtok(cloned_flags, ":|"); tmp;
  98                              tmp = strtok(NULL, ":|")) {
  99                                 int i = 0;
 100 #ifdef DEBUG
 101                                 printf("flags = %s\n",tmp);
 102 #endif
 103                                 flag_set = 1;
 104                                 for (i = 0; flag_table[i].flag != -1; i++) {
 105                                         if (!strcmp(tmp, flag_table[i].string)){
 106                                                 flags |= flag_table[i].flag;
 107                                                 break;
 108                                         }
 109                                 }
 110
 111                                 if (flag_table[i].flag == -1) {
 112                                         fprintf(stderr, "No such flag: %s\n",
 113                                                 tmp);
 114                                         save_errno = -1;
 115                                         goto out;
 116                                 }
 117                         }
 118 #ifdef DEBUG
 119                         printf("flags = %x\n", flags);
 120 #endif
 121                         break;
 122                 }
 123                 case 'm':
 124 #ifdef DEBUG
 125                         printf("mode = %s\n", optarg);
 126 #endif
 127                         mode = strtol(optarg, NULL, 8);
 128                         mode_set = 1;
 129 #ifdef DEBUG
 130                         printf("mode = %o\n", mode);
 131 #endif
 132                         break;
 133                 default:
 134                         fprintf(stderr, "Bad parameters.\n");
 135                         print_usage = 1;
 136                         goto out;
 137                 }
 138         }
 139
 140         if (optind == argc) {
 141                 fprintf(stderr, "Bad parameters.\n");
 142                 print_usage = 1;
 143                 goto out;
 144         }
 145
 146         fname = argv[optind];
 147
 148         if (!flag_set) {
 149                 fprintf(stderr, "Missing flag or file-name\n");
 150                 save_errno = -1;
 151                 goto out;
 152         }
 153
 154
 155         if (flags & O_CREAT)
 156                 fd = open(fname, flags, mode);
 157         else
 158                 fd = open(fname, flags);
 159
 160         save_errno = errno;
 161
 162         if (fd != -1) {
 163                 printf("Succeed in opening file \"%s\"(flags=%s",
 164                        fname, cloned_flags);
 165
 166                 if (mode_set)
 167                         printf(", mode=%o", mode);
 168                 printf(")\n");
 169                 close(fd);
 170         } else {
 171                 fprintf(stderr, "Error in opening file \"%s\"(flags=%s",
 172                         fname, cloned_flags);
 173         }
 174
 175         if (mode_set)
 176                 fprintf(stderr, ", mode=%o", mode);
 177         fprintf(stderr, ") %d: %s\n", save_errno, strerror(save_errno));
 178
 179 out:
 180         if (cloned_flags)
 181                 free(cloned_flags);
 182         if (print_usage)
 183                 Usage_and_abort();
 184
 185         return save_errno;
 186 }
 187