parse-options.h 9.1 KB

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