Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 1 | /* $NetBSD: res_mkquery.c,v 1.6 2006/01/24 17:40:32 christos Exp $ */ |
| 2 | |
| 3 | /* |
| 4 | * Copyright (c) 1985, 1993 |
| 5 | * The Regents of the University of California. All rights reserved. |
| 6 | * |
| 7 | * Redistribution and use in source and binary forms, with or without |
| 8 | * modification, are permitted provided that the following conditions |
| 9 | * are met: |
| 10 | * 1. Redistributions of source code must retain the above copyright |
| 11 | * notice, this list of conditions and the following disclaimer. |
| 12 | * 2. Redistributions in binary form must reproduce the above copyright |
| 13 | * notice, this list of conditions and the following disclaimer in the |
| 14 | * documentation and/or other materials provided with the distribution. |
| 15 | * 3. All advertising materials mentioning features or use of this software |
| 16 | * must display the following acknowledgement: |
| 17 | * This product includes software developed by the University of |
| 18 | * California, Berkeley and its contributors. |
| 19 | * 4. Neither the name of the University nor the names of its contributors |
| 20 | * may be used to endorse or promote products derived from this software |
| 21 | * without specific prior written permission. |
| 22 | * |
| 23 | * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND |
| 24 | * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
| 25 | * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
| 26 | * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
| 27 | * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
| 28 | * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
| 29 | * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
| 30 | * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
| 31 | * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
| 32 | * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
| 33 | * SUCH DAMAGE. |
| 34 | */ |
| 35 | |
| 36 | /* |
| 37 | * Portions Copyright (c) 1993 by Digital Equipment Corporation. |
| 38 | * |
| 39 | * Permission to use, copy, modify, and distribute this software for any |
| 40 | * purpose with or without fee is hereby granted, provided that the above |
| 41 | * copyright notice and this permission notice appear in all copies, and that |
| 42 | * the name of Digital Equipment Corporation not be used in advertising or |
| 43 | * publicity pertaining to distribution of the document or software without |
| 44 | * specific, written prior permission. |
| 45 | * |
| 46 | * THE SOFTWARE IS PROVIDED "AS IS" AND DIGITAL EQUIPMENT CORP. DISCLAIMS ALL |
| 47 | * WARRANTIES WITH REGARD TO THIS SOFTWARE, INCLUDING ALL IMPLIED WARRANTIES |
| 48 | * OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL DIGITAL EQUIPMENT |
| 49 | * CORPORATION BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL |
| 50 | * DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR |
| 51 | * PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS |
| 52 | * ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS |
| 53 | * SOFTWARE. |
| 54 | */ |
| 55 | |
| 56 | /* |
| 57 | * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC") |
| 58 | * Portions Copyright (c) 1996-1999 by Internet Software Consortium. |
| 59 | * |
| 60 | * Permission to use, copy, modify, and distribute this software for any |
| 61 | * purpose with or without fee is hereby granted, provided that the above |
| 62 | * copyright notice and this permission notice appear in all copies. |
| 63 | * |
| 64 | * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES |
| 65 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF |
| 66 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR |
| 67 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES |
| 68 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN |
| 69 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT |
| 70 | * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. |
| 71 | */ |
| 72 | |
Ken Chen | 5471dca | 2019-04-15 15:25:35 +0800 | [diff] [blame] | 73 | #define LOG_TAG "resolv" |
Bernie Innocenti | 3952ccc | 2019-03-03 19:39:53 +0900 | [diff] [blame] | 74 | |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 75 | #include <algorithm> // std::min() |
| 76 | |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 77 | #include <arpa/nameser.h> |
| 78 | #include <netdb.h> |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 79 | #include <netinet/in.h> |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 80 | #include <string.h> |
Bernie Innocenti | f33a63f | 2018-08-30 12:04:03 +0900 | [diff] [blame] | 81 | |
chenbruce | acb832c | 2019-02-20 19:45:50 +0800 | [diff] [blame] | 82 | #include <android-base/logging.h> |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 83 | |
chenbruce | acb832c | 2019-02-20 19:45:50 +0800 | [diff] [blame] | 84 | #include "resolv_private.h" |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 85 | |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 86 | // Queries will be padded to a multiple of this length when EDNS0 is active. |
| 87 | constexpr uint16_t kEdns0Padding = 128; |
| 88 | |
Mike Yu | bfb1b34 | 2018-11-06 15:42:36 +0800 | [diff] [blame] | 89 | extern const char* const _res_opcodes[] = { |
| 90 | "QUERY", "IQUERY", "CQUERYM", "CQUERYU", /* experimental */ |
| 91 | "NOTIFY", /* experimental */ |
| 92 | "UPDATE", "6", "7", "8", "9", "10", |
| 93 | "11", "12", "13", "ZONEINIT", "ZONEREF", |
| 94 | }; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 95 | |
| 96 | /* |
| 97 | * Form all types of queries. |
| 98 | * Returns the size of the result or -1. |
| 99 | */ |
chenbruce | c51f121 | 2019-09-12 16:59:33 +0800 | [diff] [blame] | 100 | int res_nmkquery(res_state statp, int op, /* opcode of query */ |
| 101 | const char* dname, /* domain name */ |
| 102 | int cl, int type, /* class and type of query */ |
| 103 | const uint8_t* data, /* resource record data */ |
| 104 | int datalen, /* length of data */ |
| 105 | const uint8_t* /*newrr_in*/, /* new rr for modify or append */ |
| 106 | uint8_t* buf, /* buffer to put query */ |
| 107 | int buflen) /* size of buffer */ |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 108 | { |
Bernie Innocenti | 9c57593 | 2018-09-07 21:10:25 +0900 | [diff] [blame] | 109 | HEADER* hp; |
chenbruce | c51f121 | 2019-09-12 16:59:33 +0800 | [diff] [blame] | 110 | uint8_t *cp, *ep; |
Bernie Innocenti | 9c57593 | 2018-09-07 21:10:25 +0900 | [diff] [blame] | 111 | int n; |
chenbruce | c51f121 | 2019-09-12 16:59:33 +0800 | [diff] [blame] | 112 | uint8_t *dnptrs[20], **dpp, **lastdnptr; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 113 | |
Ken Chen | ffc224a | 2019-03-19 17:41:28 +0800 | [diff] [blame] | 114 | LOG(DEBUG) << __func__ << ": (" << _res_opcodes[op] << ", " << p_class(cl) << ", " |
| 115 | << p_type(type) << ")"; |
chenbruce | acb832c | 2019-02-20 19:45:50 +0800 | [diff] [blame] | 116 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 117 | /* |
| 118 | * Initialize header fields. |
| 119 | */ |
| 120 | if ((buf == NULL) || (buflen < HFIXEDSZ)) return (-1); |
| 121 | memset(buf, 0, HFIXEDSZ); |
| 122 | hp = (HEADER*) (void*) buf; |
Luke Huang | dda920f | 2019-02-13 14:05:02 +0800 | [diff] [blame] | 123 | hp->id = htons(arc4random_uniform(65536)); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 124 | hp->opcode = op; |
chenbruce | 018fdb2 | 2019-06-12 18:08:04 +0800 | [diff] [blame] | 125 | hp->rd = true; |
chenbruce | d8cbb9b | 2019-06-20 18:25:28 +0800 | [diff] [blame] | 126 | hp->ad = (statp->netcontext_flags & NET_CONTEXT_FLAG_USE_DNS_OVER_TLS) != 0U; |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 127 | hp->rcode = NOERROR; |
| 128 | cp = buf + HFIXEDSZ; |
| 129 | ep = buf + buflen; |
| 130 | dpp = dnptrs; |
| 131 | *dpp++ = buf; |
| 132 | *dpp++ = NULL; |
| 133 | lastdnptr = dnptrs + sizeof dnptrs / sizeof dnptrs[0]; |
| 134 | /* |
| 135 | * perform opcode specific processing |
| 136 | */ |
| 137 | switch (op) { |
Bernie Innocenti | f40b3bd | 2018-10-10 22:30:12 +0900 | [diff] [blame] | 138 | case QUERY: |
| 139 | [[fallthrough]]; |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 140 | case NS_NOTIFY_OP: |
| 141 | if (ep - cp < QFIXEDSZ) return (-1); |
| 142 | if ((n = dn_comp(dname, cp, ep - cp - QFIXEDSZ, dnptrs, lastdnptr)) < 0) return (-1); |
| 143 | cp += n; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 144 | *reinterpret_cast<uint16_t*>(cp) = htons(type); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 145 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 146 | *reinterpret_cast<uint16_t*>(cp) = htons(cl); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 147 | cp += INT16SZ; |
| 148 | hp->qdcount = htons(1); |
| 149 | if (op == QUERY || data == NULL) break; |
| 150 | /* |
| 151 | * Make an additional record for completion domain. |
| 152 | */ |
| 153 | if ((ep - cp) < RRFIXEDSZ) return (-1); |
| 154 | n = dn_comp((const char*) data, cp, ep - cp - RRFIXEDSZ, dnptrs, lastdnptr); |
| 155 | if (n < 0) return (-1); |
| 156 | cp += n; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 157 | *reinterpret_cast<uint16_t*>(cp) = htons(ns_t_null); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 158 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 159 | *reinterpret_cast<uint16_t*>(cp) = htons(cl); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 160 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 161 | *reinterpret_cast<uint32_t*>(cp) = htonl(0); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 162 | cp += INT32SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 163 | *reinterpret_cast<uint16_t*>(cp) = htons(0); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 164 | cp += INT16SZ; |
| 165 | hp->arcount = htons(1); |
| 166 | break; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 167 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 168 | case IQUERY: |
| 169 | /* |
| 170 | * Initialize answer section |
| 171 | */ |
| 172 | if (ep - cp < 1 + RRFIXEDSZ + datalen) return (-1); |
| 173 | *cp++ = '\0'; /* no domain name */ |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 174 | *reinterpret_cast<uint16_t*>(cp) = htons(type); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 175 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 176 | *reinterpret_cast<uint16_t*>(cp) = htons(cl); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 177 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 178 | *reinterpret_cast<uint32_t*>(cp) = htonl(0); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 179 | cp += INT32SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 180 | *reinterpret_cast<uint16_t*>(cp) = htons(datalen); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 181 | cp += INT16SZ; |
| 182 | if (datalen) { |
| 183 | memcpy(cp, data, (size_t) datalen); |
| 184 | cp += datalen; |
| 185 | } |
| 186 | hp->ancount = htons(1); |
| 187 | break; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 188 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 189 | default: |
| 190 | return (-1); |
| 191 | } |
| 192 | return (cp - buf); |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 193 | } |
| 194 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 195 | int res_nopt(res_state statp, int n0, /* current offset in buffer */ |
chenbruce | c51f121 | 2019-09-12 16:59:33 +0800 | [diff] [blame] | 196 | uint8_t* buf, /* buffer to put query */ |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 197 | int buflen, /* size of buffer */ |
| 198 | int anslen) /* UDP answer buffer size */ |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 199 | { |
Bernie Innocenti | 9c57593 | 2018-09-07 21:10:25 +0900 | [diff] [blame] | 200 | HEADER* hp; |
chenbruce | c51f121 | 2019-09-12 16:59:33 +0800 | [diff] [blame] | 201 | uint8_t *cp, *ep; |
| 202 | uint16_t flags = 0; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 203 | |
Bernie Innocenti | 3952ccc | 2019-03-03 19:39:53 +0900 | [diff] [blame] | 204 | LOG(DEBUG) << __func__; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 205 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 206 | hp = (HEADER*) (void*) buf; |
| 207 | cp = buf + n0; |
| 208 | ep = buf + buflen; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 209 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 210 | if ((ep - cp) < 1 + RRFIXEDSZ) return (-1); |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 211 | |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 212 | *cp++ = 0; /* "." */ |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 213 | |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 214 | // Attach OPT pseudo-RR, as documented in RFC2671 (EDNS0). |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 215 | *reinterpret_cast<uint16_t*>(cp) = htons(ns_t_opt); /* TYPE */ |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 216 | cp += INT16SZ; |
| 217 | if (anslen > 0xffff) anslen = 0xffff; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 218 | *reinterpret_cast<uint16_t*>(cp) = htons(anslen); /* CLASS = UDP payload size */ |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 219 | cp += INT16SZ; |
| 220 | *cp++ = NOERROR; /* extended RCODE */ |
| 221 | *cp++ = 0; /* EDNS version */ |
chenbruce | d8cbb9b | 2019-06-20 18:25:28 +0800 | [diff] [blame] | 222 | if (statp->netcontext_flags & NET_CONTEXT_FLAG_USE_DNS_OVER_TLS) { |
Bernie Innocenti | 3952ccc | 2019-03-03 19:39:53 +0900 | [diff] [blame] | 223 | LOG(DEBUG) << __func__ << ": ENDS0 DNSSEC"; |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 224 | flags |= NS_OPT_DNSSEC_OK; |
| 225 | } |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 226 | *reinterpret_cast<uint16_t*>(cp) = htons(flags); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 227 | cp += INT16SZ; |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 228 | |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 229 | // EDNS0 padding |
| 230 | const uint16_t minlen = static_cast<uint16_t>(cp - buf) + 3 * INT16SZ; |
| 231 | const uint16_t extra = minlen % kEdns0Padding; |
| 232 | uint16_t padlen = (kEdns0Padding - extra) % kEdns0Padding; |
| 233 | if (minlen > buflen) { |
| 234 | return -1; |
| 235 | } |
| 236 | padlen = std::min(padlen, static_cast<uint16_t>(buflen - minlen)); |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 237 | *reinterpret_cast<uint16_t*>(cp) = htons(padlen + 2 * INT16SZ); /* RDLEN */ |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 238 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 239 | *reinterpret_cast<uint16_t*>(cp) = htons(NS_OPT_PADDING); /* OPTION-CODE */ |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 240 | cp += INT16SZ; |
chenbruce | 0d47042 | 2019-03-28 18:44:37 +0800 | [diff] [blame] | 241 | *reinterpret_cast<uint16_t*>(cp) = htons(padlen); /* OPTION-LENGTH */ |
Bernie Innocenti | 136af7b | 2018-10-01 20:46:20 +0900 | [diff] [blame] | 242 | cp += INT16SZ; |
| 243 | memset(cp, 0, padlen); |
| 244 | cp += padlen; |
| 245 | |
| 246 | hp->arcount = htons(ntohs(hp->arcount) + 1); |
Bernie Innocenti | 8ad893f | 2018-08-31 14:09:46 +0900 | [diff] [blame] | 247 | return (cp - buf); |
Bernie Innocenti | 318ed2d | 2018-08-30 04:05:20 +0900 | [diff] [blame] | 248 | } |