1 /*

     2  * Copyright (c) 1998, 2014, 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 import java.security.InvalidKeyException;

    25 import java.security.NoSuchAlgorithmException;

    26 import java.security.NoSuchProviderException;

    27 import java.security.SecureRandom;

    28 import javax.crypto.Mac;

    29 import javax.crypto.spec.SecretKeySpec;

    30 

    31 /**

    32  * @test

    33  * @bug 8048603

    34  * @summary Check if doFinal and update operation result in same Mac

    35  * @author Yu-Ching Valerie Peng, Bill Situ, Alexander Fomin

    36  * @build Utils

    37  * @run main MacSameTest

    38  */

    39 public class MacSameTest implements MacTest {

    40 

    41     private static final int MESSAGE_SIZE = 25;

    42     private static final int OFFSET = 5;

    43     private static final int KEY_SIZE = 70;

    44 

    45     /**

    46      * Initialize a message, instantiate a Mac object,

    47      * initialize the object with a SecretKey,

    48      * feed the message into the Mac object

    49      * all at once and get the output MAC as result1.

    50      * Reset the Mac object, chop the message into three pieces,

    51      * feed into the Mac object sequentially, and get the output MAC as result2.

    52      * Finally, compare result1 and result2 and see if they are the same.

    53      *

    54      * @param args the command line arguments

    55      */

    56     public static void main(String[] args) {

    57         Utils.runTests(new MacSameTest());

    58     }

    59 

    60     @Override

    61     public void doTest(String algo) throws NoSuchAlgorithmException,

    62             NoSuchProviderException, InvalidKeyException {

    63         Mac mac;

    64         try {

    65             mac = Mac.getInstance(algo, "SunJCE");

    66         } catch (NoSuchAlgorithmException nsae) {

    67             // depending on Solaris configuration,

    68             // it can support HMAC or not with Mac

    69             System.out.println("Expected NoSuchAlgorithmException thrown: "

    70                     + nsae);

    71             return;

    72         }

    73 

    74         byte[] plain = new byte[MESSAGE_SIZE];

    75         for (int i = 0; i < MESSAGE_SIZE; i++) {

    76             plain[i] = (byte) (i % 256);

    77         }

    78 

    79         byte[] tail = new byte[plain.length - OFFSET];

    80         System.arraycopy(plain, OFFSET, tail, 0, tail.length);

    81 

    82         SecureRandom srdm = new SecureRandom();

    83         byte[] keyVal = new byte[KEY_SIZE];

    84         srdm.nextBytes(keyVal);

    85         SecretKeySpec keySpec = new SecretKeySpec(keyVal, "HMAC");

    86 

    87         mac.init(keySpec);

    88         byte[] result1 = mac.doFinal(plain);

    89 

    90         mac.reset();

    91         mac.update(plain[0]);

    92         mac.update(plain, 1, OFFSET - 1);

    93         byte[] result2 = mac.doFinal(tail);

    94 

    95         if (!java.util.Arrays.equals(result1, result2)) {

    96             throw new RuntimeException("result1 and result2 are not the same");

    97         }

    98     }

    99 

   100 }