aboutsummaryrefslogtreecommitdiff
path: root/zsig/sum.go
diff options
context:
space:
mode:
Diffstat (limited to 'zsig/sum.go')
-rw-r--r--zsig/sum.go7
1 files changed, 5 insertions, 2 deletions
diff --git a/zsig/sum.go b/zsig/sum.go
index 7d706e8..846a140 100644
--- a/zsig/sum.go
+++ b/zsig/sum.go
@@ -6,11 +6,14 @@ import (
)
// Sum calculates SHA512/256
-func Sum(r io.Reader) chan []byte {
+func Sum(r io.Reader, blockSize int) chan []byte {
c := make(chan []byte, 1)
+ if blockSize == 0 {
+ blockSize = BlockSize
+ }
go func() {
defer close(c)
- buf := make([]byte, BlockSz)
+ buf := make([]byte, blockSize)
for {
n, err := io.ReadFull(r, buf)
if err == io.EOF {