package test;
    import com.github.javafaker.Faker;
    import org.junit.Test;
    import javax.crypto.*;
    import javax.crypto.spec.IvParameterSpec;
    import java.security.*;
    import java.security.spec.InvalidKeySpecException;
    import java.security.spec.PKCS8EncodedKeySpec;
    public class RSA {
        @Test
        public void rsaEncode() throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeySpecException, InvalidKeyException, BadPaddingException, IllegalBlockSizeException {
            var keyPairGen = KeyPairGenerator.getInstance("RSA");
            var pair = keyPairGen.generateKeyPair();
            var privateKey = pair.getPrivate().getEncoded();
            var faker = new Faker();
            var start = System.currentTimeMillis();
            var keyFactory = KeyFactory.getInstance("RSA");
            var keySpec = new PKCS8EncodedKeySpec(privateKey);
            var key = keyFactory.generatePrivate(keySpec);
            for(var i = 0; i < 10000; i++) {
                var randomBytes = faker.shakespeare().asYouLikeItQuote().getBytes();
                var cipher = Cipher.getInstance(keyFactory.getAlgorithm());
                cipher.init(Cipher.ENCRYPT_MODE, key);
                var resultBytes = cipher.doFinal(randomBytes);
                if(i % 100 == 0) {
                    System.out.format("%d/10000 done.\n", i);
                }
            }
            System.out.format("time: %dms\n", System.currentTimeMillis() - start);
        }
        @Test
        public void aesEncode() throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidAlgorithmParameterException, InvalidKeyException, BadPaddingException, IllegalBlockSizeException {
            var keyGenerator = KeyGenerator.getInstance("AES");
            keyGenerator.init(128);
            var key = keyGenerator.generateKey();
            var faker = new Faker();
            var ivBytes = new byte[16];
            var start = System.currentTimeMillis();
            for(var i = 0; i < 10000; i++) {
                var randomBytes = faker.shakespeare().asYouLikeItQuote().getBytes();
                var cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
                new SecureRandom().nextBytes(ivBytes);
                var iv = new IvParameterSpec(ivBytes);
                cipher.init(Cipher.ENCRYPT_MODE, key, iv);
                var result = cipher.doFinal(randomBytes);
            }
            System.out.format("time: %dms\n", System.currentTimeMillis() - start);
        }
    }