xref: /freebsd/lib/libc/resolv/res_mkquery.c (revision 62cfcf62f627e5093fb37026a6d8c98e4d2ef04c)
1 /*-
2  * SPDX-License-Identifier: (ISC AND BSD-3-Clause)
3  *
4  * Portions Copyright (C) 2004, 2005, 2008  Internet Systems Consortium, Inc. ("ISC")
5  * Portions Copyright (C) 1996, 1997, 1988, 1999, 2001, 2003  Internet Software Consortium.
6  *
7  * Permission to use, copy, modify, and/or 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 WITH
12  * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
13  * AND FITNESS.  IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
14  * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
15  * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
16  * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
17  * PERFORMANCE OF THIS SOFTWARE.
18  */
19 
20 /*-
21  * Copyright (c) 1985, 1993
22  *    The Regents of the University of California.  All rights reserved.
23  *
24  * Redistribution and use in source and binary forms, with or without
25  * modification, are permitted provided that the following conditions
26  * are met:
27  * 1. Redistributions of source code must retain the above copyright
28  *    notice, this list of conditions and the following disclaimer.
29  * 2. Redistributions in binary form must reproduce the above copyright
30  *    notice, this list of conditions and the following disclaimer in the
31  *    documentation and/or other materials provided with the distribution.
32  * 3. Neither the name of the University nor the names of its contributors
33  *    may be used to endorse or promote products derived from this software
34  *    without specific prior written permission.
35  *
36  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
37  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
38  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
39  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
40  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
41  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
42  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
43  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
44  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
45  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
46  * SUCH DAMAGE.
47  */
48 
49 /*
50  * Portions Copyright (c) 1993 by Digital Equipment Corporation.
51  *
52  * Permission to use, copy, modify, and distribute this software for any
53  * purpose with or without fee is hereby granted, provided that the above
54  * copyright notice and this permission notice appear in all copies, and that
55  * the name of Digital Equipment Corporation not be used in advertising or
56  * publicity pertaining to distribution of the document or software without
57  * specific, written prior permission.
58  *
59  * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
60  * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
61  * OF MERCHANTABILITY AND FITNESS.   IN NO EVENT SHALL DIGITAL EQUIPMENT
62  * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
63  * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
64  * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
65  * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
66  * SOFTWARE.
67  */
68 
69 #if defined(LIBC_SCCS) && !defined(lint)
70 static const char sccsid[] = "@(#)res_mkquery.c	8.1 (Berkeley) 6/4/93";
71 static const char rcsid[] = "$Id: res_mkquery.c,v 1.10 2008/12/11 09:59:00 marka Exp $";
72 #endif /* LIBC_SCCS and not lint */
73 #include <sys/cdefs.h>
74 __FBSDID("$FreeBSD$");
75 
76 #include "port_before.h"
77 #include <sys/param.h>
78 #include <netinet/in.h>
79 #include <arpa/nameser.h>
80 #include <netdb.h>
81 #include <resolv.h>
82 #include <stdio.h>
83 #include <string.h>
84 #include "port_after.h"
85 
86 /* Options.  Leave them on. */
87 #ifndef	DEBUG
88 #define	DEBUG
89 #endif
90 
91 extern const char *_res_opcodes[];
92 
93 /*%
94  * Form all types of queries.
95  * Returns the size of the result or -1.
96  */
97 int
98 res_nmkquery(res_state statp,
99 	     int op,			/*!< opcode of query  */
100 	     const char *dname,		/*!< domain name  */
101 	     int class, int type,	/*!< class and type of query  */
102 	     const u_char *data,	/*!< resource record data  */
103 	     int datalen,		/*!< length of data  */
104 	     const u_char *newrr_in,	/*!< new rr for modify or append  */
105 	     u_char *buf,		/*!< buffer to put query  */
106 	     int buflen)		/*!< size of buffer  */
107 {
108 	HEADER *hp;
109 	u_char *cp, *ep;
110 	int n;
111 	u_char *dnptrs[20], **dpp, **lastdnptr;
112 
113 	UNUSED(newrr_in);
114 
115 #ifdef DEBUG
116 	if (statp->options & RES_DEBUG)
117 		printf(";; res_nmkquery(%s, %s, %s, %s)\n",
118 		       _res_opcodes[op], dname, p_class(class), p_type(type));
119 #endif
120 	/*
121 	 * Initialize header fields.
122 	 */
123 	if ((buf == NULL) || (buflen < HFIXEDSZ))
124 		return (-1);
125 	memset(buf, 0, HFIXEDSZ);
126 	hp = (HEADER *) buf;
127 	statp->id = res_nrandomid(statp);
128 	hp->id = htons(statp->id);
129 	hp->opcode = op;
130 	hp->rd = (statp->options & RES_RECURSE) != 0U;
131 	hp->rcode = NOERROR;
132 	cp = buf + HFIXEDSZ;
133 	ep = buf + buflen;
134 	dpp = dnptrs;
135 	*dpp++ = buf;
136 	*dpp++ = NULL;
137 	lastdnptr = dnptrs + nitems(dnptrs);
138 	/*
139 	 * perform opcode specific processing
140 	 */
141 	switch (op) {
142 	case QUERY:	/*FALLTHROUGH*/
143 	case NS_NOTIFY_OP:
144 		if (ep - cp < QFIXEDSZ)
145 			return (-1);
146 		if ((n = dn_comp(dname, cp, ep - cp - QFIXEDSZ, dnptrs,
147 		    lastdnptr)) < 0)
148 			return (-1);
149 		cp += n;
150 		ns_put16(type, cp);
151 		cp += INT16SZ;
152 		ns_put16(class, cp);
153 		cp += INT16SZ;
154 		hp->qdcount = htons(1);
155 		if (op == QUERY || data == NULL)
156 			break;
157 		/*
158 		 * Make an additional record for completion domain.
159 		 */
160 		if ((ep - cp) < RRFIXEDSZ)
161 			return (-1);
162 		n = dn_comp((const char *)data, cp, ep - cp - RRFIXEDSZ,
163 			    dnptrs, lastdnptr);
164 		if (n < 0)
165 			return (-1);
166 		cp += n;
167 		ns_put16(T_NULL, cp);
168 		cp += INT16SZ;
169 		ns_put16(class, cp);
170 		cp += INT16SZ;
171 		ns_put32(0, cp);
172 		cp += INT32SZ;
173 		ns_put16(0, cp);
174 		cp += INT16SZ;
175 		hp->arcount = htons(1);
176 		break;
177 
178 	case IQUERY:
179 		/*
180 		 * Initialize answer section
181 		 */
182 		if (ep - cp < 1 + RRFIXEDSZ + datalen)
183 			return (-1);
184 		*cp++ = '\0';	/*%< no domain name */
185 		ns_put16(type, cp);
186 		cp += INT16SZ;
187 		ns_put16(class, cp);
188 		cp += INT16SZ;
189 		ns_put32(0, cp);
190 		cp += INT32SZ;
191 		ns_put16(datalen, cp);
192 		cp += INT16SZ;
193 		if (datalen) {
194 			memcpy(cp, data, datalen);
195 			cp += datalen;
196 		}
197 		hp->ancount = htons(1);
198 		break;
199 
200 	default:
201 		return (-1);
202 	}
203 	return (cp - buf);
204 }
205 
206 #ifdef RES_USE_EDNS0
207 /* attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). */
208 
209 int
210 res_nopt(res_state statp,
211 	 int n0,		/*%< current offset in buffer */
212 	 u_char *buf,		/*%< buffer to put query */
213 	 int buflen,		/*%< size of buffer */
214 	 int anslen)		/*%< UDP answer buffer size */
215 {
216 	HEADER *hp;
217 	u_char *cp, *ep;
218 	u_int16_t flags = 0;
219 
220 #ifdef DEBUG
221 	if ((statp->options & RES_DEBUG) != 0U)
222 		printf(";; res_nopt()\n");
223 #endif
224 
225 	hp = (HEADER *) buf;
226 	cp = buf + n0;
227 	ep = buf + buflen;
228 
229 	if ((ep - cp) < 1 + RRFIXEDSZ)
230 		return (-1);
231 
232 	*cp++ = 0;				/*%< "." */
233 	ns_put16(ns_t_opt, cp);			/*%< TYPE */
234 	cp += INT16SZ;
235 	if (anslen > 0xffff)
236 		anslen = 0xffff;		/* limit to 16bit value */
237 	ns_put16(anslen & 0xffff, cp);		/*%< CLASS = UDP payload size */
238 	cp += INT16SZ;
239 	*cp++ = NOERROR;			/*%< extended RCODE */
240 	*cp++ = 0;				/*%< EDNS version */
241 
242 	if (statp->options & RES_USE_DNSSEC) {
243 #ifdef DEBUG
244 		if (statp->options & RES_DEBUG)
245 			printf(";; res_opt()... ENDS0 DNSSEC\n");
246 #endif
247 		flags |= NS_OPT_DNSSEC_OK;
248 	}
249 	ns_put16(flags, cp);
250 	cp += INT16SZ;
251 
252 	ns_put16(0U, cp);			/*%< RDLEN */
253 	cp += INT16SZ;
254 
255 	hp->arcount = htons(ntohs(hp->arcount) + 1);
256 
257 	return (cp - buf);
258 }
259 
260 /*
261  * Construct variable data (RDATA) block for OPT psuedo-RR, append it
262  * to the buffer, then update the RDLEN field (previously set to zero by
263  * res_nopt()) with the new RDATA length.
264  */
265 int
266 res_nopt_rdata(res_state statp,
267 	  int n0,	 	/*%< current offset in buffer */
268 	  u_char *buf,	 	/*%< buffer to put query */
269 	  int buflen,		/*%< size of buffer */
270 	  u_char *rdata,	/*%< ptr to start of opt rdata */
271 	  u_short code,		/*%< OPTION-CODE */
272 	  u_short len,		/*%< OPTION-LENGTH */
273 	  u_char *data)		/*%< OPTION_DATA */
274 {
275 	register u_char *cp, *ep;
276 
277 #ifdef DEBUG
278 	if ((statp->options & RES_DEBUG) != 0U)
279 		printf(";; res_nopt_rdata()\n");
280 #endif
281 
282 	cp = buf + n0;
283 	ep = buf + buflen;
284 
285 	if ((ep - cp) < (4 + len))
286 		return (-1);
287 
288 	if (rdata < (buf + 2) || rdata >= ep)
289 		return (-1);
290 
291 	ns_put16(code, cp);
292 	cp += INT16SZ;
293 
294 	ns_put16(len, cp);
295 	cp += INT16SZ;
296 
297 	memcpy(cp, data, len);
298 	cp += len;
299 
300 	len = cp - rdata;
301 	ns_put16(len, rdata - 2);	/* Update RDLEN field */
302 
303 	return (cp - buf);
304 }
305 #endif
306 
307 /*! \file */
308