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