src/cypherpunks.ru/gogost/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, either version 3 of the License, or
   7 // (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 package gost28147
  18
  19 import (
  20         "bytes"
  21         "crypto/cipher"
  22         "testing"
  23         "testing/quick"
  24 )
  25
  26 func TestCBCCrypter(t *testing.T) {
  27         f := func(key [KeySize]byte, iv [BlockSize]byte, pt []byte) bool {
  28                 c := NewCipher(key[:], SboxDefault)
  29                 for i := 0; i < BlockSize; i++ {
  30                         pt = append(pt, pt...)
  31                 }
  32                 ct := make([]byte, len(pt))
  33                 e := cipher.NewCBCEncrypter(c, iv[:])
  34                 e.CryptBlocks(ct, pt)
  35                 d := cipher.NewCBCDecrypter(c, iv[:])
  36                 d.CryptBlocks(ct, ct)
  37                 return bytes.Compare(pt, ct) == 0
  38         }
  39         if err := quick.Check(f, nil); err != nil {
  40                 t.Error(err)
  41         }
  42 }