mirror of
https://github.com/matrix-construct/construct
synced 2024-08-24 18:42:16 +02:00
279 lines
7.2 KiB
C++
279 lines
7.2 KiB
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/rb.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 = (FIXED_INFO *)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 = (FIXED_INFO *)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 = (char *)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
|