X-Git-Url: https://git.r.bdr.sh/rbdr/mobius/blobdiff_plain/df1ade5433b027f9cb905e584921692313e647f5..e17f2303e8c1fb940b7550af2b02b6eb35d58f96:/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 } }