Home | History | Annotate | Line # | Download | only in irs
nis_sv.c revision 1.1.1.1.6.2
      1 /*	$NetBSD: nis_sv.c,v 1.1.1.1.6.2 2011/01/09 20:42:56 riz 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 *
     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
    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 *
    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 *
    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
    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 *
    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
    212 sv_minimize(struct irs_sv *this) {
    213 	UNUSED(this);
    214 	/* NOOP */
    215 }
    216 
    217 /* Private */
    218 
    219 static struct servent *
    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
    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