80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package tools
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/cipher"
|
||
"encoding/hex"
|
||
"github.com/tjfoc/gmsm/sm4"
|
||
)
|
||
|
||
// ECB 模式加密结构体
|
||
type ecb struct {
|
||
b cipher.Block
|
||
blockSize int
|
||
}
|
||
|
||
// 实现 ECB 加密
|
||
type ecbEncrypter ecb
|
||
|
||
func newECB(b cipher.Block) *ecb {
|
||
return &ecb{
|
||
b: b,
|
||
blockSize: b.BlockSize(),
|
||
}
|
||
}
|
||
|
||
func newECBEncrypter(b cipher.Block) cipher.BlockMode {
|
||
return (*ecbEncrypter)(newECB(b))
|
||
}
|
||
|
||
func (x *ecbEncrypter) BlockSize() int {
|
||
return x.blockSize
|
||
}
|
||
|
||
func (x *ecbEncrypter) CryptBlocks(dst, src []byte) {
|
||
if len(src)%x.blockSize != 0 {
|
||
panic("crypto/cipher: input not full blocks")
|
||
}
|
||
if len(dst) < len(src) {
|
||
panic("crypto/cipher: output smaller than input")
|
||
}
|
||
for len(src) > 0 {
|
||
x.b.Encrypt(dst, src[:x.blockSize])
|
||
src = src[x.blockSize:]
|
||
dst = dst[x.blockSize:]
|
||
}
|
||
}
|
||
|
||
// PKCS5Padding 补全填充
|
||
func PKCS5Padding(data []byte, blockSize int) []byte {
|
||
padding := blockSize - len(data)%blockSize
|
||
padText := bytes.Repeat([]byte{byte(padding)}, padding)
|
||
return append(data, padText...)
|
||
}
|
||
|
||
// SM4Encrypt 使用 SM4 对手机号进行加密（ECB 模式）
|
||
func SM4Encrypt(key, plaintext string) (string, error) {
|
||
// 将秘钥和明文转换为字节数组
|
||
keyBytes := []byte(key)
|
||
plaintextBytes := []byte(plaintext)
|
||
|
||
// 创建SM4的加密block
|
||
block, err := sm4.NewCipher(keyBytes)
|
||
if err != nil {
|
||
return "", err
|
||
}
|
||
|
||
// 填充明文到块大小（SM4块大小为16字节）
|
||
paddingText := PKCS5Padding(plaintextBytes, block.BlockSize())
|
||
|
||
// 创建 ECB 加密器
|
||
mode := newECBEncrypter(block)
|
||
|
||
// 加密
|
||
ciphertext := make([]byte, len(paddingText))
|
||
mode.CryptBlocks(ciphertext, paddingText)
|
||
|
||
// 将加密后的数据转换为hex编码
|
||
return hex.EncodeToString(ciphertext), nil
|
||
}
|