summaryrefslogtreecommitdiff
path: root/ares_init.c
diff options
context:
space:
mode:
authorDavid Stuart <dstuart@counterpath.com>2011-05-17 11:53:13 +0200
committerDaniel Stenberg <daniel@haxx.se>2011-05-17 11:53:13 +0200
commit6518b56a5e2bc9e39d720f17fa7dd322a28dc33c (patch)
tree3a7bed56f287de887aaa110391f7116f9f87499f /ares_init.c
parentd60f07d09461b05e6b82a6f4a853963f8102c39a (diff)
downloadc-ares-6518b56a5e2bc9e39d720f17fa7dd322a28dc33c.tar.gz
c-ares-6518b56a5e2bc9e39d720f17fa7dd322a28dc33c.tar.bz2
c-ares-6518b56a5e2bc9e39d720f17fa7dd322a28dc33c.zip
IPv6-on-windows: find DNS servers correctly
Diffstat (limited to 'ares_init.c')
-rw-r--r--ares_init.c239
1 files changed, 181 insertions, 58 deletions
diff --git a/ares_init.c b/ares_init.c
index 75064e9..4ee31ef 100644
--- a/ares_init.c
+++ b/ares_init.c
@@ -69,6 +69,7 @@
#include "ares_nowarn.h"
#include "ares_platform.h"
#include "ares_private.h"
+#include "inet_ntop.h"
#ifdef ANDROID
#include <sys/system_properties.h>
@@ -595,73 +596,195 @@ static int get_res_interfaces_nt(HKEY hKey, const char *subkey, char **obuf)
return 0;
}
+/**
+ * The desired output for this method is that we set "ret_buf" to
+ * something like:
+ *
+ * 192.168.0.1,dns01.my.domain,fe80::200:f8ff:fe21:67cf
+ *
+ * The only ordering requirement is that primary servers are listed
+ * before secondary. There is no requirement that IPv4 addresses should
+ * necessarily be before IPv6.
+ *
+ * Note that ret_size should ideally be big enough to hold around
+ * 2-3 IPv4 and 2-3 IPv6 addresses.
+ *
+ * Finally, we need to return the total number of DNS servers located.
+ */
static int get_iphlpapi_dns_info (char *ret_buf, size_t ret_size)
{
- FIXED_INFO *fi, *newfi;
- DWORD size = sizeof (*fi);
- IP_ADDR_STRING *ipAddr;
- int i, count = 0;
- int debug = 0;
- size_t ip_size = sizeof("255.255.255.255,")-1;
- size_t left = ret_size;
- char *ret = ret_buf;
- HRESULT res;
-
- fi = malloc(size);
- if (!fi)
- return 0;
-
- res = (*ares_fpGetNetworkParams) (fi, &size);
- if ((res != ERROR_BUFFER_OVERFLOW) && (res != ERROR_SUCCESS))
- goto quit;
-
- newfi = realloc(fi, size);
- if (!newfi)
- goto quit;
-
- fi = newfi;
- res = (*ares_fpGetNetworkParams) (fi, &size);
- if (res != ERROR_SUCCESS)
- goto quit;
-
- if (debug)
+ const size_t ipv4_size = INET_ADDRSTRLEN + 1; /* +1 for ',' at end */
+ const size_t ipv6_size = INET6_ADDRSTRLEN + 12; /* +12 for "%0123456789," at end */
+ size_t left = ret_size;
+ char *ret = ret_buf;
+ int count = 0;
+
+ /* Use the GetAdaptersAddresses method if it's available, otherwise
+ fall back to GetNetworkParams. */
+ if (ares_fpGetAdaptersAddresses != ZERO_NULL)
{
- printf ("Host Name: %s\n", fi->HostName);
- printf ("Domain Name: %s\n", fi->DomainName);
- printf ("DNS Servers:\n"
- " %s (primary)\n", fi->DnsServerList.IpAddress.String);
+ const ULONG working_buf_size = 15000;
+ IP_ADAPTER_ADDRESSES *pFirstEntry = NULL;
+ IP_ADAPTER_ADDRESSES *pEntry = NULL;
+ ULONG bufSize = 0;
+ ULONG result = 0;
+
+ /* According to MSDN, the recommended way to do this is to use a temporary
+ buffer of 15K, to "dramatically reduce the chance that the GetAdaptersAddresses
+ method returns ERROR_BUFFER_OVERFLOW" */
+ pFirstEntry = ( IP_ADAPTER_ADDRESSES * ) malloc( working_buf_size );
+ bufSize = working_buf_size;
+ if( !pFirstEntry )
+ return 0;
+
+ /* Call the method one time */
+ result = ( *ares_fpGetAdaptersAddresses )( AF_UNSPEC, 0, 0, pFirstEntry, &bufSize );
+ if( result == ERROR_BUFFER_OVERFLOW )
+ {
+ /* Reallocate, bufSize should now be set to the required size */
+ pFirstEntry = ( IP_ADAPTER_ADDRESSES * ) realloc( pFirstEntry, bufSize );
+ if( !pFirstEntry )
+ return 0;
+
+ /* Call the method a second time */
+ result = ( *ares_fpGetAdaptersAddresses )( AF_UNSPEC, 0, 0, pFirstEntry, &bufSize );
+ if( result == ERROR_BUFFER_OVERFLOW )
+ {
+ /* Reallocate, bufSize should now be set to the required size */
+ pFirstEntry = ( IP_ADAPTER_ADDRESSES * ) realloc( pFirstEntry, bufSize );
+ if( !pFirstEntry )
+ return 0;
+
+ /* Call the method a third time. The maximum number of times we're going to do
+ this is 3. Three shall be the number thou shalt count, and the number of the
+ counting shall be three. Five is right out. */
+ result = ( *ares_fpGetAdaptersAddresses )( AF_UNSPEC, 0, 0, pFirstEntry, &bufSize );
+ }
+ }
+
+ /* Check the current result for failure */
+ if( result != ERROR_SUCCESS )
+ {
+ free( pFirstEntry );
+ return 0;
+ }
+
+ /* process the results */
+ for( pEntry = pFirstEntry ; pEntry != NULL ; pEntry = pEntry->Next )
+ {
+ IP_ADAPTER_DNS_SERVER_ADDRESS* pDNSAddr = pEntry->FirstDnsServerAddress;
+ for( ; pDNSAddr != NULL ; pDNSAddr = pDNSAddr->Next )
+ {
+ struct sockaddr *pGenericAddr = pDNSAddr->Address.lpSockaddr;
+ int stringlen = 0;
+
+ if( pGenericAddr->sa_family == AF_INET && left > ipv4_size )
+ {
+ /* Handle the v4 case */
+ struct sockaddr_in *pIPv4Addr = ( struct sockaddr_in * ) pGenericAddr;
+ ares_inet_ntop( AF_INET, &pIPv4Addr->sin_addr, ret, ipv4_size - 1 ); /* -1 for comma */
+
+ /* Append a comma to the end, THEN NULL. Should be OK because we
+ already tested the size at the top of the if statement. */
+ stringlen = strlen( ret );
+ ret[ stringlen ] = ',';
+ ret[ stringlen + 1 ] = '\0';
+ ret += stringlen + 1;
+ left -= ret - ret_buf;
+ ++count;
+ }
+ else if( pGenericAddr->sa_family == AF_INET6 && left > ipv6_size )
+ {
+ /* Handle the v6 case */
+ struct sockaddr_in6 *pIPv6Addr = ( struct sockaddr_in6 * ) pGenericAddr;
+ ares_inet_ntop( AF_INET6, &pIPv6Addr->sin6_addr, ret, ipv6_size - 1 ); /* -1 for comma */
+
+ /* Append a comma to the end, THEN NULL. Should be OK because we
+ already tested the size at the top of the if statement. */
+ stringlen = strlen( ret );
+ ret[ stringlen ] = ',';
+ ret[ stringlen + 1 ] = '\0';
+ ret += stringlen + 1;
+ left -= ret - ret_buf;
+ ++count;
+
+ /* NB on Windows this also returns stuff in the fec0::/10 range,
+ seems to be hard-coded somehow. Do we need to ignore them? */
+ }
+ }
+ }
+
+ if( pFirstEntry )
+ free( pFirstEntry );
+ if (ret > ret_buf)
+ ret[-1] = '\0';
+ return count;
}
- if (strlen(fi->DnsServerList.IpAddress.String) > 0 &&
- inet_addr(fi->DnsServerList.IpAddress.String) != INADDR_NONE &&
- left > ip_size)
+ else
{
- ret += sprintf (ret, "%s,", fi->DnsServerList.IpAddress.String);
- left -= ret - ret_buf;
- count++;
- }
+ FIXED_INFO *fi, *newfi;
+ DWORD size = sizeof (*fi);
+ IP_ADDR_STRING *ipAddr;
+ int i;
+ int debug = 0;
+ HRESULT res;
+
+ fi = malloc(size);
+ if (!fi)
+ return 0;
+
+ res = (*ares_fpGetNetworkParams) (fi, &size);
+ if ((res != ERROR_BUFFER_OVERFLOW) && (res != ERROR_SUCCESS))
+ goto quit;
+
+ newfi = realloc(fi, size);
+ if (!newfi)
+ goto quit;
+
+ fi = newfi;
+ res = (*ares_fpGetNetworkParams) (fi, &size);
+ if (res != ERROR_SUCCESS)
+ goto quit;
- for (i = 0, ipAddr = fi->DnsServerList.Next; ipAddr && left > ip_size;
- ipAddr = ipAddr->Next, i++)
- {
- if (inet_addr(ipAddr->IpAddress.String) != INADDR_NONE)
+ if (debug)
{
- ret += sprintf (ret, "%s,", ipAddr->IpAddress.String);
- left -= ret - ret_buf;
- count++;
+ printf ("Host Name: %s\n", fi->HostName);
+ printf ("Domain Name: %s\n", fi->DomainName);
+ printf ("DNS Servers:\n"
+ " %s (primary)\n", fi->DnsServerList.IpAddress.String);
+ }
+ if (strlen(fi->DnsServerList.IpAddress.String) > 0 &&
+ inet_addr(fi->DnsServerList.IpAddress.String) != INADDR_NONE &&
+ left > ipv4_size)
+ {
+ ret += sprintf (ret, "%s,", fi->DnsServerList.IpAddress.String);
+ left -= ret - ret_buf;
+ ++count;
+ }
+
+ for (i = 0, ipAddr = fi->DnsServerList.Next; ipAddr && left > ipv4_size;
+ ipAddr = ipAddr->Next, i++)
+ {
+ if (inet_addr(ipAddr->IpAddress.String) != INADDR_NONE)
+ {
+ ret += sprintf (ret, "%s,", ipAddr->IpAddress.String);
+ left -= ret - ret_buf;
+ ++count;
+ }
+ if (debug)
+ printf (" %s (secondary %d)\n", ipAddr->IpAddress.String, i+1);
}
- if (debug)
- printf (" %s (secondary %d)\n", ipAddr->IpAddress.String, i+1);
- }
quit:
- if (fi)
- free(fi);
-
- if (debug && left <= ip_size)
- printf ("Too many nameservers. Truncating to %d addressess", count);
- if (ret > ret_buf)
- ret[-1] = '\0';
- return count;
+ if (fi)
+ free(fi);
+
+ if (debug && left <= ipv4_size)
+ printf ("Too many nameservers. Truncating to %d addressess", count);
+ if (ret > ret_buf)
+ ret[-1] = '\0';
+ return count;
+ }
}
#endif
@@ -704,7 +827,7 @@ DhcpNameServer
DWORD data_type;
DWORD bytes;
DWORD result;
- char buf[256];
+ char buf[512];
win_platform platform;
if (channel->nservers > -1) /* don't override ARES_OPT_SERVER */