From d4db4bfd7262871f94b512706aa88ddcb5315587 Mon Sep 17 00:00:00 2001
From: Anton Baskanov <baskanov@gmail.com>
Date: Thu, 28 Jul 2016 21:59:53 +0600
Subject: [PATCH] amstream: Implement AMAudioData::SetBuffer.

Signed-off-by: Anton Baskanov <baskanov@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
---
 dlls/amstream/audiodata.c      | 38 +++++++++++++++++++++++++--
 dlls/amstream/tests/amstream.c | 48 ++++++++++++++++++++++++++++++++++
 2 files changed, 84 insertions(+), 2 deletions(-)

diff --git a/dlls/amstream/audiodata.c b/dlls/amstream/audiodata.c
index 842fc95b960..6ffd1df49cb 100644
--- a/dlls/amstream/audiodata.c
+++ b/dlls/amstream/audiodata.c
@@ -32,6 +32,7 @@ typedef struct {
     LONG ref;
     DWORD size;
     BYTE *data;
+    BOOL data_owned;
     DWORD actual_data;
 } AMAudioDataImpl;
 
@@ -75,7 +76,14 @@ static ULONG WINAPI IAudioDataImpl_Release(IAudioData* iface)
     TRACE("(%p)->(): new ref = %u\n", iface, This->ref);
 
     if (!ref)
+    {
+        if (This->data_owned)
+        {
+            CoTaskMemFree(This->data);
+        }
+
         HeapFree(GetProcessHeap(), 0, This);
+    }
 
     return ref;
 }
@@ -83,9 +91,35 @@ static ULONG WINAPI IAudioDataImpl_Release(IAudioData* iface)
 /*** IMemoryData methods ***/
 static HRESULT WINAPI IAudioDataImpl_SetBuffer(IAudioData* iface, DWORD size, BYTE *data, DWORD flags)
 {
-    FIXME("(%p)->(%u,%p,%x): stub\n", iface, size, data, flags);
+    AMAudioDataImpl *This = impl_from_IAudioData(iface);
 
-    return E_NOTIMPL;
+    TRACE("(%p)->(%u,%p,%x)\n", iface, size, data, flags);
+
+    if (!size)
+    {
+        return E_INVALIDARG;
+    }
+
+    if (This->data_owned)
+    {
+        CoTaskMemFree(This->data);
+        This->data_owned = FALSE;
+    }
+
+    This->size = size;
+    This->data = data;
+
+    if (!This->data)
+    {
+        This->data = CoTaskMemAlloc(This->size);
+        This->data_owned = TRUE;
+        if (!This->data)
+        {
+            return E_OUTOFMEMORY;
+        }
+    }
+
+    return S_OK;
 }
 
 static HRESULT WINAPI IAudioDataImpl_GetInfo(IAudioData* iface, DWORD *length, BYTE **data, DWORD *actual_data)
diff --git a/dlls/amstream/tests/amstream.c b/dlls/amstream/tests/amstream.c
index 307868d4c14..a38ae64ec96 100644
--- a/dlls/amstream/tests/amstream.c
+++ b/dlls/amstream/tests/amstream.c
@@ -650,6 +650,53 @@ out_unknown:
     IUnknown_Release(unknown);
 }
 
+static void test_audiodata_set_buffer(void)
+{
+    IUnknown *unknown = create_audio_data();
+    IAudioData *audio_data = NULL;
+    BYTE buffer[100] = {0};
+    DWORD length = 0;
+    BYTE *data = NULL;
+
+    HRESULT result;
+
+    result = IUnknown_QueryInterface(unknown, &IID_IAudioData, (void **)&audio_data);
+    if (FAILED(result))
+    {
+        /* test_audiodata_query_interface handles this case */
+        skip("No IAudioData\n");
+        goto out_unknown;
+    }
+
+    result = IAudioData_SetBuffer(audio_data, 100, NULL, 0);
+    ok(S_OK == result, "got 0x%08x\n", result);
+
+    data = (BYTE *)0xdeadbeef;
+    length = 0xdeadbeef;
+    result = IAudioData_GetInfo(audio_data, &length, &data, NULL);
+    ok(S_OK == result, "got 0x%08x\n", result);
+    ok(100 == length, "got %u\n", length);
+    ok(NULL != data, "got %p\n", data);
+
+    result = IAudioData_SetBuffer(audio_data, 0, buffer, 0);
+    ok(E_INVALIDARG == result, "got 0x%08x\n", result);
+
+    result = IAudioData_SetBuffer(audio_data, sizeof(buffer), buffer, 0);
+    ok(S_OK == result, "got 0x%08x\n", result);
+
+    data = (BYTE *)0xdeadbeef;
+    length = 0xdeadbeef;
+    result = IAudioData_GetInfo(audio_data, &length, &data, NULL);
+    ok(S_OK == result, "got 0x%08x\n", result);
+    ok(sizeof(buffer) == length, "got %u\n", length);
+    ok(buffer == data, "got %p\n", data);
+
+    IAudioData_Release(audio_data);
+
+out_unknown:
+    IUnknown_Release(unknown);
+}
+
 START_TEST(amstream)
 {
     HANDLE file;
@@ -670,6 +717,7 @@ START_TEST(amstream)
 
     test_audiodata_query_interface();
     test_audiodata_get_info();
+    test_audiodata_set_buffer();
 
     CoUninitialize();
 }