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
__weak_alias(getservbyname_r,_getservbyname_r)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 *
getservbyname_r(const char * name,const char * proto,struct servent * sp,struct servent_data * sd)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