package com.ulisesbocchio.jasyptspringboot;
import com.ulisesbocchio.jasyptspringboot.encryptor.SimpleAsymmetricConfig;
import com.ulisesbocchio.jasyptspringboot.encryptor.SimpleAsymmetricStringEncryptor;
import com.ulisesbocchio.jasyptspringboot.encryptor.SimplePBEByteEncryptor;
import com.ulisesbocchio.jasyptspringboot.encryptor.SimplePBEStringEncryptor;
import com.ulisesbocchio.jasyptspringboot.util.AsymmetricCryptography;
import lombok.SneakyThrows;
import org.jasypt.salt.RandomSaltGenerator;
import org.junit.Assert;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.springframework.core.io.ClassPathResource;
import org.springframework.util.FileCopyUtils;
import java.io.InputStreamReader;
import java.nio.charset.StandardCharsets;
import java.security.Provider;
import java.security.Security;
import java.util.Base64;
import java.util.stream.Collectors;
import java.util.stream.Stream;
public class EncryptorTest {
private SimplePBEByteEncryptor _PBEWITHHMACSHA512ANDAES_256 = null;
private SimplePBEStringEncryptor stringEncryptor = null;
private SimpleAsymmetricStringEncryptor keyFileEncryptor = null;
private SimpleAsymmetricStringEncryptor keyResourceEncryptor = null;
private SimpleAsymmetricStringEncryptor keyStringEncryptor = null;
private SimpleAsymmetricStringEncryptor keyFilePemEncryptor = null;
private SimpleAsymmetricStringEncryptor keyResourcePemEncryptor = null;
private SimpleAsymmetricStringEncryptor keyStringPemEncryptor = null;
@BeforeClass
public static void setupClass() {
System.out.println("====== Algorithms ==========");
System.out.println(Stream.of(Security.getProviders()).map(EncryptorTest::getProviderDetails).collect(Collectors.joining("\n")));
System.out.println("===========================");
System.out.println();
}
@Before
public void setup() {
setup_PBEWITHHMACSHA512ANDAES_256();
setup_stringEncryptor();
setup_keyFileEncryptor();
setup_keyResourceEncryptor();
setup_keyStringEncryptor();
setup_keyFilePemEncryptor();
setup_keyResourcePemEncryptor();
setup_keyStringPemEncryptor();
}
private void setup_keyFileEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
config.setPrivateKeyLocation("classpath:private_key.der");
config.setPublicKeyLocation("classpath:public_key.der");
keyFileEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
private void setup_keyResourceEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
config.setPrivateKeyResource(new ClassPathResource("private_key.der"));
config.setPublicKeyResource(new ClassPathResource("public_key.der"));
keyResourceEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
@SneakyThrows
private void setup_keyStringEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
String privateKey = Base64.getEncoder().encodeToString(FileCopyUtils.copyToByteArray(new ClassPathResource("private_key.der").getInputStream()));
String publicKey = Base64.getEncoder().encodeToString(FileCopyUtils.copyToByteArray(new ClassPathResource("public_key.der").getInputStream()));
config.setPrivateKey(privateKey);
config.setPublicKey(publicKey);
keyStringEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
private void setup_keyFilePemEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
config.setPrivateKeyLocation("classpath:private_key.pem");
config.setPublicKeyLocation("classpath:public_key.pem");
config.setPrivateKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
config.setPublicKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
keyFilePemEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
private void setup_keyResourcePemEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
config.setPrivateKeyResource(new ClassPathResource("private_key.pem"));
config.setPublicKeyResource(new ClassPathResource("public_key.pem"));
config.setPrivateKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
config.setPublicKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
keyResourcePemEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
@SneakyThrows
private void setup_keyStringPemEncryptor() {
SimpleAsymmetricConfig config = new SimpleAsymmetricConfig();
String privateKey = FileCopyUtils.copyToString(new InputStreamReader(new ClassPathResource("private_key.pem").getInputStream()));
String publicKey = FileCopyUtils.copyToString(new InputStreamReader(new ClassPathResource("public_key.pem").getInputStream()));
config.setPrivateKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
config.setPublicKeyFormat(AsymmetricCryptography.KeyFormat.PEM);
config.setPrivateKey(privateKey);
config.setPublicKey(publicKey);
keyStringPemEncryptor = new SimpleAsymmetricStringEncryptor(config);
}
private static String getProviderDetails(Provider provider) {
final String serviceName = "Cipher";
final String type = "PBE";
return provider.getName() + "\n\t"
+ provider.keySet().stream()
.map(k -> ((String) k).toUpperCase())
.filter(k -> k.startsWith(serviceName.toUpperCase()) && !k.contains(" "))
.map(k -> k.substring(serviceName.length() + 1))
.filter(k -> k.startsWith(type))
.collect(Collectors.joining("\n\t"));
}
private void setup_stringEncryptor() {
stringEncryptor = new SimplePBEStringEncryptor(_PBEWITHHMACSHA512ANDAES_256);
}
@SneakyThrows
private void setup_PBEWITHHMACSHA512ANDAES_256() {
SimplePBEByteEncryptor encryptor = new SimplePBEByteEncryptor();
encryptor.setPassword("some password loco");
encryptor.setSaltGenerator(new RandomSaltGenerator());
encryptor.setIterations(1000);
encryptor.setAlgorithm("PBEWITHHMACSHA512ANDAES_256");
_PBEWITHHMACSHA512ANDAES_256 = encryptor;
}
@Test
public void test_asymmetric_key_file_encryption() {
final String message = "This is the secret message... BOOHOOO!";
final String encrypted = keyFileEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyFileEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_asymmetric_key_resource_encryption() {
final String message = "This is the secret resource message... BOOHOOO!";
final String encrypted = keyResourceEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyResourceEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_asymmetric_key_string_encryption() {
final String message = "This is the secret string message... BOOHOOO!";
final String encrypted = keyStringEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyStringEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_asymmetric_key_file_pem_encryption() {
final String message = "This is the secret pem message... BOOHOOO!";
final String encrypted = keyFilePemEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyFilePemEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_asymmetric_key_resource_pem_encryption() {
final String message = "This is the secret resource pem message... BOOHOOO!";
final String encrypted = keyResourcePemEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyResourcePemEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_asymmetric_key_string_pem_encryption() {
final String message = "chupacabras";
final String encrypted = keyStringPemEncryptor.encrypt(message);
System.out.println(encrypted);
final String decrypted = keyStringPemEncryptor.decrypt(encrypted);
System.out.println(decrypted);
Assert.assertEquals(decrypted, message);
}
@Test
public void test_PBEWITHHMACSHA512ANDAES_256_encryption() {
final String message = "chupacabras";
final byte[] messageBytes = message.getBytes(StandardCharsets.US_ASCII);
final byte[] ciphertext = _PBEWITHHMACSHA512ANDAES_256.encrypt(messageBytes);
final String decrypted = new String(
_PBEWITHHMACSHA512ANDAES_256.decrypt(ciphertext),
StandardCharsets.US_ASCII);
Assert.assertEquals(message, decrypted);
}
@Test
public void test_StringEncrytor_encryption() {
final String message = "This is the secret message... BOOHOOO!";
final String ciphertext = stringEncryptor.encrypt(message);
final String decrypted = stringEncryptor.decrypt(ciphertext);
Assert.assertEquals(message, decrypted);
}
}
