1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
/*
* reslist.c - get nameservers from windows *
*
* Copyright 1998 by the Massachusetts Institute of Technology.
* Copyright (C) 2007-2008 by Daniel Stenberg
*
* Permission to use, copy, modify, and distribute this
* software and its documentation for any purpose and without
* fee is hereby granted, provided that the above copyright
* notice appear in all copies and that both that copyright
* notice and this permission notice appear in supporting
* documentation, and that the name of M.I.T. not be used in
* advertising or publicity pertaining to distribution of the
* software without specific, written prior permission.
* M.I.T. makes no representations about the suitability of
* this software for any purpose. It is provided "as is"
* without express or implied warranty.
*/
#ifdef _WIN32
#include "mowgli.h"
#include <windows.h>
#include <iphlpapi.h>
int
mowgli_dns_get_windows_nameservers(char *ret_buf, size_t ret_size)
{
FIXED_INFO *fixedinfo, tfixedinfo;
DWORD size = sizeof(*fixedinfo);
typedef DWORD(WINAPI * get_net_param_func) (FIXED_INFO *, DWORD *);
get_net_param_func get_network_params;
HMODULE handle;
IP_ADDR_STRING *ip_addr;
int i, count = 0;
size_t ip_size = sizeof("255.255.255.255");
size_t left = ret_size;
char *ret = ret_buf;
HRESULT res;
if (!(handle = LoadLibrary("iphlpapi.dll")))
return 0;
if (!(get_network_params = (get_net_param_func)GetProcAddress(handle, "GetNetworkParams")))
goto quit;
res = (*get_network_params)(&tfixedinfo, &size);
if((res != ERROR_BUFFER_OVERFLOW) && (res != ERROR_SUCCESS))
goto quit;
fixedinfo = mowgli_alloc(size);
if(!fixedinfo || (*get_network_params)(fixedinfo, &size) != ERROR_SUCCESS)
goto quit;
#ifdef DEBUG
mowgli_log("Host Name: %s\n", fixedinfo->HostName);
mowgli_log("Domain Name: %s\n", fixedinfo->DomainName);
mowgli_log("DNS Servers:\n\t%s (primary)\n", fixedinfo->DnsServerList.IpAddress.String);
#endif
if(strlen(fixedinfo->DnsServerList.IpAddress.String) > 0 &&
inet_addr(fixedinfo->DnsServerList.IpAddress.String) != INADDR_NONE && left > ip_size)
{
ret += sprintf(ret, "%s,", fixedinfo->DnsServerList.IpAddress.String);
left -= ret - ret_buf;
count++;
}
for(i = 0, ip_addr = fixedinfo->DnsServerList.Next; ip_addr && left > ip_size; ip_addr = ip_addr->Next, i++)
{
if(inet_addr(ip_addr->IpAddress.String) != INADDR_NONE)
{
ret += sprintf(ret, "%s,", ip_addr->IpAddress.String);
left -= ret - ret_buf;
count++;
}
#ifdef DEBUG
mowgli_log("\t%s (secondary %d)\n", ip_addr->IpAddress.String, i + 1);
#endif
}
mowgli_free(fixedinfo);
quit:
if(handle)
FreeLibrary(handle);
if(left <= ip_size)
mowgli_log("Too many nameservers. Truncating to %d addressess", count);
if(ret > ret_buf)
ret[-1] = '\0';
return count;
}
#endif
|
