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