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
24 /*
25 * @test
26 * @bug 8014374
27 * @summary Test basic CipherInputStream/OutputStream func w/ GCM mode.
28 * @author Valerie Peng
29 * @key randomness
30 */
31
32 import java.security.*;
33 import javax.crypto.*;
34 import javax.crypto.spec.*;
35 import java.math.*;
36 import java.io.*;
37 import com.sun.crypto.provider.*;
38
39 import java.util.*;
40
41 public class TestCICOWithGCM extends UcryptoTest {
42 public static void main(String[] args) throws Exception {
43 main(new TestCICOWithGCM(), null);
44 }
45
46 public void doTest(Provider p) throws Exception {
47 // check if GCM support exists
48 try {
49 Cipher.getInstance("AES/GCM/NoPadding", p);
50 } catch (NoSuchAlgorithmException nsae) {
51 System.out.println("Skipping Test due to no GCM support");
52 return;
53 }
54
55 Random rdm = new Random();
56
57 //init Secret Key
58 byte[] keyValue = new byte[16];
59 rdm.nextBytes(keyValue);
60 SecretKey key = new SecretKeySpec(keyValue, "AES");
61
62 //do initialization of the plainText
63 byte[] plainText = new byte[800];
64 rdm.nextBytes(plainText);
65
66 //init ciphers
67 Cipher encCipher = Cipher.getInstance("AES/GCM/NoPadding", p);
68 encCipher.init(Cipher.ENCRYPT_MODE, key);
69 Cipher decCipher = Cipher.getInstance("AES/GCM/NoPadding", p);
70 decCipher.init(Cipher.DECRYPT_MODE, key, encCipher.getParameters());
71
72 //init cipher streams
73 ByteArrayInputStream baInput = new ByteArrayInputStream(plainText);
74 CipherInputStream ciInput = new CipherInputStream(baInput, encCipher);
75 ByteArrayOutputStream baOutput = new ByteArrayOutputStream();
76 CipherOutputStream ciOutput = new CipherOutputStream(baOutput, decCipher);
77
78 //do test
79 byte[] buffer = new byte[800];
80 int len = ciInput.read(buffer);
81 System.out.println("read " + len + " bytes from input buffer");
82
83 while (len != -1) {
84 ciOutput.write(buffer, 0, len);
85 System.out.println("wite " + len + " bytes to output buffer");
86 len = ciInput.read(buffer);
87 if (len != -1) {
88 System.out.println("read " + len + " bytes from input buffer");
89 } else {
90 System.out.println("finished reading");
91 }
92 }
93
94 ciOutput.flush();
95 ciInput.close();
96 ciOutput.close();
97 byte[] recovered = baOutput.toByteArray();
98 System.out.println("recovered " + recovered.length + " bytes");
99 if (!Arrays.equals(plainText, recovered)) {
100 throw new RuntimeException("diff check failed!");
101 } else {
102 System.out.println("diff check passed");
103 }
104 }
105 }