java中AES工具类对称加密

发布于:2023-09-15 ⋅ 阅读:(130) ⋅ 点赞:(0)
import lombok.SneakyThrows;

import javax.crypto.Cipher;
import javax.crypto.KeyGenerator;
import javax.crypto.SecretKey;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.StandardCharsets;
import java.security.NoSuchAlgorithmException;
import java.security.SecureRandom;
import java.util.Base64;

public class AESUtils {

    private static final String KEY_ALGORITHM = "AES";
    public static final String RSA_TYPE = "AES/ECB/PKCS5Padding";
    private static final int KEY_SIZE = 128;
    public static final String SIGNATURE_ALGORITHM = "SHA1PRNG";

    @SneakyThrows
    public static String encryptAES(String plainText, String secretKey) {
        Cipher cipher = Cipher.getInstance(RSA_TYPE);
        cipher.init(Cipher.ENCRYPT_MODE, getKey(secretKey));
        byte[] encryptedBytes = cipher.doFinal(plainText.getBytes(StandardCharsets.UTF_8));
        return Base64.getEncoder().encodeToString(encryptedBytes);
    }

    @SneakyThrows
    public static String decryptAES(String encryptedText, String secretKey) {
        Cipher cipher = Cipher.getInstance(RSA_TYPE);
        cipher.init(Cipher.DECRYPT_MODE, getKey(secretKey));
        byte[] decryptedBytes = cipher.doFinal(Base64.getDecoder().decode(encryptedText));
        return new String(decryptedBytes, StandardCharsets.UTF_8);

    }

    private static SecretKeySpec getKey(String secretKey) throws NoSuchAlgorithmException {
        KeyGenerator gen = KeyGenerator.getInstance(KEY_ALGORITHM);
        SecureRandom secureRandom = SecureRandom.getInstance(SIGNATURE_ALGORITHM);
        secureRandom.setSeed(secretKey.getBytes());
        gen.init(KEY_SIZE, secureRandom);
        SecretKey key = gen.generateKey();
        return new SecretKeySpec(key.getEncoded(), KEY_ALGORITHM);
    }


}