gost28147/cbc_test.go

   1 // GoGOST -- Pure Go GOST cryptographic functions library
   2 // Copyright (C) 2015-2019 Sergey Matveev <stargrave@stargrave.org>
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU General Public License as published by
   6 // the Free Software Foundation, version 3 of the License.
   7 //
   8 // This program is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  11 // GNU General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU General Public License
  14 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  15
  16 package gost28147
  17
  18 import (
  19         "bytes"
  20         "crypto/cipher"
  21         "testing"
  22         "testing/quick"
  23 )
  24
  25 func TestCBCCrypter(t *testing.T) {
  26         f := func(key [KeySize]byte, iv [BlockSize]byte, pt []byte) bool {
  27                 c := NewCipher(key[:], SboxDefault)
  28                 for i := 0; i < BlockSize; i++ {
  29                         pt = append(pt, pt...)
  30                 }
  31                 ct := make([]byte, len(pt))
  32                 e := cipher.NewCBCEncrypter(c, iv[:])
  33                 e.CryptBlocks(ct, pt)
  34                 d := cipher.NewCBCDecrypter(c, iv[:])
  35                 d.CryptBlocks(ct, ct)
  36                 return bytes.Compare(pt, ct) == 0
  37         }
  38         if err := quick.Check(f, nil); err != nil {
  39                 t.Error(err)
  40         }
  41 }