com.starmcc.qmframework.tools.operation.QmAesUtil Maven / Gradle / Ivy

Go to download
package com.starmcc.qmframework.tools.operation;

import com.starmcc.qmframework.config.AesConfiguration;
import org.apache.tomcat.util.codec.binary.Base64;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.security.SecureRandom;
import java.util.Date;

/**
 * AES对称加密技术
 *
 * @Author starmcc
 */
public class QmAesUtil {

    private static final Logger LOG = LoggerFactory.getLogger(QmAesUtil.class);

    /**
     * 加密封装(使用配置的key和次数)
     *
     * @param data 加密数据
     * @return
     * @throws Exception
     */
    public static String encryptAes(String data) throws Exception {
        Date date = new Date();
        String str = data;
        for (int i = 0; i < AesConfiguration.number; i++) {
            str = encryptAes(str, AesConfiguration.key);
        }
        LOG.debug("加密用时：" + (System.currentTimeMillis() - date.getTime()));
        return str;
    }


    /**
     * 解密封装(使用配置的key和次数)
     *
     * @param data 密文
     * @return
     * @throws Exception
     */
    public static String decryptAes(String data) throws Exception {
        Date date = new Date();
        String str = data;
        for (int i = 0; i < AesConfiguration.number; i++) {
            str = decryptAes(str, AesConfiguration.key);
        }
        LOG.debug("解密用时：" + (System.currentTimeMillis() - date.getTime()));
        return str;
    }

    /**
     * 加密封装
     *
     * @param data 加密数据
     * @param num  加密次数
     * @param key  秘钥
     * @return
     * @throws Exception
     */
    public static String encryptAes(String data, int num, String key) throws Exception {
        Date date = new Date();
        String str = data;
        for (int i = 0; i < num; i++) {
            str = encryptAes(str, key);
        }
        LOG.debug("加密用时：" + (System.currentTimeMillis() - date.getTime()));
        return str;
    }

    /**
     * 解密封装
     *
     * @param data 密文
     * @param num  解密次数
     * @param key  秘钥
     * @return
     * @throws Exception
     */
    public static String decryptAes(String data, int num, String key) throws Exception {
        Date date = new Date();
        String str = data;
        for (int i = 0; i < num; i++) {
            str = decryptAes(str, key);
        }
        LOG.debug("解密用时：" + (System.currentTimeMillis() - date.getTime()));
        return str;
    }


    /**
     * 加密
     *
     * @param data
     * @param key
     * @return
     * @throws Exception
     */
    private static String encryptAes(String data, String key) throws Exception {
        KeyGenerator kgen = KeyGenerator.getInstance("AES");
        SecureRandom secureRandom = SecureRandom.getInstance("SHA1PRNG");
        secureRandom.setSeed(key.getBytes(AesConfiguration.encoding));
        kgen.init(128, secureRandom);
        SecretKey skey = kgen.generateKey();
        byte[] raw = skey.getEncoded();
        SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
        Cipher cipher = Cipher.getInstance("AES");
        cipher.init(Cipher.ENCRYPT_MODE, skeySpec);
        byte[] encryptedData = cipher.doFinal(data.getBytes(AesConfiguration.encoding));
        String hexStr = Base64.encodeBase64String(encryptedData);
        return hexStr;
    }

    /**
     * 解密
     *
     * @param data
     * @param key
     * @return
     * @throws Exception
     */
    private static String decryptAes(String data, String key) throws Exception {
        KeyGenerator kgen = KeyGenerator.getInstance("AES");
        SecureRandom secureRandom = SecureRandom.getInstance("SHA1PRNG");
        secureRandom.setSeed(key.getBytes(AesConfiguration.encoding));
        kgen.init(128, secureRandom);
        SecretKey skey = kgen.generateKey();
        byte[] raw = skey.getEncoded();
        SecretKeySpec skeySpec = new SecretKeySpec(raw, "AES");
        Cipher cipher = Cipher.getInstance("AES");
        cipher.init(Cipher.DECRYPT_MODE, skeySpec);
        byte[] decryptedData = cipher.doFinal(Base64.decodeBase64(data));
        String respStr = new String(decryptedData, AesConfiguration.encoding);
        return respStr;
    }


}