opt: add support for showing default value.
[ccan] / ccan / opt / test / utils.c
1 #define _GNU_SOURCE
2 #include <ccan/tap/tap.h>
3 #include <stdarg.h>
4 #include <stdlib.h>
5 #include <ccan/opt/opt.h>
6 #include <getopt.h>
7 #include <string.h>
8 #include <stdio.h>
9 #include "utils.h"
10
11 unsigned int test_cb_called;
12 char *test_noarg(void *arg)
13 {
14         test_cb_called++;
15         return NULL;
16 }
17
18 char *test_arg(const char *optarg, const char *arg)
19 {
20         test_cb_called++;
21         ok1(strcmp(optarg, arg) == 0);
22         return NULL;
23 }
24
25 void show_arg(char buf[OPT_SHOW_LEN], const char *arg)
26 {
27         strncpy(buf, arg, OPT_SHOW_LEN);
28 }
29
30 char *err_output = NULL;
31
32 static void save_err_output(const char *fmt, ...)
33 {
34         va_list ap;
35         char *p;
36
37         va_start(ap, fmt);
38         vasprintf(&p, fmt, ap);
39         va_end(ap);
40
41         if (err_output) {
42                 err_output = realloc(err_output,
43                                      strlen(err_output) + strlen(p) + 1);
44                 strcat(err_output, p);
45                 free(p);
46         } else
47                 err_output = p;
48 }       
49
50 /* FIXME: This leaks, BTW. */
51 bool parse_args(int *argc, char ***argv, ...)
52 {
53         char **a;
54         va_list ap;
55
56         va_start(ap, argv);
57         *argc = 1;
58         a = malloc(sizeof(*a) * (*argc + 1));
59         a[0] = (*argv)[0];
60         while ((a[*argc] = va_arg(ap, char *)) != NULL) {
61                 (*argc)++;
62                 a = realloc(a, sizeof(*a) * (*argc + 1));
63         }
64         *argv = a;
65         /* Re-set before parsing. */
66         optind = 0;
67
68         return opt_parse(argc, *argv, save_err_output);
69 }
70
71 struct opt_table short_table[] = {
72         /* Short opts, different args. */
73         { OPT_WITHOUT_ARG(NULL, 'a', test_noarg, "a"), "Description of a" },
74         { OPT_WITH_ARG(NULL, 'b', test_arg, show_arg, "b"), "Description of b" },
75         OPT_ENDTABLE
76 };
77
78 struct opt_table long_table[] = {
79         /* Long opts, different args. */
80         { OPT_WITHOUT_ARG("ddd", 0, test_noarg, "ddd"), "Description of ddd" },
81         { OPT_WITH_ARG("eee", 0, test_arg, show_arg, "eee"), },
82         OPT_ENDTABLE
83 };
84
85 struct opt_table long_and_short_table[] = {
86         /* Short and long, different args. */
87         { OPT_WITHOUT_ARG("ggg", 'g', test_noarg, "ggg"),
88           "Description of ggg" },
89         { OPT_WITH_ARG("hhh", 'h', test_arg, NULL, "hhh"),
90           "Description of hhh"},
91         OPT_ENDTABLE
92 };
93
94 /* Sub-table test. */
95 struct opt_table subtables[] = {
96         /* Short and long, no description */
97         { OPT_WITH_ARG("jjj", 'j', test_arg, show_arg, "jjj") },
98         OPT_SUBTABLE(short_table, NULL),
99         OPT_SUBTABLE(long_table, "long table options"),
100         OPT_SUBTABLE(long_and_short_table, opt_table_hidden),
101         OPT_ENDTABLE
102 };