more logic fixes

adamhathcock · Apr 11, 2024 · 373637e · 373637e
1 parent cb22321
commit 373637e
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/SharpCompress/Utility.cs b/src/SharpCompress/Utility.cs
@@ -291,12 +291,12 @@ public static long TransferTo(this Stream source, Stream destination)
     public static long TransferTo(this Stream source, Stream destination, long maxLength)
     {
         var array = GetTransferByteArray();
+        var maxReadSize = array.Length;
         try
         {
             long total = 0;
             var remaining = maxLength;
-            var maxReadSize = array.Length;
-            if (remaining < maxLength)
+            if (remaining < maxReadSize)
             {
                 maxReadSize = (int)remaining;
             }
@@ -309,6 +309,10 @@ public static long TransferTo(this Stream source, Stream destination, long maxLe
                     break;
                 }
                 remaining -= count;
+                if (remaining < maxReadSize)
+                {
+                    maxReadSize = (int)remaining;
+                }
             }
             return total;
         }