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 * 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