snapshot; everything works other than fetches and database lookups.

bin/tests/adb_test.c

@@ -22,6 +22,7 @@
 #include <unistd.h>
 #include <string.h>
 
+#include <isc/app.h>
 #include <isc/assertions.h>
 #include <isc/buffer.h>
 #include <isc/error.h>
@@ -38,6 +39,10 @@
 isc_mem_t *mctx;
 isc_taskmgr_t *manager;
 
+static void lookup_callback(isc_task_t *, isc_event_t *);
+static void fatal(char *, ...);
+static inline void check_result(isc_result_t, char *);
+
 static void
 fatal(char *format, ...)
 {
@@ -57,6 +62,22 @@ check_result(isc_result_t result, char *msg)
 		fatal("%s: %s", msg, isc_result_totext(result));
 }
 
+static void
+lookup_callback(isc_task_t *task, isc_event_t *ev)
+{
+	dns_name_t *name;
+	dns_adbhandle_t *handle;
+
+	printf("Task %p got event %p type %08x from %p, arg %p\n",
+	       task, ev, ev->type, ev->sender, ev->arg);
+
+	name = ev->arg;
+	handle = ev->sender;
+
+	isc_event_free(&ev);
+	isc_app_shutdown();
+}
+
 int
 main(int argc, char **argv)
 {
@@ -76,6 +97,8 @@ main(int argc, char **argv)
 	(void)argv;
 
 	dns_result_register();
+	result = isc_app_start();
+	check_result(result, "isc_app_start()");
 
 	/*
 	 * EVERYTHING needs a memory context.
@@ -159,20 +182,18 @@ main(int argc, char **argv)
 	check_result(result, "dns_adb_insert 1.2.3.5");
 	printf("Added 1.2.3.5\n");
 
-	isc_task_detach(&t1);
-	isc_task_detach(&t2);
-
-	dns_adb_dump(adb, stderr);
-
 	/*
 	 * Try to look up a name or two.
 	 */
 	handle = NULL;
-	result = dns_adb_lookup(adb, NULL, NULL, NULL,
+	result = dns_adb_lookup(adb, t2, lookup_callback, &name1,
 				&name1, &name1, &handle);
 	check_result(result, "dns_adb_lookup name1");
 	check_result(handle->result, "handle->result");
 
+	dns_adb_dump(adb, stderr);
+	dns_adb_dumphandle(adb, handle, stderr);
+
 	/*
 	 * delete the names, and kill the adb
 	 */
@@ -182,9 +203,15 @@ main(int argc, char **argv)
 	check_result(result, "dns_adb_deletename name2");
 
 	dns_adb_dump(adb, stderr);
+	dns_adb_dumphandle(adb, handle, stderr);
+
+	isc_app_run();
 
 	dns_adb_done(adb, &handle);
 
+	isc_task_detach(&t1);
+	isc_task_detach(&t2);
+
 	isc_mem_stats(mctx, stdout);
 	dns_adb_destroy(&adb);
 
@@ -196,5 +223,7 @@ main(int argc, char **argv)
 	isc_mem_stats(mctx, stdout);
 	isc_mem_destroy(&mctx);
 
+	isc_app_finish();
+
 	return (0);
 }

lib/dns/adb.c

@@ -38,6 +38,7 @@
 #include <isc/event.h>
 
 #include <dns/address.h>
+#include <dns/events.h>
 #include <dns/name.h>
 #include <dns/rdata.h>
 #include <dns/rdataset.h>
@@ -183,6 +184,7 @@ static inline dns_adbentry_t *find_entry_and_lock(dns_adb_t *,
 						  isc_sockaddr_t *, int *);
 static void print_dns_name(FILE *, dns_name_t *);
 static void print_namehook_list(FILE *, dns_adbname_t *);
+static void print_handle_list(FILE *, dns_adbname_t *);
 static inline void inc_adb_refcnt(dns_adb_t *, isc_boolean_t);
 static inline void dec_adb_refcnt(dns_adb_t *, isc_boolean_t);
 static inline void inc_entry_refcnt(dns_adb_t *, dns_adbentry_t *,
@@ -807,6 +809,7 @@ dns_adb_lookup(dns_adb_t *adb, isc_task_t *task, isc_taskaction_t action,
 	dns_adbname_t *adbname;
 	int bucket;
 	isc_result_t result;
+	isc_boolean_t attach_to_task;
 
 	REQUIRE(DNS_ADB_VALID(adb));
 	if (task != NULL) {
@@ -817,6 +820,7 @@ dns_adb_lookup(dns_adb_t *adb, isc_task_t *task, isc_taskaction_t action,
 	REQUIRE(handlep != NULL && *handlep == NULL);
 
 	result = ISC_R_UNEXPECTED;
+	attach_to_task = ISC_FALSE;
 
 	/*
 	 * Look up the name in our internal database.
@@ -843,10 +847,6 @@ dns_adb_lookup(dns_adb_t *adb, isc_task_t *task, isc_taskaction_t action,
 		goto out;
 	}
 
-	handle->event.sender = task; /* store it here for a while */
-	handle->event.action = action;
-	handle->event.arg = arg;
-
 	/*
 	 * Look things up in our database first.
 	 */
@@ -876,6 +876,7 @@ dns_adb_lookup(dns_adb_t *adb, isc_task_t *task, isc_taskaction_t action,
 			handle->adbname = adbname;
 			handle->name_bucket = bucket;
 			ISC_LIST_APPEND(adbname->handles, handle, link);
+			attach_to_task = ISC_TRUE;
 		} else {
 			/*
 			 * We are done with this name, so release the lock
@@ -903,9 +904,20 @@ dns_adb_lookup(dns_adb_t *adb, isc_task_t *task, isc_taskaction_t action,
 	free_adbhandle(adb, &handle);
 	
  out:
-	if (handle != NULL)
+	if (handle != NULL) {
 		*handlep = handle;
 
+		if (attach_to_task) {
+			isc_task_t *taskp;
+
+			taskp = NULL;
+			isc_task_attach(task, &taskp);
+			handle->event.sender = taskp;
+			handle->event.action = action;
+			handle->event.arg = arg;
+		}
+	}
+
 	if (bucket != DNS_ADB_INVALIDBUCKET)
 		UNLOCK(&adb->namelocks[bucket]);
 
@@ -919,6 +931,7 @@ dns_adb_deletename(dns_adb_t *adb, dns_name_t *host)
 	dns_adbname_t *name;
 	dns_adbentry_t *entry;
 	dns_adbnamehook_t *namehook;
+	dns_adbhandle_t *handle;
 
 	REQUIRE(DNS_ADB_VALID(adb));
 	REQUIRE(host != NULL);
@@ -946,6 +959,31 @@ dns_adb_deletename(dns_adb_t *adb, dns_name_t *host)
 	/* XXX
 	 * If fetches are running for this name, cancel them all.
 	 */
+	handle = ISC_LIST_HEAD(name->handles);
+	while (handle != NULL) {
+		isc_event_t *ev;
+		isc_task_t *task;
+
+		LOCK(&handle->lock);
+
+		ev = &handle->event;
+		task = ev->sender;
+		ev->sender = handle;
+		ev->type = DNS_EVENT_ADBNAMEDELETED;
+		isc_task_sendanddetach(&task, &ev);
+
+		/*
+		 * Unlink the handle from the name, letting the caller
+		 * call dns_adb_done() on it to clean it up later.
+		 */
+		ISC_LIST_UNLINK(name->handles, handle, link);
+		handle->adbname = NULL;
+		handle->name_bucket = DNS_ADB_INVALIDBUCKET;
+
+		UNLOCK(&handle->lock);
+
+		handle = ISC_LIST_HEAD(name->handles);
+	}
 
 	/*
 	 * Loop through the name and kill any namehooks and entries they
@@ -1227,6 +1265,9 @@ dns_adb_dump(dns_adb_t *adb, FILE *f)
 			print_dns_name(f, &name->name);
 			fprintf(f, "\n");
 			print_namehook_list(f, name);
+			fprintf(f, "\n");
+			print_handle_list(f, name);
+			fprintf(f, "\n");
 
 			name = ISC_LIST_NEXT(name, link);
 		}
@@ -1286,6 +1327,60 @@ dns_adb_dump(dns_adb_t *adb, FILE *f)
 	UNLOCK(&adb->lock);
 }
 
+void
+dns_adb_dumphandle(dns_adb_t *adb, dns_adbhandle_t *handle, FILE *f)
+{
+	char tmp[512];
+	const char *tmpp;
+	dns_adbaddrinfo_t *ai;
+	isc_sockaddr_t *sa;
+
+	/*
+	 * Not used currently, in the API Just In Case we
+	 * want to dump out the name and/or entries too.
+	 */
+	(void)adb;
+
+	LOCK(&handle->lock);
+
+	fprintf(f, "Handle %p\n", handle);
+	fprintf(f, "\tquery_pending %d, result %d (%s)\n",
+		handle->query_pending, handle->result,
+		isc_result_totext(handle->result));
+	fprintf(f, "\tname_bucket %d, name %p, event sender %p\n",
+		handle->name_bucket, handle->adbname, handle->event.sender);
+
+	ai = ISC_LIST_HEAD(handle->list);
+	if (ai != NULL)
+		fprintf(f, "\tAddresses:\n");
+	while (ai != NULL) {
+		sa = ai->sockaddr;
+		switch (sa->type.sa.sa_family) {
+		case AF_INET:
+			tmpp = inet_ntop(AF_INET, &sa->type.sin.sin_addr,
+					 tmp, sizeof tmp);
+			break;
+		case AF_INET6:
+			tmpp = inet_ntop(AF_INET6, &sa->type.sin6.sin6_addr,
+					 tmp, sizeof tmp);
+			break;
+		default:
+			tmpp = "UnkFamily";
+		}
+
+		if (tmpp == NULL)
+			tmpp = "CANNOT TRANSLATE ADDRESS!";
+
+		fprintf(f, "\t\tentry %p, flags %08x goodness %d"
+			" srtt %u addr %s\n",
+			ai->entry, ai->flags, ai->goodness, ai->srtt, tmpp);
+
+		ai = ISC_LIST_NEXT(ai, link);
+	}
+
+	UNLOCK(&handle->lock);
+}
+
 static void
 print_dns_name(FILE *f, dns_name_t *name)
 {
@@ -1315,3 +1410,18 @@ print_namehook_list(FILE *f, dns_adbname_t *n)
 		nh = ISC_LIST_NEXT(nh, link);
 	}
 }
+
+static void
+print_handle_list(FILE *f, dns_adbname_t *name)
+{
+	dns_adbhandle_t *handle;
+
+	handle = ISC_LIST_HEAD(name->handles);
+	if (handle == NULL)
+		fprintf(f, "\t\tNo handles\n");
+
+	while (handle != NULL) {
+		fprintf(f, "\t\tHandle %p\n", handle);
+		handle = ISC_LIST_NEXT(handle, link);
+	}
+}

lib/dns/include/dns/adb.h

@@ -163,10 +163,17 @@ struct dns_adbaddrinfo {
  *
  * This is simply a standard event, with the "type" set to:
  *
- *	DNS_EVENT_MOREADDRESSES   -- another address resolved.
- *	DNS_EVENT_NOMOREADDRESSES -- all pending addresses failed,
+ *	DNS_EVENT_ADBMOREADDRESSES   -- another address resolved.
+ *	DNS_EVENT_ADBNOMOREADDRESSES -- all pending addresses failed,
  *					were canceled, or otherwise will
  *					not be usable.
+ *	DNS_EVENT_ADBCANCELED	     -- The request was canceled by a
+ *					3rd party.
+ *	DNS_EVENT_ADBNAMEDELETED     -- The name was deleted, so this request
+ *					was canceled.
+ *
+ * In each of these cases, the addresses returned by the initial call
+ * to dns_adb_lookup() can still be used until they are no longer needed.
  */
 
 /****
@@ -343,6 +350,20 @@ dns_adb_dump(dns_adb_t *adb, FILE *f);
  *	f != NULL, and be a file open for writing.
  */
 
+void
+dns_adb_dumphandle(dns_adb_t *adb, dns_adbhandle_t *handle, FILE *f);
+/*
+ * Dump the data associated with a handle.
+ *
+ * Requires:
+ *
+ *	adb be valid.
+ *
+ *	adbhandle be valid.
+ *
+ *	f != NULL, and be a file open for writing.
+ */
+
 
 /*
  * XXX Need functions/macros to: