des.go

package xcrypto

import (
	"crypto/cipher"
	"crypto/des"
	"encoding/hex"

	"github.com/fufuok/utils"
	"github.com/fufuok/utils/base58"
)

// DesCBCEnStringHex 加密, ZerosPadding
func DesCBCEnStringHex(s string, key []byte) string {
	return hex.EncodeToString(DesCBCEncrypt(false, utils.S2B(s), key))
}

// DesCBCEnPKCS7StringHex 加密, Pkcs7Padding
func DesCBCEnPKCS7StringHex(s string, key []byte) string {
	return hex.EncodeToString(DesCBCEncrypt(true, utils.S2B(s), key))
}

// DesCBCEnHex 加密, ZerosPadding
func DesCBCEnHex(b, key []byte) string {
	return hex.EncodeToString(DesCBCEncrypt(false, b, key))
}

// DesCBCEnPKCS7Hex 加密, Pkcs7Padding
func DesCBCEnPKCS7Hex(b, key []byte) string {
	return hex.EncodeToString(DesCBCEncrypt(true, b, key))
}

// DesCBCDeStringHex 解密, ZerosPadding
func DesCBCDeStringHex(s string, key []byte) string {
	return utils.B2S(DesCBCDeHex(s, key))
}

// DesCBCDePKCS7StringHex 解密, Pkcs7Padding
func DesCBCDePKCS7StringHex(s string, key []byte) string {
	return utils.B2S(DesCBCDePKCS7Hex(s, key))
}

// DesCBCDeHex 解密, ZerosPadding
func DesCBCDeHex(s string, key []byte) []byte {
	if data, err := hex.DecodeString(s); err == nil {
		return DesCBCDecrypt(false, data, key)
	}

	return nil
}

// DesCBCDePKCS7Hex 解密, Pkcs7Padding
func DesCBCDePKCS7Hex(s string, key []byte) []byte {
	if data, err := hex.DecodeString(s); err == nil {
		return DesCBCDecrypt(true, data, key)
	}

	return nil
}

// DesCBCEnStringB58 加密, ZerosPadding
func DesCBCEnStringB58(s string, key []byte) string {
	return base58.Encode(DesCBCEncrypt(false, utils.S2B(s), key))
}

// DesCBCEnPKCS7StringB58 加密, Pkcs7Padding
func DesCBCEnPKCS7StringB58(s string, key []byte) string {
	return base58.Encode(DesCBCEncrypt(true, utils.S2B(s), key))
}

// DesCBCEnB58 加密, ZerosPadding
func DesCBCEnB58(b, key []byte) string {
	return base58.Encode(DesCBCEncrypt(false, b, key))
}

// DesCBCEnPKCS7B58 加密, Pkcs7Padding
func DesCBCEnPKCS7B58(b, key []byte) string {
	return base58.Encode(DesCBCEncrypt(true, b, key))
}

// DesCBCDeStringB58 解密, ZerosPadding
func DesCBCDeStringB58(s string, key []byte) string {
	return utils.B2S(DesCBCDeB58(s, key))
}

// DesCBCDePKCS7StringB58 解密, Pkcs7Padding
func DesCBCDePKCS7StringB58(s string, key []byte) string {
	return utils.B2S(DesCBCDePKCS7B58(s, key))
}

// DesCBCDeB58 解密, ZerosPadding
func DesCBCDeB58(s string, key []byte) []byte {
	return DesCBCDecrypt(false, base58.Decode(s), key)
}

// DesCBCDePKCS7B58 解密, Pkcs7Padding
func DesCBCDePKCS7B58(s string, key []byte) []byte {
	return DesCBCDecrypt(true, base58.Decode(s), key)
}

// DesCBCEnStringB64 加密, ZerosPadding
func DesCBCEnStringB64(s string, key []byte) string {
	return utils.B64UrlEncode(DesCBCEncrypt(false, utils.S2B(s), key))
}

// DesCBCEnPKCS7StringB64 加密, Pkcs7Padding
func DesCBCEnPKCS7StringB64(s string, key []byte) string {
	return utils.B64UrlEncode(DesCBCEncrypt(true, utils.S2B(s), key))
}

// DesCBCEnB64 加密, ZerosPadding
func DesCBCEnB64(b, key []byte) string {
	return utils.B64UrlEncode(DesCBCEncrypt(false, b, key))
}

// DesCBCEnPKCS7B64 加密, Pkcs7Padding
func DesCBCEnPKCS7B64(b, key []byte) string {
	return utils.B64UrlEncode(DesCBCEncrypt(true, b, key))
}

// DesCBCDeStringB64 解密, ZerosPadding
func DesCBCDeStringB64(s string, key []byte) string {
	return utils.B2S(DesCBCDeB64(s, key))
}

// DesCBCDePKCS7StringB64 解密, Pkcs7Padding
func DesCBCDePKCS7StringB64(s string, key []byte) string {
	return utils.B2S(DesCBCDePKCS7B64(s, key))
}

// DesCBCDeB64 解密, ZerosPadding
func DesCBCDeB64(s string, key []byte) []byte {
	return DesCBCDecrypt(false, utils.B64UrlDecode(s), key)
}

// DesCBCDePKCS7B64 解密, Pkcs7Padding
func DesCBCDePKCS7B64(s string, key []byte) []byte {
	return DesCBCDecrypt(true, utils.B64UrlDecode(s), key)
}

// DesCBCEncrypt AES-CBC 加密
func DesCBCEncrypt(asPKCS7 bool, plaintext, key []byte, ivs ...[]byte) (ciphertext []byte) {
	ciphertext, _ = DesCBCEncryptE(asPKCS7, plaintext, key, ivs...)
	return
}

// DesCBCDecrypt AES-CBC 解密
func DesCBCDecrypt(asPKCS7 bool, ciphertext, key []byte, ivs ...[]byte) (plaintext []byte) {
	plaintext, _ = DesCBCDecryptE(asPKCS7, ciphertext, key, ivs...)
	return
}

// DesCBCEncryptE DES-CBC 加密, 密码分组链接模式 (Cipher Block Chaining (CBC))
// key 长度固定为 8
// asPKCS7: false (ZerosPadding), true (Pkcs7Padding)
func DesCBCEncryptE(asPKCS7 bool, plaintext, key []byte, ivs ...[]byte) ([]byte, error) {
	block, err := des.NewCipher(key)
	if err != nil {
		return nil, err
	}

	defer func() {
		if r := recover(); r != nil {
			return
		}
	}()

	bSize := block.BlockSize()
	plaintext = Padding(plaintext, bSize, asPKCS7)

	// 向量无效时自动为 key[:blockSize]
	var iv []byte
	if len(ivs) > 0 && len(ivs[0]) == bSize {
		iv = ivs[0]
	} else {
		iv = key[:bSize]
	}

	ciphertext := make([]byte, len(plaintext))
	mode := cipher.NewCBCEncrypter(block, iv)
	mode.CryptBlocks(ciphertext, plaintext)

	return ciphertext, nil
}

// DesCBCDecryptE DES-CBC 解密, 密码分组链接模式 (Cipher Block Chaining (CBC))
func DesCBCDecryptE(asPKCS7 bool, ciphertext, key []byte, ivs ...[]byte) ([]byte, error) {
	block, err := des.NewCipher(key)
	if err != nil {
		return nil, err
	}

	defer func() {
		if r := recover(); r != nil {
			return
		}
	}()

	bSize := block.BlockSize()

	// 向量无效时自动为 key[:blockSize]
	var iv []byte
	if len(ivs) > 0 && len(ivs[0]) == bSize {
		iv = ivs[0]
	} else {
		iv = key[:bSize]
	}

	plaintext := make([]byte, len(ciphertext))
	mode := cipher.NewCBCDecrypter(block, iv)
	mode.CryptBlocks(plaintext, ciphertext)
	plaintext = UnPadding(plaintext, asPKCS7)

	return plaintext, nil
}