From 0ced3ecc87f19aa35f9e6919ca822ac71f57b20a Mon Sep 17 00:00:00 2001 From: Dimitri Sokolyuk Date: Wed, 26 Jul 2017 00:09:58 +0200 Subject: ... --- zsig/sum.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'zsig') diff --git a/zsig/sum.go b/zsig/sum.go index 5768b95..40bf24c 100644 --- a/zsig/sum.go +++ b/zsig/sum.go @@ -14,11 +14,17 @@ const ( ) func (z ZHeader) Verify(r io.Reader) error { + if z.Alg != Alg { + return errors.New("invalid hash algorithm") + } h := sha512.New512_256() // from z.Alg blockCount := len(z.Sums) for _, sum := range z.Sums { h.Reset() - _, err := io.CopyN(h, r, z.BlockSize) + n, err := io.CopyN(h, r, z.BlockSize) + if n == 0 && err == io.EOF { + break + } if err != nil && err != io.EOF { return err } -- cgit v1.2.3