dlls/wsdapi/discovery.c

   1 /*
   2  * Web Services on Devices
   3  *
   4  * Copyright 2017 Owen Rudge for CodeWeavers
   5  *
   6  * This library is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 2.1 of the License, or (at your option) any later version.
  10  *
  11  * This library is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public
  17  * License along with this library; if not, write to the Free Software
  18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
  19  */
  20
  21 #include <stdarg.h>
  22
  23 #define COBJMACROS
  24 #define INITGUID
  25
  26 #include "wsdapi_internal.h"
  27 #include "wine/debug.h"
  28 #include "wine/heap.h"
  29 #include "guiddef.h"
  30
  31 WINE_DEFAULT_DEBUG_CHANNEL(wsdapi);
  32
  33 static inline IWSDiscoveryPublisherImpl *impl_from_IWSDiscoveryPublisher(IWSDiscoveryPublisher *iface)
  34 {
  35     return CONTAINING_RECORD(iface, IWSDiscoveryPublisherImpl, IWSDiscoveryPublisher_iface);
  36 }
  37
  38 static HRESULT WINAPI IWSDiscoveryPublisherImpl_QueryInterface(IWSDiscoveryPublisher *iface, REFIID riid, void **ppv)
  39 {
  40     IWSDiscoveryPublisherImpl *This = impl_from_IWSDiscoveryPublisher(iface);
  41
  42     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppv);
  43
  44     if (!ppv)
  45     {
  46         WARN("Invalid parameter\n");
  47         return E_INVALIDARG;
  48     }
  49
  50     *ppv = NULL;
  51
  52     if (IsEqualIID(riid, &IID_IUnknown) ||
  53         IsEqualIID(riid, &IID_IWSDiscoveryPublisher))
  54     {
  55         *ppv = &This->IWSDiscoveryPublisher_iface;
  56     }
  57     else
  58     {
  59         WARN("Unknown IID %s\n", debugstr_guid(riid));
  60         return E_NOINTERFACE;
  61     }
  62
  63     IUnknown_AddRef((IUnknown*)*ppv);
  64     return S_OK;
  65 }
  66
  67 static ULONG WINAPI IWSDiscoveryPublisherImpl_AddRef(IWSDiscoveryPublisher *iface)
  68 {
  69     IWSDiscoveryPublisherImpl *This = impl_from_IWSDiscoveryPublisher(iface);
  70     ULONG ref = InterlockedIncrement(&This->ref);
  71
  72     TRACE("(%p) ref=%d\n", This, ref);
  73     return ref;
  74 }
  75
  76 static ULONG WINAPI IWSDiscoveryPublisherImpl_Release(IWSDiscoveryPublisher *iface)
  77 {
  78     IWSDiscoveryPublisherImpl *This = impl_from_IWSDiscoveryPublisher(iface);
  79     ULONG ref = InterlockedDecrement(&This->ref);
  80     struct notificationSink *sink, *cursor;
  81
  82     TRACE("(%p) ref=%d\n", This, ref);
  83
  84     if (ref == 0)
  85     {
  86         terminate_networking(This);
  87
  88         if (This->xmlContext != NULL)
  89         {
  90             IWSDXMLContext_Release(This->xmlContext);
  91         }
  92
  93         LIST_FOR_EACH_ENTRY_SAFE(sink, cursor, &This->notificationSinks, struct notificationSink, entry)
  94         {
  95             IWSDiscoveryPublisherNotify_Release(sink->notificationSink);
  96             list_remove(&sink->entry);
  97             HeapFree(GetProcessHeap(), 0, sink);
  98         }
  99
 100         HeapFree(GetProcessHeap(), 0, This);
 101     }
 102
 103     return ref;
 104 }
 105
 106 static HRESULT WINAPI IWSDiscoveryPublisherImpl_SetAddressFamily(IWSDiscoveryPublisher *This, DWORD dwAddressFamily)
 107 {
 108     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 109
 110     TRACE("(%p, %d)\n", This, dwAddressFamily);
 111
 112     /* Has the address family already been set? */
 113     if (impl->addressFamily != 0)
 114     {
 115         return STG_E_INVALIDFUNCTION;
 116     }
 117
 118     if ((dwAddressFamily == WSDAPI_ADDRESSFAMILY_IPV4) || (dwAddressFamily == WSDAPI_ADDRESSFAMILY_IPV6) ||
 119         (dwAddressFamily == (WSDAPI_ADDRESSFAMILY_IPV4 | WSDAPI_ADDRESSFAMILY_IPV6)))
 120     {
 121         /* TODO: Check that the address family is supported by the system */
 122         impl->addressFamily = dwAddressFamily;
 123         return S_OK;
 124     }
 125
 126     return E_INVALIDARG;
 127 }
 128
 129 static HRESULT WINAPI IWSDiscoveryPublisherImpl_RegisterNotificationSink(IWSDiscoveryPublisher *This, IWSDiscoveryPublisherNotify *pSink)
 130 {
 131     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 132     struct notificationSink *sink;
 133
 134     TRACE("(%p, %p)\n", This, pSink);
 135
 136     if (pSink == NULL)
 137     {
 138         return E_INVALIDARG;
 139     }
 140
 141     sink = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*sink));
 142
 143     if (!sink)
 144     {
 145         return E_OUTOFMEMORY;
 146     }
 147
 148     sink->notificationSink = pSink;
 149     IWSDiscoveryPublisherNotify_AddRef(pSink);
 150
 151     list_add_tail(&impl->notificationSinks, &sink->entry);
 152
 153     if ((!impl->publisherStarted) && (!init_networking(impl)))
 154         return E_FAIL;
 155
 156     return S_OK;
 157 }
 158
 159 static HRESULT WINAPI IWSDiscoveryPublisherImpl_UnRegisterNotificationSink(IWSDiscoveryPublisher *This, IWSDiscoveryPublisherNotify *pSink)
 160 {
 161     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 162     struct notificationSink *sink;
 163
 164     TRACE("(%p, %p)\n", This, pSink);
 165
 166     if (pSink == NULL)
 167     {
 168         return E_INVALIDARG;
 169     }
 170
 171     LIST_FOR_EACH_ENTRY(sink, &impl->notificationSinks, struct notificationSink, entry)
 172     {
 173         if (sink->notificationSink == pSink)
 174         {
 175             IWSDiscoveryPublisherNotify_Release(pSink);
 176             list_remove(&sink->entry);
 177             HeapFree(GetProcessHeap(), 0, sink);
 178
 179             return S_OK;
 180         }
 181     }
 182
 183     /* Notification sink is not registered */
 184     return E_FAIL;
 185 }
 186
 187 static HRESULT WINAPI IWSDiscoveryPublisherImpl_Publish(IWSDiscoveryPublisher *This, LPCWSTR pszId, ULONGLONG ullMetadataVersion, ULONGLONG ullInstanceId,
 188                                                         ULONGLONG ullMessageNumber, LPCWSTR pszSessionId, const WSD_NAME_LIST *pTypesList,
 189                                                         const WSD_URI_LIST *pScopesList, const WSD_URI_LIST *pXAddrsList)
 190 {
 191     return IWSDiscoveryPublisher_PublishEx(This, pszId, ullMetadataVersion, ullInstanceId, ullMessageNumber,
 192         pszSessionId, pTypesList, pScopesList, pXAddrsList, NULL, NULL, NULL, NULL, NULL);
 193 }
 194
 195 static HRESULT WINAPI IWSDiscoveryPublisherImpl_UnPublish(IWSDiscoveryPublisher *This, LPCWSTR pszId, ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber,
 196                                                           LPCWSTR pszSessionId, const WSDXML_ELEMENT *pAny)
 197 {
 198     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 199
 200     TRACE("(%p, %s, %s, %s, %s, %p)\n", This, debugstr_w(pszId), wine_dbgstr_longlong(ullInstanceId),
 201         wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId), pAny);
 202
 203     if ((!impl->publisherStarted) || (pszId == NULL) || (lstrlenW(pszId) > WSD_MAX_TEXT_LENGTH) ||
 204         ((pszSessionId != NULL) && (lstrlenW(pszSessionId) > WSD_MAX_TEXT_LENGTH)))
 205     {
 206         return E_INVALIDARG;
 207     }
 208
 209     return send_bye_message(impl, pszId, ullInstanceId, ullMessageNumber, pszSessionId, pAny);
 210 }
 211
 212 static HRESULT WINAPI IWSDiscoveryPublisherImpl_MatchProbe(IWSDiscoveryPublisher *This, const WSD_SOAP_MESSAGE *pProbeMessage,
 213                                                            IWSDMessageParameters *pMessageParameters, LPCWSTR pszId, ULONGLONG ullMetadataVersion,
 214                                                            ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber, LPCWSTR pszSessionId,
 215                                                            const WSD_NAME_LIST *pTypesList, const WSD_URI_LIST *pScopesList,
 216                                                            const WSD_URI_LIST *pXAddrsList)
 217 {
 218     FIXME("(%p, %p, %p, %s, %s, %s, %s, %s, %p, %p, %p)\n", This, pProbeMessage, pMessageParameters, debugstr_w(pszId),
 219         wine_dbgstr_longlong(ullMetadataVersion), wine_dbgstr_longlong(ullInstanceId), wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId),
 220         pTypesList, pScopesList, pXAddrsList);
 221
 222     return E_NOTIMPL;
 223 }
 224
 225 static HRESULT WINAPI IWSDiscoveryPublisherImpl_MatchResolve(IWSDiscoveryPublisher *This, const WSD_SOAP_MESSAGE *pResolveMessage,
 226                                                              IWSDMessageParameters *pMessageParameters, LPCWSTR pszId, ULONGLONG ullMetadataVersion,
 227                                                              ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber, LPCWSTR pszSessionId,
 228                                                              const WSD_NAME_LIST *pTypesList, const WSD_URI_LIST *pScopesList,
 229                                                              const WSD_URI_LIST *pXAddrsList)
 230 {
 231     FIXME("(%p, %p, %p, %s, %s, %s, %s, %s, %p, %p, %p)\n", This, pResolveMessage, pMessageParameters, debugstr_w(pszId),
 232         wine_dbgstr_longlong(ullMetadataVersion), wine_dbgstr_longlong(ullInstanceId), wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId),
 233         pTypesList, pScopesList, pXAddrsList);
 234
 235     return E_NOTIMPL;
 236 }
 237
 238 static HRESULT WINAPI IWSDiscoveryPublisherImpl_PublishEx(IWSDiscoveryPublisher *This, LPCWSTR pszId, ULONGLONG ullMetadataVersion,
 239                                                           ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber, LPCWSTR pszSessionId,
 240                                                           const WSD_NAME_LIST *pTypesList, const WSD_URI_LIST *pScopesList,
 241                                                           const WSD_URI_LIST *pXAddrsList, const WSDXML_ELEMENT *pHeaderAny,
 242                                                           const WSDXML_ELEMENT *pReferenceParameterAny, const WSDXML_ELEMENT *pPolicyAny,
 243                                                           const WSDXML_ELEMENT *pEndpointReferenceAny, const WSDXML_ELEMENT *pAny)
 244 {
 245     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 246
 247     TRACE("(%p, %s, %s, %s, %s, %s, %p, %p, %p, %p, %p, %p, %p, %p)\n", This, debugstr_w(pszId), wine_dbgstr_longlong(ullMetadataVersion),
 248         wine_dbgstr_longlong(ullInstanceId), wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId), pTypesList, pScopesList, pXAddrsList,
 249         pHeaderAny, pReferenceParameterAny, pPolicyAny, pEndpointReferenceAny, pAny);
 250
 251     if ((!impl->publisherStarted) || (pszId == NULL) || (lstrlenW(pszId) > WSD_MAX_TEXT_LENGTH) ||
 252         ((pszSessionId != NULL) && (lstrlenW(pszSessionId) > WSD_MAX_TEXT_LENGTH)))
 253     {
 254         return E_INVALIDARG;
 255     }
 256
 257     return send_hello_message(impl, pszId, ullMetadataVersion, ullInstanceId, ullMessageNumber, pszSessionId,
 258         pTypesList, pScopesList, pXAddrsList, pHeaderAny, pReferenceParameterAny, pEndpointReferenceAny, pAny);
 259 }
 260
 261 static HRESULT WINAPI IWSDiscoveryPublisherImpl_MatchProbeEx(IWSDiscoveryPublisher *This, const WSD_SOAP_MESSAGE *pProbeMessage,
 262                                                              IWSDMessageParameters *pMessageParameters, LPCWSTR pszId, ULONGLONG ullMetadataVersion,
 263                                                              ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber, LPCWSTR pszSessionId,
 264                                                              const WSD_NAME_LIST *pTypesList, const WSD_URI_LIST *pScopesList,
 265                                                              const WSD_URI_LIST *pXAddrsList, const WSDXML_ELEMENT *pHeaderAny,
 266                                                              const WSDXML_ELEMENT *pReferenceParameterAny, const WSDXML_ELEMENT *pPolicyAny,
 267                                                              const WSDXML_ELEMENT *pEndpointReferenceAny, const WSDXML_ELEMENT *pAny)
 268 {
 269     FIXME("(%p, %p, %p, %s, %s, %s, %s, %s, %p, %p, %p, %p, %p, %p, %p, %p)\n", This, pProbeMessage, pMessageParameters, debugstr_w(pszId),
 270         wine_dbgstr_longlong(ullMetadataVersion), wine_dbgstr_longlong(ullInstanceId), wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId),
 271         pTypesList, pScopesList, pXAddrsList, pHeaderAny, pReferenceParameterAny, pPolicyAny, pEndpointReferenceAny, pAny);
 272
 273     return E_NOTIMPL;
 274 }
 275
 276 static HRESULT WINAPI IWSDiscoveryPublisherImpl_MatchResolveEx(IWSDiscoveryPublisher *This, const WSD_SOAP_MESSAGE *pResolveMessage,
 277                                                                IWSDMessageParameters *pMessageParameters, LPCWSTR pszId, ULONGLONG ullMetadataVersion,
 278                                                                ULONGLONG ullInstanceId, ULONGLONG ullMessageNumber, LPCWSTR pszSessionId,
 279                                                                const WSD_NAME_LIST *pTypesList, const WSD_URI_LIST *pScopesList,
 280                                                                const WSD_URI_LIST *pXAddrsList, const WSDXML_ELEMENT *pHeaderAny,
 281                                                                const WSDXML_ELEMENT *pReferenceParameterAny, const WSDXML_ELEMENT *pPolicyAny,
 282                                                                const WSDXML_ELEMENT *pEndpointReferenceAny, const WSDXML_ELEMENT *pAny)
 283 {
 284     FIXME("(%p, %p, %p, %s, %s, %s, %s, %s, %p, %p, %p, %p, %p, %p, %p, %p)\n", This, pResolveMessage, pMessageParameters, debugstr_w(pszId),
 285         wine_dbgstr_longlong(ullMetadataVersion), wine_dbgstr_longlong(ullInstanceId), wine_dbgstr_longlong(ullMessageNumber), debugstr_w(pszSessionId),
 286         pTypesList, pScopesList, pXAddrsList, pHeaderAny, pReferenceParameterAny, pPolicyAny, pEndpointReferenceAny, pAny);
 287
 288     return E_NOTIMPL;
 289 }
 290
 291 static HRESULT WINAPI IWSDiscoveryPublisherImpl_RegisterScopeMatchingRule(IWSDiscoveryPublisher *This, IWSDScopeMatchingRule *pScopeMatchingRule)
 292 {
 293     FIXME("(%p, %p)\n", This, pScopeMatchingRule);
 294     return E_NOTIMPL;
 295 }
 296
 297 static HRESULT WINAPI IWSDiscoveryPublisherImpl_UnRegisterScopeMatchingRule(IWSDiscoveryPublisher *This, IWSDScopeMatchingRule *pScopeMatchingRule)
 298 {
 299     FIXME("(%p, %p)\n", This, pScopeMatchingRule);
 300     return E_NOTIMPL;
 301 }
 302
 303 static HRESULT WINAPI IWSDiscoveryPublisherImpl_GetXMLContext(IWSDiscoveryPublisher *This, IWSDXMLContext **ppContext)
 304 {
 305     IWSDiscoveryPublisherImpl *impl = impl_from_IWSDiscoveryPublisher(This);
 306
 307     TRACE("%p, %p)\n", This, ppContext);
 308
 309     if (ppContext == NULL)
 310         return E_INVALIDARG;
 311
 312     if (impl->xmlContext != NULL)
 313     {
 314         IWSDXMLContext_AddRef(impl->xmlContext);
 315     }
 316
 317     *ppContext = impl->xmlContext;
 318     return S_OK;
 319 }
 320
 321 static const IWSDiscoveryPublisherVtbl publisher_vtbl =
 322 {
 323     IWSDiscoveryPublisherImpl_QueryInterface,
 324     IWSDiscoveryPublisherImpl_AddRef,
 325     IWSDiscoveryPublisherImpl_Release,
 326     IWSDiscoveryPublisherImpl_SetAddressFamily,
 327     IWSDiscoveryPublisherImpl_RegisterNotificationSink,
 328     IWSDiscoveryPublisherImpl_UnRegisterNotificationSink,
 329     IWSDiscoveryPublisherImpl_Publish,
 330     IWSDiscoveryPublisherImpl_UnPublish,
 331     IWSDiscoveryPublisherImpl_MatchProbe,
 332     IWSDiscoveryPublisherImpl_MatchResolve,
 333     IWSDiscoveryPublisherImpl_PublishEx,
 334     IWSDiscoveryPublisherImpl_MatchProbeEx,
 335     IWSDiscoveryPublisherImpl_MatchResolveEx,
 336     IWSDiscoveryPublisherImpl_RegisterScopeMatchingRule,
 337     IWSDiscoveryPublisherImpl_UnRegisterScopeMatchingRule,
 338     IWSDiscoveryPublisherImpl_GetXMLContext
 339 };
 340
 341 HRESULT WINAPI WSDCreateDiscoveryPublisher(IWSDXMLContext *pContext, IWSDiscoveryPublisher **ppPublisher)
 342 {
 343     IWSDiscoveryPublisherImpl *obj;
 344     HRESULT ret;
 345
 346     TRACE("(%p, %p)\n", pContext, ppPublisher);
 347
 348     if (ppPublisher == NULL)
 349     {
 350         WARN("Invalid parameter: ppPublisher == NULL\n");
 351         return E_POINTER;
 352     }
 353
 354     *ppPublisher = NULL;
 355
 356     obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*obj));
 357
 358     if (!obj)
 359     {
 360         WARN("Out of memory\n");
 361         return E_OUTOFMEMORY;
 362     }
 363
 364     obj->IWSDiscoveryPublisher_iface.lpVtbl = &publisher_vtbl;
 365     obj->ref = 1;
 366
 367     if (pContext == NULL)
 368     {
 369         ret = WSDXMLCreateContext(&obj->xmlContext);
 370
 371         if (FAILED(ret))
 372         {
 373             WARN("Unable to create XML context\n");
 374             heap_free(obj);
 375             return ret;
 376         }
 377     }
 378     else
 379     {
 380         obj->xmlContext = pContext;
 381         IWSDXMLContext_AddRef(pContext);
 382     }
 383
 384     ret = register_namespaces(obj->xmlContext);
 385
 386     if (FAILED(ret))
 387     {
 388         WARN("Unable to register default namespaces\n");
 389         heap_free(obj);
 390
 391         return ret;
 392     }
 393
 394     list_init(&obj->notificationSinks);
 395
 396     *ppPublisher = &obj->IWSDiscoveryPublisher_iface;
 397     TRACE("Returning iface %p\n", *ppPublisher);
 398
 399     return S_OK;
 400 }