parse-options.h 8.3 KB

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