Merge pull request #103 from jhalter/handle_tracker_list_errors

[rbdr/mobius] / hotline / transfer.go

diff --git a/hotline/transfer.go b/hotline/transfer.go
index 4c80581511bbd1276a7bcf5cd0ef309519b9de89..2acb5d6b6e6b689c59e776daca8db8e2ef2f5f28 100644 (file)
--- 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
                }
        }