1 /*
   2  * Copyright (c) 2014, 2016, 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     7088989 8000415
  27  * @summary Ensure the various message digests works correctly
  28  * @key randomness
  29  */
  30 
  31 import java.io.*;
  32 import java.security.*;
  33 import java.security.spec.*;
  34 import java.util.*;
  35 import javax.crypto.*;
  36 import javax.crypto.spec.*;
  37 
  38 public class TestDigest extends UcryptoTest {
  39 
  40     private static final String[] MD_ALGOS = {
  41         "MD5",
  42         "SHA",
  43         "SHA-224",
  44         "SHA-256",
  45         "SHA-384",
  46         "SHA-512",
  47         "SHA3-224",
  48         "SHA3-256",
  49         "SHA3-384",
  50         "SHA3-512"
  51     };
  52 
  53     public static void main(String[] args) throws Exception {
  54         main(new TestDigest(), null);
  55     }
  56 
  57     public void doTest(Provider p) throws Exception {
  58         boolean testPassed = true;
  59         byte[] msg = new byte[200];
  60         (new SecureRandom()).nextBytes(msg);
  61         String interopProvName = "SUN";
  62 
  63         MessageDigest md, md2;
  64 
  65         for (String a : MD_ALGOS) {
  66             System.out.println("Testing " + a);
  67             try {
  68                 md = MessageDigest.getInstance(a, p);
  69             } catch (NoSuchAlgorithmException nsae) {
  70                 System.out.println("=> Skip, unsupported");
  71                 continue;
  72             }
  73             try {
  74                 md2 = MessageDigest.getInstance(a, interopProvName);
  75             } catch (NoSuchAlgorithmException nsae) {
  76                 System.out.println("=> Skip, no interop provider found");
  77                 continue;
  78             }
  79 
  80             // Test Interoperability for update+digest calls
  81             for (int i = 0; i < 3; i++) {
  82                 md.update(msg);
  83                 byte[] digest = md.digest();
  84                 md2.update(msg);
  85                 byte[] digest2 = md2.digest();
  86                 if (!Arrays.equals(digest, digest2)) {
  87                     System.out.println("DIFF1 FAILED at iter " + i);
  88                     testPassed = false;
  89                 } else {
  90                     System.out.println("...diff1 test passed");
  91                 }
  92             }
  93 
  94             // Test Interoperability for digest calls
  95             md = MessageDigest.getInstance(a, p);
  96             md2 = MessageDigest.getInstance(a, interopProvName);
  97 
  98             for (int i = 0; i < 3; i++) {
  99                 byte[] digest = md.digest();
 100                 byte[] digest2 = md2.digest();
 101                 if (!Arrays.equals(digest, digest2)) {
 102                     System.out.println("DIFF2 FAILED at iter " + i);
 103                     testPassed = false;
 104                 } else {
 105                     System.out.println("...diff2 test passed");
 106                 }
 107             }
 108 
 109             // Test Cloning functionality if supported
 110             md = MessageDigest.getInstance(a, p);
 111             try {
 112                 md2 = (MessageDigest) md.clone(); // clone right after construction
 113             } catch (CloneNotSupportedException cnse) {
 114                 System.out.println("...no clone support");
 115                 continue;
 116             }
 117             byte[] digest = md.digest();
 118             byte[] digest2 = md2.digest();
 119             if (!Arrays.equals(digest, digest2)) {
 120                 System.out.println("DIFF-3.1 FAILED");
 121                 testPassed = false;
 122             } else {
 123                 System.out.println("...diff3.1 tests passed");
 124             }
 125             md.update(msg);
 126             md2 = (MessageDigest) md.clone(); // clone again after update call
 127             digest = md.digest();
 128             digest2 = md2.digest();
 129             if (!Arrays.equals(digest, digest2)) {
 130                 System.out.println("DIFF-3.2 FAILED");
 131                 testPassed = false;
 132             } else {
 133                 System.out.println("...diff3.2 tests passed");
 134             }
 135             md2 = (MessageDigest) md.clone(); // clone after digest
 136             digest = md.digest();
 137             digest2 = md2.digest();
 138             if (!Arrays.equals(digest, digest2)) {
 139                 System.out.println("DIFF-3.3 FAILED");
 140                 testPassed = false;
 141             } else {
 142                 System.out.println("...diff3.3 tests passed");
 143             }
 144         }
 145         if (!testPassed) {
 146             throw new RuntimeException("One or more MD test failed!");
 147         } else {
 148             System.out.println("MD Tests Passed");
 149         }
 150     }
 151 }