]>
Commit | Line | Data |
---|---|---|
db9ecf05 | 1 | /* SPDX-License-Identifier: LGPL-2.1-or-later */ |
74b2466e LP |
2 | #pragma once |
3 | ||
a0166609 | 4 | #include <netinet/ip.h> |
4565863f | 5 | #include <netinet/ip6.h> |
71d35b6b | 6 | #include <netinet/udp.h> |
74b2466e | 7 | |
74b2466e | 8 | #include "hashmap.h" |
7e8e0422 | 9 | #include "in-addr-util.h" |
71d35b6b TA |
10 | #include "macro.h" |
11 | #include "sparse-endian.h" | |
7e8e0422 LP |
12 | |
13 | typedef struct DnsPacketHeader DnsPacketHeader; | |
14 | typedef struct DnsPacket DnsPacket; | |
15 | ||
51323288 | 16 | #include "resolved-def.h" |
71d35b6b TA |
17 | #include "resolved-dns-answer.h" |
18 | #include "resolved-dns-question.h" | |
19 | #include "resolved-dns-rr.h" | |
74b2466e | 20 | |
1716f6dc LP |
21 | typedef enum DnsProtocol { |
22 | DNS_PROTOCOL_DNS, | |
23 | DNS_PROTOCOL_MDNS, | |
24 | DNS_PROTOCOL_LLMNR, | |
25 | _DNS_PROTOCOL_MAX, | |
2d93c20e | 26 | _DNS_PROTOCOL_INVALID = -EINVAL, |
1716f6dc LP |
27 | } DnsProtocol; |
28 | ||
74b2466e LP |
29 | struct DnsPacketHeader { |
30 | uint16_t id; | |
31 | be16_t flags; | |
32 | be16_t qdcount; | |
33 | be16_t ancount; | |
34 | be16_t nscount; | |
35 | be16_t arcount; | |
4565863f | 36 | } _packed_; |
74b2466e LP |
37 | |
38 | #define DNS_PACKET_HEADER_SIZE sizeof(DnsPacketHeader) | |
4565863f LP |
39 | #define UDP4_PACKET_HEADER_SIZE (sizeof(struct iphdr) + sizeof(struct udphdr)) |
40 | #define UDP6_PACKET_HEADER_SIZE (sizeof(struct ip6_hdr) + sizeof(struct udphdr)) | |
41 | ||
42 | assert_cc(sizeof(struct ip6_hdr) == 40); | |
43 | assert_cc(sizeof(struct iphdr) == 20); | |
44 | assert_cc(sizeof(struct udphdr) == 8); | |
45 | assert_cc(sizeof(DnsPacketHeader) == 12); | |
c73ce96b | 46 | |
4565863f LP |
47 | /* The various DNS protocols deviate in how large a packet can grow, but the TCP transport has a 16bit size |
48 | * field, hence that appears to be the absolute maximum. */ | |
64a21fda | 49 | #define DNS_PACKET_SIZE_MAX 0xFFFFu |
e1c95994 | 50 | |
86d06089 ZJS |
51 | /* The default size to use for allocation when we don't know how large |
52 | * the packet will turn out to be. */ | |
53 | #define DNS_PACKET_SIZE_START 512u | |
54 | ||
e1c95994 | 55 | /* RFC 1035 say 512 is the maximum, for classic unicast DNS */ |
64a21fda | 56 | #define DNS_PACKET_UNICAST_SIZE_MAX 512u |
e1c95994 | 57 | |
d74fb368 | 58 | /* With EDNS0 we can use larger packets, default to 4096, which is what is commonly used */ |
64a21fda | 59 | #define DNS_PACKET_UNICAST_SIZE_LARGE_MAX 4096u |
d74fb368 | 60 | |
74b2466e | 61 | struct DnsPacket { |
cf4b2f99 | 62 | unsigned n_ref; |
1716f6dc | 63 | DnsProtocol protocol; |
20a001bd | 64 | size_t size, allocated, rindex, max_size, fragsize; |
faa133f3 | 65 | void *_data; /* don't access directly, use DNS_PACKET_DATA()! */ |
74b2466e | 66 | Hashmap *names; /* For name compression */ |
519ef046 | 67 | size_t opt_start, opt_size; |
faa133f3 LP |
68 | |
69 | /* Parsed data */ | |
70 | DnsQuestion *question; | |
71 | DnsAnswer *answer; | |
d75acfb0 | 72 | DnsResourceRecord *opt; |
1716f6dc | 73 | |
dc61b7e4 | 74 | /* Packet reception metadata */ |
1716f6dc | 75 | int ifindex; |
623a4c97 | 76 | int family, ipproto; |
1716f6dc | 77 | union in_addr_union sender, destination; |
623a4c97 | 78 | uint16_t sender_port, destination_port; |
0dd25fb9 | 79 | uint32_t ttl; |
5777c613 | 80 | usec_t timestamp; /* CLOCK_BOOTTIME (or CLOCK_MONOTONIC if the former doesn't exist) */ |
a4076574 | 81 | |
9c491563 DM |
82 | /* For support of truncated packets */ |
83 | DnsPacket *more; | |
84 | ||
a8812dd7 LP |
85 | bool on_stack:1; |
86 | bool extracted:1; | |
87 | bool refuse_compression:1; | |
88 | bool canonical_form:1; | |
74b2466e LP |
89 | }; |
90 | ||
7130db91 | 91 | static inline uint8_t* DNS_PACKET_DATA(const DnsPacket *p) { |
74b2466e LP |
92 | if (_unlikely_(!p)) |
93 | return NULL; | |
94 | ||
faa133f3 LP |
95 | if (p->_data) |
96 | return p->_data; | |
74b2466e LP |
97 | |
98 | return ((uint8_t*) p) + ALIGN(sizeof(DnsPacket)); | |
99 | } | |
100 | ||
101 | #define DNS_PACKET_HEADER(p) ((DnsPacketHeader*) DNS_PACKET_DATA(p)) | |
3cb10d3a LP |
102 | #define DNS_PACKET_ID(p) DNS_PACKET_HEADER(p)->id |
103 | #define DNS_PACKET_QR(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 15) & 1) | |
104 | #define DNS_PACKET_OPCODE(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 11) & 15) | |
8b757a38 | 105 | #define DNS_PACKET_AA(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 10) & 1) |
ad867662 | 106 | #define DNS_PACKET_TC(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 9) & 1) |
8b757a38 DM |
107 | #define DNS_PACKET_RD(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 8) & 1) |
108 | #define DNS_PACKET_RA(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 7) & 1) | |
109 | #define DNS_PACKET_AD(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 5) & 1) | |
110 | #define DNS_PACKET_CD(p) ((be16toh(DNS_PACKET_HEADER(p)->flags) >> 4) & 1) | |
f3abbe25 | 111 | |
b30bf55d LP |
112 | #define DNS_PACKET_FLAG_TC (UINT16_C(1) << 9) |
113 | ||
f3abbe25 LP |
114 | static inline uint16_t DNS_PACKET_RCODE(DnsPacket *p) { |
115 | uint16_t rcode; | |
116 | ||
117 | if (p->opt) | |
118 | rcode = (uint16_t) (p->opt->ttl >> 24); | |
119 | else | |
120 | rcode = 0; | |
121 | ||
b30bf55d LP |
122 | return rcode | (be16toh(DNS_PACKET_HEADER(p)->flags) & 0xF); |
123 | } | |
124 | ||
125 | static inline uint16_t DNS_PACKET_PAYLOAD_SIZE_MAX(DnsPacket *p) { | |
126 | ||
e6eed944 | 127 | /* Returns the advertised maximum size for replies, or the DNS default if there's nothing defined. */ |
b30bf55d | 128 | |
acd74d61 LP |
129 | if (p->ipproto == IPPROTO_TCP) /* we ignore EDNS(0) size data on TCP, like everybody else */ |
130 | return DNS_PACKET_SIZE_MAX; | |
131 | ||
b30bf55d LP |
132 | if (p->opt) |
133 | return MAX(DNS_PACKET_UNICAST_SIZE_MAX, p->opt->key->class); | |
134 | ||
135 | return DNS_PACKET_UNICAST_SIZE_MAX; | |
136 | } | |
137 | ||
138 | static inline bool DNS_PACKET_DO(DnsPacket *p) { | |
139 | if (!p->opt) | |
140 | return false; | |
141 | ||
142 | return !!(p->opt->ttl & (1U << 15)); | |
143 | } | |
144 | ||
145 | static inline bool DNS_PACKET_VERSION_SUPPORTED(DnsPacket *p) { | |
146 | /* Returns true if this packet is in a version we support. Which means either non-EDNS or EDNS(0), but not EDNS | |
147 | * of any newer versions */ | |
148 | ||
149 | if (!p->opt) | |
150 | return true; | |
151 | ||
152 | return DNS_RESOURCE_RECORD_OPT_VERSION_SUPPORTED(p->opt); | |
f3abbe25 | 153 | } |
8b757a38 DM |
154 | |
155 | /* LLMNR defines some bits differently */ | |
156 | #define DNS_PACKET_LLMNR_C(p) DNS_PACKET_AA(p) | |
157 | #define DNS_PACKET_LLMNR_T(p) DNS_PACKET_RD(p) | |
158 | ||
3cb10d3a LP |
159 | #define DNS_PACKET_QDCOUNT(p) be16toh(DNS_PACKET_HEADER(p)->qdcount) |
160 | #define DNS_PACKET_ANCOUNT(p) be16toh(DNS_PACKET_HEADER(p)->ancount) | |
161 | #define DNS_PACKET_NSCOUNT(p) be16toh(DNS_PACKET_HEADER(p)->nscount) | |
162 | #define DNS_PACKET_ARCOUNT(p) be16toh(DNS_PACKET_HEADER(p)->arcount) | |
163 | ||
164 | #define DNS_PACKET_MAKE_FLAGS(qr, opcode, aa, tc, rd, ra, ad, cd, rcode) \ | |
f0258e47 LP |
165 | (((uint16_t) !!(qr) << 15) | \ |
166 | ((uint16_t) ((opcode) & 15) << 11) | \ | |
167 | ((uint16_t) !!(aa) << 10) | /* on LLMNR: c */ \ | |
168 | ((uint16_t) !!(tc) << 9) | \ | |
169 | ((uint16_t) !!(rd) << 8) | /* on LLMNR: t */ \ | |
170 | ((uint16_t) !!(ra) << 7) | \ | |
171 | ((uint16_t) !!(ad) << 5) | \ | |
172 | ((uint16_t) !!(cd) << 4) | \ | |
173 | ((uint16_t) ((rcode) & 15))) | |
3cb10d3a | 174 | |
322345fd LP |
175 | static inline unsigned DNS_PACKET_RRCOUNT(DnsPacket *p) { |
176 | return | |
177 | (unsigned) DNS_PACKET_ANCOUNT(p) + | |
178 | (unsigned) DNS_PACKET_NSCOUNT(p) + | |
179 | (unsigned) DNS_PACKET_ARCOUNT(p); | |
180 | } | |
74b2466e | 181 | |
51027656 | 182 | int dns_packet_new(DnsPacket **p, DnsProtocol protocol, size_t min_alloc_dsize, size_t max_size); |
46a58596 | 183 | int dns_packet_new_query(DnsPacket **p, DnsProtocol protocol, size_t min_alloc_dsize, bool dnssec_checking_disabled); |
74b2466e | 184 | |
1a6cd020 LP |
185 | int dns_packet_dup(DnsPacket **ret, DnsPacket *p); |
186 | ||
dbfbb6e7 DM |
187 | void dns_packet_set_flags(DnsPacket *p, bool dnssec_checking_disabled, bool truncated); |
188 | ||
74b2466e LP |
189 | DnsPacket *dns_packet_ref(DnsPacket *p); |
190 | DnsPacket *dns_packet_unref(DnsPacket *p); | |
191 | ||
192 | DEFINE_TRIVIAL_CLEANUP_FUNC(DnsPacket*, dns_packet_unref); | |
193 | ||
194 | int dns_packet_validate(DnsPacket *p); | |
195 | int dns_packet_validate_reply(DnsPacket *p); | |
623a4c97 | 196 | int dns_packet_validate_query(DnsPacket *p); |
74b2466e | 197 | |
8af5b883 LP |
198 | int dns_packet_is_reply_for(DnsPacket *p, const DnsResourceKey *key); |
199 | ||
623a4c97 | 200 | int dns_packet_append_blob(DnsPacket *p, const void *d, size_t sz, size_t *start); |
74b2466e LP |
201 | int dns_packet_append_uint8(DnsPacket *p, uint8_t v, size_t *start); |
202 | int dns_packet_append_uint16(DnsPacket *p, uint16_t v, size_t *start); | |
623a4c97 | 203 | int dns_packet_append_uint32(DnsPacket *p, uint32_t v, size_t *start); |
74b2466e | 204 | int dns_packet_append_string(DnsPacket *p, const char *s, size_t *start); |
2001c805 | 205 | int dns_packet_append_raw_string(DnsPacket *p, const void *s, size_t size, size_t *start); |
a3db237b LP |
206 | int dns_packet_append_label(DnsPacket *p, const char *s, size_t l, bool canonical_candidate, size_t *start); |
207 | int dns_packet_append_name(DnsPacket *p, const char *name, bool allow_compression, bool canonical_candidate, size_t *start); | |
58ab31d5 DR |
208 | int dns_packet_append_key(DnsPacket *p, const DnsResourceKey *key, const DnsAnswerFlags flags, size_t *start); |
209 | int dns_packet_append_rr(DnsPacket *p, const DnsResourceRecord *rr, const DnsAnswerFlags flags, size_t *start, size_t *rdata_start); | |
4a6eb824 | 210 | int dns_packet_append_opt(DnsPacket *p, uint16_t max_udp_size, bool edns0_do, bool include_rfc6975, const char *nsid, int rcode, size_t *ret_start); |
f471bc11 | 211 | int dns_packet_append_question(DnsPacket *p, DnsQuestion *q); |
6f76e68a | 212 | int dns_packet_append_answer(DnsPacket *p, DnsAnswer *a, unsigned *completed); |
74b2466e | 213 | |
93748b26 | 214 | int dns_packet_patch_max_udp_size(DnsPacket *p, uint16_t max_udp_size); |
81b4d94d | 215 | int dns_packet_patch_ttls(DnsPacket *p, usec_t timestamp); |
74b2466e | 216 | |
9c5e12a4 | 217 | void dns_packet_truncate(DnsPacket *p, size_t sz); |
519ef046 | 218 | int dns_packet_truncate_opt(DnsPacket *p); |
9c5e12a4 | 219 | |
74b2466e | 220 | int dns_packet_read(DnsPacket *p, size_t sz, const void **ret, size_t *start); |
623a4c97 | 221 | int dns_packet_read_blob(DnsPacket *p, void *d, size_t sz, size_t *start); |
74b2466e LP |
222 | int dns_packet_read_uint8(DnsPacket *p, uint8_t *ret, size_t *start); |
223 | int dns_packet_read_uint16(DnsPacket *p, uint16_t *ret, size_t *start); | |
224 | int dns_packet_read_uint32(DnsPacket *p, uint32_t *ret, size_t *start); | |
225 | int dns_packet_read_string(DnsPacket *p, char **ret, size_t *start); | |
2001c805 LP |
226 | int dns_packet_read_raw_string(DnsPacket *p, const void **ret, size_t *size, size_t *start); |
227 | int dns_packet_read_name(DnsPacket *p, char **ret, bool allow_compression, size_t *start); | |
d2579eec LP |
228 | int dns_packet_read_key(DnsPacket *p, DnsResourceKey **ret, bool *ret_cache_flush, size_t *start); |
229 | int dns_packet_read_rr(DnsPacket *p, DnsResourceRecord **ret, bool *ret_cache_flush, size_t *start); | |
74b2466e | 230 | |
8ba9fd9c LP |
231 | void dns_packet_rewind(DnsPacket *p, size_t idx); |
232 | ||
74b2466e | 233 | int dns_packet_skip_question(DnsPacket *p); |
faa133f3 | 234 | int dns_packet_extract(DnsPacket *p); |
74b2466e | 235 | |
a9fd8837 LP |
236 | bool dns_packet_equal(const DnsPacket *a, const DnsPacket *b); |
237 | ||
4a6eb824 LP |
238 | int dns_packet_has_nsid_request(DnsPacket *p); |
239 | ||
f3abbe25 | 240 | /* https://www.iana.org/assignments/dns-parameters/dns-parameters.xhtml#dns-parameters-6 */ |
74b2466e LP |
241 | enum { |
242 | DNS_RCODE_SUCCESS = 0, | |
243 | DNS_RCODE_FORMERR = 1, | |
244 | DNS_RCODE_SERVFAIL = 2, | |
245 | DNS_RCODE_NXDOMAIN = 3, | |
246 | DNS_RCODE_NOTIMP = 4, | |
247 | DNS_RCODE_REFUSED = 5, | |
248 | DNS_RCODE_YXDOMAIN = 6, | |
249 | DNS_RCODE_YXRRSET = 7, | |
250 | DNS_RCODE_NXRRSET = 8, | |
251 | DNS_RCODE_NOTAUTH = 9, | |
252 | DNS_RCODE_NOTZONE = 10, | |
253 | DNS_RCODE_BADVERS = 16, | |
254 | DNS_RCODE_BADSIG = 16, /* duplicate value! */ | |
255 | DNS_RCODE_BADKEY = 17, | |
256 | DNS_RCODE_BADTIME = 18, | |
257 | DNS_RCODE_BADMODE = 19, | |
258 | DNS_RCODE_BADNAME = 20, | |
259 | DNS_RCODE_BADALG = 21, | |
260 | DNS_RCODE_BADTRUNC = 22, | |
6f21e066 | 261 | DNS_RCODE_BADCOOKIE = 23, |
f2ed4c69 LP |
262 | _DNS_RCODE_MAX_DEFINED, |
263 | _DNS_RCODE_MAX = 4095 /* 4 bit rcode in the header plus 8 bit rcode in OPT, makes 12 bit */ | |
74b2466e LP |
264 | }; |
265 | ||
266 | const char* dns_rcode_to_string(int i) _const_; | |
267 | int dns_rcode_from_string(const char *s) _pure_; | |
1716f6dc LP |
268 | |
269 | const char* dns_protocol_to_string(DnsProtocol p) _const_; | |
270 | DnsProtocol dns_protocol_from_string(const char *s) _pure_; | |
271 | ||
272 | #define LLMNR_MULTICAST_IPV4_ADDRESS ((struct in_addr) { .s_addr = htobe32(224U << 24 | 252U) }) | |
273 | #define LLMNR_MULTICAST_IPV6_ADDRESS ((struct in6_addr) { .s6_addr = { 0xFF, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x01, 0x00, 0x03 } }) | |
8db0d2f5 | 274 | |
0db4c90a | 275 | #define MDNS_MULTICAST_IPV4_ADDRESS ((struct in_addr) { .s_addr = htobe32(224U << 24 | 251U) }) |
b6c5d46b | 276 | #define MDNS_MULTICAST_IPV6_ADDRESS ((struct in6_addr) { .s6_addr = { 0xFF, 0x02, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0xfb } }) |
0db4c90a | 277 | |
98767d75 IT |
278 | extern const struct hash_ops dns_packet_hash_ops; |
279 | ||
43fc4baa LP |
280 | static inline uint64_t SD_RESOLVED_FLAGS_MAKE( |
281 | DnsProtocol protocol, | |
282 | int family, | |
283 | bool authenticated, | |
284 | bool confidential) { | |
931851e8 | 285 | uint64_t f; |
51323288 | 286 | |
931851e8 LP |
287 | /* Converts a protocol + family into a flags field as used in queries and responses */ |
288 | ||
43fc4baa LP |
289 | f = (authenticated ? SD_RESOLVED_AUTHENTICATED : 0) | |
290 | (confidential ? SD_RESOLVED_CONFIDENTIAL : 0); | |
51323288 | 291 | |
106784eb DM |
292 | switch (protocol) { |
293 | case DNS_PROTOCOL_DNS: | |
931851e8 | 294 | return f|SD_RESOLVED_DNS; |
51323288 | 295 | |
106784eb | 296 | case DNS_PROTOCOL_LLMNR: |
931851e8 | 297 | return f|(family == AF_INET6 ? SD_RESOLVED_LLMNR_IPV6 : SD_RESOLVED_LLMNR_IPV4); |
51323288 | 298 | |
4e5bf5e1 | 299 | case DNS_PROTOCOL_MDNS: |
12e1893a | 300 | return f|(family == AF_INET6 ? SD_RESOLVED_MDNS_IPV6 : SD_RESOLVED_MDNS_IPV4); |
4e5bf5e1 | 301 | |
106784eb | 302 | default: |
12e1893a | 303 | return f; |
106784eb | 304 | } |
51323288 | 305 | } |
51027656 LP |
306 | |
307 | static inline size_t dns_packet_size_max(DnsPacket *p) { | |
308 | assert(p); | |
309 | ||
310 | /* Why not insist on a fully initialized max_size during DnsPacket construction? Well, this way it's easy to | |
311 | * allocate a transient, throw-away DnsPacket on the stack by simple zero initialization, without having to | |
312 | * deal with explicit field initialization. */ | |
313 | ||
314 | return p->max_size != 0 ? p->max_size : DNS_PACKET_SIZE_MAX; | |
315 | } | |
4565863f LP |
316 | |
317 | static inline size_t udp_header_size(int af) { | |
318 | ||
319 | switch (af) { | |
320 | case AF_INET: | |
321 | return UDP4_PACKET_HEADER_SIZE; | |
322 | case AF_INET6: | |
323 | return UDP6_PACKET_HEADER_SIZE; | |
324 | default: | |
325 | assert_not_reached("Unexpected address family"); | |
326 | } | |
327 | } |