[PATCH v4 2/3] cmd: Refactor WCMD_copy_loop out of WCMD_ManualCopy, and stop copy loop at EOF for /a mode.

[Joe Souza]

From: Joe Souza

---
 programs/cmd/builtins.c | 71 +++++++++++++++++++++++++----------------
 1 file changed, 44 insertions(+), 27 deletions(-)

diff --git a/programs/cmd/builtins.c b/programs/cmd/builtins.c
index dd9808aa9f2..8425781d664 100644
--- a/programs/cmd/builtins.c
+++ b/programs/cmd/builtins.c
@@ -540,6 +540,49 @@ end:
   return ret;
 }

+/****************************************************************************
+ * WCMD_copy_loop
+ *
+ * Copies from a file
+ *    optionally reading only until EOF (ascii copy)
+ * Returns TRUE on success
+ */
+static BOOL WCMD_copy_loop(HANDLE in, HANDLE out, BOOL ascii)
+{
+    BOOL   ok;
+    DWORD  bytesread, byteswritten;
+    char *eof = NULL;
+
+    /* Loop copying data from source to destination until EOF read */
+    do
+    {
+        char buffer[MAXSTRING];
+
+        ok = ReadFile(in, buffer, MAXSTRING, &bytesread, NULL);
+        if (ok) {
+
+            /* Stop at first EOF */
+            if (ascii) {
+                eof = (char *)memchr((void *)buffer, '\x1a', bytesread);
+                if (eof) bytesread = (eof - buffer);
+            }
+
+            if (bytesread) {
+                ok = WriteFile(out, buffer, bytesread, &byteswritten, NULL);
+                if (!ok || byteswritten != bytesread) {
+                    WINE_ERR("Unexpected failure writing, rc=%ld\n",
+                             GetLastError());
+                }
+            }
+        } else {
+            WINE_ERR("Unexpected failure reading, rc=%ld\n",
+                     GetLastError());
+        }
+    } while (ok && bytesread > 0 && !eof);
+
+    return ok;
+}
+
 /****************************************************************************
  * WCMD_ManualCopy
  *
@@ -552,7 +595,6 @@ static BOOL WCMD_ManualCopy(WCHAR *srcname, WCHAR *dstname, BOOL ascii, BOOL app
 {
     HANDLE in,out;
     BOOL   ok;
-    DWORD  bytesread, byteswritten;

WINE_TRACE("Manual Copying %s to %s (ascii: %u) (append: %u)\n",
                wine_dbgstr_w(srcname), wine_dbgstr_w(dstname), ascii, append);
@@ -578,32 +620,7 @@ static BOOL WCMD_ManualCopy(WCHAR *srcname, WCHAR *dstname, BOOL ascii, BOOL app
       SetFilePointer(out, 0, NULL, FILE_END);
     }

-    /* Loop copying data from source to destination until EOF read */
-    do
-    {
-      char buffer[MAXSTRING];
-
-      ok = ReadFile(in, buffer, MAXSTRING, &bytesread, NULL);
-      if (ok) {
-
-        /* Stop at first EOF */
-        if (ascii) {
-          char *ptr = (char *)memchr((void *)buffer, '\x1a', bytesread);
-          if (ptr) bytesread = (ptr - buffer);
-        }
-
-        if (bytesread) {
-          ok = WriteFile(out, buffer, bytesread, &byteswritten, NULL);
-          if (!ok || byteswritten != bytesread) {
-            WINE_ERR("Unexpected failure writing to %s, rc=%ld\n",
-                     wine_dbgstr_w(dstname), GetLastError());
-          }
-        }
-      } else {
-        WINE_ERR("Unexpected failure reading from %s, rc=%ld\n",
-                 wine_dbgstr_w(srcname), GetLastError());
-      }
-    } while (ok && bytesread > 0);
+    ok = WCMD_copy_loop(in, out, ascii);

CloseHandle(out);
     CloseHandle(in);