疯狂的狮子Li
2023-08-07 86a8f5a700a19d8e55da5f5f0d5edd6b1284934f
ruoyi-common/ruoyi-common-encrypt/src/main/java/org/dromara/common/encrypt/core/encryptor/Sm2Encryptor.java
@@ -1,13 +1,10 @@
package org.dromara.common.encrypt.core.encryptor;
import cn.hutool.core.codec.Base64;
import cn.hutool.crypto.SmUtil;
import cn.hutool.crypto.asymmetric.KeyType;
import cn.hutool.crypto.asymmetric.SM2;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.encrypt.core.EncryptContext;
import org.dromara.common.encrypt.enumd.AlgorithmType;
import org.dromara.common.encrypt.enumd.EncodeType;
import org.dromara.common.encrypt.utils.EncryptUtils;
/**
 * sm2算法实现
@@ -17,7 +14,7 @@
 */
public class Sm2Encryptor extends AbstractEncryptor {
    private final SM2 sm2;
    private final EncryptContext context;
    public Sm2Encryptor(EncryptContext context) {
        super(context);
@@ -26,7 +23,7 @@
        if (StringUtils.isAnyEmpty(privateKey, publicKey)) {
            throw new IllegalArgumentException("SM2公私钥均需要提供,公钥加密,私钥解密。");
        }
        this.sm2 = SmUtil.sm2(Base64.decode(privateKey), Base64.decode(publicKey));
        this.context = context;
    }
    /**
@@ -46,9 +43,9 @@
    @Override
    public String encrypt(String value, EncodeType encodeType) {
        if (encodeType == EncodeType.HEX) {
            return sm2.encryptHex(value, KeyType.PublicKey);
            return EncryptUtils.encryptBySm2Hex(value, context.getPublicKey());
        } else {
            return sm2.encryptBase64(value, KeyType.PublicKey);
            return EncryptUtils.encryptBySm2(value, context.getPublicKey());
        }
    }
@@ -59,6 +56,6 @@
     */
    @Override
    public String decrypt(String value) {
        return this.sm2.decryptStr(value, KeyType.PrivateKey);
        return EncryptUtils.decryptBySm2(value, context.getPrivateKey());
    }
}