Reputation: 231
I have this simply AES encryption code:
public static void main(String[] args) throws Exception
{
byte[] input = new byte[] {
0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77,
(byte)0x88, (byte)0x99, (byte)0xaa, (byte)0xbb,
(byte)0xcc, (byte)0xdd, (byte)0xee, (byte)0xff };
byte[] keyBytes = new byte[] {
0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07,
0x08, 0x09, 0x0a, 0x0b, 0x0c, 0x0d, 0x0e, 0x0f,
0x10, 0x11, 0x12, 0x13, 0x14, 0x15, 0x16, 0x17 };
SecretKeySpec key = new SecretKeySpec(keyBytes, "AES");
Cipher cipher = Cipher.getInstance("AES/ECB/NoPadding", "BC");
System.out.println("input text : " + Utils.toHex(input));
// encryption pass
byte[] cipherText = new byte[input.length];
cipher.init(Cipher.ENCRYPT_MODE, key);
int ctLength = cipher.update(input, 0, input.length, cipherText, 0);
ctLength += cipher.doFinal(cipherText, ctLength);
System.out.println("cipher text: " + Utils.toHex(cipherText)
+ " bytes: " + ctLength);
// decryption pass
byte[] plainText = new byte[ctLength];
cipher.init(Cipher.DECRYPT_MODE, key);
int ptLength = cipher.update(cipherText, 0, ctLength, plainText, 0);
ptLength += cipher.doFinal(plainText, ptLength);
System.out.println("plain text : " + Utils.toHex(plainText)
+ " bytes: " + ptLength);
}
I tried to decompose this void main to 3 methods : keycreation() , encryption(), and decryption(), but I failed because the encryption() method returns 2 values, byte[] cipher and int ctLength... So can someone help me to decompose this code in 3 methods?
Upvotes: 2
Views: 6834
Reputation: 2328
I would recommend organizing your methods into a class like this. Then for each encryption/ decryption task with the same key, you can create an object of this class and call methods on it.
class AESEncryption {
byte[] keyBytes;
Cipher cipher;
SecretKeySpec key;
public AESEncryption() {
try {
cipher = Cipher.getInstance("AES/ECB/NoPadding", "BC");
} catch (NoSuchAlgorithmException | NoSuchProviderException | NoSuchPaddingException e) {
System.out.println(e.getMessage());
}
}
public void createKey(byte[] keyBytes) {
this.keyBytes = keyBytes;
key = new SecretKeySpec(keyBytes, "AES");
}
public byte[] encrypt(byte[] plainText) {
try {
cipher.init(Cipher.ENCRYPT_MODE, key);
return cipher.doFinal(plainText);
} catch (InvalidKeyException | IllegalBlockSizeException | BadPaddingException e) {
System.out.println(e.getMessage());
return null;
}
}
}
You can create an object and call methods like this.
AESEncryption aes = new AESEncryption();
// call this to create the key
aes.createKey(keyBytes);
// call this on encrypt button click
byte[] encrypted = aes.encrypt(input);
// call this on decrypt button click
byte[] decrypted = aes.decrypt(encrypted);
Upvotes: 4