You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
129 lines
3.2 KiB
129 lines
3.2 KiB
package publicmethod
|
|
|
|
import (
|
|
"appPlatform/overall"
|
|
"crypto/cipher"
|
|
"encoding/hex"
|
|
"strings"
|
|
|
|
"github.com/tjfoc/gmsm/sm4"
|
|
)
|
|
|
|
// 判断字符串长度，多余的删除，不足的补充
|
|
func ProcessString(s string, length int, pad string) string {
|
|
// 如果字符串长度已经符合要求，直接返回
|
|
if len(s) == length {
|
|
return s
|
|
}
|
|
|
|
// 如果字符串长度超过了要求，截取指定长度的部分
|
|
if len(s) > length {
|
|
return s[:length]
|
|
}
|
|
|
|
// 如果字符串长度不足，使用pad字符串进行补充
|
|
return s + padString(pad, length-len(s))
|
|
}
|
|
|
|
// 获取加密随机安全码
|
|
func getSm4IvRandString(length int) (sm4Iv string) {
|
|
for {
|
|
sm4IvStr, err := GenerateRandomString(length)
|
|
if err == nil {
|
|
sm4Iv = sm4IvStr
|
|
break
|
|
}
|
|
}
|
|
return
|
|
|
|
}
|
|
|
|
/*
|
|
*
|
|
@ 作者: 秦东
|
|
@ 时间: 2026-01-20 14:38:09
|
|
@ 功能: SM4加密方法
|
|
@ 参数
|
|
|
|
#data 要加密的字符
|
|
#key 附加选项
|
|
|
|
@ 返回值
|
|
|
|
#加密后的字符
|
|
#error
|
|
*/
|
|
func SM4EncryptNew(data string, class int, key ...string) (string, string, error) {
|
|
SM4Key := overall.CONSTANT_CONFIG.Appsetup.Sm4Key
|
|
if len(key) > 0 {
|
|
SM4Key = strings.Join(key, "-")
|
|
}
|
|
SM4Key = ProcessString(SM4Key, 16, "s")
|
|
//字符串转byte切片
|
|
plainText := []byte(data)
|
|
SM4Iv := getSm4IvRandString(16)
|
|
if class == 1 {
|
|
SM4Iv = overall.CONSTANT_CONFIG.Appsetup.Sm4Token
|
|
}
|
|
sm4Iv16 := ProcessString(SM4Iv, 16, "s")
|
|
iv := []byte(sm4Iv16)
|
|
keySm4 := []byte(SM4Key)
|
|
//实例化sm4加密对象
|
|
block, err := sm4.NewCipher(keySm4)
|
|
if err != nil {
|
|
return "1", SM4Iv, err
|
|
}
|
|
//明文数据填充
|
|
paddingData := paddingLastGroup(plainText, block.BlockSize())
|
|
//声明SM4的加密工作模式
|
|
blockMode := cipher.NewCBCEncrypter(block, iv)
|
|
//为填充后的数据进行加密处理
|
|
cipherText := make([]byte, len(paddingData))
|
|
//使用CryptBlocks这个核心方法，将paddingData进行加密处理，将加密处理后的值赋值到cipherText中
|
|
blockMode.CryptBlocks(cipherText, paddingData)
|
|
//加密结果使用hex转成字符串，方便外部调用
|
|
cipherString := hex.EncodeToString(cipherText)
|
|
return cipherString, SM4Iv, nil
|
|
}
|
|
|
|
/*
|
|
*
|
|
@ 作者: 秦东
|
|
@ 时间: 2026-01-20 14:44:50
|
|
@ 功能: SM4解密新版
|
|
*/
|
|
func SM4DecryptNew(data, sm4IvStr string, key ...string) ([]byte, error) {
|
|
SM4Key := overall.CONSTANT_CONFIG.Appsetup.Sm4Key
|
|
if len(key) > 0 {
|
|
SM4Key = strings.Join(key, "-")
|
|
}
|
|
if len(key) > 0 {
|
|
SM4Key = strings.Join(key, "-")
|
|
}
|
|
SM4Key = ProcessString(SM4Key, 16, "s")
|
|
//sm4随机安全码
|
|
SM4Iv := overall.CONSTANT_CONFIG.Appsetup.Sm4Token
|
|
if sm4IvStr != "" {
|
|
SM4Iv = sm4IvStr
|
|
}
|
|
sm4Iv16 := ProcessString(SM4Iv, 16, "s")
|
|
iv := []byte(sm4Iv16)
|
|
keySm4 := []byte(SM4Key)
|
|
block, err := sm4.NewCipher(keySm4)
|
|
if err != nil {
|
|
return []byte(""), err
|
|
}
|
|
//使用hex解码
|
|
decodeString, err := hex.DecodeString(data)
|
|
if err != nil {
|
|
return []byte(""), err
|
|
}
|
|
//CBC模式 优点：具有较好的安全性，能够隐藏明文的模式和重复性。 缺点：加密过程是串行的，不适合并行处理。
|
|
blockMode := cipher.NewCBCDecrypter(block, iv)
|
|
//下文有详解这段代码的含义
|
|
blockMode.CryptBlocks(decodeString, decodeString)
|
|
//去掉明文后面的填充数据
|
|
plainText := unPaddingLastGroup(decodeString)
|
|
|
|
return plainText, nil
|
|
}
|
|
|