[solanum.git] / authd / reslist.c

/*
 * reslist.c - get nameservers from windows *
 *
 * ircd-ratbox related changes are as follows
 *
 * Copyright (C) 2008 Aaron Sethman <androsyn@ratbox.org>
 * Copyright (C) 2008-2012 ircd-ratbox development team
 *
 * pretty much all of this was yanked from c-ares ares_init.c here is the original
 * header from there --
 *
 * Id: ares_init.c,v 1.72 2008-05-15 00:00:19 yangtse Exp $
 * 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 <rb_lib.h>

#include <windows.h>
#include <iphlpapi.h>

const char *get_windows_nameservers(void);

#ifndef INADDR_NONE
#define INADDR_NONE ((unsigned int) 0xffffffff)
#endif /* INADDR_NONE */

#define IS_NT()	       ((int)GetVersion() > 0)
#define WIN_NS_9X      "System\\CurrentControlSet\\Services\\VxD\\MSTCP"
#define WIN_NS_NT_KEY  "System\\CurrentControlSet\\Services\\Tcpip\\Parameters"
#define NAMESERVER     "NameServer"
#define DHCPNAMESERVER "DhcpNameServer"
#define DATABASEPATH   "DatabasePath"
#define WIN_PATH_HOSTS	"\\hosts"

static int
get_iphlpapi_dns_info(char *ret_buf, size_t ret_size)
{
	FIXED_INFO *fi = alloca(sizeof(*fi));
	DWORD size = sizeof(*fi);
	typedef DWORD(WINAPI * get_net_param_func) (FIXED_INFO *, DWORD *);
	get_net_param_func xxGetNetworkParams;	/* available only on Win-98/2000+ */
	HMODULE handle;
	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;

	if(!fi)
		return (0);

	handle = LoadLibrary("iphlpapi.dll");
	if(!handle)
		return (0);

	xxGetNetworkParams = (get_net_param_func) GetProcAddress(handle, "GetNetworkParams");
	if(!xxGetNetworkParams)
		goto quit;

	res = (*xxGetNetworkParams) (fi, &size);
	if((res != ERROR_BUFFER_OVERFLOW) && (res != ERROR_SUCCESS))
		goto quit;

	fi = alloca(size);
	if(!fi || (*xxGetNetworkParams) (fi, &size) != ERROR_SUCCESS)
		goto quit;

	if(debug)
	{
		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 > ip_size)
	{
		ret += sprintf(ret, "%s,", fi->DnsServerList.IpAddress.String);
		left -= ret - ret_buf;
		count++;
	}

	for(i = 0, ipAddr = fi->DnsServerList.Next; ipAddr && left > ip_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);
	}

      quit:
	if(handle)
		FreeLibrary(handle);

	if(debug && left <= ip_size)
		printf("Too many nameservers. Truncating to %d addressess", count);
	if(ret > ret_buf)
		ret[-1] = '\0';
	return (count);
}

/*
 * Warning: returns a dynamically allocated buffer, the user MUST
 * use free() / rb_free() if the function returns 1
 */
static int
get_res_nt(HKEY hKey, const char *subkey, char **obuf)
{
	/* Test for the size we need */
	DWORD size = 0;
	int result;

	result = RegQueryValueEx(hKey, subkey, 0, NULL, NULL, &size);
	if((result != ERROR_SUCCESS && result != ERROR_MORE_DATA) || !size)
		return 0;
	*obuf = rb_malloc(size + 1);
	if(!*obuf)
		return 0;

	if(RegQueryValueEx(hKey, subkey, 0, NULL, (LPBYTE) * obuf, &size) != ERROR_SUCCESS)
	{
		rb_free(*obuf);
		return 0;
	}
	if(size == 1)
	{
		rb_free(*obuf);
		return 0;
	}
	return 1;
}

static int
get_res_interfaces_nt(HKEY hKey, const char *subkey, char **obuf)
{
	char enumbuf[39];	/* GUIDs are 38 chars + 1 for NULL */
	DWORD enum_size = 39;
	int idx = 0;
	HKEY hVal;

	while(RegEnumKeyEx(hKey, idx++, enumbuf, &enum_size, 0,
			   NULL, NULL, NULL) != ERROR_NO_MORE_ITEMS)
	{
		int rc;

		enum_size = 39;
		if(RegOpenKeyEx(hKey, enumbuf, 0, KEY_QUERY_VALUE, &hVal) != ERROR_SUCCESS)
			continue;
		rc = get_res_nt(hVal, subkey, obuf);
		RegCloseKey(hVal);
		if(rc)
			return 1;
	}
	return 0;
}

const char *
get_windows_nameservers(void)
{
	/*
	   NameServer info via IPHLPAPI (IP helper API):
	   GetNetworkParams() should be the trusted source for this.
	   Available in Win-98/2000 and later. If that fail, fall-back to
	   registry information.

	   NameServer Registry:

	   On Windows 9X, the DNS server can be found in:
	   HKEY_LOCAL_MACHINE\System\CurrentControlSet\Services\VxD\MSTCP\NameServer

	   On Windows NT/2000/XP/2003:
	   HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\NameServer
	   or
	   HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\DhcpNameServer
	   or
	   HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\{AdapterID}\
	   NameServer
	   or
	   HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\{AdapterID}\
	   DhcpNameServer
	 */
	static char namelist[512];
	HKEY mykey;
	HKEY subkey;
	DWORD data_type;
	DWORD bytes;
	DWORD result;
	char *line = NULL;
	memset(&namelist, 0, sizeof(namelist));
	if(get_iphlpapi_dns_info(namelist, sizeof(namelist)) > 0)
	{
		return namelist;
	}

	if(IS_NT())
	{
		if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_NT_KEY, 0,
				KEY_READ, &mykey) == ERROR_SUCCESS)
		{
			RegOpenKeyEx(mykey, "Interfaces", 0,
				     KEY_QUERY_VALUE | KEY_ENUMERATE_SUB_KEYS, &subkey);
			if(get_res_nt(mykey, NAMESERVER, &line))
			{
				rb_strlcpy(namelist, line, sizeof(namelist));
				return namelist;
			}
			else if(get_res_nt(mykey, DHCPNAMESERVER, &line))
			{
				rb_strlcpy(namelist, line, sizeof(namelist));
				rb_free(line);
			}
			/* Try the interfaces */
			else if(get_res_interfaces_nt(subkey, NAMESERVER, &line))
			{
				rb_strlcpy(namelist, line, sizeof(namelist));
				rb_free(line);
			}
			else if(get_res_interfaces_nt(subkey, DHCPNAMESERVER, &line))
			{
				rb_strlcpy(namelist, line, sizeof(namelist));
				rb_free(line);
			}
			RegCloseKey(subkey);
			RegCloseKey(mykey);
		}
	}
	else
	{
		if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_9X, 0,
				KEY_READ, &mykey) == ERROR_SUCCESS)
		{
			if((result = RegQueryValueEx(mykey, NAMESERVER, NULL, &data_type,
						     NULL, &bytes)) == ERROR_SUCCESS ||
			   result == ERROR_MORE_DATA)
			{
				if(bytes)
				{
					line = (char *)rb_malloc(bytes + 1);
					if(RegQueryValueEx(mykey, NAMESERVER, NULL, &data_type,
							   (unsigned char *)line, &bytes) ==
					   ERROR_SUCCESS)
					{
						rb_strlcpy(namelist, line, sizeof(namelist));
					}
					free(line);
				}
			}
		}
		RegCloseKey(mykey);
	}
	if(strlen(namelist) > 0)
		return namelist;
	return NULL;
}


#endif
Commit	Line	Data
76ebf6c4 AC	1	/*
76ebf6c4 AC	2	* reslist.c - get nameservers from windows *
4094d2fa	3	*
76ebf6c4	4	* ircd-ratbox related changes are as follows
4094d2fa	5	*
76ebf6c4 AC	6	* Copyright (C) 2008 Aaron Sethman <androsyn@ratbox.org>
76ebf6c4 AC	7	* Copyright (C) 2008-2012 ircd-ratbox development team
4094d2fa EM	8	*
4094d2fa EM	9	* pretty much all of this was yanked from c-ares ares_init.c here is the original
76ebf6c4 AC	10	* header from there --
76ebf6c4 AC	11	*
4094d2fa	12	* Id: ares_init.c,v 1.72 2008-05-15 00:00:19 yangtse Exp $
76ebf6c4 AC	13	* Copyright 1998 by the Massachusetts Institute of Technology.
	14	* Copyright (C) 2007-2008 by Daniel Stenberg
	15	*
	16	* Permission to use, copy, modify, and distribute this
	17	* software and its documentation for any purpose and without
	18	* fee is hereby granted, provided that the above copyright
	19	* notice appear in all copies and that both that copyright
	20	* notice and this permission notice appear in supporting
	21	* documentation, and that the name of M.I.T. not be used in
	22	* advertising or publicity pertaining to distribution of the
	23	* software without specific, written prior permission.
	24	* M.I.T. makes no representations about the suitability of
	25	* this software for any purpose. It is provided "as is"
	26	* without express or implied warranty.
	27	*
	28	*
	29	*/
	30
	31	#ifdef _WIN32
2dea53a2	32	#include <rb_lib.h>
76ebf6c4 AC	33
	34	#include <windows.h>
	35	#include <iphlpapi.h>
	36
	37	const char *get_windows_nameservers(void);
	38
b143df9a AJ	39	#ifndef INADDR_NONE
	40	#define INADDR_NONE ((unsigned int) 0xffffffff)
	41	#endif /* INADDR_NONE */
76ebf6c4 AC	42
	43	#define IS_NT() ((int)GetVersion() > 0)
	44	#define WIN_NS_9X "System\\CurrentControlSet\\Services\\VxD\\MSTCP"
	45	#define WIN_NS_NT_KEY "System\\CurrentControlSet\\Services\\Tcpip\\Parameters"
	46	#define NAMESERVER "NameServer"
	47	#define DHCPNAMESERVER "DhcpNameServer"
	48	#define DATABASEPATH "DatabasePath"
	49	#define WIN_PATH_HOSTS "\\hosts"
	50
	51	static int
	52	get_iphlpapi_dns_info(char *ret_buf, size_t ret_size)
	53	{
	54	FIXED_INFO fi = alloca(sizeof(fi));
	55	DWORD size = sizeof(*fi);
	56	typedef DWORD(WINAPI * get_net_param_func) (FIXED_INFO , DWORD );
	57	get_net_param_func xxGetNetworkParams; /* available only on Win-98/2000+ */
	58	HMODULE handle;
	59	IP_ADDR_STRING *ipAddr;
	60	int i, count = 0;
	61	int debug = 0;
	62	size_t ip_size = sizeof("255.255.255.255,") - 1;
	63	size_t left = ret_size;
	64	char *ret = ret_buf;
	65	HRESULT res;
	66
	67	if(!fi)
	68	return (0);
	69
	70	handle = LoadLibrary("iphlpapi.dll");
	71	if(!handle)
	72	return (0);
	73
	74	xxGetNetworkParams = (get_net_param_func) GetProcAddress(handle, "GetNetworkParams");
	75	if(!xxGetNetworkParams)
	76	goto quit;
	77
	78	res = (*xxGetNetworkParams) (fi, &size);
	79	if((res != ERROR_BUFFER_OVERFLOW) && (res != ERROR_SUCCESS))
	80	goto quit;
	81
	82	fi = alloca(size);
	83	if(!fi \|\| (*xxGetNetworkParams) (fi, &size) != ERROR_SUCCESS)
	84	goto quit;
	85
	86	if(debug)
	87	{
	88	printf("Host Name: %s\n", fi->HostName);
	89	printf("Domain Name: %s\n", fi->DomainName);
	90	printf("DNS Servers:\n" " %s (primary)\n", fi->DnsServerList.IpAddress.String);
	91	}
	92	if(strlen(fi->DnsServerList.IpAddress.String) > 0 &&
	93	inet_addr(fi->DnsServerList.IpAddress.String) != INADDR_NONE && left > ip_size)
	94	{
	95	ret += sprintf(ret, "%s,", fi->DnsServerList.IpAddress.String);
	96	left -= ret - ret_buf;
	97	count++;
	98	}
	99
	100	for(i = 0, ipAddr = fi->DnsServerList.Next; ipAddr && left > ip_size;
	101	ipAddr = ipAddr->Next, i++)
	102	{
	103	if(inet_addr(ipAddr->IpAddress.String) != INADDR_NONE)
	104	{
	105	ret += sprintf(ret, "%s,", ipAddr->IpAddress.String);
106	left -= ret - ret_buf;
107	count++;
108	}
109	if(debug)
110	printf(" %s (secondary %d)\n", ipAddr->IpAddress.String, i + 1);
111	}
112
113	quit:
114	if(handle)
115	FreeLibrary(handle);
116
117	if(debug && left <= ip_size)
118	printf("Too many nameservers. Truncating to %d addressess", count);
119	if(ret > ret_buf)
120	ret[-1] = '\0';
121	return (count);
122	}
123
124	/*
125	* Warning: returns a dynamically allocated buffer, the user MUST
126	* use free() / rb_free() if the function returns 1
127	*/
128	static int
129	get_res_nt(HKEY hKey, const char subkey, char *obuf)
130	{
131	/* Test for the size we need */
132	DWORD size = 0;
133	int result;
134
135	result = RegQueryValueEx(hKey, subkey, 0, NULL, NULL, &size);
136	if((result != ERROR_SUCCESS && result != ERROR_MORE_DATA) \|\| !size)
137	return 0;
138	*obuf = rb_malloc(size + 1);
139	if(!*obuf)
140	return 0;
141
142	if(RegQueryValueEx(hKey, subkey, 0, NULL, (LPBYTE) * obuf, &size) != ERROR_SUCCESS)
143	{
144	rb_free(*obuf);
145	return 0;
146	}
147	if(size == 1)
148	{
149	rb_free(*obuf);
150	return 0;
151	}
152	return 1;
153	}
154
155	static int
156	get_res_interfaces_nt(HKEY hKey, const char subkey, char *obuf)
157	{
158	char enumbuf[39]; /* GUIDs are 38 chars + 1 for NULL */
159	DWORD enum_size = 39;
160	int idx = 0;
161	HKEY hVal;
162
163	while(RegEnumKeyEx(hKey, idx++, enumbuf, &enum_size, 0,
164	NULL, NULL, NULL) != ERROR_NO_MORE_ITEMS)
165	{
166	int rc;
167
168	enum_size = 39;
169	if(RegOpenKeyEx(hKey, enumbuf, 0, KEY_QUERY_VALUE, &hVal) != ERROR_SUCCESS)
170	continue;
171	rc = get_res_nt(hVal, subkey, obuf);
172	RegCloseKey(hVal);
173	if(rc)
174	return 1;
175	}
176	return 0;
177	}
178
179	const char *
180	get_windows_nameservers(void)
181	{
182	/*
183	NameServer info via IPHLPAPI (IP helper API):
184	GetNetworkParams() should be the trusted source for this.
185	Available in Win-98/2000 and later. If that fail, fall-back to
186	registry information.
187
188	NameServer Registry:
189
190	On Windows 9X, the DNS server can be found in:
191	HKEY_LOCAL_MACHINE\System\CurrentControlSet\Services\VxD\MSTCP\NameServer
192
193	On Windows NT/2000/XP/2003:
194	HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\NameServer
195	or
196	HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\DhcpNameServer
197	or
198	HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\{AdapterID}\
199	NameServer
200	or
201	HKEY_LOCAL_MACHINE\SYSTEM\CurrentControlSet\Services\Tcpip\Parameters\{AdapterID}\
202	DhcpNameServer
203	*/
204	static char namelist[512];
205	HKEY mykey;
206	HKEY subkey;
207	DWORD data_type;
208	DWORD bytes;
209	DWORD result;
210	char *line = NULL;
211	memset(&namelist, 0, sizeof(namelist));
212	if(get_iphlpapi_dns_info(namelist, sizeof(namelist)) > 0)
213	{
214	return namelist;
215	}
216
217	if(IS_NT())
218	{
219	if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_NT_KEY, 0,
220	KEY_READ, &mykey) == ERROR_SUCCESS)
221	{
222	RegOpenKeyEx(mykey, "Interfaces", 0,
223	KEY_QUERY_VALUE \| KEY_ENUMERATE_SUB_KEYS, &subkey);
224	if(get_res_nt(mykey, NAMESERVER, &line))
225	{
226	rb_strlcpy(namelist, line, sizeof(namelist));
227	return namelist;
228	}
229	else if(get_res_nt(mykey, DHCPNAMESERVER, &line))
230	{
231	rb_strlcpy(namelist, line, sizeof(namelist));
232	rb_free(line);
233	}
234	/* Try the interfaces */
235	else if(get_res_interfaces_nt(subkey, NAMESERVER, &line))
236	{
237	rb_strlcpy(namelist, line, sizeof(namelist));
238	rb_free(line);
239	}
240	else if(get_res_interfaces_nt(subkey, DHCPNAMESERVER, &line))
241	{
242	rb_strlcpy(namelist, line, sizeof(namelist));
243	rb_free(line);
244	}
245	RegCloseKey(subkey);
246	RegCloseKey(mykey);
247	}
248	}
249	else
250	{
251	if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, WIN_NS_9X, 0,
252	KEY_READ, &mykey) == ERROR_SUCCESS)
253	{
254	if((result = RegQueryValueEx(mykey, NAMESERVER, NULL, &data_type,
255	NULL, &bytes)) == ERROR_SUCCESS \|\|
256	result == ERROR_MORE_DATA)
257	{
258	if(bytes)
259	{
260	line = (char *)rb_malloc(bytes + 1);
261	if(RegQueryValueEx(mykey, NAMESERVER, NULL, &data_type,
262	(unsigned char *)line, &bytes) ==
263	ERROR_SUCCESS)
264	{
265	rb_strlcpy(namelist, line, sizeof(namelist));
266	}
267	free(line);
268	}
269	}
270	}
271	RegCloseKey(mykey);
272	}
273	if(strlen(namelist) > 0)
274	return namelist;
275	return NULL;
276	}
277
278
279	#endif