0f3671e526d838264d78d8f68b2120acbaa71e44
[ccan] / ccan / opt / helpers.c
1 #include <ccan/opt/opt.h>
2 #include <string.h>
3 #include <stdlib.h>
4 #include <errno.h>
5 #include <stdio.h>
6 #include "private.h"
7
8 /* Upper bound to sprintf this simple type?  Each 3 bits < 1 digit. */
9 #define CHAR_SIZE(type) (((sizeof(type)*CHAR_BIT + 2) / 3) + 1)
10
11 /* FIXME: asprintf module? */
12 static char *arg_bad(const char *fmt, const char *arg)
13 {
14         char *str = malloc(strlen(fmt) + strlen(arg));
15         sprintf(str, fmt, arg);
16         return str;
17 }
18
19 char *opt_set_bool(bool *b)
20 {
21         *b = true;
22         return NULL;
23 }
24
25 char *opt_set_invbool(bool *b)
26 {
27         *b = false;
28         return NULL;
29 }
30
31 char *opt_set_bool_arg(const char *arg, bool *b)
32 {
33         if (!strcasecmp(arg, "yes") || !strcasecmp(arg, "true"))
34                 return opt_set_bool(b);
35         if (!strcasecmp(arg, "no") || !strcasecmp(arg, "false"))
36                 return opt_set_invbool(b);
37
38         return opt_invalid_argument(arg);
39 }
40
41 char *opt_set_invbool_arg(const char *arg, bool *b)
42 {
43         char *err = opt_set_bool_arg(arg, b);
44
45         if (!err)
46                 *b = !*b;
47         return err;
48 }
49
50 /* Set a char *. */
51 char *opt_set_charp(const char *arg, char **p)
52 {
53         *p = (char *)arg;
54         return NULL;
55 }
56
57 /* Set an integer value, various forms.  Sets to 1 on arg == NULL. */
58 char *opt_set_intval(const char *arg, int *i)
59 {
60         long l;
61         char *err = opt_set_longval(arg, &l);
62
63         if (err)
64                 return err;
65         *i = l;
66         /* Beware truncation... */
67         if (*i != l)
68                 return arg_bad("value '%s' does not fit into an integer", arg);
69         return err;
70 }
71
72 char *opt_set_uintval(const char *arg, unsigned int *ui)
73 {
74         int i;
75         char *err = opt_set_intval(arg, &i);
76
77         if (err)
78                 return err;
79         if (i < 0)
80                 return arg_bad("'%s' is negative", arg);
81         *ui = i;
82         return NULL;
83 }
84
85 char *opt_set_longval(const char *arg, long *l)
86 {
87         char *endp;
88
89         /* This is how the manpage says to do it.  Yech. */
90         errno = 0;
91         *l = strtol(arg, &endp, 0);
92         if (*endp || !arg[0])
93                 return arg_bad("'%s' is not a number", arg);
94         if (errno == ERANGE)
95                 return arg_bad("'%s' is out of range", arg);
96         if (errno)
97                 return opt_invalid_argument(arg);
98         return NULL;
99 }
100
101 char *opt_set_ulongval(const char *arg, unsigned long *ul)
102 {
103         long int l;
104         char *err;
105         
106         err = opt_set_longval(arg, &l);
107         if (err)
108                 return err;
109         *ul = l;
110         if (l < 0)
111                 return arg_bad("'%s' is negative", arg);
112         return NULL;
113 }
114
115 char *opt_inc_intval(int *i)
116 {
117         (*i)++;
118         return NULL;
119 }
120
121 /* Display version string. */
122 char *opt_version_and_exit(const char *version)
123 {
124         printf("%s\n", version);
125         exit(0);
126 }
127
128 char *opt_usage_and_exit(const char *extra)
129 {
130         printf("%s", opt_usage(opt_argv0, extra));
131         exit(0);
132 }
133
134 void opt_show_bool(char buf[OPT_SHOW_LEN], const bool *b)
135 {
136         strncpy(buf, *b ? "true" : "false", OPT_SHOW_LEN);
137 }
138
139 void opt_show_invbool(char buf[OPT_SHOW_LEN], const bool *b)
140 {
141         strncpy(buf, *b ? "false" : "true", OPT_SHOW_LEN);
142 }
143
144 void opt_show_charp(char buf[OPT_SHOW_LEN], char *const *p)
145 {
146         size_t len = strlen(*p);
147         buf[0] = '"';
148         if (len > OPT_SHOW_LEN - 2)
149                 len = OPT_SHOW_LEN - 2;
150         strncpy(buf+1, *p, len);
151         buf[1+len] = '"';
152 }
153
154 /* Set an integer value, various forms.  Sets to 1 on arg == NULL. */
155 void opt_show_intval(char buf[OPT_SHOW_LEN], const int *i)
156 {
157         snprintf(buf, OPT_SHOW_LEN, "%i", *i);
158 }
159
160 void opt_show_uintval(char buf[OPT_SHOW_LEN], const unsigned int *ui)
161 {
162         snprintf(buf, OPT_SHOW_LEN, "%u", *ui);
163 }
164
165 void opt_show_longval(char buf[OPT_SHOW_LEN], const long *l)
166 {
167         snprintf(buf, OPT_SHOW_LEN, "%li", *l);
168 }
169
170 void opt_show_ulongval(char buf[OPT_SHOW_LEN], const unsigned long *ul)
171 {
172         snprintf(buf, OPT_SHOW_LEN, "%lu", *ul);
173 }