1/*
2 * Copyright (c) 1985, 1993
3 * The Regents of the University of California. All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 * notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 * notice, this list of conditions and the following disclaimer in the
12 * documentation and/or other materials provided with the distribution.
13 * 4. Neither the name of the University nor the names of its contributors
14 * may be used to endorse or promote products derived from this software
15 * without specific prior written permission.
16 *
17 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
21 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27 * SUCH DAMAGE.
28 */
29
30/*
31 * Portions Copyright (c) 1993 by Digital Equipment Corporation.
32 *
33 * Permission to use, copy, modify, and distribute this software for any
34 * purpose with or without fee is hereby granted, provided that the above
35 * copyright notice and this permission notice appear in all copies, and that
36 * the name of Digital Equipment Corporation not be used in advertising or
37 * publicity pertaining to distribution of the document or software without
38 * specific, written prior permission.
39 *
40 * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL
41 * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES
42 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL DIGITAL EQUIPMENT
43 * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
44 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
45 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
46 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
47 * SOFTWARE.
48 */
49
50/*
51 * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
52 *
53 * Permission to use, copy, modify, and distribute this software for any
54 * purpose with or without fee is hereby granted, provided that the above
55 * copyright notice and this permission notice appear in all copies.
56 *
57 * THE SOFTWARE IS PROVIDED "AS IS" AND INTERNET SOFTWARE CONSORTIUM DISCLAIMS
58 * ALL WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES
59 * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL INTERNET SOFTWARE
60 * CONSORTIUM BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL
61 * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR
62 * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
63 * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS
64 * SOFTWARE.
65 */
66
67#if defined(LIBC_SCCS) && !defined(lint)
68static const char sccsid[] = "@(#)res_mkquery.c 8.1 (Berkeley) 6/4/93";
69static const char rcsid[] = "$BINDId: res_mkquery.c,v 8.12 1999/10/13 16:39:40 vixie Exp $";
70#endif /* LIBC_SCCS and not lint */
71
72#include <sys/types.h>
73#include <sys/param.h>
74#include <netinet/in.h>
75#include <arpa/nameser.h>
76#include <netdb.h>
77#include <resolv.h>
78#include <stdio.h>
79#include <string.h>
80#include <sys/time.h>
81
82/* Options. Leave them on. */
83/* #define DEBUG */
84
85#ifdef _LIBC
86# include <hp-timing.h>
87# include <stdint.h>
88# if HP_TIMING_AVAIL
89# define RANDOM_BITS(Var) { uint64_t v64; HP_TIMING_NOW (v64); Var = v64; }
90# endif
91#endif
92
93/*
94 * Form all types of queries.
95 * Returns the size of the result or -1.
96 */
97int
98res_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;
110 int n;
111 u_char *dnptrs[20], **dpp, **lastdnptr;
112
113#ifdef DEBUG
114 if (statp->options & RES_DEBUG)
115 printf(";; res_nmkquery(%s, %s, %s, %s)\n",
116 _res_opcodes[op], dname, p_class(class), p_type(type));
117#endif
118 /*
119 * Initialize header fields.
120 */
121 if ((buf == NULL) || (buflen < HFIXEDSZ))
122 return (-1);
123 memset(buf, 0, HFIXEDSZ);
124 hp = (HEADER *) buf;
125 /* We randomize the IDs every time. The old code just
126 incremented by one after the initial randomization which
127 still predictable if the application does multiple
128 requests. */
129 int randombits;
130 do
131 {
132#ifdef RANDOM_BITS
133 RANDOM_BITS (randombits);
134#else
135 struct timeval tv;
136 __gettimeofday (&tv, NULL);
137 randombits = (tv.tv_sec << 8) ^ tv.tv_usec;
138#endif
139 }
140 while ((randombits & 0xffff) == 0);
141 statp->id = (statp->id + randombits) & 0xffff;
142 hp->id = statp->id;
143 hp->opcode = op;
144 hp->rd = (statp->options & RES_RECURSE) != 0;
145 hp->rcode = NOERROR;
146 cp = buf + HFIXEDSZ;
147 buflen -= HFIXEDSZ;
148 dpp = dnptrs;
149 *dpp++ = buf;
150 *dpp++ = NULL;
151 lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0];
152 /*
153 * perform opcode specific processing
154 */
155 switch (op) {
156 case NS_NOTIFY_OP:
157 if ((buflen -= QFIXEDSZ + (data == NULL ? 0 : RRFIXEDSZ)) < 0)
158 return (-1);
159 goto compose;
160
161 case QUERY:
162 if ((buflen -= QFIXEDSZ) < 0)
163 return (-1);
164 compose:
165 n = ns_name_compress(dname, cp, buflen,
166 (const u_char **) dnptrs,
167 (const u_char **) lastdnptr);
168 if (n < 0)
169 return (-1);
170 cp += n;
171 buflen -= n;
172 NS_PUT16 (type, cp);
173 NS_PUT16 (class, cp);
174 hp->qdcount = htons(1);
175 if (op == QUERY || data == NULL)
176 break;
177 /*
178 * Make an additional record for completion domain.
179 */
180 n = ns_name_compress((char *)data, cp, buflen,
181 (const u_char **) dnptrs,
182 (const u_char **) lastdnptr);
183 if (__glibc_unlikely (n < 0))
184 return (-1);
185 cp += n;
186 buflen -= n;
187 NS_PUT16 (T_NULL, cp);
188 NS_PUT16 (class, cp);
189 NS_PUT32 (0, cp);
190 NS_PUT16 (0, cp);
191 hp->arcount = htons(1);
192 break;
193
194 case IQUERY:
195 /*
196 * Initialize answer section
197 */
198 if (__glibc_unlikely (buflen < 1 + RRFIXEDSZ + datalen))
199 return (-1);
200 *cp++ = '\0'; /* no domain name */
201 NS_PUT16 (type, cp);
202 NS_PUT16 (class, cp);
203 NS_PUT32 (0, cp);
204 NS_PUT16 (datalen, cp);
205 if (datalen) {
206 memcpy(cp, data, datalen);
207 cp += datalen;
208 }
209 hp->ancount = htons(1);
210 break;
211
212 default:
213 return (-1);
214 }
215 return (cp - buf);
216}
217libresolv_hidden_def (res_nmkquery)
218
219
220/* attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). */
221#ifndef T_OPT
222#define T_OPT 41
223#endif
224
225int
226__res_nopt(res_state statp,
227 int n0, /* current offset in buffer */
228 u_char *buf, /* buffer to put query */
229 int buflen, /* size of buffer */
230 int anslen) /* UDP answer buffer size */
231{
232 u_int16_t flags = 0;
233
234#ifdef DEBUG
235 if ((statp->options & RES_DEBUG) != 0U)
236 printf(";; res_nopt()\n");
237#endif
238
239 HEADER *hp = (HEADER *) buf;
240 u_char *cp = buf + n0;
241 u_char *ep = buf + buflen;
242
243 if ((ep - cp) < 1 + RRFIXEDSZ)
244 return -1;
245
246 *cp++ = 0; /* "." */
247
248 NS_PUT16(T_OPT, cp); /* TYPE */
249 NS_PUT16(MIN(anslen, 0xffff), cp); /* CLASS = UDP payload size */
250 *cp++ = NOERROR; /* extended RCODE */
251 *cp++ = 0; /* EDNS version */
252
253 if (statp->options & RES_USE_DNSSEC) {
254#ifdef DEBUG
255 if (statp->options & RES_DEBUG)
256 printf(";; res_opt()... ENDS0 DNSSEC\n");
257#endif
258 flags |= NS_OPT_DNSSEC_OK;
259 }
260
261 NS_PUT16(flags, cp);
262 NS_PUT16(0, cp); /* RDLEN */
263 hp->arcount = htons(ntohs(hp->arcount) + 1);
264
265 return cp - buf;
266}
267libresolv_hidden_def (__res_nopt)
268