1/*
2 * Copyright (c) 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
24import java.io.ByteArrayInputStream;
25import java.io.ByteArrayOutputStream;
26import java.io.IOException;
27import java.security.InvalidAlgorithmParameterException;
28import java.security.InvalidKeyException;
29import java.security.NoSuchAlgorithmException;
30import java.security.NoSuchProviderException;
31import java.security.spec.AlgorithmParameterSpec;
32import java.util.Random;
33import javax.crypto.Cipher;
34import javax.crypto.CipherInputStream;
35import javax.crypto.CipherOutputStream;
36import javax.crypto.KeyGenerator;
37import javax.crypto.NoSuchPaddingException;
38import javax.crypto.SecretKey;
39import javax.crypto.spec.IvParameterSpec;
40
41/**
42 * @test
43 * @bug 8043836
44 * @summary Test AES ciphers with different modes and padding schemes (ECB mode
45 *          doesn't use IV). The test tries 3 different read methods of
46 *          CipherInputStream.
47 * @key randomness
48 */
49public class CICO {
50    private static final String ALGORITHM = "aEs";
51    private static final String[] MODES = { "PCBC", "ECb", "cbC", "cFB",
52        "cFB24", "cFB32", "Cfb40", "CFB72", "OfB", "OfB20", "OfB48",
53        "OfB56", "OFB64", "OFB112", "CFB112", "pCbC" };
54    private static final String[] PADDING = { "noPadding", "pkcs5padding" };
55    private static final String PROVIDER = "SunJCE";
56    private static final int NREADS = 3;
57    private static final int KEY_LENGTH = 128;
58
59    private final byte[] plainText = new byte[1600000];
60
61
62    public static void main(String argv[]) throws Exception {
63        CICO test = new CICO();
64        for (String mode : MODES) {
65            for (String pad : PADDING) {
66                for (int m = 0; m < NREADS; m++) {
67                    test.runTest(ALGORITHM, mode, pad, m);
68                }
69            }
70        }
71    }
72
73    public void runTest(String algo, String mo, String pad, int whichRead) throws Exception {
74        Cipher ci1 = null;
75        Cipher ci2 = null;
76        byte[] iv = null;
77        AlgorithmParameterSpec aps = null;
78        SecretKey key = null;
79
80        try {
81            // Do initialization
82            Random rdm = new Random();
83            rdm.nextBytes(plainText);
84            KeyGenerator kg = KeyGenerator.getInstance(algo, PROVIDER);
85            if (!kg.getAlgorithm().equals(algo)) {
86                throw new RuntimeException("Unexpected algorithm <"
87                        + kg.getAlgorithm() + ">, expected value is <" + algo
88                        + ">");
89            }
90
91            kg.init(KEY_LENGTH);
92            key = kg.generateKey();
93
94            ci1 = Cipher.getInstance(algo + "/" + mo + "/" + pad, PROVIDER);
95
96            if (mo.equalsIgnoreCase("ECB")) {
97                ci1.init(Cipher.ENCRYPT_MODE, key);
98            } else {
99                ci1.init(Cipher.ENCRYPT_MODE, key, aps);
100            }
101
102            if (!mo.equalsIgnoreCase("ECB")) {
103                iv = ci1.getIV();
104                aps = new IvParameterSpec(iv);
105            } else {
106                aps = null;
107            }
108
109            ci2 = Cipher.getInstance(algo + "/" + mo + "/" + pad, PROVIDER);
110            if (mo.equalsIgnoreCase("ECB")) {
111                ci2.init(Cipher.DECRYPT_MODE, key);
112            } else {
113                ci2.init(Cipher.DECRYPT_MODE, key, aps);
114            }
115
116            ByteArrayInputStream baInput = new ByteArrayInputStream(plainText);
117            ByteArrayOutputStream baOutput = new ByteArrayOutputStream();
118            try (CipherInputStream ciInput = new CipherInputStream(baInput, ci1);
119                    CipherOutputStream ciOutput = new CipherOutputStream(
120                            baOutput, ci2)) {
121                // According to specification, CipherInputStream does not support the
122                // mark and reset methods
123                if (ciInput.markSupported()) {
124                    throw new RuntimeException(
125                            "CipherInputStream unexpectedly supports the mark and reset methods");
126                }
127
128                // Read from the input and write to the output using 2 types
129                // of buffering : byte[] and int
130                switch (whichRead) {
131                case 0:
132                    int buffer0 = ciInput.read();
133                    while (buffer0 != -1) {
134                        ciOutput.write(buffer0);
135                        buffer0 = ciInput.read();
136                    }
137                    break;
138                case 1:
139                    byte[] buffer1 = new byte[20];
140                    int len1 = ciInput.read(buffer1);
141                    while (len1 != -1) {
142                        ciOutput.write(buffer1, 0, len1);
143                        len1 = ciInput.read(buffer1);
144                    }
145                    break;
146                case NREADS - 1:
147                    byte[] buffer2 = new byte[ci1
148                                              .getOutputSize(plainText.length)];
149                    int offset2 = 0;
150                    int len2 = 0;
151                    while (len2 != -1) {
152                        len2 = ciInput.read(buffer2, offset2, buffer2.length
153                                - offset2);
154                        offset2 += len2;
155                    }
156                    ciOutput.write(buffer2, 0, buffer2.length);
157                    break;
158                }
159            }
160
161            // Get the output
162            byte[] recoveredText = new byte[baOutput.size()];
163            recoveredText = baOutput.toByteArray();
164            if (!java.util.Arrays.equals(plainText, recoveredText)) {
165                throw new RuntimeException(
166                        "Original text is not equal with recovered text, with "
167                                + algo + "/" + mo + "/" + pad + "/" + whichRead);
168            }
169
170            // Compare input and output
171
172        } catch (NoSuchAlgorithmException e) {
173            //OFB20 is for negative testing
174            if (!mo.equalsIgnoreCase("OFB20")) {
175                System.out.println("Unexpected NoSuchAlgorithmException with "
176                        + algo + "/" + mo + "/" + pad + "/" + whichRead);
177                throw new RuntimeException("Test failed!");
178            }
179        } catch (IOException | NoSuchProviderException | NoSuchPaddingException
180                | InvalidKeyException | InvalidAlgorithmParameterException e) {
181            System.out.println("Unexpected Exception with "
182                    + algo + "/" + mo + "/" + pad + "/" + whichRead);
183            System.out.println("Test failed!");
184            throw e;
185        }
186    }
187}
188