1 /*
2  * Copyright (c) 2003, 2007, 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 4959235
27  * @summary Verify that ARCFOUR KeyGenerator generate keys according
28  * to the key length explicitly specified in the init() call.
29  *
30  * @author Valerie Peng
31  */
32 import java.security.*;
33 import javax.crypto.*;
34 import java.util.*;
35 
36 public class TestExplicitKeyLength {
37 
38     private static final String ALGOS[] = { "RC2", "ARCFOUR" };
39 
40     private static final int KEY_SIZES[] =
41         { 64, 80 }; // in bits
42 
runTest(String algo, int keysize)43     public static void runTest(String algo, int keysize) throws Exception {
44         KeyGenerator kg = KeyGenerator.getInstance(algo, "SunJCE");
45         kg.init(keysize);
46         Key generatedKey = kg.generateKey();
47         int actualSizeInBits = generatedKey.getEncoded().length*8;
48         if (actualSizeInBits != keysize) {
49             throw new Exception("generated key has wrong length: " +
50                                 actualSizeInBits + " bits");
51         }
52     }
53 
main(String[] args)54     public static void main (String[] args) throws Exception {
55         Security.addProvider(new com.sun.crypto.provider.SunJCE());
56         for (int i = 0; i < ALGOS.length; i++) {
57             System.out.println("Testing " + ALGOS[i] + " KeyGenerator with " +
58                                KEY_SIZES[i] + "-bit keysize");
59             runTest(ALGOS[i], KEY_SIZES[i]);
60         }
61         System.out.println("Tests Passed");
62     }
63 }
64