getcap.c revision 1.42 1 /* $NetBSD: getcap.c,v 1.42 2005/11/29 03:11:59 christos Exp $ */
2
3 /*-
4 * Copyright (c) 1992, 1993
5 * The Regents of the University of California. All rights reserved.
6 *
7 * This code is derived from software contributed to Berkeley by
8 * Casey Leedom of Lawrence Livermore National Laboratory.
9 *
10 * Redistribution and use in source and binary forms, with or without
11 * modification, are permitted provided that the following conditions
12 * are met:
13 * 1. Redistributions of source code must retain the above copyright
14 * notice, this list of conditions and the following disclaimer.
15 * 2. Redistributions in binary form must reproduce the above copyright
16 * notice, this list of conditions and the following disclaimer in the
17 * documentation and/or other materials provided with the distribution.
18 * 3. Neither the name of the University nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 */
34
35 #if HAVE_NBTOOL_CONFIG_H
36 #include "nbtool_config.h"
37 #endif
38
39 #include <sys/cdefs.h>
40 #if defined(LIBC_SCCS) && !defined(lint)
41 #if 0
42 static char sccsid[] = "@(#)getcap.c 8.3 (Berkeley) 3/25/94";
43 #else
44 __RCSID("$NetBSD: getcap.c,v 1.42 2005/11/29 03:11:59 christos Exp $");
45 #endif
46 #endif /* LIBC_SCCS and not lint */
47
48 #ifndef SMALL
49 #include "namespace.h"
50 #endif
51 #include <sys/types.h>
52 #include <sys/param.h>
53
54 #include <assert.h>
55 #include <ctype.h>
56 #ifndef SMALL
57 #include <db.h>
58 #endif
59 #include <errno.h>
60 #include <fcntl.h>
61 #include <limits.h>
62 #include <stdio.h>
63 #include <stdlib.h>
64 #include <string.h>
65 #include <unistd.h>
66
67 #ifdef __weak_alias
68 __weak_alias(cgetcap,_cgetcap)
69 __weak_alias(cgetclose,_cgetclose)
70 __weak_alias(cgetent,_cgetent)
71 __weak_alias(cgetfirst,_cgetfirst)
72 __weak_alias(cgetmatch,_cgetmatch)
73 __weak_alias(cgetnext,_cgetnext)
74 __weak_alias(cgetnum,_cgetnum)
75 __weak_alias(cgetset,_cgetset)
76 __weak_alias(cgetstr,_cgetstr)
77 __weak_alias(cgetustr,_cgetustr)
78 #endif
79
80 #define BFRAG 1024
81 #define BSIZE 1024
82 #define ESC ('[' & 037) /* ASCII ESC */
83 #define MAX_RECURSION 32 /* maximum getent recursion */
84 #define SFRAG 100 /* cgetstr mallocs in SFRAG chunks */
85
86 #define RECOK (char)0
87 #define TCERR (char)1
88 #define SHADOW (char)2
89
90 static size_t topreclen; /* toprec length */
91 static char *toprec; /* Additional record specified by cgetset() */
92 static int gottoprec; /* Flag indicating retrieval of toprecord */
93
94 #ifndef SMALL
95 static int cdbget(DB *, char **, const char *);
96 #endif
97 static int getent(char **, size_t *, const char * const *, int,
98 const char *, int, char *);
99 static int nfcmp(char *, char *);
100
101 /*
102 * Cgetset() allows the addition of a user specified buffer to be added
103 * to the database array, in effect "pushing" the buffer on top of the
104 * virtual database. 0 is returned on success, -1 on failure.
105 */
106 int
107 cgetset(const char *ent)
108 {
109 const char *source, *check;
110 char *dest;
111
112 if (ent == NULL) {
113 if (toprec)
114 free(toprec);
115 toprec = NULL;
116 topreclen = 0;
117 return (0);
118 }
119 topreclen = strlen(ent);
120 if ((toprec = malloc (topreclen + 1)) == NULL) {
121 errno = ENOMEM;
122 return (-1);
123 }
124 gottoprec = 0;
125
126 source=ent;
127 dest=toprec;
128 while (*source) { /* Strip whitespace */
129 *dest++ = *source++; /* Do not check first field */
130 while (*source == ':') {
131 check=source+1;
132 while (*check && (isspace((unsigned char)*check) ||
133 (*check=='\\' && isspace((unsigned char)check[1]))))
134 ++check;
135 if( *check == ':' )
136 source=check;
137 else
138 break;
139
140 }
141 }
142 *dest=0;
143
144 return (0);
145 }
146
147 /*
148 * Cgetcap searches the capability record buf for the capability cap with
149 * type `type'. A pointer to the value of cap is returned on success, NULL
150 * if the requested capability couldn't be found.
151 *
152 * Specifying a type of ':' means that nothing should follow cap (:cap:).
153 * In this case a pointer to the terminating ':' or NUL will be returned if
154 * cap is found.
155 *
156 * If (cap, '@') or (cap, terminator, '@') is found before (cap, terminator)
157 * return NULL.
158 */
159 char *
160 cgetcap(buf, cap, type)
161 char *buf;
162 const char *cap;
163 int type;
164 {
165 char *bp;
166 const char *cp;
167
168 _DIAGASSERT(buf != NULL);
169 _DIAGASSERT(cap != NULL);
170
171 bp = buf;
172 for (;;) {
173 /*
174 * Skip past the current capability field - it's either the
175 * name field if this is the first time through the loop, or
176 * the remainder of a field whose name failed to match cap.
177 */
178 for (;;)
179 if (*bp == '\0')
180 return (NULL);
181 else
182 if (*bp++ == ':')
183 break;
184
185 /*
186 * Try to match (cap, type) in buf.
187 */
188 for (cp = cap; *cp == *bp && *bp != '\0'; cp++, bp++)
189 continue;
190 if (*cp != '\0')
191 continue;
192 if (*bp == '@')
193 return (NULL);
194 if (type == ':') {
195 if (*bp != '\0' && *bp != ':')
196 continue;
197 return(bp);
198 }
199 if (*bp != type)
200 continue;
201 bp++;
202 return (*bp == '@' ? NULL : bp);
203 }
204 /* NOTREACHED */
205 }
206
207 /*
208 * Cgetent extracts the capability record name from the NULL terminated file
209 * array db_array and returns a pointer to a malloc'd copy of it in buf.
210 * Buf must be retained through all subsequent calls to cgetcap, cgetnum,
211 * cgetflag, and cgetstr, but may then be free'd. 0 is returned on success,
212 * -1 if the requested record couldn't be found, -2 if a system error was
213 * encountered (couldn't open/read a file, etc.), and -3 if a potential
214 * reference loop is detected.
215 */
216 int
217 cgetent(char **buf, const char * const *db_array, const char *name)
218 {
219 size_t dummy;
220
221 _DIAGASSERT(buf != NULL);
222 _DIAGASSERT(db_array != NULL);
223 _DIAGASSERT(name != NULL);
224
225 return (getent(buf, &dummy, db_array, -1, name, 0, NULL));
226 }
227
228 /*
229 * Getent implements the functions of cgetent. If fd is non-negative,
230 * *db_array has already been opened and fd is the open file descriptor. We
231 * do this to save time and avoid using up file descriptors for tc=
232 * recursions.
233 *
234 * Getent returns the same success/failure codes as cgetent. On success, a
235 * pointer to a malloc'ed capability record with all tc= capabilities fully
236 * expanded and its length (not including trailing ASCII NUL) are left in
237 * *cap and *len.
238 *
239 * Basic algorithm:
240 * + Allocate memory incrementally as needed in chunks of size BFRAG
241 * for capability buffer.
242 * + Recurse for each tc=name and interpolate result. Stop when all
243 * names interpolated, a name can't be found, or depth exceeds
244 * MAX_RECURSION.
245 */
246 static int
247 getent(char **cap, size_t *len, const char * const *db_array, int fd,
248 const char *name, int depth, char *nfield)
249 {
250 #ifndef SMALL
251 DB *capdbp;
252 char pbuf[MAXPATHLEN];
253 char *cbuf;
254 int retval;
255 size_t clen;
256 #endif
257 char *record, *newrecord;
258 char *r_end, *rp = NULL; /* pacify gcc */
259 const char * const *db_p;
260 int myfd = 0, eof, foundit;
261 int tc_not_resolved;
262
263 _DIAGASSERT(cap != NULL);
264 _DIAGASSERT(len != NULL);
265 _DIAGASSERT(db_array != NULL);
266 /* fd may be -1 */
267 _DIAGASSERT(name != NULL);
268 /* nfield may be NULL */
269
270 /*
271 * Return with ``loop detected'' error if we've recursed more than
272 * MAX_RECURSION times.
273 */
274 if (depth > MAX_RECURSION)
275 return (-3);
276
277 /*
278 * Check if we have a top record from cgetset().
279 */
280 if (depth == 0 && toprec != NULL && cgetmatch(toprec, name) == 0) {
281 if ((record = malloc (topreclen + BFRAG)) == NULL) {
282 errno = ENOMEM;
283 return (-2);
284 }
285 (void)strcpy(record, toprec); /* XXX: strcpy is safe */
286 db_p = db_array;
287 rp = record + topreclen + 1;
288 r_end = rp + BFRAG;
289 goto tc_exp;
290 }
291 /*
292 * Allocate first chunk of memory.
293 */
294 if ((record = malloc(BFRAG)) == NULL) {
295 errno = ENOMEM;
296 return (-2);
297 }
298 r_end = record + BFRAG;
299 foundit = 0;
300 /*
301 * Loop through database array until finding the record.
302 */
303
304 for (db_p = db_array; *db_p != NULL; db_p++) {
305 eof = 0;
306
307 /*
308 * Open database if not already open.
309 */
310
311 if (fd >= 0) {
312 (void)lseek(fd, (off_t)0, SEEK_SET);
313 } else {
314 #ifndef SMALL
315 (void)snprintf(pbuf, sizeof(pbuf), "%s.db", *db_p);
316 if ((capdbp = dbopen(pbuf, O_RDONLY, 0, DB_HASH, 0))
317 != NULL) {
318 free(record);
319 retval = cdbget(capdbp, &record, name);
320 if (retval < 0) {
321 /* no record available */
322 (void)capdbp->close(capdbp);
323 return (retval);
324 }
325 /* save the data; close frees it */
326 clen = strlen(record);
327 cbuf = malloc(clen + 1);
328 memmove(cbuf, record, clen + 1);
329 if (capdbp->close(capdbp) < 0) {
330 int serrno = errno;
331
332 free(cbuf);
333 errno = serrno;
334 return (-2);
335 }
336 *len = clen;
337 *cap = cbuf;
338 return (retval);
339 } else
340 #endif
341 {
342 fd = open(*db_p, O_RDONLY, 0);
343 if (fd < 0) {
344 /* No error on unfound file. */
345 continue;
346 }
347 myfd = 1;
348 }
349 }
350 /*
351 * Find the requested capability record ...
352 */
353 {
354 char buf[BUFSIZ];
355 char *b_end, *bp, *cp;
356 int c, slash;
357
358 /*
359 * Loop invariants:
360 * There is always room for one more character in record.
361 * R_end always points just past end of record.
362 * Rp always points just past last character in record.
363 * B_end always points just past last character in buf.
364 * Bp always points at next character in buf.
365 * Cp remembers where the last colon was.
366 */
367 b_end = buf;
368 bp = buf;
369 cp = 0;
370 slash = 0;
371 for (;;) {
372
373 /*
374 * Read in a line implementing (\, newline)
375 * line continuation.
376 */
377 rp = record;
378 for (;;) {
379 if (bp >= b_end) {
380 int n;
381
382 n = read(fd, buf, sizeof(buf));
383 if (n <= 0) {
384 if (myfd)
385 (void)close(fd);
386 if (n < 0) {
387 int serrno = errno;
388
389 free(record);
390 errno = serrno;
391 return (-2);
392 } else {
393 fd = -1;
394 eof = 1;
395 break;
396 }
397 }
398 b_end = buf+n;
399 bp = buf;
400 }
401
402 c = *bp++;
403 if (c == '\n') {
404 if (slash) {
405 slash = 0;
406 rp--;
407 continue;
408 } else
409 break;
410 }
411 if (slash) {
412 slash = 0;
413 cp = 0;
414 }
415 if (c == ':') {
416 /*
417 * If the field was `empty' (i.e.
418 * contained only white space), back up
419 * to the colon (eliminating the
420 * field).
421 */
422 if (cp)
423 rp = cp;
424 else
425 cp = rp;
426 } else if (c == '\\') {
427 slash = 1;
428 } else if (c != ' ' && c != '\t') {
429 /*
430 * Forget where the colon was, as this
431 * is not an empty field.
432 */
433 cp = 0;
434 }
435 *rp++ = c;
436
437 /*
438 * Enforce loop invariant: if no room
439 * left in record buffer, try to get
440 * some more.
441 */
442 if (rp >= r_end) {
443 u_int pos;
444 size_t newsize;
445
446 pos = rp - record;
447 newsize = r_end - record + BFRAG;
448 newrecord = realloc(record, newsize);
449 if (newrecord == NULL) {
450 free(record);
451 if (myfd)
452 (void)close(fd);
453 errno = ENOMEM;
454 return (-2);
455 }
456 record = newrecord;
457 r_end = record + newsize;
458 rp = record + pos;
459 }
460 }
461 /* Eliminate any white space after the last colon. */
462 if (cp)
463 rp = cp + 1;
464 /* Loop invariant lets us do this. */
465 *rp++ = '\0';
466
467 /*
468 * If encountered eof check next file.
469 */
470 if (eof)
471 break;
472
473 /*
474 * Toss blank lines and comments.
475 */
476 if (*record == '\0' || *record == '#')
477 continue;
478
479 /*
480 * See if this is the record we want ...
481 */
482 if (cgetmatch(record, name) == 0) {
483 if (nfield == NULL || !nfcmp(nfield, record)) {
484 foundit = 1;
485 break; /* found it! */
486 }
487 }
488 }
489 }
490 if (foundit)
491 break;
492 }
493
494 if (!foundit)
495 return (-1);
496
497 /*
498 * Got the capability record, but now we have to expand all tc=name
499 * references in it ...
500 */
501 tc_exp: {
502 char *newicap, *s;
503 size_t ilen, newilen;
504 int diff, iret, tclen;
505 char *icap, *scan, *tc, *tcstart, *tcend;
506
507 /*
508 * Loop invariants:
509 * There is room for one more character in record.
510 * R_end points just past end of record.
511 * Rp points just past last character in record.
512 * Scan points at remainder of record that needs to be
513 * scanned for tc=name constructs.
514 */
515 scan = record;
516 tc_not_resolved = 0;
517 for (;;) {
518 if ((tc = cgetcap(scan, "tc", '=')) == NULL)
519 break;
520
521 /*
522 * Find end of tc=name and stomp on the trailing `:'
523 * (if present) so we can use it to call ourselves.
524 */
525 s = tc;
526 for (;;)
527 if (*s == '\0')
528 break;
529 else
530 if (*s++ == ':') {
531 *(s - 1) = '\0';
532 break;
533 }
534 tcstart = tc - 3;
535 tclen = s - tcstart;
536 tcend = s;
537
538 iret = getent(&icap, &ilen, db_p, fd, tc, depth+1,
539 NULL);
540 newicap = icap; /* Put into a register. */
541 newilen = ilen;
542 if (iret != 0) {
543 /* an error */
544 if (iret < -1) {
545 if (myfd)
546 (void)close(fd);
547 free(record);
548 return (iret);
549 }
550 if (iret == 1)
551 tc_not_resolved = 1;
552 /* couldn't resolve tc */
553 if (iret == -1) {
554 *(s - 1) = ':';
555 scan = s - 1;
556 tc_not_resolved = 1;
557 continue;
558
559 }
560 }
561 /* not interested in name field of tc'ed record */
562 s = newicap;
563 for (;;)
564 if (*s == '\0')
565 break;
566 else
567 if (*s++ == ':')
568 break;
569 newilen -= s - newicap;
570 newicap = s;
571
572 /* make sure interpolated record is `:'-terminated */
573 s += newilen;
574 if (*(s-1) != ':') {
575 *s = ':'; /* overwrite NUL with : */
576 newilen++;
577 }
578
579 /*
580 * Make sure there's enough room to insert the
581 * new record.
582 */
583 diff = newilen - tclen;
584 if (diff >= r_end - rp) {
585 u_int pos, tcpos, tcposend;
586 size_t newsize;
587
588 pos = rp - record;
589 newsize = r_end - record + diff + BFRAG;
590 tcpos = tcstart - record;
591 tcposend = tcend - record;
592 newrecord = realloc(record, newsize);
593 if (newrecord == NULL) {
594 free(record);
595 if (myfd)
596 (void)close(fd);
597 free(icap);
598 errno = ENOMEM;
599 return (-2);
600 }
601 record = newrecord;
602 r_end = record + newsize;
603 rp = record + pos;
604 tcstart = record + tcpos;
605 tcend = record + tcposend;
606 }
607
608 /*
609 * Insert tc'ed record into our record.
610 */
611 s = tcstart + newilen;
612 memmove(s, tcend, (size_t)(rp - tcend));
613 memmove(tcstart, newicap, newilen);
614 rp += diff;
615 free(icap);
616
617 /*
618 * Start scan on `:' so next cgetcap works properly
619 * (cgetcap always skips first field).
620 */
621 scan = s-1;
622 }
623
624 }
625 /*
626 * Close file (if we opened it), give back any extra memory, and
627 * return capability, length and success.
628 */
629 if (myfd)
630 (void)close(fd);
631 *len = rp - record - 1; /* don't count NUL */
632 if (r_end > rp) {
633 if ((newrecord =
634 realloc(record, (size_t)(rp - record))) == NULL) {
635 free(record);
636 errno = ENOMEM;
637 return (-2);
638 }
639 record = newrecord;
640 }
641
642 *cap = record;
643 if (tc_not_resolved)
644 return (1);
645 return (0);
646 }
647
648 #ifndef SMALL
649 static int
650 cdbget(DB *capdbp, char **bp, const char *name)
651 {
652 DBT key;
653 DBT data;
654
655 _DIAGASSERT(capdbp != NULL);
656 _DIAGASSERT(bp != NULL);
657 _DIAGASSERT(name != NULL);
658
659 key.data = __UNCONST(name);
660 key.size = strlen(name);
661
662 for (;;) {
663 /* Get the reference. */
664 switch(capdbp->get(capdbp, &key, &data, 0)) {
665 case -1:
666 return (-2);
667 case 1:
668 return (-1);
669 }
670
671 /* If not an index to another record, leave. */
672 if (((char *)data.data)[0] != SHADOW)
673 break;
674
675 key.data = (char *)data.data + 1;
676 key.size = data.size - 1;
677 }
678
679 *bp = (char *)data.data + 1;
680 return (((char *)(data.data))[0] == TCERR ? 1 : 0);
681 }
682 #endif
683
684 /*
685 * Cgetmatch will return 0 if name is one of the names of the capability
686 * record buf, -1 if not.
687 */
688 int
689 cgetmatch(const char *buf, const char *name)
690 {
691 const char *np, *bp;
692
693 _DIAGASSERT(buf != NULL);
694 _DIAGASSERT(name != NULL);
695
696 /*
697 * Start search at beginning of record.
698 */
699 bp = buf;
700 for (;;) {
701 /*
702 * Try to match a record name.
703 */
704 np = name;
705 for (;;)
706 if (*np == '\0') {
707 if (*bp == '|' || *bp == ':' || *bp == '\0')
708 return (0);
709 else
710 break;
711 } else
712 if (*bp++ != *np++)
713 break;
714
715 /*
716 * Match failed, skip to next name in record.
717 */
718 if (bp > buf)
719 bp--; /* a '|' or ':' may have stopped the match */
720 else
721 return (-1);
722 for (;;)
723 if (*bp == '\0' || *bp == ':')
724 return (-1); /* match failed totally */
725 else
726 if (*bp++ == '|')
727 break; /* found next name */
728 }
729 }
730
731 int
732 cgetfirst(char **buf, const char * const *db_array)
733 {
734
735 _DIAGASSERT(buf != NULL);
736 _DIAGASSERT(db_array != NULL);
737
738 (void)cgetclose();
739 return (cgetnext(buf, db_array));
740 }
741
742 static FILE *pfp;
743 static int slash;
744 static const char * const *dbp;
745
746 int
747 cgetclose(void)
748 {
749 if (pfp != NULL) {
750 (void)fclose(pfp);
751 pfp = NULL;
752 }
753 dbp = NULL;
754 gottoprec = 0;
755 slash = 0;
756 return(0);
757 }
758
759 /*
760 * Cgetnext() gets either the first or next entry in the logical database
761 * specified by db_array. It returns 0 upon completion of the database, 1
762 * upon returning an entry with more remaining, and -1 if an error occurs.
763 */
764 int
765 cgetnext(char **bp, const char * const *db_array)
766 {
767 size_t len;
768 int status, done;
769 char *cp, *line, *rp, *np, buf[BSIZE], nbuf[BSIZE];
770 size_t dummy;
771
772 _DIAGASSERT(bp != NULL);
773 _DIAGASSERT(db_array != NULL);
774
775 if (dbp == NULL)
776 dbp = db_array;
777
778 if (pfp == NULL && (pfp = fopen(*dbp, "r")) == NULL) {
779 (void)cgetclose();
780 return (-1);
781 }
782 for(;;) {
783 if (toprec && !gottoprec) {
784 gottoprec = 1;
785 line = toprec;
786 } else {
787 line = fgetln(pfp, &len);
788 if (line == NULL && pfp) {
789 if (ferror(pfp)) {
790 (void)cgetclose();
791 return (-1);
792 } else {
793 (void)fclose(pfp);
794 pfp = NULL;
795 if (*++dbp == NULL) {
796 (void)cgetclose();
797 return (0);
798 } else if ((pfp =
799 fopen(*dbp, "r")) == NULL) {
800 (void)cgetclose();
801 return (-1);
802 } else
803 continue;
804 }
805 } else
806 line[len - 1] = '\0';
807 if (len == 1) {
808 slash = 0;
809 continue;
810 }
811 if (isspace((unsigned char)*line) ||
812 *line == ':' || *line == '#' || slash) {
813 if (line[len - 2] == '\\')
814 slash = 1;
815 else
816 slash = 0;
817 continue;
818 }
819 if (line[len - 2] == '\\')
820 slash = 1;
821 else
822 slash = 0;
823 }
824
825
826 /*
827 * Line points to a name line.
828 */
829 if (len > sizeof(nbuf))
830 return -1;
831 done = 0;
832 np = nbuf;
833 for (;;) {
834 for (cp = line; *cp != '\0'; cp++) {
835 if (*cp == ':') {
836 *np++ = ':';
837 done = 1;
838 break;
839 }
840 if (*cp == '\\')
841 break;
842 *np++ = *cp;
843 }
844 if (done) {
845 *np = '\0';
846 break;
847 } else { /* name field extends beyond the line */
848 line = fgetln(pfp, &len);
849 if (line == NULL && pfp) {
850 if (ferror(pfp)) {
851 (void)cgetclose();
852 return (-1);
853 }
854 (void)fclose(pfp);
855 pfp = NULL;
856 *np = '\0';
857 break;
858 } else
859 line[len - 1] = '\0';
860 }
861 }
862 if (len > sizeof(buf))
863 return -1;
864 rp = buf;
865 for(cp = nbuf; *cp != '\0'; cp++)
866 if (*cp == '|' || *cp == ':')
867 break;
868 else
869 *rp++ = *cp;
870
871 *rp = '\0';
872 /*
873 * XXX
874 * Last argument of getent here should be nbuf if we want true
875 * sequential access in the case of duplicates.
876 * With NULL, getent will return the first entry found
877 * rather than the duplicate entry record. This is a
878 * matter of semantics that should be resolved.
879 */
880 status = getent(bp, &dummy, db_array, -1, buf, 0, NULL);
881 if (status == -2 || status == -3)
882 (void)cgetclose();
883
884 return (status + 1);
885 }
886 /* NOTREACHED */
887 }
888
889 /*
890 * Cgetstr retrieves the value of the string capability cap from the
891 * capability record pointed to by buf. A pointer to a decoded, NUL
892 * terminated, malloc'd copy of the string is returned in the char *
893 * pointed to by str. The length of the string not including the trailing
894 * NUL is returned on success, -1 if the requested string capability
895 * couldn't be found, -2 if a system error was encountered (storage
896 * allocation failure).
897 */
898 int
899 cgetstr(char *buf, const char *cap, char **str)
900 {
901 u_int m_room;
902 const char *bp;
903 char *mp;
904 int len;
905 char *mem, *newmem;
906
907 _DIAGASSERT(buf != NULL);
908 _DIAGASSERT(cap != NULL);
909 _DIAGASSERT(str != NULL);
910
911 /*
912 * Find string capability cap
913 */
914 bp = cgetcap(buf, cap, '=');
915 if (bp == NULL)
916 return (-1);
917
918 /*
919 * Conversion / storage allocation loop ... Allocate memory in
920 * chunks SFRAG in size.
921 */
922 if ((mem = malloc(SFRAG)) == NULL) {
923 errno = ENOMEM;
924 return (-2); /* couldn't even allocate the first fragment */
925 }
926 m_room = SFRAG;
927 mp = mem;
928
929 while (*bp != ':' && *bp != '\0') {
930 /*
931 * Loop invariants:
932 * There is always room for one more character in mem.
933 * Mp always points just past last character in mem.
934 * Bp always points at next character in buf.
935 */
936 if (*bp == '^') {
937 bp++;
938 if (*bp == ':' || *bp == '\0')
939 break; /* drop unfinished escape */
940 *mp++ = *bp++ & 037;
941 } else if (*bp == '\\') {
942 bp++;
943 if (*bp == ':' || *bp == '\0')
944 break; /* drop unfinished escape */
945 if ('0' <= *bp && *bp <= '7') {
946 int n, i;
947
948 n = 0;
949 i = 3; /* maximum of three octal digits */
950 do {
951 n = n * 8 + (*bp++ - '0');
952 } while (--i && '0' <= *bp && *bp <= '7');
953 *mp++ = n;
954 }
955 else switch (*bp++) {
956 case 'b': case 'B':
957 *mp++ = '\b';
958 break;
959 case 't': case 'T':
960 *mp++ = '\t';
961 break;
962 case 'n': case 'N':
963 *mp++ = '\n';
964 break;
965 case 'f': case 'F':
966 *mp++ = '\f';
967 break;
968 case 'r': case 'R':
969 *mp++ = '\r';
970 break;
971 case 'e': case 'E':
972 *mp++ = ESC;
973 break;
974 case 'c': case 'C':
975 *mp++ = ':';
976 break;
977 default:
978 /*
979 * Catches '\', '^', and
980 * everything else.
981 */
982 *mp++ = *(bp-1);
983 break;
984 }
985 } else
986 *mp++ = *bp++;
987 m_room--;
988
989 /*
990 * Enforce loop invariant: if no room left in current
991 * buffer, try to get some more.
992 */
993 if (m_room == 0) {
994 size_t size = mp - mem;
995
996 if ((newmem = realloc(mem, size + SFRAG)) == NULL) {
997 free(mem);
998 return (-2);
999 }
1000 mem = newmem;
1001 m_room = SFRAG;
1002 mp = mem + size;
1003 }
1004 }
1005 *mp++ = '\0'; /* loop invariant let's us do this */
1006 m_room--;
1007 len = mp - mem - 1;
1008
1009 /*
1010 * Give back any extra memory and return value and success.
1011 */
1012 if (m_room != 0) {
1013 if ((newmem = realloc(mem, (size_t)(mp - mem))) == NULL) {
1014 free(mem);
1015 return (-2);
1016 }
1017 mem = newmem;
1018 }
1019 *str = mem;
1020 return (len);
1021 }
1022
1023 /*
1024 * Cgetustr retrieves the value of the string capability cap from the
1025 * capability record pointed to by buf. The difference between cgetustr()
1026 * and cgetstr() is that cgetustr does not decode escapes but rather treats
1027 * all characters literally. A pointer to a NUL terminated malloc'd
1028 * copy of the string is returned in the char pointed to by str. The
1029 * length of the string not including the trailing NUL is returned on success,
1030 * -1 if the requested string capability couldn't be found, -2 if a system
1031 * error was encountered (storage allocation failure).
1032 */
1033 int
1034 cgetustr(char *buf, const char *cap, char **str)
1035 {
1036 u_int m_room;
1037 const char *bp;
1038 char *mp;
1039 int len;
1040 char *mem, *newmem;
1041
1042 _DIAGASSERT(buf != NULL);
1043 _DIAGASSERT(cap != NULL);
1044 _DIAGASSERT(str != NULL);
1045
1046 /*
1047 * Find string capability cap
1048 */
1049 if ((bp = cgetcap(buf, cap, '=')) == NULL)
1050 return (-1);
1051
1052 /*
1053 * Conversion / storage allocation loop ... Allocate memory in
1054 * chunks SFRAG in size.
1055 */
1056 if ((mem = malloc(SFRAG)) == NULL) {
1057 errno = ENOMEM;
1058 return (-2); /* couldn't even allocate the first fragment */
1059 }
1060 m_room = SFRAG;
1061 mp = mem;
1062
1063 while (*bp != ':' && *bp != '\0') {
1064 /*
1065 * Loop invariants:
1066 * There is always room for one more character in mem.
1067 * Mp always points just past last character in mem.
1068 * Bp always points at next character in buf.
1069 */
1070 *mp++ = *bp++;
1071 m_room--;
1072
1073 /*
1074 * Enforce loop invariant: if no room left in current
1075 * buffer, try to get some more.
1076 */
1077 if (m_room == 0) {
1078 size_t size = mp - mem;
1079
1080 if ((newmem = realloc(mem, size + SFRAG)) == NULL) {
1081 free(mem);
1082 return (-2);
1083 }
1084 mem = newmem;
1085 m_room = SFRAG;
1086 mp = mem + size;
1087 }
1088 }
1089 *mp++ = '\0'; /* loop invariant let's us do this */
1090 m_room--;
1091 len = mp - mem - 1;
1092
1093 /*
1094 * Give back any extra memory and return value and success.
1095 */
1096 if (m_room != 0) {
1097 if ((newmem = realloc(mem, (size_t)(mp - mem))) == NULL) {
1098 free(mem);
1099 return (-2);
1100 }
1101 mem = newmem;
1102 }
1103 *str = mem;
1104 return (len);
1105 }
1106
1107 /*
1108 * Cgetnum retrieves the value of the numeric capability cap from the
1109 * capability record pointed to by buf. The numeric value is returned in
1110 * the long pointed to by num. 0 is returned on success, -1 if the requested
1111 * numeric capability couldn't be found.
1112 */
1113 int
1114 cgetnum(char *buf, const char *cap, long *num)
1115 {
1116 long n;
1117 int base, digit;
1118 const char *bp;
1119
1120 _DIAGASSERT(buf != NULL);
1121 _DIAGASSERT(cap != NULL);
1122 _DIAGASSERT(num != NULL);
1123
1124 /*
1125 * Find numeric capability cap
1126 */
1127 bp = cgetcap(buf, cap, '#');
1128 if (bp == NULL)
1129 return (-1);
1130
1131 /*
1132 * Look at value and determine numeric base:
1133 * 0x... or 0X... hexadecimal,
1134 * else 0... octal,
1135 * else decimal.
1136 */
1137 if (*bp == '0') {
1138 bp++;
1139 if (*bp == 'x' || *bp == 'X') {
1140 bp++;
1141 base = 16;
1142 } else
1143 base = 8;
1144 } else
1145 base = 10;
1146
1147 /*
1148 * Conversion loop ...
1149 */
1150 n = 0;
1151 for (;;) {
1152 if ('0' <= *bp && *bp <= '9')
1153 digit = *bp - '0';
1154 else if ('a' <= *bp && *bp <= 'f')
1155 digit = 10 + *bp - 'a';
1156 else if ('A' <= *bp && *bp <= 'F')
1157 digit = 10 + *bp - 'A';
1158 else
1159 break;
1160
1161 if (digit >= base)
1162 break;
1163
1164 n = n * base + digit;
1165 bp++;
1166 }
1167
1168 /*
1169 * Return value and success.
1170 */
1171 *num = n;
1172 return (0);
1173 }
1174
1175
1176 /*
1177 * Compare name field of record.
1178 */
1179 static int
1180 nfcmp(char *nf, char *rec)
1181 {
1182 char *cp, tmp;
1183 int ret;
1184
1185 _DIAGASSERT(nf != NULL);
1186 _DIAGASSERT(rec != NULL);
1187
1188 for (cp = rec; *cp != ':'; cp++)
1189 ;
1190
1191 tmp = *(cp + 1);
1192 *(cp + 1) = '\0';
1193 ret = strcmp(nf, rec);
1194 *(cp + 1) = tmp;
1195
1196 return (ret);
1197 }
1198