dlls/wsdapi/msgparams.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
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "wine/debug.h"
  28 #include "wsdapi.h"
  29
  30 WINE_DEFAULT_DEBUG_CHANNEL(wsdapi);
  31
  32 typedef struct IWSDMessageParametersImpl {
  33     IWSDMessageParameters IWSDMessageParameters_iface;
  34     LONG                  ref;
  35     IWSDAddress           *localAddress;
  36     IWSDAddress           *remoteAddress;
  37 } IWSDMessageParametersImpl;
  38
  39 typedef struct IWSDUdpMessageParametersImpl {
  40     IWSDMessageParametersImpl base;
  41     WSDUdpRetransmitParams    retransmitParams;
  42 } IWSDUdpMessageParametersImpl;
  43
  44 static inline IWSDMessageParametersImpl *impl_from_IWSDMessageParameters(IWSDMessageParameters *iface)
  45 {
  46     return CONTAINING_RECORD(iface, IWSDMessageParametersImpl, IWSDMessageParameters_iface);
  47 }
  48
  49 static inline IWSDUdpMessageParametersImpl *impl_from_IWSDUdpMessageParameters(IWSDUdpMessageParameters *iface)
  50 {
  51     return CONTAINING_RECORD(iface, IWSDUdpMessageParametersImpl, base.IWSDMessageParameters_iface);
  52 }
  53
  54 /* IWSDMessageParameters implementation */
  55
  56 static ULONG IWSDMessageParametersImpl_AddRef(IWSDMessageParameters *iface)
  57 {
  58     IWSDMessageParametersImpl *This = impl_from_IWSDMessageParameters(iface);
  59     ULONG ref = InterlockedIncrement(&This->ref);
  60
  61     TRACE("(%p) ref=%d\n", This, ref);
  62     return ref;
  63 }
  64
  65 static ULONG IWSDMessageParametersImpl_Release(IWSDMessageParameters *iface)
  66 {
  67     IWSDMessageParametersImpl *This = impl_from_IWSDMessageParameters(iface);
  68     ULONG ref = InterlockedDecrement(&This->ref);
  69
  70     TRACE("(%p) ref=%d\n", This, ref);
  71
  72     if (ref == 0)
  73     {
  74         if (This->localAddress != NULL)
  75         {
  76             IWSDAddress_Release(This->localAddress);
  77         }
  78
  79         if (This->remoteAddress != NULL)
  80         {
  81             IWSDAddress_Release(This->remoteAddress);
  82         }
  83
  84         HeapFree(GetProcessHeap(), 0, This);
  85     }
  86
  87     return ref;
  88 }
  89
  90 static HRESULT IWSDMessageParametersImpl_GetLocalAddress(IWSDMessageParameters *This, IWSDAddress **ppAddress)
  91 {
  92     IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
  93
  94     TRACE("(%p, %p)\n", impl, ppAddress);
  95
  96     if (ppAddress == NULL)
  97     {
  98         return E_POINTER;
  99     }
 100
 101     if (impl->localAddress == NULL)
 102     {
 103         return E_ABORT;
 104     }
 105
 106     *ppAddress = impl->localAddress;
 107     IWSDAddress_AddRef(*ppAddress);
 108
 109     return S_OK;
 110 }
 111
 112 static HRESULT IWSDMessageParametersImpl_SetLocalAddress(IWSDMessageParameters *This, IWSDAddress *pAddress)
 113 {
 114     IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
 115
 116     TRACE("(%p, %p)\n", impl, pAddress);
 117
 118     if (pAddress == NULL)
 119     {
 120         return E_POINTER;
 121     }
 122
 123     if (impl->localAddress != NULL)
 124     {
 125         IWSDAddress_Release(impl->localAddress);
 126     }
 127
 128     impl->localAddress = pAddress;
 129     IWSDAddress_AddRef(pAddress);
 130
 131     return S_OK;
 132 }
 133
 134 static HRESULT IWSDMessageParametersImpl_GetRemoteAddress(IWSDMessageParameters *This, IWSDAddress **ppAddress)
 135 {
 136     IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
 137
 138     TRACE("(%p, %p)\n", impl, ppAddress);
 139
 140     if (ppAddress == NULL)
 141     {
 142         return E_POINTER;
 143     }
 144
 145     if (impl->remoteAddress == NULL)
 146     {
 147         return E_ABORT;
 148     }
 149
 150     *ppAddress = impl->remoteAddress;
 151     IWSDAddress_AddRef(*ppAddress);
 152
 153     return S_OK;
 154 }
 155
 156 static HRESULT IWSDMessageParametersImpl_SetRemoteAddress(IWSDMessageParameters *This, IWSDAddress *pAddress)
 157 {
 158     IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
 159
 160     TRACE("(%p, %p)\n", impl, pAddress);
 161
 162     if (pAddress == NULL)
 163     {
 164         return E_POINTER;
 165     }
 166
 167     if (impl->remoteAddress != NULL)
 168     {
 169         IWSDAddress_Release(impl->remoteAddress);
 170     }
 171
 172     impl->remoteAddress = pAddress;
 173     IWSDAddress_AddRef(pAddress);
 174
 175     return S_OK;
 176 }
 177
 178 static HRESULT IWSDMessageParametersImpl_GetLowerParameters(IWSDMessageParameters *This, IWSDMessageParameters **ppTxParams)
 179 {
 180     FIXME("(%p, %p)\n", This, ppTxParams);
 181     return E_NOTIMPL;
 182 }
 183
 184 /* IWSDUdpMessageParameters implementation */
 185 static HRESULT WINAPI IWSDUdpMessageParametersImpl_QueryInterface(IWSDUdpMessageParameters *iface, REFIID riid, void **ppv)
 186 {
 187     IWSDUdpMessageParametersImpl *This = impl_from_IWSDUdpMessageParameters(iface);
 188
 189     TRACE("(%p, %s, %p)\n", This, debugstr_guid(riid), ppv);
 190
 191     if (!ppv)
 192     {
 193         WARN("Invalid parameter\n");
 194         return E_INVALIDARG;
 195     }
 196
 197     *ppv = NULL;
 198
 199     if (IsEqualIID(riid, &IID_IUnknown) ||
 200         IsEqualIID(riid, &IID_IWSDMessageParameters) ||
 201         IsEqualIID(riid, &IID_IWSDUdpMessageParameters))
 202     {
 203         *ppv = &This->base.IWSDMessageParameters_iface;
 204     }
 205     else
 206     {
 207         WARN("Unknown IID %s\n", debugstr_guid(riid));
 208         return E_NOINTERFACE;
 209     }
 210
 211     IUnknown_AddRef((IUnknown*)*ppv);
 212     return S_OK;
 213 }
 214
 215 static ULONG WINAPI IWSDUdpMessageParametersImpl_AddRef(IWSDUdpMessageParameters *iface)
 216 {
 217     return IWSDMessageParametersImpl_AddRef((IWSDMessageParameters *)iface);
 218 }
 219
 220 static ULONG WINAPI IWSDUdpMessageParametersImpl_Release(IWSDUdpMessageParameters *iface)
 221 {
 222     return IWSDMessageParametersImpl_Release((IWSDMessageParameters *)iface);
 223 }
 224
 225 static HRESULT WINAPI IWSDUdpMessageParametersImpl_GetLocalAddress(IWSDUdpMessageParameters *This, IWSDAddress **ppAddress)
 226 {
 227     return IWSDMessageParametersImpl_GetLocalAddress((IWSDMessageParameters *)This, ppAddress);
 228 }
 229
 230 static HRESULT WINAPI IWSDUdpMessageParametersImpl_SetLocalAddress(IWSDUdpMessageParameters *This, IWSDAddress *pAddress)
 231 {
 232     return IWSDMessageParametersImpl_SetLocalAddress((IWSDMessageParameters *)This, pAddress);
 233 }
 234
 235 static HRESULT WINAPI IWSDUdpMessageParametersImpl_GetRemoteAddress(IWSDUdpMessageParameters *This, IWSDAddress **ppAddress)
 236 {
 237     return IWSDMessageParametersImpl_GetRemoteAddress((IWSDMessageParameters *)This, ppAddress);
 238 }
 239
 240 static HRESULT WINAPI IWSDUdpMessageParametersImpl_SetRemoteAddress(IWSDUdpMessageParameters *This, IWSDAddress *pAddress)
 241 {
 242     return IWSDMessageParametersImpl_SetRemoteAddress((IWSDMessageParameters *)This, pAddress);
 243 }
 244
 245 static HRESULT WINAPI IWSDUdpMessageParametersImpl_GetLowerParameters(IWSDUdpMessageParameters *This, IWSDMessageParameters **ppTxParams)
 246 {
 247     return IWSDMessageParametersImpl_GetLowerParameters((IWSDMessageParameters *)This, ppTxParams);
 248 }
 249
 250 static HRESULT WINAPI IWSDUdpMessageParametersImpl_SetRetransmitParams(IWSDUdpMessageParameters *This, const WSDUdpRetransmitParams *pParams)
 251 {
 252     IWSDUdpMessageParametersImpl *impl = impl_from_IWSDUdpMessageParameters(This);
 253
 254     TRACE("(%p, %p)\n", impl, pParams);
 255
 256     if (pParams == NULL)
 257     {
 258         return E_INVALIDARG;
 259     }
 260
 261     impl->retransmitParams = *pParams;
 262     return S_OK;
 263 }
 264
 265 static HRESULT WINAPI IWSDUdpMessageParametersImpl_GetRetransmitParams(IWSDUdpMessageParameters *This, WSDUdpRetransmitParams *pParams)
 266 {
 267     IWSDUdpMessageParametersImpl *impl = impl_from_IWSDUdpMessageParameters(This);
 268
 269     TRACE("(%p, %p)\n", impl, pParams);
 270
 271     if (pParams == NULL)
 272     {
 273         return E_POINTER;
 274     }
 275
 276     * pParams = impl->retransmitParams;
 277     return S_OK;
 278 }
 279
 280 static const IWSDUdpMessageParametersVtbl udpMsgParamsVtbl =
 281 {
 282     /* IUnknown */
 283     IWSDUdpMessageParametersImpl_QueryInterface,
 284     IWSDUdpMessageParametersImpl_AddRef,
 285     IWSDUdpMessageParametersImpl_Release,
 286
 287     /* IWSDMessageParameters */
 288     IWSDUdpMessageParametersImpl_GetLocalAddress,
 289     IWSDUdpMessageParametersImpl_SetLocalAddress,
 290     IWSDUdpMessageParametersImpl_GetRemoteAddress,
 291     IWSDUdpMessageParametersImpl_SetRemoteAddress,
 292     IWSDUdpMessageParametersImpl_GetLowerParameters,
 293
 294     /* IWSDUdpMessageParameters */
 295     IWSDUdpMessageParametersImpl_SetRetransmitParams,
 296     IWSDUdpMessageParametersImpl_GetRetransmitParams
 297 };
 298
 299 HRESULT WINAPI WSDCreateUdpMessageParameters(IWSDUdpMessageParameters **ppTxParams)
 300 {
 301     IWSDUdpMessageParametersImpl *obj;
 302
 303     TRACE("(%p)\n", ppTxParams);
 304
 305     if (ppTxParams == NULL)
 306     {
 307         WARN("Invalid parameter: ppTxParams == NULL\n");
 308         return E_POINTER;
 309     }
 310
 311     *ppTxParams = NULL;
 312
 313     obj = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, sizeof(*obj));
 314     if (!obj) return E_OUTOFMEMORY;
 315
 316     obj->base.IWSDMessageParameters_iface.lpVtbl = (IWSDMessageParametersVtbl *)&udpMsgParamsVtbl;
 317     obj->base.ref = 1;
 318
 319     /* Populate default retransmit parameters */
 320     obj->retransmitParams.ulSendDelay = 0;
 321     obj->retransmitParams.ulRepeat = 1;
 322     obj->retransmitParams.ulRepeatMinDelay = 50;
 323     obj->retransmitParams.ulRepeatMaxDelay = 250;
 324     obj->retransmitParams.ulRepeatUpperDelay = 450;
 325
 326     *ppTxParams = (IWSDUdpMessageParameters *)&obj->base.IWSDMessageParameters_iface;
 327     TRACE("Returning iface %p\n", *ppTxParams);
 328
 329     return S_OK;
 330 }