Home | History | Annotate | Line # | Download | only in net
getservbyname_r.c revision 1.9
      1 /*	$NetBSD: getservbyname_r.c,v 1.9 2012/03/13 21:13:41 christos Exp $	*/
      2 
      3 /*
      4  * Copyright (c) 1983, 1993
      5  *	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
     35 static char sccsid[] = "@(#)getservbyname.c	8.1 (Berkeley) 6/4/93";
     36 #else
     37 __RCSID("$NetBSD: getservbyname_r.c,v 1.9 2012/03/13 21:13:41 christos 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 
     54 static 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] = (uint8_t)namelen;
     82 		buf[1] = (uint8_t)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;
    129 gotname:
    130 			if (proto == NULL || strcmp(sp->s_proto, proto) == 0)
    131 				return sp;
    132 		}
    133 		return NULL;
    134 	}
    135 }
    136 
    137 struct servent *
    138 getservbyname_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