1 /*

     2  * Copyright 2003 Sun Microsystems, Inc.  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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

    20  * CA 95054 USA or visit www.sun.com if you need additional information or

    21  * have any questions.

    22  */

    23 

    24 /**

    25  * @test

    26  * @bug 4856966

    27  * @summary

    28  * @author Andreas Sterbenz

    29  * @library ..

    30  */

    31 

    32 import java.util.*;

    33 

    34 import java.security.*;

    35 

    36 public class ReinitDigest extends PKCS11Test {

    37 

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

    39         main(new ReinitDigest());

    40     }

    41 

    42     public void main(Provider p) throws Exception {

    43         if (p.getService("MessageDigest", "MD5") == null) {

    44             System.out.println("Provider does not support MD5, skipping");

    45             return;

    46         }

    47         Random r = new Random();

    48         byte[] data1 = new byte[10 * 1024];

    49         byte[] data2 = new byte[10 * 1024];

    50         r.nextBytes(data1);

    51         r.nextBytes(data2);

    52         MessageDigest md;

    53         md = MessageDigest.getInstance("MD5", "SUN");

    54         byte[] d1 = md.digest(data1);

    55         md = MessageDigest.getInstance("MD5", p);

    56         byte[] d2 = md.digest(data1);

    57         check(d1, d2);

    58         byte[] d3 = md.digest(data1);

    59         check(d1, d3);

    60         md.update(data2);

    61         md.update((byte)0);

    62         md.reset();

    63         byte[] d4 = md.digest(data1);

    64         check(d1, d4);

    65         System.out.println("All tests passed");

    66     }

    67 

    68     private static void check(byte[] d1, byte[] d2) throws Exception {

    69         if (Arrays.equals(d1, d2) == false) {

    70             throw new RuntimeException("Digest mismatch");

    71         }

    72     }

    73 }