1 /* $NetBSD: nis_sv.c,v 1.1.1.2 2012/09/09 16:07:51 christos Exp $ */
2
3 /*
4 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
5 * Copyright (c) 1996,1999 by Internet Software Consortium.
6 *
7 * Permission to use, copy, modify, and distribute this software for any
8 * purpose with or without fee is hereby granted, provided that the above
9 * copyright notice and this permission notice appear in all copies.
10 *
11 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
12 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
13 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
14 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
15 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
16 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
17 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
18 */
19
20 #if defined(LIBC_SCCS) && !defined(lint)
21 static const char rcsid[] = "Id: nis_sv.c,v 1.4 2005/04/27 04:56:34 sra Exp ";
22 #endif /* LIBC_SCCS and not lint */
23
24 /* Imports */
25
26 #include "port_before.h"
27
28 #ifndef WANT_IRS_NIS
29 static int __bind_irs_nis_unneeded;
30 #else
31
32 #include <sys/types.h>
33 #include <netinet/in.h>
34 #include <arpa/nameser.h>
35 #include <resolv.h>
36 #include <sys/socket.h>
37 #ifdef T_NULL
38 #undef T_NULL /* Silence re-definition warning of T_NULL. */
39 #endif
40 #include <rpc/rpc.h>
41 #include <rpc/xdr.h>
42 #include <rpcsvc/yp_prot.h>
43 #include <rpcsvc/ypclnt.h>
44
45 #include <ctype.h>
46 #include <errno.h>
47 #include <stdio.h>
48 #include <stdlib.h>
49 #include <string.h>
50
51 #include <isc/memcluster.h>
52 #include <irs.h>
53
54 #include "port_after.h"
55
56 #include "irs_p.h"
57 #include "nis_p.h"
58
59 /* Definitions */
60
61 struct pvt {
62 int needrewind;
63 char * nis_domain;
64 char * curkey_data;
65 int curkey_len;
66 char * curval_data;
67 int curval_len;
68 char line[BUFSIZ+1];
69 struct servent serv;
70 char * svbuf;
71 };
72
73 enum do_what { do_none = 0x0, do_key = 0x1, do_val = 0x2, do_all = 0x3 };
74
75 static /*const*/ char services_byname[] = "services.byname";
76
77 /* Forward */
78
79 static void sv_close(struct irs_sv*);
80 static struct servent * sv_next(struct irs_sv *);
81 static struct servent * sv_byname(struct irs_sv *, const char *,
82 const char *);
83 static struct servent * sv_byport(struct irs_sv *, int, const char *);
84 static void sv_rewind(struct irs_sv *);
85 static void sv_minimize(struct irs_sv *);
86
87 static struct servent * makeservent(struct irs_sv *this);
88 static void nisfree(struct pvt *, enum do_what);
89
90 /* Public */
91
92 struct irs_sv *
irs_nis_sv(struct irs_acc * this)93 irs_nis_sv(struct irs_acc *this) {
94 struct irs_sv *sv;
95 struct pvt *pvt;
96
97 if (!(sv = memget(sizeof *sv))) {
98 errno = ENOMEM;
99 return (NULL);
100 }
101 memset(sv, 0x5e, sizeof *sv);
102 if (!(pvt = memget(sizeof *pvt))) {
103 memput(sv, sizeof *sv);
104 errno = ENOMEM;
105 return (NULL);
106 }
107 memset(pvt, 0, sizeof *pvt);
108 pvt->needrewind = 1;
109 pvt->nis_domain = ((struct nis_p *)this->private)->domain;
110 sv->private = pvt;
111 sv->close = sv_close;
112 sv->next = sv_next;
113 sv->byname = sv_byname;
114 sv->byport = sv_byport;
115 sv->rewind = sv_rewind;
116 sv->minimize = sv_minimize;
117 sv->res_get = NULL;
118 sv->res_set = NULL;
119 return (sv);
120 }
121
122 /* Methods */
123
124 static void
sv_close(struct irs_sv * this)125 sv_close(struct irs_sv *this) {
126 struct pvt *pvt = (struct pvt *)this->private;
127
128 nisfree(pvt, do_all);
129 if (pvt->serv.s_aliases)
130 free(pvt->serv.s_aliases);
131 if (pvt->svbuf)
132 free(pvt->svbuf);
133 memput(pvt, sizeof *pvt);
134 memput(this, sizeof *this);
135 }
136
137 static struct servent *
sv_byname(struct irs_sv * this,const char * name,const char * proto)138 sv_byname(struct irs_sv *this, const char *name, const char *proto) {
139 struct servent *serv;
140 char **sap;
141
142 sv_rewind(this);
143 while ((serv = sv_next(this)) != NULL) {
144 if (proto != NULL && strcmp(proto, serv->s_proto))
145 continue;
146 if (!strcmp(name, serv->s_name))
147 break;
148 for (sap = serv->s_aliases; sap && *sap; sap++)
149 if (!strcmp(name, *sap))
150 break;
151 }
152 return (serv);
153 }
154
155 static struct servent *
sv_byport(struct irs_sv * this,int port,const char * proto)156 sv_byport(struct irs_sv *this, int port, const char *proto) {
157 struct servent *serv;
158
159 sv_rewind(this);
160 while ((serv = sv_next(this)) != NULL) {
161 if (proto != NULL && strcmp(proto, serv->s_proto))
162 continue;
163 if (serv->s_port == port)
164 break;
165 }
166 return (serv);
167 }
168
169 static void
sv_rewind(struct irs_sv * this)170 sv_rewind(struct irs_sv *this) {
171 struct pvt *pvt = (struct pvt *)this->private;
172
173 pvt->needrewind = 1;
174 }
175
176 static struct servent *
sv_next(struct irs_sv * this)177 sv_next(struct irs_sv *this) {
178 struct pvt *pvt = (struct pvt *)this->private;
179 struct servent *rval;
180 int r;
181
182 do {
183 if (pvt->needrewind) {
184 nisfree(pvt, do_all);
185 r = yp_first(pvt->nis_domain, services_byname,
186 &pvt->curkey_data, &pvt->curkey_len,
187 &pvt->curval_data, &pvt->curval_len);
188 pvt->needrewind = 0;
189 } else {
190 char *newkey_data;
191 int newkey_len;
192
193 nisfree(pvt, do_val);
194 r = yp_next(pvt->nis_domain, services_byname,
195 pvt->curkey_data, pvt->curkey_len,
196 &newkey_data, &newkey_len,
197 &pvt->curval_data, &pvt->curval_len);
198 nisfree(pvt, do_key);
199 pvt->curkey_data = newkey_data;
200 pvt->curkey_len = newkey_len;
201 }
202 if (r != 0) {
203 errno = ENOENT;
204 return (NULL);
205 }
206 rval = makeservent(this);
207 } while (rval == NULL);
208 return (rval);
209 }
210
211 static void
sv_minimize(struct irs_sv * this)212 sv_minimize(struct irs_sv *this) {
213 UNUSED(this);
214 /* NOOP */
215 }
216
217 /* Private */
218
219 static struct servent *
makeservent(struct irs_sv * this)220 makeservent(struct irs_sv *this) {
221 struct pvt *pvt = (struct pvt *)this->private;
222 static const char spaces[] = " \t";
223 char *p, **t;
224 int n, m;
225
226 if (pvt->svbuf)
227 free(pvt->svbuf);
228 pvt->svbuf = pvt->curval_data;
229 pvt->curval_data = NULL;
230
231 if (pvt->serv.s_aliases) {
232 free(pvt->serv.s_aliases);
233 pvt->serv.s_aliases = NULL;
234 }
235
236 if ((p = strpbrk(pvt->svbuf, "#\n")))
237 *p = '\0';
238
239 p = pvt->svbuf;
240
241 pvt->serv.s_name = p;
242 p += strcspn(p, spaces);
243 if (!*p)
244 goto cleanup;
245 *p++ = '\0';
246 p += strspn(p, spaces);
247
248 pvt->serv.s_port = htons((u_short) atoi(p));
249 pvt->serv.s_proto = NULL;
250
251 while (*p && !isspace((unsigned char)*p))
252 if (*p++ == '/')
253 pvt->serv.s_proto = p;
254 if (!pvt->serv.s_proto)
255 goto cleanup;
256 if (*p) {
257 *p++ = '\0';
258 p += strspn(p, spaces);
259 }
260
261 n = m = 0;
262 while (*p) {
263 if ((n + 1) >= m || !pvt->serv.s_aliases) {
264 m += 10;
265 t = realloc(pvt->serv.s_aliases, m * sizeof(char *));
266 if (!t) {
267 errno = ENOMEM;
268 goto cleanup;
269 }
270 pvt->serv.s_aliases = t;
271 }
272 pvt->serv.s_aliases[n++] = p;
273 p += strcspn(p, spaces);
274 if (!*p)
275 break;
276 *p++ = '\0';
277 p += strspn(p, spaces);
278 }
279 if (!pvt->serv.s_aliases)
280 pvt->serv.s_aliases = malloc(sizeof(char *));
281 if (!pvt->serv.s_aliases)
282 goto cleanup;
283 pvt->serv.s_aliases[n] = NULL;
284 return (&pvt->serv);
285
286 cleanup:
287 if (pvt->serv.s_aliases) {
288 free(pvt->serv.s_aliases);
289 pvt->serv.s_aliases = NULL;
290 }
291 if (pvt->svbuf) {
292 free(pvt->svbuf);
293 pvt->svbuf = NULL;
294 }
295 return (NULL);
296 }
297
298 static void
nisfree(struct pvt * pvt,enum do_what do_what)299 nisfree(struct pvt *pvt, enum do_what do_what) {
300 if ((do_what & do_key) && pvt->curkey_data) {
301 free(pvt->curkey_data);
302 pvt->curkey_data = NULL;
303 }
304 if ((do_what & do_val) && pvt->curval_data) {
305 free(pvt->curval_data);
306 pvt->curval_data = NULL;
307 }
308 }
309
310 #endif /*WANT_IRS_NIS*/
311
312 /*! \file */
313