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