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);
}
}