com.github.houbb.heaven.util.secrect.AesUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of heaven Show documentation
Collect base useful tools for dev.
The newest version!
package com.github.houbb.heaven.util.secrect;

import com.github.houbb.heaven.response.exception.CommonRuntimeException;
import com.github.houbb.heaven.util.common.ArgUtil;
import com.github.houbb.heaven.util.util.internal.Base64;

import javax.crypto.Cipher;
import javax.crypto.spec.IvParameterSpec;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;

/**
 * AES 加解密工具类
 *
 * @author binbin.hou
 * @since 0.1.98
 */
public final class AesUtil {

    private AesUtil() {
    }

    /**
     * 加密
     *
     * @param sourceText 原始文本
     * @param key        密匙
     * @return 结果
     * @since 0.1.98
     */
    public static String encrypt(String sourceText, String key) {
        ArgUtil.notEmpty(sourceText, "sourceText");
        ArgUtil.notEmpty(key, "key");

        try {
            byte[] raw = key.getBytes();
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            //"算法/模式/补码方式"
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            //使用CBC模式，需要一个向量iv，可增加加密算法的强度
            IvParameterSpec iv = new IvParameterSpec("0102030405060708".getBytes());
            cipher.init(Cipher.ENCRYPT_MODE, skeySpec, iv);
            byte[] encrypted = cipher.doFinal(sourceText.getBytes());

            //此处使用BASE64做转码功能，同时能起到2次加密的作用。
            return Base64.getEncoder().encodeToString(encrypted);
        } catch (Exception e) {
            throw new CommonRuntimeException(e);
        }
    }

    /**
     * 解密
     *
     * @param sourceText 原始文本
     * @param key        密匙
     * @return 结果
     * @since 0.1.98
     */
    public static String decrypt(String sourceText, String key) {
        ArgUtil.notEmpty(sourceText, "sourceText");
        ArgUtil.notEmpty(key, "key");

        try {
            byte[] raw = key.getBytes(StandardCharsets.UTF_8);
            SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
            Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
            IvParameterSpec iv = new IvParameterSpec("0102030405060708"
                    .getBytes());
            cipher.init(Cipher.DECRYPT_MODE, skeySpec, iv);
            //先用base64解密
            byte[] encrypted1 = Base64.getDecoder().decode(sourceText);
            byte[] original = cipher.doFinal(encrypted1);

            return new String(original);
        } catch (Exception ex) {
            throw new CommonRuntimeException(ex);
        }
    }

}