]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/resolve/resolved-etc-hosts.c
tree-wide: use set_ensure_put()
[thirdparty/systemd.git] / src / resolve / resolved-etc-hosts.c
CommitLineData
53e1b683 1/* SPDX-License-Identifier: LGPL-2.1+ */
dd0bc0f1 2
ca78ad1d
ZJS
3#include <sys/stat.h>
4#include <sys/types.h>
5#include <unistd.h>
6
dd0bc0f1
LP
7#include "fd-util.h"
8#include "fileio.h"
9#include "hostname-util.h"
dd0bc0f1 10#include "resolved-dns-synthesize.h"
7c777a77 11#include "resolved-etc-hosts.h"
5c3fa98d 12#include "socket-netlink.h"
dd0bc0f1
LP
13#include "string-util.h"
14#include "strv.h"
15#include "time-util.h"
16
17/* Recheck /etc/hosts at most once every 2s */
18#define ETC_HOSTS_RECHECK_USEC (2*USEC_PER_SEC)
19
a1e92eee 20static void etc_hosts_item_free(EtcHostsItem *item) {
37b7cc8d
ZJS
21 strv_free(item->names);
22 free(item);
23}
dd0bc0f1 24
a1e92eee 25static void etc_hosts_item_by_name_free(EtcHostsItemByName *item) {
37b7cc8d
ZJS
26 free(item->name);
27 free(item->addresses);
28 free(item);
29}
dd0bc0f1 30
78fc21a1 31void etc_hosts_free(EtcHosts *hosts) {
37b7cc8d
ZJS
32 hosts->by_address = hashmap_free_with_destructor(hosts->by_address, etc_hosts_item_free);
33 hosts->by_name = hashmap_free_with_destructor(hosts->by_name, etc_hosts_item_by_name_free);
fd373593 34 hosts->no_address = set_free_free(hosts->no_address);
37b7cc8d 35}
dd0bc0f1 36
37b7cc8d
ZJS
37void manager_etc_hosts_flush(Manager *m) {
38 etc_hosts_free(&m->etc_hosts);
dd0bc0f1 39 m->etc_hosts_mtime = USEC_INFINITY;
aa5408e2
GS
40 m->etc_hosts_ino = 0;
41 m->etc_hosts_dev = 0;
dd0bc0f1
LP
42}
43
37b7cc8d 44static int parse_line(EtcHosts *hosts, unsigned nr, const char *line) {
0351eb11
YW
45 _cleanup_free_ char *address_str = NULL;
46 struct in_addr_data address = {};
fd373593 47 bool found = false;
dd0bc0f1 48 EtcHostsItem *item;
dd0bc0f1
LP
49 int r;
50
37b7cc8d 51 assert(hosts);
0351eb11 52 assert(line);
dd0bc0f1 53
0351eb11
YW
54 r = extract_first_word(&line, &address_str, NULL, EXTRACT_RELAX);
55 if (r < 0)
bd005277
ZJS
56 return log_error_errno(r, "/etc/hosts:%u: failed to extract address: %m", nr);
57 assert(r > 0); /* We already checked that the line is not empty, so it should contain *something* */
0351eb11 58
7bf8c3de 59 r = in_addr_ifindex_from_string_auto(address_str, &address.family, &address.address, NULL);
bd005277
ZJS
60 if (r < 0) {
61 log_warning_errno(r, "/etc/hosts:%u: address '%s' is invalid, ignoring: %m", nr, address_str);
62 return 0;
63 }
0351eb11
YW
64
65 r = in_addr_is_null(address.family, &address.address);
bd005277
ZJS
66 if (r < 0) {
67 log_warning_errno(r, "/etc/hosts:%u: address '%s' is invalid, ignoring: %m", nr, address_str);
68 return 0;
69 }
dd0bc0f1
LP
70 if (r > 0)
71 /* This is an 0.0.0.0 or :: item, which we assume means that we shall map the specified hostname to
72 * nothing. */
73 item = NULL;
74 else {
78fc21a1 75 /* If this is a normal address, then simply add entry mapping it to the specified names */
dd0bc0f1 76
37b7cc8d 77 item = hashmap_get(hosts->by_address, &address);
0351eb11 78 if (!item) {
37b7cc8d 79 r = hashmap_ensure_allocated(&hosts->by_address, &in_addr_data_hash_ops);
dd0bc0f1
LP
80 if (r < 0)
81 return log_oom();
82
0351eb11 83 item = new0(EtcHostsItem, 1);
dd0bc0f1
LP
84 if (!item)
85 return log_oom();
86
0351eb11 87 item->address = address;
dd0bc0f1 88
37b7cc8d 89 r = hashmap_put(hosts->by_address, &item->address, item);
dd0bc0f1
LP
90 if (r < 0) {
91 free(item);
92 return log_oom();
93 }
94 }
95 }
96
0351eb11
YW
97 for (;;) {
98 _cleanup_free_ char *name = NULL;
dd0bc0f1
LP
99 EtcHostsItemByName *bn;
100
0351eb11
YW
101 r = extract_first_word(&line, &name, NULL, EXTRACT_RELAX);
102 if (r < 0)
38b38500 103 return log_error_errno(r, "/etc/hosts:%u: couldn't extract hostname: %m", nr);
0351eb11
YW
104 if (r == 0)
105 break;
106
fd373593
ZJS
107 found = true;
108
7470cc4c 109 r = dns_name_is_valid_ldh(name);
bd005277
ZJS
110 if (r <= 0) {
111 log_warning_errno(r, "/etc/hosts:%u: hostname \"%s\" is not valid, ignoring.", nr, name);
112 continue;
113 }
114
fd373593 115 if (is_localhost(name))
0351eb11 116 /* Suppress the "localhost" line that is often seen */
0351eb11 117 continue;
0351eb11 118
fd373593
ZJS
119 if (!item) {
120 /* Optimize the case where we don't need to store any addresses, by storing
121 * only the name in a dedicated Set instead of the hashmap */
122
de7fef4b 123 r = set_ensure_put(&hosts->no_address, &dns_name_hash_ops, name);
fd373593
ZJS
124 if (r < 0)
125 return r;
126
127 TAKE_PTR(name);
128 continue;
0351eb11
YW
129 }
130
fd373593
ZJS
131 r = strv_extend(&item->names, name);
132 if (r < 0)
133 return log_oom();
134
37b7cc8d 135 bn = hashmap_get(hosts->by_name, name);
dd0bc0f1 136 if (!bn) {
37b7cc8d 137 r = hashmap_ensure_allocated(&hosts->by_name, &dns_name_hash_ops);
dd0bc0f1
LP
138 if (r < 0)
139 return log_oom();
140
141 bn = new0(EtcHostsItemByName, 1);
142 if (!bn)
143 return log_oom();
144
37b7cc8d 145 r = hashmap_put(hosts->by_name, name, bn);
dd0bc0f1 146 if (r < 0) {
dd0bc0f1
LP
147 free(bn);
148 return log_oom();
149 }
0351eb11
YW
150
151 bn->name = TAKE_PTR(name);
dd0bc0f1
LP
152 }
153
fd373593
ZJS
154 if (!GREEDY_REALLOC(bn->addresses, bn->n_allocated, bn->n_addresses + 1))
155 return log_oom();
dd0bc0f1 156
fd373593 157 bn->addresses[bn->n_addresses++] = &item->address;
dd0bc0f1
LP
158 }
159
baaa35ad 160 if (!found)
38b38500 161 log_warning("/etc/hosts:%u: line is missing any hostnames", nr);
dd0bc0f1 162
0351eb11 163 return 0;
dd0bc0f1
LP
164}
165
78fc21a1 166int etc_hosts_parse(EtcHosts *hosts, FILE *f) {
37b7cc8d 167 _cleanup_(etc_hosts_free) EtcHosts t = {};
37b7cc8d
ZJS
168 unsigned nr = 0;
169 int r;
170
7c777a77
LP
171 for (;;) {
172 _cleanup_free_ char *line = NULL;
37b7cc8d
ZJS
173 char *l;
174
7c777a77
LP
175 r = read_line(f, LONG_LINE_MAX, &line);
176 if (r < 0)
177 return log_error_errno(r, "Failed to read /etc/hosts: %m");
178 if (r == 0)
179 break;
180
37b7cc8d
ZJS
181 nr++;
182
bd005277
ZJS
183 l = strchr(line, '#');
184 if (l)
185 *l = '\0';
186
37b7cc8d
ZJS
187 l = strstrip(line);
188 if (isempty(l))
189 continue;
37b7cc8d
ZJS
190
191 r = parse_line(&t, nr, l);
192 if (r < 0)
193 return r;
194 }
195
be19cdf3 196 etc_hosts_free(hosts);
37b7cc8d
ZJS
197 *hosts = t;
198 t = (EtcHosts) {}; /* prevent cleanup */
199 return 0;
200}
201
452ca091 202static int manager_etc_hosts_read(Manager *m) {
dd0bc0f1 203 _cleanup_fclose_ FILE *f = NULL;
dd0bc0f1
LP
204 struct stat st;
205 usec_t ts;
dd0bc0f1
LP
206 int r;
207
208 assert_se(sd_event_now(m->event, clock_boottime_or_monotonic(), &ts) >= 0);
209
210 /* See if we checked /etc/hosts recently already */
211 if (m->etc_hosts_last != USEC_INFINITY && m->etc_hosts_last + ETC_HOSTS_RECHECK_USEC > ts)
212 return 0;
213
214 m->etc_hosts_last = ts;
215
216 if (m->etc_hosts_mtime != USEC_INFINITY) {
217 if (stat("/etc/hosts", &st) < 0) {
37b7cc8d
ZJS
218 if (errno != ENOENT)
219 return log_error_errno(errno, "Failed to stat /etc/hosts: %m");
dd0bc0f1 220
37b7cc8d
ZJS
221 manager_etc_hosts_flush(m);
222 return 0;
dd0bc0f1
LP
223 }
224
aa5408e2
GS
225 /* Did the mtime or ino/dev change? If not, there's no point in re-reading the file. */
226 if (timespec_load(&st.st_mtim) == m->etc_hosts_mtime &&
227 st.st_ino == m->etc_hosts_ino && st.st_dev == m->etc_hosts_dev)
dd0bc0f1
LP
228 return 0;
229 }
230
231 f = fopen("/etc/hosts", "re");
232 if (!f) {
37b7cc8d
ZJS
233 if (errno != ENOENT)
234 return log_error_errno(errno, "Failed to open /etc/hosts: %m");
dd0bc0f1 235
37b7cc8d
ZJS
236 manager_etc_hosts_flush(m);
237 return 0;
dd0bc0f1
LP
238 }
239
240 /* Take the timestamp at the beginning of processing, so that any changes made later are read on the next
241 * invocation */
242 r = fstat(fileno(f), &st);
243 if (r < 0)
244 return log_error_errno(errno, "Failed to fstat() /etc/hosts: %m");
245
37b7cc8d 246 r = etc_hosts_parse(&m->etc_hosts, f);
be19cdf3 247 if (r < 0)
37b7cc8d 248 return r;
dd0bc0f1
LP
249
250 m->etc_hosts_mtime = timespec_load(&st.st_mtim);
aa5408e2
GS
251 m->etc_hosts_ino = st.st_ino;
252 m->etc_hosts_dev = st.st_dev;
dd0bc0f1
LP
253 m->etc_hosts_last = ts;
254
255 return 1;
dd0bc0f1
LP
256}
257
258int manager_etc_hosts_lookup(Manager *m, DnsQuestion* q, DnsAnswer **answer) {
259 bool found_a = false, found_aaaa = false;
0351eb11 260 struct in_addr_data k = {};
dd0bc0f1 261 EtcHostsItemByName *bn;
dd0bc0f1
LP
262 DnsResourceKey *t;
263 const char *name;
264 unsigned i;
265 int r;
266
267 assert(m);
268 assert(q);
269 assert(answer);
270
86317087
YW
271 if (!m->read_etc_hosts)
272 return 0;
273
be19cdf3 274 (void) manager_etc_hosts_read(m);
dd0bc0f1
LP
275
276 name = dns_question_first_name(q);
277 if (!name)
278 return 0;
279
280 r = dns_name_address(name, &k.family, &k.address);
281 if (r > 0) {
282 EtcHostsItem *item;
283 DnsResourceKey *found_ptr = NULL;
284
37b7cc8d 285 item = hashmap_get(m->etc_hosts.by_address, &k);
dd0bc0f1
LP
286 if (!item)
287 return 0;
288
289 /* We have an address in /etc/hosts that matches the queried name. Let's return successful. Actual data
290 * we'll only return if the request was for PTR. */
291
292 DNS_QUESTION_FOREACH(t, q) {
293 if (!IN_SET(t->type, DNS_TYPE_PTR, DNS_TYPE_ANY))
294 continue;
295 if (!IN_SET(t->class, DNS_CLASS_IN, DNS_CLASS_ANY))
296 continue;
297
1c02e7ba 298 r = dns_name_equal(dns_resource_key_name(t), name);
dd0bc0f1
LP
299 if (r < 0)
300 return r;
301 if (r > 0) {
302 found_ptr = t;
303 break;
304 }
305 }
306
307 if (found_ptr) {
308 char **n;
309
310 r = dns_answer_reserve(answer, strv_length(item->names));
311 if (r < 0)
312 return r;
313
314 STRV_FOREACH(n, item->names) {
315 _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL;
316
317 rr = dns_resource_record_new(found_ptr);
318 if (!rr)
319 return -ENOMEM;
320
321 rr->ptr.name = strdup(*n);
322 if (!rr->ptr.name)
323 return -ENOMEM;
324
325 r = dns_answer_add(*answer, rr, 0, DNS_ANSWER_AUTHENTICATED);
326 if (r < 0)
327 return r;
328 }
329 }
330
331 return 1;
332 }
333
37b7cc8d 334 bn = hashmap_get(m->etc_hosts.by_name, name);
fd373593
ZJS
335 if (bn) {
336 r = dns_answer_reserve(answer, bn->n_addresses);
337 if (r < 0)
338 return r;
339 } else {
340 /* Check if name was listed with no address. If yes, continue to return an answer. */
341 if (!set_contains(m->etc_hosts.no_address, name))
342 return 0;
343 }
dd0bc0f1
LP
344
345 DNS_QUESTION_FOREACH(t, q) {
346 if (!IN_SET(t->type, DNS_TYPE_A, DNS_TYPE_AAAA, DNS_TYPE_ANY))
347 continue;
348 if (!IN_SET(t->class, DNS_CLASS_IN, DNS_CLASS_ANY))
349 continue;
350
1c02e7ba 351 r = dns_name_equal(dns_resource_key_name(t), name);
dd0bc0f1
LP
352 if (r < 0)
353 return r;
354 if (r == 0)
355 continue;
356
357 if (IN_SET(t->type, DNS_TYPE_A, DNS_TYPE_ANY))
358 found_a = true;
359 if (IN_SET(t->type, DNS_TYPE_AAAA, DNS_TYPE_ANY))
360 found_aaaa = true;
361
362 if (found_a && found_aaaa)
363 break;
364 }
365
fd373593 366 for (i = 0; bn && i < bn->n_addresses; i++) {
dd0bc0f1
LP
367 _cleanup_(dns_resource_record_unrefp) DnsResourceRecord *rr = NULL;
368
0351eb11
YW
369 if ((!found_a && bn->addresses[i]->family == AF_INET) ||
370 (!found_aaaa && bn->addresses[i]->family == AF_INET6))
dd0bc0f1
LP
371 continue;
372
0351eb11 373 r = dns_resource_record_new_address(&rr, bn->addresses[i]->family, &bn->addresses[i]->address, bn->name);
dd0bc0f1
LP
374 if (r < 0)
375 return r;
376
377 r = dns_answer_add(*answer, rr, 0, DNS_ANSWER_AUTHENTICATED);
378 if (r < 0)
379 return r;
380 }
381
4050e04b 382 return found_a || found_aaaa;
dd0bc0f1 383}