system/wine
system/wine
1
0
Fork 0
mirror of https://github.com/wine-mirror/wine synced 2024-07-03 08:19:41 +00:00
Code Issues Projects Releases Wiki Activity
master
wine/dlls/dnsapi/tests/query.c

378 lines
15 KiB
C
Raw Permalink Blame History

/*
* Copyright 2020 Dmitry Timoshkov
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
*/
#include <stdarg.h>
#include <stdio.h>
#include "windef.h"
#include "winbase.h"
#include "winnls.h"
#include "windns.h"
#include "winsock2.h"
#include "ws2ipdef.h"
#include "ws2tcpip.h"
#include "iphlpapi.h"
#include "wine/test.h"
#define NS_MAXDNAME 1025
static void dump_dns_records(DNS_RECORDW *rec)
{
while (rec)
{
if (rec->wType == DNS_TYPE_CNAME)
printf(" %s type=CNAME -> %s\n", wine_dbgstr_w(rec->pName), wine_dbgstr_w(rec->Data.CNAME.pNameHost));
else if (rec->wType == DNS_TYPE_A)
{
char buf[16];
const char *addr;
addr = inet_ntop(AF_INET, &rec->Data.A.IpAddress, buf, sizeof(buf));
printf(" %s type=A -> %s\n", wine_dbgstr_w(rec->pName), addr ? addr : "(unknown IPv4 address)");
}
else if (rec->wType == DNS_TYPE_AAAA)
{
char buf[46];
const char *addr;
addr = inet_ntop(AF_INET6, &rec->Data.AAAA.Ip6Address, buf, sizeof(buf));
printf(" %s type=AAAA -> %s\n", wine_dbgstr_w(rec->pName), addr ? addr : "(unknown IPv6 address)");
}
else
printf(" %s type=%d\n", wine_dbgstr_w(rec->pName), rec->wType);
rec = rec->pNext;
}
}
static void test_DnsQuery(void)
{
WCHAR domain[MAX_PATH];
WCHAR name[NS_MAXDNAME];
DWORD ret, size;
DNS_RECORDW *rec, *ptr;
DNS_STATUS status;
/* IP in name. */
status = DnsQuery_W(L" 192.168.111.11", DNS_TYPE_A, 0, NULL, &rec, NULL);
ok(status != ERROR_SUCCESS, "got %lu.\n", status);
status = DnsQuery_W(L"192.168.111.11 ", DNS_TYPE_A, 0, NULL, &rec, NULL);
ok(status != ERROR_SUCCESS, "got %lu.\n", status);
status = DnsQuery_W(L"192.168.111.11", DNS_TYPE_A, 0, NULL, &rec, NULL);
ok(!status, "got %lu.\n", status);
ok(rec->wType == DNS_TYPE_A, "got %#x.\n", rec->wType);
ok(rec->wDataLength == sizeof(rec->Data.A), "got %u.\n", rec->wDataLength);
ok(rec->Data.A.IpAddress == 0x0b6fa8c0, "got %#lx.\n", rec->Data.A.IpAddress);
ok(!rec->pNext, "got %p.\n", rec->pNext);
ok(rec->dwTtl == 604800, "got %lu.\n", rec->dwTtl);
ok(!wcscmp(rec->pName, L"192.168.111.11"), "got %s.\n", debugstr_w(rec->pName));
ok(!rec->Flags.S.Section, "got %u.\n", rec->Flags.S.Section);
ok(!rec->Flags.S.Delete, "got %u.\n", rec->Flags.S.Delete);
ok(rec->Flags.S.CharSet == DnsCharSetUnicode, "got %u.\n", rec->Flags.S.CharSet);
ok(!rec->Flags.S.Unused, "got %u.\n", rec->Flags.S.Unused);
ok(rec->Flags.S.Reserved == 0x20, "got %u.\n", rec->Flags.S.Reserved);
DnsRecordListFree(rec, DnsFreeRecordList);
status = DnsQuery_W(L"2001:db8:3333:4444:5555:6666:7777:8888", DNS_TYPE_AAAA, 0, NULL, &rec, NULL);
ok(!status, "got %lu.\n", status);
ok(rec->wType == DNS_TYPE_AAAA, "got %#x.\n", rec->wType);
ok(rec->wDataLength == sizeof(rec->Data.AAAA), "got %u.\n", rec->wDataLength);
ok(rec->Data.AAAA.Ip6Address.IP6Dword[0] == 0xb80d0120, "got %#lx.\n", rec->Data.AAAA.Ip6Address.IP6Dword[0]);
ok(rec->Data.AAAA.Ip6Address.IP6Dword[1] == 0x44443333, "got %#lx.\n", rec->Data.AAAA.Ip6Address.IP6Dword[1]);
ok(rec->Data.AAAA.Ip6Address.IP6Dword[2] == 0x66665555, "got %#lx.\n", rec->Data.AAAA.Ip6Address.IP6Dword[2]);
ok(rec->Data.AAAA.Ip6Address.IP6Dword[3] == 0x88887777, "got %#lx.\n", rec->Data.AAAA.Ip6Address.IP6Dword[3]);
ok(!rec->pNext, "got %p.\n", rec->pNext);
ok(rec->dwTtl == 604800, "got %lu.\n", rec->dwTtl);
ok(!wcscmp(rec->pName, L"2001:db8:3333:4444:5555:6666:7777:8888"), "got %s.\n", debugstr_w(rec->pName));
ok(!rec->Flags.S.Section, "got %u.\n", rec->Flags.S.Section);
ok(!rec->Flags.S.Delete, "got %u.\n", rec->Flags.S.Delete);
ok(rec->Flags.S.CharSet == DnsCharSetUnicode, "got %u.\n", rec->Flags.S.CharSet);
ok(!rec->Flags.S.Unused, "got %u.\n", rec->Flags.S.Unused);
ok(rec->Flags.S.Reserved == 0x20, "got %u.\n", rec->Flags.S.Reserved);
DnsRecordListFree(rec, DnsFreeRecordList);
rec = NULL;
status = DnsQuery_W(L"winehq.org", DNS_TYPE_A, DNS_QUERY_STANDARD, NULL, &rec, NULL);
if (status == ERROR_TIMEOUT)
{
skip("query timed out\n");
return;
}
ok(status == ERROR_SUCCESS, "got %ld\n", status);
DnsRecordListFree(rec, DnsFreeRecordList);
/* Show that DNS_TYPE_A returns CNAMEs too */
rec = NULL;
wcscpy(domain, L"test.winehq.org"); /* should be a CNAME */
status = DnsQuery_W(domain, DNS_TYPE_A, DNS_QUERY_STANDARD, NULL, &rec, NULL);
if (status == ERROR_TIMEOUT)
{
skip("query timed out\n");
return;
}
ok(status == ERROR_SUCCESS, "got %ld\n", status);
if (status == ERROR_SUCCESS && winetest_debug > 1)
{
trace("DnsQuery DNS_TYPE_A:\n");
dump_dns_records(rec);
}
ptr = rec; /* CNAMEs come first */
ok(!wcscmp(domain, ptr->pName), "expected record name %s, got %s\n", wine_dbgstr_w(domain), wine_dbgstr_w(ptr->pName));
ok(DNS_TYPE_CNAME == ptr->wType, "expected record type %d, got %d\n", DNS_TYPE_CNAME, ptr->wType);
wcscpy(domain, L"winehq.org");
if (ptr->wType == DNS_TYPE_CNAME)
ok(!wcscmp(domain, ptr->Data.CNAME.pNameHost), "expected CNAME target %s, got %s\n", wine_dbgstr_w(domain), wine_dbgstr_w(ptr->Data.CNAME.pNameHost));
ptr = ptr->pNext;
ok(!wcscmp(domain, ptr->pName), "expected record name %s, got %s\n",
wine_dbgstr_w(domain), wine_dbgstr_w(ptr->pName));
ok(DNS_TYPE_A == ptr->wType, "expected record type %d, got %d\n",
DNS_TYPE_A, ptr->wType);
ptr = ptr->pNext;
while (ptr)
{
ok(wcscmp(domain, ptr->pName), "did not expect a record for %s\n",
wine_dbgstr_w(ptr->pName));
ok(ptr->wType == DNS_TYPE_A || ptr->wType == DNS_TYPE_AAAA || ptr->wType == DNS_TYPE_OPT,
"unexpected record type %d\n", ptr->wType);
ptr = ptr->pNext;
}
DnsRecordListFree(rec, DnsFreeRecordList);
/* DNS_TYPE_CNAME does not return a record for its target but it may still
* return other related records!
*/
rec = NULL;
wcscpy(domain, L"test.winehq.org");
status = DnsQuery_W(L"test.winehq.org", DNS_TYPE_CNAME, DNS_QUERY_STANDARD, NULL, &rec, NULL);
if (status == ERROR_TIMEOUT)
{
skip("query timed out\n");
return;
}
if (status == DNS_INFO_NO_RECORDS)
{
skip("no CNAME records\n");
return;
}
ok(status == ERROR_SUCCESS, "got %ld\n", status);
if (status == ERROR_SUCCESS && winetest_debug > 1)
{
trace("DnsQuery DNS_TYPE_CNAME:\n");
dump_dns_records(rec);
}
ptr = rec;
ok(!wcscmp(domain, ptr->pName), "expected record name %s, got %s\n", wine_dbgstr_w(domain), wine_dbgstr_w(ptr->pName));
ok(DNS_TYPE_CNAME == ptr->wType, "expected record type %d, got %d\n", DNS_TYPE_CNAME, ptr->wType);
wcscpy(domain, L"winehq.org");
if (ptr->wType == DNS_TYPE_CNAME)
ok(!wcscmp(domain, ptr->Data.CNAME.pNameHost), "expected CNAME target %s, got %s\n", wine_dbgstr_w(domain), wine_dbgstr_w(ptr->Data.CNAME.pNameHost));
DnsRecordListFree(rec, DnsFreeRecordList);
status = DnsQuery_W(L"", DNS_TYPE_SRV, DNS_QUERY_STANDARD, NULL, &rec, NULL);
ok(status == DNS_ERROR_RCODE_NAME_ERROR || status == DNS_INFO_NO_RECORDS || status == ERROR_INVALID_NAME /* XP */,
"got %ld\n", status);
wcscpy(domain, L"_ldap._tcp.deadbeef");
status = DnsQuery_W(domain, DNS_TYPE_SRV, DNS_QUERY_STANDARD, NULL, &rec, NULL);
ok(status == DNS_ERROR_RCODE_NAME_ERROR || status == DNS_INFO_NO_RECORDS || status == ERROR_INVALID_NAME /* XP */,
"got %ld\n", status);
wcscpy(domain, L"_ldap._tcp.dc._msdcs.");
size = ARRAY_SIZE(domain) - wcslen(domain);
ret = GetComputerNameExW(ComputerNameDnsDomain, domain + wcslen(domain), &size);
ok(ret, "GetComputerNameEx error %lu\n", GetLastError());
if (!size)
{
skip("computer is not in a domain\n");
return;
}
status = DnsQuery_W(domain, DNS_TYPE_SRV, DNS_QUERY_STANDARD, NULL, &rec, NULL);
trace("DnsQuery_W(%s) => %ld\n", wine_dbgstr_w(domain), status);
if (status != ERROR_SUCCESS)
{
skip("domain %s doesn't have an SRV entry\n", wine_dbgstr_w(domain));
return;
}
trace("target %s, port %d\n", wine_dbgstr_w(rec->Data.Srv.pNameTarget), rec->Data.Srv.wPort);
lstrcpynW(name, rec->Data.Srv.pNameTarget, ARRAY_SIZE(name));
DnsRecordListFree(rec, DnsFreeRecordList);
/* IPv4 */
status = DnsQuery_W(name, DNS_TYPE_A, DNS_QUERY_STANDARD, NULL, &rec, NULL);
ok(status == ERROR_SUCCESS || status == DNS_ERROR_RCODE_NAME_ERROR, "DnsQuery_W(%s) => %ld\n",
wine_dbgstr_w(name), status);
if (status == ERROR_SUCCESS)
{
SOCKADDR_IN addr;
WCHAR buf[IP4_ADDRESS_STRING_LENGTH];
addr.sin_family = AF_INET;
addr.sin_port = 0;
addr.sin_addr.s_addr = rec->Data.A.IpAddress;
size = sizeof(buf);
ret = WSAAddressToStringW((SOCKADDR *)&addr, sizeof(addr), NULL, buf, &size);
ok(!ret, "WSAAddressToStringW error %lu\n", ret);
trace("WSAAddressToStringW => %s\n", wine_dbgstr_w(buf));
DnsRecordListFree(rec, DnsFreeRecordList);
}
/* IPv6 */
status = DnsQuery_W(name, DNS_TYPE_AAAA, DNS_QUERY_STANDARD, NULL, &rec, NULL);
ok(status == ERROR_SUCCESS || status == DNS_ERROR_RCODE_NAME_ERROR, "DnsQuery_W(%s) => %ld\n",
wine_dbgstr_w(name), status);
if (status == ERROR_SUCCESS)
{
SOCKADDR_IN6 addr;
WCHAR buf[IP6_ADDRESS_STRING_LENGTH];
addr.sin6_family = AF_INET6;
addr.sin6_port = 0;
addr.sin6_scope_id = 0;
memcpy(addr.sin6_addr.s6_addr, &rec->Data.AAAA.Ip6Address, sizeof(rec->Data.AAAA.Ip6Address));
size = sizeof(buf);
ret = WSAAddressToStringW((SOCKADDR *)&addr, sizeof(addr), NULL, buf, &size);
ok(!ret, "WSAAddressToStringW error %lu\n", ret);
trace("WSAAddressToStringW => %s\n", wine_dbgstr_w(buf));
DnsRecordListFree(rec, DnsFreeRecordList);
}
}
static IP_ADAPTER_ADDRESSES *get_adapters(void)
{
ULONG err, size = 1024;
IP_ADAPTER_ADDRESSES *ret = malloc( size );
for (;;)
{
err = GetAdaptersAddresses( AF_UNSPEC, GAA_FLAG_SKIP_ANYCAST | GAA_FLAG_SKIP_MULTICAST |
GAA_FLAG_SKIP_FRIENDLY_NAME,
NULL, ret, &size );
if (err != ERROR_BUFFER_OVERFLOW) break;
ret = realloc( ret, size );
}
if (err == ERROR_SUCCESS) return ret;
free( ret );
return NULL;
}
static void test_DnsQueryConfig( void )
{
DNS_STATUS err;
DWORD size, i, ipv6_count;
WCHAR name[MAX_ADAPTER_NAME_LENGTH + 1];
IP_ADAPTER_ADDRESSES *adapters, *ptr;
DNS_ADDR_ARRAY *ipv4, *ipv6, *unspec;
IP4_ARRAY *ip4_array;
if (!(adapters = get_adapters())) return;
for (ptr = adapters; ptr; ptr = ptr->Next)
{
MultiByteToWideChar( CP_ACP, 0, ptr->AdapterName, -1, name, ARRAY_SIZE(name) );
if (ptr->IfType == IF_TYPE_SOFTWARE_LOOPBACK) continue;
size = 0;
err = DnsQueryConfig( DnsConfigDnsServersIpv4, 0, name, NULL, NULL, &size );
if (err) continue;
ipv4 = malloc( size );
size--;
err = DnsQueryConfig( DnsConfigDnsServersIpv4, 0, name, NULL, ipv4, &size );
ok( err == ERROR_MORE_DATA, "got %ld\n", err );
size++;
err = DnsQueryConfig( DnsConfigDnsServersIpv4, 0, name, NULL, ipv4, &size );
ok( !err, "got %ld\n", err );
ok( ipv4->AddrCount == ipv4->MaxCount, "got %lu vs %lu\n", ipv4->AddrCount, ipv4->MaxCount );
ok( !ipv4->Tag, "got %#lx\n", ipv4->Tag );
ok( !ipv4->Family, "got %d\n", ipv4->Family );
ok( !ipv4->WordReserved, "got %#x\n", ipv4->WordReserved );
ok( !ipv4->Flags, "got %#lx\n", ipv4->Flags );
ok( !ipv4->MatchFlag, "got %#lx\n", ipv4->MatchFlag );
ok( !ipv4->Reserved1, "got %#lx\n", ipv4->Reserved1 );
ok( !ipv4->Reserved2, "got %#lx\n", ipv4->Reserved2 );
size = 0;
err = DnsQueryConfig( DnsConfigDnsServerList, 0, name, NULL, NULL, &size );
ok( !err, "got %ld\n", err );
ip4_array = malloc( size );
err = DnsQueryConfig( DnsConfigDnsServerList, 0, name, NULL, ip4_array, &size );
ok( !err, "got %ld\n", err );
ok( ipv4->AddrCount == ip4_array->AddrCount, "got %lu vs %lu\n", ipv4->AddrCount, ip4_array->AddrCount );
for (i = 0; i < ipv4->AddrCount; i++)
{
SOCKADDR_IN *sa = (SOCKADDR_IN *)ipv4->AddrArray[i].MaxSa;
ok( sa->sin_family == AF_INET, "got %d\n", sa->sin_family );
ok( sa->sin_addr.s_addr == ip4_array->AddrArray[i], "got %#lx vs %#lx\n",
sa->sin_addr.s_addr, ip4_array->AddrArray[i] );
ok( ipv4->AddrArray[i].Data.DnsAddrUserDword[0] == sizeof(*sa), "got %lu\n",
ipv4->AddrArray[i].Data.DnsAddrUserDword[0] );
}
size = 0;
err = DnsQueryConfig( DnsConfigDnsServersIpv6, 0, name, NULL, NULL, &size );
ok( !err || err == DNS_ERROR_NO_DNS_SERVERS, "got %ld\n", err );
ipv6_count = 0;
ipv6 = NULL;
if (!err)
{
ipv6 = malloc( size );
err = DnsQueryConfig( DnsConfigDnsServersIpv6, 0, name, NULL, ipv6, &size );
ok( !err, "got %ld\n", err );
ipv6_count = ipv6->AddrCount;
}
size = 0;
err = DnsQueryConfig( DnsConfigDnsServersUnspec, 0, name, NULL, NULL, &size );
ok( !err, "got %ld\n", err );
unspec = malloc( size );
err = DnsQueryConfig( DnsConfigDnsServersUnspec, 0, name, NULL, unspec, &size );
ok( !err, "got %ld\n", err );
ok( unspec->AddrCount == ipv4->AddrCount + ipv6_count, "got %lu vs %lu + %lu\n",
unspec->AddrCount, ipv4->AddrCount, ipv6_count );
free( ip4_array );
free( unspec );
free( ipv6 );
free( ipv4 );
}
free( adapters );
}
START_TEST(query)
{
WSADATA data;
WSAStartup(MAKEWORD(2, 2), &data);
test_DnsQuery();
test_DnsQueryConfig();
}
Reference in New Issue View Git Blame Copy Permalink