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