Strolch
/
strolch
mirror of https://github.com/strolch-li/strolch.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[New] Handling illegal key size in AesCryptoHelperTest

This commit is contained in:
Robert von Burg 2015-10-16 17:39:06 +02:00
parent 1de56d0043
commit d0691e4d35
2 changed files with 57 additions and 74 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/main/java/ch/eitchnet/utils/helper/ExceptionHelper.java
View File

@ -95,4 +95,20 @@ public class ExceptionHelper {
String root = formatExceptionMessage(t.getCause());
return getExceptionMessage(t) + "\ncause:\n" + root;
}
/**
* Returns the root cause for the given {@link Throwable}
*
* @param t
* the {@link Throwable} for which to get the root cause
*
* @return the root cause of the given {@link Throwable}
*/
public static Throwable getRootCause(Throwable t) {
while (t.getCause() != null) {
t = t.getCause();
}
return t;
}
}

115
src/test/java/ch/eitchnet/utils/helper/AesCryptoHelperTest.java
View File

@ -9,7 +9,6 @@ import java.io.File;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.security.InvalidKeyException;
import org.junit.Test;
import org.slf4j.Logger;
@ -59,8 +58,7 @@ public class AesCryptoHelperTest {
assertArrayEquals(clearTextBytes, decryptedBytes);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
if (ExceptionHelper.getRootCause(e).getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
@ -78,8 +76,7 @@ public class AesCryptoHelperTest {
assertArrayEquals(clearTextBytes, decryptedBytes);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
if (ExceptionHelper.getRootCause(e).getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
@ -88,99 +85,69 @@ public class AesCryptoHelperTest {
@Test
public void shouldEncryptShortFile() {
try {
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_short.txt";
// encrypted file
String encryptedFileS = "target/encrypted_short.aes";
// encrypted file
String decryptedFileS = "target/decrypted_short.txt";
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_short.txt";
// encrypted file
String encryptedFileS = "target/encrypted_short.aes";
// encrypted file
String decryptedFileS = "target/decrypted_short.txt";
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
}
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
}
@Test
public void shouldEncryptMiddleFile() {
try {
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_middle.txt";
// encrypted file
String encryptedFileS = "target/encrypted_middle.aes";
// encrypted file
String decryptedFileS = "target/decrypted_middle.txt";
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_middle.txt";
// encrypted file
String encryptedFileS = "target/encrypted_middle.aes";
// encrypted file
String decryptedFileS = "target/decrypted_middle.txt";
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
}
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
}
@Test
public void shouldEncryptLongFile() {
try {
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_long.txt";
// encrypted file
String encryptedFileS = "target/encrypted_long.aes";
// encrypted file
String decryptedFileS = "target/decrypted_long.txt";
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_long.txt";
// encrypted file
String encryptedFileS = "target/encrypted_long.aes";
// encrypted file
String decryptedFileS = "target/decrypted_long.txt";
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
}
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
}
@Test
public void shouldEncryptBinaryFile() {
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_image.ico";
// encrypted file
String encryptedFileS = "target/encrypted_image.aes";
// encrypted file
String decryptedFileS = "target/decrypted_image.ico";
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
}
private static void testCrypto(String clearTextFileS, String encryptedFileS, String decryptedFileS) {
try {
// file to be encrypted
String clearTextFileS = "src/test/resources/crypto_test_image.ico";
// encrypted file
String encryptedFileS = "target/encrypted_image.aes";
// encrypted file
String decryptedFileS = "target/decrypted_image.ico";
AesCryptoHelper.encrypt(password, salt, clearTextFileS, encryptedFileS);
AesCryptoHelper.decrypt(password, salt, encryptedFileS, decryptedFileS);
testCrypto(clearTextFileS, encryptedFileS, decryptedFileS);
String inputSha256 = StringHelper.getHexString(FileHelper.hashFileSha256(new File(clearTextFileS)));
String doutputSha256 = StringHelper.getHexString(FileHelper.hashFileSha256(new File(decryptedFileS)));
assertEquals(inputSha256, doutputSha256);
} catch (RuntimeException e) {
if (e.getCause() instanceof InvalidKeyException
&& e.getCause().getMessage().equals("Illegal key size or default parameters"))
if (ExceptionHelper.getRootCause(e).getMessage().equals("Illegal key size or default parameters"))
logger.warn("YOU ARE MISSING THE UNLIMITED JCE POLICIES and can not do AES encryption!");
else
throw e;
}
}
private static void testCrypto(String clearTextFileS, String encryptedFileS, String decryptedFileS) {
AesCryptoHelper.encrypt(password, salt, clearTextFileS, encryptedFileS);
AesCryptoHelper.decrypt(password, salt, encryptedFileS, decryptedFileS);
String inputSha256 = StringHelper.getHexString(FileHelper.hashFileSha256(new File(clearTextFileS)));
String doutputSha256 = StringHelper.getHexString(FileHelper.hashFileSha256(new File(decryptedFileS)));
assertEquals(inputSha256, doutputSha256);
}
}