getprotoent.c revision 157779 
1/*
2 * Copyright (c) 1983, 1993
3 *	The Regents of the University of California.  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 the University of
16 *	California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#if defined(LIBC_SCCS) && !defined(lint)
35static char sccsid[] = "@(#)getprotoent.c	8.1 (Berkeley) 6/4/93";
36#endif /* LIBC_SCCS and not lint */
37#include <sys/cdefs.h>
38__FBSDID("$FreeBSD: head/lib/libc/net/getprotoent.c 157779 2006-04-15 16:20:27Z ume $");
39
40#include <sys/param.h>
41#include <sys/types.h>
42#include <sys/socket.h>
43#include <errno.h>
44#include <limits.h>
45#include <netdb.h>
46#include <stdio.h>
47#include <stdlib.h>
48#include <string.h>
49#include "namespace.h"
50#include "reentrant.h"
51#include "un-namespace.h"
52#include "netdb_private.h"
53
54NETDB_THREAD_ALLOC(protoent_data)
55NETDB_THREAD_ALLOC(protodata)
56
57static void
58protoent_data_clear(struct protoent_data *ped)
59{
60	if (ped->fp) {
61		fclose(ped->fp);
62		ped->fp = NULL;
63	}
64}
65
66static void
67protoent_data_free(void *ptr)
68{
69	struct protoent_data *ped = ptr;
70
71	protoent_data_clear(ped);
72	free(ped);
73}
74
75static void
76protodata_free(void *ptr)
77{
78	free(ptr);
79}
80
81int
82__copy_protoent(struct protoent *pe, struct protoent *pptr, char *buf,
83    size_t buflen)
84{
85	char *cp;
86	int i, n;
87	int numptr, len;
88
89	/* Find out the amount of space required to store the answer. */
90	numptr = 1; /* NULL ptr */
91	len = (char *)ALIGN(buf) - buf;
92	for (i = 0; pe->p_aliases[i]; i++, numptr++) {
93		len += strlen(pe->p_aliases[i]) + 1;
94	}
95	len += strlen(pe->p_name) + 1;
96	len += numptr * sizeof(char*);
97
98	if (len > (int)buflen) {
99		errno = ERANGE;
100		return (-1);
101	}
102
103	/* copy protocol value*/
104	pptr->p_proto = pe->p_proto;
105
106	cp = (char *)ALIGN(buf) + numptr * sizeof(char *);
107
108	/* copy official name */
109	n = strlen(pe->p_name) + 1;
110	strcpy(cp, pe->p_name);
111	pptr->p_name = cp;
112	cp += n;
113
114	/* copy aliases */
115	pptr->p_aliases = (char **)ALIGN(buf);
116	for (i = 0 ; pe->p_aliases[i]; i++) {
117		n = strlen(pe->p_aliases[i]) + 1;
118		strcpy(cp, pe->p_aliases[i]);
119		pptr->p_aliases[i] = cp;
120		cp += n;
121	}
122	pptr->p_aliases[i] = NULL;
123
124	return (0);
125}
126
127void
128__setprotoent_p(int f, struct protoent_data *ped)
129{
130	if (ped->fp == NULL)
131		ped->fp = fopen(_PATH_PROTOCOLS, "r");
132	else
133		rewind(ped->fp);
134	ped->stayopen |= f;
135}
136
137void
138__endprotoent_p(struct protoent_data *ped)
139{
140	if (ped->fp) {
141		fclose(ped->fp);
142		ped->fp = NULL;
143	}
144	ped->stayopen = 0;
145}
146
147int
148__getprotoent_p(struct protoent *pe, struct protoent_data *ped)
149{
150	char *p;
151	char *cp, **q, *endp;
152	long l;
153
154	if (ped->fp == NULL && (ped->fp = fopen(_PATH_PROTOCOLS, "r")) == NULL)
155		return (-1);
156again:
157	if ((p = fgets(ped->line, sizeof ped->line, ped->fp)) == NULL)
158		return (-1);
159	if (*p == '#')
160		goto again;
161	cp = strpbrk(p, "#\n");
162	if (cp != NULL)
163		*cp = '\0';
164	pe->p_name = p;
165	cp = strpbrk(p, " \t");
166	if (cp == NULL)
167		goto again;
168	*cp++ = '\0';
169	while (*cp == ' ' || *cp == '\t')
170		cp++;
171	p = strpbrk(cp, " \t");
172	if (p != NULL)
173		*p++ = '\0';
174	l = strtol(cp, &endp, 10);
175	if (endp == cp || *endp != '\0' || l < 0 || l > USHRT_MAX)
176		goto again;
177	pe->p_proto = l;
178	q = pe->p_aliases = ped->aliases;
179	if (p != NULL) {
180		cp = p;
181		while (cp && *cp) {
182			if (*cp == ' ' || *cp == '\t') {
183				cp++;
184				continue;
185			}
186			if (q < &ped->aliases[_MAXALIASES - 1])
187				*q++ = cp;
188			cp = strpbrk(cp, " \t");
189			if (cp != NULL)
190				*cp++ = '\0';
191		}
192	}
193	*q = NULL;
194	return (0);
195}
196
197int
198getprotoent_r(struct protoent *pptr, char *buffer, size_t buflen,
199    struct protoent **result)
200{
201	struct protoent pe;
202	struct protoent_data *ped;
203
204	if ((ped = __protoent_data_init()) == NULL)
205		return (-1);
206
207	if (__getprotoent_p(&pe, ped) != 0)
208		return (-1);
209	if (__copy_protoent(&pe, pptr, buffer, buflen) != 0)
210		return (-1);
211	*result = pptr;
212	return (0);
213}
214
215void
216setprotoent(int f)
217{
218	struct protoent_data *ped;
219
220	if ((ped = __protoent_data_init()) == NULL)
221		return;
222	__setprotoent_p(f, ped);
223}
224
225void
226endprotoent(void)
227{
228	struct protoent_data *ped;
229
230	if ((ped = __protoent_data_init()) == NULL)
231		return;
232	__endprotoent_p(ped);
233}
234
235struct protoent *
236getprotoent(void)
237{
238	struct protodata *pd;
239	struct protoent *rval;
240
241	if ((pd = __protodata_init()) == NULL)
242		return (NULL);
243	if (getprotoent_r(&pd->proto, pd->data, sizeof(pd->data), &rval) != 0)
244		return (NULL);
245	return (rval);
246}
247