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, char **arg,
 681         const char *prefix, void *opt)
 682 {
 683         const char *name;
 684         unsigned *p = (unsigned *)(((char *)opt) + decl->offset);
 685
 686         if (skip_name(decl, arg[0], prefix, 0, NULL)) {
 687                 if ((decl->flags && ISL_ARG_BOOL_ARG) && arg[1]) {
 688                         char *endptr;
 689                         int val = strtol(arg[1], &endptr, 0);
 690                         if (*endptr == '\0' && (val == 0 || val == 1)) {
 691                                 if (decl->u.b.set)
 692                                         decl->u.b.set(opt, val);
 693                                 else if (decl->offset != (size_t) -1)
 694                                         *p = val;
 695                                 return 2;
 696                         }
 697                 }
 698                 if (decl->u.b.set)
 699                         decl->u.b.set(opt, 1);
 700                 else if (decl->offset != (size_t) -1)
 701                         *p = 1;
 702
 703                 return 1;
 704         }
 705
 706         if (!decl->long_name)
 707                 return 0;
 708
 709         name = skip_dash_dash(decl, arg[0]);
 710         if (!name)
 711                 return 0;
 712
 713         if (prefix) {
 714                 size_t prefix_len = strlen(prefix);
 715                 if (strncmp(name, prefix, prefix_len) == 0 &&
 716                     name[prefix_len] == '-') {
 717                         name += prefix_len + 1;
 718                         prefix = NULL;
 719                 }
 720         }
 721
 722         if (strncmp(name, "no-", 3))
 723                 return 0;
 724         name += 3;
 725
 726         if (prefix) {
 727                 size_t prefix_len = strlen(prefix);
 728                 if (strncmp(name, prefix, prefix_len) == 0 &&
 729                     name[prefix_len] == '-')
 730                         name += prefix_len + 1;
 731         }
 732
 733         if (match_long_name(decl, name, name + strlen(name))) {
 734                 if (decl->u.b.set)
 735                         decl->u.b.set(opt, 0);
 736                 else if (decl->offset != (size_t) -1)
 737                         *p = 0;
 738
 739                 return 1;
 740         }
 741
 742         return 0;
 743 }
 744
 745 static int parse_str_option(struct isl_arg *decl, char **arg,
 746         const char *prefix, void *opt)
 747 {
 748         int has_argument;
 749         const char *s;
 750         char **p = (char **)(((char *)opt) + decl->offset);
 751
 752         s = skip_name(decl, arg[0], prefix, 0, &has_argument);
 753         if (!s)
 754                 return 0;
 755
 756         if (has_argument) {
 757                 free(*p);
 758                 *p = strdup(s);
 759                 return 1;
 760         }
 761
 762         if (arg[1]) {
 763                 free(*p);
 764                 *p = strdup(arg[1]);
 765                 return 2;
 766         }
 767
 768         return 0;
 769 }
 770
 771 static int parse_long_option(struct isl_arg *decl, char **arg,
 772         const char *prefix, void *opt)
 773 {
 774         int has_argument;
 775         const char *val;
 776         char *endptr;
 777         long *p = (long *)(((char *)opt) + decl->offset);
 778
 779         val = skip_name(decl, arg[0], prefix, 0, &has_argument);
 780         if (!val)
 781                 return 0;
 782
 783         if (has_argument) {
 784                 long l = strtol(val, NULL, 0);
 785                 if (decl->u.l.set)
 786                         decl->u.l.set(opt, l);
 787                 else
 788                         *p = l;
 789                 return 1;
 790         }
 791
 792         if (arg[1]) {
 793                 long l = strtol(arg[1], &endptr, 0);
 794                 if (*endptr == '\0') {
 795                         if (decl->u.l.set)
 796                                 decl->u.l.set(opt, l);
 797                         else
 798                                 *p = l;
 799                         return 2;
 800                 }
 801         }
 802
 803         if (decl->u.l.default_value != decl->u.l.default_selected) {
 804                 if (decl->u.l.set)
 805                         decl->u.l.set(opt, decl->u.l.default_selected);
 806                 else
 807                         *p = decl->u.l.default_selected;
 808                 return 1;
 809         }
 810
 811         return 0;
 812 }
 813
 814 static int parse_ulong_option(struct isl_arg *decl, char **arg,
 815         const char *prefix, void *opt)
 816 {
 817         int has_argument;
 818         const char *val;
 819         char *endptr;
 820         unsigned long *p = (unsigned long *)(((char *)opt) + decl->offset);
 821
 822         val = skip_name(decl, arg[0], prefix, 0, &has_argument);
 823         if (!val)
 824                 return 0;
 825
 826         if (has_argument) {
 827                 *p = strtoul(val, NULL, 0);
 828                 return 1;
 829         }
 830
 831         if (arg[1]) {
 832                 unsigned long ul = strtoul(arg[1], &endptr, 0);
 833                 if (*endptr == '\0') {
 834                         *p = ul;
 835                         return 2;
 836                 }
 837         }
 838
 839         return 0;
 840 }
 841
 842 static int parse_option(struct isl_arg *decl, char **arg,
 843         const char *prefix, void *opt);
 844
 845 static int parse_child_option(struct isl_arg *decl, char **arg, void *opt)
 846 {
 847         return parse_option(decl->u.child.child, arg, decl->long_name,
 848                                 *(void **)(((char *)opt) + decl->offset));
 849 }
 850
 851 static int parse_option(struct isl_arg *decl, char **arg,
 852         const char *prefix, void *opt)
 853 {
 854         int i;
 855
 856         for (i = 0; decl[i].type != isl_arg_end; ++i) {
 857                 int parsed = 0;
 858                 switch (decl[i].type) {
 859                 case isl_arg_choice:
 860                         parsed = parse_choice_option(&decl[i], arg, prefix, opt);
 861                         break;
 862                 case isl_arg_flags:
 863                         parsed = parse_flags_option(&decl[i], arg, prefix, opt);
 864                         break;
 865                 case isl_arg_long:
 866                         parsed = parse_long_option(&decl[i], arg, prefix, opt);
 867                         break;
 868                 case isl_arg_ulong:
 869                         parsed = parse_ulong_option(&decl[i], arg, prefix, opt);
 870                         break;
 871                 case isl_arg_bool:
 872                         parsed = parse_bool_option(&decl[i], arg, prefix, opt);
 873                         break;
 874                 case isl_arg_str:
 875                         parsed = parse_str_option(&decl[i], arg, prefix, opt);
 876                         break;
 877                 case isl_arg_child:
 878                         parsed = parse_child_option(&decl[i], arg, opt);
 879                         break;
 880                 case isl_arg_alias:
 881                 case isl_arg_arg:
 882                 case isl_arg_user:
 883                 case isl_arg_version:
 884                 case isl_arg_end:
 885                         break;
 886                 }
 887                 if (parsed)
 888                         return parsed;
 889         }
 890
 891         return 0;
 892 }
 893
 894 static void print_version(struct isl_arg *decl)
 895 {
 896         int i;
 897
 898         for (i = 0; decl[i].type != isl_arg_end; ++i) {
 899                 switch (decl[i].type) {
 900                 case isl_arg_version:
 901                         decl[i].u.version.print_version();
 902                         break;
 903                 case isl_arg_child:
 904                         print_version(decl[i].u.child.child);
 905                         break;
 906                 default:
 907                         break;
 908                 }
 909         }
 910 }
 911
 912 static void print_version_and_exit(struct isl_arg *decl)
 913 {
 914         print_version(decl);
 915
 916         exit(0);
 917 }
 918
 919 static int drop_argument(int argc, char **argv, int drop, int n)
 920 {
 921         for (; drop < argc; ++drop)
 922                 argv[drop] = argv[drop + n];
 923
 924         return argc - n;
 925 }
 926
 927 static int n_arg(struct isl_arg *arg)
 928 {
 929         int i;
 930         int n_arg = 0;
 931
 932         for (i = 0; arg[i].type != isl_arg_end; ++i)
 933                 if (arg[i].type == isl_arg_arg)
 934                         n_arg++;
 935
 936         return n_arg;
 937 }
 938
 939 static int next_arg(struct isl_arg *arg, int a)
 940 {
 941         for (++a; arg[a].type != isl_arg_end; ++a)
 942                 if (arg[a].type == isl_arg_arg)
 943                         return a;
 944
 945         return -1;
 946 }
 947
 948 int isl_arg_parse(struct isl_arg *arg, int argc, char **argv, void *opt,
 949         unsigned flags)
 950 {
 951         int a = -1;
 952         int skip = 0;
 953         int i;
 954         int n;
 955
 956         n = n_arg(arg);
 957
 958         for (i = 1; i < argc; ++i) {
 959                 if (strcmp(argv[i], "--help") == 0)
 960                         print_help_and_exit(arg, argv[0]);
 961         }
 962
 963         for (i = 1; i < argc; ++i) {
 964                 if ((strcmp(argv[i], "--version") == 0 ||
 965                      strcmp(argv[i], "-V") == 0) && any_version(arg))
 966                         print_version_and_exit(arg);
 967         }
 968
 969         while (argc > 1 + skip) {
 970                 int parsed;
 971                 if (argv[1 + skip][0] != '-') {
 972                         a = next_arg(arg, a);
 973                         if (a >= 0) {
 974                                 char **p;
 975                                 p = (char **)(((char *)opt)+arg[a].offset);
 976                                 free(*p);
 977                                 *p = strdup(argv[1 + skip]);
 978                                 argc = drop_argument(argc, argv, 1 + skip, 1);
 979                                 --n;
 980                         } else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
 981                                 fprintf(stderr, "%s: extra argument: %s\n",
 982                                             prog_name(argv[0]), argv[1 + skip]);
 983                                 exit(-1);
 984                         } else
 985                                 ++skip;
 986                         continue;
 987                 }
 988                 parsed = parse_option(arg, &argv[1 + skip], NULL, opt);
 989                 if (parsed)
 990                         argc = drop_argument(argc, argv, 1 + skip, parsed);
 991                 else if (ISL_FL_ISSET(flags, ISL_ARG_ALL)) {
 992                         fprintf(stderr, "%s: unrecognized option: %s\n",
 993                                         prog_name(argv[0]), argv[1 + skip]);
 994                         exit(-1);
 995                 } else
 996                         ++skip;
 997         }
 998
 999         if (n > 0) {
1000                 fprintf(stderr, "%s: expecting %d more argument(s)\n",
1001                                 prog_name(argv[0]), n);
1002                 exit(-1);
1003         }
1004
1005         return argc;
1006 }