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 4853305 4963723 27 * @summary Test signing/verifying using all the signature algorithms 28 * @author Andreas Sterbenz 29 * @key randomness 30 */ 31 32import java.io.*; 33import java.util.*; 34 35import java.security.*; 36import java.security.interfaces.*; 37 38public class TestSignatures { 39 40 private final static String BASE = System.getProperty("test.src", "."); 41 42 private static final char[] password = "test12".toCharArray(); 43 44 private static Provider provider; 45 46 private static byte[] data; 47 48 static KeyStore getKeyStore() throws Exception { 49 InputStream in = new FileInputStream(new File(BASE, "rsakeys.ks")); 50 KeyStore ks = KeyStore.getInstance("JKS"); 51 ks.load(in, password); 52 in.close(); 53 return ks; 54 } 55 56 private static void testSignature(String algorithm, PrivateKey privateKey, PublicKey publicKey) throws Exception { 57 System.out.println("Testing " + algorithm + "..."); 58 Signature s = Signature.getInstance(algorithm, provider); 59 s.initSign(privateKey); 60 s.update(data); 61 byte[] sig = s.sign(); 62 s.initVerify(publicKey); 63 s.update(data); 64 boolean result; 65 result = s.verify(sig); 66 if (result == false) { 67 throw new Exception("Verification 1 failed"); 68 } 69 s.update(data); 70 result = s.verify(sig); 71 if (result == false) { 72 throw new Exception("Verification 2 failed"); 73 } 74 result = s.verify(sig); 75 if (result == true) { 76 throw new Exception("Verification 3 succeeded"); 77 } 78 } 79 80 private static void test(PrivateKey privateKey, PublicKey publicKey) throws Exception { 81 testSignature("MD2withRSA", privateKey, publicKey); 82 testSignature("MD5withRSA", privateKey, publicKey); 83 testSignature("SHA1withRSA", privateKey, publicKey); 84 testSignature("SHA224withRSA", privateKey, publicKey); 85 testSignature("SHA256withRSA", privateKey, publicKey); 86 RSAPublicKey rsaKey = (RSAPublicKey)publicKey; 87 if (rsaKey.getModulus().bitLength() > 512) { 88 // for SHA384 and SHA512 the data is too long for 512 bit keys 89 testSignature("SHA384withRSA", privateKey, publicKey); 90 testSignature("SHA512withRSA", privateKey, publicKey); 91 } 92 } 93 94 public static void main(String[] args) throws Exception { 95 long start = System.currentTimeMillis(); 96 provider = Security.getProvider("SunRsaSign"); 97 data = new byte[2048]; 98 new Random().nextBytes(data); 99 KeyStore ks = getKeyStore(); 100 KeyFactory kf = KeyFactory.getInstance("RSA", provider); 101 for (Enumeration e = ks.aliases(); e.hasMoreElements(); ) { 102 String alias = (String)e.nextElement(); 103 if (ks.isKeyEntry(alias)) { 104 System.out.println("* Key " + alias + "..."); 105 PrivateKey privateKey = (PrivateKey)ks.getKey(alias, password); 106 PublicKey publicKey = ks.getCertificate(alias).getPublicKey(); 107 privateKey = (PrivateKey)kf.translateKey(privateKey); 108 publicKey = (PublicKey)kf.translateKey(publicKey); 109 test(privateKey, publicKey); 110 } 111 } 112 long stop = System.currentTimeMillis(); 113 System.out.println("All tests passed (" + (stop - start) + " ms)."); 114 } 115} 116