Modifications to have dighost.c use liblwres to parse the resolv.conf file. Updates to dig, host and nslookup to increase modularity and use more functions in dighost.c

bin/dig/dighost.c

@@ -15,7 +15,7 @@
  * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
-/* $Id: dighost.c,v 1.248 2002/08/06 00:35:01 marka Exp $ */
+/* $Id: dighost.c,v 1.249 2002/08/12 18:25:25 mayer Exp $ */
 
 /*
  * Notice to programmers:  Do not use this code as an example of how to
@@ -60,10 +60,16 @@
 #include <isc/types.h>
 #include <isc/util.h>
 
+#include <lwres/lwres.h>
+#include <lwres/net.h>
+
 #include <bind9/getaddresses.h>
 
 #include <dig/dig.h>
 
+static lwres_context_t *lwctx = NULL;
+static lwres_conf_t *lwconf;
+
 ISC_LIST(dig_lookup_t) lookup_list;
 dig_serverlist_t server_list;
 ISC_LIST(dig_searchlist_t) search_list;
@@ -144,6 +150,17 @@ connect_timeout(isc_task_t *task, isc_event_t *event);
 static void
 launch_next_query(dig_query_t *query, isc_boolean_t include_question);
 
+
+static void *
+mem_alloc(void *arg, size_t size) {
+	return (isc_mem_get(arg, size));
+}
+
+static void
+mem_free(void *arg, void *mem, size_t size) {
+	isc_mem_put(arg, mem, size);
+}
+
 char *
 next_token(char **stringp, const char *delim) {
 	char *res;
@@ -331,6 +348,86 @@ make_server(const char *servname) {
 	ISC_LINK_INIT(srv, link);
 	return (srv);
 }
+static int
+addr2af(int lwresaddrtype)
+{
+	int af = 0;
+
+	switch (lwresaddrtype) {
+	case LWRES_ADDRTYPE_V4:
+		af = AF_INET;
+		break;
+
+	case LWRES_ADDRTYPE_V6:
+		af = AF_INET6;
+		break;
+	}
+
+	return (af);
+}
+/*
+ * Create a copy of the server list from the lwres configuration structure.
+ * The dest list must have already had ISC_LIST_INIT applied.
+ */
+void
+copy_server_list(lwres_conf_t *confdata, dig_serverlist_t *dest) {
+	dig_server_t *newsrv;
+	char tmp[sizeof("ffff:ffff:ffff:ffff:ffff:ffff:255.255.255.255")];
+	int af;
+	int i;
+
+	debug("copy_server_list()");
+	for (i = 0; i < confdata->nsnext; i++) {
+		af = addr2af(confdata->nameservers[i].family);
+
+		lwres_net_ntop(af, confdata->nameservers[i].address,
+				   tmp, sizeof(tmp));
+		newsrv = make_server(tmp);
+		ISC_LINK_INIT(newsrv, link);
+		ISC_LIST_ENQUEUE(*dest, newsrv, link);
+	}
+}
+void
+flush_server_list(void) {
+	dig_server_t *s, *ps;
+
+	debug("flush_server_list()");
+	s = ISC_LIST_HEAD(server_list);
+	while (s != NULL) {
+		ps = s;
+		s = ISC_LIST_NEXT(s, link);
+		ISC_LIST_DEQUEUE(server_list, ps, link);
+		isc_mem_free(mctx, ps);
+	}
+}
+void
+set_nameserver(char *opt) {
+	dig_server_t *srv;
+
+	if (opt == NULL)
+		return;
+
+	flush_server_list();
+	srv = make_server(opt);
+	if (srv == NULL)
+		fatal("memory allocation failure");
+	ISC_LIST_INITANDAPPEND(server_list, srv, link);
+}
+
+isc_result_t
+add_nameserver(lwres_conf_t *confdata, char *addr, int af) {
+
+	int i = confdata->nsnext;
+
+	if (confdata->nsnext >= LWRES_CONFMAXNAMESERVERS)
+		return (ISC_R_FAILURE);
+
+	if (lwres_net_pton(af, addr, &confdata->nameservers[i]) == 1) {
+		confdata->nsnext++;
+		return (ISC_R_SUCCESS);
+	}
+	return (ISC_R_FAILURE);
+}
 
 /*
  * Produce a cloned server list.  The dest list must have already had
@@ -594,86 +691,62 @@ make_searchlist_entry(char *domain) {
 	return (search);
 }
 
+void
+create_search_list(lwres_conf_t *confdata) {
+	int i;
+	dig_searchlist_t *search;
+
+	debug("create_search_list()");
+	ISC_LIST_INIT(search_list);
+
+	for (i = 0; i < confdata->searchnxt; i++) {
+		search = make_searchlist_entry(confdata->search[i]);
+		ISC_LIST_APPEND(search_list, search, link);
+	}
+}
+
 /*
  * Setup the system as a whole, reading key information and resolv.conf
  * settings.
  */
 void
 setup_system(void) {
-	char rcinput[MXNAME];
+	dig_searchlist_t *domain = NULL;
-	FILE *fp;
+	lwres_result_t lwresult;
-	char *ptr;
-	dig_server_t *srv;
-	dig_searchlist_t *search, *domain = NULL;
-	isc_boolean_t get_servers;
-	char *input;
 
 	debug("setup_system()");
 
-	free_now = ISC_FALSE;
+	lwresult = lwres_context_create(&lwctx, mctx, mem_alloc, mem_free, 1);
-	get_servers = ISC_TF(server_list.head == NULL);
+	if (lwresult != LWRES_R_SUCCESS)
-	fp = fopen(RESOLV_CONF, "r");
+		fatal("lwres_context_create failed");
-	/* XXX Use lwres resolv.conf reader */
-	if (fp == NULL)
-		goto no_file;
 
-	while (fgets(rcinput, MXNAME, fp) != 0) {
+	(void)lwres_conf_parse(lwctx, RESOLV_CONF);
-		input = rcinput;
+	lwconf = lwres_conf_get(lwctx);
-		ptr = next_token(&input, " \t\r\n");
+
-		if (ptr != NULL) {
+	/* Make the search list */
-			if (get_servers &&
+	if (lwconf->searchnxt > 0)
-			    strcasecmp(ptr, "nameserver") == 0) {
+		create_search_list(lwconf);
-				debug("got a nameserver line");
+	else {
-				ptr = next_token(&input, " \t\r\n");
+		/* No search list. Use the domain name if any */
-				if (ptr != NULL) {
+		if (lwconf->domainname != NULL) {
-					srv = make_server(ptr);
+			domain = make_searchlist_entry(lwconf->domainname);
-					ISC_LIST_APPEND(server_list, srv, link);
+			ISC_LIST_INITANDAPPEND(search_list, domain, link);
-				}
+			domain  = NULL;
-			} else if (strcasecmp(ptr, "options") == 0) {
-				ptr = next_token(&input, " \t\r\n");
-				if (ptr != NULL) {
-					if (strncasecmp(ptr, "ndots:", 6) == 0
-					    && ndots == -1)
-					{
-						ndots = atoi(&ptr[6]);
-						debug("ndots is %d.", ndots);
-					}
-				}
-			} else if (strcasecmp(ptr, "search") == 0){
-				while ((ptr = next_token(&input, " \t\r\n"))
-				       != NULL) {
-					debug("adding search %s", ptr);
-					search = make_searchlist_entry(ptr);
-					ISC_LIST_INITANDAPPEND(search_list,
-							       search, link);
-				}
-			} else if (strcasecmp(ptr, "domain") == 0) {
-				while ((ptr = next_token(&input, " \t\r\n"))
-				       != NULL) {
-					if (domain != NULL)
-						isc_mem_free(mctx, domain);
-					domain = make_searchlist_entry(ptr);
-				}
-			}
 		}
 	}
-	fclose(fp);
+			
- no_file:
+	ndots = lwconf->ndots;
+	debug("ndots is %d.", ndots);
 
-	if (ISC_LIST_EMPTY(search_list) && domain != NULL) {
+	/* If we don't find a nameserver fall back to localhost */
-		ISC_LIST_INITANDAPPEND(search_list, domain, link);
+	if (lwconf->nsnext == 0) {
-		domain = NULL;
+		lwresult = add_nameserver(lwconf, "127.0.0.1", AF_INET);
+		if(lwresult != ISC_R_SUCCESS)
+			fatal("add_nameserver failed");
 	}
-	if (domain != NULL)
-		isc_mem_free(mctx, domain);
-	
-	if (ndots == -1)
-		ndots = 1;
 
-	if (server_list.head == NULL) {
+	if (ISC_LIST_EMPTY(server_list))
-		srv = make_server("127.0.0.1");
+		copy_server_list(lwconf, &server_list);
-		ISC_LIST_APPEND(server_list, srv, link);
-	}
 
 	if (keyfile[0] != 0)
 		setup_file_key();
@@ -1691,7 +1764,7 @@ send_udp(dig_query_t *query) {
  */
 static void
 connect_timeout(isc_task_t *task, isc_event_t *event) {
-	dig_lookup_t *l = NULL, *n;
+	dig_lookup_t *l = NULL;
 	dig_query_t *query = NULL, *cq;
 
 	UNUSED(task);
@@ -1727,8 +1800,7 @@ connect_timeout(isc_task_t *task, isc_event_t *event) {
 			debug("making new TCP request, %d tries left",
 			      l->retries);
 			l->retries--;
-			n = requeue_lookup(l, ISC_TRUE);
+			requeue_lookup(l, ISC_TRUE);
-			n->origin = query->lookup->origin;
 			cancel_lookup(l);
 		}
 	} else {
@@ -2300,7 +2372,6 @@ recv_done(isc_task_t *task, isc_event_t *event) {
 		printf(";; Truncated, retrying in TCP mode.\n");
 		n = requeue_lookup(l, ISC_TRUE);
 		n->tcp_mode = ISC_TRUE;
-		n->origin = query->lookup->origin;
 		dns_message_destroy(&msg);
 		isc_event_free(&event);
 		clear_query(query);
@@ -2574,8 +2645,6 @@ cancel_all(void) {
  */
 void
 destroy_libs(void) {
-	void *ptr;
-	dig_server_t *s;
 
 	debug("destroy_libs()");
 	if (global_task != NULL) {
@@ -2601,13 +2670,11 @@ destroy_libs(void) {
 
 	free_now = ISC_TRUE;
 
-	s = ISC_LIST_HEAD(server_list);
+	lwres_conf_clear(lwctx);
-	while (s != NULL) {
+	lwres_context_destroy(&lwctx);
-		debug("freeing global server %p", s);
+
-		ptr = s;
+	flush_server_list();
-		s = ISC_LIST_NEXT(s, link);
+
-		isc_mem_free(mctx, ptr);
-	}
 	clear_searchlist();
 	if (commctx != NULL) {
 		debug("freeing commctx");

bin/dig/host.c

@@ -15,7 +15,7 @@
  * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
-/* $Id: host.c,v 1.84 2002/07/25 05:46:07 marka Exp $ */
+/* $Id: host.c,v 1.85 2002/08/12 18:25:25 mayer Exp $ */
 
 #include <config.h>
 #include <stdlib.h>
@@ -41,7 +41,7 @@
 #include <dig/dig.h>
 
 extern ISC_LIST(dig_lookup_t) lookup_list;
-extern ISC_LIST(dig_server_t) server_list;
+extern dig_serverlist_t server_list;
 extern ISC_LIST(dig_searchlist_t) search_list;
 
 extern isc_boolean_t usesearch;
@@ -472,7 +472,6 @@ printmessage(dig_query_t *query, dns_message_t *msg, isc_boolean_t headers) {
 static void
 parse_args(isc_boolean_t is_batchfile, int argc, char **argv) {
 	char hostname[MXNAME];
-	dig_server_t *srv;
 	dig_lookup_t *lookup;
 	int c;
 	char store[MXNAME];
@@ -586,9 +585,8 @@ parse_args(isc_boolean_t is_batchfile, int argc, char **argv) {
 	strncpy(hostname, argv[isc_commandline_index], sizeof(hostname));
 	hostname[sizeof(hostname)-1]=0;
 	if (argc > isc_commandline_index + 1) {
-		srv = make_server(argv[isc_commandline_index+1]);
+		set_nameserver(argv[isc_commandline_index+1]);
-		debug("server is %s", srv->servername);
+		debug("server is %s", argv[isc_commandline_index+1]);
-		ISC_LIST_APPEND(server_list, srv, link);
 		listed_server = ISC_TRUE;
 	}
 

bin/dig/include/dig/dig.h

@@ -15,7 +15,7 @@
  * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
-/* $Id: dig.h,v 1.78 2002/07/25 05:46:07 marka Exp $ */
+/* $Id: dig.h,v 1.79 2002/08/12 18:25:25 mayer Exp $ */
 
 #ifndef DIG_H
 #define DIG_H
@@ -235,6 +235,12 @@ clone_lookup(dig_lookup_t *lookold, isc_boolean_t servers);
 dig_server_t *
 make_server(const char *servname);
 
+void
+flush_server_list(void);
+
+void
+set_nameserver(char *opt);
+
 void
 clone_server_list(dig_serverlist_t src,
 		  dig_serverlist_t *dest);

bin/dig/nslookup.c

@@ -15,7 +15,7 @@
  * WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  */
 
-/* $Id: nslookup.c,v 1.98 2002/08/01 03:28:01 mayer Exp $ */
+/* $Id: nslookup.c,v 1.99 2002/08/12 18:25:25 mayer Exp $ */
 
 #include <config.h>
 
@@ -45,7 +45,7 @@
 #include <dig/dig.h>
 
 extern ISC_LIST(dig_lookup_t) lookup_list;
-extern ISC_LIST(dig_server_t) server_list;
+extern dig_serverlist_t server_list;
 extern ISC_LIST(dig_searchlist_t) search_list;
 
 extern isc_boolean_t usesearch, debugging;
@@ -699,39 +699,6 @@ addlookup(char *opt) {
 	debug("looking up %s", lookup->textname);
 }
 
-static void
-flush_server_list(void) {
-	dig_server_t *s, *ps;
-
-	debug("flush_server_list()");
-	s = ISC_LIST_HEAD(server_list);
-	while (s != NULL) {
-		ps = s;
-		s = ISC_LIST_NEXT(s, link);
-		ISC_LIST_DEQUEUE(server_list, ps, link);
-		isc_mem_free(mctx, ps);
-	}
-}
-
-/*
- * This works on the global server list, instead of on a per-lookup
- * server list, since the change is persistent.
- */
-static void
-setsrv(char *opt) {
-	dig_server_t *srv;
-
-	if (opt == NULL)
-		return;
-
-	flush_server_list();
-	srv = isc_mem_allocate(mctx, sizeof(struct dig_server));
-	if (srv == NULL)
-		fatal("memory allocation failure");
-	safecpy(srv->servername, opt, sizeof(srv->servername));
-	ISC_LIST_INITANDAPPEND(server_list, srv, link);
-}
-
 static void
 get_next_command(void) {
 	char *buf;
@@ -759,7 +726,7 @@ get_next_command(void) {
 		setoption(arg);
 	else if ((strcasecmp(ptr, "server") == 0) ||
 		 (strcasecmp(ptr, "lserver") == 0)) {
-		setsrv(arg);
+		set_nameserver(arg);
 		show_settings(ISC_TRUE, ISC_TRUE);
 	} else if (strcasecmp(ptr, "exit") == 0) {
 		in_use = ISC_FALSE;
@@ -801,7 +768,7 @@ parse_args(int argc, char **argv) {
 				addlookup(argv[0]);
 			}
 			else
-				setsrv(argv[0]);
+				set_nameserver(argv[0]);
 		}
 	}
 }