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