package src;

import java.io.*;
import javax.crypto.*;
import java.security.*;

public class AESEncryption {
	public static void main(String[] args) {

		KeyGenerator kg = null;
		try {
			String beforeEncrypt = "original.txt";
			String afterEncrypt = "cipher.txt";

			kg = KeyGenerator.getInstance("AES");
			kg.init(128);
			SecretKey key = kg.generateKey();

			byte[] aesKey = key.getEncoded();
			PublicKey rsaKey = GetRSAKey.GetPublicKey();
			byte[] cipherKey = RSAEncryption.encrypt(rsaKey, aesKey);

			KeyCipher.saveKey(afterEncrypt, cipherKey);

			Cipher cipher = Cipher.getInstance("AES");

			cipher.init(Cipher.ENCRYPT_MODE, key);
			BufferedInputStream in = new BufferedInputStream(
					new FileInputStream(beforeEncrypt));
			CipherOutputStream out = new CipherOutputStream(
					new BufferedOutputStream(new FileOutputStream(afterEncrypt,
							true)), cipher);
			int i;
			do {
				i = in.read();
				if (i != -1)
					out.write(i);
			} while (i != -1);

			in.close();
			out.close();
			System.out.println("Encryption is Successful!");
		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}