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