eventdns: Apply a couple of patches from AGL; start working on windows compat; note...

svn:r6621

diff --git a/src/or/eventdns.c b/src/or/eventdns.c
index e955e541ca4127d2039ca56bea7f3048c75d3042..7830cd526bc8c2e485693f68988801ba6074025c 100644 (file)
--- a/src/or/eventdns.c
+++ b/src/or/eventdns.c
@@ -1,3 +1,12 @@
+/* $Id$ */
+
+// Modified from agl's original; see CVS for more info.
+// Try to keep this re-mergeable by Adam.  Don't make it depend on Tor.
+// TODO:
+//   - Check all malloc return values.
+//   - Better logging support
+//   - Learn about nameservers on win32.
+
 /* Async DNS Library
  * Adam Langley <agl@imperialviolet.org>
  * http://www.imperialviolet.org/eventdns.html
@@ -75,7 +84,7 @@
  *
  * In order for this library to be a good replacment for glibc's resolver it
  * supports searching. This involves setting a list of default domains, in
- * which names will be queried for. The number of nots in the query name
+ * which names will be queried for. The number of dots in the query name
  * determines the order in which this list is used.
  *
  * Searching appears to be a single lookup from the point of view of the API,
@@ -112,12 +121,12 @@
  * int eventdns_nameserver_add(unsigned long int addresss)
  *   Add a nameserver. The address should be an IP address in
  *   network byte order. The type of address is chosen so that
- *   it match in_addr.s_addr.
+ *   it matches in_addr.s_addr.
  *   Returns non-zero on error.
  *
  * int eventdns_nameserer_ip_add(const char *ip_as_string)
  *   This wraps the above function by parsing a string as an IP
- *   address and add it as a nameserver.
+ *   address and adds it as a nameserver.
  *   Returns non-zero on error
  *
  * int eventdns_resolve(const char *name, int flags,
@@ -238,6 +247,7 @@
 #include <unistd.h>
 #include <limits.h>
 #include <sys/stat.h>
+#include <ctype.h>
 
 #ifndef HOST_NAME_MAX
 #define HOST_NAME_MAX 255
@@ -387,6 +397,9 @@ error_is_eagain(int sock)
 #define error_is_eagain(sock) (errno == EAGAIN)
 #endif
 
+#define ISSPACE(c) isspace((int)(unsigned char)(c))
+#define ISDIGIT(c) isdigit((int)(unsigned char)(c))
+
 #ifndef NDEBUG
 static const char *
 debug_ntoa(u32 address)
@@ -1180,6 +1193,29 @@ eventdns_nameserver_ip_add(const char *ip_as_string) {
        return eventdns_nameserver_add(ina.s_addr);
 }
 
+/* Add multiple nameservers from a space-or-comma-separated list. */
+static int
+eventdns_nameserver_ip_add_line(const char *ips) {
+       const char *addr;
+       char *buf;
+       int r;
+       while (*ips) {
+               while (ISSPACE(*ips) || *ips == ',' || *ips == '\t')
+                       ++ips;
+               addr = ips;
+               while (ISDIGIT(*ips) || *ips == '.')
+                       ++ips;
+               buf = malloc(ips-addr+1);
+               if (!buf) return 4;
+               memcpy(buf, addr, ips-addr);
+               buf[ips-addr] = '\0';
+               r = eventdns_nameserver_ip_add(buf);
+               free(buf);
+               if (r) return r;
+       }
+       return 0;
+}
+
 // insert into the tail of the queue
 static void
 eventdns_request_insert(struct request *req, struct request **head) {
@@ -1651,3 +1687,60 @@ out1:
        close(fd);
        return err;
 }
+
+#ifdef MS_WINDOWS
+static int
+load_nameservers_with_getnetworkparams(void)
+{
+       FIXED_INFO *fixed;
+       HMODULE handle = 0;
+       ULONG size = sizeof(FIXED_INFO);
+       void *buf = NULL;
+       int status = 0, r, added_any;
+       IP_ADDR_STRING *ns;
+       DWORD (WINAPI *fn)(FIXED_INFO*, DWORD*);
+
+       if (!(handle = LoadLibrary("iphlpapi.dll")))
+               goto done;
+       if (!(fn = GetProcAddress(handle, "GetNetworkParams")))
+               goto done;
+
+       buf = malloc(size);
+       if (!buf) { status = 4; goto done; }
+       fixed = buf;
+       r = fn(fixed, &size);
+       if (r != ERROR_SUCCESS && r != ERROR_BUFFER_OVERFLOW) {
+               status = -1;
+               goto done;
+       }
+       if (r != ERROR_SUCCESS) {
+               free(buf);
+               buf = malloc(size);
+               if (!buf) { status = 4; goto done; }
+               fixed = buf;
+               r = fn(fixed, &size);
+               if (r != ERROR_SUCCESS) { status = -1; goto done; }
+       }
+
+       assert(fixed);
+       added_any = 0;
+       ns = fixed->DnsServerList;
+       while (ns) {
+               r = eventdns_nameserver_ip_add_line(ns->IpAddress.String);
+               if (r) { status = r; goto done; }
+               added_any = 0;
+               ns = ns->next;
+       }
+
+       if (!added_any)
+               status = -1;
+
+ done:
+       if (buf)
+               free(buf);
+       if (handle)
+               FreeLibrary(handle);
+       return status;
+}
+
+#endif

diff --git a/src/or/eventdns.h b/src/or/eventdns.h
index 43f474e3f7f1aeece0e5a68c8ed5b35c61ed16d6..c2dab163a8adc6931fdf131f84fbe8c1ac4596a8 100644 (file)
--- a/src/or/eventdns.h
+++ b/src/or/eventdns.h
@@ -16,14 +16,25 @@
 #ifndef EVENTDNS_H
 #define EVENTDNS_H
 
+/* Error codes 0-5 are as described in RFC 1035. */
 #define DNS_ERR_NONE 0
+/* The name server was unable to interpret the query */
 #define DNS_ERR_FORMAT 1
+/* The name server was unable to process this query due to a problem with the
+ * name server */
 #define DNS_ERR_SERVERFAILED 2
+/* The domain name does not exist */
 #define DNS_ERR_NOTEXIST 3
+/* The name server does not support the requested kind of query */
 #define DNS_ERR_NOTIMPL 4
+/* The name server refuses to reform the specified operation for policy
+ * reasons */
 #define DNS_ERR_REFUSED 5
+/* The reply was truncated or ill-formated */
 #define DNS_ERR_TRUNCATED 65
+/* An unknown error occurred */
 #define DNS_ERR_UNKNOWN 66
+/* Communication with the server timed out */
 #define DNS_ERR_TIMEOUT 67
 
 #define DNS_IPv4_A 1