From 1a7d1fb948f08e940aed67070f5313799f19e6d0 Mon Sep 17 00:00:00 2001
From: Jeff Halter <868228+jhalter@users.noreply.github.com>
Date: Sun, 26 Jun 2022 08:31:18 -0700
Subject: Fix file upload stall

---
 hotline/transfer.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'hotline')

diff --git a/hotline/transfer.go b/hotline/transfer.go
index 4c80581..2acb5d6 100644
--- a/hotline/transfer.go
+++ b/hotline/transfer.go
@@ -42,7 +42,7 @@ func receiveFile(r io.Reader, targetFile, resForkFile, infoFork, counterWriter i
 		return err
 	}
 
-	if _, err = io.Copy(targetFile, io.TeeReader(r, counterWriter)); err != nil {
+	if _, err = io.CopyN(targetFile, io.TeeReader(r, counterWriter), ffo.dataSize()); err != nil {
 		return err
 	}
 
@@ -51,7 +51,7 @@ func receiveFile(r io.Reader, targetFile, resForkFile, infoFork, counterWriter i
 			return err
 		}
 
-		if _, err = io.Copy(resForkFile, io.TeeReader(r, counterWriter)); err != nil {
+		if _, err = io.CopyN(resForkFile, io.TeeReader(r, counterWriter), ffo.rsrcSize()); err != nil {
 			return err
 		}
 	}
-- 
cgit