scripts/mkflags.c

   1 // Take three word input lines on stdin (the three space separated words are
   2 // command name, option string with current config, option string from
   3 // allyesconfig; space separated, the last two are and double quotes)
   4 // and produce flag #defines to stdout.
   5
   6 // This is intentionally crappy code because we control the inputs. It leaks
   7 // memory like a sieve and segfaults if malloc returns null, but does the job.
   8
   9 #include <unistd.h>
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13 #include <errno.h>
  14 #include <ctype.h>
  15
  16 struct flag {
  17   struct flag *next;
  18   char *command;
  19   struct flag *lopt;
  20 };
  21
  22 // replace chopped out USE_BLAH() sections with low-ascii characters
  23 // showing how many flags got skipped
  24
  25 char *mark_gaps(char *flags, char *all)
  26 {
  27   char *n, *new, c;
  28
  29   // Shell feeds in " " for blank args, leading space not meaningful.
  30   while (isspace(*flags)) flags++;
  31   while (isspace(*all)) all++;
  32
  33   n = new = strdup(all);
  34   while (*all) {
  35     if (*flags == *all) {
  36       *(new++) = *(all++);
  37       *flags++;
  38       continue;
  39     }
  40
  41     c = *(all++);
  42     if (strchr("?&^-:#|@*; ", c));
  43     else if (strchr("=<>", c)) while (isdigit(*all)) all++;
  44     else if (c == '(') while(*(all++) != ')');
  45     else *(new++) = 1;
  46   }
  47   *new = 0;
  48
  49   return n;
  50 }
  51
  52 // Break down a command string into struct flag list.
  53
  54 struct flag *digest(char *string)
  55 {
  56   struct flag *list = NULL;
  57   char *err = string;
  58
  59   while (*string) {
  60     // Groups must be at end.
  61     if (*string == '[') break;
  62
  63     // Longopts
  64     if (*string == '(') {
  65       struct flag *new = calloc(sizeof(struct flag), 1);
  66
  67       new->command = ++string;
  68
  69       // Attach longopt to previous short opt, if any.
  70       if (list && list->command) {
  71         new->next = list->lopt;
  72         list->lopt = new;
  73       } else {
  74         struct flag *blank = calloc(sizeof(struct flag), 1);
  75
  76         blank->next = list;
  77         blank->lopt = new;
  78         list = blank;
  79       }
  80       // An empty longopt () would break this.
  81       while (*++string != ')') if (*string == '-') *string = '_';
  82       *(string++) = 0;
  83       continue;
  84     }
  85
  86     if (strchr("?&^-:#|@*; ", *string)) string++;
  87     else if (strchr("=<>", *string)) {
  88       if (!isdigit(string[1])) {
  89         fprintf(stderr, "%c without number in '%s'", *string, err);
  90         exit(1);
  91       }
  92       while (isdigit(*++string)) {
  93         if (!list) {
  94            string++;
  95            break;
  96         }
  97       }
  98     } else {
  99       struct flag *new = calloc(sizeof(struct flag), 1);
 100
 101       new->command = string++;
 102       new->next = list;
 103       list = new;
 104     }
 105   }
 106
 107   return list;
 108 }
 109
 110 int main(int argc, char *argv[])
 111 {
 112   char command[256], flags[1023], allflags[1024];
 113   char *out, *outbuf = malloc(1024*1024);
 114
 115   // Yes, the output buffer is 1 megabyte with no bounds checking.
 116   // See "intentionally crappy", above.
 117   if (!(out = outbuf)) return 1;
 118
 119   printf("#ifdef FORCE_FLAGS\n#define FORCED_FLAG 1\n"
 120          "#else\n#define FORCED_FLAG 0\n#endif\n\n");
 121
 122   for (;;) {
 123     struct flag *flist, *aflist, *offlist;
 124     char *gaps, *mgaps, c;
 125     unsigned bit;
 126
 127     *command = *flags = *allflags = 0;
 128     bit = fscanf(stdin, "%255s \"%1023[^\"]\" \"%1023[^\"]\"\n",
 129                     command, flags, allflags);
 130
 131     if (getenv("DEBUG"))
 132       fprintf(stderr, "command=%s, flags=%s, allflags=%s\n",
 133         command, flags, allflags);
 134
 135     if (!*command) break;
 136     if (bit != 3) {
 137       fprintf(stderr, "\nError in %s (duplicate command?)\n", command);
 138       exit(1);
 139     }
 140
 141     bit = 0;
 142     printf("// %s %s %s\n", command, flags, allflags);
 143     mgaps = mark_gaps(flags, allflags);
 144     for (gaps = mgaps; *gaps == 1; gaps++);
 145     if (*gaps) c = '"';
 146     else {
 147       c = ' ';
 148       gaps = "0";
 149     }
 150     printf("#undef OPTSTR_%s\n#define OPTSTR_%s %c%s%c\n",
 151             command, command, c, gaps, c);
 152     free(mgaps);
 153
 154     flist = digest(flags);
 155     offlist = aflist = digest(allflags);
 156
 157     printf("#ifdef CLEANUP_%s\n#undef CLEANUP_%s\n#undef FOR_%s\n",
 158            command, command, command);
 159
 160     while (offlist) {
 161       struct flag *f = offlist->lopt;
 162       while (f) {
 163         printf("#undef FLAG_%s\n", f->command);
 164         f = f->next;
 165       }
 166       if (offlist->command) printf("#undef FLAG_%c\n", *offlist->command);
 167       offlist = offlist->next;
 168     }
 169     printf("#endif\n\n");
 170
 171     sprintf(out, "#ifdef FOR_%s\n#ifndef TT\n#define TT this.%s\n#endif\n",
 172             command, command);
 173     out += strlen(out);
 174
 175     while (aflist) {
 176       if (aflist->lopt) {
 177         if (flist && flist->lopt &&
 178             !strcmp(flist->lopt->command, aflist->lopt->command))
 179         {
 180           sprintf(out, "#define FLAG_%s (1<<%d)\n", flist->lopt->command, bit);
 181           flist->lopt = flist->lopt->next;
 182         } else sprintf(out, "#define FLAG_%s (FORCED_FLAG<<%d)\n",
 183                        aflist->lopt->command, bit);
 184         aflist->lopt = aflist->lopt->next;
 185         if (!aflist->command) {
 186           aflist = aflist->next;
 187           bit++;
 188           if (flist) flist = flist->next;
 189         }
 190       } else if (aflist->command) {
 191         if (flist && (!flist->command || *aflist->command == *flist->command)) {
 192           if (aflist->command)
 193             sprintf(out, "#define FLAG_%c (1<<%d)\n", *aflist->command, bit);
 194           flist = flist->next;
 195         } else sprintf(out, "#define FLAG_%c (FORCED_FLAG<<%d)\n",
 196                        *aflist->command, bit);
 197         bit++;
 198         aflist = aflist->next;
 199       }
 200       out += strlen(out);
 201     }
 202     out = stpcpy(out, "#endif\n\n");
 203   }
 204
 205   if (fflush(0) && ferror(stdout)) return 1;
 206
 207   out = outbuf;
 208   while (*out) {
 209     int i = write(1, outbuf, strlen(outbuf));
 210
 211     if (i<0) return 1;
 212     out += i;
 213   }
 214
 215   return 0;
 216 }