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