Update namespace major version

[gogost.git] / gost3410 / vko2001.go

diff --git a/gost3410/vko2001.go b/gost3410/vko2001.go
index 02f976f9b286e0f27753baef60c7ca2dfe9a8f70..7ba7ba44373eb08a3b0b803fc9a531da32afe4e0 100644 (file)
--- a/gost3410/vko2001.go
+++ b/gost3410/vko2001.go
@@ -19,21 +19,23 @@ import (
        "errors"
        "math/big"
 
-       "go.cypherpunks.ru/gogost/v4/gost28147"
-       "go.cypherpunks.ru/gogost/v4/gost341194"
+       "go.cypherpunks.ru/gogost/v5/gost28147"
+       "go.cypherpunks.ru/gogost/v5/gost341194"
 )
 
 // RFC 4357 VKO GOST R 34.10-2001 key agreement function.
 // UKM is user keying material, also called VKO-factor.
 func (prv *PrivateKey) KEK2001(pub *PublicKey, ukm *big.Int) ([]byte, error) {
-       if prv.Mode != Mode2001 {
-               return nil, errors.New("gogost/gost3410: KEK2001 can not be used in Mode2012")
+       if prv.C.PointSize() != 32 {
+               return nil, errors.New("gogost/gost3410: KEK2001 is only for 256-bit curves")
        }
        key, err := prv.KEK(pub, ukm)
        if err != nil {
                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
 }