lib/dns/lookup.c

1 /*	$NetBSD: lookup.c,v 1.8 2024/02/21 22:52:06 christos Exp $	*/
29 #include <dns/lookup.h>
61 lookup_find(dns_lookup_t *lookup, dns_fetchevent_t *event);
65 	dns_lookup_t *lookup = event->ev_arg;
70 	REQUIRE(VALID_LOOKUP(lookup));
71 	REQUIRE(lookup->task == task);
73 	REQUIRE(fevent->fetch == lookup->fetch);
75 	lookup_find(lookup, fevent);
79 start_fetch(dns_lookup_t *lookup) {
83 	 * The caller must be holding the lookup's lock.
86 	REQUIRE(lookup->fetch == NULL);
89 		lookup->view->resolver, dns_fixedname_name(&lookup->name),
90 		lookup->type, NULL, NULL, NULL, NULL, 0, 0, 0, NULL,
91 		lookup->task, fetch_done, lookup, &lookup->rdataset,
92 		&lookup->sigrdataset, &lookup->fetch);
98 build_event(dns_lookup_t *lookup) {
103 	name = isc_mem_get(lookup->mctx, sizeof(dns_name_t));
105 	dns_name_dup(dns_fixedname_name(&lookup->name), lookup->mctx, name);
107 	if (dns_rdataset_isassociated(&lookup->rdataset)) {
108 		rdataset = isc_mem_get(lookup->mctx, sizeof(dns_rdataset_t));
110 		dns_rdataset_clone(&lookup->rdataset, rdataset);
113 	if (dns_rdataset_isassociated(&lookup->sigrdataset)) {
114 		sigrdataset = isc_mem_get(lookup->mctx, sizeof(dns_rdataset_t));
116 		dns_rdataset_clone(&lookup->sigrdataset, sigrdataset);
119 	lookup->event->name = name;
120 	lookup->event->rdataset = rdataset;
121 	lookup->event->sigrdataset = sigrdataset;
125 view_find(dns_lookup_t *lookup, dns_name_t *foundname) {
127 	dns_name_t *name = dns_fixedname_name(&lookup->name);
130 	if (lookup->type == dns_rdatatype_rrsig) {
133 		type = lookup->type;
136 	result = dns_view_find(lookup->view, name, type, 0, 0, false, false,
137 			       &lookup->event->db, &lookup->event->node,
138 			       foundname, &lookup->rdataset,
139 			       &lookup->sigrdataset);
144 lookup_find(dns_lookup_t *lookup, dns_fetchevent_t *event) {
157 	REQUIRE(VALID_LOOKUP(lookup));
159 	LOCK(&lookup->lock);
161 	name = dns_fixedname_name(&lookup->name);
164 		lookup->restarts++;
168 		if (event == NULL && !lookup->canceled) {
170 			INSIST(!dns_rdataset_isassociated(&lookup->rdataset));
172 				&lookup->sigrdataset));
176 			if (lookup->event->node != NULL) {
177 				INSIST(lookup->event->db != NULL);
178 				dns_db_detachnode(lookup->event->db,
179 						  &lookup->event->node);
181 			if (lookup->event->db != NULL) {
182 				dns_db_detach(&lookup->event->db);
184 			result = view_find(lookup, fname);
190 				if (lookup->event->node != NULL) {
191 					INSIST(lookup->event->db != NULL);
192 					dns_db_detachnode(lookup->event->db,
193 							  &lookup->event->node);
195 				if (lookup->event->db != NULL) {
196 					dns_db_detach(&lookup->event->db);
198 				result = start_fetch(lookup);
207 			dns_resolver_destroyfetch(&lookup->fetch);
208 			INSIST(event->rdataset == &lookup->rdataset);
209 			INSIST(event->sigrdataset == &lookup->sigrdataset);
215 		if (lookup->canceled) {
221 			build_event(lookup);
226 				dns_db_attach(event->db, &lookup->event->db);
229 				dns_db_attachnode(lookup->event->db,
231 						  &lookup->event->node);
236 			 * Copy the CNAME's target into the lookup's
239 			result = dns_rdataset_first(&lookup->rdataset);
243 			dns_rdataset_current(&lookup->rdataset, &rdata);
261 			result = dns_rdataset_first(&lookup->rdataset);
265 			dns_rdataset_current(&lookup->rdataset, &rdata);
288 		if (dns_rdataset_isassociated(&lookup->rdataset)) {
289 			dns_rdataset_disassociate(&lookup->rdataset);
291 		if (dns_rdataset_isassociated(&lookup->sigrdataset)) {
292 			dns_rdataset_disassociate(&lookup->sigrdataset);
309 		if (want_restart && lookup->restarts == MAX_RESTARTS) {
317 		lookup->event->result = result;
318 		lookup->event->ev_sender = lookup;
319 		isc_task_sendanddetach(&lookup->task,
320 				       (isc_event_t **)(void *)&lookup->event);
321 		dns_view_detach(&lookup->view);
324 	UNLOCK(&lookup->lock);
363 	dns_lookup_t *lookup;
366 	lookup = isc_mem_get(mctx, sizeof(*lookup));
367 	lookup->mctx = NULL;
368 	isc_mem_attach(mctx, &lookup->mctx);
369 	lookup->options = options;
371 	ievent = isc_event_allocate(mctx, lookup, DNS_EVENT_LOOKUPDONE, action,
372 				    arg, sizeof(*lookup->event));
373 	lookup->event = (dns_lookupevent_t *)ievent;
374 	lookup->event->ev_destroy = levent_destroy;
375 	lookup->event->ev_destroy_arg = mctx;
376 	lookup->event->result = ISC_R_FAILURE;
377 	lookup->event->name = NULL;
378 	lookup->event->rdataset = NULL;
379 	lookup->event->sigrdataset = NULL;
380 	lookup->event->db = NULL;
381 	lookup->event->node = NULL;
383 	lookup->task = NULL;
384 	isc_task_attach(task, &lookup->task);
386 	isc_mutex_init(&lookup->lock);
388 	dns_fixedname_init(&lookup->name);
390 	dns_name_copy(name, dns_fixedname_name(&lookup->name));
392 	lookup->type = type;
393 	lookup->view = NULL;
394 	dns_view_attach(view, &lookup->view);
395 	lookup->fetch = NULL;
396 	lookup->restarts = 0;
397 	lookup->canceled = false;
398 	dns_rdataset_init(&lookup->rdataset);
399 	dns_rdataset_init(&lookup->sigrdataset);
400 	lookup->magic = LOOKUP_MAGIC;
402 	*lookupp = lookup;
404 	lookup_find(lookup, NULL);
410 dns_lookup_cancel(dns_lookup_t *lookup) {
411 	REQUIRE(VALID_LOOKUP(lookup));
413 	LOCK(&lookup->lock);
415 	if (!lookup->canceled) {
416 		lookup->canceled = true;
417 		if (lookup->fetch != NULL) {
418 			INSIST(lookup->view != NULL);
419 			dns_resolver_cancelfetch(lookup->fetch);
423 	UNLOCK(&lookup->lock);
428 	dns_lookup_t *lookup;
431 	lookup = *lookupp;
433 	REQUIRE(VALID_LOOKUP(lookup));
434 	REQUIRE(lookup->event == NULL);
435 	REQUIRE(lookup->task == NULL);
436 	REQUIRE(lookup->view == NULL);
437 	if (dns_rdataset_isassociated(&lookup->rdataset)) {
438 		dns_rdataset_disassociate(&lookup->rdataset);
440 	if (dns_rdataset_isassociated(&lookup->sigrdataset)) {
441 		dns_rdataset_disassociate(&lookup->sigrdataset);
444 	isc_mutex_destroy(&lookup->lock);
445 	lookup->magic = 0;
446 	isc_mem_putanddetach(&lookup->mctx, lookup, sizeof(*lookup));