1 /* 2 * Copyright (c) 2003, 2012, Oracle and/or its affiliates. All rights reserved. 3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. 4 * 5 * This code is free software; you can redistribute it and/or modify it 6 * under the terms of the GNU General Public License version 2 only, as 7 * published by the Free Software Foundation. 8 * 9 * This code is distributed in the hope that it will be useful, but WITHOUT 10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or 11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License 12 * version 2 for more details (a copy is included in the LICENSE file that 13 * accompanied this code). 14 * 15 * You should have received a copy of the GNU General Public License version 16 * 2 along with this work; if not, write to the Free Software Foundation, 17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. 18 * 19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 /* 25 * @test 26 * @bug 4923484 27 * @summary encryption/decryption test for using OAEPParameterSpec. 28 * @author Valerie Peng 29 */ 30 31 import java.util.*; 32 33 import java.security.*; 34 import java.security.spec.MGF1ParameterSpec; 35 import javax.crypto.*; 36 import javax.crypto.spec.PSource; 37 import javax.crypto.spec.OAEPParameterSpec; 38 39 public class TestOAEPWithParams { 40 41 private static Provider cp; 42 43 private static PrivateKey privateKey; 44 45 private static PublicKey publicKey; 46 47 private static Random random = new Random(); 48 49 private static String MD[] = { 50 "MD5", "SHA1", "SHA-224", "SHA-256" 51 }; 52 private static int DATA_LENGTH[] = { 53 62, 54, 34, 30 54 }; 55 public static void main(String[] args) throws Exception { 56 long start = System.currentTimeMillis(); 57 cp = Security.getProvider("SunJCE"); 58 System.out.println("Testing provider " + cp.getName() + "..."); 59 Provider kfp = Security.getProvider("SunRsaSign"); 60 KeyPairGenerator kpg = KeyPairGenerator.getInstance("RSA", kfp); 61 kpg.initialize(768); 62 KeyPair kp = kpg.generateKeyPair(); 63 privateKey = kp.getPrivate(); 64 publicKey = kp.getPublic(); 65 66 for (int i = 0; i < MD.length; i++) { 67 // basic test using MD5 68 testEncryptDecrypt(MD[i], DATA_LENGTH[i]); 69 } 70 71 long stop = System.currentTimeMillis(); 72 System.out.println("Done (" + (stop - start) + " ms)."); 73 } 74 75 private static void testEncryptDecrypt(String hashAlg, int dataLength) 76 throws Exception { 77 System.out.println("Testing OAEP with hash " + hashAlg + ", " + dataLength + " bytes"); 78 Cipher c = Cipher.getInstance("RSA/ECB/OAEPwith" + hashAlg + 79 "andMGF1Padding", cp); 80 byte[] pSrc1 = { (byte) 0x01, (byte) 0x01, (byte) 0x01, (byte) 0x01, 81 (byte) 0x02, (byte) 0x02, (byte) 0x02, (byte) 0x02 82 }; 83 byte[] pSrc2 = { (byte) 0x01, (byte) 0x01, (byte) 0x01, (byte) 0x01, 84 (byte) 0x02, (byte) 0x02, (byte) 0x03, (byte) 0x04 85 }; 86 OAEPParameterSpec spec1 = new OAEPParameterSpec(hashAlg, 87 "MGF1", MGF1ParameterSpec.SHA1, new PSource.PSpecified(pSrc1)); 88 OAEPParameterSpec spec2 = new OAEPParameterSpec(hashAlg, 89 "MGF1", MGF1ParameterSpec.SHA1, new PSource.PSpecified(pSrc2)); 90 byte[] plainText = new byte[dataLength]; 91 byte[] cipherText, recovered; 92 // do encryption with parameters#1 93 System.out.println("Testing with user-supplied parameters..."); 94 c.init(Cipher.ENCRYPT_MODE, publicKey, spec1); 95 cipherText = c.doFinal(plainText); 96 97 // test#1: decrypt with parameters#1 98 c.init(Cipher.DECRYPT_MODE, privateKey, spec1); 99 recovered = c.doFinal(cipherText); 100 if (Arrays.equals(plainText, recovered) == false) { 101 throw new Exception("Decrypted data does not match"); 102 } 103 104 // test#2: decrypt without parameters 105 c.init(Cipher.DECRYPT_MODE, privateKey); 106 try { 107 recovered = c.doFinal(cipherText); 108 throw new Exception("Should throw BadPaddingException"); 109 } catch (BadPaddingException bpe) { 110 // expected 111 } 112 // test#3: decrypt with different parameters 113 c.init(Cipher.DECRYPT_MODE, privateKey, spec2); 114 try { 115 recovered = c.doFinal(cipherText); 116 throw new Exception("Should throw BadPaddingException"); 117 } catch (BadPaddingException bpe) { 118 // expected 119 } 120 // do encryption without parameters 121 System.out.println("Testing with cipher default parameters..."); 122 c.init(Cipher.ENCRYPT_MODE, publicKey); 123 cipherText = c.doFinal(plainText); 124 125 // test#1: decrypt with parameters got from cipher 126 AlgorithmParameters params = c.getParameters(); 127 c.init(Cipher.DECRYPT_MODE, privateKey, params); 128 recovered = c.doFinal(cipherText); 129 if (Arrays.equals(plainText, recovered) == false) { 130 throw new Exception("Decrypted data does not match"); 131 } 132 133 // test#2: decrypt without parameters 134 c.init(Cipher.DECRYPT_MODE, privateKey); 135 recovered = c.doFinal(cipherText); 136 if (Arrays.equals(plainText, recovered) == false) { 137 throw new Exception("Decrypted data does not match"); 138 } 139 140 // test#3: decrypt with different parameters 141 c.init(Cipher.DECRYPT_MODE, privateKey, spec2); 142 try { 143 recovered = c.doFinal(cipherText); 144 throw new Exception("Should throw BadPaddingException"); 145 } catch (BadPaddingException bpe) { 146 // expected 147 } 148 } 149 }