github/kcp-go
4
0
Fork 0
mirror of https://github.com/xtaci/kcp-go.git synced 2026-10-21 06:43:27 +00:00
Code Issues Releases Wiki Activity

pisces

Browse Source
This commit is contained in:
xtaci
2016-08-20 12:14:07 +08:00
parent 50597b1e10
commit fbb6bf6788
2 changed files with 58 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crypt.go
+41 -42
View File
@@ -10,6 +10,7 @@ import (
"golang.org/x/crypto/cast5"
"golang.org/x/crypto/pbkdf2"
"golang.org/x/crypto/tea"
"golang.org/x/crypto/twofish"
)
var (
@@ -28,6 +29,32 @@ type BlockCrypt interface {
Decrypt(dst, src []byte)
}
// TwofishBlockCrypt implements BlockCrypt
type TwofishBlockCrypt struct {
encbuf []byte
decbuf []byte
block cipher.Block
}
// NewTwofishBlockCrypt initates AES BlockCrypt by the given key
func NewTwofishBlockCrypt(key []byte) (BlockCrypt, error) {
c := new(TwofishBlockCrypt)
block, err := twofish.NewCipher(key)
if err != nil {
return nil, err
}
c.block = block
c.encbuf = make([]byte, twofish.BlockSize)
c.decbuf = make([]byte, 2*twofish.BlockSize)
return c, nil
}
// Encrypt implements Encrypt interface
func (c *TwofishBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *TwofishBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// TripleDESBlockCrypt implements BlockCrypt
type TripleDESBlockCrypt struct {
encbuf []byte
@@ -49,14 +76,10 @@ func NewTripleDESBlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *TripleDESBlockCrypt) Encrypt(dst, src []byte) {
encrypt(c.block, dst, src, c.encbuf)
}
func (c *TripleDESBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *TripleDESBlockCrypt) Decrypt(dst, src []byte) {
decrypt(c.block, dst, src, c.decbuf)
}
func (c *TripleDESBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// Cast5BlockCrypt implements BlockCrypt
type Cast5BlockCrypt struct {
@@ -79,14 +102,10 @@ func NewCast5BlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *Cast5BlockCrypt) Encrypt(dst, src []byte) {
encrypt(c.block, dst, src, c.encbuf)
}
func (c *Cast5BlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *Cast5BlockCrypt) Decrypt(dst, src []byte) {
decrypt(c.block, dst, src, c.decbuf)
}
func (c *Cast5BlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// BlowfishBlockCrypt implements BlockCrypt
type BlowfishBlockCrypt struct {
@@ -109,14 +128,10 @@ func NewBlowfishBlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *BlowfishBlockCrypt) Encrypt(dst, src []byte) {
encrypt(c.block, dst, src, c.encbuf)
}
func (c *BlowfishBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *BlowfishBlockCrypt) Decrypt(dst, src []byte) {
decrypt(c.block, dst, src, c.decbuf)
}
func (c *BlowfishBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// AESBlockCrypt implements BlockCrypt with AES
type AESBlockCrypt struct {
@@ -139,14 +154,10 @@ func NewAESBlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *AESBlockCrypt) Encrypt(dst, src []byte) {
encrypt(c.block, dst, src, c.encbuf)
}
func (c *AESBlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *AESBlockCrypt) Decrypt(dst, src []byte) {
decrypt(c.block, dst, src, c.decbuf)
}
func (c *AESBlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// TEABlockCrypt implements BlockCrypt with TEA
type TEABlockCrypt struct {
@@ -169,14 +180,10 @@ func NewTEABlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *TEABlockCrypt) Encrypt(dst, src []byte) {
encrypt(c.block, dst, src, c.encbuf)
}
func (c *TEABlockCrypt) Encrypt(dst, src []byte) { encrypt(c.block, dst, src, c.encbuf) }
// Decrypt implements Decrypt interface
func (c *TEABlockCrypt) Decrypt(dst, src []byte) {
decrypt(c.block, dst, src, c.decbuf)
}
func (c *TEABlockCrypt) Decrypt(dst, src []byte) { decrypt(c.block, dst, src, c.decbuf) }
// SimpleXORBlockCrypt implements BlockCrypt with simple xor to a table
type SimpleXORBlockCrypt struct {
@@ -191,14 +198,10 @@ func NewSimpleXORBlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *SimpleXORBlockCrypt) Encrypt(dst, src []byte) {
xorBytes(dst, src, c.xortbl)
}
func (c *SimpleXORBlockCrypt) Encrypt(dst, src []byte) { xorBytes(dst, src, c.xortbl) }
// Decrypt implements Decrypt interface
func (c *SimpleXORBlockCrypt) Decrypt(dst, src []byte) {
xorBytes(dst, src, c.xortbl)
}
func (c *SimpleXORBlockCrypt) Decrypt(dst, src []byte) { xorBytes(dst, src, c.xortbl) }
// NoneBlockCrypt simple returns the plaintext
type NoneBlockCrypt struct {
@@ -211,14 +214,10 @@ func NewNoneBlockCrypt(key []byte) (BlockCrypt, error) {
}
// Encrypt implements Encrypt interface
func (c *NoneBlockCrypt) Encrypt(dst, src []byte) {
copy(dst, src)
}
func (c *NoneBlockCrypt) Encrypt(dst, src []byte) { copy(dst, src) }
// Decrypt implements Decrypt interface
func (c *NoneBlockCrypt) Decrypt(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) {
crypt_test.go
+17
View File
@@ -135,3 +135,20 @@ func TestTripleDES(t *testing.T) {
t.Fail()
}
}
func TestTwofish(t *testing.T) {
pass := pbkdf2.Key(key, []byte(salt), 4096, 32, sha1.New)
bc, err := NewTwofishBlockCrypt(pass)
if err != nil {
t.Fatal(err)
}
data := make([]byte, mtuLimit)
io.ReadFull(rand.Reader, data)
dec := make([]byte, mtuLimit)
enc := make([]byte, mtuLimit)
bc.Encrypt(enc, data)
bc.Decrypt(dec, enc)
if !bytes.Equal(data, dec) {
t.Fail()
}
}