inetcomm/tests: Fix test stream leak (Valgrind).

Signed-off-by: Nikolay Sivov <nsivov@codeweavers.com> Signed-off-by: Alexandre Julliard <julliard@winehq.org>

inetcomm/tests: Fix test stream leak (Valgrind).
8b67cfb8 · Nikolay Sivov · Alexandre Julliard · 59103375 · 8b67cfb8
Commit 8b67cfb8 authored Mar 12, 2017 by Nikolay Sivov Committed by Alexandre Julliard Mar 13, 2017
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 1 deletion

mimeole.c dlls/inetcomm/tests/mimeole.c +6 -1

No files found.
--- a/dlls/inetcomm/tests/mimeole.c
+++ b/dlls/inetcomm/tests/mimeole.c
@@ -323,7 +323,12 @@ static ULONG WINAPI Stream_AddRef(IStream *iface)
 static ULONG WINAPI Stream_Release(IStream *iface)
 {
    TestStream *This = impl_from_IStream(iface);
-    return InterlockedDecrement(&This->ref);
+    ULONG ref = InterlockedDecrement(&This->ref);
+
+    if (!ref)
+        HeapFree(GetProcessHeap(), 0, This);
+
+    return ref;
 }

 static HRESULT WINAPI Stream_Read(IStream *iface, void *pv, ULONG cb, ULONG *pcbRead)