1168404Spjd/*	$NetBSD: getservbyname_r.c,v 1.7 2010/04/25 00:54:46 joerg Exp $	*/
2168404Spjd
3168404Spjd/*
4168404Spjd * Copyright (c) 1983, 1993
5168404Spjd *	The Regents of the University of California.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 3. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY 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 <sys/cdefs.h>
33#if defined(LIBC_SCCS) && !defined(lint)
34#if 0
35static char sccsid[] = "@(#)getservbyname.c	8.1 (Berkeley) 6/4/93";
36#else
37__RCSID("$NetBSD: getservbyname_r.c,v 1.7 2010/04/25 00:54:46 joerg Exp $");
38#endif
39#endif /* LIBC_SCCS and not lint */
40
41#include "namespace.h"
42#include <assert.h>
43#include <cdbr.h>
44#include <netdb.h>
45#include <stdlib.h>
46#include <string.h>
47
48#include "servent.h"
49
50#ifdef __weak_alias
51__weak_alias(getservbyname_r,_getservbyname_r)
52#endif
53
54static struct servent *
55_servent_getbyname(struct servent_data *sd, struct servent *sp,
56    const char *name, const char *proto)
57{
58
59	if ((sd->flags & (_SV_CDB | _SV_PLAINFILE)) == 0)
60		return NULL;
61
62	if (sd->flags & _SV_CDB) {
63		uint8_t buf[255 * 2 + 2];
64		size_t namelen, protolen;
65		const uint8_t *data, *data_end;
66		const void *data_ptr;
67		size_t datalen;
68
69		namelen = strlen(name);
70		if (namelen == 0 || namelen > 255)
71			return NULL;
72		if (proto != NULL) {
73			protolen = strlen(proto);
74			if (protolen == 0 || protolen > 255)
75				return NULL;
76		} else
77			protolen = 0;
78		if (namelen + protolen > 255)
79			return NULL;
80
81		buf[0] = namelen;
82		buf[1] = protolen;
83		memcpy(buf + 2, name, namelen);
84		memcpy(buf + 2 + namelen, proto, protolen);
85
86		if (cdbr_find(sd->cdb, buf, 2 + namelen + protolen,
87		    &data_ptr, &datalen))
88			return NULL;
89
90		if (datalen < namelen + protolen + 6)
91			return NULL;
92
93		data = data_ptr;
94		data_end = data + datalen;
95		if (protolen) {
96			if (data[2] != protolen)
97				return NULL;
98			if (memcmp(data + 3, proto, protolen + 1))
99				return NULL;
100		}
101		data += 3 + data[2] + 1;
102		if (data > data_end)
103			return NULL;
104		while (data != data_end) {
105			if (*data == '\0')
106				return NULL;
107			if (data + data[0] + 2 > data_end)
108				return NULL;
109			if (data[0] == namelen &&
110			    memcmp(data + 1, name, namelen + 1) == 0)
111				return _servent_parsedb(sd, sp, data_ptr,
112				    datalen);
113			data += data[0] + 2;
114		}
115		return NULL;
116	} else {
117		while (_servent_getline(sd) != -1) {
118			char **cp;
119			if (_servent_parseline(sd, sp) == NULL)
120				continue;
121
122			if (strcmp(name, sp->s_name) == 0)
123				goto gotname;
124
125			for (cp = sp->s_aliases; *cp; cp++)
126				if (strcmp(name, *cp) == 0)
127					goto gotname;
128			continue;
129gotname:
130			if (proto == NULL || strcmp(sp->s_proto, proto) == 0)
131				return sp;
132		}
133		return NULL;
134	}
135}
136
137struct servent *
138getservbyname_r(const char *name, const char *proto, struct servent *sp,
139    struct servent_data *sd)
140{
141	_DIAGASSERT(name != NULL);
142	/* proto may be NULL */
143
144	setservent_r(sd->flags & _SV_STAYOPEN, sd);
145	sp = _servent_getbyname(sd, sp, name, proto);
146	if (!(sd->flags & _SV_STAYOPEN))
147		_servent_close(sd);
148	return sp;
149}
150