isl_arg.c

   1 /*
   2  * Copyright 2008-2009 Katholieke Universiteit Leuven
   3  *
   4  * Use of this software is governed by the GNU LGPLv2.1 license
   5  *
   6  * Written by Sven Verdoolaege, K.U.Leuven, Departement
   7  * Computerwetenschappen, Celestijnenlaan 200A, B-3001 Leuven, Belgium
   8  */
   9
  10 #include <stdio.h>
  11 #include <stdlib.h>
  12 #include <string.h>
  13
  14 #include <isl/arg.h>
  15 #include <isl/ctx.h>
  16
  17 static void set_default_choice(struct isl_arg *arg, void *opt)
  18 {
  19         *(unsigned *)(((char *)opt) + arg->offset) = arg->u.choice.default_value;
  20 }
  21
  22 static void set_default_flags(struct isl_arg *arg, void *opt)
  23 {
  24         *(unsigned *)(((char *)opt) + arg->offset) = arg->u.flags.default_value;
  25 }
  26
  27 static void set_default_bool(struct isl_arg *arg, void *opt)
  28 {
  29         *(unsigned *)(((char *)opt) + arg->offset) = arg->u.b.default_value;
  30 }
  31
  32 static void set_default_child(struct isl_arg *arg, void *opt)
  33 {
  34         void *child = calloc(1, arg->u.child.size);
  35
  36         if (child)
  37                 isl_arg_set_defaults(arg->u.child.child, child);
  38
  39         *(void **)(((char *)opt) + arg->offset) = child;
  40 }
  41
  42 static void set_default_user(struct isl_arg *arg, void *opt)
  43 {
  44         arg->u.user.init(((char *)opt) + arg->offset);
  45 }
  46
  47 static void set_default_long(struct isl_arg *arg, void *opt)
  48 {
  49         *(long *)(((char *)opt) + arg->offset) = arg->u.l.default_value;
  50 }
  51
  52 static void set_default_ulong(struct isl_arg *arg, void *opt)
  53 {
  54         *(unsigned long *)(((char *)opt) + arg->offset) = arg->u.ul.default_value;
  55 }
  56
  57 static void set_default_str(struct isl_arg *arg, void *opt)
  58 {
  59         const char *str = NULL;
  60         if (arg->u.str.default_value)
  61                 str = strdup(arg->u.str.default_value);
  62         *(const char **)(((char *)opt) + arg->offset) = str;
  63 }
  64
  65 void isl_arg_set_defaults(struct isl_arg *arg, void *opt)
  66 {
  67         int i;
  68
  69         for (i = 0; arg[i].type != isl_arg_end; ++i) {
  70                 switch (arg[i].type) {
  71                 case isl_arg_choice:
  72                         set_default_choice(&arg[i], opt);
  73                         break;
  74                 case isl_arg_flags:
  75                         set_default_flags(&arg[i], opt);
  76                         break;
  77                 case isl_arg_bool:
  78                         set_default_bool(&arg[i], opt);
  79                         break;
  80                 case isl_arg_child:
  81                         set_default_child(&arg[i], opt);
  82                         break;
  83                 case isl_arg_user:
  84                         set_default_user(&arg[i], opt);
  85                         break;
  86                 case isl_arg_long:
  87                         set_default_long(&arg[i], opt);
  88                         break;
  89                 case isl_arg_ulong:
  90                         set_default_ulong(&arg[i], opt);
  91                         break;
  92                 case isl_arg_arg:
  93                 case isl_arg_str:
  94                         set_default_str(&arg[i], opt);
  95                         break;
  96                 case isl_arg_version:
  97                 case isl_arg_end:
  98                         break;
  99                 }
 100         }
 101 }
 102
 103 void isl_arg_free(struct isl_arg *arg, void *opt)
 104 {
 105         int i;
 106
 107         if (!opt)
 108                 return;
 109
 110         for (i = 0; arg[i].type != isl_arg_end; ++i) {
 111                 switch (arg[i].type) {
 112                 case isl_arg_child:
 113                         isl_arg_free(arg[i].u.child.child,
 114                                 *(void **)(((char *)opt) + arg[i].offset));
 115                         break;
 116                 case isl_arg_arg:
 117                 case isl_arg_str:
 118                         free(*(char **)(((char *)opt) + arg[i].offset));
 119                         break;
 120                 case isl_arg_user:
 121                         if (arg[i].u.user.clear)
 122                                 arg[i].u.user.clear(((char *)opt) + arg[i].offset);
 123                         break;
 124                 case isl_arg_bool:
 125                 case isl_arg_choice:
 126                 case isl_arg_flags:
 127                 case isl_arg_long:
 128                 case isl_arg_ulong:
 129                 case isl_arg_version:
 130                 case isl_arg_end:
 131                         break;
 132                 }
 133         }
 134
 135         free(opt);
 136 }
 137
 138 static int print_arg_help(struct isl_arg *decl, const char *prefix, int no)
 139 {
 140         int len = 0;
 141
 142         if (!decl->long_name) {
 143                 printf("  -%c", decl->short_name);
 144                 return 4;
 145         }
 146
 147         if (decl->short_name)
 148                 printf("  -%c, --", decl->short_name);
 149         else
 150                 printf("      --");
 151         len += 8;
 152
 153         if (prefix) {
 154                 printf("%s-", prefix);
 155                 len += strlen(prefix) + 1;
 156         }
 157         if (no) {
 158                 printf("no-");
 159                 len += 3;
 160         }
 161         printf("%s", decl->long_name);
 162         len += strlen(decl->long_name);
 163
 164         return len;
 165 }
 166
 167 const void *isl_memrchr(const void *s, int c, size_t n)
 168 {
 169         const char *p = s;
 170         while (n-- > 0)
 171                 if (p[n] == c)
 172                         return p + n;
 173         return NULL;
 174 }
 175
 176 static int print_help_msg(struct isl_arg *decl, int pos)
 177 {
 178         int len;
 179         const char *s;
 180
 181         if (!decl->help_msg)
 182                 return pos;
 183
 184         if (pos >= 29)
 185                 printf("\n%30s", "");
 186         else
 187                 printf("%*s", 30 - pos, "");
 188
 189         s = decl->help_msg;
 190         len = strlen(s);
 191         while (len > 45) {
 192                 const char *space = isl_memrchr(s, ' ', 45);
 193                 int l;
 194
 195                 if (!space)
 196                         space = strchr(s + 45, ' ');
 197                 if (!space)
 198                         break;
 199                 l = space - s;
 200                 printf("%.*s", l, s);
 201                 s = space + 1;
 202                 len -= l + 1;
 203                 printf("\n%30s", "");
 204         }
 205
 206         printf("%s", s);
 207         return len;
 208 }
 209
 210 static void print_default(struct isl_arg *decl, const char *def, int pos)
 211 {
 212         int i;
 213         const char *default_prefix = "[default: ";
 214         const char *default_suffix = "]";
 215         int len;
 216
 217         len = strlen(default_prefix) + strlen(def) + strlen(default_suffix);
 218
 219         if (!decl->help_msg) {
 220                 if (pos >= 29)
 221                         printf("\n%30s", "");
 222                 else
 223                         printf("%*s", 30 - pos, "");
 224                 pos = 0;
 225         } else {
 226                 if (pos + len >= 48)
 227                         printf("\n%30s", "");
 228                 else
 229                         printf(" ");
 230         }
 231         printf("%s%s%s", default_prefix, def, default_suffix);
 232 }
 233
 234 static void print_default_choice(struct isl_arg *decl, int pos)
 235 {
 236         int i;
 237         const char *s = "none";
 238
 239         for (i = 0; decl->u.choice.choice[i].name; ++i)
 240                 if (decl->u.choice.choice[i].value == decl->u.choice.default_value) {
 241                         s = decl->u.choice.choice[i].name;
 242                         break;
 243                 }
 244
 245         print_default(decl, s, pos);
 246 }
 247
 248 static void print_choice_help(struct isl_arg *decl, const char *prefix)
 249 {
 250         int i;
 251         int pos;
 252
 253         pos = print_arg_help(decl, prefix, 0);
 254         printf("=");
 255         pos++;
 256
 257         for (i = 0; decl->u.choice.choice[i].name; ++i) {
 258                 if (i) {
 259                         printf("|");
 260                         pos++;
 261                 }
 262                 printf("%s", decl->u.choice.choice[i].name);
 263                 pos += strlen(decl->u.choice.choice[i].name);
 264         }
 265
 266         pos = print_help_msg(decl, pos);
 267         print_default_choice(decl, pos);
 268
 269         printf("\n");
 270 }
 271
 272 static void print_default_flags(struct isl_arg *decl, int pos)
 273 {
 274         int i, first;
 275         const char *default_prefix = "[default: ";
 276         const char *default_suffix = "]";
 277         int len = strlen(default_prefix) + strlen(default_suffix);
 278
 279         for (i = 0; decl->u.flags.flags[i].name; ++i)
 280                 if ((decl->u.flags.default_value & decl->u.flags.flags[i].mask) ==
 281                      decl->u.flags.flags[i].value)
 282                         len += strlen(decl->u.flags.flags[i].name);
 283
 284         if (!decl->help_msg) {
 285                 if (pos >= 29)
 286                         printf("\n%30s", "");
 287                 else
 288                         printf("%*s", 30 - pos, "");
 289                 pos = 0;
 290         } else {
 291                 if (pos + len >= 48)
 292                         printf("\n%30s", "");
 293                 else
 294                         printf(" ");
 295         }
 296         printf("%s", default_prefix);
 297
 298         for (first = 1, i = 0; decl->u.flags.flags[i].name; ++i)
 299                 if ((decl->u.flags.default_value & decl->u.flags.flags[i].mask) ==
 300                      decl->u.flags.flags[i].value) {
 301                         if (!first)
 302                                 printf(",");
 303                         printf("%s", decl->u.flags.flags[i].name);
 304                         first = 0;
 305                 }
 306
 307         printf("%s", default_suffix);
 308 }
 309
 310 static void print_flags_help(struct isl_arg *decl, const char *prefix)
 311 {
 312         int i, j;
 313         int pos;
 314
 315         pos = print_arg_help(decl, prefix, 0);
 316         printf("=");
 317         pos++;
 318
 319         for (i = 0; decl->u.flags.flags[i].name; ++i) {
 320                 if (i) {
 321                         printf(",");
 322                         pos++;
 323                 }
 324                 for (j = i;
 325                      decl->u.flags.flags[j].mask == decl->u.flags.flags[i].mask;
 326                      ++j) {
 327                         if (j != i) {
 328                                 printf("|");
 329                                 pos++;
 330                         }
 331                         printf("%s", decl->u.flags.flags[j].name);
 332                         pos += strlen(decl->u.flags.flags[j].name);
 333                 }
 334                 i = j - 1;
 335         }
 336
 337         pos = print_help_msg(decl, pos);
 338         print_default_flags(decl, pos);
 339
 340         printf("\n");
 341 }
 342
 343 static void print_bool_help(struct isl_arg *decl, const char *prefix)
 344 {
 345         int pos;
 346         int no = decl->u.b.default_value == 1;
 347         pos = print_arg_help(decl, prefix, no);
 348         pos = print_help_msg(decl, pos);
 349         print_default(decl, no ? "yes" : "no", pos);
 350         printf("\n");
 351 }
 352
 353 static int print_argument_name(struct isl_arg *decl, const char *name, int pos)
 354 {
 355         printf("%c<%s>", decl->long_name ? '=' : ' ', name);
 356         return pos + 3 + strlen(name);
 357 }
 358
 359 static void print_long_help(struct isl_arg *decl, const char *prefix)
 360 {
 361         int pos;
 362         pos = print_arg_help(decl, prefix, 0);
 363         if (decl->u.l.default_value != decl->u.l.default_selected) {
 364                 printf("[");
 365                 pos++;
 366         }
 367         printf("=long");
 368         pos += 5;
 369         if (decl->u.l.default_value != decl->u.l.default_selected) {
 370                 printf("]");
 371                 pos++;
 372         }
 373         print_help_msg(decl, pos);
 374         printf("\n");
 375 }
 376
 377 static void print_ulong_help(struct isl_arg *decl, const char *prefix)
 378 {
 379         int pos;
 380         pos = print_arg_help(decl, prefix, 0);
 381         printf("=ulong");
 382         pos += 6;
 383         print_help_msg(decl, pos);
 384         printf("\n");
 385 }
 386
 387 static void print_str_help(struct isl_arg *decl, const char *prefix)
 388 {
 389         int pos;
 390         const char *a = decl->argument_name ? decl->argument_name : "string";
 391         pos = print_arg_help(decl, prefix, 0);
 392         pos = print_argument_name(decl, a, pos);
 393         print_help_msg(decl, pos);
 394         printf("\n");
 395 }
 396
 397 static void print_help(struct isl_arg *arg, const char *prefix)
 398 {
 399         int i;
 400
 401         for (i = 0; arg[i].type != isl_arg_end; ++i) {
 402                 switch (arg[i].type) {
 403                 case isl_arg_flags:
 404                         print_flags_help(&arg[i], prefix);
 405                         break;
 406                 case isl_arg_choice:
 407                         print_choice_help(&arg[i], prefix);
 408                         break;
 409                 case isl_arg_bool:
 410                         print_bool_help(&arg[i], prefix);
 411                         break;
 412                 case isl_arg_long:
 413                         print_long_help(&arg[i], prefix);
 414                         break;
 415                 case isl_arg_ulong:
 416                         print_ulong_help(&arg[i], prefix);
 417                         break;
 418                 case isl_arg_str:
 419                         print_str_help(&arg[i], prefix);
 420                         break;
 421                 case isl_arg_version:
 422                         printf("  -V, --version\n");
 423                         break;
 424                 case isl_arg_arg:
 425                 case isl_arg_child:
 426                 case isl_arg_user:
 427                 case isl_arg_end:
 428                         break;
 429                 }
 430         }
 431
 432         for (i = 0; arg[i].type != isl_arg_end; ++i) {
 433                 if (arg[i].type != isl_arg_child)
 434                         continue;
 435
 436                 printf("\n");
 437                 if (arg[i].help_msg)
 438                         printf(" %s\n", arg[i].help_msg);
 439                 print_help(arg[i].u.child.child, arg[i].long_name);
 440         }
 441 }
 442
 443 static const char *prog_name(const char *prog)
 444 {
 445         const char *slash;
 446
 447         slash = strrchr(prog, '/');
 448         if (slash)
 449                 prog = slash + 1;
 450         if (strncmp(prog, "lt-", 3) == 0)
 451                 prog += 3;
 452
 453         return prog;
 454 }
 455
 456 static void print_help_and_exit(struct isl_arg *arg, const char *prog)
 457 {
 458         int i;
 459
 460         printf("Usage: %s [OPTION...]", prog_name(prog));
 461
 462         for (i = 0; arg[i].type != isl_arg_end; ++i)
 463                 if (arg[i].type == isl_arg_arg)
 464                         printf(" %s", arg[i].argument_name);
 465
 466         printf("\n\n");
 467
 468         print_help(arg, NULL);
 469
 470         exit(0);
 471 }
 472
 473 static const char *skip_name(struct isl_arg *decl, const char *arg,
 474         const char *prefix, int need_argument, int *has_argument)
 475 {
 476         const char *equal;
 477         const char *name;
 478         const char *end;
 479
 480         if (arg[0] == '-' && arg[1] && arg[1] == decl->short_name) {
 481                 if (need_argument && !arg[2])
 482                         return NULL;
 483                 if (has_argument)
 484                         *has_argument = arg[2] != '\0';
 485                 return arg + 2;
 486         }
 487         if (!decl->long_name)
 488                 return NULL;
 489
 490         if (strncmp(arg, "--", 2))
 491                 return NULL;
 492
 493         name = arg + 2;
 494         equal = strchr(name, '=');
 495         if (need_argument && !equal)
 496                 return NULL;
 497
 498         if (has_argument)
 499                 *has_argument = !!equal;
 500         end = equal ? equal : name + strlen(name);
 501
 502         if (prefix) {
 503                 size_t prefix_len = strlen(prefix);
 504                 if (strncmp(name, prefix, prefix_len) == 0 &&
 505                     name[prefix_len] == '-')
 506                         name += prefix_len + 1;
 507         }
 508
 509         if (end - name != strlen(decl->long_name) ||
 510             strncmp(name, decl->long_name, end - name))
 511                 return NULL;
 512
 513         return equal ? equal + 1 : end;
 514 }
 515
 516 static int parse_choice_option(struct isl_arg *decl, char **arg,
 517         const char *prefix, void *opt)
 518 {
 519         int i;
 520         int has_argument;
 521         const char *choice;
 522
 523         choice = skip_name(decl, arg[0], prefix, 0, &has_argument);
 524         if (!choice)
 525                 return 0;
 526
 527         if (!has_argument && (!arg[1] || arg[1][0] == '-')) {
 528                 unsigned u = decl->u.choice.default_selected;
 529                 if (decl->u.choice.set)
 530                         decl->u.choice.set(opt, u);
 531                 else
 532                         *(unsigned *)(((char *)opt) + decl->offset) = u;
 533
 534                 return 1;
 535         }
 536
 537         if (!has_argument)
 538                 choice = arg[1];
 539
 540         for (i = 0; decl->u.choice.choice[i].name; ++i) {
 541                 unsigned u;
 542
 543                 if (strcmp(choice, decl->u.choice.choice[i].name))
 544                         continue;
 545
 546                 u = decl->u.choice.choice[i].value;
 547                 if (decl->u.choice.set)
 548                         decl->u.choice.set(opt, u);
 549                 else
 550                         *(unsigned *)(((char *)opt) + decl->offset) = u;
 551
 552                 return has_argument ? 1 : 2;
 553         }
 554
 555         return 0;
 556 }
 557
 558 static int set_flag(struct isl_arg *decl, unsigned *val, const char *flag,
 559         size_t len)
 560 {
 561         int i;
 562
 563         for (i = 0; decl->u.flags.flags[i].name; ++i) {
 564                 if (strncmp(flag, decl->u.flags.flags[i].name, len))
 565                         continue;
 566
 567                 *val &= ~decl->u.flags.flags[i].mask;
 568                 *val |= decl->u.flags.flags[i].value;
 569
 570                 return 1;
 571         }
 572
 573         return 0;
 574 }
 575
 576 static int parse_flags_option(struct isl_arg *decl, char **arg,
 577         const char *prefix, void *opt)
 578 {
 579         int has_argument;
 580         const char *flags;
 581         const char *comma;
 582         unsigned val;
 583
 584         flags = skip_name(decl, arg[0], prefix, 0, &has_argument);
 585         if (!flags)
 586                 return 0;
 587
 588         if (!has_argument && !arg[1])
 589                 return 0;
 590
 591         if (!has_argument)
 592                 flags = arg[1];
 593
 594         val = *(unsigned *)(((char *)opt) + decl->offset);
 595
 596         while ((comma = strchr(flags, ',')) != NULL) {
 597                 if (!set_flag(decl, &val, flags, comma - flags))
 598                         return 0;
 599                 flags = comma + 1;
 600         }
 601         if (!set_flag(decl, &val, flags, strlen(flags)))
 602                 return 0;
 603
 604         *(unsigned *)(((char *)opt) + decl->offset) = val;
 605
 606         return has_argument ? 1 : 2;
 607 }
 608
 609 static int parse_bool_option(struct isl_arg *decl, const char *arg,
 610         const char *prefix, void *opt)
 611 {
 612         if (skip_name(decl, arg, prefix, 0, NULL)) {
 613                 *(unsigned *)(((char *)opt) + decl->offset) = 1;
 614
 615                 return 1;
 616         }
 617
 618         if (!decl->long_name)
 619                 return 0;
 620
 621         if (strncmp(arg, "--", 2))
 622                 return 0;
 623         arg += 2;
 624
 625         if (prefix) {
 626                 size_t prefix_len = strlen(prefix);
 627                 if (strncmp(arg, prefix, prefix_len) == 0 &&
 628                     arg[prefix_len] == '-') {
 629                         arg += prefix_len + 1;
 630                         prefix = NULL;
 631                 }
 632         }
 633
 634         if (strncmp(arg, "no-", 3))
 635                 return 0;
 636         arg += 3;
 637
 638         if (prefix) {
 639                 size_t prefix_len = strlen(prefix);
 640                 if (strncmp(arg, prefix, prefix_len) == 0 &&
 641                     arg[prefix_len] == '-')
 642                         arg += prefix_len + 1;
 643         }
 644
 645         if (!strcmp(arg, decl->long_name)) {
 646                 *(unsigned *)(((char *)opt) + decl->offset) = 0;
 647
 648                 return 1;
 649         }
 650
 651         return 0;
 652 }
 653
 654 static int parse_str_option(struct isl_arg *decl, char **arg,
 655         const char *prefix, void *opt)
 656 {
 657         int has_argument;
 658         const char *s;
 659         char **p = (char **)(((char *)opt) + decl->offset);
 660
 661         s = skip_name(decl, arg[0], prefix, 0, &has_argument);
 662         if (!s)
 663                 return 0;
 664
 665         if (has_argument) {
 666                 *p = strdup(s);
 667                 return 1;
 668         }
 669
 670         if (arg[1]) {
 671                 *p = strdup(arg[1]);
 672                 return 2;
 673         }
 674
 675         return 0;
 676 }
 677
 678 static int parse_long_option(struct isl_arg *decl, char **arg,
 679         const char *prefix, void *opt)
 680 {
 681         int has_argument;
 682         const char *val;
 683         char *endptr;
 684         long *p = (long *)(((char *)opt) + decl->offset);
 685
 686         val = skip_name(decl, arg[0], prefix, 0, &has_argument);
 687         if (!val)
 688                 return 0;
 689
 690         if (has_argument) {
 691                 long l = strtol(val, NULL, 0);
 692                 if (decl->u.l.set)
 693                         decl->u.l.set(opt, l);
 694                 else
 695                         *p = l;
 696                 return 1;
 697         }
 698
 699         if (arg[1]) {
 700                 long l = strtol(arg[1], &endptr, 0);
 701                 if (*endptr == '\0') {
 702                         if (decl->u.l.set)
 703                                 decl->u.l.set(opt, l);
 704                         else
 705                                 *p = l;
 706                         return 2;
 707                 }
 708         }
 709
 710         if (decl->u.l.default_value != decl->u.l.default_selected) {
 711                 if (decl->u.l.set)
 712                         decl->u.l.set(opt, decl->u.l.default_selected);
 713                 else
 714                         *p = decl->u.l.default_selected;
 715                 return 1;
 716         }
 717
 718         return 0;
 719 }
 720
 721 static int parse_ulong_option(struct isl_arg *decl, char **arg,
 722         const char *prefix, void *opt)
 723 {
 724         int has_argument;
 725         const char *val;
 726         char *endptr;
 727         unsigned long *p = (unsigned long *)(((char *)opt) + decl->offset);
 728
 729         val = skip_name(decl, arg[0], prefix, 0, &has_argument);
 730         if (!val)
 731                 return 0;
 732
 733         if (has_argument) {
 734                 *p = strtoul(val, NULL, 0);
 735                 return 1;
 736         }
 737
 738         if (arg[1]) {
 739                 unsigned long ul = strtoul(arg[1], &endptr, 0);
 740                 if (*endptr == '\0') {
 741                         *p = ul;
 742                         return 2;
 743                 }
 744         }
 745
 746         return 0;
 747 }
 748
 749 static int parse_option(struct isl_arg *decl, char **arg,
 750         const char *prefix, void *opt);
 751
 752 static int parse_child_option(struct isl_arg *decl, char **arg, void *opt)
 753 {
 754         return parse_option(decl->u.child.child, arg, decl->long_name,
 755                                 *(void **)(((char *)opt) + decl->offset));
 756 }
 757
 758 static int parse_option(struct isl_arg *decl, char **arg,
 759         const char *prefix, void *opt)
 760 {
 761         int i;
 762
 763         for (i = 0; decl[i].type != isl_arg_end; ++i) {
 764                 int parsed = 0;
 765                 switch (decl[i].type) {
 766                 case isl_arg_choice:
 767                         parsed = parse_choice_option(&decl[i], arg, prefix, opt);
 768                         break;
 769                 case isl_arg_flags:
 770                         parsed = parse_flags_option(&decl[i], arg, prefix, opt);
 771                         break;
 772                 case isl_arg_long:
 773                         parsed = parse_long_option(&decl[i], arg, prefix, opt);
 774                         break;
 775                 case isl_arg_ulong:
 776                         parsed = parse_ulong_option(&decl[i], arg, prefix, opt);
 777                         break;
 778                 case isl_arg_bool:
 779                         parsed = parse_bool_option(&decl[i], *arg, prefix, opt);
 780                         break;
 781                 case isl_arg_str:
 782                         parsed = parse_str_option(&decl[i], arg, prefix, opt);
 783                         break;
 784                 case isl_arg_child:
 785                         parsed = parse_child_option(&decl[i], arg, opt);
 786                         break;
 787                 case isl_arg_arg:
 788                 case isl_arg_user:
 789                 case isl_arg_version:
 790                 case isl_arg_end:
 791                         break;
 792                 }
 793                 if (parsed)
 794                         return parsed;
 795         }
 796
 797         return 0;
 798 }
 799
 800 static int any_version(struct isl_arg *decl)
 801 {
 802         int i;
 803
 804         for (i = 0; decl[i].type != isl_arg_end; ++i) {
 805                 switch (decl[i].type) {
 806                 case isl_arg_version:
 807                         return 1;
 808                 case isl_arg_child:
 809                         if (any_version(decl[i].u.child.child))
 810                                 return 1;
 811                         break;
 812                 default:
 813                         break;
 814                 }
 815         }
 816
 817         return 0;
 818 }
 819
 820 static void print_version(struct isl_arg *decl)
 821 {
 822         int i;
 823
 824         for (i = 0; decl[i].type != isl_arg_end; ++i) {
 825                 switch (decl[i].type) {
 826                 case isl_arg_version:
 827                         decl[i].u.version.print_version();
 828                         break;
 829                 case isl_arg_child:
 830                         print_version(decl[i].u.child.child);
 831                         break;
 832                 default:
 833                         break;
 834                 }
 835         }
 836 }
 837
 838 static void print_version_and_exit(struct isl_arg *decl)
 839 {
 840         print_version(decl);
 841
 842         exit(0);
 843 }
 844
 845 static int drop_argument(int argc, char **argv, int drop, int n)
 846 {
 847         for (; drop < argc; ++drop)
 848                 argv[drop] = argv[drop + n];
 849
 850         return argc - n;
 851 }
 852
 853 static int n_arg(struct isl_arg *arg)
 854 {
 855         int i;
 856         int n_arg = 0;
 857
 858         for (i = 0; arg[i].type != isl_arg_end; ++i)
 859                 if (arg[i].type == isl_arg_arg)
 860                         n_arg++;
 861
 862         return n_arg;
 863 }
 864
 865 int isl_arg_parse(struct isl_arg *arg, int argc, char **argv, void *opt,
 866         unsigned flags)
 867 {
 868         int skip = 0;
 869         int i;
 870         int n;
 871
 872         n = n_arg(arg);
 873
 874         for (i = 1; i < argc; ++i) {
 875                 if (strcmp(argv[i], "--help") == 0)
 876                         print_help_and_exit(arg, argv[0]);
 877         }
 878
 879         for (i = 1; i < argc; ++i) {
 880                 if ((strcmp(argv[i], "--version") == 0 ||
 881                      strcmp(argv[i], "-V") == 0) && any_version(arg))
 882                         print_version_and_exit(arg);
 883         }
 884
 885         while (argc > 1 + skip) {
 886                 int parsed;
 887                 if (argv[1 + skip][0] != '-')
 888                         break;
 889                 parsed = parse_option(arg, &argv[1 + skip], NULL, opt);
 890                 if (parsed)
 891                         argc = drop_argument(argc, argv, 1 + skip, parsed);
 892                 else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
 893                         fprintf(stderr, "%s: unrecognized option: %s\n",
 894                                         prog_name(argv[0]), argv[1 + skip]);
 895                         exit(-1);
 896                 } else
 897                         ++skip;
 898         }
 899
 900         if (ISL_FL_ISSET(flags, ISL_ARG_ALL) ? argc != 1 + n
 901                                              : argc < 1 + skip + n) {
 902                 fprintf(stderr, "%s: expecting %d arguments\n",
 903                                 prog_name(argv[0]), n);
 904                 exit(-1);
 905         }
 906
 907         for (i = 0; arg[i].type != isl_arg_end; ++i) {
 908                 const char *str;
 909                 if (arg[i].type != isl_arg_arg)
 910                         continue;
 911                 str = strdup(argv[1 + skip]);
 912                 *(const char **)(((char *)opt) + arg[i].offset) = str;
 913                 argc = drop_argument(argc, argv, 1 + skip, 1);
 914         }
 915
 916         return argc;
 917 }