X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/c6a3fa2cdbf7eb78964a5df7622482cdd8782d74..d005ef04cfaa26943e6dd33807d741577ffb232a:/hotline/transfer.go?ds=sidebyside 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 } }