/* * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC") * Copyright (c) 1996,1999 by Internet Software Consortium. * * Permission to use, copy, modify, and distribute this software for any * purpose with or without fee is hereby granted, provided that the above * copyright notice and this permission notice appear in all copies. * * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. */#if defined(LIBC_SCCS) && !defined(lint)static const char rcsid[] = "$Id: nis_ho.c,v 1.2.2.1.4.1 2004/03/09 08:33:38 marka Exp $";#endif /* LIBC_SCCS and not lint *//* Imports */#include "port_before.h"#ifndef WANT_IRS_NISstatic int __bind_irs_nis_unneeded;#else#include <sys/types.h>#include <sys/param.h>#include <sys/socket.h>#include <netinet/in.h>#include <arpa/inet.h>#include <arpa/nameser.h>#ifdef T_NULL#undef T_NULL			/* Silence re-definition warning of T_NULL. */#endif#include <rpc/rpc.h>#include <rpc/xdr.h>#include <rpcsvc/yp_prot.h>#include <rpcsvc/ypclnt.h>#include <ctype.h>#include <errno.h>#include <stdlib.h>#include <netdb.h>#include <resolv.h>#include <stdio.h>#include <string.h>#include <isc/memcluster.h>#include <irs.h>#include "port_after.h"#include "irs_p.h"#include "nis_p.h"/* Definitions */#define	MAXALIASES	35#define	MAXADDRS	35#if PACKETSZ > 1024#define	MAXPACKET	PACKETSZ#else#define	MAXPACKET	1024#endifstruct pvt {	int		needrewind;	char *		nis_domain;	char *		curkey_data;	int		curkey_len;	char *		curval_data;	int		curval_len;	struct hostent	host;	char *		h_addr_ptrs[MAXADDRS + 1];	char *		host_aliases[MAXALIASES + 1];	char		hostbuf[8*1024];	u_char		host_addr[16];	/* IPv4 or IPv6 */	struct __res_state  *res;	void		(*free_res)(void *);};enum do_what { do_none = 0x0, do_key = 0x1, do_val = 0x2, do_all = 0x3 };static const u_char mapped[] = { 0,0, 0,0, 0,0, 0,0, 0,0, 0xff,0xff };static const u_char tunnelled[] = { 0,0, 0,0, 0,0, 0,0, 0,0, 0,0 };static /*const*/ char hosts_byname[] = "hosts.byname";static /*const*/ char hosts_byaddr[] = "hosts.byaddr";static /*const*/ char ipnode_byname[] = "ipnode.byname";static /*const*/ char ipnode_byaddr[] = "ipnode.byaddr";static /*const*/ char yp_multi[] = "YP_MULTI_";/* Forwards */static void		ho_close(struct irs_ho *this);static struct hostent *	ho_byname(struct irs_ho *this, const char *name);static struct hostent *	ho_byname2(struct irs_ho *this, const char *name,					int af);static struct hostent *	ho_byaddr(struct irs_ho *this, const void *addr,				       int len, int af);static struct hostent *	ho_next(struct irs_ho *this);static void		ho_rewind(struct irs_ho *this);static void		ho_minimize(struct irs_ho *this);static struct __res_state * ho_res_get(struct irs_ho *this);static void		ho_res_set(struct irs_ho *this,				   struct __res_state *res,				   void (*free_res)(void *));static struct addrinfo * ho_addrinfo(struct irs_ho *this, const char *name,				     const struct addrinfo *pai);static struct hostent *	makehostent(struct irs_ho *this);static void		nisfree(struct pvt *, enum do_what);static int		init(struct irs_ho *this);/* Public */struct irs_ho *irs_nis_ho(struct irs_acc *this) {	struct irs_ho *ho;	struct pvt *pvt;	if (!(pvt = memget(sizeof *pvt))) {		errno = ENOMEM;		return (NULL);	}	memset(pvt, 0, sizeof *pvt);	if (!(ho = memget(sizeof *ho))) {		memput(pvt, sizeof *pvt);		errno = ENOMEM;		return (NULL);	}	memset(ho, 0x5e, sizeof *ho);	pvt->needrewind = 1;	pvt->nis_domain = ((struct nis_p *)this->private)->domain;	ho->private = pvt;	ho->close = ho_close;	ho->byname = ho_byname;	ho->byname2 = ho_byname2;	ho->byaddr = ho_byaddr;	ho->next = ho_next;	ho->rewind = ho_rewind;	ho->minimize = ho_minimize;	ho->res_set = ho_res_set;	ho->res_get = ho_res_get;	ho->addrinfo = ho_addrinfo;	return (ho);}/* Methods */static voidho_close(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	ho_minimize(this);	nisfree(pvt, do_all);	if (pvt->res && pvt->free_res)		(*pvt->free_res)(pvt->res);	memput(pvt, sizeof *pvt);	memput(this, sizeof *this);}static struct hostent *ho_byname(struct irs_ho *this, const char *name) {	struct pvt *pvt = (struct pvt *)this->private;	struct hostent *hp;	if (init(this) == -1)		return (NULL);	if (pvt->res->options & RES_USE_INET6) {		hp = ho_byname2(this, name, AF_INET6);		if (hp)			return (hp);	}	return (ho_byname2(this, name, AF_INET));}static struct hostent *ho_byname2(struct irs_ho *this, const char *name, int af) {	struct pvt *pvt = (struct pvt *)this->private;	int r;	char *tmp;	UNUSED(af);		if (init(this) == -1)		return (NULL);	nisfree(pvt, do_val);	strcpy(pvt->hostbuf, yp_multi);	strncat(pvt->hostbuf, name, sizeof(pvt->hostbuf) - sizeof(yp_multi));	pvt->hostbuf[sizeof(pvt->hostbuf) - 1] = '\0';	for (r = sizeof(yp_multi) - 1; pvt->hostbuf[r] != '\0'; r++)		if (isupper((unsigned char)pvt->hostbuf[r]))			tolower(pvt->hostbuf[r]);	tmp = pvt->hostbuf;	r = yp_match(pvt->nis_domain, ipnode_byname, tmp,		     strlen(tmp), &pvt->curval_data, &pvt->curval_len);	if (r != 0) {		tmp = pvt->hostbuf + sizeof(yp_multi) - 1;		r = yp_match(pvt->nis_domain, ipnode_byname, tmp,			     strlen(tmp), &pvt->curval_data, &pvt->curval_len);	}	if (r != 0) {		tmp = pvt->hostbuf;		r = yp_match(pvt->nis_domain, hosts_byname, tmp,			     strlen(tmp), &pvt->curval_data, &pvt->curval_len);	}	if (r != 0) {		tmp = pvt->hostbuf + sizeof(yp_multi) - 1;		r = yp_match(pvt->nis_domain, hosts_byname, tmp,			     strlen(tmp), &pvt->curval_data, &pvt->curval_len);	}	if (r != 0) {		RES_SET_H_ERRNO(pvt->res, HOST_NOT_FOUND);		return (NULL);	}	return (makehostent(this));}static struct hostent *ho_byaddr(struct irs_ho *this, const void *addr, int len, int af) {	struct pvt *pvt = (struct pvt *)this->private;	char tmp[sizeof "ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255"];	const u_char *uaddr = addr;	int r;		if (init(this) == -1)		return (NULL);	if (af == AF_INET6 && len == IN6ADDRSZ &&	    (!memcmp(uaddr, mapped, sizeof mapped) ||	     !memcmp(uaddr, tunnelled, sizeof tunnelled))) {		/* Unmap. */		addr = (const u_char *)addr + sizeof mapped;		uaddr += sizeof mapped;		af = AF_INET;		len = INADDRSZ;	}	if (inet_ntop(af, uaddr, tmp, sizeof tmp) == NULL) {		RES_SET_H_ERRNO(pvt->res, NETDB_INTERNAL);		return (NULL);	}	nisfree(pvt, do_val);	r = yp_match(pvt->nis_domain, ipnode_byaddr, tmp, strlen(tmp),		     &pvt->curval_data, &pvt->curval_len);	if (r != 0)		r = yp_match(pvt->nis_domain, hosts_byaddr, tmp, strlen(tmp),			     &pvt->curval_data, &pvt->curval_len);	if (r != 0) {		RES_SET_H_ERRNO(pvt->res, HOST_NOT_FOUND);		return (NULL);	}	return (makehostent(this));}static struct hostent *ho_next(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	struct hostent *rval;	int r;	if (init(this) == -1)		return (NULL);	do {		if (pvt->needrewind) {			nisfree(pvt, do_all);			r = yp_first(pvt->nis_domain, hosts_byaddr,				     &pvt->curkey_data, &pvt->curkey_len,				     &pvt->curval_data, &pvt->curval_len);			pvt->needrewind = 0;		} else {			char *newkey_data;			int newkey_len;			nisfree(pvt, do_val);			r = yp_next(pvt->nis_domain, hosts_byaddr,				    pvt->curkey_data, pvt->curkey_len,				    &newkey_data, &newkey_len,				    &pvt->curval_data, &pvt->curval_len);			nisfree(pvt, do_key);			pvt->curkey_data = newkey_data;			pvt->curkey_len = newkey_len;		}		if (r != 0) {			RES_SET_H_ERRNO(pvt->res, HOST_NOT_FOUND);			return (NULL);		}		rval = makehostent(this);	} while (rval == NULL);	return (rval);}static voidho_rewind(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	pvt->needrewind = 1;}static voidho_minimize(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	if (pvt->res)		res_nclose(pvt->res);}static struct __res_state *ho_res_get(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	if (!pvt->res) {		struct __res_state *res;		res = (struct __res_state *)malloc(sizeof *res);		if (!res) {			errno = ENOMEM;			return (NULL);		}		memset(res, 0, sizeof *res);		ho_res_set(this, res, free);	}	return (pvt->res);}static voidho_res_set(struct irs_ho *this, struct __res_state *res,		void (*free_res)(void *)) {	struct pvt *pvt = (struct pvt *)this->private;	if (pvt->res && pvt->free_res) {		res_nclose(pvt->res);		(*pvt->free_res)(pvt->res);	}	pvt->res = res;	pvt->free_res = free_res;}struct nis_res_target {	struct nis_res_target *next;	int family;};/* XXX */extern struct addrinfo *hostent2addrinfo __P((struct hostent *,					      const struct addrinfo *pai));static struct addrinfo *ho_addrinfo(struct irs_ho *this, const char *name, const struct addrinfo *pai){	struct pvt *pvt = (struct pvt *)this->private;	struct hostent *hp;	struct nis_res_target q, q2, *p;	struct addrinfo sentinel, *cur;	memset(&q, 0, sizeof(q2));	memset(&q2, 0, sizeof(q2));	memset(&sentinel, 0, sizeof(sentinel));	cur = &sentinel;	switch(pai->ai_family) {	case AF_UNSPEC:		/* INET6 then INET4 */		q.family = AF_INET6;		q.next = &q2;		q2.family = AF_INET;		break;	case AF_INET6:		q.family = AF_INET6;		break;	case AF_INET:		q.family = AF_INET;		break;	default:		RES_SET_H_ERRNO(pvt->res, NO_RECOVERY); /* ??? */		return(NULL);	}	for (p = &q; p; p = p->next) {		struct addrinfo *ai;		hp = (*this->byname2)(this, name, p->family);		if (hp == NULL) {			/* byname2 should've set an appropriate error */			continue;		}		if ((hp->h_name == NULL) || (hp->h_name[0] == 0) ||		    (hp->h_addr_list[0] == NULL)) {			RES_SET_H_ERRNO(pvt->res, NO_RECOVERY);			continue;		}		ai = hostent2addrinfo(hp, pai);		if (ai) {			cur->ai_next = ai;			while (cur && cur->ai_next)				cur = cur->ai_next;		}	}	if (sentinel.ai_next == NULL)		RES_SET_H_ERRNO(pvt->res, HOST_NOT_FOUND);	return(sentinel.ai_next);}/* Private *//*ipnodes:::1             localhost127.0.0.1       localhost1.2.3.4         FOO bar1.2.6.4         FOO bar1.2.6.5         hostipnodes.byname:YP_MULTI_localhost ::1,127.0.0.1        localhostYP_MULTI_foo 1.2.3.4,1.2.6.4    FOO barYP_MULTI_bar 1.2.3.4,1.2.6.4    FOO barhost 1.2.6.5    hosthosts.byname:localhost 127.0.0.1     localhosthost 1.2.6.5    hostYP_MULTI_foo 1.2.3.4,1.2.6.4    FOO barYP_MULTI_bar 1.2.3.4,1.2.6.4    FOO bar*/static struct hostent *makehostent(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;	static const char spaces[] = " \t";	char *cp, **q, *p, *comma, *ap;	int af = 0, len = 0;	int multi = 0;	int addr = 0;	p = pvt->curval_data;	if ((cp = strpbrk(p, "#\n")) != NULL)		*cp = '\0';	if (!(cp = strpbrk(p, spaces)))		return (NULL);	*cp++ = '\0';	ap = pvt->hostbuf;	do {		if ((comma = strchr(p, ',')) != NULL) {			*comma++ = '\0';			multi = 1;		}		if ((ap + IN6ADDRSZ) > (pvt->hostbuf + sizeof(pvt->hostbuf)))			break;		if ((pvt->res->options & RES_USE_INET6) &&		    inet_pton(AF_INET6, p, ap) > 0) {			af = AF_INET6;			len = IN6ADDRSZ;		} else if (inet_pton(AF_INET, p, pvt->host_addr) > 0) {			if (pvt->res->options & RES_USE_INET6) {				map_v4v6_address((char*)pvt->host_addr, ap);				af = AF_INET6;				len = IN6ADDRSZ;			} else {				af = AF_INET;				len = INADDRSZ;			}		} else {			if (!multi)				return (NULL);			continue;		}		if (addr < MAXADDRS) {			pvt->h_addr_ptrs[addr++] = ap;			pvt->h_addr_ptrs[addr] = NULL;			ap += len;		}	} while ((p = comma) != NULL);	if (ap == pvt->hostbuf)		return (NULL);	pvt->host.h_addr_list = pvt->h_addr_ptrs;	pvt->host.h_length = len;	pvt->host.h_addrtype = af;	cp += strspn(cp, spaces);	pvt->host.h_name = cp;	q = pvt->host.h_aliases = pvt->host_aliases;	if ((cp = strpbrk(cp, spaces)) != NULL)		*cp++ = '\0';	while (cp && *cp) {		if (*cp == ' ' || *cp == '\t') {			cp++;			continue;		}		if (q < &pvt->host_aliases[MAXALIASES])			*q++ = cp;		if ((cp = strpbrk(cp, spaces)) != NULL)			*cp++ = '\0';	}	*q = NULL;	RES_SET_H_ERRNO(pvt->res, NETDB_SUCCESS);	return (&pvt->host);}static voidnisfree(struct pvt *pvt, enum do_what do_what) {	if ((do_what & do_key) && pvt->curkey_data) {		free(pvt->curkey_data);		pvt->curkey_data = NULL;	}	if ((do_what & do_val) && pvt->curval_data) {		free(pvt->curval_data);		pvt->curval_data = NULL;	}}static intinit(struct irs_ho *this) {	struct pvt *pvt = (struct pvt *)this->private;		if (!pvt->res && !ho_res_get(this))		return (-1);	if (((pvt->res->options & RES_INIT) == 0) &&	    res_ninit(pvt->res) == -1)		return (-1);	return (0);}#endif /*WANT_IRS_NIS*/