AesCryptor.java

package org.n3r.eql.dbfieldcryptor.refer.aes;

import lombok.SneakyThrows;
import org.n3r.eql.util.S;

import javax.crypto.Cipher;
import javax.crypto.spec.SecretKeySpec;

/**
 * Advanced Encryption Standard.
 *
 * @author Bingoo Huang
 */
public class AesCryptor extends BaseCryptor {

    // 加密用
    private Cipher encryptCipher;
    // 解密用
    private Cipher decryptCipher;

    /**
     * 默认构造函数.
     */
    public AesCryptor() {
        super();
        initCipher();
    }

    /**
     * 使用密钥构造.
     *
     * @param key 密钥
     */
    public AesCryptor(String key) {
        super(key);

        initCipher();
    }

    @SneakyThrows
    private void initCipher() {
        final byte[] rawkey = S.alignRight(getKey(), 16, 'L').getBytes("UTF-8");
        SecretKeySpec key1 = new SecretKeySpec(rawkey, "AES");
        encryptCipher = Cipher.getInstance("AES");
        encryptCipher.init(Cipher.ENCRYPT_MODE, key1);
        decryptCipher = Cipher.getInstance("AES");
        decryptCipher.init(Cipher.DECRYPT_MODE, key1);
    }

    /**
     * 取得Cipher对象.
     *
     * @param isEncrypt 是否加密
     * @return Cipher对象
     */
    @Override
    protected Cipher getCipher(boolean isEncrypt) {
        return isEncrypt ? encryptCipher : decryptCipher;
    }
}