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)
35 static char sccsid[] = "@(#)getservent.c 8.1 (Berkeley) 6/4/93";
36 #endif /* LIBC_SCCS and not lint */
37
38 #include <sys/types.h>
39 #include <sys/socket.h>
40 #include <netdb.h>
41 #include <stdio.h>
42 #include <string.h>
43 #include <stdlib.h>
44 #ifdef YP
45 #include <rpc/rpc.h>
46 #include <rpcsvc/yp_prot.h>
47 #include <rpcsvc/ypclnt.h>
48 static int serv_stepping_yp = 0;
49 extern int _yp_check __P(( char ** ));
50 #endif
51
52
53 #define MAXALIASES 35
54
55 static FILE *servf = NULL;
56 static char line[BUFSIZ+1];
57 static struct servent serv;
58 static char *serv_aliases[MAXALIASES];
59 int _serv_stayopen;
60
61 #ifdef YP
62 char *___getservbyname_yp = NULL;
63 char *___getservbyproto_yp = NULL;
64 int ___getservbyport_yp = 0;
65 static char *yp_domain = NULL;
66
67 static int
_getservbyport_yp(line)68 _getservbyport_yp(line)
69 char *line;
70 {
71 char *result;
72 int resultlen;
73 char buf[YPMAXRECORD + 2];
74 int rv;
75
76 snprintf(buf, sizeof(buf), "%d/%s", ntohs(___getservbyport_yp),
77 ___getservbyproto_yp);
78
79 ___getservbyport_yp = 0;
80 ___getservbyproto_yp = NULL;
81
82 if(!yp_domain) {
83 if(yp_get_default_domain(&yp_domain))
84 return (0);
85 }
86
87 /*
88 * We have to be a little flexible here. Ideally you're supposed
89 * to have both a services.byname and a services.byport map, but
90 * some systems have only services.byname. FreeBSD cheats a little
91 * by putting the services.byport information in the same map as
92 * services.byname so that either case will work. We allow for both
93 * possibilities here: if there is no services.byport map, we try
94 * services.byname instead.
95 */
96 if ((rv = yp_match(yp_domain, "services.byport", buf, strlen(buf),
97 &result, &resultlen))) {
98 if (rv == YPERR_MAP) {
99 if (yp_match(yp_domain, "services.byname", buf,
100 strlen(buf), &result, &resultlen))
101 return(0);
102 } else
103 return(0);
104 }
105
106 /* getservent() expects lines terminated with \n -- make it happy */
107 snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
108
109 free(result);
110 return(1);
111 }
112
113 static int
_getservbyname_yp(line)114 _getservbyname_yp(line)
115 char *line;
116 {
117 char *result;
118 int resultlen;
119 char buf[YPMAXRECORD + 2];
120
121 if(!yp_domain) {
122 if(yp_get_default_domain(&yp_domain))
123 return (0);
124 }
125
126 snprintf(buf, sizeof(buf), "%s/%s", ___getservbyname_yp,
127 ___getservbyproto_yp);
128
129 ___getservbyname_yp = 0;
130 ___getservbyproto_yp = NULL;
131
132 if (yp_match(yp_domain, "services.byname", buf, strlen(buf),
133 &result, &resultlen)) {
134 return(0);
135 }
136
137 /* getservent() expects lines terminated with \n -- make it happy */
138 snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
139
140 free(result);
141 return(1);
142 }
143
144 static int
_getservent_yp(line)145 _getservent_yp(line)
146 char *line;
147 {
148 static char *key = NULL;
149 static int keylen;
150 char *lastkey, *result;
151 int resultlen;
152 int rv;
153
154 if(!yp_domain) {
155 if(yp_get_default_domain(&yp_domain))
156 return (0);
157 }
158
159 if (!serv_stepping_yp) {
160 if (key)
161 free(key);
162 if ((rv = yp_first(yp_domain, "services.byname", &key, &keylen,
163 &result, &resultlen))) {
164 serv_stepping_yp = 0;
165 return(0);
166 }
167 serv_stepping_yp = 1;
168 } else {
169 lastkey = key;
170 rv = yp_next(yp_domain, "services.byname", key, keylen, &key,
171 &keylen, &result, &resultlen);
172 free(lastkey);
173 if (rv) {
174 serv_stepping_yp = 0;
175 return (0);
176 }
177 }
178
179 /* getservent() expects lines terminated with \n -- make it happy */
180 snprintf(line, BUFSIZ, "%.*s\n", resultlen, result);
181
182 free(result);
183
184 return(1);
185 }
186 #endif
187
188 void
setservent(int f)189 setservent(int f)
190 {
191 if (servf == NULL)
192 servf = fopen(_PATH_SERVICES, "r" );
193 else
194 rewind(servf);
195 _serv_stayopen |= f;
196 }
197
198 void
endservent()199 endservent()
200 {
201 if (servf) {
202 fclose(servf);
203 servf = NULL;
204 }
205 _serv_stayopen = 0;
206 }
207
208 struct servent *
getservent()209 getservent()
210 {
211 char *p;
212 register char *cp, **q;
213
214 #ifdef YP
215 if (serv_stepping_yp && _getservent_yp(line)) {
216 p = (char *)&line;
217 goto unpack;
218 }
219 tryagain:
220 #endif
221 if (servf == NULL && (servf = fopen(_PATH_SERVICES, "r" )) == NULL)
222 return (NULL);
223 again:
224 if ((p = fgets(line, BUFSIZ, servf)) == NULL)
225 return (NULL);
226 #ifdef YP
227 if (*p == '+' && _yp_check(NULL)) {
228 if (___getservbyname_yp != NULL) {
229 if (!_getservbyname_yp(line))
230 goto tryagain;
231 }
232 else if (___getservbyport_yp != 0) {
233 if (!_getservbyport_yp(line))
234 goto tryagain;
235 }
236 else if (!_getservent_yp(line))
237 goto tryagain;
238 }
239 unpack:
240 #endif
241 if (*p == '#')
242 goto again;
243 cp = strpbrk(p, "#\n");
244 if (cp == NULL)
245 goto again;
246 *cp = '\0';
247 serv.s_name = p;
248 p = strpbrk(p, " \t");
249 if (p == NULL)
250 goto again;
251 *p++ = '\0';
252 while (*p == ' ' || *p == '\t')
253 p++;
254 cp = strpbrk(p, ",/");
255 if (cp == NULL)
256 goto again;
257 *cp++ = '\0';
258 serv.s_port = htons((u_short)atoi(p));
259 serv.s_proto = cp;
260 q = serv.s_aliases = serv_aliases;
261 cp = strpbrk(cp, " \t");
262 if (cp != NULL)
263 *cp++ = '\0';
264 while (cp && *cp) {
265 if (*cp == ' ' || *cp == '\t') {
266 cp++;
267 continue;
268 }
269 if (q < &serv_aliases[MAXALIASES - 1])
270 *q++ = cp;
271 cp = strpbrk(cp, " \t");
272 if (cp != NULL)
273 *cp++ = '\0';
274 }
275 *q = NULL;
276 return (&serv);
277 }
278