parse.c revision 1.17 1 /* $NetBSD: parse.c,v 1.17 2013/03/01 18:25:16 joerg Exp $ */
2
3 /*-
4 * Copyright (c) 2008 David Young. All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
11 * 2. Redistributions in binary form must reproduce the above copyright
12 * notice, this list of conditions and the following disclaimer in the
13 * documentation and/or other materials provided with the distribution.
14 *
15 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25 * SUCH DAMAGE.
26 */
27
28 #include <sys/cdefs.h>
29 #ifndef lint
30 __RCSID("$NetBSD: parse.c,v 1.17 2013/03/01 18:25:16 joerg Exp $");
31 #endif /* not lint */
32
33 #include <err.h>
34 #include <errno.h>
35 #include <limits.h>
36 #include <netdb.h>
37 #include <stddef.h>
38 #include <stdio.h>
39 #include <stdlib.h>
40 #include <string.h>
41
42 #include <arpa/inet.h>
43 #include <sys/param.h>
44 #include <net/if.h>
45 #include <net/if_dl.h>
46 #include <netatalk/at.h>
47
48 #include "env.h"
49 #include "parse.h"
50 #include "util.h"
51
52 #define dbg_warnx(__fmt, ...) /* empty */
53
54 static int parser_default_init(struct parser *);
55 static int pbranch_init(struct parser *);
56 static int pkw_init(struct parser *);
57
58 static int pterm_match(const struct parser *, const struct match *,
59 struct match *, int, const char *);
60
61 static int paddr_match(const struct parser *, const struct match *,
62 struct match *, int, const char *);
63
64 static int pbranch_match(const struct parser *, const struct match *,
65 struct match *, int, const char *);
66
67 static int piface_match(const struct parser *, const struct match *,
68 struct match *, int, const char *);
69
70 static int pstr_match(const struct parser *, const struct match *,
71 struct match *, int, const char *);
72
73 static int pinteger_match(const struct parser *, const struct match *,
74 struct match *, int, const char *);
75
76 static int pkw_match(const struct parser *, const struct match *,
77 struct match *, int, const char *);
78
79 const struct parser_methods pterm_methods = {
80 .pm_match = pterm_match
81 , .pm_init = NULL
82 };
83
84 const struct parser_methods pstr_methods = {
85 .pm_match = pstr_match
86 , .pm_init = parser_default_init
87 };
88
89 const struct parser_methods pinteger_methods = {
90 .pm_match = pinteger_match
91 , .pm_init = parser_default_init
92 };
93
94 const struct parser_methods paddr_methods = {
95 .pm_match = paddr_match
96 , .pm_init = parser_default_init
97 };
98
99 const struct parser_methods piface_methods = {
100 .pm_match = piface_match
101 , .pm_init = parser_default_init
102 };
103
104 const struct parser_methods pbranch_methods = {
105 .pm_match = pbranch_match
106 , .pm_init = pbranch_init
107 };
108
109 const struct parser_methods pkw_methods = {
110 .pm_match = pkw_match
111 , .pm_init = pkw_init
112 };
113
114 static int
115 match_setenv(const struct match *im, struct match *om, const char *key,
116 prop_object_t o)
117 {
118 if (im == NULL)
119 om->m_env = prop_dictionary_create();
120 else
121 om->m_env = prop_dictionary_copy(im->m_env);
122
123 if (om->m_env == NULL)
124 goto delobj;
125
126 if (key != NULL && !prop_dictionary_set(om->m_env, key, o))
127 goto deldict;
128
129 if (o != NULL)
130 prop_object_release((prop_object_t)o);
131
132 return 0;
133 deldict:
134 prop_object_release((prop_object_t)om->m_env);
135 om->m_env = NULL;
136 delobj:
137 prop_object_release((prop_object_t)o);
138 errno = ENOMEM;
139 return -1;
140 }
141
142 int
143 pstr_match(const struct parser *p, const struct match *im, struct match *om,
144 int argidx, const char *arg)
145 {
146 prop_object_t o;
147 const struct pstr *ps = (const struct pstr *)p;
148 uint8_t buf[128];
149 int len;
150
151 if (arg == NULL) {
152 errno = EINVAL;
153 return -1;
154 }
155
156 len = (int)sizeof(buf);
157 if (get_string(arg, NULL, buf, &len, ps->ps_hexok) == NULL) {
158 errno = EINVAL;
159 return -1;
160 }
161
162 o = (prop_object_t)prop_data_create_data(buf, len);
163
164 if (o == NULL) {
165 errno = ENOMEM;
166 return -1;
167 }
168
169 if (match_setenv(im, om, ps->ps_key, o) == -1)
170 return -1;
171
172 om->m_argidx = argidx;
173 om->m_parser = p;
174 om->m_nextparser = p->p_nextparser;
175
176 return 0;
177 }
178
179 int
180 pinteger_match(const struct parser *p, const struct match *im, struct match *om,
181 int argidx, const char *arg)
182 {
183 prop_object_t o;
184 const struct pinteger *pi = (const struct pinteger *)p;
185 char *end;
186 int64_t val;
187
188 if (arg == NULL) {
189 errno = EINVAL;
190 return -1;
191 }
192
193 val = strtoimax(arg, &end, pi->pi_base);
194 if ((val == INTMAX_MIN || val == INTMAX_MAX) && errno == ERANGE)
195 return -1;
196
197 if (*end != '\0') {
198 errno = EINVAL;
199 return -1;
200 }
201
202 if (val < pi->pi_min || val > pi->pi_max) {
203 errno = ERANGE;
204 return -1;
205 }
206
207 o = (prop_object_t)prop_number_create_integer(val);
208
209 if (o == NULL) {
210 errno = ENOMEM;
211 return -1;
212 }
213
214 if (match_setenv(im, om, pi->pi_key, o) == -1)
215 return -1;
216
217 om->m_argidx = argidx;
218 om->m_parser = p;
219 om->m_nextparser = p->p_nextparser;
220
221 return 0;
222 }
223
224 static int
225 parse_linkaddr(const char *addr, struct sockaddr_storage *ss)
226 {
227 static const size_t maxlen =
228 sizeof(*ss) - offsetof(struct sockaddr_dl, sdl_data[0]);
229 enum {
230 LLADDR_S_INITIAL = 0,
231 LLADDR_S_ONE_OCTET = 1,
232 LLADDR_S_TWO_OCTETS = 2,
233 LLADDR_S_COLON = 3
234 } state = LLADDR_S_INITIAL;
235 uint8_t octet = 0, val;
236 struct sockaddr_dl *sdl;
237 const char *p;
238 size_t i;
239
240 memset(ss, 0, sizeof(*ss));
241 ss->ss_family = AF_LINK;
242 sdl = (struct sockaddr_dl *)ss;
243
244 for (i = 0, p = addr; i < maxlen; p++) {
245 dbg_warnx("%s.%d: *p == %c, state %d", __func__, __LINE__, *p,
246 state);
247 if (*p == '\0') {
248 dbg_warnx("%s.%d", __func__, __LINE__);
249 if (state != LLADDR_S_ONE_OCTET &&
250 state != LLADDR_S_TWO_OCTETS)
251 return -1;
252 dbg_warnx("%s.%d", __func__, __LINE__);
253 sdl->sdl_data[i++] = octet;
254 sdl->sdl_len = offsetof(struct sockaddr_dl, sdl_data)
255 + i * sizeof(sdl->sdl_data[0]);
256 sdl->sdl_alen = i;
257 return 0;
258 }
259 if (*p == ':') {
260 dbg_warnx("%s.%d", __func__, __LINE__);
261 if (state != LLADDR_S_ONE_OCTET &&
262 state != LLADDR_S_TWO_OCTETS)
263 return -1;
264 dbg_warnx("%s.%d", __func__, __LINE__);
265 sdl->sdl_data[i++] = octet;
266 state = LLADDR_S_COLON;
267 continue;
268 }
269 if ('a' <= *p && *p <= 'f')
270 val = 10 + *p - 'a';
271 else if ('A' <= *p && *p <= 'F')
272 val = 10 + *p - 'A';
273 else if ('0' <= *p && *p <= '9')
274 val = *p - '0';
275 else
276 return -1;
277
278 dbg_warnx("%s.%d", __func__, __LINE__);
279 if (state == LLADDR_S_ONE_OCTET) {
280 state = LLADDR_S_TWO_OCTETS;
281 octet <<= 4;
282 octet |= val;
283 } else if (state != LLADDR_S_INITIAL && state != LLADDR_S_COLON)
284 return -1;
285 else {
286 state = LLADDR_S_ONE_OCTET;
287 octet = val;
288 }
289 dbg_warnx("%s.%d", __func__, __LINE__);
290 }
291 return -1;
292 }
293
294 static int
295 paddr_match(const struct parser *p, const struct match *im, struct match *om,
296 int argidx, const char *arg0)
297 {
298 unsigned int net, node;
299 int nread;
300 union {
301 struct sockaddr sa;
302 struct sockaddr_at sat;
303 struct sockaddr_in sin;
304 struct sockaddr_storage ss;
305 } u;
306 const struct paddr *pa = (const struct paddr *)p;
307 prop_data_t d;
308 prop_object_t o;
309 int64_t af0;
310 int af, rc;
311 struct paddr_prefix *pfx, *mask;
312 const struct sockaddr *sa = NULL;
313 struct addrinfo hints, *result = NULL;
314 char *arg, *end, *plen = NULL, *servname0;
315 const char *servname;
316 long prefixlen = -1;
317 size_t len;
318
319 if (arg0 == NULL) {
320 errno = EINVAL;
321 return -1;
322 }
323
324 if (pa->pa_activator != NULL &&
325 prop_dictionary_get(im->m_env, pa->pa_activator) == NULL)
326 return -1;
327
328 if (pa->pa_deactivator != NULL &&
329 prop_dictionary_get(im->m_env, pa->pa_deactivator) != NULL)
330 return -1;
331
332 if (!prop_dictionary_get_int64(im->m_env, "af", &af0))
333 af = AF_UNSPEC;
334 else
335 af = af0;
336
337 memset(&u, 0, sizeof(u));
338
339 switch (af) {
340 case AF_UNSPEC:
341 case AF_INET:
342 case AF_INET6:
343 if ((arg = strdup(arg0)) == NULL)
344 return -1;
345
346 servname0 = arg;
347 (void)strsep(&servname0, ",");
348 servname = (servname0 == NULL) ? "0" : servname0;
349
350 if (pa->pa_maskkey == NULL)
351 ;
352 else if ((plen = strrchr(arg, '/')) != NULL)
353 *plen++ = '\0';
354
355 memset(&hints, 0, sizeof(hints));
356
357 hints.ai_flags = AI_NUMERICHOST | AI_PASSIVE;
358 hints.ai_family = af;
359 hints.ai_socktype = SOCK_DGRAM;
360
361 for (;;) {
362 rc = getaddrinfo(arg, servname, &hints, &result);
363 if (rc == 0) {
364 if (result->ai_next == NULL)
365 sa = result->ai_addr;
366 else
367 errno = EMLINK;
368 break;
369 } else if ((hints.ai_flags & AI_NUMERICHOST) != 0 &&
370 (af == AF_INET || af == AF_UNSPEC) &&
371 inet_aton(arg, &u.sin.sin_addr) == 1) {
372 u.sin.sin_family = AF_INET;
373 u.sin.sin_len = sizeof(u.sin);
374 sa = &u.sa;
375 break;
376 } else if ((hints.ai_flags & AI_NUMERICHOST) == 0 ||
377 rc != EAI_NONAME) {
378 errno = ENOENT;
379 break;
380 }
381 hints.ai_flags &= ~AI_NUMERICHOST;
382 }
383
384
385 if (plen == NULL)
386 prefixlen = -1;
387 else {
388 prefixlen = strtol(plen, &end, 10);
389 if (end != NULL && *end != '\0')
390 sa = NULL;
391 if (prefixlen < 0 || prefixlen >= UINT8_MAX) {
392 errno = ERANGE;
393 sa = NULL;
394 }
395 }
396
397 free(arg);
398 if (sa != NULL || af != AF_UNSPEC)
399 break;
400 /*FALLTHROUGH*/
401 case AF_APPLETALK:
402 if (sscanf(arg0, "%u.%u%n", &net, &node, &nread) == 2 &&
403 net != 0 && net <= 0xffff && node != 0 && node <= 0xfe &&
404 arg0[nread] == '\0') {
405 u.sat.sat_family = AF_APPLETALK;
406 u.sat.sat_len = sizeof(u.sat);
407 u.sat.sat_addr.s_net = htons(net);
408 u.sat.sat_addr.s_node = node;
409 sa = &u.sa;
410 }
411 break;
412 case AF_LINK:
413 if (parse_linkaddr(arg0, &u.ss) == -1)
414 sa = NULL;
415 else
416 sa = &u.sa;
417 break;
418 }
419
420 if (sa == NULL)
421 return -1;
422
423 len = offsetof(struct paddr_prefix, pfx_addr) + sa->sa_len;
424
425 if ((pfx = malloc(len)) == NULL)
426 return -1;
427
428 #if 0
429 {
430 int i;
431
432 for (i = 0; i < sa->sa_len; i++)
433 printf(" %02x", ((const uint8_t *)sa)[i]);
434 printf("\n");
435 }
436 #endif
437
438 pfx->pfx_len = (int16_t)prefixlen;
439 memcpy(&pfx->pfx_addr, sa, sa->sa_len);
440 af = sa->sa_family;
441
442 if (result != NULL)
443 freeaddrinfo(result);
444
445 o = (prop_object_t)prop_data_create_data(pfx, len);
446
447 free(pfx);
448
449 if (o == NULL)
450 return -1;
451
452 if (match_setenv(im, om, pa->pa_addrkey, o) == -1)
453 return -1;
454
455 if (pa->pa_maskkey != NULL && plen != NULL) {
456 size_t masklen;
457
458 if ((mask = prefixlen_to_mask(af, prefixlen)) == NULL) {
459 err(EXIT_FAILURE, "%s: prefixlen_to_mask(%d, %ld)",
460 __func__, af, prefixlen);
461 return -1;
462 }
463
464 masklen = paddr_prefix_size(mask);
465
466 d = prop_data_create_data(mask, masklen);
467 free(mask);
468
469 if (d == NULL) {
470 err(EXIT_FAILURE, "%s: prop_data_create_data",
471 __func__);
472 return -1;
473 }
474
475 rc = prop_dictionary_set(om->m_env, pa->pa_maskkey,
476 (prop_object_t)d) ? 0 : -1;
477
478 prop_object_release((prop_object_t)d);
479
480 if (rc != 0) {
481 err(EXIT_FAILURE, "%s: prop_dictionary_set", __func__);
482 return rc;
483 }
484 }
485
486 om->m_argidx = argidx;
487 om->m_parser = p;
488 om->m_nextparser = p->p_nextparser;
489 return 0;
490 }
491
492 static int
493 pterm_match(const struct parser *p, const struct match *im,
494 struct match *om, int argidx, const char *arg)
495 {
496 const struct pterm *pt = (const struct pterm *)p;
497 prop_bool_t b;
498
499 if (arg != NULL) {
500 errno = EINVAL;
501 return -1;
502 }
503 b = prop_bool_create(true);
504
505 if (match_setenv(im, om, pt->pt_key, (prop_object_t)b) == -1)
506 return -1;
507
508 om->m_argidx = argidx;
509 om->m_parser = p;
510 om->m_nextparser = NULL;
511 return 0;
512 }
513
514 static int
515 piface_match(const struct parser *p, const struct match *im,
516 struct match *om, int argidx, const char *arg)
517 {
518 const struct piface *pif = (const struct piface *)p;
519 prop_object_t o;
520
521 if (arg == NULL || strlen(arg) > IFNAMSIZ) {
522 errno = EINVAL;
523 return -1;
524 }
525
526 if ((o = (prop_object_t)prop_string_create_cstring(arg)) == NULL) {
527 errno = ENOMEM;
528 return -1;
529 }
530
531 if (match_setenv(im, om, pif->pif_key, o) == -1)
532 return -1;
533
534 om->m_argidx = argidx;
535 om->m_parser = p;
536 om->m_nextparser = p->p_nextparser;
537 return 0;
538 }
539
540 static void
541 match_cleanup(struct match *dst)
542 {
543 if (dst->m_env != NULL)
544 prop_object_release((prop_object_t)dst->m_env);
545 memset(dst, 0, sizeof(*dst));
546 }
547
548 static void
549 match_copy(struct match *dst, const struct match *src)
550 {
551 match_cleanup(dst);
552
553 prop_object_retain((prop_object_t)src->m_env);
554 *dst = *src;
555 }
556
557 static int
558 pbranch_match(const struct parser *p, const struct match *im,
559 struct match *om, int argidx, const char *arg)
560 {
561 const struct parser *nextp;
562 struct branch *b;
563 const struct pbranch *pb = (const struct pbranch *)p;
564 struct match tmpm;
565 int nforbid = 0, nmatch = 0, rc;
566 parser_match_t matchfunc;
567
568 memset(&tmpm, 0, sizeof(tmpm));
569
570 SIMPLEQ_FOREACH(b, &pb->pb_branches, b_next) {
571 dbg_warnx("%s: b->b_nextparser %p", __func__,
572 (const void *)b->b_nextparser);
573 nextp = b->b_nextparser;
574 if (nextp == NULL) {
575 if (arg == NULL) {
576 nmatch++;
577 match_setenv(im, om, NULL, NULL);
578 om->m_nextparser = NULL;
579 om->m_parser = p;
580 om->m_argidx = argidx;
581 }
582 continue;
583 }
584 matchfunc = nextp->p_methods->pm_match;
585 rc = (*matchfunc)(nextp, im, &tmpm, argidx, arg);
586 if (rc == 0) {
587 match_copy(om, &tmpm);
588 match_cleanup(&tmpm);
589 nmatch++;
590 dbg_warnx("%s: branch %s ok", __func__, nextp->p_name);
591 if (pb->pb_match_first)
592 break;
593 } else if (rc == 1) {
594 nforbid++;
595 if (pb->pb_match_first)
596 break;
597 } else {
598 dbg_warnx("%s: fail branch %s", __func__,
599 nextp->p_name);
600 }
601 }
602 switch (nmatch) {
603 case 0:
604 errno = ENOENT;
605 return (nforbid == 0) ? -1 : 1;
606 case 1:
607 dbg_warnx("%s: branch ok", __func__);
608 return 0;
609 default:
610 match_cleanup(om);
611 errno = EMLINK;
612 return -1;
613 }
614 }
615
616 static int
617 pkw_match(const struct parser *p, const struct match *im,
618 struct match *om, int argidx, const char *arg)
619 {
620 prop_object_t o = NULL;
621 struct kwinst *k;
622 union kwval *u = NULL;
623 const struct pkw *pk = (const struct pkw *)p;
624
625 if (arg == NULL) {
626 errno = EINVAL;
627 return -1;
628 }
629
630 SIMPLEQ_FOREACH(k, &pk->pk_keywords, k_next) {
631 if (k->k_act != NULL &&
632 prop_dictionary_get(im->m_env, k->k_act) == NULL)
633 continue;
634
635 if (k->k_neg && arg[0] == '-' &&
636 strcmp(k->k_word, arg + 1) == 0)
637 u = &k->k_negu;
638 else if (strcmp(k->k_word, arg) == 0)
639 u = &k->k_u;
640 else
641 continue;
642
643 if (k->k_altdeact != NULL &&
644 prop_dictionary_get(im->m_env, k->k_altdeact) != NULL)
645 return 1;
646
647 if (k->k_deact != NULL &&
648 prop_dictionary_get(im->m_env, k->k_deact) != NULL)
649 return 1;
650 break;
651 }
652 if (k == NULL) {
653 errno = ENOENT;
654 return -1;
655 }
656 switch (k->k_type) {
657 case KW_T_NONE:
658 break;
659 case KW_T_BOOL:
660 o = (prop_object_t)prop_bool_create(u->u_bool);
661 if (o == NULL)
662 goto err;
663 break;
664 case KW_T_INT:
665 o = (prop_object_t)prop_number_create_integer(u->u_sint);
666 if (o == NULL)
667 goto err;
668 break;
669 case KW_T_UINT:
670 o = (prop_object_t)prop_number_create_unsigned_integer(
671 u->u_uint);
672 if (o == NULL)
673 goto err;
674 break;
675 case KW_T_OBJ:
676 o = u->u_obj;
677 break;
678 case KW_T_STR:
679 o = (prop_object_t)prop_string_create_cstring_nocopy(u->u_str);
680 if (o == NULL)
681 goto err;
682 break;
683 default:
684 errx(EXIT_FAILURE, "unknown keyword type %d", k->k_type);
685 }
686
687 if (match_setenv(im, om, (o == NULL) ? NULL : k->k_key, o) == -1)
688 return -1;
689
690 om->m_argidx = argidx;
691 om->m_parser = p;
692 om->m_nextparser = k->k_nextparser;
693 om->m_exec = k->k_exec;
694 return 0;
695 err:
696 errno = ENOMEM;
697 return -1;
698 }
699
700 struct paddr *
701 paddr_create(const char *name, parser_exec_t pexec, const char *addrkey,
702 const char *maskkey, struct parser *next)
703 {
704 struct paddr *pa;
705
706 if ((pa = calloc(sizeof(*pa), 1)) == NULL)
707 return NULL;
708
709 pa->pa_parser.p_methods = &paddr_methods;
710 pa->pa_parser.p_exec = pexec;
711 pa->pa_parser.p_name = name;
712 pa->pa_parser.p_nextparser = next;
713
714 pa->pa_addrkey = addrkey;
715 pa->pa_maskkey = maskkey;
716
717 return pa;
718 }
719
720 struct piface *
721 piface_create(const char *name, parser_exec_t pexec, const char *defkey,
722 struct parser *defnext)
723 {
724 struct piface *pif;
725
726 if ((pif = calloc(sizeof(*pif), 1)) == NULL)
727 return NULL;
728
729 pif->pif_parser.p_methods = &piface_methods;
730 pif->pif_parser.p_exec = pexec;
731 pif->pif_parser.p_name = name;
732 pif->pif_parser.p_nextparser = defnext;
733
734 pif->pif_key = defkey;
735
736 return pif;
737 }
738
739 int
740 pbranch_addbranch(struct pbranch *pb, struct parser *p)
741 {
742 struct branch *b;
743
744 if ((b = malloc(sizeof(*b))) == NULL)
745 return -1;
746 b->b_nextparser = p;
747 SIMPLEQ_INSERT_HEAD(&pb->pb_branches, b, b_next);
748 pb->pb_parser.p_initialized = false;
749 return parser_init(&pb->pb_parser);
750 }
751
752 int
753 pbranch_setbranches(struct pbranch *pb, const struct branch *brs, size_t nbr)
754 {
755 struct branch *b;
756 size_t i;
757
758 dbg_warnx("%s: nbr %zu", __func__, nbr);
759
760 while ((b = SIMPLEQ_FIRST(&pb->pb_branches)) != NULL) {
761 SIMPLEQ_REMOVE_HEAD(&pb->pb_branches, b_next);
762 free(b);
763 }
764
765 for (i = 0; i < nbr; i++) {
766 if ((b = malloc(sizeof(*b))) == NULL)
767 goto err;
768 *b = brs[i];
769 dbg_warnx("%s: b->b_nextparser %p", __func__,
770 (const void *)b->b_nextparser);
771 SIMPLEQ_INSERT_TAIL(&pb->pb_branches, b, b_next);
772 }
773
774 return 0;
775 err:
776 while ((b = SIMPLEQ_FIRST(&pb->pb_branches)) != NULL) {
777 SIMPLEQ_REMOVE_HEAD(&pb->pb_branches, b_next);
778 free(b);
779 }
780 return -1;
781 }
782
783 static int
784 pbranch_init(struct parser *p)
785 {
786 struct branch *b;
787 struct pbranch *pb = (struct pbranch *)p;
788 struct parser *np;
789
790 if (pb->pb_nbrinit == 0)
791 ;
792 else if (pbranch_setbranches(pb, pb->pb_brinit, pb->pb_nbrinit) == -1)
793 return -1;
794
795 pb->pb_nbrinit = 0;
796
797 SIMPLEQ_FOREACH(b, &pb->pb_branches, b_next) {
798 np = b->b_nextparser;
799 if (np != NULL && parser_init(np) == -1)
800 return -1;
801 }
802 return 0;
803 }
804
805 struct pbranch *
806 pbranch_create(const char *name, const struct branch *brs, size_t nbr,
807 bool match_first)
808 {
809 struct pbranch *pb;
810
811 dbg_warnx("%s: nbr %zu", __func__, nbr);
812
813 if ((pb = calloc(1, sizeof(*pb))) == NULL)
814 return NULL;
815
816 pb->pb_parser.p_methods = &pbranch_methods;
817 pb->pb_parser.p_name = name;
818
819 SIMPLEQ_INIT(&pb->pb_branches);
820
821 if (pbranch_setbranches(pb, brs, nbr) == -1)
822 goto post_pb_err;
823
824 pb->pb_match_first = match_first;
825 return pb;
826 post_pb_err:
827 free(pb);
828 return NULL;
829 }
830
831 static int
832 parser_default_init(struct parser *p)
833 {
834 struct parser *np;
835
836 np = p->p_nextparser;
837 if (np != NULL && parser_init(np) == -1)
838 return -1;
839
840 return 0;
841 }
842
843 static int
844 pkw_setwords(struct pkw *pk, parser_exec_t defexec, const char *defkey,
845 const struct kwinst *kws, size_t nkw, struct parser *defnext)
846 {
847 struct kwinst *k;
848 size_t i;
849
850 for (i = 0; i < nkw; i++) {
851 if (kws[i].k_word == NULL)
852 continue;
853 if ((k = malloc(sizeof(*k))) == NULL)
854 goto post_pk_err;
855 *k = kws[i];
856 if (k->k_nextparser == NULL)
857 k->k_nextparser = defnext;
858 if (k->k_key == NULL)
859 k->k_key = defkey;
860 if (k->k_exec == NULL)
861 k->k_exec = defexec;
862 SIMPLEQ_INSERT_TAIL(&pk->pk_keywords, k, k_next);
863 }
864 return 0;
865
866 post_pk_err:
867 while ((k = SIMPLEQ_FIRST(&pk->pk_keywords)) != NULL) {
868 SIMPLEQ_REMOVE_HEAD(&pk->pk_keywords, k_next);
869 free(k);
870 }
871 return -1;
872 }
873
874 static int
875 pkw_init(struct parser *p)
876 {
877 struct kwinst *k;
878 struct pkw *pk = (struct pkw *)p;
879 struct parser *np;
880
881 if (pk->pk_nkwinit == 0)
882 ;
883 else if (pkw_setwords(pk, pk->pk_execinit, pk->pk_keyinit,
884 pk->pk_kwinit, pk->pk_nkwinit, pk->pk_nextinit) == -1)
885 return -1;
886
887 pk->pk_nkwinit = 0;
888
889 SIMPLEQ_FOREACH(k, &pk->pk_keywords, k_next) {
890 np = k->k_nextparser;
891 if (np != NULL && parser_init(np) == -1)
892 return -1;
893 }
894 return 0;
895 }
896
897 struct pkw *
898 pkw_create(const char *name, parser_exec_t defexec, const char *defkey,
899 const struct kwinst *kws, size_t nkw, struct parser *defnext)
900 {
901 struct pkw *pk;
902
903 if ((pk = calloc(1, sizeof(*pk))) == NULL)
904 return NULL;
905
906 pk->pk_parser.p_methods = &pkw_methods;
907 pk->pk_parser.p_exec = defexec;
908 pk->pk_parser.p_name = name;
909
910 SIMPLEQ_INIT(&pk->pk_keywords);
911
912 if (pkw_setwords(pk, defexec, defkey, kws, nkw, defnext) == -1)
913 goto err;
914
915 return pk;
916 err:
917 free(pk);
918 return NULL;
919 }
920
921 int
922 parse(int argc, char **argv, const struct parser *p0, struct match *matches,
923 size_t *nmatch, int *narg)
924 {
925 int i, rc = 0;
926 struct match *lastm = NULL, *m = matches;
927 const struct parser *p = p0;
928
929 for (i = 0; i < argc && p != NULL; i++) {
930 if ((size_t)(m - matches) >= *nmatch) {
931 errno = EFBIG;
932 rc = -1;
933 break;
934 }
935 rc = (*p->p_methods->pm_match)(p, lastm, m, i, argv[i]);
936 if (rc != 0)
937 goto out;
938 p = m->m_nextparser;
939 lastm = m++;
940 }
941 for (; (size_t)(m - matches) < *nmatch && p != NULL; ) {
942 rc = (*p->p_methods->pm_match)(p, lastm, m, i, NULL);
943 if (rc != 0)
944 break;
945 p = m->m_nextparser;
946 lastm = m++;
947 }
948 out:
949 *nmatch = m - matches;
950 *narg = i;
951 return rc;
952 }
953
954 int
955 matches_exec(const struct match *matches, prop_dictionary_t oenv, size_t nmatch)
956 {
957 size_t i;
958 int rc = 0;
959 const struct match *m;
960 parser_exec_t pexec;
961 prop_dictionary_t d;
962
963 for (i = 0; i < nmatch; i++) {
964 m = &matches[i];
965 dbg_warnx("%s.%d: i %zu", __func__, __LINE__, i);
966 pexec = (m->m_parser->p_exec != NULL)
967 ? m->m_parser->p_exec : m->m_exec;
968 if (pexec == NULL)
969 continue;
970 dbg_warnx("%s.%d: m->m_parser->p_name %s", __func__, __LINE__,
971 m->m_parser->p_name);
972 d = prop_dictionary_augment(m->m_env, oenv);
973 rc = (*pexec)(d, oenv);
974 prop_object_release((prop_object_t)d);
975 if (rc == -1)
976 break;
977 }
978 return rc;
979 }
980
981 int
982 parser_init(struct parser *p)
983 {
984 if (p->p_initialized)
985 return 0;
986 p->p_initialized = true;
987 if (p->p_methods->pm_init == NULL)
988 return 0;
989 return (*p->p_methods->pm_init)(p);
990 }
991