import java.security.InvalidAlgorithmParameterException; import java.security.InvalidKeyException; import java.security.NoSuchAlgorithmException; import java.security.SecureRandom; import javax.crypto.BadPaddingException; import javax.crypto.Cipher; import javax.crypto.IllegalBlockSizeException; import javax.crypto.KeyGenerator; import javax.crypto.NoSuchPaddingException; import javax.crypto.SecretKey; import javax.crypto.spec.IvParameterSpec; import javax.crypto.spec.SecretKeySpec; public class Sample { public static class EncryptedData { public byte[] iv; public byte[] data; } public static void main(String[] args) throws Exception { // 鍵を生成する。 // この鍵を保存して暗号文を交換する2者間で安全に共有する。 final byte[] key = generateKey(); print("[Key]", key); // 暗号化対象の平文 final byte[] input = "hello world".getBytes(); print("[Input]", input); // 鍵で平文を暗号化する。 // 復号化するためには、「鍵」と「IV(initial vector)」と「暗号文」が必要。 // 「鍵」は事前に安全に共有されていると仮定して、ここでは「IV」と「暗号文」を求める。 final EncryptedData encrypted = encrypt(key, input); // 試しに暗号化したデータを表示してみる。 print("[IV]", encrypted.iv); print("[Encrypted]", encrypted.data); // 「鍵」と「IV」と「暗号文」を渡して復号化する。 final byte[] decrypted = decrypt(key, encrypted.iv, encrypted.data); // 試しに復号化したデータを表示してみる。 print("[Decrypted]", decrypted); // さらに、同じ鍵で同じ平文を暗号化・復号化してみる。 final EncryptedData encrypted2 = encrypt(key, input); print("[IV(2)]", encrypted2.iv); print("[Encrypted(2)]", encrypted2.data); final byte[] decrypted2 = decrypt(key, encrypted.iv, encrypted.data); print("[Decrypted(2)]", decrypted2); } public static byte[] generateKey() throws NoSuchAlgorithmException { final SecureRandom random = SecureRandom.getInstance("SHA1PRNG"); //random.setSeed(getRandomSeedBytes()); // ここで本物の乱数で初期化する final KeyGenerator keyGen = KeyGenerator.getInstance("AES"); keyGen.init(256, random); return keyGen.generateKey().getEncoded(); } public static EncryptedData encrypt(byte[] key, byte[] input) throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeyException, IllegalBlockSizeException, BadPaddingException { final SecureRandom random = SecureRandom.getInstance("SHA1PRNG"); //random.setSeed(getRandomSeedBytes()); // ここで本物の乱数で初期化する final SecretKey secretKey = new SecretKeySpec(key, "AES"); final Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding"); cipher.init(Cipher.ENCRYPT_MODE, secretKey, random); final EncryptedData result = new EncryptedData(); result.iv = cipher.getIV(); result.data = cipher.doFinal(input); return result; } public static byte[] decrypt(byte[] key, byte[] iv, byte[] input) throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeyException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException { final SecretKey secretKey = new SecretKeySpec(key, "AES"); final Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding"); cipher.init(Cipher.DECRYPT_MODE, secretKey, new IvParameterSpec(iv)); return cipher.doFinal(input); } public static void print(String tag, byte[] bs) { System.out.print(tag); for (int i = 0; i < bs.length; ++i) { if (i % 16 == 0) { System.out.println(); } System.out.print(String.format(" %02X", bs[i])); } System.out.println(); } }
2011年2月28日月曜日
ついでにAESで暗号化・復号化するサンプルコード書いてみた
登録:
コメントの投稿 (Atom)
0 件のコメント:
コメントを投稿