src/cypherpunks.ru/gogost/gost28147/cfb_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 TestCFBCryptomanager(t *testing.T) {
  27         key := []byte{
  28                 0x75, 0x71, 0x31, 0x34, 0xB6, 0x0F, 0xEC, 0x45,
  29                 0xA6, 0x07, 0xBB, 0x83, 0xAA, 0x37, 0x46, 0xAF,
  30                 0x4F, 0xF9, 0x9D, 0xA6, 0xD1, 0xB5, 0x3B, 0x5B,
  31                 0x1B, 0x40, 0x2A, 0x1B, 0xAA, 0x03, 0x0D, 0x1B,
  32         }
  33         sbox := &SboxIdGostR341194TestParamSet
  34         pt := []byte{
  35                 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88,
  36                 0x99, 0xAA, 0xBB, 0xCC, 0xDD, 0x80, 0x00, 0x00,
  37         }
  38         ct := []byte{
  39                 0x6E, 0xE8, 0x45, 0x86, 0xDD, 0x2B, 0xCA, 0x0C,
  40                 0xAD, 0x36, 0x16, 0x94, 0x0E, 0x16, 0x42, 0x42,
  41         }
  42         c := NewCipher(key, sbox)
  43         iv := []byte{0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08}
  44         tmp := make([]byte, 16)
  45         fe := c.NewCFBEncrypter(iv)
  46         fe.XORKeyStream(tmp, pt)
  47         if bytes.Compare(tmp, ct) != 0 {
  48                 t.Fail()
  49         }
  50         fd := c.NewCFBDecrypter(iv)
  51         fd.XORKeyStream(tmp, ct)
  52         if bytes.Compare(tmp, pt) != 0 {
  53                 t.Fail()
  54         }
  55 }
  56
  57 func TestCFBRandom(t *testing.T) {
  58         f := func(key [KeySize]byte, iv [BlockSize]byte, pt []byte) bool {
  59                 if len(pt) == 0 {
  60                         return true
  61                 }
  62                 c := NewCipher(key[:], SboxDefault)
  63                 ct := make([]byte, len(pt))
  64                 fe := c.NewCFBEncrypter(iv[:])
  65                 fe.XORKeyStream(ct, pt)
  66                 fd := c.NewCFBDecrypter(iv[:])
  67                 pt2 := make([]byte, len(ct))
  68                 fd.XORKeyStream(pt2, ct)
  69                 return bytes.Compare(pt2, pt) == 0
  70         }
  71         if err := quick.Check(f, nil); err != nil {
  72                 t.Error(err)
  73         }
  74 }
  75
  76 func TestCFBInterface(t *testing.T) {
  77         var key [32]byte
  78         var iv [8]byte
  79         c := NewCipher(key[:], SboxDefault)
  80         var _ cipher.Stream = c.NewCFBEncrypter(iv[:])
  81         var _ cipher.Stream = c.NewCFBDecrypter(iv[:])
  82 }