Update namespace major version

[gogost.git] / gost3410 / vko2012.go

diff --git a/gost3410/vko2012.go b/gost3410/vko2012.go
index 01f3e85770df3c7c14769845baea846b850cba82..76fa0a4d15138e57581e776dadefe020d3ae6adb 100644 (file)
--- a/gost3410/vko2012.go
+++ b/gost3410/vko2012.go
@@ -18,8 +18,8 @@ package gost3410
 import (
        "math/big"
 
-       "go.cypherpunks.ru/gogost/v4/gost34112012256"
-       "go.cypherpunks.ru/gogost/v4/gost34112012512"
+       "go.cypherpunks.ru/gogost/v5/gost34112012256"
+       "go.cypherpunks.ru/gogost/v5/gost34112012512"
 )
 
 // RFC 7836 VKO GOST R 34.10-2012 256-bit key agreement function.
@@ -30,7 +30,9 @@ func (prv *PrivateKey) KEK2012256(pub *PublicKey, ukm *big.Int) ([]byte, error)
                return nil, err
        }
        h := gost34112012256.New()
-       h.Write(key)
+       if _, err = h.Write(key); err != nil {
+               return nil, err
+       }
        return h.Sum(key[:0]), nil
 }
 
@@ -42,6 +44,8 @@ func (prv *PrivateKey) KEK2012512(pub *PublicKey, ukm *big.Int) ([]byte, error)
                return nil, err
        }
        h := gost34112012512.New()
-       h.Write(key)
+       if _, err = h.Write(key); err != nil {
+               return nil, err
+       }
        return h.Sum(key[:0]), nil
 }