src/cypherpunks.ru/gogost/gost28147/cbc_test.go

   1 // GoGOST -- Pure Go GOST cryptographic functions library
   2 // Copyright (C) 2015-2016 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         "crypto/rand"
  23         "testing"
  24         "testing/quick"
  25 )
  26
  27 func TestCBCCrypter(t *testing.T) {
  28         var key [KeySize]byte
  29         var iv [BlockSize]byte
  30         rand.Read(key[:])
  31         rand.Read(iv[:])
  32         c := NewCipher(key, SboxDefault)
  33         f := func(pt []byte) bool {
  34                 for i := 0; i < BlockSize; i++ {
  35                         pt = append(pt, pt...)
  36                 }
  37                 ct := make([]byte, len(pt))
  38                 e := cipher.NewCBCEncrypter(c, iv[:])
  39                 e.CryptBlocks(ct, pt)
  40                 d := cipher.NewCBCDecrypter(c, iv[:])
  41                 d.CryptBlocks(ct, ct)
  42                 return bytes.Compare(pt, ct) == 0
  43         }
  44         if err := quick.Check(f, nil); err != nil {
  45                 t.Error(err)
  46         }
  47 }