103 lines
3.5 KiB
Go
103 lines
3.5 KiB
Go
// 提供3DES-EDE加解密算法，本解密算法采用`(K1, K2, K3)`的安全长密钥。
|
||
package tdes
|
||
|
||
import (
|
||
"crypto/cipher"
|
||
"crypto/des"
|
||
"crypto/sha256"
|
||
"fmt"
|
||
|
||
"archgrid.xyz/ag/toolsbox/encryption"
|
||
)
|
||
|
||
type KeyGenerator func([]byte) [3][8]byte
|
||
|
||
// 直接采用分组的方式，将密钥分为4个部分，取前三个部分作为3DES的密钥。
|
||
func PartitionKeyGenerator(key []byte) [3][8]byte {
|
||
hashedKey := sha256.Sum256(key)
|
||
return [3][8]byte{
|
||
([8]byte)(hashedKey[0:8]),
|
||
([8]byte)(hashedKey[8:16]),
|
||
([8]byte)(hashedKey[16:24]),
|
||
}
|
||
}
|
||
|
||
// 对给定密钥进行Sha256散列以后，将散列结果分为三部分，分别作为3DES的密钥，前三组分别与第四组异或，得到3组新的密钥。
|
||
func XorKeyGenerator(key []byte) [3][8]byte {
|
||
hashedKey := sha256.Sum256(key)
|
||
var result [3][8]byte
|
||
for i := 0; i < 3; i++ {
|
||
for j := 0; j < 8; j++ {
|
||
result[i][j] = hashedKey[i*8+j] ^ hashedKey[3*8+j]
|
||
}
|
||
}
|
||
return result
|
||
}
|
||
|
||
// 对给定的数据使用给定的密钥进行加密。IV与密钥相同。
|
||
func rawDesEncrypt(data []byte, key [8]byte) ([]byte, error) {
|
||
block, err := des.NewCipher(key[:])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("创建加密单元失败，%w", err)
|
||
}
|
||
iv := key[:]
|
||
cipherText := make([]byte, len(data))
|
||
mode := cipher.NewCBCEncrypter(block, iv)
|
||
mode.CryptBlocks(cipherText, data)
|
||
return cipherText, nil
|
||
}
|
||
|
||
// 对给定的数据使用给定的密钥进行解密。IV与密钥相同。
|
||
func rawDesDecrypt(data []byte, key [8]byte) ([]byte, error) {
|
||
block, err := des.NewCipher(key[:])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("创建加密单元失败，%w", err)
|
||
}
|
||
iv := key[:]
|
||
plainText := make([]byte, len(data))
|
||
mode := cipher.NewCBCDecrypter(block, iv)
|
||
mode.CryptBlocks(plainText, data)
|
||
return plainText, nil
|
||
}
|
||
|
||
// 对给定的数据进行加密。
|
||
// 3DES-EDE加密IV为8字节，默认与运算后的密钥相同。
|
||
// 参数padding指定了填充方式，可选值为NoPadding、ZeroPadding和PKCS7Padding，默认采用ZeroPadding。
|
||
func Encrypt(data []byte, key []byte, padding encryption.PaddingMode, keyGenerator ...KeyGenerator) ([]byte, error) {
|
||
desKeys := append(keyGenerator, XorKeyGenerator)[0](key)
|
||
plainText := encryption.Padding(data, des.BlockSize, padding)
|
||
cipher1Text, err := rawDesEncrypt(plainText, desKeys[0])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第一阶段加密失败，%w", err)
|
||
}
|
||
plain2Text, err := rawDesDecrypt(cipher1Text, desKeys[1])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第二阶段解密失败，%w", err)
|
||
}
|
||
cipher3Text, err := rawDesEncrypt(plain2Text, desKeys[2])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第三阶段加密失败，%w", err)
|
||
}
|
||
return cipher3Text, nil
|
||
}
|
||
|
||
// 对给定的数据进行解密。
|
||
// 3DES-EDE解密IV为8字节，默认与运算后的密钥相同。
|
||
// 参数padding指定了填充方式，可选值为NoPadding、ZeroPadding和PKCS7Padding，默认采用ZeroPadding。
|
||
func Decrypt(data []byte, key []byte, padding encryption.PaddingMode, keyGenerator ...KeyGenerator) ([]byte, error) {
|
||
desKeys := append(keyGenerator, XorKeyGenerator)[0](key)
|
||
plainText, err := rawDesDecrypt(data, desKeys[2])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第三阶段解密失败，%w", err)
|
||
}
|
||
cipher2Text, err := rawDesEncrypt(plainText, desKeys[1])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第二阶段加密失败，%w", err)
|
||
}
|
||
plain3Text, err := rawDesDecrypt(cipher2Text, desKeys[0])
|
||
if err != nil {
|
||
return nil, fmt.Errorf("第一阶段解密失败，%w", err)
|
||
}
|
||
return encryption.Unpadding(plain3Text, des.BlockSize, padding), nil
|
||
}
|