Split long lines

[gogost.git] / src / cypherpunks.ru / gogost / gost3410 / 2001_test.go

diff --git a/src/cypherpunks.ru/gogost/gost3410/2001_test.go b/src/cypherpunks.ru/gogost/gost3410/2001_test.go
index 0bc14291ce39be4fed1cb8dda18cc5974afd5851..8bccc62d71e0307d727b6227699b0fd0c82aa036 100644 (file)
--- a/src/cypherpunks.ru/gogost/gost3410/2001_test.go
+++ b/src/cypherpunks.ru/gogost/gost3410/2001_test.go
@@ -1,5 +1,5 @@
 // GoGOST -- Pure Go GOST cryptographic functions library
-// Copyright (C) 2015-2016 Sergey Matveev <stargrave@stargrave.org>
+// Copyright (C) 2015-2019 Sergey Matveev <stargrave@stargrave.org>
 //
 // This program is free software: you can redistribute it and/or modify
 // it under the terms of the GNU General Public License as published by
@@ -63,7 +63,7 @@ func TestRFCVectors(t *testing.T) {
        if err != nil {
                t.FailNow()
        }
-       prv, err := NewPrivateKey(c, DigestSize2001, priv)
+       prv, err := NewPrivateKey(c, Mode2001, priv)
        if err != nil {
                t.FailNow()
        }
@@ -96,7 +96,7 @@ func TestRandom2001(t *testing.T) {
        f := func(data [31]byte, digest [32]byte) bool {
                prv, err := NewPrivateKey(
                        c,
-                       DigestSize2001,
+                       Mode2001,
                        append([]byte{0xde}, data[:]...),
                )
                if err != nil {
@@ -107,7 +107,7 @@ func TestRandom2001(t *testing.T) {
                        return false
                }
                pubRaw := pub.Raw()
-               pub, err = NewPublicKey(c, DigestSize2001, pubRaw)
+               pub, err = NewPublicKey(c, Mode2001, pubRaw)
                if err != nil {
                        return false
                }
@@ -128,7 +128,7 @@ func TestRandom2001(t *testing.T) {
 
 func BenchmarkSign2001(b *testing.B) {
        c, _ := NewCurveFromParams(CurveParamsGostR34102001Test)
-       prv, err := GenPrivateKey(c, DigestSize2001, rand.Reader)
+       prv, err := GenPrivateKey(c, Mode2001, rand.Reader)
        if err != nil {
                b.FailNow()
        }
@@ -142,7 +142,7 @@ func BenchmarkSign2001(b *testing.B) {
 
 func BenchmarkVerify2001(b *testing.B) {
        c, _ := NewCurveFromParams(CurveParamsGostR34102001Test)
-       prv, err := GenPrivateKey(c, DigestSize2001, rand.Reader)
+       prv, err := GenPrivateKey(c, Mode2001, rand.Reader)
        if err != nil {
                b.FailNow()
        }
@@ -161,3 +161,29 @@ func BenchmarkVerify2001(b *testing.B) {
                pub.VerifyDigest(digest, sign)
        }
 }
+
+func TestPrvEqualsTo1(t *testing.T) {
+       c, _ := NewCurveFromParams(CurveParamsGostR34102001Test)
+       prv, err := NewPrivateKey(c, Mode2001, []byte{0x01})
+       if err != nil {
+               t.FailNow()
+       }
+       pub, err := prv.PublicKey()
+       if err != nil {
+               t.FailNow()
+       }
+       digest := []byte{
+               0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
+               0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
+               0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
+               0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01, 0x01,
+       }
+       sign, err := prv.SignDigest(digest, rand.Reader)
+       if err != nil {
+               t.FailNow()
+       }
+       valid, err := pub.VerifyDigest(digest, sign)
+       if err != nil || !valid {
+               t.FailNow()
+       }
+}