xglcomp

annotate src/optcfg/optcfg.c @ 0:d9b3fba68705

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