getcap.c revision 1.45 1 /* $NetBSD: getcap.c,v 1.45 2006/04/02 03:26:03 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.45 2006/04/02 03:26:03 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 /* coverity[+alloc : arg-*0] */
217 int
218 cgetent(char **buf, const char * const *db_array, const char *name)
219 {
220 size_t dummy;
221
222 _DIAGASSERT(buf != NULL);
223 _DIAGASSERT(db_array != NULL);
224 _DIAGASSERT(name != NULL);
225
226 return (getent(buf, &dummy, db_array, -1, name, 0, NULL));
227 }
228
229 /*
230 * Getent implements the functions of cgetent. If fd is non-negative,
231 * *db_array has already been opened and fd is the open file descriptor. We
232 * do this to save time and avoid using up file descriptors for tc=
233 * recursions.
234 *
235 * Getent returns the same success/failure codes as cgetent. On success, a
236 * pointer to a malloc'ed capability record with all tc= capabilities fully
237 * expanded and its length (not including trailing ASCII NUL) are left in
238 * *cap and *len.
239 *
240 * Basic algorithm:
241 * + Allocate memory incrementally as needed in chunks of size BFRAG
242 * for capability buffer.
243 * + Recurse for each tc=name and interpolate result. Stop when all
244 * names interpolated, a name can't be found, or depth exceeds
245 * MAX_RECURSION.
246 */
247 /* coverity[+alloc : arg-*0] */
248 static int
249 getent(char **cap, size_t *len, const char * const *db_array, int fd,
250 const char *name, int depth, char *nfield)
251 {
252 #ifndef SMALL
253 DB *capdbp;
254 char pbuf[MAXPATHLEN];
255 char *cbuf;
256 int retval;
257 size_t clen;
258 #endif
259 char *record, *newrecord;
260 char *r_end, *rp = NULL; /* pacify gcc */
261 const char * const *db_p;
262 int myfd = 0, eof, foundit;
263 int tc_not_resolved;
264
265 _DIAGASSERT(cap != NULL);
266 _DIAGASSERT(len != NULL);
267 _DIAGASSERT(db_array != NULL);
268 /* fd may be -1 */
269 _DIAGASSERT(name != NULL);
270 /* nfield may be NULL */
271
272 /*
273 * Return with ``loop detected'' error if we've recursed more than
274 * MAX_RECURSION times.
275 */
276 if (depth > MAX_RECURSION)
277 return (-3);
278
279 /*
280 * Check if we have a top record from cgetset().
281 */
282 if (depth == 0 && toprec != NULL && cgetmatch(toprec, name) == 0) {
283 if ((record = malloc (topreclen + BFRAG)) == NULL) {
284 errno = ENOMEM;
285 return (-2);
286 }
287 (void)strcpy(record, toprec); /* XXX: strcpy is safe */
288 db_p = db_array;
289 rp = record + topreclen + 1;
290 r_end = rp + BFRAG;
291 goto tc_exp;
292 }
293 /*
294 * Allocate first chunk of memory.
295 */
296 if ((record = malloc(BFRAG)) == NULL) {
297 errno = ENOMEM;
298 return (-2);
299 }
300 r_end = record + BFRAG;
301 foundit = 0;
302 /*
303 * Loop through database array until finding the record.
304 */
305
306 for (db_p = db_array; *db_p != NULL; db_p++) {
307 eof = 0;
308
309 /*
310 * Open database if not already open.
311 */
312
313 if (fd >= 0) {
314 (void)lseek(fd, (off_t)0, SEEK_SET);
315 } else {
316 #ifndef SMALL
317 (void)snprintf(pbuf, sizeof(pbuf), "%s.db", *db_p);
318 if ((capdbp = dbopen(pbuf, O_RDONLY, 0, DB_HASH, 0))
319 != NULL) {
320 free(record);
321 retval = cdbget(capdbp, &record, name);
322 if (retval < 0) {
323 /* no record available */
324 (void)capdbp->close(capdbp);
325 return (retval);
326 }
327 /* save the data; close frees it */
328 clen = strlen(record);
329 cbuf = malloc(clen + 1);
330 memmove(cbuf, record, clen + 1);
331 if (capdbp->close(capdbp) < 0) {
332 int serrno = errno;
333
334 free(cbuf);
335 errno = serrno;
336 return (-2);
337 }
338 *len = clen;
339 *cap = cbuf;
340 return (retval);
341 } else
342 #endif
343 {
344 fd = open(*db_p, O_RDONLY, 0);
345 if (fd < 0) {
346 /* No error on unfound file. */
347 continue;
348 }
349 myfd = 1;
350 }
351 }
352 /*
353 * Find the requested capability record ...
354 */
355 {
356 char buf[BUFSIZ];
357 char *b_end, *bp, *cp;
358 int c, slash;
359
360 /*
361 * Loop invariants:
362 * There is always room for one more character in record.
363 * R_end always points just past end of record.
364 * Rp always points just past last character in record.
365 * B_end always points just past last character in buf.
366 * Bp always points at next character in buf.
367 * Cp remembers where the last colon was.
368 */
369 b_end = buf;
370 bp = buf;
371 cp = 0;
372 slash = 0;
373 for (;;) {
374
375 /*
376 * Read in a line implementing (\, newline)
377 * line continuation.
378 */
379 rp = record;
380 for (;;) {
381 if (bp >= b_end) {
382 int n;
383
384 n = read(fd, buf, sizeof(buf));
385 if (n <= 0) {
386 if (myfd)
387 (void)close(fd);
388 if (n < 0) {
389 int serrno = errno;
390
391 free(record);
392 errno = serrno;
393 return (-2);
394 } else {
395 fd = -1;
396 eof = 1;
397 break;
398 }
399 }
400 b_end = buf+n;
401 bp = buf;
402 }
403
404 c = *bp++;
405 if (c == '\n') {
406 if (slash) {
407 slash = 0;
408 rp--;
409 continue;
410 } else
411 break;
412 }
413 if (slash) {
414 slash = 0;
415 cp = 0;
416 }
417 if (c == ':') {
418 /*
419 * If the field was `empty' (i.e.
420 * contained only white space), back up
421 * to the colon (eliminating the
422 * field).
423 */
424 if (cp)
425 rp = cp;
426 else
427 cp = rp;
428 } else if (c == '\\') {
429 slash = 1;
430 } else if (c != ' ' && c != '\t') {
431 /*
432 * Forget where the colon was, as this
433 * is not an empty field.
434 */
435 cp = 0;
436 }
437 *rp++ = c;
438
439 /*
440 * Enforce loop invariant: if no room
441 * left in record buffer, try to get
442 * some more.
443 */
444 if (rp >= r_end) {
445 u_int pos;
446 size_t newsize;
447
448 pos = rp - record;
449 newsize = r_end - record + BFRAG;
450 newrecord = realloc(record, newsize);
451 if (newrecord == NULL) {
452 free(record);
453 if (myfd)
454 (void)close(fd);
455 errno = ENOMEM;
456 return (-2);
457 }
458 record = newrecord;
459 r_end = record + newsize;
460 rp = record + pos;
461 }
462 }
463 /* Eliminate any white space after the last colon. */
464 if (cp)
465 rp = cp + 1;
466 /* Loop invariant lets us do this. */
467 *rp++ = '\0';
468
469 /*
470 * If encountered eof check next file.
471 */
472 if (eof)
473 break;
474
475 /*
476 * Toss blank lines and comments.
477 */
478 if (*record == '\0' || *record == '#')
479 continue;
480
481 /*
482 * See if this is the record we want ...
483 */
484 if (cgetmatch(record, name) == 0) {
485 if (nfield == NULL || !nfcmp(nfield, record)) {
486 foundit = 1;
487 break; /* found it! */
488 }
489 }
490 }
491 }
492 if (foundit)
493 break;
494 }
495
496 if (!foundit)
497 return (-1);
498
499 /*
500 * Got the capability record, but now we have to expand all tc=name
501 * references in it ...
502 */
503 tc_exp: {
504 char *newicap, *s;
505 size_t ilen, newilen;
506 int diff, iret, tclen;
507 char *icap, *scan, *tc, *tcstart, *tcend;
508
509 /*
510 * Loop invariants:
511 * There is room for one more character in record.
512 * R_end points just past end of record.
513 * Rp points just past last character in record.
514 * Scan points at remainder of record that needs to be
515 * scanned for tc=name constructs.
516 */
517 scan = record;
518 tc_not_resolved = 0;
519 for (;;) {
520 if ((tc = cgetcap(scan, "tc", '=')) == NULL)
521 break;
522
523 /*
524 * Find end of tc=name and stomp on the trailing `:'
525 * (if present) so we can use it to call ourselves.
526 */
527 s = tc;
528 for (;;)
529 if (*s == '\0')
530 break;
531 else
532 if (*s++ == ':') {
533 *(s - 1) = '\0';
534 break;
535 }
536 tcstart = tc - 3;
537 tclen = s - tcstart;
538 tcend = s;
539
540 iret = getent(&icap, &ilen, db_p, fd, tc, depth+1,
541 NULL);
542 newicap = icap; /* Put into a register. */
543 newilen = ilen;
544 if (iret != 0) {
545 /* an error */
546 if (iret < -1) {
547 if (myfd)
548 (void)close(fd);
549 free(record);
550 return (iret);
551 }
552 if (iret == 1)
553 tc_not_resolved = 1;
554 /* couldn't resolve tc */
555 if (iret == -1) {
556 *(s - 1) = ':';
557 scan = s - 1;
558 tc_not_resolved = 1;
559 continue;
560
561 }
562 }
563 /* not interested in name field of tc'ed record */
564 s = newicap;
565 for (;;)
566 if (*s == '\0')
567 break;
568 else
569 if (*s++ == ':')
570 break;
571 newilen -= s - newicap;
572 newicap = s;
573
574 /* make sure interpolated record is `:'-terminated */
575 s += newilen;
576 if (*(s-1) != ':') {
577 *s = ':'; /* overwrite NUL with : */
578 newilen++;
579 }
580
581 /*
582 * Make sure there's enough room to insert the
583 * new record.
584 */
585 diff = newilen - tclen;
586 if (diff >= r_end - rp) {
587 u_int pos, tcpos, tcposend;
588 size_t newsize;
589
590 pos = rp - record;
591 newsize = r_end - record + diff + BFRAG;
592 tcpos = tcstart - record;
593 tcposend = tcend - record;
594 newrecord = realloc(record, newsize);
595 if (newrecord == NULL) {
596 free(record);
597 if (myfd)
598 (void)close(fd);
599 free(icap);
600 errno = ENOMEM;
601 return (-2);
602 }
603 record = newrecord;
604 r_end = record + newsize;
605 rp = record + pos;
606 tcstart = record + tcpos;
607 tcend = record + tcposend;
608 }
609
610 /*
611 * Insert tc'ed record into our record.
612 */
613 s = tcstart + newilen;
614 memmove(s, tcend, (size_t)(rp - tcend));
615 memmove(tcstart, newicap, newilen);
616 rp += diff;
617 free(icap);
618
619 /*
620 * Start scan on `:' so next cgetcap works properly
621 * (cgetcap always skips first field).
622 */
623 scan = s-1;
624 }
625
626 }
627 /*
628 * Close file (if we opened it), give back any extra memory, and
629 * return capability, length and success.
630 */
631 if (myfd)
632 (void)close(fd);
633 *len = rp - record - 1; /* don't count NUL */
634 if (r_end > rp) {
635 if ((newrecord =
636 realloc(record, (size_t)(rp - record))) == NULL) {
637 free(record);
638 errno = ENOMEM;
639 return (-2);
640 }
641 record = newrecord;
642 }
643
644 *cap = record;
645 if (tc_not_resolved)
646 return (1);
647 return (0);
648 }
649
650 #ifndef SMALL
651 static int
652 cdbget(DB *capdbp, char **bp, const char *name)
653 {
654 DBT key;
655 DBT data;
656
657 _DIAGASSERT(capdbp != NULL);
658 _DIAGASSERT(bp != NULL);
659 _DIAGASSERT(name != NULL);
660
661 key.data = __UNCONST(name);
662 key.size = strlen(name);
663
664 for (;;) {
665 /* Get the reference. */
666 switch(capdbp->get(capdbp, &key, &data, 0)) {
667 case -1:
668 return (-2);
669 case 1:
670 return (-1);
671 }
672
673 /* If not an index to another record, leave. */
674 if (((char *)data.data)[0] != SHADOW)
675 break;
676
677 key.data = (char *)data.data + 1;
678 key.size = data.size - 1;
679 }
680
681 *bp = (char *)data.data + 1;
682 return (((char *)(data.data))[0] == TCERR ? 1 : 0);
683 }
684 #endif
685
686 /*
687 * Cgetmatch will return 0 if name is one of the names of the capability
688 * record buf, -1 if not.
689 */
690 int
691 cgetmatch(const char *buf, const char *name)
692 {
693 const char *np, *bp;
694
695 _DIAGASSERT(buf != NULL);
696 _DIAGASSERT(name != NULL);
697
698 /*
699 * Start search at beginning of record.
700 */
701 bp = buf;
702 for (;;) {
703 /*
704 * Try to match a record name.
705 */
706 np = name;
707 for (;;)
708 if (*np == '\0') {
709 if (*bp == '|' || *bp == ':' || *bp == '\0')
710 return (0);
711 else
712 break;
713 } else
714 if (*bp++ != *np++)
715 break;
716
717 /*
718 * Match failed, skip to next name in record.
719 */
720 if (bp > buf)
721 bp--; /* a '|' or ':' may have stopped the match */
722 else
723 return (-1);
724 for (;;)
725 if (*bp == '\0' || *bp == ':')
726 return (-1); /* match failed totally */
727 else
728 if (*bp++ == '|')
729 break; /* found next name */
730 }
731 }
732
733 int
734 cgetfirst(char **buf, const char * const *db_array)
735 {
736
737 _DIAGASSERT(buf != NULL);
738 _DIAGASSERT(db_array != NULL);
739
740 (void)cgetclose();
741 return (cgetnext(buf, db_array));
742 }
743
744 static FILE *pfp;
745 static int slash;
746 static const char * const *dbp;
747
748 int
749 cgetclose(void)
750 {
751 if (pfp != NULL) {
752 (void)fclose(pfp);
753 pfp = NULL;
754 }
755 dbp = NULL;
756 gottoprec = 0;
757 slash = 0;
758 return(0);
759 }
760
761 /*
762 * Cgetnext() gets either the first or next entry in the logical database
763 * specified by db_array. It returns 0 upon completion of the database, 1
764 * upon returning an entry with more remaining, and -1 if an error occurs.
765 */
766 /* coverity[+alloc : arg-*0] */
767 int
768 cgetnext(char **bp, const char * const *db_array)
769 {
770 size_t len = 0;
771 int status, done;
772 char *cp, *line, *rp, *np, buf[BSIZE], nbuf[BSIZE];
773 size_t dummy;
774
775 _DIAGASSERT(bp != NULL);
776 _DIAGASSERT(db_array != NULL);
777
778 if (dbp == NULL)
779 dbp = db_array;
780
781 if (pfp == NULL && (pfp = fopen(*dbp, "r")) == NULL) {
782 (void)cgetclose();
783 return (-1);
784 }
785 for(;;) {
786 if (toprec && !gottoprec) {
787 gottoprec = 1;
788 line = toprec;
789 } else {
790 line = fgetln(pfp, &len);
791 if (line == NULL) {
792 if (pfp == NULL)
793 return -1;
794 if (ferror(pfp)) {
795 (void)cgetclose();
796 return (-1);
797 } else {
798 (void)fclose(pfp);
799 pfp = NULL;
800 if (*++dbp == NULL) {
801 (void)cgetclose();
802 return (0);
803 } else if ((pfp =
804 fopen(*dbp, "r")) == NULL) {
805 (void)cgetclose();
806 return (-1);
807 } else
808 continue;
809 }
810 } else
811 line[len - 1] = '\0';
812 if (len == 1) {
813 slash = 0;
814 continue;
815 }
816 if (isspace((unsigned char)*line) ||
817 *line == ':' || *line == '#' || slash) {
818 if (line[len - 2] == '\\')
819 slash = 1;
820 else
821 slash = 0;
822 continue;
823 }
824 if (line[len - 2] == '\\')
825 slash = 1;
826 else
827 slash = 0;
828 }
829
830
831 /*
832 * Line points to a name line.
833 */
834 if (len > sizeof(nbuf))
835 return -1;
836 done = 0;
837 np = nbuf;
838 for (;;) {
839 for (cp = line; *cp != '\0'; cp++) {
840 if (*cp == ':') {
841 *np++ = ':';
842 done = 1;
843 break;
844 }
845 if (*cp == '\\')
846 break;
847 *np++ = *cp;
848 }
849 if (done) {
850 *np = '\0';
851 break;
852 } else { /* name field extends beyond the line */
853 line = fgetln(pfp, &len);
854 if (line == NULL && pfp) {
855 if (ferror(pfp)) {
856 (void)cgetclose();
857 return (-1);
858 }
859 (void)fclose(pfp);
860 pfp = NULL;
861 *np = '\0';
862 break;
863 } else
864 line[len - 1] = '\0';
865 }
866 }
867 if (len > sizeof(buf))
868 return -1;
869 rp = buf;
870 for(cp = nbuf; *cp != '\0'; cp++)
871 if (*cp == '|' || *cp == ':')
872 break;
873 else
874 *rp++ = *cp;
875
876 *rp = '\0';
877 /*
878 * XXX
879 * Last argument of getent here should be nbuf if we want true
880 * sequential access in the case of duplicates.
881 * With NULL, getent will return the first entry found
882 * rather than the duplicate entry record. This is a
883 * matter of semantics that should be resolved.
884 */
885 status = getent(bp, &dummy, db_array, -1, buf, 0, NULL);
886 if (status == -2 || status == -3)
887 (void)cgetclose();
888
889 return (status + 1);
890 }
891 /* NOTREACHED */
892 }
893
894 /*
895 * Cgetstr retrieves the value of the string capability cap from the
896 * capability record pointed to by buf. A pointer to a decoded, NUL
897 * terminated, malloc'd copy of the string is returned in the char *
898 * pointed to by str. The length of the string not including the trailing
899 * NUL is returned on success, -1 if the requested string capability
900 * couldn't be found, -2 if a system error was encountered (storage
901 * allocation failure).
902 */
903 int
904 cgetstr(char *buf, const char *cap, char **str)
905 {
906 u_int m_room;
907 const char *bp;
908 char *mp;
909 int len;
910 char *mem, *newmem;
911
912 _DIAGASSERT(buf != NULL);
913 _DIAGASSERT(cap != NULL);
914 _DIAGASSERT(str != NULL);
915
916 /*
917 * Find string capability cap
918 */
919 bp = cgetcap(buf, cap, '=');
920 if (bp == NULL)
921 return (-1);
922
923 /*
924 * Conversion / storage allocation loop ... Allocate memory in
925 * chunks SFRAG in size.
926 */
927 if ((mem = malloc(SFRAG)) == NULL) {
928 errno = ENOMEM;
929 return (-2); /* couldn't even allocate the first fragment */
930 }
931 m_room = SFRAG;
932 mp = mem;
933
934 while (*bp != ':' && *bp != '\0') {
935 /*
936 * Loop invariants:
937 * There is always room for one more character in mem.
938 * Mp always points just past last character in mem.
939 * Bp always points at next character in buf.
940 */
941 if (*bp == '^') {
942 bp++;
943 if (*bp == ':' || *bp == '\0')
944 break; /* drop unfinished escape */
945 *mp++ = *bp++ & 037;
946 } else if (*bp == '\\') {
947 bp++;
948 if (*bp == ':' || *bp == '\0')
949 break; /* drop unfinished escape */
950 if ('0' <= *bp && *bp <= '7') {
951 int n, i;
952
953 n = 0;
954 i = 3; /* maximum of three octal digits */
955 do {
956 n = n * 8 + (*bp++ - '0');
957 } while (--i && '0' <= *bp && *bp <= '7');
958 *mp++ = n;
959 }
960 else switch (*bp++) {
961 case 'b': case 'B':
962 *mp++ = '\b';
963 break;
964 case 't': case 'T':
965 *mp++ = '\t';
966 break;
967 case 'n': case 'N':
968 *mp++ = '\n';
969 break;
970 case 'f': case 'F':
971 *mp++ = '\f';
972 break;
973 case 'r': case 'R':
974 *mp++ = '\r';
975 break;
976 case 'e': case 'E':
977 *mp++ = ESC;
978 break;
979 case 'c': case 'C':
980 *mp++ = ':';
981 break;
982 default:
983 /*
984 * Catches '\', '^', and
985 * everything else.
986 */
987 *mp++ = *(bp-1);
988 break;
989 }
990 } else
991 *mp++ = *bp++;
992 m_room--;
993
994 /*
995 * Enforce loop invariant: if no room left in current
996 * buffer, try to get some more.
997 */
998 if (m_room == 0) {
999 size_t size = mp - mem;
1000
1001 if ((newmem = realloc(mem, size + SFRAG)) == NULL) {
1002 free(mem);
1003 return (-2);
1004 }
1005 mem = newmem;
1006 m_room = SFRAG;
1007 mp = mem + size;
1008 }
1009 }
1010 *mp++ = '\0'; /* loop invariant let's us do this */
1011 m_room--;
1012 len = mp - mem - 1;
1013
1014 /*
1015 * Give back any extra memory and return value and success.
1016 */
1017 if (m_room != 0) {
1018 if ((newmem = realloc(mem, (size_t)(mp - mem))) == NULL) {
1019 free(mem);
1020 return (-2);
1021 }
1022 mem = newmem;
1023 }
1024 *str = mem;
1025 return (len);
1026 }
1027
1028 /*
1029 * Cgetustr retrieves the value of the string capability cap from the
1030 * capability record pointed to by buf. The difference between cgetustr()
1031 * and cgetstr() is that cgetustr does not decode escapes but rather treats
1032 * all characters literally. A pointer to a NUL terminated malloc'd
1033 * copy of the string is returned in the char pointed to by str. The
1034 * length of the string not including the trailing NUL is returned on success,
1035 * -1 if the requested string capability couldn't be found, -2 if a system
1036 * error was encountered (storage allocation failure).
1037 */
1038 int
1039 cgetustr(char *buf, const char *cap, char **str)
1040 {
1041 u_int m_room;
1042 const char *bp;
1043 char *mp;
1044 int len;
1045 char *mem, *newmem;
1046
1047 _DIAGASSERT(buf != NULL);
1048 _DIAGASSERT(cap != NULL);
1049 _DIAGASSERT(str != NULL);
1050
1051 /*
1052 * Find string capability cap
1053 */
1054 if ((bp = cgetcap(buf, cap, '=')) == NULL)
1055 return (-1);
1056
1057 /*
1058 * Conversion / storage allocation loop ... Allocate memory in
1059 * chunks SFRAG in size.
1060 */
1061 if ((mem = malloc(SFRAG)) == NULL) {
1062 errno = ENOMEM;
1063 return (-2); /* couldn't even allocate the first fragment */
1064 }
1065 m_room = SFRAG;
1066 mp = mem;
1067
1068 while (*bp != ':' && *bp != '\0') {
1069 /*
1070 * Loop invariants:
1071 * There is always room for one more character in mem.
1072 * Mp always points just past last character in mem.
1073 * Bp always points at next character in buf.
1074 */
1075 *mp++ = *bp++;
1076 m_room--;
1077
1078 /*
1079 * Enforce loop invariant: if no room left in current
1080 * buffer, try to get some more.
1081 */
1082 if (m_room == 0) {
1083 size_t size = mp - mem;
1084
1085 if ((newmem = realloc(mem, size + SFRAG)) == NULL) {
1086 free(mem);
1087 return (-2);
1088 }
1089 mem = newmem;
1090 m_room = SFRAG;
1091 mp = mem + size;
1092 }
1093 }
1094 *mp++ = '\0'; /* loop invariant let's us do this */
1095 m_room--;
1096 len = mp - mem - 1;
1097
1098 /*
1099 * Give back any extra memory and return value and success.
1100 */
1101 if (m_room != 0) {
1102 if ((newmem = realloc(mem, (size_t)(mp - mem))) == NULL) {
1103 free(mem);
1104 return (-2);
1105 }
1106 mem = newmem;
1107 }
1108 *str = mem;
1109 return (len);
1110 }
1111
1112 /*
1113 * Cgetnum retrieves the value of the numeric capability cap from the
1114 * capability record pointed to by buf. The numeric value is returned in
1115 * the long pointed to by num. 0 is returned on success, -1 if the requested
1116 * numeric capability couldn't be found.
1117 */
1118 int
1119 cgetnum(char *buf, const char *cap, long *num)
1120 {
1121 long n;
1122 int base, digit;
1123 const char *bp;
1124
1125 _DIAGASSERT(buf != NULL);
1126 _DIAGASSERT(cap != NULL);
1127 _DIAGASSERT(num != NULL);
1128
1129 /*
1130 * Find numeric capability cap
1131 */
1132 bp = cgetcap(buf, cap, '#');
1133 if (bp == NULL)
1134 return (-1);
1135
1136 /*
1137 * Look at value and determine numeric base:
1138 * 0x... or 0X... hexadecimal,
1139 * else 0... octal,
1140 * else decimal.
1141 */
1142 if (*bp == '0') {
1143 bp++;
1144 if (*bp == 'x' || *bp == 'X') {
1145 bp++;
1146 base = 16;
1147 } else
1148 base = 8;
1149 } else
1150 base = 10;
1151
1152 /*
1153 * Conversion loop ...
1154 */
1155 n = 0;
1156 for (;;) {
1157 if ('0' <= *bp && *bp <= '9')
1158 digit = *bp - '0';
1159 else if ('a' <= *bp && *bp <= 'f')
1160 digit = 10 + *bp - 'a';
1161 else if ('A' <= *bp && *bp <= 'F')
1162 digit = 10 + *bp - 'A';
1163 else
1164 break;
1165
1166 if (digit >= base)
1167 break;
1168
1169 n = n * base + digit;
1170 bp++;
1171 }
1172
1173 /*
1174 * Return value and success.
1175 */
1176 *num = n;
1177 return (0);
1178 }
1179
1180
1181 /*
1182 * Compare name field of record.
1183 */
1184 static int
1185 nfcmp(char *nf, char *rec)
1186 {
1187 char *cp, tmp;
1188 int ret;
1189
1190 _DIAGASSERT(nf != NULL);
1191 _DIAGASSERT(rec != NULL);
1192
1193 for (cp = rec; *cp != ':'; cp++)
1194 ;
1195
1196 tmp = *(cp + 1);
1197 *(cp + 1) = '\0';
1198 ret = strcmp(nf, rec);
1199 *(cp + 1) = tmp;
1200
1201 return (ret);
1202 }
1203