File:  [ELWIX - Embedded LightWeight unIX -] / embedaddon / dhcp / minires / res_findzonecut.c
Revision 1.1: download - view: text, annotated - select for diffs - revision graph
Tue Feb 21 22:30:18 2012 UTC (12 years, 4 months ago) by misho
CVS tags: MAIN, HEAD
Initial revision

    1: #if !defined(lint) && !defined(SABER)
    2: static const char rcsid[] = "$Id: res_findzonecut.c,v 1.1 2012/02/21 22:30:18 misho Exp $";
    3: #endif /* not lint */
    4: 
    5: /*
    6:  * Copyright (c) 2009-2010 by Internet Systems Consortium, Inc. ("ISC")
    7:  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
    8:  * Copyright (c) 1999-2003 by Internet Software Consortium
    9:  *
   10:  * Permission to use, copy, modify, and distribute this software for any
   11:  * purpose with or without fee is hereby granted, provided that the above
   12:  * copyright notice and this permission notice appear in all copies.
   13:  *
   14:  * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
   15:  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
   16:  * MERCHANTABILITY AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR
   17:  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
   18:  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
   19:  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
   20:  * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
   21:  *
   22:  *   Internet Systems Consortium, Inc.
   23:  *   950 Charter Street
   24:  *   Redwood City, CA 94063
   25:  *   <info@isc.org>
   26:  *   https://www.isc.org/
   27:  */
   28: 
   29: /* Import. */
   30: 
   31: #include <sys/param.h>
   32: #include <sys/socket.h>
   33: #include <sys/time.h>
   34: 
   35: #include <netinet/in.h>
   36: #include <arpa/inet.h>
   37: 
   38: #include <errno.h>
   39: #include <limits.h>
   40: #include <netdb.h>
   41: #include <stdarg.h>
   42: #include <stdio.h>
   43: #include <stdlib.h>
   44: #include <string.h>
   45: 
   46: #include <isc-dhcp/list.h>
   47: 
   48: #include "minires/minires.h"
   49: #include "arpa/nameser.h"
   50: 
   51: /* Data structures. */
   52: 
   53: typedef struct rr_a {
   54: 	ISC_LINK(struct rr_a)	link;
   55: 	struct in_addr		addr;
   56: } rr_a;
   57: typedef ISC_LIST(rr_a) rrset_a;
   58: 
   59: typedef struct rr_ns {
   60: 	ISC_LINK(struct rr_ns) link;
   61: 	char *name;
   62: 	rrset_a addrs;
   63: } rr_ns;
   64: typedef ISC_LIST(rr_ns) rrset_ns;
   65: 
   66: /* Forward. */
   67: 
   68: static int	satisfy(res_state,
   69: 			const char *, rrset_ns *, struct in_addr *, int);
   70: static int	add_addrs(res_state, rr_ns *, struct in_addr *, int);
   71: static isc_result_t get_soa(res_state, const char *, ns_class,
   72: 			char *, size_t, char *, size_t,
   73: 			rrset_ns *);
   74: static isc_result_t get_ns(res_state, const char *, ns_class, rrset_ns *);
   75: static isc_result_t get_glue(res_state, ns_class, rrset_ns *);
   76: static isc_result_t save_ns(res_state, ns_msg *, ns_sect,
   77: 			    const char *, ns_class, rrset_ns *);
   78: static isc_result_t save_a(res_state, ns_msg *, ns_sect,
   79: 			   const char *, ns_class, rrset_a *);
   80: static void	free_nsrrset(rrset_ns *);
   81: static void	free_nsrr(rrset_ns *, rr_ns *);
   82: static rr_ns *	find_ns(rrset_ns *, const char *);
   83: static isc_result_t do_query(res_state, const char *, ns_class, ns_type,
   84: 			     double *, ns_msg *, int *);
   85: 
   86: /* Public. */
   87: 
   88: /*
   89:  * int
   90:  * res_findzonecut(res, dname, class, zname, zsize, addrs, naddrs)
   91:  *	find enclosing zone for a <dname,class>, and some server addresses
   92:  * parameters:
   93:  *	res - resolver context to work within (is modified)
   94:  *	dname - domain name whose enclosing zone is desired
   95:  *	class - class of dname (and its enclosing zone)
   96:  *	zname - found zone name
   97:  *	zsize - allocated size of zname
   98:  *	addrs - found server addresses
   99:  *	naddrs - max number of addrs
  100:  * return values:
  101:  *	< 0 - an error occurred (check errno)
  102:  *	= 0 - zname is now valid, but addrs[] wasn't changed
  103:  *	> 0 - zname is now valid, and return value is number of addrs[] found
  104:  * notes:
  105:  *	this function calls res_nsend() which means it depends on correctly
  106:  *	functioning recursive nameservers (usually defined in /etc/resolv.conf
  107:  *	or its local equivilent).
  108:  *
  109:  *	we start by asking for an SOA<dname,class>.  if we get one as an
  110:  *	answer, that just means <dname,class> is a zone top, which is fine.
  111:  *	more than likely we'll be told to go pound sand, in the form of a
  112:  *	negative answer.
  113:  *
  114:  *	note that we are not prepared to deal with referrals since that would
  115:  *	only come from authority servers and our correctly functioning local
  116:  *	recursive server would have followed the referral and got us something
  117:  *	more definite.
  118:  *
  119:  *	if the authority section contains an SOA, this SOA should also be the
  120:  *	closest enclosing zone, since any intermediary zone cuts would've been
  121:  *	returned as referrals and dealt with by our correctly functioning local
  122:  *	recursive name server.  but an SOA in the authority section should NOT
  123:  *	match our dname (since that would have been returned in the answer
  124:  *	section).  an authority section SOA has to be "above" our dname.
  125:  *
  126:  *	we cannot fail to find an SOA in this way.  ultimately we'll return
  127:  *	a zname indicating the root zone if that's the closest enclosing zone.
  128:  *	however, since authority section SOA's were once optional, it's
  129:  *	possible that we'll have to go hunting for the enclosing SOA by
  130:  *	ripping labels off the front of our dname -- this is known as "doing
  131:  *	it the hard way."
  132:  *
  133:  *	ultimately we want some server addresses, which are ideally the ones
  134:  *	pertaining to the SOA.MNAME, but only if there is a matching NS RR.
  135:  *	so the second phase (after we find an SOA) is to go looking for the
  136:  *	NS RRset for that SOA's zone.
  137:  *
  138:  *	no answer section processed by this code is allowed to contain CNAME
  139:  *	or DNAME RR's.  for the SOA query this means we strip a label and
  140:  *	keep going.  for the NS and A queries this means we just give up.
  141:  */
  142: 
  143: isc_result_t
  144: res_findzonecut(res_state statp, const char *dname, ns_class class, int opts,
  145: 		char *zname, size_t zsize, struct in_addr *addrs, int naddrs,
  146: 		int *count, void *zcookie)
  147: {
  148: 	char mname[NS_MAXDNAME];
  149: 	u_long save_pfcode;
  150: 	rrset_ns nsrrs;
  151: 	int n = 0;
  152: 	isc_result_t rcode;
  153: 
  154: 	DPRINTF(("START dname='%s' class=%s, zsize=%ld, naddrs=%d",
  155: 		 dname, p_class(class), (long)zsize, naddrs));
  156: 	save_pfcode = statp->pfcode;
  157: 	statp->pfcode |= RES_PRF_HEAD2 | RES_PRF_HEAD1 | RES_PRF_HEADX |
  158: 			 RES_PRF_QUES | RES_PRF_ANS |
  159: 			 RES_PRF_AUTH | RES_PRF_ADD;
  160: 	ISC_LIST_INIT(nsrrs);
  161: 
  162: 	DPRINTF (("look for a predefined zone statement"));
  163: 	rcode = find_cached_zone (dname, class, zname, zsize,
  164: 				  addrs, naddrs, &n, zcookie);
  165: 	if (rcode == ISC_R_SUCCESS)
  166: 		goto done;
  167: 
  168: 	DPRINTF(("get the soa, and see if it has enough glue"));
  169: 	if ((rcode = get_soa(statp, dname, class, zname, zsize,
  170: 			     mname, sizeof mname, &nsrrs)) != ISC_R_SUCCESS ||
  171: 	    ((opts & RES_EXHAUSTIVE) == 0 &&
  172: 	     (n = satisfy(statp, mname, &nsrrs, addrs, naddrs)) > 0))
  173: 		goto done;
  174: 
  175: 	DPRINTF(("get the ns rrset and see if it has enough glue"));
  176: 	if ((rcode = get_ns(statp, zname, class, &nsrrs)) != ISC_R_SUCCESS ||
  177: 	    ((opts & RES_EXHAUSTIVE) == 0 &&
  178: 	     (n = satisfy(statp, mname, &nsrrs, addrs, naddrs)) > 0))
  179: 		goto done;
  180: 
  181: 	DPRINTF(("get the missing glue and see if it's finally enough"));
  182: 	if ((rcode = get_glue(statp, class, &nsrrs)) == ISC_R_SUCCESS)
  183: 		n = satisfy(statp, mname, &nsrrs, addrs, naddrs);
  184: 
  185: 	/* If we found the zone, cache it. */
  186: 	if (n > 0)
  187: 		cache_found_zone (class, zname, addrs, n);
  188:  done:
  189: 	DPRINTF(("FINISH n=%d (%s)", n, (n < 0) ? strerror(errno) : "OK"));
  190: 	free_nsrrset(&nsrrs);
  191: 	statp->pfcode = save_pfcode;
  192: 	if (count)
  193: 		*count = n;
  194: 	return rcode;
  195: }
  196: 
  197: /* Private. */
  198: 
  199: static int
  200: satisfy(res_state statp,
  201: 	const char *mname, rrset_ns *nsrrsp, struct in_addr *addrs, int naddrs)
  202: {
  203: 	rr_ns *nsrr;
  204: 	int n, x;
  205: 
  206: 	n = 0;
  207: 	nsrr = find_ns(nsrrsp, mname);
  208: 	if (nsrr != NULL) {
  209: 		x = add_addrs(statp, nsrr, addrs, naddrs);
  210: 		addrs += x;
  211: 		naddrs -= x;
  212: 		n += x;
  213: 	}
  214: 	for (nsrr = ISC_LIST_HEAD(*nsrrsp);
  215: 	     nsrr != NULL && naddrs > 0;
  216: 	     nsrr = ISC_LIST_NEXT(nsrr, link))
  217: 		if (ns_samename(nsrr->name, mname) != 1) {
  218: 			x = add_addrs(statp, nsrr, addrs, naddrs);
  219: 			addrs += x;
  220: 			naddrs -= x;
  221: 			n += x;
  222: 		}
  223: 	DPRINTF(("satisfy(%s): %d", mname, n));
  224: 	return (n);
  225: }
  226: 
  227: static int
  228: add_addrs(res_state statp, rr_ns *nsrr, struct in_addr *addrs, int naddrs) {
  229: 	rr_a *arr;
  230: 	int n = 0;
  231: 
  232: 	for (arr = ISC_LIST_HEAD(nsrr->addrs);
  233: 	     arr != NULL; arr = ISC_LIST_NEXT(arr, link)) {
  234: 		if (naddrs <= 0)
  235: 			return (0);
  236: 		*addrs++ = arr->addr;
  237: 		naddrs--;
  238: 		n++;
  239: 	}
  240: 	DPRINTF(("add_addrs: %d", n));
  241: 	return (n);
  242: }
  243: 
  244: static isc_result_t
  245: get_soa(res_state statp, const char *dname, ns_class class,
  246: 	char *zname, size_t zsize, char *mname, size_t msize,
  247: 	rrset_ns *nsrrsp)
  248: {
  249: 	char tname[NS_MAXDNAME];
  250: 	double resp[NS_PACKETSZ / sizeof (double)];
  251: 	int n, i, ancount, nscount;
  252: 	ns_sect sect;
  253: 	ns_msg msg;
  254: 	isc_result_t rcode;
  255: 
  256: 	/*
  257: 	 * Find closest enclosing SOA, even if it's for the root zone.
  258: 	 */
  259: 
  260: 	/* First canonicalize dname (exactly one unescaped trailing "."). */
  261: 	rcode = ns_makecanon(dname, tname, sizeof tname);
  262: 	if (rcode != ISC_R_SUCCESS)
  263: 		return rcode;
  264: 	dname = tname;
  265: 
  266: 	/* Now grovel the subdomains, hunting for an SOA answer or auth. */
  267: 	for (;;) {
  268: 		/* Leading or inter-label '.' are skipped here. */
  269: 		while (*dname == '.')
  270: 			dname++;
  271: 
  272: 		/* Is there an SOA? */
  273: 		rcode = do_query(statp, dname, class, ns_t_soa,
  274: 				 resp, &msg, &n);
  275: 		if (rcode != ISC_R_SUCCESS) {
  276: 			DPRINTF(("get_soa: do_query('%s', %s) failed (%d)",
  277: 				 dname, p_class(class), n));
  278: 			return rcode;
  279: 		}
  280: 		if (n > 0) {
  281: 			DPRINTF(("get_soa: CNAME or DNAME found"));
  282: 			sect = ns_s_max, n = 0;
  283: 		} else {
  284: 			ancount = ns_msg_count(msg, ns_s_an);
  285: 			nscount = ns_msg_count(msg, ns_s_ns);
  286: 			if (ancount > 0 && rcode == ISC_R_SUCCESS)
  287: 				sect = ns_s_an, n = ancount;
  288: 			else if (nscount > 0)
  289: 				sect = ns_s_ns, n = nscount;
  290: 			else
  291: 				sect = ns_s_max, n = 0;
  292: 		}
  293: 		for (i = 0; i < n; i++) {
  294: 			const char *t;
  295: 			const u_char *rdata;
  296: 			int rdlen;
  297: 			ns_rr rr;
  298: 
  299: 			rcode = ns_parserr(&msg, sect, i, &rr) < 0;
  300: 			if (rcode != ISC_R_SUCCESS) {
  301: 				DPRINTF(("get_soa: ns_parserr(%s, %d) failed",
  302: 					 p_section(sect, ns_o_query), i));
  303: 				return rcode;
  304: 			}
  305: 			if (ns_rr_type(rr) == ns_t_cname ||
  306: 			    ns_rr_type(rr) == ns_t_dname)
  307: 				break;
  308: 			if (ns_rr_type(rr) != ns_t_soa ||
  309: 			    ns_rr_class(rr) != class)
  310: 				continue;
  311: 			t = ns_rr_name(rr);
  312: 			switch (sect) {
  313: 			case ns_s_an:
  314: 				if (ns_samedomain(dname, t) == 0) {
  315: 					DPRINTF(("get_soa: %s'%s', '%s') == 0",
  316: 						 "ns_samedomain(", dname, t));
  317: 					return ISC_R_NOTZONE;
  318: 				}
  319: 				break;
  320: 			case ns_s_ns:
  321: 				if (ns_samename(dname, t) == 1 ||
  322: 				    ns_samedomain(dname, t) == 0) {
  323: 					DPRINTF(("get_soa: %smain('%s', '%s')",
  324: 						 "ns_samename() || !ns_samedo",
  325: 						 dname, t));
  326: 					return ISC_R_NOTZONE;
  327: 				}
  328: 				break;
  329: 			default:
  330: 				abort();
  331: 			}
  332: 			if (strlen(t) + 1 > zsize) {
  333: 				DPRINTF(("get_soa: zname(%d) too small (%d)",
  334: 					 zsize, strlen(t) + 1));
  335: 				return ISC_R_NOSPACE;
  336: 			}
  337: 			strcpy(zname, t);
  338: 			rdata = ns_rr_rdata(rr);
  339: 			rdlen = ns_rr_rdlen(rr);
  340: 			if (ns_name_uncompress((u_char *)resp,
  341: 					       ns_msg_end(msg), rdata,
  342: 					       mname, msize) < 0) {
  343: 				DPRINTF(("get_soa: %s failed",
  344: 					 "ns_name_uncompress"));
  345: 				return ISC_R_NOMEMORY;
  346: 			}
  347: 			rcode = save_ns(statp, &msg,
  348: 					ns_s_ns, zname, class, nsrrsp);
  349: 			if (rcode != ISC_R_SUCCESS) {
  350: 				DPRINTF(("get_soa: save_ns failed"));
  351: 				return rcode;
  352: 			}
  353: 			return ISC_R_SUCCESS;
  354: 		}
  355: 
  356: 		/* If we're out of labels, then not even "." has an SOA! */
  357: 		if (*dname == '\0')
  358: 			break;
  359: 
  360: 		/* Find label-terminating "."; top of loop will skip it. */
  361: 		while (*dname != '.') {
  362: 			if (*dname == '\\')
  363: 				if (*++dname == '\0') {
  364: 					return ISC_R_NOSPACE;
  365: 				}
  366: 			dname++;
  367: 		}
  368: 	}
  369: 	DPRINTF(("get_soa: out of labels"));
  370: 	return ISC_R_DESTADDRREQ;
  371: }
  372: 
  373: static isc_result_t
  374: get_ns(res_state statp, const char *zname, ns_class class, rrset_ns *nsrrsp) {
  375: 	double resp[NS_PACKETSZ / sizeof (double)];
  376: 	ns_msg msg;
  377: 	int n;
  378: 	isc_result_t rcode;
  379: 
  380: 	/* Go and get the NS RRs for this zone. */
  381: 	rcode = do_query(statp, zname, class, ns_t_ns, resp, &msg, &n);
  382: 	if (rcode != ISC_R_SUCCESS) {
  383: 		DPRINTF(("get_ns: do_query('zname', %s) failed (%d)",
  384: 			 zname, p_class(class), rcode));
  385: 		return rcode;
  386: 	}
  387: 
  388: 	/* Remember the NS RRs and associated A RRs that came back. */
  389: 	rcode = save_ns(statp, &msg, ns_s_an, zname, class, nsrrsp);
  390: 	if (rcode != ISC_R_SUCCESS) {
  391: 		DPRINTF(("get_ns save_ns('%s', %s) failed",
  392: 			 zname, p_class(class)));
  393: 		return rcode;
  394: 	}
  395: 
  396: 	return ISC_R_SUCCESS;
  397: }
  398: 
  399: static isc_result_t
  400: get_glue(res_state statp, ns_class class, rrset_ns *nsrrsp) {
  401: 	rr_ns *nsrr, *nsrr_n;
  402: 
  403: 	/* Go and get the A RRs for each empty NS RR on our list. */
  404: 	for (nsrr = ISC_LIST_HEAD(*nsrrsp); nsrr != NULL; nsrr = nsrr_n) {
  405: 		double resp[NS_PACKETSZ / sizeof (double)];
  406: 		ns_msg msg;
  407: 		int n;
  408: 		isc_result_t rcode;
  409: 
  410: 		nsrr_n = ISC_LIST_NEXT(nsrr, link);
  411: 
  412: 		if (ISC_LIST_EMPTY(nsrr->addrs)) {
  413: 			rcode = do_query(statp, nsrr->name, class, ns_t_a,
  414: 					 resp, &msg, &n);
  415: 			if (rcode != ISC_R_SUCCESS) {
  416: 			    DPRINTF(("get_glue: do_query('%s', %s') failed",
  417: 				     nsrr->name, p_class(class)));
  418: 			    return rcode;
  419: 			}
  420: 			if (n > 0) {
  421: 				DPRINTF((
  422: 			"get_glue: do_query('%s', %s') CNAME or DNAME found",
  423: 					 nsrr->name, p_class(class)));
  424: 			}
  425: 			rcode = save_a(statp, &msg, ns_s_an, nsrr->name, class,
  426: 				       &nsrr->addrs);
  427: 			if (rcode != ISC_R_SUCCESS) {
  428: 				DPRINTF(("get_glue: save_r('%s', %s) failed",
  429: 					 nsrr->name, p_class(class)));
  430: 				return rcode;
  431: 			}
  432: 			/* If it's still empty, it's just chaff. */
  433: 			if (ISC_LIST_EMPTY(nsrr->addrs)) {
  434: 				DPRINTF(("get_glue: removing empty '%s' NS",
  435: 					 nsrr->name));
  436: 				free_nsrr(nsrrsp, nsrr);
  437: 			}
  438: 		}
  439: 	}
  440: 	return ISC_R_SUCCESS;
  441: }
  442: 
  443: static isc_result_t
  444: save_ns(res_state statp, ns_msg *msg, ns_sect sect,
  445: 	const char *owner, ns_class class,
  446: 	rrset_ns *nsrrsp)
  447: {
  448: 	int i;
  449: 	isc_result_t rcode;
  450: 
  451: 	for (i = 0; i < ns_msg_count(*msg, sect); i++) {
  452: 		char tname[MAXDNAME];
  453: 		const u_char *rdata;
  454: 		rr_ns *nsrr;
  455: 		ns_rr rr;
  456: 		int rdlen;
  457: 
  458: 		rcode = ns_parserr(msg, sect, i, &rr);
  459: 		if (rcode != ISC_R_SUCCESS) {
  460: 			DPRINTF(("save_ns: ns_parserr(%s, %d) failed",
  461: 				 p_section(sect, ns_o_query), i));
  462: 			return rcode;
  463: 		}
  464: 		if (ns_rr_type(rr) != ns_t_ns ||
  465: 		    ns_rr_class(rr) != class ||
  466: 		    ns_samename(ns_rr_name(rr), owner) != 1)
  467: 			continue;
  468: 		nsrr = find_ns(nsrrsp, ns_rr_name(rr));
  469: 		if (nsrr == NULL) {
  470: 			nsrr = malloc(sizeof *nsrr);
  471: 			if (nsrr == NULL) {
  472: 				DPRINTF(("save_ns: malloc failed"));
  473: 				return ISC_R_NOMEMORY;
  474: 			}
  475: 			rdata = ns_rr_rdata(rr);
  476: 			rdlen = ns_rr_rdlen(rr);
  477: 			if (ns_name_uncompress(ns_msg_base(*msg),
  478: 					       ns_msg_end(*msg), rdata,
  479: 					       tname, sizeof tname) < 0) {
  480: 				DPRINTF(("save_ns: ns_name_uncompress failed"));
  481: 				free(nsrr);
  482: 				return ISC_R_NOMEMORY;
  483: 			}
  484: 			nsrr->name = strdup(tname);
  485: 			if (nsrr->name == NULL) {
  486: 				DPRINTF(("save_ns: strdup failed"));
  487: 				free(nsrr);
  488: 				return ISC_R_NOMEMORY;
  489: 			}
  490: 			ISC_LIST_INIT(nsrr->addrs);
  491: 			ISC_LIST_APPEND(*nsrrsp, nsrr, link);
  492: 		}
  493: 		rcode = save_a(statp, msg, ns_s_ar,
  494: 			       nsrr->name, class, &nsrr->addrs);
  495: 		if (rcode != ISC_R_SUCCESS) {
  496: 			DPRINTF(("save_ns: save_r('%s', %s) failed",
  497: 				 nsrr->name, p_class(class)));
  498: 			return rcode;
  499: 		}
  500: 	}
  501: 	return ISC_R_SUCCESS;
  502: }
  503: 
  504: static isc_result_t
  505: save_a(res_state statp, ns_msg *msg, ns_sect sect,
  506:        const char *owner, ns_class class,
  507:        rrset_a *arrsp)
  508: {
  509: 	int i;
  510: 	isc_result_t rcode;
  511: 
  512: 	for (i = 0; i < ns_msg_count(*msg, sect); i++) {
  513: 		ns_rr rr;
  514: 		rr_a *arr;
  515: 
  516: 		rcode = ns_parserr(msg, sect, i, &rr);
  517: 		if (rcode != ISC_R_SUCCESS) {
  518: 			DPRINTF(("save_a: ns_parserr(%s, %d) failed",
  519: 				 p_section(sect, ns_o_query), i));
  520: 			return rcode;
  521: 		}
  522: 		if (ns_rr_type(rr) != ns_t_a ||
  523: 		    ns_rr_class(rr) != class ||
  524: 		    ns_samename(ns_rr_name(rr), owner) != 1 ||
  525: 		    ns_rr_rdlen(rr) != NS_INADDRSZ)
  526: 			continue;
  527: 		arr = malloc(sizeof *arr);
  528: 		if (arr == NULL) {
  529: 			DPRINTF(("save_a: malloc failed"));
  530: 			return ISC_R_NOMEMORY;
  531: 		}
  532: 		memcpy(&arr->addr, ns_rr_rdata(rr), NS_INADDRSZ);
  533: 		ISC_LIST_APPEND(*arrsp, arr, link);
  534: 	}
  535: 	return ISC_R_SUCCESS;
  536: }
  537: 
  538: static void
  539: free_nsrrset(rrset_ns *nsrrsp) {
  540: 	rr_ns *nsrr;
  541: 
  542: 	while ((nsrr = ISC_LIST_HEAD(*nsrrsp)) != NULL)
  543: 		free_nsrr(nsrrsp, nsrr);
  544: }
  545: 
  546: static void
  547: free_nsrr(rrset_ns *nsrrsp, rr_ns *nsrr) {
  548: 	rr_a *arr;
  549: 
  550: 	while ((arr = ISC_LIST_HEAD(nsrr->addrs)) != NULL) {
  551: 		ISC_LIST_UNLINK(nsrr->addrs, arr, link);
  552: 		free(arr);
  553: 	}
  554: 	free((char *)nsrr->name);
  555: 	ISC_LIST_UNLINK(*nsrrsp, nsrr, link);
  556: 	free(nsrr);
  557: }
  558: 
  559: static rr_ns *
  560: find_ns(rrset_ns *nsrrsp, const char *dname) {
  561: 	rr_ns *nsrr;
  562: 
  563: 	for (nsrr = ISC_LIST_HEAD(*nsrrsp);
  564: 	     nsrr != NULL; nsrr = ISC_LIST_NEXT(nsrr, link))
  565: 		if (ns_samename(nsrr->name, dname) == 1)
  566: 			return (nsrr);
  567: 	return (NULL);
  568: }
  569: 
  570: static isc_result_t
  571: do_query(res_state statp, const char *dname, ns_class class, ns_type qtype,
  572: 	 double *resp, ns_msg *msg, int *alias_count)
  573: {
  574: 	double req[NS_PACKETSZ / sizeof (double)];
  575: 	int i;
  576: 	unsigned n;
  577: 	isc_result_t status;
  578: 
  579: 	status = res_nmkquery(statp, ns_o_query, dname, class, qtype,
  580: 			      NULL, 0, NULL, req, NS_PACKETSZ, &n);
  581: 	if (status != ISC_R_SUCCESS) {
  582: 		DPRINTF(("do_query: res_nmkquery failed"));
  583: 		return status;
  584: 	}
  585: 	status = res_nsend(statp, req, n, resp, NS_PACKETSZ, &n);
  586: 	if (status != ISC_R_SUCCESS) {
  587: 		DPRINTF(("do_query: res_nsend failed"));
  588: 		return status;
  589: 	}
  590: 	if (n == 0) {
  591: 		DPRINTF(("do_query: res_nsend returned 0"));
  592: 		return ISC_R_NOTFOUND;
  593: 	}
  594: 	if (ns_initparse((u_char *)resp, n, msg) < 0) {
  595: 		DPRINTF(("do_query: ns_initparse failed"));
  596: 		return ISC_R_NOSPACE;
  597: 	}
  598: 	n = 0;
  599: 	for (i = 0; i < ns_msg_count(*msg, ns_s_an); i++) {
  600: 		ns_rr rr;
  601: 
  602: 		status = ns_parserr(msg, ns_s_an, i, &rr);
  603: 		if (status != ISC_R_SUCCESS) {
  604: 			DPRINTF(("do_query: ns_parserr failed"));
  605: 			return status;
  606: 		}
  607: 		n += (ns_rr_class(rr) == class &&
  608: 		      (ns_rr_type(rr) == ns_t_cname ||
  609: 		       ns_rr_type(rr) == ns_t_dname));
  610: 	}
  611: 	if (alias_count)
  612: 		*alias_count = n;
  613: 	return ISC_R_SUCCESS;
  614: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>