parse-options.c

   1 #include "git-compat-util.h"
   2 #include "parse-options.h"
   3 #include "cache.h"
   4 #include "config.h"
   5 #include "commit.h"
   6 #include "color.h"
   7 #include "utf8.h"
   8
   9 static int disallow_abbreviated_options;
  10
  11 enum opt_parsed {
  12         OPT_LONG  = 0,
  13         OPT_SHORT = 1<<0,
  14         OPT_UNSET = 1<<1,
  15 };
  16
  17 static void optbug(const struct option *opt, const char *reason)
  18 {
  19         if (opt->long_name && opt->short_name)
  20                 bug("switch '%c' (--%s) %s", opt->short_name,
  21                     opt->long_name, reason);
  22         else if (opt->long_name)
  23                 bug("option '%s' %s", opt->long_name, reason);
  24         else
  25                 bug("switch '%c' %s", opt->short_name, reason);
  26 }
  27
  28 static const char *optname(const struct option *opt, enum opt_parsed flags)
  29 {
  30         static struct strbuf sb = STRBUF_INIT;
  31
  32         strbuf_reset(&sb);
  33         if (flags & OPT_SHORT)
  34                 strbuf_addf(&sb, "switch `%c'", opt->short_name);
  35         else if (flags & OPT_UNSET)
  36                 strbuf_addf(&sb, "option `no-%s'", opt->long_name);
  37         else if (flags == OPT_LONG)
  38                 strbuf_addf(&sb, "option `%s'", opt->long_name);
  39         else
  40                 BUG("optname() got unknown flags %d", flags);
  41
  42         return sb.buf;
  43 }
  44
  45 static enum parse_opt_result get_arg(struct parse_opt_ctx_t *p,
  46                                      const struct option *opt,
  47                                      enum opt_parsed flags, const char **arg)
  48 {
  49         if (p->opt) {
  50                 *arg = p->opt;
  51                 p->opt = NULL;
  52         } else if (p->argc == 1 && (opt->flags & PARSE_OPT_LASTARG_DEFAULT)) {
  53                 *arg = (const char *)opt->defval;
  54         } else if (p->argc > 1) {
  55                 p->argc--;
  56                 *arg = *++p->argv;
  57         } else
  58                 return error(_("%s requires a value"), optname(opt, flags));
  59         return 0;
  60 }
  61
  62 static void fix_filename(const char *prefix, const char **file)
  63 {
  64         if (!file || !*file || !prefix || is_absolute_path(*file)
  65             || !strcmp("-", *file))
  66                 return;
  67         *file = prefix_filename(prefix, *file);
  68 }
  69
  70 static enum parse_opt_result opt_command_mode_error(
  71         const struct option *opt,
  72         const struct option *all_opts,
  73         enum opt_parsed flags)
  74 {
  75         const struct option *that;
  76         struct strbuf that_name = STRBUF_INIT;
  77
  78         /*
  79          * Find the other option that was used to set the variable
  80          * already, and report that this is not compatible with it.
  81          */
  82         for (that = all_opts; that->type != OPTION_END; that++) {
  83                 if (that == opt ||
  84                     !(that->flags & PARSE_OPT_CMDMODE) ||
  85                     that->value != opt->value ||
  86                     that->defval != *(int *)opt->value)
  87                         continue;
  88
  89                 if (that->long_name)
  90                         strbuf_addf(&that_name, "--%s", that->long_name);
  91                 else
  92                         strbuf_addf(&that_name, "-%c", that->short_name);
  93                 error(_("%s is incompatible with %s"),
  94                       optname(opt, flags), that_name.buf);
  95                 strbuf_release(&that_name);
  96                 return PARSE_OPT_ERROR;
  97         }
  98         return error(_("%s : incompatible with something else"),
  99                      optname(opt, flags));
 100 }
 101
 102 static enum parse_opt_result get_value(struct parse_opt_ctx_t *p,
 103                                        const struct option *opt,
 104                                        const struct option *all_opts,
 105                                        enum opt_parsed flags)
 106 {
 107         const char *s, *arg;
 108         const int unset = flags & OPT_UNSET;
 109         int err;
 110
 111         if (unset && p->opt)
 112                 return error(_("%s takes no value"), optname(opt, flags));
 113         if (unset && (opt->flags & PARSE_OPT_NONEG))
 114                 return error(_("%s isn't available"), optname(opt, flags));
 115         if (!(flags & OPT_SHORT) && p->opt && (opt->flags & PARSE_OPT_NOARG))
 116                 return error(_("%s takes no value"), optname(opt, flags));
 117
 118         /*
 119          * Giving the same mode option twice, although unnecessary,
 120          * is not a grave error, so let it pass.
 121          */
 122         if ((opt->flags & PARSE_OPT_CMDMODE) &&
 123             *(int *)opt->value && *(int *)opt->value != opt->defval)
 124                 return opt_command_mode_error(opt, all_opts, flags);
 125
 126         switch (opt->type) {
 127         case OPTION_LOWLEVEL_CALLBACK:
 128                 return opt->ll_callback(p, opt, NULL, unset);
 129
 130         case OPTION_BIT:
 131                 if (unset)
 132                         *(int *)opt->value &= ~opt->defval;
 133                 else
 134                         *(int *)opt->value |= opt->defval;
 135                 return 0;
 136
 137         case OPTION_NEGBIT:
 138                 if (unset)
 139                         *(int *)opt->value |= opt->defval;
 140                 else
 141                         *(int *)opt->value &= ~opt->defval;
 142                 return 0;
 143
 144         case OPTION_BITOP:
 145                 if (unset)
 146                         BUG("BITOP can't have unset form");
 147                 *(int *)opt->value &= ~opt->extra;
 148                 *(int *)opt->value |= opt->defval;
 149                 return 0;
 150
 151         case OPTION_COUNTUP:
 152                 if (*(int *)opt->value < 0)
 153                         *(int *)opt->value = 0;
 154                 *(int *)opt->value = unset ? 0 : *(int *)opt->value + 1;
 155                 return 0;
 156
 157         case OPTION_SET_INT:
 158                 *(int *)opt->value = unset ? 0 : opt->defval;
 159                 return 0;
 160
 161         case OPTION_STRING:
 162                 if (unset)
 163                         *(const char **)opt->value = NULL;
 164                 else if (opt->flags & PARSE_OPT_OPTARG && !p->opt)
 165                         *(const char **)opt->value = (const char *)opt->defval;
 166                 else
 167                         return get_arg(p, opt, flags, (const char **)opt->value);
 168                 return 0;
 169
 170         case OPTION_FILENAME:
 171                 err = 0;
 172                 if (unset)
 173                         *(const char **)opt->value = NULL;
 174                 else if (opt->flags & PARSE_OPT_OPTARG && !p->opt)
 175                         *(const char **)opt->value = (const char *)opt->defval;
 176                 else
 177                         err = get_arg(p, opt, flags, (const char **)opt->value);
 178
 179                 if (!err)
 180                         fix_filename(p->prefix, (const char **)opt->value);
 181                 return err;
 182
 183         case OPTION_CALLBACK:
 184         {
 185                 const char *p_arg = NULL;
 186                 int p_unset;
 187
 188                 if (unset)
 189                         p_unset = 1;
 190                 else if (opt->flags & PARSE_OPT_NOARG)
 191                         p_unset = 0;
 192                 else if (opt->flags & PARSE_OPT_OPTARG && !p->opt)
 193                         p_unset = 0;
 194                 else if (get_arg(p, opt, flags, &arg))
 195                         return -1;
 196                 else {
 197                         p_unset = 0;
 198                         p_arg = arg;
 199                 }
 200                 if (opt->callback)
 201                         return (*opt->callback)(opt, p_arg, p_unset) ? (-1) : 0;
 202                 else
 203                         return (*opt->ll_callback)(p, opt, p_arg, p_unset);
 204         }
 205         case OPTION_INTEGER:
 206                 if (unset) {
 207                         *(int *)opt->value = 0;
 208                         return 0;
 209                 }
 210                 if (opt->flags & PARSE_OPT_OPTARG && !p->opt) {
 211                         *(int *)opt->value = opt->defval;
 212                         return 0;
 213                 }
 214                 if (get_arg(p, opt, flags, &arg))
 215                         return -1;
 216                 if (!*arg)
 217                         return error(_("%s expects a numerical value"),
 218                                      optname(opt, flags));
 219                 *(int *)opt->value = strtol(arg, (char **)&s, 10);
 220                 if (*s)
 221                         return error(_("%s expects a numerical value"),
 222                                      optname(opt, flags));
 223                 return 0;
 224
 225         case OPTION_MAGNITUDE:
 226                 if (unset) {
 227                         *(unsigned long *)opt->value = 0;
 228                         return 0;
 229                 }
 230                 if (opt->flags & PARSE_OPT_OPTARG && !p->opt) {
 231                         *(unsigned long *)opt->value = opt->defval;
 232                         return 0;
 233                 }
 234                 if (get_arg(p, opt, flags, &arg))
 235                         return -1;
 236                 if (!git_parse_ulong(arg, opt->value))
 237                         return error(_("%s expects a non-negative integer value"
 238                                        " with an optional k/m/g suffix"),
 239                                      optname(opt, flags));
 240                 return 0;
 241
 242         default:
 243                 BUG("opt->type %d should not happen", opt->type);
 244         }
 245 }
 246
 247 static enum parse_opt_result parse_short_opt(struct parse_opt_ctx_t *p,
 248                                              const struct option *options)
 249 {
 250         const struct option *all_opts = options;
 251         const struct option *numopt = NULL;
 252
 253         for (; options->type != OPTION_END; options++) {
 254                 if (options->short_name == *p->opt) {
 255                         p->opt = p->opt[1] ? p->opt + 1 : NULL;
 256                         return get_value(p, options, all_opts, OPT_SHORT);
 257                 }
 258
 259                 /*
 260                  * Handle the numerical option later, explicit one-digit
 261                  * options take precedence over it.
 262                  */
 263                 if (options->type == OPTION_NUMBER)
 264                         numopt = options;
 265         }
 266         if (numopt && isdigit(*p->opt)) {
 267                 size_t len = 1;
 268                 char *arg;
 269                 int rc;
 270
 271                 while (isdigit(p->opt[len]))
 272                         len++;
 273                 arg = xmemdupz(p->opt, len);
 274                 p->opt = p->opt[len] ? p->opt + len : NULL;
 275                 if (numopt->callback)
 276                         rc = (*numopt->callback)(numopt, arg, 0) ? (-1) : 0;
 277                 else
 278                         rc = (*numopt->ll_callback)(p, numopt, arg, 0);
 279                 free(arg);
 280                 return rc;
 281         }
 282         return PARSE_OPT_UNKNOWN;
 283 }
 284
 285 static int has_string(const char *it, const char **array)
 286 {
 287         while (*array)
 288                 if (!strcmp(it, *(array++)))
 289                         return 1;
 290         return 0;
 291 }
 292
 293 static int is_alias(struct parse_opt_ctx_t *ctx,
 294                     const struct option *one_opt,
 295                     const struct option *another_opt)
 296 {
 297         const char **group;
 298
 299         if (!ctx->alias_groups)
 300                 return 0;
 301
 302         if (!one_opt->long_name || !another_opt->long_name)
 303                 return 0;
 304
 305         for (group = ctx->alias_groups; *group; group += 3) {
 306                 /* it and other are from the same family? */
 307                 if (has_string(one_opt->long_name, group) &&
 308                     has_string(another_opt->long_name, group))
 309                         return 1;
 310         }
 311         return 0;
 312 }
 313
 314 static enum parse_opt_result parse_long_opt(
 315         struct parse_opt_ctx_t *p, const char *arg,
 316         const struct option *options)
 317 {
 318         const struct option *all_opts = options;
 319         const char *arg_end = strchrnul(arg, '=');
 320         const struct option *abbrev_option = NULL, *ambiguous_option = NULL;
 321         enum opt_parsed abbrev_flags = OPT_LONG, ambiguous_flags = OPT_LONG;
 322
 323         for (; options->type != OPTION_END; options++) {
 324                 const char *rest, *long_name = options->long_name;
 325                 enum opt_parsed flags = OPT_LONG, opt_flags = OPT_LONG;
 326
 327                 if (options->type == OPTION_SUBCOMMAND)
 328                         continue;
 329                 if (!long_name)
 330                         continue;
 331
 332 again:
 333                 if (!skip_prefix(arg, long_name, &rest))
 334                         rest = NULL;
 335                 if (!rest) {
 336                         /* abbreviated? */
 337                         if (!(p->flags & PARSE_OPT_KEEP_UNKNOWN_OPT) &&
 338                             !strncmp(long_name, arg, arg_end - arg)) {
 339 is_abbreviated:
 340                                 if (abbrev_option &&
 341                                     !is_alias(p, abbrev_option, options)) {
 342                                         /*
 343                                          * If this is abbreviated, it is
 344                                          * ambiguous. So when there is no
 345                                          * exact match later, we need to
 346                                          * error out.
 347                                          */
 348                                         ambiguous_option = abbrev_option;
 349                                         ambiguous_flags = abbrev_flags;
 350                                 }
 351                                 if (!(flags & OPT_UNSET) && *arg_end)
 352                                         p->opt = arg_end + 1;
 353                                 abbrev_option = options;
 354                                 abbrev_flags = flags ^ opt_flags;
 355                                 continue;
 356                         }
 357                         /* negation allowed? */
 358                         if (options->flags & PARSE_OPT_NONEG)
 359                                 continue;
 360                         /* negated and abbreviated very much? */
 361                         if (starts_with("no-", arg)) {
 362                                 flags |= OPT_UNSET;
 363                                 goto is_abbreviated;
 364                         }
 365                         /* negated? */
 366                         if (!starts_with(arg, "no-")) {
 367                                 if (skip_prefix(long_name, "no-", &long_name)) {
 368                                         opt_flags |= OPT_UNSET;
 369                                         goto again;
 370                                 }
 371                                 continue;
 372                         }
 373                         flags |= OPT_UNSET;
 374                         if (!skip_prefix(arg + 3, long_name, &rest)) {
 375                                 /* abbreviated and negated? */
 376                                 if (starts_with(long_name, arg + 3))
 377                                         goto is_abbreviated;
 378                                 else
 379                                         continue;
 380                         }
 381                 }
 382                 if (*rest) {
 383                         if (*rest != '=')
 384                                 continue;
 385                         p->opt = rest + 1;
 386                 }
 387                 return get_value(p, options, all_opts, flags ^ opt_flags);
 388         }
 389
 390         if (disallow_abbreviated_options && (ambiguous_option || abbrev_option))
 391                 die("disallowed abbreviated or ambiguous option '%.*s'",
 392                     (int)(arg_end - arg), arg);
 393
 394         if (ambiguous_option) {
 395                 error(_("ambiguous option: %s "
 396                         "(could be --%s%s or --%s%s)"),
 397                         arg,
 398                         (ambiguous_flags & OPT_UNSET) ?  "no-" : "",
 399                         ambiguous_option->long_name,
 400                         (abbrev_flags & OPT_UNSET) ?  "no-" : "",
 401                         abbrev_option->long_name);
 402                 return PARSE_OPT_HELP;
 403         }
 404         if (abbrev_option)
 405                 return get_value(p, abbrev_option, all_opts, abbrev_flags);
 406         return PARSE_OPT_UNKNOWN;
 407 }
 408
 409 static enum parse_opt_result parse_nodash_opt(struct parse_opt_ctx_t *p,
 410                                               const char *arg,
 411                                               const struct option *options)
 412 {
 413         const struct option *all_opts = options;
 414
 415         for (; options->type != OPTION_END; options++) {
 416                 if (!(options->flags & PARSE_OPT_NODASH))
 417                         continue;
 418                 if (options->short_name == arg[0] && arg[1] == '\0')
 419                         return get_value(p, options, all_opts, OPT_SHORT);
 420         }
 421         return PARSE_OPT_ERROR;
 422 }
 423
 424 static enum parse_opt_result parse_subcommand(const char *arg,
 425                                               const struct option *options)
 426 {
 427         for (; options->type != OPTION_END; options++)
 428                 if (options->type == OPTION_SUBCOMMAND &&
 429                     !strcmp(options->long_name, arg)) {
 430                         *(parse_opt_subcommand_fn **)options->value = options->subcommand_fn;
 431                         return PARSE_OPT_SUBCOMMAND;
 432                 }
 433
 434         return PARSE_OPT_UNKNOWN;
 435 }
 436
 437 static void check_typos(const char *arg, const struct option *options)
 438 {
 439         if (strlen(arg) < 3)
 440                 return;
 441
 442         if (starts_with(arg, "no-")) {
 443                 error(_("did you mean `--%s` (with two dashes)?"), arg);
 444                 exit(129);
 445         }
 446
 447         for (; options->type != OPTION_END; options++) {
 448                 if (!options->long_name)
 449                         continue;
 450                 if (starts_with(options->long_name, arg)) {
 451                         error(_("did you mean `--%s` (with two dashes)?"), arg);
 452                         exit(129);
 453                 }
 454         }
 455 }
 456
 457 static void parse_options_check(const struct option *opts)
 458 {
 459         char short_opts[128];
 460         void *subcommand_value = NULL;
 461
 462         memset(short_opts, '\0', sizeof(short_opts));
 463         for (; opts->type != OPTION_END; opts++) {
 464                 if ((opts->flags & PARSE_OPT_LASTARG_DEFAULT) &&
 465                     (opts->flags & PARSE_OPT_OPTARG))
 466                         optbug(opts, "uses incompatible flags "
 467                                "LASTARG_DEFAULT and OPTARG");
 468                 if (opts->short_name) {
 469                         if (0x7F <= opts->short_name)
 470                                 optbug(opts, "invalid short name");
 471                         else if (short_opts[opts->short_name]++)
 472                                 optbug(opts, "short name already used");
 473                 }
 474                 if (opts->flags & PARSE_OPT_NODASH &&
 475                     ((opts->flags & PARSE_OPT_OPTARG) ||
 476                      !(opts->flags & PARSE_OPT_NOARG) ||
 477                      !(opts->flags & PARSE_OPT_NONEG) ||
 478                      opts->long_name))
 479                         optbug(opts, "uses feature "
 480                                "not supported for dashless options");
 481                 switch (opts->type) {
 482                 case OPTION_COUNTUP:
 483                 case OPTION_BIT:
 484                 case OPTION_NEGBIT:
 485                 case OPTION_SET_INT:
 486                 case OPTION_NUMBER:
 487                         if ((opts->flags & PARSE_OPT_OPTARG) ||
 488                             !(opts->flags & PARSE_OPT_NOARG))
 489                                 optbug(opts, "should not accept an argument");
 490                         break;
 491                 case OPTION_CALLBACK:
 492                         if (!opts->callback && !opts->ll_callback)
 493                                 optbug(opts, "OPTION_CALLBACK needs one callback");
 494                         else if (opts->callback && opts->ll_callback)
 495                                 optbug(opts, "OPTION_CALLBACK can't have two callbacks");
 496                         break;
 497                 case OPTION_LOWLEVEL_CALLBACK:
 498                         if (!opts->ll_callback)
 499                                 optbug(opts, "OPTION_LOWLEVEL_CALLBACK needs a callback");
 500                         if (opts->callback)
 501                                 optbug(opts, "OPTION_LOWLEVEL_CALLBACK needs no high level callback");
 502                         break;
 503                 case OPTION_ALIAS:
 504                         optbug(opts, "OPT_ALIAS() should not remain at this point. "
 505                                "Are you using parse_options_step() directly?\n"
 506                                "That case is not supported yet.");
 507                         break;
 508                 case OPTION_SUBCOMMAND:
 509                         if (!opts->value || !opts->subcommand_fn)
 510                                 optbug(opts, "OPTION_SUBCOMMAND needs a value and a subcommand function");
 511                         if (!subcommand_value)
 512                                 subcommand_value = opts->value;
 513                         else if (subcommand_value != opts->value)
 514                                 optbug(opts, "all OPTION_SUBCOMMANDs need the same value");
 515                         break;
 516                 default:
 517                         ; /* ok. (usually accepts an argument) */
 518                 }
 519                 if (opts->argh &&
 520                     strcspn(opts->argh, " _") != strlen(opts->argh))
 521                         optbug(opts, "multi-word argh should use dash to separate words");
 522         }
 523         BUG_if_bug("invalid 'struct option'");
 524 }
 525
 526 static int has_subcommands(const struct option *options)
 527 {
 528         for (; options->type != OPTION_END; options++)
 529                 if (options->type == OPTION_SUBCOMMAND)
 530                         return 1;
 531         return 0;
 532 }
 533
 534 static void parse_options_start_1(struct parse_opt_ctx_t *ctx,
 535                                   int argc, const char **argv, const char *prefix,
 536                                   const struct option *options,
 537                                   enum parse_opt_flags flags)
 538 {
 539         ctx->argc = argc;
 540         ctx->argv = argv;
 541         if (!(flags & PARSE_OPT_ONE_SHOT)) {
 542                 ctx->argc--;
 543                 ctx->argv++;
 544         }
 545         ctx->total = ctx->argc;
 546         ctx->out   = argv;
 547         ctx->prefix = prefix;
 548         ctx->cpidx = ((flags & PARSE_OPT_KEEP_ARGV0) != 0);
 549         ctx->flags = flags;
 550         ctx->has_subcommands = has_subcommands(options);
 551         if (!ctx->has_subcommands && (flags & PARSE_OPT_SUBCOMMAND_OPTIONAL))
 552                 BUG("Using PARSE_OPT_SUBCOMMAND_OPTIONAL without subcommands");
 553         if (ctx->has_subcommands) {
 554                 if (flags & PARSE_OPT_STOP_AT_NON_OPTION)
 555                         BUG("subcommands are incompatible with PARSE_OPT_STOP_AT_NON_OPTION");
 556                 if (!(flags & PARSE_OPT_SUBCOMMAND_OPTIONAL)) {
 557                         if (flags & PARSE_OPT_KEEP_UNKNOWN_OPT)
 558                                 BUG("subcommands are incompatible with PARSE_OPT_KEEP_UNKNOWN_OPT unless in combination with PARSE_OPT_SUBCOMMAND_OPTIONAL");
 559                         if (flags & PARSE_OPT_KEEP_DASHDASH)
 560                                 BUG("subcommands are incompatible with PARSE_OPT_KEEP_DASHDASH unless in combination with PARSE_OPT_SUBCOMMAND_OPTIONAL");
 561                 }
 562         }
 563         if ((flags & PARSE_OPT_KEEP_UNKNOWN_OPT) &&
 564             (flags & PARSE_OPT_STOP_AT_NON_OPTION) &&
 565             !(flags & PARSE_OPT_ONE_SHOT))
 566                 BUG("STOP_AT_NON_OPTION and KEEP_UNKNOWN don't go together");
 567         if ((flags & PARSE_OPT_ONE_SHOT) &&
 568             (flags & PARSE_OPT_KEEP_ARGV0))
 569                 BUG("Can't keep argv0 if you don't have it");
 570         parse_options_check(options);
 571 }
 572
 573 void parse_options_start(struct parse_opt_ctx_t *ctx,
 574                          int argc, const char **argv, const char *prefix,
 575                          const struct option *options,
 576                          enum parse_opt_flags flags)
 577 {
 578         memset(ctx, 0, sizeof(*ctx));
 579         parse_options_start_1(ctx, argc, argv, prefix, options, flags);
 580 }
 581
 582 static void show_negated_gitcomp(const struct option *opts, int show_all,
 583                                  int nr_noopts)
 584 {
 585         int printed_dashdash = 0;
 586
 587         for (; opts->type != OPTION_END; opts++) {
 588                 int has_unset_form = 0;
 589                 const char *name;
 590
 591                 if (!opts->long_name)
 592                         continue;
 593                 if (!show_all &&
 594                         (opts->flags & (PARSE_OPT_HIDDEN | PARSE_OPT_NOCOMPLETE)))
 595                         continue;
 596                 if (opts->flags & PARSE_OPT_NONEG)
 597                         continue;
 598
 599                 switch (opts->type) {
 600                 case OPTION_STRING:
 601                 case OPTION_FILENAME:
 602                 case OPTION_INTEGER:
 603                 case OPTION_MAGNITUDE:
 604                 case OPTION_CALLBACK:
 605                 case OPTION_BIT:
 606                 case OPTION_NEGBIT:
 607                 case OPTION_COUNTUP:
 608                 case OPTION_SET_INT:
 609                         has_unset_form = 1;
 610                         break;
 611                 default:
 612                         break;
 613                 }
 614                 if (!has_unset_form)
 615                         continue;
 616
 617                 if (skip_prefix(opts->long_name, "no-", &name)) {
 618                         if (nr_noopts < 0)
 619                                 printf(" --%s", name);
 620                 } else if (nr_noopts >= 0) {
 621                         if (nr_noopts && !printed_dashdash) {
 622                                 printf(" --");
 623                                 printed_dashdash = 1;
 624                         }
 625                         printf(" --no-%s", opts->long_name);
 626                         nr_noopts++;
 627                 }
 628         }
 629 }
 630
 631 static int show_gitcomp(const struct option *opts, int show_all)
 632 {
 633         const struct option *original_opts = opts;
 634         int nr_noopts = 0;
 635
 636         for (; opts->type != OPTION_END; opts++) {
 637                 const char *prefix = "--";
 638                 const char *suffix = "";
 639
 640                 if (!opts->long_name)
 641                         continue;
 642                 if (!show_all &&
 643                         (opts->flags & (PARSE_OPT_HIDDEN | PARSE_OPT_NOCOMPLETE | PARSE_OPT_FROM_ALIAS)))
 644                         continue;
 645
 646                 switch (opts->type) {
 647                 case OPTION_SUBCOMMAND:
 648                         prefix = "";
 649                         break;
 650                 case OPTION_GROUP:
 651                         continue;
 652                 case OPTION_STRING:
 653                 case OPTION_FILENAME:
 654                 case OPTION_INTEGER:
 655                 case OPTION_MAGNITUDE:
 656                 case OPTION_CALLBACK:
 657                         if (opts->flags & PARSE_OPT_NOARG)
 658                                 break;
 659                         if (opts->flags & PARSE_OPT_OPTARG)
 660                                 break;
 661                         if (opts->flags & PARSE_OPT_LASTARG_DEFAULT)
 662                                 break;
 663                         suffix = "=";
 664                         break;
 665                 default:
 666                         break;
 667                 }
 668                 if (opts->flags & PARSE_OPT_COMP_ARG)
 669                         suffix = "=";
 670                 if (starts_with(opts->long_name, "no-"))
 671                         nr_noopts++;
 672                 printf("%s%s%s%s", opts == original_opts ? "" : " ",
 673                        prefix, opts->long_name, suffix);
 674         }
 675         show_negated_gitcomp(original_opts, show_all, -1);
 676         show_negated_gitcomp(original_opts, show_all, nr_noopts);
 677         fputc('\n', stdout);
 678         return PARSE_OPT_COMPLETE;
 679 }
 680
 681 /*
 682  * Scan and may produce a new option[] array, which should be used
 683  * instead of the original 'options'.
 684  *
 685  * Right now this is only used to preprocess and substitute
 686  * OPTION_ALIAS.
 687  *
 688  * The returned options should be freed using free_preprocessed_options.
 689  */
 690 static struct option *preprocess_options(struct parse_opt_ctx_t *ctx,
 691                                          const struct option *options)
 692 {
 693         struct option *newopt;
 694         int i, nr, alias;
 695         int nr_aliases = 0;
 696
 697         for (nr = 0; options[nr].type != OPTION_END; nr++) {
 698                 if (options[nr].type == OPTION_ALIAS)
 699                         nr_aliases++;
 700         }
 701
 702         if (!nr_aliases)
 703                 return NULL;
 704
 705         DUP_ARRAY(newopt, options, nr + 1);
 706
 707         /* each alias has two string pointers and NULL */
 708         CALLOC_ARRAY(ctx->alias_groups, 3 * (nr_aliases + 1));
 709
 710         for (alias = 0, i = 0; i < nr; i++) {
 711                 int short_name;
 712                 const char *long_name;
 713                 const char *source;
 714                 struct strbuf help = STRBUF_INIT;
 715                 int j;
 716
 717                 if (newopt[i].type != OPTION_ALIAS)
 718                         continue;
 719
 720                 short_name = newopt[i].short_name;
 721                 long_name = newopt[i].long_name;
 722                 source = newopt[i].value;
 723
 724                 if (!long_name)
 725                         BUG("An alias must have long option name");
 726                 strbuf_addf(&help, _("alias of --%s"), source);
 727
 728                 for (j = 0; j < nr; j++) {
 729                         const char *name = options[j].long_name;
 730
 731                         if (!name || strcmp(name, source))
 732                                 continue;
 733
 734                         if (options[j].type == OPTION_ALIAS)
 735                                 BUG("No please. Nested aliases are not supported.");
 736
 737                         memcpy(newopt + i, options + j, sizeof(*newopt));
 738                         newopt[i].short_name = short_name;
 739                         newopt[i].long_name = long_name;
 740                         newopt[i].help = strbuf_detach(&help, NULL);
 741                         newopt[i].flags |= PARSE_OPT_FROM_ALIAS;
 742                         break;
 743                 }
 744
 745                 if (j == nr)
 746                         BUG("could not find source option '%s' of alias '%s'",
 747                             source, newopt[i].long_name);
 748                 ctx->alias_groups[alias * 3 + 0] = newopt[i].long_name;
 749                 ctx->alias_groups[alias * 3 + 1] = options[j].long_name;
 750                 ctx->alias_groups[alias * 3 + 2] = NULL;
 751                 alias++;
 752         }
 753
 754         return newopt;
 755 }
 756
 757 static void free_preprocessed_options(struct option *options)
 758 {
 759         int i;
 760
 761         if (!options)
 762                 return;
 763
 764         for (i = 0; options[i].type != OPTION_END; i++) {
 765                 if (options[i].flags & PARSE_OPT_FROM_ALIAS)
 766                         free((void *)options[i].help);
 767         }
 768         free(options);
 769 }
 770
 771 static enum parse_opt_result usage_with_options_internal(struct parse_opt_ctx_t *,
 772                                                          const char * const *,
 773                                                          const struct option *,
 774                                                          int, int);
 775
 776 enum parse_opt_result parse_options_step(struct parse_opt_ctx_t *ctx,
 777                                          const struct option *options,
 778                                          const char * const usagestr[])
 779 {
 780         int internal_help = !(ctx->flags & PARSE_OPT_NO_INTERNAL_HELP);
 781
 782         /* we must reset ->opt, unknown short option leave it dangling */
 783         ctx->opt = NULL;
 784
 785         for (; ctx->argc; ctx->argc--, ctx->argv++) {
 786                 const char *arg = ctx->argv[0];
 787
 788                 if (ctx->flags & PARSE_OPT_ONE_SHOT &&
 789                     ctx->argc != ctx->total)
 790                         break;
 791
 792                 if (*arg != '-' || !arg[1]) {
 793                         if (parse_nodash_opt(ctx, arg, options) == 0)
 794                                 continue;
 795                         if (!ctx->has_subcommands) {
 796                                 if (ctx->flags & PARSE_OPT_STOP_AT_NON_OPTION)
 797                                         return PARSE_OPT_NON_OPTION;
 798                                 ctx->out[ctx->cpidx++] = ctx->argv[0];
 799                                 continue;
 800                         }
 801                         switch (parse_subcommand(arg, options)) {
 802                         case PARSE_OPT_SUBCOMMAND:
 803                                 return PARSE_OPT_SUBCOMMAND;
 804                         case PARSE_OPT_UNKNOWN:
 805                                 if (ctx->flags & PARSE_OPT_SUBCOMMAND_OPTIONAL)
 806                                         /*
 807                                          * arg is neither a short or long
 808                                          * option nor a subcommand.  Since
 809                                          * this command has a default
 810                                          * operation mode, we have to treat
 811                                          * this arg and all remaining args
 812                                          * as args meant to that default
 813                                          * operation mode.
 814                                          * So we are done parsing.
 815                                          */
 816                                         return PARSE_OPT_DONE;
 817                                 error(_("unknown subcommand: `%s'"), arg);
 818                                 usage_with_options(usagestr, options);
 819                         case PARSE_OPT_COMPLETE:
 820                         case PARSE_OPT_HELP:
 821                         case PARSE_OPT_ERROR:
 822                         case PARSE_OPT_DONE:
 823                         case PARSE_OPT_NON_OPTION:
 824                                 /* Impossible. */
 825                                 BUG("parse_subcommand() cannot return these");
 826                         }
 827                 }
 828
 829                 /* lone -h asks for help */
 830                 if (internal_help && ctx->total == 1 && !strcmp(arg + 1, "h"))
 831                         goto show_usage;
 832
 833                 /*
 834                  * lone --git-completion-helper and --git-completion-helper-all
 835                  * are asked by git-completion.bash
 836                  */
 837                 if (ctx->total == 1 && !strcmp(arg, "--git-completion-helper"))
 838                         return show_gitcomp(options, 0);
 839                 if (ctx->total == 1 && !strcmp(arg, "--git-completion-helper-all"))
 840                         return show_gitcomp(options, 1);
 841
 842                 if (arg[1] != '-') {
 843                         ctx->opt = arg + 1;
 844                         switch (parse_short_opt(ctx, options)) {
 845                         case PARSE_OPT_ERROR:
 846                                 return PARSE_OPT_ERROR;
 847                         case PARSE_OPT_UNKNOWN:
 848                                 if (ctx->opt)
 849                                         check_typos(arg + 1, options);
 850                                 if (internal_help && *ctx->opt == 'h')
 851                                         goto show_usage;
 852                                 goto unknown;
 853                         case PARSE_OPT_NON_OPTION:
 854                         case PARSE_OPT_SUBCOMMAND:
 855                         case PARSE_OPT_HELP:
 856                         case PARSE_OPT_COMPLETE:
 857                                 BUG("parse_short_opt() cannot return these");
 858                         case PARSE_OPT_DONE:
 859                                 break;
 860                         }
 861                         if (ctx->opt)
 862                                 check_typos(arg + 1, options);
 863                         while (ctx->opt) {
 864                                 switch (parse_short_opt(ctx, options)) {
 865                                 case PARSE_OPT_ERROR:
 866                                         return PARSE_OPT_ERROR;
 867                                 case PARSE_OPT_UNKNOWN:
 868                                         if (internal_help && *ctx->opt == 'h')
 869                                                 goto show_usage;
 870
 871                                         /* fake a short option thing to hide the fact that we may have
 872                                          * started to parse aggregated stuff
 873                                          *
 874                                          * This is leaky, too bad.
 875                                          */
 876                                         ctx->argv[0] = xstrdup(ctx->opt - 1);
 877                                         *(char *)ctx->argv[0] = '-';
 878                                         goto unknown;
 879                                 case PARSE_OPT_NON_OPTION:
 880                                 case PARSE_OPT_SUBCOMMAND:
 881                                 case PARSE_OPT_COMPLETE:
 882                                 case PARSE_OPT_HELP:
 883                                         BUG("parse_short_opt() cannot return these");
 884                                 case PARSE_OPT_DONE:
 885                                         break;
 886                                 }
 887                         }
 888                         continue;
 889                 }
 890
 891                 if (!arg[2] /* "--" */ ||
 892                     !strcmp(arg + 2, "end-of-options")) {
 893                         if (!(ctx->flags & PARSE_OPT_KEEP_DASHDASH)) {
 894                                 ctx->argc--;
 895                                 ctx->argv++;
 896                         }
 897                         break;
 898                 }
 899
 900                 if (internal_help && !strcmp(arg + 2, "help-all"))
 901                         return usage_with_options_internal(ctx, usagestr, options, 1, 0);
 902                 if (internal_help && !strcmp(arg + 2, "help"))
 903                         goto show_usage;
 904                 switch (parse_long_opt(ctx, arg + 2, options)) {
 905                 case PARSE_OPT_ERROR:
 906                         return PARSE_OPT_ERROR;
 907                 case PARSE_OPT_UNKNOWN:
 908                         goto unknown;
 909                 case PARSE_OPT_HELP:
 910                         goto show_usage;
 911                 case PARSE_OPT_NON_OPTION:
 912                 case PARSE_OPT_SUBCOMMAND:
 913                 case PARSE_OPT_COMPLETE:
 914                         BUG("parse_long_opt() cannot return these");
 915                 case PARSE_OPT_DONE:
 916                         break;
 917                 }
 918                 continue;
 919 unknown:
 920                 if (ctx->flags & PARSE_OPT_ONE_SHOT)
 921                         break;
 922                 if (ctx->has_subcommands &&
 923                     (ctx->flags & PARSE_OPT_SUBCOMMAND_OPTIONAL) &&
 924                     (ctx->flags & PARSE_OPT_KEEP_UNKNOWN_OPT)) {
 925                         /*
 926                          * Found an unknown option given to a command with
 927                          * subcommands that has a default operation mode:
 928                          * we treat this option and all remaining args as
 929                          * arguments meant to that default operation mode.
 930                          * So we are done parsing.
 931                          */
 932                         return PARSE_OPT_DONE;
 933                 }
 934                 if (!(ctx->flags & PARSE_OPT_KEEP_UNKNOWN_OPT))
 935                         return PARSE_OPT_UNKNOWN;
 936                 ctx->out[ctx->cpidx++] = ctx->argv[0];
 937                 ctx->opt = NULL;
 938         }
 939         return PARSE_OPT_DONE;
 940
 941  show_usage:
 942         return usage_with_options_internal(ctx, usagestr, options, 0, 0);
 943 }
 944
 945 int parse_options_end(struct parse_opt_ctx_t *ctx)
 946 {
 947         if (ctx->flags & PARSE_OPT_ONE_SHOT)
 948                 return ctx->total - ctx->argc;
 949
 950         MOVE_ARRAY(ctx->out + ctx->cpidx, ctx->argv, ctx->argc);
 951         ctx->out[ctx->cpidx + ctx->argc] = NULL;
 952         return ctx->cpidx + ctx->argc;
 953 }
 954
 955 int parse_options(int argc, const char **argv,
 956                   const char *prefix,
 957                   const struct option *options,
 958                   const char * const usagestr[],
 959                   enum parse_opt_flags flags)
 960 {
 961         struct parse_opt_ctx_t ctx;
 962         struct option *real_options;
 963
 964         disallow_abbreviated_options =
 965                 git_env_bool("GIT_TEST_DISALLOW_ABBREVIATED_OPTIONS", 0);
 966
 967         memset(&ctx, 0, sizeof(ctx));
 968         real_options = preprocess_options(&ctx, options);
 969         if (real_options)
 970                 options = real_options;
 971         parse_options_start_1(&ctx, argc, argv, prefix, options, flags);
 972         switch (parse_options_step(&ctx, options, usagestr)) {
 973         case PARSE_OPT_HELP:
 974         case PARSE_OPT_ERROR:
 975                 exit(129);
 976         case PARSE_OPT_COMPLETE:
 977                 exit(0);
 978         case PARSE_OPT_NON_OPTION:
 979         case PARSE_OPT_SUBCOMMAND:
 980                 break;
 981         case PARSE_OPT_DONE:
 982                 if (ctx.has_subcommands &&
 983                     !(flags & PARSE_OPT_SUBCOMMAND_OPTIONAL)) {
 984                         error(_("need a subcommand"));
 985                         usage_with_options(usagestr, options);
 986                 }
 987                 break;
 988         case PARSE_OPT_UNKNOWN:
 989                 if (ctx.argv[0][1] == '-') {
 990                         error(_("unknown option `%s'"), ctx.argv[0] + 2);
 991                 } else if (isascii(*ctx.opt)) {
 992                         error(_("unknown switch `%c'"), *ctx.opt);
 993                 } else {
 994                         error(_("unknown non-ascii option in string: `%s'"),
 995                               ctx.argv[0]);
 996                 }
 997                 usage_with_options(usagestr, options);
 998         }
 999
1000         precompose_argv_prefix(argc, argv, NULL);
1001         free_preprocessed_options(real_options);
1002         free(ctx.alias_groups);
1003         return parse_options_end(&ctx);
1004 }
1005
1006 static int usage_argh(const struct option *opts, FILE *outfile)
1007 {
1008         const char *s;
1009         int literal = (opts->flags & PARSE_OPT_LITERAL_ARGHELP) ||
1010                 !opts->argh || !!strpbrk(opts->argh, "()<>[]|");
1011         if (opts->flags & PARSE_OPT_OPTARG)
1012                 if (opts->long_name)
1013                         s = literal ? "[=%s]" : "[=<%s>]";
1014                 else
1015                         s = literal ? "[%s]" : "[<%s>]";
1016         else
1017                 s = literal ? " %s" : " <%s>";
1018         return utf8_fprintf(outfile, s, opts->argh ? _(opts->argh) : _("..."));
1019 }
1020
1021 #define USAGE_OPTS_WIDTH 24
1022 #define USAGE_GAP         2
1023
1024 static enum parse_opt_result usage_with_options_internal(struct parse_opt_ctx_t *ctx,
1025                                                          const char * const *usagestr,
1026                                                          const struct option *opts,
1027                                                          int full, int err)
1028 {
1029         FILE *outfile = err ? stderr : stdout;
1030         int need_newline;
1031
1032         const char *usage_prefix = _("usage: %s");
1033         /*
1034          * The translation could be anything, but we can count on
1035          * msgfmt(1)'s --check option to have asserted that "%s" is in
1036          * the translation. So compute the length of the "usage: "
1037          * part. We are assuming that the translator wasn't overly
1038          * clever and used e.g. "%1$s" instead of "%s", there's only
1039          * one "%s" in "usage_prefix" above, so there's no reason to
1040          * do so even with a RTL language.
1041          */
1042         size_t usage_len = strlen(usage_prefix) - strlen("%s");
1043         /*
1044          * TRANSLATORS: the colon here should align with the
1045          * one in "usage: %s" translation.
1046          */
1047         const char *or_prefix = _("   or: %s");
1048         /*
1049          * TRANSLATORS: You should only need to translate this format
1050          * string if your language is a RTL language (e.g. Arabic,
1051          * Hebrew etc.), not if it's a LTR language (e.g. German,
1052          * Russian, Chinese etc.).
1053          *
1054          * When a translated usage string has an embedded "\n" it's
1055          * because options have wrapped to the next line. The line
1056          * after the "\n" will then be padded to align with the
1057          * command name, such as N_("git cmd [opt]\n<8
1058          * spaces>[opt2]"), where the 8 spaces are the same length as
1059          * "git cmd ".
1060          *
1061          * This format string prints out that already-translated
1062          * line. The "%*s" is whitespace padding to account for the
1063          * padding at the start of the line that we add in this
1064          * function. The "%s" is a line in the (hopefully already
1065          * translated) N_() usage string, which contained embedded
1066          * newlines before we split it up.
1067          */
1068         const char *usage_continued = _("%*s%s");
1069         const char *prefix = usage_prefix;
1070         int saw_empty_line = 0;
1071
1072         if (!usagestr)
1073                 return PARSE_OPT_HELP;
1074
1075         if (!err && ctx && ctx->flags & PARSE_OPT_SHELL_EVAL)
1076                 fprintf(outfile, "cat <<\\EOF\n");
1077
1078         while (*usagestr) {
1079                 const char *str = _(*usagestr++);
1080                 struct string_list list = STRING_LIST_INIT_DUP;
1081                 unsigned int j;
1082
1083                 if (!saw_empty_line && !*str)
1084                         saw_empty_line = 1;
1085
1086                 string_list_split(&list, str, '\n', -1);
1087                 for (j = 0; j < list.nr; j++) {
1088                         const char *line = list.items[j].string;
1089
1090                         if (saw_empty_line && *line)
1091                                 fprintf_ln(outfile, _("    %s"), line);
1092                         else if (saw_empty_line)
1093                                 fputc('\n', outfile);
1094                         else if (!j)
1095                                 fprintf_ln(outfile, prefix, line);
1096                         else
1097                                 fprintf_ln(outfile, usage_continued,
1098                                            (int)usage_len, "", line);
1099                 }
1100                 string_list_clear(&list, 0);
1101
1102                 prefix = or_prefix;
1103         }
1104
1105         need_newline = 1;
1106
1107         for (; opts->type != OPTION_END; opts++) {
1108                 size_t pos;
1109                 int pad;
1110
1111                 if (opts->type == OPTION_SUBCOMMAND)
1112                         continue;
1113                 if (opts->type == OPTION_GROUP) {
1114                         fputc('\n', outfile);
1115                         need_newline = 0;
1116                         if (*opts->help)
1117                                 fprintf(outfile, "%s\n", _(opts->help));
1118                         continue;
1119                 }
1120                 if (!full && (opts->flags & PARSE_OPT_HIDDEN))
1121                         continue;
1122
1123                 if (need_newline) {
1124                         fputc('\n', outfile);
1125                         need_newline = 0;
1126                 }
1127
1128                 pos = fprintf(outfile, "    ");
1129                 if (opts->short_name) {
1130                         if (opts->flags & PARSE_OPT_NODASH)
1131                                 pos += fprintf(outfile, "%c", opts->short_name);
1132                         else
1133                                 pos += fprintf(outfile, "-%c", opts->short_name);
1134                 }
1135                 if (opts->long_name && opts->short_name)
1136                         pos += fprintf(outfile, ", ");
1137                 if (opts->long_name)
1138                         pos += fprintf(outfile, "--%s", opts->long_name);
1139                 if (opts->type == OPTION_NUMBER)
1140                         pos += utf8_fprintf(outfile, _("-NUM"));
1141
1142                 if ((opts->flags & PARSE_OPT_LITERAL_ARGHELP) ||
1143                     !(opts->flags & PARSE_OPT_NOARG))
1144                         pos += usage_argh(opts, outfile);
1145
1146                 if (pos <= USAGE_OPTS_WIDTH)
1147                         pad = USAGE_OPTS_WIDTH - pos;
1148                 else {
1149                         fputc('\n', outfile);
1150                         pad = USAGE_OPTS_WIDTH;
1151                 }
1152                 if (opts->type == OPTION_ALIAS) {
1153                         fprintf(outfile, "%*s", pad + USAGE_GAP, "");
1154                         fprintf_ln(outfile, _("alias of --%s"),
1155                                    (const char *)opts->value);
1156                         continue;
1157                 }
1158                 fprintf(outfile, "%*s%s\n", pad + USAGE_GAP, "", _(opts->help));
1159         }
1160         fputc('\n', outfile);
1161
1162         if (!err && ctx && ctx->flags & PARSE_OPT_SHELL_EVAL)
1163                 fputs("EOF\n", outfile);
1164
1165         return PARSE_OPT_HELP;
1166 }
1167
1168 void NORETURN usage_with_options(const char * const *usagestr,
1169                         const struct option *opts)
1170 {
1171         usage_with_options_internal(NULL, usagestr, opts, 0, 1);
1172         exit(129);
1173 }
1174
1175 void NORETURN usage_msg_opt(const char *msg,
1176                    const char * const *usagestr,
1177                    const struct option *options)
1178 {
1179         die_message("%s\n", msg); /* The extra \n is intentional */
1180         usage_with_options(usagestr, options);
1181 }
1182
1183 void NORETURN usage_msg_optf(const char * const fmt,
1184                              const char * const *usagestr,
1185                              const struct option *options, ...)
1186 {
1187         struct strbuf msg = STRBUF_INIT;
1188         va_list ap;
1189         va_start(ap, options);
1190         strbuf_vaddf(&msg, fmt, ap);
1191         va_end(ap);
1192
1193         usage_msg_opt(msg.buf, usagestr, options);
1194 }
1195
1196 void die_for_incompatible_opt4(int opt1, const char *opt1_name,
1197                                int opt2, const char *opt2_name,
1198                                int opt3, const char *opt3_name,
1199                                int opt4, const char *opt4_name)
1200 {
1201         int count = 0;
1202         const char *options[4];
1203
1204         if (opt1)
1205                 options[count++] = opt1_name;
1206         if (opt2)
1207                 options[count++] = opt2_name;
1208         if (opt3)
1209                 options[count++] = opt3_name;
1210         if (opt4)
1211                 options[count++] = opt4_name;
1212         switch (count) {
1213         case 4:
1214                 die(_("options '%s', '%s', '%s', and '%s' cannot be used together"),
1215                     opt1_name, opt2_name, opt3_name, opt4_name);
1216                 break;
1217         case 3:
1218                 die(_("options '%s', '%s', and '%s' cannot be used together"),
1219                     options[0], options[1], options[2]);
1220                 break;
1221         case 2:
1222                 die(_("options '%s' and '%s' cannot be used together"),
1223                     options[0], options[1]);
1224                 break;
1225         default:
1226                 break;
1227         }
1228 }