1 /*

     2  * Copyright (c) 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     7088989

    27  * @summary Ensure the various message digests works correctly

    28  */

    29 import java.io.*;

    30 import java.security.*;

    31 import java.security.spec.*;

    32 import java.util.*;

    33 import javax.crypto.*;

    34 import javax.crypto.spec.*;

    35 

    36 public class TestDigest extends UcryptoTest {

    37 

    38     private static final String[] MD_ALGOS = {

    39         "MD5",

    40         "SHA",

    41         "SHA-256",

    42         "SHA-384",

    43         "SHA-512"

    44     };

    45 

    46     public static void main(String[] args) throws Exception {

    47         main(new TestDigest(), null);

    48     }

    49 

    50     public void doTest(Provider p) {

    51         boolean testPassed = true;

    52         byte[] msg = new byte[200];

    53         (new SecureRandom()).nextBytes(msg);

    54         String interopProvName = "SUN";

    55 

    56         for (String a : MD_ALGOS) {

    57             try {

    58                 MessageDigest md, md2;

    59                 try {

    60                     md = MessageDigest.getInstance(a, p);

    61                 } catch (NoSuchAlgorithmException nsae) {

    62                     System.out.println("Skipping Unsupported MD algo: " + a);

    63                     continue;

    64                 }

    65                 md2 = MessageDigest.getInstance(a, interopProvName);

    66                 // Test Interoperability for update+digest calls

    67                 for (int i = 0; i < 3; i++) {

    68                     md.update(msg);

    69                     byte[] digest = md.digest();

    70                     md2.update(msg);

    71                     byte[] digest2 = md2.digest();

    72                     if (!Arrays.equals(digest, digest2)) {

    73                         System.out.println("DIFF1 FAILED for: " + a + " at iter " + i);

    74                         testPassed = false;

    75                     }

    76                 }

    77 

    78                 // Test Interoperability for digest calls

    79                 md = MessageDigest.getInstance(a, p);

    80                 md2 = MessageDigest.getInstance(a, interopProvName);

    81 

    82                 for (int i = 0; i < 3; i++) {

    83                     byte[] digest = md.digest();

    84                     byte[] digest2 = md2.digest();

    85                     if (!Arrays.equals(digest, digest2)) {

    86                         System.out.println("DIFF2 FAILED for: " + a + " at iter " + i);

    87                         testPassed = false;

    88                     }

    89                 }

    90 

    91                 // Test Cloning functionality

    92                 md = MessageDigest.getInstance(a, p);

    93                 md2 = (MessageDigest) md.clone(); // clone right after construction

    94                 byte[] digest = md.digest();

    95                 byte[] digest2 = md2.digest();

    96                 if (!Arrays.equals(digest, digest2)) {

    97                     System.out.println("DIFF-3.1 FAILED for: " + a);

    98                     testPassed = false;

    99                 }

   100                 md.update(msg);

   101                 md2 = (MessageDigest) md.clone(); // clone again after update call

   102                 digest = md.digest();

   103                 digest2 = md2.digest();

   104                 if (!Arrays.equals(digest, digest2)) {

   105                     System.out.println("DIFF-3.2 FAILED for: " + a);

   106                     testPassed = false;

   107                 }

   108                 md2 = (MessageDigest) md.clone(); // clone after digest

   109                 digest = md.digest();

   110                 digest2 = md2.digest();

   111                 if (!Arrays.equals(digest, digest2)) {

   112                     System.out.println("DIFF-3.3 FAILED for: " + a);

   113                     testPassed = false;

   114                 }

   115             } catch(Exception ex) {

   116                 System.out.println("Unexpected Exception: " + a);

   117                 ex.printStackTrace();

   118                 testPassed = false;

   119             }

   120         }

   121         if (!testPassed) {

   122             throw new RuntimeException("One or more MD test failed!");

   123         } else {

   124             System.out.println("MD Tests Passed");

   125         }

   126     }

   127 }