1 /* $NetBSD: resolv.h,v 1.31 2005/12/26 19:01:47 perry Exp $ */
2
3 /*
4 * Copyright (c) 1983, 1987, 1989
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. Neither the name of the University nor the names of its contributors
16 * may be used to endorse or promote products derived from this software
17 * without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32 /*
33 * Copyright (c) 2004 by Internet Systems Consortium, Inc. ("ISC")
34 * Portions Copyright (c) 1996-1999 by Internet Software Consortium.
35 *
36 * Permission to use, copy, modify, and distribute this software for any
37 * purpose with or without fee is hereby granted, provided that the above
38 * copyright notice and this permission notice appear in all copies.
39 *
40 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES
41 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
42 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR
43 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
44 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
45 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT
46 * OF OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
47 */
48
49 #pragma once
50
51 #include <android-base/logging.h>
52 #include <android-base/unique_fd.h>
53 #include <net/if.h>
54 #include <time.h>
55 #include <string>
56 #include <vector>
57
58 #include "DnsResolver.h"
59 #include "netd_resolv/resolv.h"
60 #include "params.h"
61 #include "stats.pb.h"
62
63 // Linux defines MAXHOSTNAMELEN as 64, while the domain name limit in
64 // RFC 1034 and RFC 1035 is 255 octets.
65 #ifdef MAXHOSTNAMELEN
66 #undef MAXHOSTNAMELEN
67 #endif
68 #define MAXHOSTNAMELEN 256
69
70 /*
71 * Global defines and variables for resolver stub.
72 */
73 #define RES_TIMEOUT 5000 /* min. milliseconds between retries */
74 #define RES_DFLRETRY 2 /* Default #/tries. */
75
76 // Flags for res_state->_flags
77 #define RES_F_VC 0x00000001 // socket is TCP
78 #define RES_F_EDNS0ERR 0x00000004 // EDNS0 caused errors
79
80 // Holds either a sockaddr_in or a sockaddr_in6.
81 union sockaddr_union {
82 struct sockaddr sa;
83 struct sockaddr_in sin;
84 struct sockaddr_in6 sin6;
85 };
86 constexpr int MAXPACKET = 8 * 1024;
87
88 struct ResState {
closeSocketsResState89 void closeSockets() {
90 tcp_nssock.reset();
91 _flags &= ~RES_F_VC;
92
93 for (auto& sock : nssocks) {
94 sock.reset();
95 }
96 }
97
nameserverCountResState98 int nameserverCount() { return nsaddrs.size(); }
99
100 // clang-format off
101 unsigned netid; // NetId: cache key and socket mark
102 uid_t uid; // uid of the app that sent the DNS lookup
103 pid_t pid; // pid of the app that sent the DNS lookup
104 uint16_t id; // current message id
105 std::vector<std::string> search_domains{}; // domains to search
106 std::vector<android::netdutils::IPSockAddr> nsaddrs;
107 android::base::unique_fd nssocks[MAXNS]; // UDP sockets to nameservers
108 unsigned ndots : 4; // threshold for initial abs. query
109 unsigned _mark; // If non-0 SET_MARK to _mark on all request sockets
110 android::base::unique_fd tcp_nssock; // TCP socket (but why not one per nameserver?)
111 uint32_t _flags = 0; // See RES_F_* defines below
112 android::net::NetworkDnsEventReported* event;
113 uint32_t netcontext_flags;
114 int tc_mode = 0;
115 bool enforce_dns_uid = false;
116 // clang-format on
117 };
118
119 // TODO: remove these legacy aliases
120 typedef ResState* res_state;
121
122 /* End of stats related definitions */
123
124 /*
125 * Error code extending h_errno codes defined in bionic/libc/include/netdb.h.
126 *
127 * This error code, including legacy h_errno, is returned from res_nquery(), res_nsearch(),
128 * res_nquerydomain(), res_queryN(), res_searchN() and res_querydomainN() for DNS metrics.
129 *
130 * TODO: Consider mapping legacy and extended h_errno into a unified resolver error code mapping.
131 */
132 #define NETD_RESOLV_H_ERRNO_EXT_TIMEOUT RCODE_TIMEOUT
133
134 extern const char* const _res_opcodes[];
135
136 int res_nameinquery(const char*, int, int, const uint8_t*, const uint8_t*);
137 int res_queriesmatch(const uint8_t*, const uint8_t*, const uint8_t*, const uint8_t*);
138
139 int res_nquery(res_state, const char*, int, int, uint8_t*, int, int*);
140 int res_nsearch(res_state, const char*, int, int, uint8_t*, int, int*);
141 int res_nquerydomain(res_state, const char*, const char*, int, int, uint8_t*, int, int*);
142 int res_nmkquery(int op, const char* qname, int cl, int type, const uint8_t* data, int datalen,
143 uint8_t* buf, int buflen, int netcontext_flags);
144 int res_nsend(res_state statp, const uint8_t* buf, int buflen, uint8_t* ans, int anssiz, int* rcode,
145 uint32_t flags, std::chrono::milliseconds sleepTimeMs = {});
146 int res_nopt(res_state, int, uint8_t*, int, int);
147
148 int getaddrinfo_numeric(const char* hostname, const char* servname, addrinfo hints,
149 addrinfo** result);
150
151 // Helper function for converting h_errno to the error codes visible to netd
152 int herrnoToAiErrno(int herrno);
153
154 // switch resolver log severity
155 android::base::LogSeverity logSeverityStrToEnum(const std::string& logSeverityStr);
156
157 template <typename Dest>
saturate_cast(int64_t x)158 Dest saturate_cast(int64_t x) {
159 using DestLimits = std::numeric_limits<Dest>;
160 if (x > DestLimits::max()) return DestLimits::max();
161 if (x < DestLimits::min()) return DestLimits::min();
162 return static_cast<Dest>(x);
163 }
164
165 android::net::NsType getQueryType(const uint8_t* msg, size_t msgLen);
166
167 android::net::IpVersion ipFamilyToIPVersion(int ipFamily);
168
resolv_tag_socket(int sock,uid_t uid,pid_t pid)169 inline void resolv_tag_socket(int sock, uid_t uid, pid_t pid) {
170 if (android::net::gResNetdCallbacks.tagSocket != nullptr) {
171 if (int err = android::net::gResNetdCallbacks.tagSocket(sock, TAG_SYSTEM_DNS, uid, pid)) {
172 LOG(WARNING) << "Failed to tag socket: " << strerror(-err);
173 }
174 }
175
176 if (fchown(sock, uid, -1) == -1) {
177 LOG(WARNING) << "Failed to chown socket: " << strerror(errno);
178 }
179 }
180
addrToString(const sockaddr_storage * addr)181 inline std::string addrToString(const sockaddr_storage* addr) {
182 char out[INET6_ADDRSTRLEN] = {0};
183 getnameinfo((const sockaddr*)addr, sizeof(sockaddr_storage), out, INET6_ADDRSTRLEN, nullptr, 0,
184 NI_NUMERICHOST);
185 return std::string(out);
186 }
187