xref: /openbsd-src/usr.sbin/netgroup_mkdb/util.c (revision cbb3c7677af27f85936485864b5ea25ee018a571)
1 /*
2  * Copyright (c) 1994 Christos Zoulas
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. All advertising materials mentioning features or use of this software
14  *    must display the following acknowledgement:
15  *	This product includes software developed by Christos Zoulas.
16  * 4. The name of the author may not be used to endorse or promote products
17  *    derived from this software without specific prior written permission.
18  *
19  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS
20  * OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
21  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY
23  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29  * SUCH DAMAGE.
30  */
31 
32 #include <err.h>
33 #include <stdlib.h>
34 #include <string.h>
35 
36 #include "util.h"
37 
38 /* emalloc():
39  *	Error checked malloc
40  */
41 void *
emalloc(size_t s)42 emalloc(size_t s)
43 {
44 	void *ptr = malloc(s);
45 	if (ptr == NULL)
46 		err(1, NULL);
47 	return ptr;
48 }
49 
50 
51 /* erealloc():
52  *	Error checked realloc
53  */
54 void *
erealloc(void * p,size_t s)55 erealloc(void *p, size_t s)
56 {
57 	void *ptr = realloc(p, s);
58 	if (ptr == NULL)
59 		err(1, NULL);
60 	return ptr;
61 }
62 
63 
64 /* get_line():
65  *	Read a line from a file parsing continuations ending in \
66  *	and eliminating trailing newlines.
67  */
68 char *
get_line(FILE * fp,size_t * size)69 get_line(FILE *fp, size_t *size)
70 {
71 	size_t s, len = 0;
72 	char *buf = NULL, *ptr;
73 	int cnt = 1;
74 
75 	while (cnt) {
76 		if ((ptr = fgetln(fp, &s)) == NULL) {
77 			*size = len;
78 			return buf;
79 		}
80 		if (ptr[s - 1] == '\n')		/* the newline may be missing at EOF */
81 			s--;				/* forget newline */
82 		if (s && (cnt = (ptr[s - 1] == '\\')))	/* check for \\ */
83 			s--;				/* forget \\ */
84 
85 		buf = erealloc(buf, len + s + 1);
86 		memcpy(buf + len, ptr, s);
87 		len += s;
88 		buf[len] = '\0';
89 	}
90 	*size = len;
91 	return buf;
92 }
93