"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")
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{
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 != "" {