dlls/wsdapi/tests/discovery.c

   1 /*
   2  * Web Services on Devices
   3  * Discovery tests
   4  *
   5  * Copyright 2017 Owen Rudge for CodeWeavers
   6  *
   7  * This library is free software; you can redistribute it and/or
   8  * modify it under the terms of the GNU Lesser General Public
   9  * License as published by the Free Software Foundation; either
  10  * version 2.1 of the License, or (at your option) any later version.
  11  *
  12  * This library is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  15  * Lesser General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Lesser General Public
  18  * License along with this library; if not, write to the Free Software
  19  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  20  */
  21
  22 #define COBJMACROS
  23
  24 #include <windows.h>
  25
  26 #include "wine/test.h"
  27 #include "initguid.h"
  28 #include "objbase.h"
  29 #include "wsdapi.h"
  30
  31 typedef struct IWSDiscoveryPublisherNotifyImpl {
  32     IWSDiscoveryPublisherNotify IWSDiscoveryPublisherNotify_iface;
  33     LONG                  ref;
  34 } IWSDiscoveryPublisherNotifyImpl;
  35
  36 static inline IWSDiscoveryPublisherNotifyImpl *impl_from_IWSDiscoveryPublisherNotify(IWSDiscoveryPublisherNotify *iface)
  37 {
  38     return CONTAINING_RECORD(iface, IWSDiscoveryPublisherNotifyImpl, IWSDiscoveryPublisherNotify_iface);
  39 }
  40
  41 static HRESULT WINAPI IWSDiscoveryPublisherNotifyImpl_QueryInterface(IWSDiscoveryPublisherNotify *iface, REFIID riid, void **ppv)
  42 {
  43     IWSDiscoveryPublisherNotifyImpl *This = impl_from_IWSDiscoveryPublisherNotify(iface);
  44
  45     if (!ppv)
  46     {
  47         return E_INVALIDARG;
  48     }
  49
  50     *ppv = NULL;
  51
  52     if (IsEqualIID(riid, &IID_IUnknown) ||
  53         IsEqualIID(riid, &IID_IWSDiscoveryPublisherNotify))
  54     {
  55         *ppv = &This->IWSDiscoveryPublisherNotify_iface;
  56     }
  57     else
  58     {
  59         return E_NOINTERFACE;
  60     }
  61
  62     IUnknown_AddRef((IUnknown*)*ppv);
  63     return S_OK;
  64 }
  65
  66 static ULONG WINAPI IWSDiscoveryPublisherNotifyImpl_AddRef(IWSDiscoveryPublisherNotify *iface)
  67 {
  68     IWSDiscoveryPublisherNotifyImpl *This = impl_from_IWSDiscoveryPublisherNotify(iface);
  69     ULONG ref = InterlockedIncrement(&This->ref);
  70
  71     trace("IWSDiscoveryPublisherNotifyImpl_AddRef called (%p, ref = %d)\n", This, ref);
  72     return ref;
  73 }
  74
  75 static ULONG WINAPI IWSDiscoveryPublisherNotifyImpl_Release(IWSDiscoveryPublisherNotify *iface)
  76 {
  77     IWSDiscoveryPublisherNotifyImpl *This = impl_from_IWSDiscoveryPublisherNotify(iface);
  78     ULONG ref = InterlockedDecrement(&This->ref);
  79
  80     trace("IWSDiscoveryPublisherNotifyImpl_Release called (%p, ref = %d)\n", This, ref);
  81
  82     if (ref == 0)
  83     {
  84         HeapFree(GetProcessHeap(), 0, This);
  85     }
  86
  87     return ref;
  88 }
  89
  90 static HRESULT WINAPI IWSDiscoveryPublisherNotifyImpl_ProbeHandler(IWSDiscoveryPublisherNotify *This, const WSD_SOAP_MESSAGE *pSoap, IWSDMessageParameters *pMessageParameters)
  91 {
  92     trace("IWSDiscoveryPublisherNotifyImpl_ProbeHandler called (%p, %p, %p)\n", This, pSoap, pMessageParameters);
  93     return S_OK;
  94 }
  95
  96 static HRESULT WINAPI IWSDiscoveryPublisherNotifyImpl_ResolveHandler(IWSDiscoveryPublisherNotify *This, const WSD_SOAP_MESSAGE *pSoap, IWSDMessageParameters *pMessageParameters)
  97 {
  98     trace("IWSDiscoveryPublisherNotifyImpl_ResolveHandler called (%p, %p, %p)\n", This, pSoap, pMessageParameters);
  99     return S_OK;
 100 }
 101
 102 static const IWSDiscoveryPublisherNotifyVtbl publisherNotify_vtbl =
 103 {
 104     IWSDiscoveryPublisherNotifyImpl_QueryInterface,
 105     IWSDiscoveryPublisherNotifyImpl_AddRef,
 106     IWSDiscoveryPublisherNotifyImpl_Release,
 107     IWSDiscoveryPublisherNotifyImpl_ProbeHandler,
 108     IWSDiscoveryPublisherNotifyImpl_ResolveHandler
 109 };
 110
 111 static BOOL create_discovery_publisher_notify(IWSDiscoveryPublisherNotify **publisherNotify)
 112 {
 113     IWSDiscoveryPublisherNotifyImpl *obj;
 114
 115     *publisherNotify = NULL;
 116
 117     obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*obj));
 118
 119     if (!obj)
 120     {
 121         trace("Out of memory creating IWSDiscoveryPublisherNotify\n");
 122         return FALSE;
 123     }
 124
 125     obj->IWSDiscoveryPublisherNotify_iface.lpVtbl = &publisherNotify_vtbl;
 126     obj->ref = 1;
 127
 128     *publisherNotify = &obj->IWSDiscoveryPublisherNotify_iface;
 129
 130     return TRUE;
 131 }
 132
 133 static void CreateDiscoveryPublisher_tests(void)
 134 {
 135     IWSDiscoveryPublisher *publisher = NULL;
 136     IWSDiscoveryPublisher *publisher2;
 137     IUnknown *unknown;
 138     HRESULT rc;
 139     ULONG ref;
 140
 141     rc = WSDCreateDiscoveryPublisher(NULL, NULL);
 142     ok((rc == E_POINTER) || (rc == E_INVALIDARG), "WSDCreateDiscoveryPublisher(NULL, NULL) failed: %08x\n", rc);
 143
 144     rc = WSDCreateDiscoveryPublisher(NULL, &publisher);
 145     ok(rc == S_OK, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: %08x\n", rc);
 146     ok(publisher != NULL, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: publisher == NULL\n");
 147
 148     /* Try to query for objects */
 149     rc = IWSDiscoveryPublisher_QueryInterface(publisher, &IID_IUnknown, (LPVOID*)&unknown);
 150     ok(rc == S_OK,"IWSDiscoveryPublisher_QueryInterface(IID_IUnknown) failed: %08x\n", rc);
 151
 152     if (rc == S_OK)
 153         IUnknown_Release(unknown);
 154
 155     rc = IWSDiscoveryPublisher_QueryInterface(publisher, &IID_IWSDiscoveryPublisher, (LPVOID*)&publisher2);
 156     ok(rc == S_OK,"IWSDiscoveryPublisher_QueryInterface(IID_IWSDiscoveryPublisher) failed: %08x\n", rc);
 157
 158     if (rc == S_OK)
 159         IWSDiscoveryPublisher_Release(publisher2);
 160
 161     ref = IWSDiscoveryPublisher_Release(publisher);
 162     ok(ref == 0, "IWSDiscoveryPublisher_Release() has %d references, should have 0\n", ref);
 163 }
 164
 165 static void CreateDiscoveryPublisher_XMLContext_tests(void)
 166 {
 167     IWSDiscoveryPublisher *publisher = NULL;
 168     IWSDXMLContext *xmlContext, *returnedContext;
 169     HRESULT rc;
 170     int ref;
 171
 172     /* Test creating an XML context and supplying it to WSDCreateDiscoveryPublisher */
 173     rc = WSDXMLCreateContext(&xmlContext);
 174     ok(rc == S_OK, "WSDXMLCreateContext failed: %08x\n", rc);
 175
 176     rc = WSDCreateDiscoveryPublisher(xmlContext, &publisher);
 177     ok(rc == S_OK, "WSDCreateDiscoveryPublisher(xmlContext, &publisher) failed: %08x\n", rc);
 178     ok(publisher != NULL, "WSDCreateDiscoveryPublisher(xmlContext, &publisher) failed: publisher == NULL\n");
 179
 180     rc = IWSDiscoveryPublisher_GetXMLContext(publisher, NULL);
 181     ok(rc == E_INVALIDARG, "GetXMLContext returned unexpected value with NULL argument: %08x\n", rc);
 182
 183     rc = IWSDiscoveryPublisher_GetXMLContext(publisher, &returnedContext);
 184     ok(rc == S_OK, "GetXMLContext failed: %08x\n", rc);
 185
 186     ok(xmlContext == returnedContext, "GetXMLContext returned unexpected value: returnedContext == %p\n", returnedContext);
 187
 188     ref = IWSDXMLContext_Release(returnedContext);
 189     ok(ref == 2, "IWSDXMLContext_Release() has %d references, should have 2\n", ref);
 190
 191     ref = IWSDiscoveryPublisher_Release(publisher);
 192     ok(ref == 0, "IWSDiscoveryPublisher_Release() has %d references, should have 0\n", ref);
 193
 194     ref = IWSDXMLContext_Release(returnedContext);
 195     ok(ref == 0, "IWSDXMLContext_Release() has %d references, should have 0\n", ref);
 196
 197     /* Test using a default XML context */
 198     publisher = NULL;
 199     returnedContext = NULL;
 200
 201     rc = WSDCreateDiscoveryPublisher(NULL, &publisher);
 202     ok(rc == S_OK, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: %08x\n", rc);
 203     ok(publisher != NULL, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: publisher == NULL\n");
 204
 205     rc = IWSDiscoveryPublisher_GetXMLContext(publisher, &returnedContext);
 206     ok(rc == S_OK, "GetXMLContext failed: %08x\n", rc);
 207
 208     ref = IWSDXMLContext_Release(returnedContext);
 209     ok(ref == 1, "IWSDXMLContext_Release() has %d references, should have 1\n", ref);
 210
 211     ref = IWSDiscoveryPublisher_Release(publisher);
 212     ok(ref == 0, "IWSDiscoveryPublisher_Release() has %d references, should have 0\n", ref);
 213 }
 214
 215 static void Publish_tests(void)
 216 {
 217     IWSDiscoveryPublisher *publisher = NULL;
 218     IWSDiscoveryPublisherNotify *sink1 = NULL, *sink2 = NULL;
 219     IWSDiscoveryPublisherNotifyImpl *sink1Impl = NULL, *sink2Impl = NULL;
 220
 221     HRESULT rc;
 222     ULONG ref;
 223
 224     rc = WSDCreateDiscoveryPublisher(NULL, &publisher);
 225     ok(rc == S_OK, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: %08x\n", rc);
 226     ok(publisher != NULL, "WSDCreateDiscoveryPublisher(NULL, &publisher) failed: publisher == NULL\n");
 227
 228     /* Test SetAddressFamily */
 229     rc = IWSDiscoveryPublisher_SetAddressFamily(publisher, 12345);
 230     ok(rc == E_INVALIDARG, "IWSDiscoveryPublisher_SetAddressFamily(12345) returned unexpected result: %08x\n", rc);
 231
 232     rc = IWSDiscoveryPublisher_SetAddressFamily(publisher, WSDAPI_ADDRESSFAMILY_IPV4);
 233     ok(rc == S_OK, "IWSDiscoveryPublisher_SetAddressFamily(WSDAPI_ADDRESSFAMILY_IPV4) failed: %08x\n", rc);
 234
 235     /* Try to update the address family after already setting it */
 236     rc = IWSDiscoveryPublisher_SetAddressFamily(publisher, WSDAPI_ADDRESSFAMILY_IPV6);
 237     ok(rc == STG_E_INVALIDFUNCTION, "IWSDiscoveryPublisher_SetAddressFamily(WSDAPI_ADDRESSFAMILY_IPV6) returned unexpected result: %08x\n", rc);
 238
 239     /* Create notification sinks */
 240     ok(create_discovery_publisher_notify(&sink1) == TRUE, "create_discovery_publisher_notify failed\n");
 241     ok(create_discovery_publisher_notify(&sink2) == TRUE, "create_discovery_publisher_notify failed\n");
 242
 243     /* Get underlying implementation so we can check the ref count */
 244     sink1Impl = impl_from_IWSDiscoveryPublisherNotify(sink1);
 245     sink2Impl = impl_from_IWSDiscoveryPublisherNotify(sink2);
 246
 247     /* Attempt to unregister sink before registering it */
 248     rc = IWSDiscoveryPublisher_UnRegisterNotificationSink(publisher, sink1);
 249     ok(rc == E_FAIL, "IWSDiscoveryPublisher_UnRegisterNotificationSink returned unexpected result: %08x\n", rc);
 250
 251     /* Register notification sinks */
 252     rc = IWSDiscoveryPublisher_RegisterNotificationSink(publisher, sink1);
 253     ok(rc == S_OK, "IWSDiscoveryPublisher_RegisterNotificationSink failed: %08x\n", rc);
 254     ok(sink1Impl->ref == 2, "Ref count for sink 1 is not as expected: %d\n", sink1Impl->ref);
 255
 256     rc = IWSDiscoveryPublisher_RegisterNotificationSink(publisher, sink2);
 257     ok(rc == S_OK, "IWSDiscoveryPublisher_RegisterNotificationSink failed: %08x\n", rc);
 258     ok(sink2Impl->ref == 2, "Ref count for sink 2 is not as expected: %d\n", sink2Impl->ref);
 259
 260     /* Unregister the first sink */
 261     rc = IWSDiscoveryPublisher_UnRegisterNotificationSink(publisher, sink1);
 262     ok(rc == S_OK, "IWSDiscoveryPublisher_UnRegisterNotificationSink failed: %08x\n", rc);
 263     ok(sink1Impl->ref == 1, "Ref count for sink 1 is not as expected: %d\n", sink1Impl->ref);
 264
 265     /* TODO: Publish */
 266
 267     ref = IWSDiscoveryPublisher_Release(publisher);
 268     ok(ref == 0, "IWSDiscoveryPublisher_Release() has %d references, should have 0\n", ref);
 269
 270     /* Check that the sinks have been released by the publisher */
 271     ok(sink1Impl->ref == 1, "Ref count for sink 1 is not as expected: %d\n", sink1Impl->ref);
 272     ok(sink2Impl->ref == 1, "Ref count for sink 2 is not as expected: %d\n", sink2Impl->ref);
 273
 274     /* Release the sinks */
 275     IWSDiscoveryPublisherNotify_Release(sink1);
 276     IWSDiscoveryPublisherNotify_Release(sink2);
 277 }
 278
 279 START_TEST(discovery)
 280 {
 281     CoInitialize(NULL);
 282
 283     CreateDiscoveryPublisher_tests();
 284     CreateDiscoveryPublisher_XMLContext_tests();
 285     Publish_tests();
 286
 287     CoUninitialize();
 288 }