From: Owen Rudge <orudge@codeweavers.com>
Subject: [7/9] wsdapi: Implement Get/SetLocalAddress.
Message-Id: <df1b1b48-2d83-257d-2d5a-03b79943b5d7@codeweavers.com>
Date: Tue, 6 Jun 2017 20:56:10 +0100

Signed-off-by: Owen Rudge <orudge@codeweavers.com>
---
  dlls/wsdapi/msgparams.c       | 46 
+++++++++++++++++++++++++++++++++++++++----
  dlls/wsdapi/tests/msgparams.c | 12 +++++------
  2 files changed, 48 insertions(+), 10 deletions(-)

diff --git a/dlls/wsdapi/msgparams.c b/dlls/wsdapi/msgparams.c
index b5f71b1..2f78cd4 100644
--- a/dlls/wsdapi/msgparams.c
+++ b/dlls/wsdapi/msgparams.c
@@ -32,6 +32,7 @@ WINE_DEFAULT_DEBUG_CHANNEL(wsdapi);
 typedef struct IWSDMessageParametersImpl {
     IWSDMessageParameters IWSDMessageParameters_iface;
     LONG                  ref;
+    IWSDAddress           *localAddress;
 } IWSDMessageParametersImpl;
 
 typedef struct IWSDUdpMessageParametersImpl {
@@ -68,6 +69,11 @@ static ULONG WINAPI IWSDMessageParametersImpl_Release(IWSDMessageParameters *ifa
 
     if (ref == 0)
     {
+        if (This->localAddress != NULL)
+        {
+            IWSDAddress_Release(This->localAddress);
+        }
+
         HeapFree(GetProcessHeap(), 0, This);
     }
 
@@ -76,14 +82,46 @@ static ULONG WINAPI IWSDMessageParametersImpl_Release(IWSDMessageParameters *ifa
 
 static HRESULT WINAPI IWSDMessageParametersImpl_GetLocalAddress(IWSDMessageParameters *This, IWSDAddress **ppAddress)
 {
-    FIXME("(%p, %p)\n", This, ppAddress);
-    return E_NOTIMPL;
+    IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
+
+    TRACE("(%p, %p)\n", impl, ppAddress);
+
+    if (ppAddress == NULL)
+    {
+        return E_POINTER;
+    }
+
+    if (impl->localAddress == NULL)
+    {
+        return E_ABORT;
+    }
+
+    *ppAddress = impl->localAddress;
+    IWSDAddress_AddRef(*ppAddress);
+
+    return S_OK;
 }
 
 static HRESULT WINAPI IWSDMessageParametersImpl_SetLocalAddress(IWSDMessageParameters *This, IWSDAddress *pAddress)
 {
-    FIXME("(%p, %p)\n", This, pAddress);
-    return E_NOTIMPL;
+    IWSDMessageParametersImpl *impl = impl_from_IWSDMessageParameters(This);
+
+    TRACE("(%p, %p)\n", impl, pAddress);
+
+    if (pAddress == NULL)
+    {
+        return E_POINTER;
+    }
+
+    if (impl->localAddress != NULL)
+    {
+        IWSDAddress_Release(impl->localAddress);
+    }
+
+    impl->localAddress = pAddress;
+    IWSDAddress_AddRef(pAddress);
+
+    return S_OK;
 }
 
 static HRESULT WINAPI IWSDMessageParametersImpl_GetRemoteAddress(IWSDMessageParameters *This, IWSDAddress **ppAddress)
diff --git a/dlls/wsdapi/tests/msgparams.c b/dlls/wsdapi/tests/msgparams.c
index 246cfd6..a8f905d 100644
--- a/dlls/wsdapi/tests/msgparams.c
+++ b/dlls/wsdapi/tests/msgparams.c
@@ -84,11 +84,11 @@ static void LocalAddress_tests(void)
     ok(udpMessageParams != NULL, "WSDCreateUdpMessageParameters(NULL, &udpMessageParams) failed: udpMessageParams == NULL\n");
 
     rc = IWSDUdpMessageParameters_GetLocalAddress(udpMessageParams, NULL);
-    todo_wine ok(rc == E_POINTER, "GetLocalAddress failed: %08x\n", rc);
+    ok(rc == E_POINTER, "GetLocalAddress failed: %08x\n", rc);
     ok(returnedAddress == NULL, "GetLocalAddress returned %p\n", returnedAddress);
 
     rc = IWSDUdpMessageParameters_GetLocalAddress(udpMessageParams, &returnedAddress);
-    todo_wine ok(rc == E_ABORT, "GetLocalAddress failed: %08x\n", rc);
+    ok(rc == E_ABORT, "GetLocalAddress failed: %08x\n", rc);
     ok(returnedAddress == NULL, "GetLocalAddress returned %p\n", returnedAddress);
 
     rc = WSDCreateUdpAddress(&origUdpAddress);
@@ -99,14 +99,14 @@ static void LocalAddress_tests(void)
     todo_wine ok(rc == S_OK, "SetTransportAddress failed: %08x\n", rc);
 
     rc = IWSDUdpMessageParameters_SetLocalAddress(udpMessageParams, (IWSDAddress *)origUdpAddress);
-    todo_wine ok(rc == S_OK, "SetLocalAddress failed: %08x\n", rc);
+    ok(rc == S_OK, "SetLocalAddress failed: %08x\n", rc);
 
     rc = IWSDUdpMessageParameters_GetLocalAddress(udpMessageParams, &returnedAddress);
-    todo_wine ok(rc == S_OK, "GetLocalAddress failed: %08x\n", rc);
-    todo_wine ok(returnedAddress != NULL, "GetLocalAddress returned NULL\n");
+    ok(rc == S_OK, "GetLocalAddress failed: %08x\n", rc);
+    ok(returnedAddress != NULL, "GetLocalAddress returned NULL\n");
 
     /* Check if GetLocalAddress returns the same object */
-    todo_wine ok(returnedAddress == (IWSDAddress *)origUdpAddress, "returnedAddress != origUdpAddress\n");
+    ok(returnedAddress == (IWSDAddress *)origUdpAddress, "returnedAddress != origUdpAddress\n");
 
     ret = IWSDUdpMessageParameters_Release(udpMessageParams);
     ok(ret == 0, "IWSDUdpMessageParameters_Release() has %d references, should have 0\n", ret);