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