src/cypherpunks.ru/gogost/gost28147/cfb.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 type CFBEncrypter struct {
  20         c  *Cipher
  21         iv []byte
  22 }
  23
  24 func (c *Cipher) NewCFBEncrypter(iv []byte) *CFBEncrypter {
  25         if len(iv) != BlockSize {
  26                 panic("iv length is not equal to blocksize")
  27         }
  28         encrypter := CFBEncrypter{c: c, iv: make([]byte, BlockSize)}
  29         copy(encrypter.iv, iv)
  30         return &encrypter
  31 }
  32
  33 func (c *CFBEncrypter) XORKeyStream(dst, src []byte) {
  34         var n int
  35         i := 0
  36 MainLoop:
  37         for {
  38                 c.c.Encrypt(c.iv, c.iv)
  39                 for n = 0; n < BlockSize; n++ {
  40                         if i*BlockSize+n == len(src) {
  41                                 break MainLoop
  42                         }
  43                         c.iv[n] ^= src[i*BlockSize+n]
  44                         dst[i*BlockSize+n] = c.iv[n]
  45                 }
  46                 i++
  47         }
  48         return
  49 }
  50
  51 type CFBDecrypter struct {
  52         c  *Cipher
  53         iv []byte
  54 }
  55
  56 func (c *Cipher) NewCFBDecrypter(iv []byte) *CFBDecrypter {
  57         if len(iv) != BlockSize {
  58                 panic("iv length is not equal to blocksize")
  59         }
  60         decrypter := CFBDecrypter{c: c, iv: make([]byte, BlockSize)}
  61         copy(decrypter.iv, iv)
  62         return &decrypter
  63 }
  64
  65 func (c *CFBDecrypter) XORKeyStream(dst, src []byte) {
  66         var n int
  67         i := 0
  68 MainLoop:
  69         for {
  70                 c.c.Encrypt(c.iv, c.iv)
  71                 for n = 0; n < BlockSize; n++ {
  72                         if i*BlockSize+n == len(src) {
  73                                 break MainLoop
  74                         }
  75                         dst[i*BlockSize+n] = c.iv[n] ^ src[i*BlockSize+n]
  76                         c.iv[n] = src[i*BlockSize+n]
  77                 }
  78                 i++
  79         }
  80         return
  81 }