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 /**
26 * @author Tom Deneau
27 */
28
29 import javax.crypto.Cipher;
30 import javax.crypto.KeyGenerator;
31 import javax.crypto.SecretKey;
32 import javax.crypto.spec.IvParameterSpec;
33 import javax.crypto.spec.SecretKeySpec;
34 import java.security.AlgorithmParameters;
35
36 import java.util.Random;
37 import java.util.Arrays;
38
39 abstract public class TestAESBase {
40 int msgSize = Integer.getInteger("msgSize", 646);
41 boolean checkOutput = Boolean.getBoolean("checkOutput");
42 boolean noReinit = Boolean.getBoolean("noReinit");
43 int keySize = Integer.getInteger("keySize", 128);
44 String algorithm = System.getProperty("algorithm", "AES");
45 String mode = System.getProperty("mode", "CBC");
46 byte[] input;
47 byte[] encode;
48 byte[] expectedEncode;
49 byte[] decode;
50 byte[] expectedDecode;
51 Random random = new Random(0);
52 Cipher cipher;
53 Cipher dCipher;
54 String paddingStr = "PKCS5Padding";
55 AlgorithmParameters algParams;
56 SecretKey key;
57 int ivLen;
58
59 static int numThreads = 0;
60 int threadId;
61 static synchronized int getThreadId() {
62 int id = numThreads;
63 numThreads++;
64 return id;
65 }
66
67 abstract public void run();
68
69 public void prepare() {
70 try {
71 System.out.println("\nmsgSize=" + msgSize + ", key size=" + keySize + ", reInit=" + !noReinit + ", checkOutput=" + checkOutput);
72
73 int keyLenBytes = (keySize == 0 ? 16 : keySize/8);
74 byte keyBytes[] = new byte[keyLenBytes];
75 if (keySize == 128)
76 keyBytes = new byte[] {-8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7};
77 else
78 random.nextBytes(keyBytes);
79
80 key = new SecretKeySpec(keyBytes, algorithm);
81 if (threadId == 0) {
82 System.out.println("Algorithm: " + key.getAlgorithm() + "("
83 + key.getEncoded().length * 8 + "bit)");
84 }
85 input = new byte[msgSize];
86 for (int i=0; i<input.length; i++) {
87 input[i] = (byte) (i & 0xff);
88 }
89
90 cipher = Cipher.getInstance(algorithm + "/" + mode + "/" + paddingStr, "SunJCE");
91 dCipher = Cipher.getInstance(algorithm + "/" + mode + "/" + paddingStr, "SunJCE");
92
93 ivLen = (algorithm.equals("AES") ? 16 : algorithm.equals("DES") ? 8 : 0);
94 IvParameterSpec initVector = new IvParameterSpec(new byte[ivLen]);
95
96 cipher.init(Cipher.ENCRYPT_MODE, key, initVector);
97 algParams = cipher.getParameters();
98 dCipher.init(Cipher.DECRYPT_MODE, key, algParams);
99 if (threadId == 0) {
100 childShowCipher();
101 }
102
103 // do one encode and decode in preparation
104 // this will also create the encode buffer and decode buffer
105 encode = cipher.doFinal(input);
106 decode = dCipher.doFinal(encode);
107 if (checkOutput) {
108 expectedEncode = (byte[]) encode.clone();
109 expectedDecode = (byte[]) decode.clone();
110 showArray(key.getEncoded() , "key: ");
111 showArray(input, "input: ");
112 showArray(encode, "encode: ");
113 showArray(decode, "decode: ");
114 }
115 }
116 catch (Exception e) {
117 e.printStackTrace();
118 System.exit(1);
119 }
120 }
121
122 void showArray(byte b[], String name) {
123 System.out.format("%s [%d]: ", name, b.length);
124 for (int i=0; i<Math.min(b.length, 32); i++) {
125 System.out.format("%02x ", b[i] & 0xff);
126 }
127 System.out.println();
128 }
129
130 void compareArrays(byte b[], byte exp[]) {
131 if (b.length != exp.length) {
132 System.out.format("different lengths for actual and expected output arrays\n");
133 showArray(b, "test: ");
134 showArray(exp, "exp : ");
135 System.exit(1);
136 }
137 for (int i=0; i< exp.length; i++) {
138 if (b[i] != exp[i]) {
139 System.out.format("output error at index %d: got %02x, expected %02x\n", i, b[i] & 0xff, exp[i] & 0xff);
140 showArray(b, "test: ");
141 showArray(exp, "exp : ");
142 System.exit(1);
143 }
144 }
145 }
146
147
148 void showCipher(Cipher c, String kind) {
149 System.out.println(kind + " cipher provider: " + cipher.getProvider());
150 System.out.println(kind + " cipher algorithm: " + cipher.getAlgorithm());
151 }
152
153 abstract void childShowCipher();
154 }