1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4 This file is part of systemd.
6 Copyright 2014 Lennart Poettering
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
24 #include "resolved-dns-zone.h"
25 #include "resolved-dns-domain.h"
26 #include "resolved-dns-packet.h"
28 /* Never allow more than 1K entries */
31 typedef struct DnsZoneItem DnsZoneItem
;
34 DnsResourceRecord
*rr
;
36 LIST_FIELDS(DnsZoneItem
, by_key
);
37 LIST_FIELDS(DnsZoneItem
, by_name
);
40 static void dns_zone_item_free(DnsZoneItem
*i
) {
44 dns_resource_record_unref(i
->rr
);
48 DEFINE_TRIVIAL_CLEANUP_FUNC(DnsZoneItem
*, dns_zone_item_free
);
50 static void dns_zone_item_remove_and_free(DnsZone
*z
, DnsZoneItem
*i
) {
58 first
= hashmap_get(z
->by_key
, i
->rr
->key
);
59 LIST_REMOVE(by_key
, first
, i
);
61 assert_se(hashmap_replace(z
->by_key
, first
->rr
->key
, first
) >= 0);
63 hashmap_remove(z
->by_key
, i
->rr
->key
);
65 first
= hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(i
->rr
->key
));
66 LIST_REMOVE(by_name
, first
, i
);
68 assert_se(hashmap_replace(z
->by_name
, DNS_RESOURCE_KEY_NAME(first
->rr
->key
), first
) >= 0);
70 hashmap_remove(z
->by_name
, DNS_RESOURCE_KEY_NAME(i
->rr
->key
));
72 dns_zone_item_free(i
);
75 void dns_zone_flush(DnsZone
*z
) {
80 while ((i
= hashmap_first(z
->by_key
)))
81 dns_zone_item_remove_and_free(z
, i
);
83 assert(hashmap_size(z
->by_key
) == 0);
84 assert(hashmap_size(z
->by_name
) == 0);
86 hashmap_free(z
->by_key
);
89 hashmap_free(z
->by_name
);
93 static DnsZoneItem
* dns_zone_get(DnsZone
*z
, DnsResourceRecord
*rr
) {
99 LIST_FOREACH(by_key
, i
, hashmap_get(z
->by_key
, rr
->key
))
100 if (dns_resource_record_equal(i
->rr
, rr
))
106 void dns_zone_remove_rr(DnsZone
*z
, DnsResourceRecord
*rr
) {
112 i
= dns_zone_get(z
, rr
);
114 dns_zone_item_remove_and_free(z
, i
);
117 static int dns_zone_init(DnsZone
*z
) {
122 r
= hashmap_ensure_allocated(&z
->by_key
, dns_resource_key_hash_func
, dns_resource_key_compare_func
);
126 r
= hashmap_ensure_allocated(&z
->by_name
, dns_name_hash_func
, dns_name_compare_func
);
133 static int dns_zone_link_item(DnsZone
*z
, DnsZoneItem
*i
) {
137 first
= hashmap_get(z
->by_key
, i
->rr
->key
);
139 LIST_PREPEND(by_key
, first
, i
);
140 assert_se(hashmap_replace(z
->by_key
, first
->rr
->key
, first
) >= 0);
142 r
= hashmap_put(z
->by_key
, i
->rr
->key
, i
);
147 first
= hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(i
->rr
->key
));
149 LIST_PREPEND(by_name
, first
, i
);
150 assert_se(hashmap_replace(z
->by_name
, DNS_RESOURCE_KEY_NAME(first
->rr
->key
), first
) >= 0);
152 r
= hashmap_put(z
->by_name
, DNS_RESOURCE_KEY_NAME(i
->rr
->key
), i
);
160 int dns_zone_put(DnsZone
*z
, DnsResourceRecord
*rr
) {
161 _cleanup_(dns_zone_item_freep
) DnsZoneItem
*i
= NULL
;
162 DnsZoneItem
*existing
;
168 if (rr
->key
->class == DNS_CLASS_ANY
)
170 if (rr
->key
->type
== DNS_TYPE_ANY
)
173 existing
= dns_zone_get(z
, rr
);
177 r
= dns_zone_init(z
);
181 i
= new0(DnsZoneItem
, 1);
185 i
->rr
= dns_resource_record_ref(rr
);
187 r
= dns_zone_link_item(z
, i
);
195 int dns_zone_lookup(DnsZone
*z
, DnsQuestion
*q
, DnsAnswer
**ret_answer
, DnsAnswer
**ret_soa
) {
196 _cleanup_(dns_answer_unrefp
) DnsAnswer
*answer
= NULL
, *soa
= NULL
;
197 unsigned i
, n_answer
= 0, n_soa
= 0;
205 if (q
->n_keys
<= 0) {
211 /* First iteration, count what we have */
212 for (i
= 0; i
< q
->n_keys
; i
++) {
215 if (q
->keys
[i
]->type
== DNS_TYPE_ANY
||
216 q
->keys
[i
]->class == DNS_CLASS_ANY
) {
219 /* If this is a generic match, then we have to
220 * go through the list by the name and look
221 * for everything manually */
223 j
= hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
]));
224 LIST_FOREACH(by_name
, j
, j
) {
225 k
= dns_resource_key_match_rr(q
->keys
[i
], j
->rr
);
235 j
= hashmap_get(z
->by_key
, q
->keys
[i
]);
237 LIST_FOREACH(by_key
, j
, j
)
240 if (hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
])))
246 if (n_answer
<= 0 && n_soa
<= 0) {
253 answer
= dns_answer_new(n_answer
);
259 soa
= dns_answer_new(n_soa
);
264 /* Second iteration, actually add the RRs to the answers */
265 for (i
= 0; i
< q
->n_keys
; i
++) {
268 if (q
->keys
[i
]->type
== DNS_TYPE_ANY
||
269 q
->keys
[i
]->class == DNS_CLASS_ANY
) {
272 j
= hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
]));
273 LIST_FOREACH(by_name
, j
, j
) {
274 k
= dns_resource_key_match_rr(q
->keys
[i
], j
->rr
);
278 r
= dns_answer_add_soa(soa
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
]));
280 r
= dns_answer_add(answer
, j
->rr
);
286 j
= hashmap_get(z
->by_key
, q
->keys
[i
]);
288 LIST_FOREACH(by_key
, j
, j
) {
289 r
= dns_answer_add(answer
, j
->rr
);
294 if (hashmap_get(z
->by_name
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
]))) {
295 r
= dns_answer_add_soa(soa
, DNS_RESOURCE_KEY_NAME(q
->keys
[i
]));
303 *ret_answer
= answer
;