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 }