libc/dns/resolv/res_mkquery.c - nest-cam/4320010/libc - Git at Google

 /*	$NetBSD: res_mkquery.c,v 1.6 2006/01/24 17:40:32 christos Exp $	*/

 /*
  * Copyright (c) 2008  Android Open Source Project (query id randomization)
  * Copyright (c) 1985, 1993
  *    The Regents of the University of California.  All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  * 1. Redistributions of source code must retain the above copyright
  *    notice, this list of conditions and the following disclaimer.
  * 2. Redistributions in binary form must reproduce the above copyright
  *    notice, this list of conditions and the following disclaimer in the
  *    documentation and/or other materials provided with the distribution.
  * 3. All advertising materials mentioning features or use of this software
  *    must display the following acknowledgement:
  * 	This product includes software developed by the University of
  * 	California, Berkeley and its contributors.
  * 4. Neither the name of the University nor the names of its contributors
  *    may be used to endorse or promote products derived from this software
  *    without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  * SUCH DAMAGE.
  */

 /*
  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
  *
  * 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, and that
  * the name of Digital Equipment Corporation not be used in advertising or
  * publicity pertaining to distribution of the document or software without
  * specific, written prior permission.
  *
  * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
  * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
  * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
  * CORPORATION 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.
  */

 /*
  * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
  * Portions 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.
  */

 #include <sys/cdefs.h>
 #if defined(LIBC_SCCS) && !defined(lint)
 #ifdef notdef
 static const char sccsid[] = "@(#)res_mkquery.c	8.1 (Berkeley) 6/4/93";
 static const char rcsid[] = "Id: res_mkquery.c,v 1.1.2.2.4.2 2004/03/16 12:34:18 marka Exp";
 #else
 __RCSID("$NetBSD: res_mkquery.c,v 1.6 2006/01/24 17:40:32 christos Exp $");
 #endif
 #endif /* LIBC_SCCS and not lint */


 #include <sys/types.h>
 #include <sys/param.h>
 #include <netinet/in.h>
 #include <arpa/nameser.h>
 #include <netdb.h>
 #ifdef ANDROID_CHANGES
 #include "resolv_private.h"
 #else
 #include <resolv.h>
 #endif
 #include <stdio.h>
 #include <string.h>

 /* Options.  Leave them on. */
 #ifndef DEBUG
 #define DEBUG
 #endif

 #ifndef lint
 #define UNUSED(a)	(void)&a
 #else
 #define UNUSED(a)	a = a
 #endif

 extern const char *_res_opcodes[];

 /*
  * Form all types of queries.
  * Returns the size of the result or -1.
  */
 int
 res_nmkquery(res_state statp,
 	     int op,			/* opcode of query */
 	     const char *dname,		/* domain name */
 	     int class, int type,	/* class and type of query */
 	     const u_char *data,	/* resource record data */
 	     int datalen,		/* length of data */
 	     const u_char *newrr_in,	/* new rr for modify or append */
 	     u_char *buf,		/* buffer to put query */
 	     int buflen)		/* size of buffer */
 {
 	register HEADER *hp;
 	register u_char *cp, *ep;
 	register int n;
 	u_char *dnptrs[20], **dpp, **lastdnptr;

 	UNUSED(newrr_in);

 #ifdef DEBUG
 	if (statp->options & RES_DEBUG)
 		printf(";; res_nmkquery(%s, %s, %s, %s)\n",
 		       _res_opcodes[op], dname, p_class(class), p_type(type));
 #endif
 	/*
 	 * Initialize header fields.
 	 */
 	if ((buf == NULL) || (buflen < HFIXEDSZ))
 		return (-1);
 	memset(buf, 0, HFIXEDSZ);
 	hp = (HEADER *)(void *)buf;
 	hp->id = htons(res_randomid());
 	hp->opcode = op;
 	hp->rd = (statp->options & RES_RECURSE) != 0U;
 	hp->rcode = NOERROR;
 	cp = buf + HFIXEDSZ;
 	ep = buf + buflen;
 	dpp = dnptrs;
 	*dpp++ = buf;
 	*dpp++ = NULL;
 	lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
 	/*
 	 * perform opcode specific processing
 	 */
 	switch (op) {
 	case QUERY:	/*FALLTHROUGH*/
 	case NS_NOTIFY_OP:
 		if (ep - cp < QFIXEDSZ)
 			return (-1);
 		if ((n = dn_comp(dname, cp, ep - cp - QFIXEDSZ, dnptrs,
 		    lastdnptr)) < 0)
 			return (-1);
 		cp += n;
 		ns_put16(type, cp);
 		cp += INT16SZ;
 		ns_put16(class, cp);
 		cp += INT16SZ;
 		hp->qdcount = htons(1);
 		if (op == QUERY || data == NULL)
 			break;
 		/*
 		 * Make an additional record for completion domain.
 		 */
 		if ((ep - cp) < RRFIXEDSZ)
 			return (-1);
 		n = dn_comp((const char *)data, cp, ep - cp - RRFIXEDSZ,
 			    dnptrs, lastdnptr);
 		if (n < 0)
 			return (-1);
 		cp += n;
 		ns_put16(T_NULL, cp);
 		cp += INT16SZ;
 		ns_put16(class, cp);
 		cp += INT16SZ;
 		ns_put32(0, cp);
 		cp += INT32SZ;
 		ns_put16(0, cp);
 		cp += INT16SZ;
 		hp->arcount = htons(1);
 		break;

 	case IQUERY:
 		/*
 		 * Initialize answer section
 		 */
 		if (ep - cp < 1 + RRFIXEDSZ + datalen)
 			return (-1);
 		*cp++ = '\0';	/* no domain name */
 		ns_put16(type, cp);
 		cp += INT16SZ;
 		ns_put16(class, cp);
 		cp += INT16SZ;
 		ns_put32(0, cp);
 		cp += INT32SZ;
 		ns_put16(datalen, cp);
 		cp += INT16SZ;
 		if (datalen) {
 			memcpy(cp, data, (size_t)datalen);
 			cp += datalen;
 		}
 		hp->ancount = htons(1);
 		break;

 	default:
 		return (-1);
 	}
 	return (cp - buf);
 }

 #ifdef RES_USE_EDNS0
 /* attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). */
 #ifndef T_OPT
 #define T_OPT	41
 #endif

 int
 res_nopt(res_state statp,
 	 int n0,		/* current offset in buffer */
 	 u_char *buf,		/* buffer to put query */
 	 int buflen,		/* size of buffer */
 	 int anslen)		/* UDP answer buffer size */
 {
 	register HEADER *hp;
 	register u_char *cp, *ep;
 	u_int16_t flags = 0;

 #ifdef DEBUG
 	if ((statp->options & RES_DEBUG) != 0U)
 		printf(";; res_nopt()\n");
 #endif

 	hp = (HEADER *)(void *)buf;
 	cp = buf + n0;
 	ep = buf + buflen;

 	if ((ep - cp) < 1 + RRFIXEDSZ)
 		return (-1);

 	*cp++ = 0;	/* "." */

 	ns_put16(T_OPT, cp);	/* TYPE */
 	cp += INT16SZ;
 	ns_put16(anslen & 0xffff, cp);	/* CLASS = UDP payload size */
 	cp += INT16SZ;
 	*cp++ = NOERROR;	/* extended RCODE */
 	*cp++ = 0;		/* EDNS version */
 	if (statp->options & RES_USE_DNSSEC) {
 #ifdef DEBUG
 		if (statp->options & RES_DEBUG)
 			printf(";; res_opt()... ENDS0 DNSSEC\n");
 #endif
 		flags |= NS_OPT_DNSSEC_OK;
 	}
 	ns_put16(flags, cp);
 	cp += INT16SZ;
 	ns_put16(0, cp);	/* RDLEN */
 	cp += INT16SZ;
 	hp->arcount = htons(ntohs(hp->arcount) + 1);

 	return (cp - buf);
 }
 #endif
	/* $NetBSD: res_mkquery.c,v 1.6 2006/01/24 17:40:32 christos Exp $ */

	/*
	* Copyright (c) 2008 Android Open Source Project (query id randomization)
	* Copyright (c) 1985, 1993
	* The Regents of the University of California. All rights reserved.
	*
	* Redistribution and use in source and binary forms, with or without
	* modification, are permitted provided that the following conditions
	* are met:
	* 1. Redistributions of source code must retain the above copyright
	* notice, this list of conditions and the following disclaimer.
	* 2. Redistributions in binary form must reproduce the above copyright
	* notice, this list of conditions and the following disclaimer in the
	* documentation and/or other materials provided with the distribution.
	* 3. All advertising materials mentioning features or use of this software
	* must display the following acknowledgement:
	* This product includes software developed by the University of
	* California, Berkeley and its contributors.
	* 4. Neither the name of the University nor the names of its contributors
	* may be used to endorse or promote products derived from this software
	* without specific prior written permission.
	*
	* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
	* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
	* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
	* ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
	* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
	* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
	* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
	* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
	* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
	* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
	* SUCH DAMAGE.
	*/

	/*
	* Portions Copyright (c) 1993 by Digital Equipment Corporation.
	*
	* 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, and that
	* the name of Digital Equipment Corporation not be used in advertising or
	* publicity pertaining to distribution of the document or software without
	* specific, written prior permission.
	*
	* THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
	* WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
	* OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL DIGITAL EQUIPMENT
	* CORPORATION 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.
	*/

	/*
	* Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
	* Portions 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.
	*/

	#include <sys/cdefs.h>
	#if defined(LIBC_SCCS) && !defined(lint)
	#ifdef notdef
	static const char sccsid[] = "@(#)res_mkquery.c 8.1 (Berkeley) 6/4/93";
	static const char rcsid[] = "Id: res_mkquery.c,v 1.1.2.2.4.2 2004/03/16 12:34:18 marka Exp";
	#else
	__RCSID("$NetBSD: res_mkquery.c,v 1.6 2006/01/24 17:40:32 christos Exp $");
	#endif
	#endif /* LIBC_SCCS and not lint */



	#include <sys/types.h>
	#include <sys/param.h>
	#include <netinet/in.h>
	#include <arpa/nameser.h>
	#include <netdb.h>
	#ifdef ANDROID_CHANGES
	#include "resolv_private.h"
	#else
	#include <resolv.h>
	#endif
	#include <stdio.h>
	#include <string.h>

	/* Options. Leave them on. */
	#ifndef DEBUG
	#define DEBUG
	#endif

	#ifndef lint
	#define UNUSED(a) (void)&a
	#else
	#define UNUSED(a) a = a
	#endif

	extern const char *_res_opcodes[];

	/*
	* Form all types of queries.
	* Returns the size of the result or -1.
	*/
	int
	res_nmkquery(res_state statp,
	int op, /* opcode of query */
	const char dname, / domain name */
	int class, int type, /* class and type of query */
	const u_char data, / resource record data */
	int datalen, /* length of data */
	const u_char newrr_in, / new rr for modify or append */
	u_char buf, / buffer to put query */
	int buflen) /* size of buffer */
	{
	register HEADER *hp;
	register u_char cp, ep;
	register int n;
	u_char dnptrs[20], dpp, *lastdnptr;

	UNUSED(newrr_in);

	#ifdef DEBUG
	if (statp->options & RES_DEBUG)
	printf(";; res_nmkquery(%s, %s, %s, %s)\n",
	_res_opcodes[op], dname, p_class(class), p_type(type));
	#endif
	/*
	* Initialize header fields.
	*/
	if ((buf == NULL) \|\| (buflen < HFIXEDSZ))
	return (-1);
	memset(buf, 0, HFIXEDSZ);
	hp = (HEADER )(void )buf;
	hp->id = htons(res_randomid());
	hp->opcode = op;
	hp->rd = (statp->options & RES_RECURSE) != 0U;
	hp->rcode = NOERROR;
	cp = buf + HFIXEDSZ;
	ep = buf + buflen;
	dpp = dnptrs;
	*dpp++ = buf;
	*dpp++ = NULL;
	lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
	/*
	* perform opcode specific processing
	*/
	switch (op) {
	case QUERY: /FALLTHROUGH/
	case NS_NOTIFY_OP:
	if (ep - cp < QFIXEDSZ)
	return (-1);
	if ((n = dn_comp(dname, cp, ep - cp - QFIXEDSZ, dnptrs,
	lastdnptr)) < 0)
	return (-1);
	cp += n;
	ns_put16(type, cp);
	cp += INT16SZ;
	ns_put16(class, cp);
	cp += INT16SZ;
	hp->qdcount = htons(1);
	if (op == QUERY \|\| data == NULL)
	break;
	/*
	* Make an additional record for completion domain.
	*/
	if ((ep - cp) < RRFIXEDSZ)
	return (-1);
	n = dn_comp((const char *)data, cp, ep - cp - RRFIXEDSZ,
	dnptrs, lastdnptr);
	if (n < 0)
	return (-1);
	cp += n;
	ns_put16(T_NULL, cp);
	cp += INT16SZ;
	ns_put16(class, cp);
	cp += INT16SZ;
	ns_put32(0, cp);
	cp += INT32SZ;
	ns_put16(0, cp);
	cp += INT16SZ;
	hp->arcount = htons(1);
	break;

	case IQUERY:
	/*
	* Initialize answer section
	*/
	if (ep - cp < 1 + RRFIXEDSZ + datalen)
	return (-1);
	cp++ = '\0'; / no domain name */
	ns_put16(type, cp);
	cp += INT16SZ;
	ns_put16(class, cp);
	cp += INT16SZ;
	ns_put32(0, cp);
	cp += INT32SZ;
	ns_put16(datalen, cp);
	cp += INT16SZ;
	if (datalen) {
	memcpy(cp, data, (size_t)datalen);
	cp += datalen;
	}
	hp->ancount = htons(1);
	break;

	default:
	return (-1);
	}
	return (cp - buf);
	}

	#ifdef RES_USE_EDNS0
	/* attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). */
	#ifndef T_OPT
	#define T_OPT 41
	#endif

	int
	res_nopt(res_state statp,
	int n0, /* current offset in buffer */
	u_char buf, / buffer to put query */
	int buflen, /* size of buffer */
	int anslen) /* UDP answer buffer size */
	{
	register HEADER *hp;
	register u_char cp, ep;
	u_int16_t flags = 0;

	#ifdef DEBUG
	if ((statp->options & RES_DEBUG) != 0U)
	printf(";; res_nopt()\n");
	#endif

	hp = (HEADER )(void )buf;
	cp = buf + n0;
	ep = buf + buflen;

	if ((ep - cp) < 1 + RRFIXEDSZ)
	return (-1);

	cp++ = 0; / "." */

	ns_put16(T_OPT, cp); /* TYPE */
	cp += INT16SZ;
	ns_put16(anslen & 0xffff, cp); /* CLASS = UDP payload size */
	cp += INT16SZ;
	cp++ = NOERROR; / extended RCODE */
	cp++ = 0; / EDNS version */
	if (statp->options & RES_USE_DNSSEC) {
	#ifdef DEBUG
	if (statp->options & RES_DEBUG)
	printf(";; res_opt()... ENDS0 DNSSEC\n");
	#endif
	flags \|= NS_OPT_DNSSEC_OK;
	}
	ns_put16(flags, cp);
	cp += INT16SZ;
	ns_put16(0, cp); /* RDLEN */
	cp += INT16SZ;
	hp->arcount = htons(ntohs(hp->arcount) + 1);

	return (cp - buf);
	}
	#endif