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();
	}