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