123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785 |
- package kcp
- import (
- "crypto/aes"
- "crypto/cipher"
- "crypto/des"
- "crypto/sha1"
- "github.com/templexxx/xor"
- "github.com/tjfoc/gmsm/sm4"
- "golang.org/x/crypto/blowfish"
- "golang.org/x/crypto/cast5"
- "golang.org/x/crypto/pbkdf2"
- "golang.org/x/crypto/salsa20"
- "golang.org/x/crypto/tea"
- "golang.org/x/crypto/twofish"
- "golang.org/x/crypto/xtea"
- )
- var (
- initialVector = []byte{167, 115, 79, 156, 18, 172, 27, 1, 164, 21, 242, 193, 252, 120, 230, 107}
- saltxor = `sH3CIVoF#rWLtJo6`
- )
- // BlockCrypt defines encryption/decryption methods for a given byte slice.
- // Notes on implementing: the data to be encrypted contains a builtin
- // nonce at the first 16 bytes
- type BlockCrypt interface {
- // Encrypt encrypts the whole block in src into dst.
- // Dst and src may point at the same memory.
- Encrypt(dst, src []byte)
- // Decrypt decrypts the whole block in src into dst.
- // Dst and src may point at the same memory.
- Decrypt(dst, src []byte)
- }
- type salsa20BlockCrypt struct {
- key [32]byte
- }
- // NewSalsa20BlockCrypt https://en.wikipedia.org/wiki/Salsa20
- func NewSalsa20BlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(salsa20BlockCrypt)
- copy(c.key[:], key)
- return c, nil
- }
- func (c *salsa20BlockCrypt) Encrypt(dst, src []byte) {
- salsa20.XORKeyStream(dst[8:], src[8:], src[:8], &c.key)
- copy(dst[:8], src[:8])
- }
- func (c *salsa20BlockCrypt) Decrypt(dst, src []byte) {
- salsa20.XORKeyStream(dst[8:], src[8:], src[:8], &c.key)
- copy(dst[:8], src[:8])
- }
- type sm4BlockCrypt struct {
- encbuf [sm4.BlockSize]byte
- decbuf [2 * sm4.BlockSize]byte
- block cipher.Block
- }
- // NewSM4BlockCrypt https://github.com/tjfoc/gmsm/tree/master/sm4
- func NewSM4BlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(sm4BlockCrypt)
- block, err := sm4.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *sm4BlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *sm4BlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type twofishBlockCrypt struct {
- encbuf [twofish.BlockSize]byte
- decbuf [2 * twofish.BlockSize]byte
- block cipher.Block
- }
- // NewTwofishBlockCrypt https://en.wikipedia.org/wiki/Twofish
- func NewTwofishBlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(twofishBlockCrypt)
- block, err := twofish.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *twofishBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *twofishBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type tripleDESBlockCrypt struct {
- encbuf [des.BlockSize]byte
- decbuf [2 * des.BlockSize]byte
- block cipher.Block
- }
- // NewTripleDESBlockCrypt https://en.wikipedia.org/wiki/Triple_DES
- func NewTripleDESBlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(tripleDESBlockCrypt)
- block, err := des.NewTripleDESCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *tripleDESBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *tripleDESBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type cast5BlockCrypt struct {
- encbuf [cast5.BlockSize]byte
- decbuf [2 * cast5.BlockSize]byte
- block cipher.Block
- }
- // NewCast5BlockCrypt https://en.wikipedia.org/wiki/CAST-128
- func NewCast5BlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(cast5BlockCrypt)
- block, err := cast5.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *cast5BlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *cast5BlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type blowfishBlockCrypt struct {
- encbuf [blowfish.BlockSize]byte
- decbuf [2 * blowfish.BlockSize]byte
- block cipher.Block
- }
- // NewBlowfishBlockCrypt https://en.wikipedia.org/wiki/Blowfish_(cipher)
- func NewBlowfishBlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(blowfishBlockCrypt)
- block, err := blowfish.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *blowfishBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *blowfishBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type aesBlockCrypt struct {
- encbuf [aes.BlockSize]byte
- decbuf [2 * aes.BlockSize]byte
- block cipher.Block
- }
- // NewAESBlockCrypt https://en.wikipedia.org/wiki/Advanced_Encryption_Standard
- func NewAESBlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(aesBlockCrypt)
- block, err := aes.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *aesBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *aesBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type teaBlockCrypt struct {
- encbuf [tea.BlockSize]byte
- decbuf [2 * tea.BlockSize]byte
- block cipher.Block
- }
- // NewTEABlockCrypt https://en.wikipedia.org/wiki/Tiny_Encryption_Algorithm
- func NewTEABlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(teaBlockCrypt)
- block, err := tea.NewCipherWithRounds(key, 16)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *teaBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *teaBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type xteaBlockCrypt struct {
- encbuf [xtea.BlockSize]byte
- decbuf [2 * xtea.BlockSize]byte
- block cipher.Block
- }
- // NewXTEABlockCrypt https://en.wikipedia.org/wiki/XTEA
- func NewXTEABlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(xteaBlockCrypt)
- block, err := xtea.NewCipher(key)
- if err != nil {
- return nil, err
- }
- c.block = block
- return c, nil
- }
- func (c *xteaBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf[:]) }
- func (c *xteaBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf[:]) }
- type simpleXORBlockCrypt struct {
- xortbl []byte
- }
- // NewSimpleXORBlockCrypt simple xor with key expanding
- func NewSimpleXORBlockCrypt(key []byte) (BlockCrypt, error) {
- c := new(simpleXORBlockCrypt)
- c.xortbl = pbkdf2.Key(key, []byte(saltxor), 32, mtuLimit, sha1.New)
- return c, nil
- }
- func (c *simpleXORBlockCrypt) Encrypt(dst, src []byte) { xor.Bytes(dst, src, c.xortbl) }
- func (c *simpleXORBlockCrypt) Decrypt(dst, src []byte) { xor.Bytes(dst, src, c.xortbl) }
- type noneBlockCrypt struct{}
- // NewNoneBlockCrypt does nothing but copying
- func NewNoneBlockCrypt(key []byte) (BlockCrypt, error) {
- return new(noneBlockCrypt), nil
- }
- func (c *noneBlockCrypt) Encrypt(dst, src []byte) { copy(dst, src) }
- func (c *noneBlockCrypt) Decrypt(dst, src []byte) { copy(dst, src) }
- // packet encryption with local CFB mode
- func encrypt(block cipher.Block, dst, src, buf []byte) {
- switch block.BlockSize() {
- case 8:
- encrypt8(block, dst, src, buf)
- case 16:
- encrypt16(block, dst, src, buf)
- default:
- encryptVariant(block, dst, src, buf)
- }
- }
- // optimized encryption for the ciphers which works in 8-bytes
- func encrypt8(block cipher.Block, dst, src, buf []byte) {
- tbl := buf[:8]
- block.Encrypt(tbl, initialVector)
- n := len(src) / 8
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- s := src[base:][0:64]
- d := dst[base:][0:64]
- // 1
- xor.BytesSrc1(d[0:8], s[0:8], tbl)
- block.Encrypt(tbl, d[0:8])
- // 2
- xor.BytesSrc1(d[8:16], s[8:16], tbl)
- block.Encrypt(tbl, d[8:16])
- // 3
- xor.BytesSrc1(d[16:24], s[16:24], tbl)
- block.Encrypt(tbl, d[16:24])
- // 4
- xor.BytesSrc1(d[24:32], s[24:32], tbl)
- block.Encrypt(tbl, d[24:32])
- // 5
- xor.BytesSrc1(d[32:40], s[32:40], tbl)
- block.Encrypt(tbl, d[32:40])
- // 6
- xor.BytesSrc1(d[40:48], s[40:48], tbl)
- block.Encrypt(tbl, d[40:48])
- // 7
- xor.BytesSrc1(d[48:56], s[48:56], tbl)
- block.Encrypt(tbl, d[48:56])
- // 8
- xor.BytesSrc1(d[56:64], s[56:64], tbl)
- block.Encrypt(tbl, d[56:64])
- base += 64
- }
- switch left {
- case 7:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 6:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 5:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 4:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 3:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 2:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 1:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 8
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
- // optimized encryption for the ciphers which works in 16-bytes
- func encrypt16(block cipher.Block, dst, src, buf []byte) {
- tbl := buf[:16]
- block.Encrypt(tbl, initialVector)
- n := len(src) / 16
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- s := src[base:][0:128]
- d := dst[base:][0:128]
- // 1
- xor.BytesSrc1(d[0:16], s[0:16], tbl)
- block.Encrypt(tbl, d[0:16])
- // 2
- xor.BytesSrc1(d[16:32], s[16:32], tbl)
- block.Encrypt(tbl, d[16:32])
- // 3
- xor.BytesSrc1(d[32:48], s[32:48], tbl)
- block.Encrypt(tbl, d[32:48])
- // 4
- xor.BytesSrc1(d[48:64], s[48:64], tbl)
- block.Encrypt(tbl, d[48:64])
- // 5
- xor.BytesSrc1(d[64:80], s[64:80], tbl)
- block.Encrypt(tbl, d[64:80])
- // 6
- xor.BytesSrc1(d[80:96], s[80:96], tbl)
- block.Encrypt(tbl, d[80:96])
- // 7
- xor.BytesSrc1(d[96:112], s[96:112], tbl)
- block.Encrypt(tbl, d[96:112])
- // 8
- xor.BytesSrc1(d[112:128], s[112:128], tbl)
- block.Encrypt(tbl, d[112:128])
- base += 128
- }
- switch left {
- case 7:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 6:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 5:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 4:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 3:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 2:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 1:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += 16
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
- func encryptVariant(block cipher.Block, dst, src, buf []byte) {
- blocksize := block.BlockSize()
- tbl := buf[:blocksize]
- block.Encrypt(tbl, initialVector)
- n := len(src) / blocksize
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- // 1
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 2
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 3
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 4
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 5
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 6
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 7
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- // 8
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- }
- switch left {
- case 7:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 6:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 5:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 4:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 3:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 2:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 1:
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- block.Encrypt(tbl, dst[base:])
- base += blocksize
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
- // decryption
- func decrypt(block cipher.Block, dst, src, buf []byte) {
- switch block.BlockSize() {
- case 8:
- decrypt8(block, dst, src, buf)
- case 16:
- decrypt16(block, dst, src, buf)
- default:
- decryptVariant(block, dst, src, buf)
- }
- }
- func decrypt8(block cipher.Block, dst, src, buf []byte) {
- tbl := buf[0:8]
- next := buf[8:16]
- block.Encrypt(tbl, initialVector)
- n := len(src) / 8
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- s := src[base:][0:64]
- d := dst[base:][0:64]
- // 1
- block.Encrypt(next, s[0:8])
- xor.BytesSrc1(d[0:8], s[0:8], tbl)
- // 2
- block.Encrypt(tbl, s[8:16])
- xor.BytesSrc1(d[8:16], s[8:16], next)
- // 3
- block.Encrypt(next, s[16:24])
- xor.BytesSrc1(d[16:24], s[16:24], tbl)
- // 4
- block.Encrypt(tbl, s[24:32])
- xor.BytesSrc1(d[24:32], s[24:32], next)
- // 5
- block.Encrypt(next, s[32:40])
- xor.BytesSrc1(d[32:40], s[32:40], tbl)
- // 6
- block.Encrypt(tbl, s[40:48])
- xor.BytesSrc1(d[40:48], s[40:48], next)
- // 7
- block.Encrypt(next, s[48:56])
- xor.BytesSrc1(d[48:56], s[48:56], tbl)
- // 8
- block.Encrypt(tbl, s[56:64])
- xor.BytesSrc1(d[56:64], s[56:64], next)
- base += 64
- }
- switch left {
- case 7:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 6:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 5:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 4:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 3:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 2:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 1:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 8
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
- func decrypt16(block cipher.Block, dst, src, buf []byte) {
- tbl := buf[0:16]
- next := buf[16:32]
- block.Encrypt(tbl, initialVector)
- n := len(src) / 16
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- s := src[base:][0:128]
- d := dst[base:][0:128]
- // 1
- block.Encrypt(next, s[0:16])
- xor.BytesSrc1(d[0:16], s[0:16], tbl)
- // 2
- block.Encrypt(tbl, s[16:32])
- xor.BytesSrc1(d[16:32], s[16:32], next)
- // 3
- block.Encrypt(next, s[32:48])
- xor.BytesSrc1(d[32:48], s[32:48], tbl)
- // 4
- block.Encrypt(tbl, s[48:64])
- xor.BytesSrc1(d[48:64], s[48:64], next)
- // 5
- block.Encrypt(next, s[64:80])
- xor.BytesSrc1(d[64:80], s[64:80], tbl)
- // 6
- block.Encrypt(tbl, s[80:96])
- xor.BytesSrc1(d[80:96], s[80:96], next)
- // 7
- block.Encrypt(next, s[96:112])
- xor.BytesSrc1(d[96:112], s[96:112], tbl)
- // 8
- block.Encrypt(tbl, s[112:128])
- xor.BytesSrc1(d[112:128], s[112:128], next)
- base += 128
- }
- switch left {
- case 7:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 6:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 5:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 4:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 3:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 2:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 1:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += 16
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
- func decryptVariant(block cipher.Block, dst, src, buf []byte) {
- blocksize := block.BlockSize()
- tbl := buf[:blocksize]
- next := buf[blocksize:]
- block.Encrypt(tbl, initialVector)
- n := len(src) / blocksize
- base := 0
- repeat := n / 8
- left := n % 8
- for i := 0; i < repeat; i++ {
- // 1
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- base += blocksize
- // 2
- block.Encrypt(tbl, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], next)
- base += blocksize
- // 3
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- base += blocksize
- // 4
- block.Encrypt(tbl, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], next)
- base += blocksize
- // 5
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- base += blocksize
- // 6
- block.Encrypt(tbl, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], next)
- base += blocksize
- // 7
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- base += blocksize
- // 8
- block.Encrypt(tbl, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], next)
- base += blocksize
- }
- switch left {
- case 7:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 6:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 5:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 4:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 3:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 2:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 1:
- block.Encrypt(next, src[base:])
- xor.BytesSrc1(dst[base:], src[base:], tbl)
- tbl, next = next, tbl
- base += blocksize
- fallthrough
- case 0:
- xor.BytesSrc0(dst[base:], src[base:], tbl)
- }
- }
|