liboptcfg

view src/optcfg.c @ 1:8fd2858c6a29

find changesets by author, revision, files, or words in the commit message
example (test) program
author John Tsiombikas <nuclear@member.fsf.org>
date Sat, 14 Nov 2015 14:12:30 +0200
parents 10b89befcaa9
children 9c73004c7af3
line source
1 #include <stdio.h>
2 #include <stdlib.h>
3 #include <string.h>
4 #include <ctype.h>
5 #ifdef _MSC_VER
6 #include <malloc.h>
7 #else
8 #include <alloca.h>
9 #endif
10 #include "optcfg.h"
11
12 struct optcfg {
13 struct optcfg_option *optlist;
14
15 optcfg_opt_callback opt_func;
16 void *opt_cls;
17 optcfg_arg_callback arg_func;
18 void *arg_cls;
19
20 int err_abort;
21
22 /* argument parsing state */
23 char **argv;
24 int argidx;
25
26 /* config file parsing state */
27 const char *cfg_fname;
28 int cfg_nline;
29 char *cfg_value;
30 };
31
32 static int get_opt(struct optcfg *oc, const char *s);
33 static char *skip_spaces(char *s);
34 static void strip_comments(char *s);
35 static void strip_trailing_spaces(char *s);
36 static char *parse_keyval(char *line);
37
38
39 struct optcfg *optcfg_init(struct optcfg_option *optv)
40 {
41 struct optcfg *oc;
42
43 if(!(oc = calloc(1, sizeof *oc))) {
44 return 0;
45 }
46 oc->optlist = optv;
47 oc->err_abort = 1;
48 return oc;
49 }
50
51 void optcfg_destroy(struct optcfg *oc)
52 {
53 memset(oc, 0, sizeof *oc);
54 free(oc);
55 }
56
57 void optcfg_set_opt_callback(struct optcfg *oc, optcfg_opt_callback func, void *cls)
58 {
59 oc->opt_func = func;
60 oc->opt_cls = cls;
61 }
62
63 void optcfg_set_arg_callback(struct optcfg *oc, optcfg_arg_callback func, void *cls)
64 {
65 oc->arg_func = func;
66 oc->arg_cls = cls;
67 }
68
69 void optcfg_set_error_action(struct optcfg *oc, int act)
70 {
71 if(act == OPTCFG_ERROR_FAIL) {
72 oc->err_abort = 1;
73 } else if(act == OPTCFG_ERROR_IGNORE) {
74 oc->err_abort = 0;
75 }
76 }
77
78 int optcfg_parse_args(struct optcfg *oc, int argc, char **argv)
79 {
80 int i;
81
82 oc->argv = argv;
83
84 for(i=1; i<argc; i++) {
85 oc->argidx = i;
86
87 if(argv[i][0] == '-') {
88 if(oc->opt_func) {
89 int o = get_opt(oc, argv[i]);
90 if(o == -1 || oc->opt_func(oc, o, oc->opt_cls) == -1) {
91 if(oc->err_abort) {
92 return -1;
93 }
94 }
95 } else {
96 fprintf(stderr, "unexpected option: %s\n", argv[i]);
97 if(oc->err_abort) {
98 return -1;
99 }
100 }
101 } else {
102 if(oc->arg_func) {
103 if(oc->arg_func(oc, argv[i], oc->arg_cls) == -1) {
104 if(oc->err_abort) {
105 return -1;
106 }
107 }
108 } else {
109 fprintf(stderr, "unexpected argument: %s\n", argv[i]);
110 if(oc->err_abort) {
111 return -1;
112 }
113 }
114 }
115
116 i = oc->argidx;
117 }
118
119 oc->argidx = 0; /* done parsing args */
120 return 0;
121 }
122
123 int optcfg_parse_config_file(struct optcfg *oc, const char *fname)
124 {
125 int res;
126 FILE *fp = fopen(fname, "rb");
127 if(!fp) {
128 return -1;
129 }
130
131 oc->cfg_fname = fname;
132 res = optcfg_parse_config_stream(oc, fp);
133 oc->cfg_fname = 0;
134 fclose(fp);
135 return res;
136 }
137
138 int optcfg_parse_config_stream(struct optcfg *oc, FILE *fp)
139 {
140 char buf[512];
141
142 oc->cfg_nline = 0;
143 while(fgets(buf, sizeof buf, fp)) {
144 ++oc->cfg_nline;
145
146 if(optcfg_parse_config_line(oc, buf) == -1) {
147 if(oc->err_abort) {
148 return -1;
149 }
150 }
151 }
152 return 0;
153 }
154
155 int optcfg_parse_config_line(struct optcfg *oc, const char *line)
156 {
157 int opt, len;
158 char *start, *val, *buf;
159
160 len = strlen(line);
161 buf = alloca(len + 1);
162 memcpy(buf, line, len);
163
164 start = skip_spaces(buf);
165 strip_comments(start);
166 strip_trailing_spaces(start);
167 if(!*start) {
168 return 0;
169 }
170
171 if(!(val = parse_keyval(start))) {
172 fprintf(stderr, "error parsing %s line %d: invalid syntax\n", oc->cfg_fname ? oc->cfg_fname : "", oc->cfg_nline);
173 return -1;
174 }
175 oc->cfg_value = val;
176 if((opt = get_opt(oc, start)) == -1) {
177 fprintf(stderr, "error parsing %s line %d: unknown option: %s\n", oc->cfg_fname ? oc->cfg_fname : "",
178 oc->cfg_nline, start);
179 return -1;
180 }
181
182 if(oc->opt_func) {
183 if(oc->opt_func(oc, opt, oc->opt_cls) == -1) {
184 return -1;
185 }
186 }
187 oc->cfg_value = 0;
188 return 0;
189 }
190
191 char *optcfg_next_value(struct optcfg *oc)
192 {
193 if(oc->argidx) { /* we're in the middle of parsing arguments, so get the next one */
194 return oc->argv[++oc->argidx];
195 }
196 if(oc->cfg_value) {
197 char *val = oc->cfg_value;
198 oc->cfg_value = 0;
199 return val;
200 }
201 return 0;
202 }
203
204 void optcfg_print_options(struct optcfg *oc)
205 {
206 int i;
207 for(i=0; oc->optlist[i].opt != -1; i++) {
208 struct optcfg_option *opt = oc->optlist + i;
209
210 if(opt->c) {
211 printf(" -%c", opt->c);
212 } else {
213 printf(" ");
214 }
215 if(opt->s) {
216 printf("%c-%s: ", opt->c ? ',' : ' ', opt->s);
217 } else {
218 printf(": ");
219 }
220 printf("%s\n", opt->desc ? opt->desc : "undocumented");
221 }
222 }
223
224 int optcfg_bool_value(char *s, int *valret)
225 {
226 if(strcasecmp(s, "yes") == 0 || strcasecmp(s, "true") == 0 || strcmp(s, "1") == 0) {
227 *valret = 1;
228 return 0;
229 }
230 if(strcasecmp(s, "no") == 0 || strcasecmp(s, "false") == 0 || strcmp(s, "0") == 0) {
231 *valret = 0;
232 return 0;
233 }
234 return -1;
235 }
236
237 int optcfg_int_value(char *str, int *valret)
238 {
239 char *endp;
240 *valret = strtol(str, &endp, 0);
241 if(endp == str) {
242 return -1;
243 }
244 return 0;
245 }
246
247 int optcfg_float_value(char *str, float *valret)
248 {
249 char *endp;
250 *valret = strtod(str, &endp);
251 if(endp == str) {
252 return -1;
253 }
254 return 0;
255 }
256
257
258
259 static int get_opt(struct optcfg *oc, const char *arg)
260 {
261 int i, ndashes = 0;
262
263 while(*arg && *arg == '-') {
264 ndashes++;
265 arg++;
266 }
267
268 if(ndashes > 2) {
269 arg -= ndashes;
270 }
271
272 if(arg[1]) { /* match long options */
273 for(i=0; oc->optlist[i].opt != -1; i++) {
274 if(strcmp(arg, oc->optlist[i].s) == 0) {
275 return i;
276 }
277 }
278 } else {
279 for(i=0; oc->optlist[i].opt != -1; i++) {
280 if(arg[0] == oc->optlist[i].c) {
281 return i;
282 }
283 }
284 }
285 return -1;
286 }
287
288 static char *skip_spaces(char *s)
289 {
290 while(*s && isspace(*s)) ++s;
291 return s;
292 }
293
294 static void strip_comments(char *s)
295 {
296 while(*s && *s != '#') ++s;
297 if(*s == '#') *s = 0;
298 }
299
300 static void strip_trailing_spaces(char *s)
301 {
302 char *end = s + strlen(s) - 1;
303 while(end >= s && isspace(*end)) {
304 *end-- = 0;
305 }
306 }
307
308 static char *parse_keyval(char *line)
309 {
310 char *val;
311 char *eq = strchr(line, '=');
312 if(!eq) return 0;
313
314 *eq = 0;
315 strip_trailing_spaces(line);
316 val = skip_spaces(eq + 1);
317
318 if(!*line || !*val) {
319 return 0;
320 }
321 return val;
322 }