| /* |
| * Copyright (c) 2012, 2017, Oracle and/or its affiliates. All rights reserved. |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. |
| * |
| * This code is free software; you can redistribute it and/or modify it |
| * under the terms of the GNU General Public License version 2 only, as |
| * published by the Free Software Foundation. |
| * |
| * This code is distributed in the hope that it will be useful, but WITHOUT |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License |
| * version 2 for more details (a copy is included in the LICENSE file that |
| * accompanied this code). |
| * |
| * You should have received a copy of the GNU General Public License version |
| * 2 along with this work; if not, write to the Free Software Foundation, |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. |
| * |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA |
| * or visit www.oracle.com if you need additional information or have any |
| * questions. |
| */ |
| |
| /* |
| * @test |
| * @bug 6414899 |
| * @summary Ensure the cloning functionality works. |
| * @author Valerie Peng |
| * @library .. |
| * @key randomness |
| * @modules jdk.crypto.cryptoki |
| * @run main/othervm TestCloning |
| * @run main/othervm TestCloning sm |
| */ |
| |
| import java.security.MessageDigest; |
| import java.security.Provider; |
| import java.util.Arrays; |
| import java.util.Random; |
| |
| public class TestCloning extends PKCS11Test { |
| |
| private static final String[] ALGOS = { |
| "MD2", "MD5", "SHA1", "SHA-224", "SHA-256", "SHA-384", "SHA-512" |
| }; |
| |
| public static void main(String[] args) throws Exception { |
| main(new TestCloning(), args); |
| } |
| |
| private static final byte[] data1 = new byte[10]; |
| private static final byte[] data2 = new byte[10*1024]; |
| |
| |
| @Override |
| public void main(Provider p) throws Exception { |
| Random r = new Random(); |
| byte[] data1 = new byte[10]; |
| byte[] data2 = new byte[2*1024]; |
| r.nextBytes(data1); |
| r.nextBytes(data2); |
| System.out.println("Testing against provider " + p.getName()); |
| for (int i = 0; i < ALGOS.length; i++) { |
| if (p.getService("MessageDigest", ALGOS[i]) == null) { |
| System.out.println(ALGOS[i] + " is not supported, skipping"); |
| continue; |
| } else { |
| System.out.println("Testing " + ALGOS[i] + " of " + p.getName()); |
| MessageDigest md = MessageDigest.getInstance(ALGOS[i], p); |
| try { |
| md = testCloning(md, p); |
| // repeat the test again after generating digest once |
| for (int j = 0; j < 10; j++) { |
| md = testCloning(md, p); |
| } |
| } catch (Exception ex) { |
| if (ALGOS[i] == "MD2" && |
| p.getName().equalsIgnoreCase("SunPKCS11-NSS")) { |
| // known bug in NSS; ignore for now |
| System.out.println("Ignore Known bug in MD2 of NSS"); |
| continue; |
| } |
| throw ex; |
| } |
| } |
| } |
| } |
| |
| private static MessageDigest testCloning(MessageDigest mdObj, Provider p) |
| throws Exception { |
| |
| // copy#0: clone at state BLANK w/o any data |
| MessageDigest mdCopy0 = (MessageDigest) mdObj.clone(); |
| |
| // copy#1: clone again at state BUFFERED w/ very short data |
| mdObj.update(data1); |
| mdCopy0.update(data1); |
| MessageDigest mdCopy1 = (MessageDigest) mdObj.clone(); |
| |
| // copy#2: clone again after updating it w/ long data to trigger |
| // the state into INIT |
| mdObj.update(data2); |
| mdCopy0.update(data2); |
| mdCopy1.update(data2); |
| MessageDigest mdCopy2 = (MessageDigest) mdObj.clone(); |
| |
| // copy#3: clone again after updating it w/ very short data |
| mdObj.update(data1); |
| mdCopy0.update(data1); |
| mdCopy1.update(data1); |
| mdCopy2.update(data1); |
| MessageDigest mdCopy3 = (MessageDigest) mdObj.clone(); |
| |
| // copy#4: clone again after updating it w/ long data |
| mdObj.update(data2); |
| mdCopy0.update(data2); |
| mdCopy1.update(data2); |
| mdCopy2.update(data2); |
| mdCopy3.update(data2); |
| MessageDigest mdCopy4 = (MessageDigest) mdObj.clone(); |
| |
| // check digest equalities |
| byte[] answer = mdObj.digest(); |
| byte[] result0 = mdCopy0.digest(); |
| byte[] result1 = mdCopy1.digest(); |
| byte[] result2 = mdCopy2.digest(); |
| byte[] result3 = mdCopy3.digest(); |
| byte[] result4 = mdCopy4.digest(); |
| |
| |
| check(answer, result0, "copy0"); |
| check(answer, result1, "copy1"); |
| check(answer, result2, "copy2"); |
| check(answer, result3, "copy3"); |
| check(answer, result4, "copy4"); |
| |
| return mdCopy3; |
| } |
| |
| private static void check(byte[] d1, byte[] d2, String copyName) |
| throws Exception { |
| if (Arrays.equals(d1, d2) == false) { |
| throw new RuntimeException(copyName + " digest mismatch!"); |
| } |
| } |
| } |
| |