parse-options.h 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  1. /* SPDX-License-Identifier: GPL-2.0 */
  2. #ifndef __SUBCMD_PARSE_OPTIONS_H
  3. #define __SUBCMD_PARSE_OPTIONS_H
  4. #include <linux/kernel.h>
  5. #include <stdbool.h>
  6. #include <stdint.h>
  7. #ifndef NORETURN
  8. #define NORETURN __attribute__((__noreturn__))
  9. #endif
  10. enum parse_opt_type {
  11. /* special types */
  12. OPTION_END,
  13. OPTION_ARGUMENT,
  14. OPTION_GROUP,
  15. /* options with no arguments */
  16. OPTION_BIT,
  17. OPTION_BOOLEAN,
  18. OPTION_INCR,
  19. OPTION_SET_UINT,
  20. OPTION_SET_PTR,
  21. /* options with arguments (usually) */
  22. OPTION_STRING,
  23. OPTION_INTEGER,
  24. OPTION_LONG,
  25. OPTION_ULONG,
  26. OPTION_CALLBACK,
  27. OPTION_U64,
  28. OPTION_UINTEGER,
  29. };
  30. enum parse_opt_flags {
  31. PARSE_OPT_KEEP_DASHDASH = 1,
  32. PARSE_OPT_STOP_AT_NON_OPTION = 2,
  33. PARSE_OPT_KEEP_ARGV0 = 4,
  34. PARSE_OPT_KEEP_UNKNOWN = 8,
  35. PARSE_OPT_NO_INTERNAL_HELP = 16,
  36. };
  37. enum parse_opt_option_flags {
  38. PARSE_OPT_OPTARG = 1,
  39. PARSE_OPT_NOARG = 2,
  40. PARSE_OPT_NONEG = 4,
  41. PARSE_OPT_HIDDEN = 8,
  42. PARSE_OPT_LASTARG_DEFAULT = 16,
  43. PARSE_OPT_DISABLED = 32,
  44. PARSE_OPT_EXCLUSIVE = 64,
  45. PARSE_OPT_NOEMPTY = 128,
  46. PARSE_OPT_NOBUILD = 256,
  47. PARSE_OPT_CANSKIP = 512,
  48. };
  49. struct option;
  50. typedef int parse_opt_cb(const struct option *, const char *arg, int unset);
  51. /*
  52. * `type`::
  53. * holds the type of the option, you must have an OPTION_END last in your
  54. * array.
  55. *
  56. * `short_name`::
  57. * the character to use as a short option name, '\0' if none.
  58. *
  59. * `long_name`::
  60. * the long option name, without the leading dashes, NULL if none.
  61. *
  62. * `value`::
  63. * stores pointers to the values to be filled.
  64. *
  65. * `argh`::
  66. * token to explain the kind of argument this option wants. Keep it
  67. * homogeneous across the repository.
  68. *
  69. * `help`::
  70. * the short help associated to what the option does.
  71. * Must never be NULL (except for OPTION_END).
  72. * OPTION_GROUP uses this pointer to store the group header.
  73. *
  74. * `flags`::
  75. * mask of parse_opt_option_flags.
  76. * PARSE_OPT_OPTARG: says that the argument is optional (not for BOOLEANs)
  77. * PARSE_OPT_NOARG: says that this option takes no argument, for CALLBACKs
  78. * PARSE_OPT_NONEG: says that this option cannot be negated
  79. * PARSE_OPT_HIDDEN this option is skipped in the default usage, showed in
  80. * the long one.
  81. *
  82. * `callback`::
  83. * pointer to the callback to use for OPTION_CALLBACK.
  84. *
  85. * `defval`::
  86. * default value to fill (*->value) with for PARSE_OPT_OPTARG.
  87. * OPTION_{BIT,SET_UINT,SET_PTR} store the {mask,integer,pointer} to put in
  88. * the value when met.
  89. * CALLBACKS can use it like they want.
  90. *
  91. * `set`::
  92. * whether an option was set by the user
  93. */
  94. struct option {
  95. enum parse_opt_type type;
  96. int short_name;
  97. const char *long_name;
  98. void *value;
  99. const char *argh;
  100. const char *help;
  101. const char *build_opt;
  102. int flags;
  103. parse_opt_cb *callback;
  104. intptr_t defval;
  105. bool *set;
  106. void *data;
  107. const struct option *parent;
  108. };
  109. #define check_vtype(v, type) ( BUILD_BUG_ON_ZERO(!__builtin_types_compatible_p(typeof(v), type)) + v )
  110. #define OPT_END() { .type = OPTION_END }
  111. #define OPT_PARENT(p) { .type = OPTION_END, .parent = (p) }
  112. #define OPT_ARGUMENT(l, h) { .type = OPTION_ARGUMENT, .long_name = (l), .help = (h) }
  113. #define OPT_GROUP(h) { .type = OPTION_GROUP, .help = (h) }
  114. #define OPT_BIT(s, l, v, h, b) { .type = OPTION_BIT, .short_name = (s), .long_name = (l), .value = check_vtype(v, int *), .help = (h), .defval = (b) }
  115. #define OPT_BOOLEAN(s, l, v, h) { .type = OPTION_BOOLEAN, .short_name = (s), .long_name = (l), .value = check_vtype(v, bool *), .help = (h) }
  116. #define OPT_BOOLEAN_FLAG(s, l, v, h, f) { .type = OPTION_BOOLEAN, .short_name = (s), .long_name = (l), .value = check_vtype(v, bool *), .help = (h), .flags = (f) }
  117. #define OPT_BOOLEAN_SET(s, l, v, os, h) \
  118. { .type = OPTION_BOOLEAN, .short_name = (s), .long_name = (l), \
  119. .value = check_vtype(v, bool *), .help = (h), \
  120. .set = check_vtype(os, bool *)}
  121. #define OPT_INCR(s, l, v, h) { .type = OPTION_INCR, .short_name = (s), .long_name = (l), .value = check_vtype(v, int *), .help = (h) }
  122. #define OPT_SET_UINT(s, l, v, h, i) { .type = OPTION_SET_UINT, .short_name = (s), .long_name = (l), .value = check_vtype(v, unsigned int *), .help = (h), .defval = (i) }
  123. #define OPT_SET_PTR(s, l, v, h, p) { .type = OPTION_SET_PTR, .short_name = (s), .long_name = (l), .value = (v), .help = (h), .defval = (p) }
  124. #define OPT_INTEGER(s, l, v, h) { .type = OPTION_INTEGER, .short_name = (s), .long_name = (l), .value = check_vtype(v, int *), .help = (h) }
  125. #define OPT_UINTEGER(s, l, v, h) { .type = OPTION_UINTEGER, .short_name = (s), .long_name = (l), .value = check_vtype(v, unsigned int *), .help = (h) }
  126. #define OPT_UINTEGER_OPTARG(s, l, v, d, h) { .type = OPTION_UINTEGER, .short_name = (s), .long_name = (l), .value = check_vtype(v, unsigned int *), .help = (h), .flags = PARSE_OPT_OPTARG, .defval = (intptr_t)(d) }
  127. #define OPT_LONG(s, l, v, h) { .type = OPTION_LONG, .short_name = (s), .long_name = (l), .value = check_vtype(v, long *), .help = (h) }
  128. #define OPT_ULONG(s, l, v, h) { .type = OPTION_ULONG, .short_name = (s), .long_name = (l), .value = check_vtype(v, unsigned long *), .help = (h) }
  129. #define OPT_U64(s, l, v, h) { .type = OPTION_U64, .short_name = (s), .long_name = (l), .value = check_vtype(v, u64 *), .help = (h) }
  130. #define OPT_STRING(s, l, v, a, h) { .type = OPTION_STRING, .short_name = (s), .long_name = (l), .value = check_vtype(v, const char **), .argh = (a), .help = (h) }
  131. #define OPT_STRING_OPTARG(s, l, v, a, h, d) \
  132. { .type = OPTION_STRING, .short_name = (s), .long_name = (l), \
  133. .value = check_vtype(v, const char **), .argh =(a), .help = (h), \
  134. .flags = PARSE_OPT_OPTARG, .defval = (intptr_t)(d) }
  135. #define OPT_STRING_OPTARG_SET(s, l, v, os, a, h, d) \
  136. { .type = OPTION_STRING, .short_name = (s), .long_name = (l), \
  137. .value = check_vtype(v, const char **), .argh = (a), .help = (h), \
  138. .flags = PARSE_OPT_OPTARG, .defval = (intptr_t)(d), \
  139. .set = check_vtype(os, bool *)}
  140. #define OPT_STRING_NOEMPTY(s, l, v, a, h) { .type = OPTION_STRING, .short_name = (s), .long_name = (l), .value = check_vtype(v, const char **), .argh = (a), .help = (h), .flags = PARSE_OPT_NOEMPTY}
  141. #define OPT_DATE(s, l, v, h) \
  142. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), .value = (v), .argh = "time", .help = (h), .callback = parse_opt_approxidate_cb }
  143. #define OPT_CALLBACK(s, l, v, a, h, f) \
  144. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), .value = (v), .argh = (a), .help = (h), .callback = (f) }
  145. #define OPT_CALLBACK_SET(s, l, v, os, a, h, f) \
  146. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), .value = (v), .argh = (a), .help = (h), .callback = (f), .set = check_vtype(os, bool *)}
  147. #define OPT_CALLBACK_NOOPT(s, l, v, a, h, f) \
  148. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), .value = (v), .argh = (a), .help = (h), .callback = (f), .flags = PARSE_OPT_NOARG }
  149. #define OPT_CALLBACK_DEFAULT(s, l, v, a, h, f, d) \
  150. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), .value = (v), .argh = (a), .help = (h), .callback = (f), .defval = (intptr_t)d, .flags = PARSE_OPT_LASTARG_DEFAULT }
  151. #define OPT_CALLBACK_DEFAULT_NOOPT(s, l, v, a, h, f, d) \
  152. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l),\
  153. .value = (v), .arg = (a), .help = (h), .callback = (f), .defval = (intptr_t)d,\
  154. .flags = PARSE_OPT_LASTARG_DEFAULT | PARSE_OPT_NOARG}
  155. #define OPT_CALLBACK_OPTARG(s, l, v, d, a, h, f) \
  156. { .type = OPTION_CALLBACK, .short_name = (s), .long_name = (l), \
  157. .value = (v), .argh = (a), .help = (h), .callback = (f), \
  158. .flags = PARSE_OPT_OPTARG, .data = (d) }
  159. /* parse_options() will filter out the processed options and leave the
  160. * non-option argments in argv[].
  161. * Returns the number of arguments left in argv[].
  162. *
  163. * NOTE: parse_options() and parse_options_subcommand() may call exit() in the
  164. * case of an error (or for 'special' options like --list-cmds or --list-opts).
  165. */
  166. extern int parse_options(int argc, const char **argv,
  167. const struct option *options,
  168. const char * const usagestr[], int flags);
  169. extern int parse_options_subcommand(int argc, const char **argv,
  170. const struct option *options,
  171. const char *const subcommands[],
  172. const char *usagestr[], int flags);
  173. extern NORETURN void usage_with_options(const char * const *usagestr,
  174. const struct option *options);
  175. extern NORETURN __attribute__((format(printf,3,4)))
  176. void usage_with_options_msg(const char * const *usagestr,
  177. const struct option *options,
  178. const char *fmt, ...);
  179. /*----- incremantal advanced APIs -----*/
  180. enum {
  181. PARSE_OPT_HELP = -1,
  182. PARSE_OPT_DONE,
  183. PARSE_OPT_LIST_OPTS,
  184. PARSE_OPT_LIST_SUBCMDS,
  185. PARSE_OPT_UNKNOWN,
  186. };
  187. /*
  188. * It's okay for the caller to consume argv/argc in the usual way.
  189. * Other fields of that structure are private to parse-options and should not
  190. * be modified in any way.
  191. */
  192. struct parse_opt_ctx_t {
  193. const char **argv;
  194. const char **out;
  195. int argc, cpidx;
  196. const char *opt;
  197. const struct option *excl_opt;
  198. int flags;
  199. };
  200. extern int parse_options_usage(const char * const *usagestr,
  201. const struct option *opts,
  202. const char *optstr,
  203. bool short_opt);
  204. /*----- some often used options -----*/
  205. extern int parse_opt_abbrev_cb(const struct option *, const char *, int);
  206. extern int parse_opt_approxidate_cb(const struct option *, const char *, int);
  207. extern int parse_opt_verbosity_cb(const struct option *, const char *, int);
  208. #define OPT__VERBOSE(var) OPT_BOOLEAN('v', "verbose", (var), "be verbose")
  209. #define OPT__QUIET(var) OPT_BOOLEAN('q', "quiet", (var), "be quiet")
  210. #define OPT__VERBOSITY(var) \
  211. { OPTION_CALLBACK, 'v', "verbose", (var), NULL, "be more verbose", \
  212. PARSE_OPT_NOARG, &parse_opt_verbosity_cb, 0 }, \
  213. { OPTION_CALLBACK, 'q', "quiet", (var), NULL, "be more quiet", \
  214. PARSE_OPT_NOARG, &parse_opt_verbosity_cb, 0 }
  215. #define OPT__DRY_RUN(var) OPT_BOOLEAN('n', "dry-run", (var), "dry run")
  216. #define OPT__ABBREV(var) \
  217. { OPTION_CALLBACK, 0, "abbrev", (var), "n", \
  218. "use <n> digits to display SHA-1s", \
  219. PARSE_OPT_OPTARG, &parse_opt_abbrev_cb, 0 }
  220. extern const char *parse_options_fix_filename(const char *prefix, const char *file);
  221. void set_option_flag(struct option *opts, int sopt, const char *lopt, int flag);
  222. void set_option_nobuild(struct option *opts, int shortopt, const char *longopt,
  223. const char *build_opt, bool can_skip);
  224. #endif /* __SUBCMD_PARSE_OPTIONS_H */