Nikolay Sivov : shcore/stream: Implement generic CopyTo().

[Alexandre Julliard]

Module: wine
Branch: master
Commit: e7f60e39f68f9190477a9e022c4f785724d55ae6
URL:    https://source.winehq.org/git/wine.git/?a=commit;h=e7f60e39f68f9190477a9e022c4f785724d55ae6

Author: Nikolay Sivov <nsivov at codeweavers.com>
Date:   Mon Apr 26 08:51:39 2021 +0300

shcore/stream: Implement generic CopyTo().

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

---

 dlls/shcore/main.c | 48 +++++++++++++++++++++++++++++++++++++++++-------
 1 file changed, 41 insertions(+), 7 deletions(-)

diff --git a/dlls/shcore/main.c b/dlls/shcore/main.c
index fd21db423ea..82b236bb47d 100644
--- a/dlls/shcore/main.c
+++ b/dlls/shcore/main.c
@@ -695,20 +695,54 @@ static HRESULT WINAPI memstream_SetSize(IStream *iface, ULARGE_INTEGER new_size)
     return S_OK;
 }
 
-static HRESULT WINAPI shstream_CopyTo(IStream *iface, IStream *pstm, ULARGE_INTEGER size, ULARGE_INTEGER *read_len, ULARGE_INTEGER *written)
+static HRESULT WINAPI shstream_CopyTo(IStream *iface, IStream *dest, ULARGE_INTEGER size,
+        ULARGE_INTEGER *read_len, ULARGE_INTEGER *written)
 {
     struct shstream *stream = impl_from_IStream(iface);
+    ULARGE_INTEGER total_read, total_written;
+    HRESULT hr = S_OK;
+    BYTE buffer[0x400];
 
-    TRACE("(%p)\n", stream);
+    TRACE("(%p, %p, %s, %p, %p)\n", stream, dest, wine_dbgstr_longlong(size.QuadPart), read_len, written);
 
-    if (read_len)
-        read_len->QuadPart = 0;
+    if (!dest)
+        return E_POINTER;
+
+    total_read.QuadPart = 0;
+    total_written.QuadPart = 0;
+
+    while (size.QuadPart > 0)
+    {
+        ULONG chunk_size = size.QuadPart >= sizeof(buffer) ? sizeof(buffer) : size.u.LowPart;
+        ULONG chunk_read, chunk_written;
+
+        hr = IStream_Read(iface, buffer, chunk_size, &chunk_read);
+        if (FAILED(hr))
+            break;
 
+        total_read.QuadPart += chunk_read;
+
+        if (chunk_read)
+        {
+            hr = IStream_Write(dest, buffer, chunk_read, &chunk_written);
+            if (FAILED(hr))
+                break;
+
+            total_written.QuadPart += chunk_written;
+        }
+
+        if (chunk_read != chunk_size)
+            size.QuadPart = 0;
+        else
+            size.QuadPart -= chunk_read;
+    }
+
+    if (read_len)
+        read_len->QuadPart = total_read.QuadPart;
     if (written)
-        written->QuadPart = 0;
+        written->QuadPart = total_written.QuadPart;
 
-    /* TODO implement */
-    return E_NOTIMPL;
+    return hr;
 }
 
 static HRESULT WINAPI shstream_Commit(IStream *iface, DWORD flags)