dlls/wsdapi/tests/address.c - wine - Git at Google

 /*
  * Web Services on Devices
  * Address tests
  *
  * Copyright 2017 Owen Rudge for CodeWeavers
  *
  * 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
  */

 #define COBJMACROS

 #include <winsock2.h>
 #include <ws2tcpip.h>
 #include <windows.h>

 #include "wine/test.h"
 #include "objbase.h"
 #include "wsdapi.h"

 static void CreateUdpAddress_tests(void)
 {
     IWSDUdpAddress *udpAddress = NULL, *udpAddress2 = NULL;
     IWSDTransportAddress *transportAddress = NULL;
     IWSDAddress *address = NULL;
     IUnknown *unknown;
     HRESULT rc;
     ULONG ref;

     rc = WSDCreateUdpAddress(NULL);
     ok((rc == E_POINTER) || (rc == E_INVALIDARG), "WSDCreateUDPAddress(NULL) failed: %08x\n", rc);

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUDPAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUDPAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     /* Try to query for objects */
     rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDUdpAddress, (LPVOID*)&udpAddress2);
     ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IWSDUdpAddress) failed: %08x\n", rc);

     if (rc == S_OK)
         IWSDUdpAddress_Release(udpAddress2);

     rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDTransportAddress, (LPVOID*)&transportAddress);
     ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IID_WSDTransportAddress) failed: %08x\n", rc);

     if (rc == S_OK)
         IWSDTransportAddress_Release(transportAddress);

     rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDAddress, (LPVOID*)&address);
     ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IWSDAddress) failed: %08x\n", rc);

     if (rc == S_OK)
         IWSDAddress_Release(address);

     rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IUnknown, (LPVOID*)&unknown);
     ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IID_IUnknown) failed: %08x\n", rc);

     if (rc == S_OK)
         IUnknown_Release(unknown);

     ref = IWSDUdpAddress_Release(udpAddress);
     ok(ref == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ref);
 }

 static void GetSetTransportAddress_udp_tests(void)
 {
     IWSDUdpAddress *udpAddress = NULL;
     const WCHAR ipv4Address[] = {'1','0','.','2','0','.','3','0','.','4','0',0};
     const WCHAR ipv6Address[] = {'a','a','b','b',':','c','d',':',':','a','b','c',0};
     const WCHAR ipv4AddressWithPort[] = {'1','0','.','2','0','.','3','0','.','4','0',':','1','2','4',0};
     const WCHAR ipv6AddressWithPort[] = {'[','a','a','b','b',':','c','d',':',':','a','b','c',':','5','6','7',']',':','1','2','4',0};
     const WCHAR invalidAddress[] = {'n','o','t','/','v','a','l','i','d',0};
     LPCWSTR returnedAddress = NULL;
     WSADATA wsaData;
     HRESULT rc;
     int ret;

     ret = WSAStartup(MAKEWORD(2, 2), &wsaData);
     ok(ret == 0, "WSAStartup failed: %d\n", ret);

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
     ok(returnedAddress == NULL, "GetTransportAddress returned unexpected address: %08x\n", rc);

     /* Try setting a null address */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, NULL);
     ok(rc == E_INVALIDARG, "SetTransportAddress(NULL) returned unexpected result: %08x\n", rc);

     /* Try setting an invalid address */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, invalidAddress);
     ok(rc == HRESULT_FROM_WIN32(WSAHOST_NOT_FOUND), "SetTransportAddress(invalidAddress) returned unexpected result: %08x\n", rc);

     /* Try setting an IPv4 address */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv4Address);
     ok(rc == S_OK, "SetTransportAddress(ipv4Address) failed: %08x\n", rc);

     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, NULL);
     ok(rc == E_POINTER, "GetTransportAddress(NULL) returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
     ok(lstrcmpW(returnedAddress, ipv4Address) == 0, "Returned address != ipv4Address (%s)\n", wine_dbgstr_w(returnedAddress));

     /* Try setting an IPv4 address with a port number */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv4AddressWithPort);
     ok(rc == HRESULT_FROM_WIN32(WSAHOST_NOT_FOUND), "SetTransportAddress(ipv4Address) failed: %08x\n", rc);

     /* Try setting an IPv6 address */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv6Address);
     ok(rc == S_OK, "SetTransportAddress(ipv6Address) failed: %08x\n", rc);

     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
     ok(lstrcmpW(returnedAddress, ipv6Address) == 0, "Returned address != ipv6Address (%s)\n", wine_dbgstr_w(returnedAddress));

     /* Try setting an IPv6 address with a port number */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv6AddressWithPort);
     ok(rc == S_OK, "SetTransportAddress(ipv6AddressWithPort) failed: %08x\n", rc);

     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
     todo_wine ok(lstrcmpW(returnedAddress, ipv6AddressWithPort) == 0, "Returned address != ipv6AddressWithPort (%s)\n", wine_dbgstr_w(returnedAddress));

     /* Release the object */
     ret = IWSDUdpAddress_Release(udpAddress);
     ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

     ret = WSACleanup();
     ok(ret == 0, "WSACleanup failed: %d\n", ret);
 }

 static void GetSetPort_udp_tests(void)
 {
     IWSDUdpAddress *udpAddress = NULL;
     WORD expectedPort1 = 12345;
     WORD expectedPort2 = 8080;
     WORD actualPort = 0;
     HRESULT rc;
     int ret;

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     /* No test for GetPort(NULL) as this causes an access violation exception on Windows */

     rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(actualPort == 0, "GetPort returned unexpected port: %d\n", actualPort);

     /* Try setting a zero port */
     rc = IWSDUdpAddress_SetPort(udpAddress, 0);
     ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(actualPort == 0, "GetPort returned unexpected port: %d\n", actualPort);

     /* Set a real port */
     rc = IWSDUdpAddress_SetPort(udpAddress, expectedPort1);
     ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(actualPort == expectedPort1, "GetPort returned unexpected port: %d\n", actualPort);

     /* Now set a different port */
     rc = IWSDUdpAddress_SetPort(udpAddress, expectedPort2);
     ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(actualPort == expectedPort2, "GetPort returned unexpected port: %d\n", actualPort);

     /* Release the object */
     ret = IWSDUdpAddress_Release(udpAddress);
     ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);
 }

 static void GetSetMessageType_udp_tests(void)
 {
     WSDUdpMessageType expectedMessageType1 = TWO_WAY;
     WSDUdpMessageType expectedMessageType2 = ONE_WAY;
     WSDUdpMessageType actualMessageType = 0;
     IWSDUdpAddress *udpAddress = NULL;
     HRESULT rc;
     int ret;

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     rc = IWSDUdpAddress_GetMessageType(udpAddress, NULL);
     ok(rc == E_POINTER, "GetMessageType returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
     ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
     ok(actualMessageType == 0, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

     /* Try setting a message type */
     rc = IWSDUdpAddress_SetMessageType(udpAddress, expectedMessageType1);
     ok(rc == S_OK, "SetMessageType returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
     ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
     ok(actualMessageType == expectedMessageType1, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

     /* Set another one */
     rc = IWSDUdpAddress_SetMessageType(udpAddress, expectedMessageType2);
     ok(rc == S_OK, "SetMessageType returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
     ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
     ok(actualMessageType == expectedMessageType2, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

     /* Release the object */
     ret = IWSDUdpAddress_Release(udpAddress);
     ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);
 }

 static void GetSetSockaddr_udp_tests(void)
 {
     SOCKADDR_STORAGE storage1, storage2;
     SOCKADDR_STORAGE returnedStorage;
     struct sockaddr_in6 *sockAddr6Ptr;
     struct sockaddr_in *sockAddrPtr;
     IWSDUdpAddress *udpAddress = NULL;
     LPCWSTR returnedAddress = NULL;
     char addressBuffer[MAX_PATH];
     WSADATA wsaData;
     WORD port = 0;
     HRESULT rc;
     int ret;

     const char *ipv4Address = "1.2.3.4";
     const short ipv4Port = 1234;
     const WCHAR expectedIpv4TransportAddr[] = {'1','.','2','.','3','.','4',':','1','2','3','4',0};
     const WCHAR expectedIpv4TransportAddrNoPort[] = {'1','.','2','.','3','.','4',0};

     const char *ipv6Address = "2a00:1234:5678:dead:beef::aaaa";
     const short ipv6Port = 2345;
     const WCHAR expectedIpv6TransportAddr[] = {'[','2','a','0','0',':','1','2','3','4',':','5','6','7','8',':','d','e','a','d',':',
         'b','e','e','f',':',':','a','a','a','a',']',':','2','3','4','5',0};
     const WCHAR expectedIpv6TransportAddrNoPort[] = {'2','a','0','0',':','1','2','3','4',':','5','6','7','8',':','d','e','a','d',':',
         'b','e','e','f',':',':','a','a','a','a',0};

     ZeroMemory(&storage1, sizeof(SOCKADDR_STORAGE));
     ZeroMemory(&storage2, sizeof(SOCKADDR_STORAGE));
     ZeroMemory(&returnedStorage, sizeof(SOCKADDR_STORAGE));

     ret = WSAStartup(MAKEWORD(2, 2), &wsaData);
     ok(ret == 0, "WSAStartup failed: %d\n", ret);

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     rc = IWSDUdpAddress_GetSockaddr(udpAddress, NULL);
     ok(rc == E_POINTER, "GetSockaddr returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
     ok(rc == E_FAIL, "GetSockaddr returned unexpected result: %08x\n", rc);

     /* Try setting a transport address */
     rc = IWSDUdpAddress_SetTransportAddress(udpAddress, expectedIpv6TransportAddr);
     ok(rc == S_OK, "SetTransportAddress failed: %08x\n", rc);

     /* A socket address should be returned */
     rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
     ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);
     ok(returnedStorage.ss_family == AF_INET6, "returnedStorage.ss_family != AF_INET6 (%d)\n", returnedStorage.ss_family);

     sockAddr6Ptr = (struct sockaddr_in6 *) &returnedStorage;

     /* Windows however doesn't set the port number */
     ok(sockAddr6Ptr->sin6_port == 0, "returnedStorage.sin6_port != 0 (%d)\n", sockAddr6Ptr->sin6_port);

     ok(inet_ntop(returnedStorage.ss_family, &sockAddr6Ptr->sin6_addr, addressBuffer, MAX_PATH) != NULL, "inet_ntop failed (%d)\n", WSAGetLastError());
     ok(strcmp(addressBuffer, ipv6Address) == 0, "returnedStorage.sin6_addr != '%s' ('%s')\n", ipv6Address, addressBuffer);

     /* Release the object and create a new one */
     ret = IWSDUdpAddress_Release(udpAddress);
     ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

     rc = WSDCreateUdpAddress(&udpAddress);
     ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
     ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

     /* Try setting an IPv4 address */
     sockAddrPtr = (struct sockaddr_in *) &storage1;
     sockAddrPtr->sin_family = AF_INET;
     sockAddrPtr->sin_port = htons(ipv4Port);

     ret = inet_pton(AF_INET, ipv4Address, &sockAddrPtr->sin_addr);
     ok(ret == 1, "inet_pton(ipv4) failed: %d\n", WSAGetLastError());

     rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage1);
     ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
     ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);

     ok(returnedStorage.ss_family == storage1.ss_family, "returnedStorage.ss_family != storage1.ss_family (%d)\n", returnedStorage.ss_family);
     ok(memcmp(&returnedStorage, &storage1, sizeof(struct sockaddr_in)) == 0, "returnedStorage != storage1\n");

     /* Check that GetTransportAddress returns the address set via the socket */
     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
     ok(lstrcmpW(returnedAddress, expectedIpv4TransportAddr) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

     /* Check that GetPort doesn't return the port set via the socket */
     rc = IWSDUdpAddress_GetPort(udpAddress, &port);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(port == 0, "GetPort returned unexpected port: %d\n", port);

     /* Try setting an IPv4 address without a port */
     sockAddrPtr->sin_port = 0;

     rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage1);
     ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

     /* Check that GetTransportAddress returns the address set via the socket */
     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
     ok(lstrcmpW(returnedAddress, expectedIpv4TransportAddrNoPort) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

     /* Try setting an IPv6 address */
     sockAddr6Ptr = (struct sockaddr_in6 *) &storage2;
     sockAddr6Ptr->sin6_family = AF_INET6;
     sockAddr6Ptr->sin6_port = htons(ipv6Port);

     ret = inet_pton(AF_INET6, ipv6Address, &sockAddr6Ptr->sin6_addr);
     ok(ret == 1, "inet_pton(ipv6) failed: %d\n", WSAGetLastError());

     rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
     ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

     rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
     ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);

     ok(returnedStorage.ss_family == storage2.ss_family, "returnedStorage.ss_family != storage2.ss_family (%d)\n", returnedStorage.ss_family);
     ok(memcmp(&returnedStorage, &storage2, sizeof(struct sockaddr_in6)) == 0, "returnedStorage != storage2\n");

     /* Check that GetTransportAddress returns the address set via the socket */
     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
     ok(lstrcmpW(returnedAddress, expectedIpv6TransportAddr) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

     /* Check that GetPort doesn't return the port set via the socket */
     rc = IWSDUdpAddress_GetPort(udpAddress, &port);
     ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
     ok(port == 0, "GetPort returned unexpected port: %d\n", port);

     /* Try setting an IPv6 address without a port */
     sockAddr6Ptr->sin6_port = 0;

     rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
     ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

     /* Check that GetTransportAddress returns the address set via the socket */
     rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
     ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
     ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
     ok(lstrcmpW(returnedAddress, expectedIpv6TransportAddrNoPort) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

     rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
     ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

     /* Release the object */
     ret = IWSDUdpAddress_Release(udpAddress);
     ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

     ret = WSACleanup();
     ok(ret == 0, "WSACleanup failed: %d\n", ret);
 }

 START_TEST(address)
 {
     CoInitialize(NULL);

     CreateUdpAddress_tests();
     GetSetTransportAddress_udp_tests();
     GetSetPort_udp_tests();
     GetSetMessageType_udp_tests();
     GetSetSockaddr_udp_tests();

     CoUninitialize();
 }
	/*
	* Web Services on Devices
	* Address tests
	*
	* Copyright 2017 Owen Rudge for CodeWeavers
	*
	* 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
	*/

	#define COBJMACROS

	#include <winsock2.h>
	#include <ws2tcpip.h>
	#include <windows.h>

	#include "wine/test.h"
	#include "objbase.h"
	#include "wsdapi.h"

	static void CreateUdpAddress_tests(void)
	{
	IWSDUdpAddress udpAddress = NULL, udpAddress2 = NULL;
	IWSDTransportAddress *transportAddress = NULL;
	IWSDAddress *address = NULL;
	IUnknown *unknown;
	HRESULT rc;
	ULONG ref;

	rc = WSDCreateUdpAddress(NULL);
	ok((rc == E_POINTER) \|\| (rc == E_INVALIDARG), "WSDCreateUDPAddress(NULL) failed: %08x\n", rc);

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUDPAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUDPAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	/* Try to query for objects */
	rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDUdpAddress, (LPVOID*)&udpAddress2);
	ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IWSDUdpAddress) failed: %08x\n", rc);

	if (rc == S_OK)
	IWSDUdpAddress_Release(udpAddress2);

	rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDTransportAddress, (LPVOID*)&transportAddress);
	ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IID_WSDTransportAddress) failed: %08x\n", rc);

	if (rc == S_OK)
	IWSDTransportAddress_Release(transportAddress);

	rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IWSDAddress, (LPVOID*)&address);
	ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IWSDAddress) failed: %08x\n", rc);

	if (rc == S_OK)
	IWSDAddress_Release(address);

	rc = IWSDUdpAddress_QueryInterface(udpAddress, &IID_IUnknown, (LPVOID*)&unknown);
	ok(rc == S_OK, "IWSDUdpAddress_QueryInterface(IID_IUnknown) failed: %08x\n", rc);

	if (rc == S_OK)
	IUnknown_Release(unknown);

	ref = IWSDUdpAddress_Release(udpAddress);
	ok(ref == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ref);
	}

	static void GetSetTransportAddress_udp_tests(void)
	{
	IWSDUdpAddress *udpAddress = NULL;
	const WCHAR ipv4Address[] = {'1','0','.','2','0','.','3','0','.','4','0',0};
	const WCHAR ipv6Address[] = {'a','a','b','b',':','c','d',':',':','a','b','c',0};
	const WCHAR ipv4AddressWithPort[] = {'1','0','.','2','0','.','3','0','.','4','0',':','1','2','4',0};
	const WCHAR ipv6AddressWithPort[] = {'[','a','a','b','b',':','c','d',':',':','a','b','c',':','5','6','7',']',':','1','2','4',0};
	const WCHAR invalidAddress[] = {'n','o','t','/','v','a','l','i','d',0};
	LPCWSTR returnedAddress = NULL;
	WSADATA wsaData;
	HRESULT rc;
	int ret;

	ret = WSAStartup(MAKEWORD(2, 2), &wsaData);
	ok(ret == 0, "WSAStartup failed: %d\n", ret);

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
	ok(returnedAddress == NULL, "GetTransportAddress returned unexpected address: %08x\n", rc);

	/* Try setting a null address */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, NULL);
	ok(rc == E_INVALIDARG, "SetTransportAddress(NULL) returned unexpected result: %08x\n", rc);

	/* Try setting an invalid address */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, invalidAddress);
	ok(rc == HRESULT_FROM_WIN32(WSAHOST_NOT_FOUND), "SetTransportAddress(invalidAddress) returned unexpected result: %08x\n", rc);

	/* Try setting an IPv4 address */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv4Address);
	ok(rc == S_OK, "SetTransportAddress(ipv4Address) failed: %08x\n", rc);

	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, NULL);
	ok(rc == E_POINTER, "GetTransportAddress(NULL) returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
	ok(lstrcmpW(returnedAddress, ipv4Address) == 0, "Returned address != ipv4Address (%s)\n", wine_dbgstr_w(returnedAddress));

	/* Try setting an IPv4 address with a port number */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv4AddressWithPort);
	ok(rc == HRESULT_FROM_WIN32(WSAHOST_NOT_FOUND), "SetTransportAddress(ipv4Address) failed: %08x\n", rc);

	/* Try setting an IPv6 address */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv6Address);
	ok(rc == S_OK, "SetTransportAddress(ipv6Address) failed: %08x\n", rc);

	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
	ok(lstrcmpW(returnedAddress, ipv6Address) == 0, "Returned address != ipv6Address (%s)\n", wine_dbgstr_w(returnedAddress));

	/* Try setting an IPv6 address with a port number */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, ipv6AddressWithPort);
	ok(rc == S_OK, "SetTransportAddress(ipv6AddressWithPort) failed: %08x\n", rc);

	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress returned unexpected result: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: '%s'\n", wine_dbgstr_w(returnedAddress));
	todo_wine ok(lstrcmpW(returnedAddress, ipv6AddressWithPort) == 0, "Returned address != ipv6AddressWithPort (%s)\n", wine_dbgstr_w(returnedAddress));

	/* Release the object */
	ret = IWSDUdpAddress_Release(udpAddress);
	ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

	ret = WSACleanup();
	ok(ret == 0, "WSACleanup failed: %d\n", ret);
	}

	static void GetSetPort_udp_tests(void)
	{
	IWSDUdpAddress *udpAddress = NULL;
	WORD expectedPort1 = 12345;
	WORD expectedPort2 = 8080;
	WORD actualPort = 0;
	HRESULT rc;
	int ret;

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	/* No test for GetPort(NULL) as this causes an access violation exception on Windows */

	rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(actualPort == 0, "GetPort returned unexpected port: %d\n", actualPort);

	/* Try setting a zero port */
	rc = IWSDUdpAddress_SetPort(udpAddress, 0);
	ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(actualPort == 0, "GetPort returned unexpected port: %d\n", actualPort);

	/* Set a real port */
	rc = IWSDUdpAddress_SetPort(udpAddress, expectedPort1);
	ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(actualPort == expectedPort1, "GetPort returned unexpected port: %d\n", actualPort);

	/* Now set a different port */
	rc = IWSDUdpAddress_SetPort(udpAddress, expectedPort2);
	ok(rc == S_OK, "SetPort returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetPort(udpAddress, &actualPort);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(actualPort == expectedPort2, "GetPort returned unexpected port: %d\n", actualPort);

	/* Release the object */
	ret = IWSDUdpAddress_Release(udpAddress);
	ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);
	}

	static void GetSetMessageType_udp_tests(void)
	{
	WSDUdpMessageType expectedMessageType1 = TWO_WAY;
	WSDUdpMessageType expectedMessageType2 = ONE_WAY;
	WSDUdpMessageType actualMessageType = 0;
	IWSDUdpAddress *udpAddress = NULL;
	HRESULT rc;
	int ret;

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	rc = IWSDUdpAddress_GetMessageType(udpAddress, NULL);
	ok(rc == E_POINTER, "GetMessageType returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
	ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
	ok(actualMessageType == 0, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

	/* Try setting a message type */
	rc = IWSDUdpAddress_SetMessageType(udpAddress, expectedMessageType1);
	ok(rc == S_OK, "SetMessageType returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
	ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
	ok(actualMessageType == expectedMessageType1, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

	/* Set another one */
	rc = IWSDUdpAddress_SetMessageType(udpAddress, expectedMessageType2);
	ok(rc == S_OK, "SetMessageType returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetMessageType(udpAddress, &actualMessageType);
	ok(rc == S_OK, "GetMessageType returned unexpected result: %08x\n", rc);
	ok(actualMessageType == expectedMessageType2, "GetMessageType returned unexpected message type: %d\n", actualMessageType);

	/* Release the object */
	ret = IWSDUdpAddress_Release(udpAddress);
	ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);
	}

	static void GetSetSockaddr_udp_tests(void)
	{
	SOCKADDR_STORAGE storage1, storage2;
	SOCKADDR_STORAGE returnedStorage;
	struct sockaddr_in6 *sockAddr6Ptr;
	struct sockaddr_in *sockAddrPtr;
	IWSDUdpAddress *udpAddress = NULL;
	LPCWSTR returnedAddress = NULL;
	char addressBuffer[MAX_PATH];
	WSADATA wsaData;
	WORD port = 0;
	HRESULT rc;
	int ret;

	const char *ipv4Address = "1.2.3.4";
	const short ipv4Port = 1234;
	const WCHAR expectedIpv4TransportAddr[] = {'1','.','2','.','3','.','4',':','1','2','3','4',0};
	const WCHAR expectedIpv4TransportAddrNoPort[] = {'1','.','2','.','3','.','4',0};

	const char *ipv6Address = "2a00:1234:5678:dead:beef::aaaa";
	const short ipv6Port = 2345;
	const WCHAR expectedIpv6TransportAddr[] = {'[','2','a','0','0',':','1','2','3','4',':','5','6','7','8',':','d','e','a','d',':',
	'b','e','e','f',':',':','a','a','a','a',']',':','2','3','4','5',0};
	const WCHAR expectedIpv6TransportAddrNoPort[] = {'2','a','0','0',':','1','2','3','4',':','5','6','7','8',':','d','e','a','d',':',
	'b','e','e','f',':',':','a','a','a','a',0};

	ZeroMemory(&storage1, sizeof(SOCKADDR_STORAGE));
	ZeroMemory(&storage2, sizeof(SOCKADDR_STORAGE));
	ZeroMemory(&returnedStorage, sizeof(SOCKADDR_STORAGE));

	ret = WSAStartup(MAKEWORD(2, 2), &wsaData);
	ok(ret == 0, "WSAStartup failed: %d\n", ret);

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	rc = IWSDUdpAddress_GetSockaddr(udpAddress, NULL);
	ok(rc == E_POINTER, "GetSockaddr returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
	ok(rc == E_FAIL, "GetSockaddr returned unexpected result: %08x\n", rc);

	/* Try setting a transport address */
	rc = IWSDUdpAddress_SetTransportAddress(udpAddress, expectedIpv6TransportAddr);
	ok(rc == S_OK, "SetTransportAddress failed: %08x\n", rc);

	/* A socket address should be returned */
	rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
	ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);
	ok(returnedStorage.ss_family == AF_INET6, "returnedStorage.ss_family != AF_INET6 (%d)\n", returnedStorage.ss_family);

	sockAddr6Ptr = (struct sockaddr_in6 *) &returnedStorage;

	/* Windows however doesn't set the port number */
	ok(sockAddr6Ptr->sin6_port == 0, "returnedStorage.sin6_port != 0 (%d)\n", sockAddr6Ptr->sin6_port);

	ok(inet_ntop(returnedStorage.ss_family, &sockAddr6Ptr->sin6_addr, addressBuffer, MAX_PATH) != NULL, "inet_ntop failed (%d)\n", WSAGetLastError());
	ok(strcmp(addressBuffer, ipv6Address) == 0, "returnedStorage.sin6_addr != '%s' ('%s')\n", ipv6Address, addressBuffer);

	/* Release the object and create a new one */
	ret = IWSDUdpAddress_Release(udpAddress);
	ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

	rc = WSDCreateUdpAddress(&udpAddress);
	ok(rc == S_OK, "WSDCreateUdpAddress(NULL, &udpAddress) failed: %08x\n", rc);
	ok(udpAddress != NULL, "WSDCreateUdpAddress(NULL, &udpAddress) failed: udpAddress == NULL\n");

	/* Try setting an IPv4 address */
	sockAddrPtr = (struct sockaddr_in *) &storage1;
	sockAddrPtr->sin_family = AF_INET;
	sockAddrPtr->sin_port = htons(ipv4Port);

	ret = inet_pton(AF_INET, ipv4Address, &sockAddrPtr->sin_addr);
	ok(ret == 1, "inet_pton(ipv4) failed: %d\n", WSAGetLastError());

	rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage1);
	ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
	ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);

	ok(returnedStorage.ss_family == storage1.ss_family, "returnedStorage.ss_family != storage1.ss_family (%d)\n", returnedStorage.ss_family);
	ok(memcmp(&returnedStorage, &storage1, sizeof(struct sockaddr_in)) == 0, "returnedStorage != storage1\n");

	/* Check that GetTransportAddress returns the address set via the socket */
	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
	ok(lstrcmpW(returnedAddress, expectedIpv4TransportAddr) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

	/* Check that GetPort doesn't return the port set via the socket */
	rc = IWSDUdpAddress_GetPort(udpAddress, &port);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(port == 0, "GetPort returned unexpected port: %d\n", port);

	/* Try setting an IPv4 address without a port */
	sockAddrPtr->sin_port = 0;

	rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage1);
	ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

	/* Check that GetTransportAddress returns the address set via the socket */
	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
	ok(lstrcmpW(returnedAddress, expectedIpv4TransportAddrNoPort) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

	/* Try setting an IPv6 address */
	sockAddr6Ptr = (struct sockaddr_in6 *) &storage2;
	sockAddr6Ptr->sin6_family = AF_INET6;
	sockAddr6Ptr->sin6_port = htons(ipv6Port);

	ret = inet_pton(AF_INET6, ipv6Address, &sockAddr6Ptr->sin6_addr);
	ok(ret == 1, "inet_pton(ipv6) failed: %d\n", WSAGetLastError());

	rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
	ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

	rc = IWSDUdpAddress_GetSockaddr(udpAddress, &returnedStorage);
	ok(rc == S_OK, "GetSockaddr returned unexpected result: %08x\n", rc);

	ok(returnedStorage.ss_family == storage2.ss_family, "returnedStorage.ss_family != storage2.ss_family (%d)\n", returnedStorage.ss_family);
	ok(memcmp(&returnedStorage, &storage2, sizeof(struct sockaddr_in6)) == 0, "returnedStorage != storage2\n");

	/* Check that GetTransportAddress returns the address set via the socket */
	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
	ok(lstrcmpW(returnedAddress, expectedIpv6TransportAddr) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

	/* Check that GetPort doesn't return the port set via the socket */
	rc = IWSDUdpAddress_GetPort(udpAddress, &port);
	ok(rc == S_OK, "GetPort returned unexpected result: %08x\n", rc);
	ok(port == 0, "GetPort returned unexpected port: %d\n", port);

	/* Try setting an IPv6 address without a port */
	sockAddr6Ptr->sin6_port = 0;

	rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
	ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

	/* Check that GetTransportAddress returns the address set via the socket */
	rc = IWSDUdpAddress_GetTransportAddress(udpAddress, &returnedAddress);
	ok(rc == S_OK, "GetTransportAddress failed: %08x\n", rc);
	ok(returnedAddress != NULL, "GetTransportAddress returned unexpected address: %p\n", returnedAddress);
	ok(lstrcmpW(returnedAddress, expectedIpv6TransportAddrNoPort) == 0, "GetTransportAddress returned unexpected address: %s\n", wine_dbgstr_w(returnedAddress));

	rc = IWSDUdpAddress_SetSockaddr(udpAddress, &storage2);
	ok(rc == S_OK, "SetSockaddr returned unexpected result: %08x\n", rc);

	/* Release the object */
	ret = IWSDUdpAddress_Release(udpAddress);
	ok(ret == 0, "IWSDUdpAddress_Release() has %d references, should have 0\n", ret);

	ret = WSACleanup();
	ok(ret == 0, "WSACleanup failed: %d\n", ret);
	}

	START_TEST(address)
	{
	CoInitialize(NULL);

	CreateUdpAddress_tests();
	GetSetTransportAddress_udp_tests();
	GetSetPort_udp_tests();
	GetSetMessageType_udp_tests();
	GetSetSockaddr_udp_tests();

	CoUninitialize();
	}