GOST Streebog hash algorithms support

[meta4ra.git] / cmd / meta4-create / main.go

diff --git a/cmd/meta4-create/main.go b/cmd/meta4-create/main.go
index 0d54569e88490ca1951abb2d63f944332f56fe56..2c3a5ff0aaeae8d7500c5180569c95e5d940c66c 100644 (file)
--- a/cmd/meta4-create/main.go
+++ b/cmd/meta4-create/main.go
@@ -25,15 +25,66 @@ import (
        "encoding/hex"
        "encoding/xml"
        "flag"
+       "hash"
        "io"
        "log"
        "os"
        "path/filepath"
+       "sync"
        "time"
 
+       "go.cypherpunks.ru/gogost/v5/gost34112012256"
+       "go.cypherpunks.ru/gogost/v5/gost34112012512"
        "go.stargrave.org/meta4ra"
 )
 
+type MultiHasher struct {
+       sha256h      hash.Hash
+       sha512h      hash.Hash
+       streebog256h hash.Hash
+       streebog512h hash.Hash
+       g            sync.WaitGroup
+}
+
+func NewMultiHasher() *MultiHasher {
+       return &MultiHasher{
+               sha256h:      sha256.New(),
+               sha512h:      sha512.New(),
+               streebog256h: gost34112012256.New(),
+               streebog512h: gost34112012512.New(),
+       }
+}
+
+func (h *MultiHasher) Write(p []byte) (n int, err error) {
+       h.g.Add(4)
+       go func() {
+               if _, err := h.sha256h.Write(p); err != nil {
+                       panic(err)
+               }
+               h.g.Done()
+       }()
+       go func() {
+               if _, err := h.sha512h.Write(p); err != nil {
+                       panic(err)
+               }
+               h.g.Done()
+       }()
+       go func() {
+               if _, err := h.streebog256h.Write(p); err != nil {
+                       panic(err)
+               }
+               h.g.Done()
+       }()
+       go func() {
+               if _, err := h.streebog512h.Write(p); err != nil {
+                       panic(err)
+               }
+               h.g.Done()
+       }()
+       h.g.Wait()
+       return len(p), nil
+}
+
 func main() {
        file := flag.String("file", "", "Path to file")
        mtime := flag.String("mtime", "", "Take that file's mtime as a Published date")
@@ -54,13 +105,9 @@ func main() {
        if err != nil {
                log.Fatalln(err)
        }
-       sha256Hasher := sha256.New()
-       sha512Hasher := sha512.New()
-       _, err = io.Copy(
-               io.MultiWriter(sha256Hasher, sha512Hasher),
-               bufio.NewReader(fd),
-       )
-       if err != nil {
+       buf := make([]byte, 1<<20)
+       h := NewMultiHasher()
+       if _, err = io.CopyBuffer(h, bufio.NewReader(fd), buf); err != nil {
                log.Fatalln(err)
        }
        f := meta4ra.File{
@@ -69,8 +116,10 @@ func main() {
                Size:        uint64(fi.Size()),
                URLs:        urls,
                Hashes: []meta4ra.Hash{
-                       {Type: meta4ra.HashSHA256, Hash: hex.EncodeToString(sha256Hasher.Sum(nil))},
-                       {Type: meta4ra.HashSHA512, Hash: hex.EncodeToString(sha512Hasher.Sum(nil))},
+                       {Type: meta4ra.HashSHA256, Hash: hex.EncodeToString(h.sha256h.Sum(nil))},
+                       {Type: meta4ra.HashSHA512, Hash: hex.EncodeToString(h.sha512h.Sum(nil))},
+                       {Type: meta4ra.HashStreebog256, Hash: hex.EncodeToString(h.streebog256h.Sum(nil))},
+                       {Type: meta4ra.HashStreebog512, Hash: hex.EncodeToString(h.streebog512h.Sum(nil))},
                },
        }
        if *sig != "" {