Panic on all possible hash write errors

[gogost.git] / gost3410 / vko2001.go

diff --git a/gost3410/vko2001.go b/gost3410/vko2001.go
index 02f976f9b286e0f27753baef60c7ca2dfe9a8f70..c9aeada21a90a36f782a49497aaa8c4af347a813 100644 (file)
--- a/gost3410/vko2001.go
+++ b/gost3410/vko2001.go
@@ -34,6 +34,8 @@ func (prv *PrivateKey) KEK2001(pub *PublicKey, ukm *big.Int) ([]byte, error) {
                return nil, err
        }
        h := gost341194.New(&gost28147.SboxIdGostR341194CryptoProParamSet)
-       h.Write(key)
+       if _, err = h.Write(key); err != nil {
+               return nil, err
+       }
        return h.Sum(key[:0]), nil
 }